148 lines
2.6 KiB
C
148 lines
2.6 KiB
C
|
/*
|
||
|
|
* ps
|
||
|
|
*
|
||
|
|
* print a list of running processes
|
||
|
|
*/
|
||
|
|
|
||
|
|
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <stdint.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <syscall.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <dirent.h>
|
||
|
|
|
||
|
|
#include "lib/list.h"
|
||
|
|
|
||
|
|
#define LINE_LEN 4096
|
||
|
|
|
||
|
|
void print_username(int uid) {
|
||
|
|
FILE * passwd = fopen("/etc/passwd", "r");
|
||
|
|
char line[LINE_LEN];
|
||
|
|
|
||
|
|
while (fgets(line, LINE_LEN, passwd) != NULL) {
|
||
|
|
|
||
|
|
line[strlen(line)-1] = '\0';
|
||
|
|
|
||
|
|
char *p, *tokens[10], *last;
|
||
|
|
int i = 0;
|
||
|
|
for ((p = strtok_r(line, ":", &last)); p; (p = strtok_r(NULL, ":", &last)), i++) {
|
||
|
|
if (i < 511) tokens[i] = p;
|
||
|
|
}
|
||
|
|
tokens[i] = NULL;
|
||
|
|
|
||
|
|
if (atoi(tokens[2]) == uid) {
|
||
|
|
printf("%-8s", tokens[0]);
|
||
|
|
fclose(passwd);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
printf("%-8d", uid);
|
||
|
|
fclose(passwd);
|
||
|
|
}
|
||
|
|
|
||
|
|
void print_entry(struct dirent * dent) {
|
||
|
|
char tmp[256], buf[4096];
|
||
|
|
FILE * f;
|
||
|
|
int read = 1;
|
||
|
|
char line[LINE_LEN];
|
||
|
|
|
||
|
|
int pid, uid;
|
||
|
|
|
||
|
|
snprintf(tmp, 256, "/proc/%s/status", dent->d_name);
|
||
|
|
f = fopen(tmp, "r");
|
||
|
|
|
||
|
|
while (fgets(line, LINE_LEN, f) != NULL) {
|
||
|
|
if (strstr(line, "Pid:") == line) {
|
||
|
|
sscanf(line, "%s %d", &buf, &pid);
|
||
|
|
} else if (strstr(line, "Uid:") == line) {
|
||
|
|
sscanf(line, "%s %d", &buf, &uid);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fclose(f);
|
||
|
|
|
||
|
|
print_username(uid);
|
||
|
|
printf(" %5d ", pid);
|
||
|
|
|
||
|
|
snprintf(tmp, 256, "/proc/%s/cmdline", dent->d_name);
|
||
|
|
f = fopen(tmp, "r");
|
||
|
|
memset(buf, 0x00, 4096);
|
||
|
|
read = fread(buf, 1, 4096, f);
|
||
|
|
fclose(f);
|
||
|
|
|
||
|
|
buf[read] = '\0';
|
||
|
|
for (int i = 0; i < read; ++i) {
|
||
|
|
if (buf[i] == '\036') {
|
||
|
|
buf[i] = ' ';
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
printf("%s\n", buf);
|
||
|
|
}
|
||
|
|
|
||
|
|
void show_usage(int argc, char * argv[]) {
|
||
|
|
printf(
|
||
|
|
"ps - list running processes\n"
|
||
|
|
"\n"
|
||
|
|
"usage: %s [-A] [format]\n"
|
||
|
|
"\n"
|
||
|
|
" -A \033[3mignored\033[0m\n"
|
||
|
|
" -? \033[3mshow this help text\033[0m\n"
|
||
|
|
"\n", argv[0]);
|
||
|
|
}
|
||
|
|
|
||
|
|
int main (int argc, char * argv[]) {
|
||
|
|
|
||
|
|
/* Parse arguments */
|
||
|
|
int show_all = 0;
|
||
|
|
|
||
|
|
if (argc > 1) {
|
||
|
|
int index, c;
|
||
|
|
while ((c = getopt(argc, argv, "A?")) != -1) {
|
||
|
|
switch (c) {
|
||
|
|
case 'A':
|
||
|
|
show_all = 1;
|
||
|
|
break;
|
||
|
|
case '?':
|
||
|
|
show_usage(argc, argv);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Open the directory */
|
||
|
|
DIR * dirp = opendir("/proc");
|
||
|
|
|
||
|
|
/* Read the entries in the directory */
|
||
|
|
list_t * ents_list = list_create();
|
||
|
|
|
||
|
|
struct dirent * ent = readdir(dirp);
|
||
|
|
while (ent != NULL) {
|
||
|
|
if (ent->d_name[0] >= '0' && ent->d_name[0] <= '9') {
|
||
|
|
struct dirent * entcpy = malloc(sizeof(struct dirent));
|
||
|
|
memcpy(entcpy, ent, sizeof(struct dirent));
|
||
|
|
list_insert(ents_list, (void *)entcpy);
|
||
|
|
}
|
||
|
|
|
||
|
|
ent = readdir(dirp);
|
||
|
|
}
|
||
|
|
closedir(dirp);
|
||
|
|
|
||
|
|
foreach(entry, ents_list) {
|
||
|
|
print_entry(entry->value);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* vim: tabstop=4
|
||
|
|
* vim: shiftwidth=4
|
||
|
|
* vim: noexpandtab
|
||
|
|
*/
|