diff options
| -rw-r--r-- | toys/pending/lsof.c | 415 |
1 files changed, 415 insertions, 0 deletions
diff --git a/toys/pending/lsof.c b/toys/pending/lsof.c new file mode 100644 index 00000000..f8088b72 --- /dev/null +++ b/toys/pending/lsof.c @@ -0,0 +1,415 @@ +/* lsof.c - list open files. + * + * Copyright 2015 The Android Open Source Project + +USE_LSOF(NEWTOY(lsof, "p#", TOYFLAG_USR|TOYFLAG_BIN)) + +config LSOF + bool "lsof" + default n + help + usage: lsof + + Lists open files. + + -p for given pid only (default all pids) +*/ + +#define FOR_lsof +#include "toys.h" + +GLOBALS( + int pid; + + struct double_list *head; + int shown_header; +) + +struct proc_info { + char cmd[10]; + int pid; + char user[12]; +}; + +struct file_info { + char *next, *prev; + + struct proc_info pi; + char fd[8]; + char rw; + char locks; + char type[10]; + char device[32]; + char size_off[32]; + char node[32]; + char* name; +}; + +static void print_header() +{ + // TODO: llist_traverse to measure the columns first. + char* names[] = { + "COMMAND", "PID", "USER", "FD", "TYPE", "DEVICE", "SIZE/OFF", "NODE", "NAME" + }; + printf("%-9s %5s %10s %4s %7s %18s %9s %10s %s\n", names[0], names[1], + names[2], names[3], names[4], names[5], names[6], names[7], names[8]); + + TT.shown_header = 1; +} + +static void print_info(void *data) +{ + struct file_info *fi = data; + + if (!TT.shown_header) print_header(); + + printf("%-9s %5d %10s %4s%c%c %7s %18s %9s %10s %s\n", + fi->pi.cmd, fi->pi.pid, fi->pi.user, + fi->fd, fi->rw, fi->locks, fi->type, fi->device, fi->size_off, + fi->node, fi->name); + + if (CFG_FREE) { + free(((struct file_info *)data)->name); + free(data); + } +} + +static void fill_flags(struct file_info *fi) +{ + FILE* fp; + long long pos; + unsigned flags; + + snprintf(toybuf, sizeof(toybuf), "/proc/%d/fdinfo/%s", fi->pi.pid, fi->fd); + fp = fopen(toybuf, "r"); + if (!fp) return; + + if (fscanf(fp, "pos: %lld flags: %o", &pos, &flags) == 2) { + flags &= O_ACCMODE; + if (flags == O_RDONLY) fi->rw = 'r'; + else if (flags == O_WRONLY) fi->rw = 'w'; + else fi->rw = 'u'; + + snprintf(fi->size_off, sizeof(fi->size_off), "0t%lld", pos); + } + fclose(fp); +} + +static char *chomp(char *s) +{ + char *p = strrchr(s, '\n'); + + if (p) *p = 0; + return s; +} + +static int find_unix_socket(struct file_info *fi, long sought_inode) +{ + FILE *fp = fopen("/proc/net/unix", "r"); + char *line = NULL; + size_t line_length = 0; + + if (!fp) return 0; + + if (!getline(&line, &line_length, fp)) return 0; // Skip header. + + while (getline(&line, &line_length, fp) > 0) { + long inode; + int path_pos; + + if (sscanf(line, "%*p: %*X %*X %*X %*X %*X %lu %n", + &inode, &path_pos) >= 1) { + if (inode == sought_inode) { + char *name = chomp(line + path_pos); + + strcpy(fi->type, "unix"); + fi->name = strdup(*name ? name : "socket"); + break; + } + } + } + + free(line); + fclose(fp); + + return fi->name != 0; +} + +// Matches lines in either /proc/net/tcp or /proc/net/tcp6, depending on 'af'. +static int ip_match(int af, char* line, struct in6_addr* l, int* l_port, + struct in6_addr* r, int* r_port, int* state, long* inode) +{ + if (af == AF_INET) { + return sscanf(line, " %*d: %x:%x %x:%x %x %*x:%*x %*X:%*X %*X %*d %*d %ld", + &(l->s6_addr32[0]), l_port, &(r->s6_addr32[0]), r_port, + state, inode) == 6; + } else { + return sscanf(line, " %*d: %8x%8x%8x%8x:%x %8x%8x%8x%8x:%x %x " + "%*x:%*x %*X:%*X %*X %*d %*d %ld", + &(l->s6_addr32[0]), &(l->s6_addr32[1]), &(l->s6_addr32[2]), + &(l->s6_addr32[3]), l_port, &(r->s6_addr32[0]), + &(r->s6_addr32[1]), &(r->s6_addr32[2]), &(r->s6_addr32[3]), + r_port, state, inode) == 12; + } +} + +static int find_ip_socket(struct file_info *fi, const char *path, + int af, int type, long sought_inode) +{ + FILE *fp = fopen(path, "r"); + char *line = NULL; + size_t line_length = 0; + char *tcp_states[] = { + "UNKNOWN", "ESTABLISHED", "SYN_SENT", "SYN_RECV", "FIN_WAIT1", "FIN_WAIT2", + "TIME_WAIT", "CLOSE", "CLOSE_WAIT", "LAST_ACK", "LISTEN", "CLOSING" + }; + + if (!fp) return 0; + + if (!getline(&line, &line_length, fp)) return 0; // Skip header. + + while (getline(&line, &line_length, fp) > 0) { + struct in6_addr local, remote; + int local_port, remote_port, state; + long inode; + + if (ip_match(af, line, &local, &local_port, &remote, &remote_port, + &state, &inode)) { + if (inode == sought_inode) { + char local_ip[INET6_ADDRSTRLEN] = {0}; + char remote_ip[INET6_ADDRSTRLEN] = {0}; + + strcpy(fi->type, af == AF_INET ? "IPv4" : "IPv6"); + inet_ntop(af, &local, local_ip, sizeof(local_ip)); + inet_ntop(af, &remote, remote_ip, sizeof(remote_ip)); + if (type == SOCK_STREAM) { + if (state < 0 || state > TCP_CLOSING) state = 0; + fi->name = xmprintf(af == AF_INET ? + "TCP %s:%d->%s:%d (%s)" : + "TCP [%s]:%d->[%s]:%d (%s)", + local_ip, local_port, remote_ip, remote_port, + tcp_states[state]); + } else { + fi->name = xmprintf(af == AF_INET ? + "%s %s:%d->%s:%d" : "%s [%s]:%d->[%s]:%d", + type == SOCK_DGRAM ? "UDP" : "RAW", + local_ip, local_port, remote_ip, remote_port); + } + break; + } + } + } + + free(line); + fclose(fp); + + return fi->name != 0; +} + +static int find_socket(struct file_info *fi, long inode) +{ + // TODO: other protocols (netlink). + return find_unix_socket(fi, inode) || + find_ip_socket(fi, "/proc/net/tcp", AF_INET, SOCK_STREAM, inode) || + find_ip_socket(fi, "/proc/net/tcp6", AF_INET6, SOCK_STREAM, inode) || + find_ip_socket(fi, "/proc/net/udp", AF_INET, SOCK_DGRAM, inode) || + find_ip_socket(fi, "/proc/net/udp6", AF_INET6, SOCK_DGRAM, inode) || + find_ip_socket(fi, "/proc/net/raw", AF_INET, SOCK_RAW, inode) || + find_ip_socket(fi, "/proc/net/raw6", AF_INET6, SOCK_RAW, inode); +} + +static void fill_stat(struct file_info *fi, const char* path) +{ + struct stat sb; + long dev; + + if (stat(path, &sb)) return; + + // Fill TYPE. + switch ((sb.st_mode & S_IFMT)) { + case S_IFBLK: strcpy(fi->type, "BLK"); break; + case S_IFCHR: strcpy(fi->type, "CHR"); break; + case S_IFDIR: strcpy(fi->type, "DIR"); break; + case S_IFIFO: strcpy(fi->type, "FIFO"); break; + case S_IFLNK: strcpy(fi->type, "LINK"); break; + case S_IFREG: strcpy(fi->type, "REG"); break; + case S_IFSOCK: strcpy(fi->type, "sock"); break; + default: + snprintf(fi->type, sizeof(fi->type), "0%03o", sb.st_mode & S_IFMT); + break; + } + + if (S_ISSOCK(sb.st_mode)) find_socket(fi, sb.st_ino); + + // Fill DEVICE. + dev = (S_ISBLK(sb.st_mode) || S_ISCHR(sb.st_mode)) ? sb.st_rdev : sb.st_dev; + snprintf(fi->device, sizeof(fi->device), "%ld,%ld", + (long)major(dev), (long)minor(dev)); + + // Fill SIZE/OFF. + if (S_ISREG(sb.st_mode) || S_ISDIR(sb.st_mode)) + snprintf(fi->size_off, sizeof(fi->size_off), "%lld", + (long long)sb.st_size); + + // Fill NODE. + snprintf(fi->node, sizeof(fi->node), "%ld", (long)sb.st_ino); +} + +struct file_info *new_file_info(struct proc_info *pi, const char* fd) +{ + struct file_info *fi = xzalloc(sizeof(struct file_info)); + + dlist_add_nomalloc(&TT.head, (struct double_list *)fi); + + fi->pi = *pi; + + // Defaults. + strcpy(fi->fd, fd); + strcpy(fi->type, "unknown"); + fi->rw = fi->locks = ' '; + + return fi; +} + +static void visit_symlink(struct proc_info *pi, char* name, char* path) +{ + struct file_info *fi = new_file_info(pi, ""); + + // Get NAME. + if (name) { // "/proc/pid/[cwd]". + snprintf(fi->fd, sizeof(fi->fd), "%s", name); + snprintf(toybuf, sizeof(toybuf), "/proc/%d/%s", pi->pid, path); + } else { // "/proc/pid/fd/[3]" + snprintf(fi->fd, sizeof(fi->fd), "%s", path); + fill_flags(fi); // Clobbers toybuf. + snprintf(toybuf, sizeof(toybuf), "/proc/%d/fd/%s", pi->pid, path); + } + // TODO: code called by fill_stat would be easier to write if we didn't + // rely on toybuf being preserved here. + fill_stat(fi, toybuf); + if (!fi->name) { // We already have a name for things like sockets. + fi->name = xreadlink(toybuf); + if (!fi->name) { + fi->name = xmprintf("%s (readlink: %s)", toybuf, strerror(errno)); + } + } +} + +static void visit_maps(struct proc_info *pi) +{ + FILE *fp; + unsigned long long offset; + char device[10]; + long inode; + char *line = NULL; + size_t line_length = 0; + + snprintf(toybuf, sizeof(toybuf), "/proc/%d/maps", pi->pid); + fp = fopen(toybuf, "r"); + if (!fp) return; + + while (getline(&line, &line_length, fp) > 0) { + int name_pos; + + if (sscanf(line, "%*x-%*x %*s %llx %s %ld %n", + &offset, device, &inode, &name_pos) >= 3) { + struct file_info *fi; + + // Ignore non-file maps. + if (inode == 0 || !strcmp(device, "00:00")) continue; + // TODO: show unique maps even if they have a non-zero offset? + if (offset != 0) continue; + + fi = new_file_info(pi, "mem"); + fi->name = strdup(chomp(line + name_pos)); + fill_stat(fi, fi->name); + } + } + free(line); + fclose(fp); +} + +static void visit_fds(struct proc_info *pi) +{ + DIR *dir; + struct dirent *de; + + snprintf(toybuf, sizeof(toybuf), "/proc/%d/fd", pi->pid); + if (!(dir = opendir(toybuf))) { + struct file_info *fi = new_file_info(pi, "NOFD"); + + fi->name = xmprintf("%s (opendir: %s)", toybuf, strerror(errno)); + return; + } + + while ((de = readdir(dir))) { + if (*de->d_name == '.') continue; + visit_symlink(pi, NULL, de->d_name); + } + + closedir(dir); +} + +static void lsof_pid(int pid) +{ + struct proc_info pi; + FILE *fp; + char *line; + struct stat sb; + + // Does this process even exist? + snprintf(toybuf, sizeof(toybuf), "/proc/%d/stat", pid); + fp = fopen(toybuf, "r"); + if (!fp) return; + + // Get COMMAND. + strcpy(pi.cmd, "?"); + line = fgets(toybuf, sizeof(toybuf), fp); + fclose(fp); + if (line) { + char *open_paren = strchr(toybuf, '('); + char *close_paren = strrchr(toybuf, ')'); + + if (open_paren && close_paren) { + *close_paren = 0; + snprintf(pi.cmd, sizeof(pi.cmd), "%s", open_paren + 1); + } + } + + // We already know PID. + pi.pid = pid; + + // Get USER. + snprintf(toybuf, sizeof(toybuf), "/proc/%d", pid); + if (!stat(toybuf, &sb)) { + struct passwd *pw = getpwuid(sb.st_uid); + + if (pw) snprintf(pi.user, sizeof(pi.user), "%s", pw->pw_name); + else snprintf(pi.user, sizeof(pi.user), "%u", (unsigned)sb.st_uid); + } + + visit_symlink(&pi, "cwd", "cwd"); + visit_symlink(&pi, "rtd", "root"); + visit_symlink(&pi, "txt", "exe"); + visit_maps(&pi); + visit_fds(&pi); +} + +static int scan_slash_proc(struct dirtree *node) +{ + int pid; + + if (!node->parent) return DIRTREE_RECURSE; + if ((pid = atol(node->name))) lsof_pid(pid); + return 0; +} + +void lsof_main(void) +{ + if (toys.optflags&FLAG_p) lsof_pid(TT.pid); + else dirtree_read("/proc", scan_slash_proc); + + // TODO: path filtering. + + llist_traverse(TT.head, print_info); +} |