htop/OpenFilesScreen.c

246 lines
6.8 KiB
C
Raw Normal View History

2009-06-02 19:28:19 +00:00
/*
htop - OpenFilesScreen.c
(C) 2005-2006 Hisham H. Muhammad
Released under the GNU GPL, see the COPYING file
in the source distribution for its full text.
*/
2011-12-26 21:35:57 +00:00
#include "OpenFilesScreen.h"
#include "CRT.h"
#include "ProcessList.h"
#include "ListItem.h"
#include "IncSet.h"
#include "String.h"
2011-12-26 21:35:57 +00:00
#include <string.h>
2009-06-02 19:28:19 +00:00
#include <stdio.h>
#include <unistd.h>
#include <stdbool.h>
#include <unistd.h>
2011-12-26 21:35:57 +00:00
#include <stdlib.h>
2009-06-02 19:28:19 +00:00
#include <fcntl.h>
#include <sys/types.h>
#include <sys/wait.h>
2011-12-26 21:35:57 +00:00
/*{
2009-06-02 19:28:19 +00:00
#include "Process.h"
#include "Panel.h"
#include "FunctionBar.h"
typedef struct OpenFiles_Data_ {
char* data[256];
} OpenFiles_Data;
2009-06-02 19:28:19 +00:00
typedef struct OpenFiles_ProcessData_ {
OpenFiles_Data data;
2009-10-16 20:15:01 +00:00
int error;
struct OpenFiles_FileData_* files;
2009-06-02 19:28:19 +00:00
} OpenFiles_ProcessData;
typedef struct OpenFiles_FileData_ {
OpenFiles_Data data;
2009-06-02 19:28:19 +00:00
struct OpenFiles_FileData_* next;
} OpenFiles_FileData;
typedef struct OpenFilesScreen_ {
Process* process;
pid_t pid;
2009-06-02 19:28:19 +00:00
Panel* display;
FunctionBar* bar;
} OpenFilesScreen;
}*/
static const char* ofsFunctions[] = {"Search ", "Filter ", "Refresh", "Done ", NULL};
2009-06-02 19:28:19 +00:00
static const char* ofsKeys[] = {"F3", "F4", "F5", "Esc"};
2009-06-02 19:28:19 +00:00
static int ofsEvents[] = {KEY_F(3), KEY_F(4), KEY_F(5), 27};
2009-06-02 19:28:19 +00:00
OpenFilesScreen* OpenFilesScreen_new(Process* process) {
OpenFilesScreen* this = (OpenFilesScreen*) malloc(sizeof(OpenFilesScreen));
this->process = process;
this->display = Panel_new(0, 1, COLS, LINES-3, false, Class(ListItem));
if (Process_isThread(process))
this->pid = process->tgid;
else
this->pid = process->pid;
2009-06-02 19:28:19 +00:00
return this;
}
void OpenFilesScreen_delete(OpenFilesScreen* this) {
Panel_delete((Object*)this->display);
free(this);
}
static void OpenFilesScreen_draw(OpenFilesScreen* this, IncSet* inc) {
2009-06-02 19:28:19 +00:00
attrset(CRT_colors[METER_TEXT]);
mvhline(0, 0, ' ', COLS);
mvprintw(0, 0, "Snapshot of files open in process %d - %s", this->pid, this->process->comm);
2009-06-02 19:28:19 +00:00
attrset(CRT_colors[DEFAULT_COLOR]);
Panel_draw(this->display, true);
IncSet_drawBar(inc);
2009-06-02 19:28:19 +00:00
}
static OpenFiles_ProcessData* OpenFilesScreen_getProcessData(pid_t pid) {
2009-06-02 19:28:19 +00:00
char command[1025];
2010-09-02 20:26:16 +00:00
snprintf(command, 1024, "lsof -P -p %d -F 2> /dev/null", pid);
2009-06-02 19:28:19 +00:00
FILE* fd = popen(command, "r");
OpenFiles_ProcessData* pdata = calloc(1, sizeof(OpenFiles_ProcessData));
OpenFiles_FileData* fdata = NULL;
OpenFiles_Data* item = &(pdata->data);
2009-10-16 20:15:01 +00:00
if (!fd) {
pdata->error = 127;
return pdata;
2009-10-16 20:15:01 +00:00
}
2009-06-02 19:28:19 +00:00
while (!feof(fd)) {
int cmd = fgetc(fd);
if (cmd == EOF)
2009-06-02 19:28:19 +00:00
break;
char* entry = malloc(1024);
2010-03-03 21:10:51 +00:00
if (!fgets(entry, 1024, fd)) {
free(entry);
break;
}
2009-06-02 19:28:19 +00:00
char* newline = strrchr(entry, '\n');
*newline = '\0';
if (cmd == 'f') {
OpenFiles_FileData* nextFile = calloc(1, sizeof(OpenFiles_FileData));
if (fdata == NULL) {
pdata->files = nextFile;
2009-06-02 19:28:19 +00:00
} else {
fdata->next = nextFile;
2009-06-02 19:28:19 +00:00
}
fdata = nextFile;
item = &(fdata->data);
2009-06-02 19:28:19 +00:00
}
item->data[cmd] = entry;
}
pdata->error = pclose(fd);
return pdata;
2009-06-02 19:28:19 +00:00
}
static inline void addLine(const char* line, Vector* lines, Panel* panel, const char* incFilter) {
Vector_add(lines, (Object*) ListItem_new(line, 0));
if (!incFilter || String_contains_i(line, incFilter))
Panel_add(panel, (Object*)Vector_get(lines, Vector_size(lines)-1));
}
static inline void OpenFiles_Data_clear(OpenFiles_Data* data) {
for (int i = 0; i < 255; i++)
if (data->data[i])
free(data->data[i]);
}
static void OpenFilesScreen_scan(OpenFilesScreen* this, Vector* lines, IncSet* inc) {
2009-06-02 19:28:19 +00:00
Panel* panel = this->display;
int idx = Panel_getSelectedIndex(panel);
2009-06-02 19:28:19 +00:00
Panel_prune(panel);
OpenFiles_ProcessData* pdata = OpenFilesScreen_getProcessData(this->pid);
if (pdata->error == 127) {
addLine("Could not execute 'lsof'. Please make sure it is available in your $PATH.", lines, panel, IncSet_filter(inc));
} else if (pdata->error == 1) {
addLine("Failed listing open files.", lines, panel, IncSet_filter(inc));
2009-06-02 19:28:19 +00:00
} else {
OpenFiles_FileData* fdata = pdata->files;
while (fdata) {
2009-06-02 19:28:19 +00:00
char entry[1024];
char** data = fdata->data.data;
2009-06-02 19:28:19 +00:00
sprintf(entry, "%5s %4s %10s %10s %10s %s",
data['f'] ? data['f'] : "",
data['t'] ? data['t'] : "",
data['D'] ? data['D'] : "",
data['s'] ? data['s'] : "",
data['i'] ? data['i'] : "",
data['n'] ? data['n'] : "");
addLine(entry, lines, panel, IncSet_filter(inc));
OpenFiles_Data_clear(&fdata->data);
OpenFiles_FileData* old = fdata;
fdata = fdata->next;
2009-06-02 19:28:19 +00:00
free(old);
}
OpenFiles_Data_clear(&pdata->data);
2009-06-02 19:28:19 +00:00
}
free(pdata);
Vector_insertionSort(lines);
Vector_insertionSort(panel->items);
2010-02-25 01:43:18 +00:00
Panel_setSelected(panel, idx);
2009-06-02 19:28:19 +00:00
}
void OpenFilesScreen_run(OpenFilesScreen* this) {
Panel* panel = this->display;
Panel_setHeader(panel, " FD TYPE DEVICE SIZE NODE NAME");
FunctionBar* bar = FunctionBar_new(ofsFunctions, ofsKeys, ofsEvents);
IncSet* inc = IncSet_new(bar);
Vector* lines = Vector_new(panel->items->type, true, DEFAULT_SIZE);
OpenFilesScreen_scan(this, lines, inc);
OpenFilesScreen_draw(this, inc);
2009-06-02 19:28:19 +00:00
bool looping = true;
while (looping) {
2009-06-02 19:28:19 +00:00
Panel_draw(panel, true);
if (inc->active)
move(LINES-1, CRT_cursorX);
2009-06-02 19:28:19 +00:00
int ch = getch();
2009-06-02 19:28:19 +00:00
if (ch == KEY_MOUSE) {
MEVENT mevent;
int ok = getmouse(&mevent);
if (ok == OK)
if (mevent.y >= panel->y && mevent.y < LINES - 1) {
Panel_setSelected(panel, mevent.y - panel->y + panel->scrollV);
ch = 0;
} if (mevent.y == LINES - 1)
ch = FunctionBar_synthesizeEvent(inc->bar, mevent.x);
2009-06-02 19:28:19 +00:00
}
if (inc->active) {
IncSet_handleKey(inc, ch, panel, IncSet_getListItemValue, lines);
continue;
}
2009-06-02 19:28:19 +00:00
switch(ch) {
case ERR:
continue;
case KEY_F(3):
case '/':
IncSet_activate(inc, INC_SEARCH);
break;
case KEY_F(4):
case '\\':
IncSet_activate(inc, INC_FILTER);
break;
2009-06-02 19:28:19 +00:00
case KEY_F(5):
clear();
OpenFilesScreen_scan(this, lines, inc);
OpenFilesScreen_draw(this, inc);
2009-06-02 19:28:19 +00:00
break;
case '\014': // Ctrl+L
clear();
OpenFilesScreen_draw(this, inc);
2009-06-02 19:28:19 +00:00
break;
case 'q':
case 27:
case KEY_F(10):
2009-06-02 19:28:19 +00:00
looping = false;
break;
case KEY_RESIZE:
Panel_resize(panel, COLS, LINES-2);
OpenFilesScreen_draw(this, inc);
2009-06-02 19:28:19 +00:00
break;
default:
Panel_onKey(panel, ch);
}
}
Vector_delete(lines);
FunctionBar_delete((Object*)bar);
IncSet_delete(inc);
2009-06-02 19:28:19 +00:00
}