Use clang-format to format (somewhat) LLVM-style (fixes #44)
This commit is contained in:
Arnout Engelen
654
cui.cpp
654
cui.cpp
@@ -1,4 +1,4 @@
|
||||
/*
|
||||
/*
|
||||
* cui.cpp
|
||||
*
|
||||
* Copyright (c) 2004-2006,2008,2010,2011 Arnout Engelen
|
||||
@@ -15,11 +15,11 @@
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
|
||||
*USA.
|
||||
*
|
||||
*/
|
||||
|
||||
|
||||
/* NetHogs console UI */
|
||||
#include <string>
|
||||
#include <pwd.h>
|
||||
@@ -33,14 +33,14 @@
|
||||
#include "nethogs.h"
|
||||
#include "process.h"
|
||||
|
||||
std::string * caption;
|
||||
std::string *caption;
|
||||
extern const char version[];
|
||||
extern ProcList * processes;
|
||||
extern ProcList *processes;
|
||||
extern timeval curtime;
|
||||
|
||||
extern Process * unknowntcp;
|
||||
extern Process * unknownudp;
|
||||
extern Process * unknownip;
|
||||
extern Process *unknowntcp;
|
||||
extern Process *unknownudp;
|
||||
extern Process *unknownip;
|
||||
|
||||
extern bool sortRecv;
|
||||
|
||||
@@ -51,414 +51,392 @@ extern unsigned refreshcount;
|
||||
|
||||
#define PID_MAX 4194303
|
||||
|
||||
const int COLUMN_WIDTH_PID = 7;
|
||||
const int COLUMN_WIDTH_USER = 8;
|
||||
const int COLUMN_WIDTH_DEV = 5;
|
||||
const int COLUMN_WIDTH_PID = 7;
|
||||
const int COLUMN_WIDTH_USER = 8;
|
||||
const int COLUMN_WIDTH_DEV = 5;
|
||||
const int COLUMN_WIDTH_SENT = 11;
|
||||
const int COLUMN_WIDTH_RECEIVED = 11;
|
||||
const int COLUMN_WIDTH_UNIT = 6;
|
||||
|
||||
const char * COLUMN_FORMAT_PID = "%7d";
|
||||
const char * COLUMN_FORMAT_SENT = "%11.3f";
|
||||
const char * COLUMN_FORMAT_RECEIVED = "%11.3f";
|
||||
const char *COLUMN_FORMAT_PID = "%7d";
|
||||
const char *COLUMN_FORMAT_SENT = "%11.3f";
|
||||
const char *COLUMN_FORMAT_RECEIVED = "%11.3f";
|
||||
|
||||
class Line
|
||||
{
|
||||
class Line {
|
||||
public:
|
||||
Line (const char * name, double n_recv_value, double n_sent_value, pid_t pid, uid_t uid, const char * n_devicename)
|
||||
{
|
||||
assert (pid >= 0);
|
||||
assert (pid <= PID_MAX);
|
||||
m_name = name;
|
||||
sent_value = n_sent_value;
|
||||
recv_value = n_recv_value;
|
||||
devicename = n_devicename;
|
||||
m_pid = pid;
|
||||
m_uid = uid;
|
||||
assert (m_pid >= 0);
|
||||
}
|
||||
Line(const char *name, double n_recv_value, double n_sent_value, pid_t pid,
|
||||
uid_t uid, const char *n_devicename) {
|
||||
assert(pid >= 0);
|
||||
assert(pid <= PID_MAX);
|
||||
m_name = name;
|
||||
sent_value = n_sent_value;
|
||||
recv_value = n_recv_value;
|
||||
devicename = n_devicename;
|
||||
m_pid = pid;
|
||||
m_uid = uid;
|
||||
assert(m_pid >= 0);
|
||||
}
|
||||
|
||||
void show (int row, unsigned int proglen);
|
||||
void log ();
|
||||
void show(int row, unsigned int proglen);
|
||||
void log();
|
||||
|
||||
double sent_value;
|
||||
double recv_value;
|
||||
|
||||
double sent_value;
|
||||
double recv_value;
|
||||
private:
|
||||
const char * m_name;
|
||||
const char * devicename;
|
||||
pid_t m_pid;
|
||||
uid_t m_uid;
|
||||
const char *m_name;
|
||||
const char *devicename;
|
||||
pid_t m_pid;
|
||||
uid_t m_uid;
|
||||
};
|
||||
|
||||
#include <sstream>
|
||||
|
||||
std::string itoa(int i)
|
||||
{
|
||||
std::stringstream out;
|
||||
out << i;
|
||||
return out.str();
|
||||
std::string itoa(int i) {
|
||||
std::stringstream out;
|
||||
out << i;
|
||||
return out.str();
|
||||
}
|
||||
|
||||
/**
|
||||
* @returns the username that corresponds to this uid
|
||||
* @returns the username that corresponds to this uid
|
||||
*/
|
||||
std::string uid2username (uid_t uid)
|
||||
{
|
||||
struct passwd * pwd = NULL;
|
||||
errno = 0;
|
||||
std::string uid2username(uid_t uid) {
|
||||
struct passwd *pwd = NULL;
|
||||
errno = 0;
|
||||
|
||||
/* points to a static memory area, should not be freed */
|
||||
pwd = getpwuid(uid);
|
||||
/* points to a static memory area, should not be freed */
|
||||
pwd = getpwuid(uid);
|
||||
|
||||
if (pwd == NULL)
|
||||
if (errno == 0)
|
||||
return itoa(uid);
|
||||
else
|
||||
forceExit(false, "Error calling getpwuid(3) for uid %d: %d %s", uid, errno, strerror(errno));
|
||||
else
|
||||
return std::string(pwd->pw_name);
|
||||
if (pwd == NULL)
|
||||
if (errno == 0)
|
||||
return itoa(uid);
|
||||
else
|
||||
forceExit(false, "Error calling getpwuid(3) for uid %d: %d %s", uid,
|
||||
errno, strerror(errno));
|
||||
else
|
||||
return std::string(pwd->pw_name);
|
||||
}
|
||||
|
||||
/**
|
||||
* Render the provided text at the specified location, truncating if the length of the text exceeds a maximum. If the
|
||||
* text must be truncated, the string ".." will be rendered, followed by max_len - 2 characters of the provided text.
|
||||
* Render the provided text at the specified location, truncating if the length
|
||||
* of the text exceeds a maximum. If the
|
||||
* text must be truncated, the string ".." will be rendered, followed by max_len
|
||||
* - 2 characters of the provided text.
|
||||
*/
|
||||
static void mvaddstr_truncate_leading(int row, int col, const char* str, std::size_t str_len, std::size_t max_len)
|
||||
{
|
||||
if (str_len < max_len) {
|
||||
mvaddstr(row, col, str);
|
||||
} else {
|
||||
mvaddstr(row, col, "..");
|
||||
addnstr(str + 2, max_len - 2);
|
||||
}
|
||||
static void mvaddstr_truncate_leading(int row, int col, const char *str,
|
||||
std::size_t str_len,
|
||||
std::size_t max_len) {
|
||||
if (str_len < max_len) {
|
||||
mvaddstr(row, col, str);
|
||||
} else {
|
||||
mvaddstr(row, col, "..");
|
||||
addnstr(str + 2, max_len - 2);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Render the provided text at the specified location, truncating if the length of the text exceeds a maximum. If the
|
||||
* text must be truncated, the text will be rendered up to max_len - 2 characters and then ".." will be rendered.
|
||||
* Render the provided text at the specified location, truncating if the length
|
||||
* of the text exceeds a maximum. If the
|
||||
* text must be truncated, the text will be rendered up to max_len - 2
|
||||
* characters and then ".." will be rendered.
|
||||
*/
|
||||
static void mvaddstr_truncate_trailing(int row, int col, const char* str, std::size_t str_len, std::size_t max_len)
|
||||
{
|
||||
if (str_len < max_len) {
|
||||
mvaddstr(row, col, str);
|
||||
} else {
|
||||
mvaddnstr(row, col, str, max_len - 2);
|
||||
addstr("..");
|
||||
}
|
||||
static void mvaddstr_truncate_trailing(int row, int col, const char *str,
|
||||
std::size_t str_len,
|
||||
std::size_t max_len) {
|
||||
if (str_len < max_len) {
|
||||
mvaddstr(row, col, str);
|
||||
} else {
|
||||
mvaddnstr(row, col, str, max_len - 2);
|
||||
addstr("..");
|
||||
}
|
||||
}
|
||||
|
||||
void Line::show (int row, unsigned int proglen)
|
||||
{
|
||||
assert (m_pid >= 0);
|
||||
assert (m_pid <= PID_MAX);
|
||||
void Line::show(int row, unsigned int proglen) {
|
||||
assert(m_pid >= 0);
|
||||
assert(m_pid <= PID_MAX);
|
||||
|
||||
const int column_offset_pid = 0;
|
||||
const int column_offset_user = column_offset_pid + COLUMN_WIDTH_PID + 1;
|
||||
const int column_offset_program = column_offset_user + COLUMN_WIDTH_USER + 1;
|
||||
const int column_offset_dev = column_offset_program + proglen + 2;
|
||||
const int column_offset_sent = column_offset_dev + COLUMN_WIDTH_DEV + 1;
|
||||
const int column_offset_received = column_offset_sent + COLUMN_WIDTH_SENT + 1;
|
||||
const int column_offset_unit = column_offset_received + COLUMN_WIDTH_RECEIVED + 1;
|
||||
const int column_offset_pid = 0;
|
||||
const int column_offset_user = column_offset_pid + COLUMN_WIDTH_PID + 1;
|
||||
const int column_offset_program = column_offset_user + COLUMN_WIDTH_USER + 1;
|
||||
const int column_offset_dev = column_offset_program + proglen + 2;
|
||||
const int column_offset_sent = column_offset_dev + COLUMN_WIDTH_DEV + 1;
|
||||
const int column_offset_received = column_offset_sent + COLUMN_WIDTH_SENT + 1;
|
||||
const int column_offset_unit =
|
||||
column_offset_received + COLUMN_WIDTH_RECEIVED + 1;
|
||||
|
||||
// PID column
|
||||
if (m_pid == 0)
|
||||
mvaddch (row, column_offset_pid + COLUMN_WIDTH_PID - 1, '?');
|
||||
else
|
||||
mvprintw (row, column_offset_pid, COLUMN_FORMAT_PID, m_pid);
|
||||
// PID column
|
||||
if (m_pid == 0)
|
||||
mvaddch(row, column_offset_pid + COLUMN_WIDTH_PID - 1, '?');
|
||||
else
|
||||
mvprintw(row, column_offset_pid, COLUMN_FORMAT_PID, m_pid);
|
||||
|
||||
std::string username = uid2username(m_uid);
|
||||
mvaddstr_truncate_trailing (row, column_offset_user, username.c_str(), username.size(), COLUMN_WIDTH_USER);
|
||||
std::string username = uid2username(m_uid);
|
||||
mvaddstr_truncate_trailing(row, column_offset_user, username.c_str(),
|
||||
username.size(), COLUMN_WIDTH_USER);
|
||||
|
||||
mvaddstr_truncate_leading (row, column_offset_program, m_name, strlen (m_name), proglen);
|
||||
mvaddstr_truncate_leading(row, column_offset_program, m_name, strlen(m_name),
|
||||
proglen);
|
||||
|
||||
mvaddstr (row, column_offset_dev, devicename);
|
||||
mvaddstr(row, column_offset_dev, devicename);
|
||||
|
||||
mvprintw (row, column_offset_sent, COLUMN_FORMAT_SENT, sent_value);
|
||||
mvprintw(row, column_offset_sent, COLUMN_FORMAT_SENT, sent_value);
|
||||
|
||||
mvprintw (row, column_offset_received, COLUMN_FORMAT_RECEIVED, recv_value);
|
||||
if (viewMode == VIEWMODE_KBPS)
|
||||
{
|
||||
mvaddstr (row, column_offset_unit, "KB/sec");
|
||||
}
|
||||
else if (viewMode == VIEWMODE_TOTAL_MB)
|
||||
{
|
||||
mvaddstr (row, column_offset_unit, "MB ");
|
||||
}
|
||||
else if (viewMode == VIEWMODE_TOTAL_KB)
|
||||
{
|
||||
mvaddstr (row, column_offset_unit, "KB ");
|
||||
}
|
||||
else if (viewMode == VIEWMODE_TOTAL_B)
|
||||
{
|
||||
mvaddstr (row, column_offset_unit, "B ");
|
||||
}
|
||||
mvprintw(row, column_offset_received, COLUMN_FORMAT_RECEIVED, recv_value);
|
||||
if (viewMode == VIEWMODE_KBPS) {
|
||||
mvaddstr(row, column_offset_unit, "KB/sec");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_MB) {
|
||||
mvaddstr(row, column_offset_unit, "MB ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_KB) {
|
||||
mvaddstr(row, column_offset_unit, "KB ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_B) {
|
||||
mvaddstr(row, column_offset_unit, "B ");
|
||||
}
|
||||
}
|
||||
|
||||
void Line::log() {
|
||||
std::cout << m_name << '/' << m_pid << '/' << m_uid << "\t" << sent_value << "\t" << recv_value << std::endl;
|
||||
std::cout << m_name << '/' << m_pid << '/' << m_uid << "\t" << sent_value
|
||||
<< "\t" << recv_value << std::endl;
|
||||
}
|
||||
|
||||
int GreatestFirst (const void * ma, const void * mb)
|
||||
{
|
||||
Line ** pa = (Line **)ma;
|
||||
Line ** pb = (Line **)mb;
|
||||
Line * a = *pa;
|
||||
Line * b = *pb;
|
||||
double aValue;
|
||||
if (sortRecv)
|
||||
{
|
||||
aValue = a->recv_value;
|
||||
}
|
||||
else
|
||||
{
|
||||
aValue = a->sent_value;
|
||||
}
|
||||
int GreatestFirst(const void *ma, const void *mb) {
|
||||
Line **pa = (Line **)ma;
|
||||
Line **pb = (Line **)mb;
|
||||
Line *a = *pa;
|
||||
Line *b = *pb;
|
||||
double aValue;
|
||||
if (sortRecv) {
|
||||
aValue = a->recv_value;
|
||||
} else {
|
||||
aValue = a->sent_value;
|
||||
}
|
||||
|
||||
double bValue;
|
||||
if (sortRecv)
|
||||
{
|
||||
bValue = b->recv_value;
|
||||
}
|
||||
else
|
||||
{
|
||||
bValue = b->sent_value;
|
||||
}
|
||||
double bValue;
|
||||
if (sortRecv) {
|
||||
bValue = b->recv_value;
|
||||
} else {
|
||||
bValue = b->sent_value;
|
||||
}
|
||||
|
||||
if (aValue > bValue)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
if (aValue == bValue)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
return 1;
|
||||
if (aValue > bValue) {
|
||||
return -1;
|
||||
}
|
||||
if (aValue == bValue) {
|
||||
return 0;
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
void init_ui ()
|
||||
{
|
||||
WINDOW * screen = initscr();
|
||||
raw();
|
||||
noecho();
|
||||
cbreak();
|
||||
nodelay(screen, TRUE);
|
||||
caption = new std::string ("NetHogs");
|
||||
caption->append(getVersion());
|
||||
//caption->append(", running at ");
|
||||
void init_ui() {
|
||||
WINDOW *screen = initscr();
|
||||
raw();
|
||||
noecho();
|
||||
cbreak();
|
||||
nodelay(screen, TRUE);
|
||||
caption = new std::string("NetHogs");
|
||||
caption->append(getVersion());
|
||||
// caption->append(", running at ");
|
||||
}
|
||||
|
||||
void exit_ui ()
|
||||
{
|
||||
clear();
|
||||
endwin();
|
||||
delete caption;
|
||||
void exit_ui() {
|
||||
clear();
|
||||
endwin();
|
||||
delete caption;
|
||||
}
|
||||
|
||||
void ui_tick ()
|
||||
{
|
||||
switch (getch()) {
|
||||
case 'q':
|
||||
/* quit */
|
||||
quit_cb(0);
|
||||
break;
|
||||
case 's':
|
||||
/* sort on 'sent' */
|
||||
sortRecv = false;
|
||||
break;
|
||||
case 'r':
|
||||
/* sort on 'received' */
|
||||
sortRecv = true;
|
||||
break;
|
||||
case 'm':
|
||||
/* switch mode: total vs kb/s */
|
||||
viewMode = (viewMode + 1) % VIEWMODE_COUNT;
|
||||
break;
|
||||
}
|
||||
void ui_tick() {
|
||||
switch (getch()) {
|
||||
case 'q':
|
||||
/* quit */
|
||||
quit_cb(0);
|
||||
break;
|
||||
case 's':
|
||||
/* sort on 'sent' */
|
||||
sortRecv = false;
|
||||
break;
|
||||
case 'r':
|
||||
/* sort on 'received' */
|
||||
sortRecv = true;
|
||||
break;
|
||||
case 'm':
|
||||
/* switch mode: total vs kb/s */
|
||||
viewMode = (viewMode + 1) % VIEWMODE_COUNT;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void show_trace(Line * lines[], int nproc) {
|
||||
std::cout << "\nRefreshing:\n";
|
||||
void show_trace(Line *lines[], int nproc) {
|
||||
std::cout << "\nRefreshing:\n";
|
||||
|
||||
/* print them */
|
||||
for (int i=0; i<nproc; i++)
|
||||
{
|
||||
lines[i]->log();
|
||||
delete lines[i];
|
||||
}
|
||||
/* print them */
|
||||
for (int i = 0; i < nproc; i++) {
|
||||
lines[i]->log();
|
||||
delete lines[i];
|
||||
}
|
||||
|
||||
/* print the 'unknown' connections, for debugging */
|
||||
ConnList * curr_unknownconn = unknowntcp->connections;
|
||||
while (curr_unknownconn != NULL) {
|
||||
std::cout << "Unknown connection: " <<
|
||||
curr_unknownconn->getVal()->refpacket->gethashstring() << std::endl;
|
||||
/* print the 'unknown' connections, for debugging */
|
||||
ConnList *curr_unknownconn = unknowntcp->connections;
|
||||
while (curr_unknownconn != NULL) {
|
||||
std::cout << "Unknown connection: "
|
||||
<< curr_unknownconn->getVal()->refpacket->gethashstring()
|
||||
<< std::endl;
|
||||
|
||||
curr_unknownconn = curr_unknownconn->getNext();
|
||||
}
|
||||
curr_unknownconn = curr_unknownconn->getNext();
|
||||
}
|
||||
}
|
||||
|
||||
void show_ncurses(Line * lines[], int nproc) {
|
||||
int rows; // number of terminal rows
|
||||
int cols; // number of terminal columns
|
||||
unsigned int proglen; // max length of the "PROGRAM" column
|
||||
void show_ncurses(Line *lines[], int nproc) {
|
||||
int rows; // number of terminal rows
|
||||
int cols; // number of terminal columns
|
||||
unsigned int proglen; // max length of the "PROGRAM" column
|
||||
|
||||
double sent_global = 0;
|
||||
double recv_global = 0;
|
||||
double sent_global = 0;
|
||||
double recv_global = 0;
|
||||
|
||||
getmaxyx(stdscr, rows, cols); /* find the boundaries of the screeen */
|
||||
getmaxyx(stdscr, rows, cols); /* find the boundaries of the screeen */
|
||||
|
||||
if (cols < 62) {
|
||||
clear();
|
||||
mvprintw(0,0, "The terminal is too narrow! Please make it wider.\nI'll wait...");
|
||||
return;
|
||||
}
|
||||
if (cols < 62) {
|
||||
clear();
|
||||
mvprintw(0, 0,
|
||||
"The terminal is too narrow! Please make it wider.\nI'll wait...");
|
||||
return;
|
||||
}
|
||||
|
||||
if (cols > PROGNAME_WIDTH) cols = PROGNAME_WIDTH;
|
||||
if (cols > PROGNAME_WIDTH)
|
||||
cols = PROGNAME_WIDTH;
|
||||
|
||||
proglen = cols - 55;
|
||||
proglen = cols - 55;
|
||||
|
||||
clear();
|
||||
mvprintw (0, 0, "%s", caption->c_str());
|
||||
attron(A_REVERSE);
|
||||
mvprintw (2, 0, " PID USER %-*.*s DEV SENT RECEIVED ", proglen, proglen, "PROGRAM");
|
||||
attroff(A_REVERSE);
|
||||
clear();
|
||||
mvprintw(0, 0, "%s", caption->c_str());
|
||||
attron(A_REVERSE);
|
||||
mvprintw(2, 0,
|
||||
" PID USER %-*.*s DEV SENT RECEIVED ",
|
||||
proglen, proglen, "PROGRAM");
|
||||
attroff(A_REVERSE);
|
||||
|
||||
/* print them */
|
||||
int i;
|
||||
for (i=0; i<nproc; i++)
|
||||
{
|
||||
if (i+3 < rows)
|
||||
lines[i]->show(i+3, proglen);
|
||||
recv_global += lines[i]->recv_value;
|
||||
sent_global += lines[i]->sent_value;
|
||||
delete lines[i];
|
||||
}
|
||||
/* print them */
|
||||
int i;
|
||||
for (i = 0; i < nproc; i++) {
|
||||
if (i + 3 < rows)
|
||||
lines[i]->show(i + 3, proglen);
|
||||
recv_global += lines[i]->recv_value;
|
||||
sent_global += lines[i]->sent_value;
|
||||
delete lines[i];
|
||||
}
|
||||
|
||||
attron(A_REVERSE);
|
||||
int totalrow = std::min(rows-1, 3+1+i);
|
||||
mvprintw (totalrow, 0, " TOTAL %-*.*s %11.3f %11.3f ", proglen, proglen, " ", sent_global, recv_global);
|
||||
if (viewMode == VIEWMODE_KBPS)
|
||||
{
|
||||
mvprintw (3+1+i, cols - COLUMN_WIDTH_UNIT, "KB/sec ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_B) {
|
||||
mvprintw (3+1+i, cols - COLUMN_WIDTH_UNIT, "B ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_KB) {
|
||||
mvprintw (3+1+i, cols - COLUMN_WIDTH_UNIT, "KB ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_MB) {
|
||||
mvprintw (3+1+i, cols - COLUMN_WIDTH_UNIT, "MB ");
|
||||
}
|
||||
attroff(A_REVERSE);
|
||||
mvprintw (totalrow+1, 0, "");
|
||||
refresh();
|
||||
attron(A_REVERSE);
|
||||
int totalrow = std::min(rows - 1, 3 + 1 + i);
|
||||
mvprintw(totalrow, 0, " TOTAL %-*.*s %11.3f %11.3f ",
|
||||
proglen, proglen, " ", sent_global, recv_global);
|
||||
if (viewMode == VIEWMODE_KBPS) {
|
||||
mvprintw(3 + 1 + i, cols - COLUMN_WIDTH_UNIT, "KB/sec ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_B) {
|
||||
mvprintw(3 + 1 + i, cols - COLUMN_WIDTH_UNIT, "B ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_KB) {
|
||||
mvprintw(3 + 1 + i, cols - COLUMN_WIDTH_UNIT, "KB ");
|
||||
} else if (viewMode == VIEWMODE_TOTAL_MB) {
|
||||
mvprintw(3 + 1 + i, cols - COLUMN_WIDTH_UNIT, "MB ");
|
||||
}
|
||||
attroff(A_REVERSE);
|
||||
mvprintw(totalrow + 1, 0, "");
|
||||
refresh();
|
||||
}
|
||||
|
||||
// Display all processes and relevant network traffic using show function
|
||||
void do_refresh()
|
||||
{
|
||||
refreshconninode();
|
||||
refreshcount++;
|
||||
void do_refresh() {
|
||||
refreshconninode();
|
||||
refreshcount++;
|
||||
|
||||
ProcList * curproc = processes;
|
||||
ProcList * previousproc = NULL;
|
||||
int nproc = processes->size();
|
||||
/* initialise to null pointers */
|
||||
Line * lines [nproc];
|
||||
int n = 0;
|
||||
ProcList *curproc = processes;
|
||||
ProcList *previousproc = NULL;
|
||||
int nproc = processes->size();
|
||||
/* initialise to null pointers */
|
||||
Line *lines[nproc];
|
||||
int n = 0;
|
||||
|
||||
#ifndef NDEBUG
|
||||
// initialise to null pointers
|
||||
for (int i = 0; i < nproc; i++)
|
||||
lines[i] = NULL;
|
||||
// initialise to null pointers
|
||||
for (int i = 0; i < nproc; i++)
|
||||
lines[i] = NULL;
|
||||
#endif
|
||||
|
||||
while (curproc != NULL)
|
||||
{
|
||||
// walk though its connections, summing up their data, and
|
||||
// throwing away connections that haven't received a package
|
||||
// in the last PROCESSTIMEOUT seconds.
|
||||
assert (curproc != NULL);
|
||||
assert (curproc->getVal() != NULL);
|
||||
assert (nproc == processes->size());
|
||||
while (curproc != NULL) {
|
||||
// walk though its connections, summing up their data, and
|
||||
// throwing away connections that haven't received a package
|
||||
// in the last PROCESSTIMEOUT seconds.
|
||||
assert(curproc != NULL);
|
||||
assert(curproc->getVal() != NULL);
|
||||
assert(nproc == processes->size());
|
||||
|
||||
/* remove timed-out processes (unless it's one of the the unknown process) */
|
||||
if ((curproc->getVal()->getLastPacket() + PROCESSTIMEOUT <= curtime.tv_sec)
|
||||
&& (curproc->getVal() != unknowntcp)
|
||||
&& (curproc->getVal() != unknownudp)
|
||||
&& (curproc->getVal() != unknownip))
|
||||
{
|
||||
if (DEBUG)
|
||||
std::cout << "PROC: Deleting process\n";
|
||||
ProcList * todelete = curproc;
|
||||
Process * p_todelete = curproc->getVal();
|
||||
if (previousproc)
|
||||
{
|
||||
previousproc->next = curproc->next;
|
||||
curproc = curproc->next;
|
||||
} else {
|
||||
processes = curproc->getNext();
|
||||
curproc = processes;
|
||||
}
|
||||
delete todelete;
|
||||
delete p_todelete;
|
||||
nproc--;
|
||||
//continue;
|
||||
}
|
||||
else
|
||||
{
|
||||
// add a non-timed-out process to the list of stuff to show
|
||||
float value_sent = 0,
|
||||
value_recv = 0;
|
||||
/* remove timed-out processes (unless it's one of the the unknown process)
|
||||
*/
|
||||
if ((curproc->getVal()->getLastPacket() + PROCESSTIMEOUT <=
|
||||
curtime.tv_sec) &&
|
||||
(curproc->getVal() != unknowntcp) &&
|
||||
(curproc->getVal() != unknownudp) && (curproc->getVal() != unknownip)) {
|
||||
if (DEBUG)
|
||||
std::cout << "PROC: Deleting process\n";
|
||||
ProcList *todelete = curproc;
|
||||
Process *p_todelete = curproc->getVal();
|
||||
if (previousproc) {
|
||||
previousproc->next = curproc->next;
|
||||
curproc = curproc->next;
|
||||
} else {
|
||||
processes = curproc->getNext();
|
||||
curproc = processes;
|
||||
}
|
||||
delete todelete;
|
||||
delete p_todelete;
|
||||
nproc--;
|
||||
// continue;
|
||||
} else {
|
||||
// add a non-timed-out process to the list of stuff to show
|
||||
float value_sent = 0, value_recv = 0;
|
||||
|
||||
if (viewMode == VIEWMODE_KBPS)
|
||||
{
|
||||
//std::cout << "kbps viemode" << std::endl;
|
||||
curproc->getVal()->getkbps (&value_recv, &value_sent);
|
||||
}
|
||||
else if (viewMode == VIEWMODE_TOTAL_KB)
|
||||
{
|
||||
//std::cout << "total viemode" << std::endl;
|
||||
curproc->getVal()->gettotalkb(&value_recv, &value_sent);
|
||||
}
|
||||
else if (viewMode == VIEWMODE_TOTAL_MB)
|
||||
{
|
||||
//std::cout << "total viemode" << std::endl;
|
||||
curproc->getVal()->gettotalmb(&value_recv, &value_sent);
|
||||
}
|
||||
else if (viewMode == VIEWMODE_TOTAL_B)
|
||||
{
|
||||
//std::cout << "total viemode" << std::endl;
|
||||
curproc->getVal()->gettotalb(&value_recv, &value_sent);
|
||||
}
|
||||
else
|
||||
{
|
||||
forceExit(false, "Invalid viewMode: %d", viewMode);
|
||||
}
|
||||
uid_t uid = curproc->getVal()->getUid();
|
||||
assert (curproc->getVal()->pid >= 0);
|
||||
assert (n < nproc);
|
||||
if (viewMode == VIEWMODE_KBPS) {
|
||||
// std::cout << "kbps viemode" << std::endl;
|
||||
curproc->getVal()->getkbps(&value_recv, &value_sent);
|
||||
} else if (viewMode == VIEWMODE_TOTAL_KB) {
|
||||
// std::cout << "total viemode" << std::endl;
|
||||
curproc->getVal()->gettotalkb(&value_recv, &value_sent);
|
||||
} else if (viewMode == VIEWMODE_TOTAL_MB) {
|
||||
// std::cout << "total viemode" << std::endl;
|
||||
curproc->getVal()->gettotalmb(&value_recv, &value_sent);
|
||||
} else if (viewMode == VIEWMODE_TOTAL_B) {
|
||||
// std::cout << "total viemode" << std::endl;
|
||||
curproc->getVal()->gettotalb(&value_recv, &value_sent);
|
||||
} else {
|
||||
forceExit(false, "Invalid viewMode: %d", viewMode);
|
||||
}
|
||||
uid_t uid = curproc->getVal()->getUid();
|
||||
assert(curproc->getVal()->pid >= 0);
|
||||
assert(n < nproc);
|
||||
|
||||
lines[n] = new Line (curproc->getVal()->name, value_recv, value_sent,
|
||||
curproc->getVal()->pid, uid, curproc->getVal()->devicename);
|
||||
previousproc = curproc;
|
||||
curproc = curproc->next;
|
||||
n++;
|
||||
lines[n] =
|
||||
new Line(curproc->getVal()->name, value_recv, value_sent,
|
||||
curproc->getVal()->pid, uid, curproc->getVal()->devicename);
|
||||
previousproc = curproc;
|
||||
curproc = curproc->next;
|
||||
n++;
|
||||
#ifndef NDEBUG
|
||||
assert (nproc == processes->size());
|
||||
if (curproc == NULL)
|
||||
assert (n-1 < nproc);
|
||||
else
|
||||
assert (n < nproc);
|
||||
assert(nproc == processes->size());
|
||||
if (curproc == NULL)
|
||||
assert(n - 1 < nproc);
|
||||
else
|
||||
assert(n < nproc);
|
||||
#endif
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* sort the accumulated lines */
|
||||
qsort (lines, nproc, sizeof(Line *), GreatestFirst);
|
||||
/* sort the accumulated lines */
|
||||
qsort(lines, nproc, sizeof(Line *), GreatestFirst);
|
||||
|
||||
if (tracemode || DEBUG)
|
||||
show_trace(lines, nproc);
|
||||
else
|
||||
show_ncurses(lines, nproc);
|
||||
if (tracemode || DEBUG)
|
||||
show_trace(lines, nproc);
|
||||
else
|
||||
show_ncurses(lines, nproc);
|
||||
|
||||
if (refreshlimit != 0 && refreshcount >= refreshlimit)
|
||||
quit_cb(0);
|
||||
if (refreshlimit != 0 && refreshcount >= refreshlimit)
|
||||
quit_cb(0);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user