infosecn1nja
/
metasploit-framework
mirror of https://github.com/infosecn1nja/metasploit-framework.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'mephos-netstat' into rapid7

unstable
James Lee 2012-09-12 16:56:17 -05:00
parent 39f2cbfc3c 40b383e247
commit 823bc0a7f6
1 changed files with 51 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
external/source/meterpreter/source/extensions/stdapi/server/net/config/netstat.c vendored
View File

@ -644,11 +644,45 @@ DWORD linux_parse_proc_net_file(char * filename, struct connection_table ** tabl
DWORD linux_proc_get_program_name(struct connection_entry * connection, unsigned char * pid) DWORD linux_proc_get_program_name(struct connection_entry * connection, unsigned char * pid)
{ {
FILE *fd; FILE *fd;
char buffer[30], buffer_file[50], name[30]; char buffer[30], buffer_file[256], name[256];
char * bname;
snprintf(buffer, sizeof(buffer), "/proc/%s/status", pid); int do_status = 0;
do {
// try /proc/PID/cmdline first
snprintf(buffer, sizeof(buffer)-1, "/proc/%s/cmdline", pid);
fd = fopen(buffer, "r"); fd = fopen(buffer, "r");
// will try /proc/PID/status
if (fd == NULL) {
do_status = 1;
break;
}
if (fgets(buffer_file, sizeof(buffer_file), fd) == NULL) {
do_status = 1;
break;
}
// each entry in cmdline is seperated by '\0' so buffer_file contains first the path of the executable launched
if ((bname = basename(buffer_file)) == NULL) {
do_status = 1;
break;
}
// copy basename into name to be consistent at the end
strncpy(name, bname, sizeof(name)-1);
name[sizeof(name)-1] = '\0';
} while (0);
if (fd != NULL)
fclose(fd);
// /proc/PID/cmdline failed, try /proc/PID/status
if (do_status == 1) {
snprintf(buffer, sizeof(buffer)-1, "/proc/%s/status", pid);
fd = fopen(buffer, "r");
// will try /proc/PID/status
if (fd == NULL) if (fd == NULL)
return -1; return -1;
@ -657,13 +691,15 @@ DWORD linux_proc_get_program_name(struct connection_entry * connection, unsigned
return -1; return -1;
} }
if (sscanf(buffer_file, "Name: %s\n", name) != 1) { if (sscanf(buffer_file, "Name: %200s\n", name) != 1) {
fclose(fd); fclose(fd);
return -1; return -1;
} }
fclose(fd);
}
snprintf(connection->program_name, sizeof(connection->program_name), "%s/%s", pid, name); snprintf(connection->program_name, sizeof(connection->program_name), "%s/%s", pid, name);
fclose(fd);
return 0; return 0;
} }