Browse Source

Add killall, pidof

K. Lange 2 years ago
parent
commit
d80c893c38
2 changed files with 345 additions and 0 deletions
  1. 222 0
      apps/killall.c
  2. 123 0
      apps/pidof.c

+ 222 - 0
apps/killall.c

@@ -0,0 +1,222 @@
+/* vim: ts=4 sw=4 noexpandtab
+ * This file is part of ToaruOS and is released under the terms
+ * of the NCSA / University of Illinois License - see LICENSE.md
+ * Copyright (C) 2018 K. Lange
+ *
+ * killall
+ *
+ * Find processes by name and send them signals.
+ */
+
+
+#include <sys/stat.h>
+#include <sys/signal.h>
+#include <fcntl.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <dirent.h>
+#include <signal.h>
+#include <getopt.h>
+
+typedef struct process {
+	int pid;
+	int ppid;
+	int tgid;
+	char name[100];
+	char path[200];
+} p_t;
+
+#define LINE_LEN 4096
+
+p_t * build_entry(struct dirent * dent) {
+	char tmp[256];
+	FILE * f;
+	char line[LINE_LEN];
+
+	sprintf(tmp, "/proc/%s/status", dent->d_name);
+	f = fopen(tmp, "r");
+
+	p_t * proc = malloc(sizeof(p_t));
+
+	while (fgets(line, LINE_LEN, f) != NULL) {
+		char * n = strstr(line,"\n");
+		if (n) { *n = '\0'; }
+		char * tab = strstr(line,"\t");
+		if (tab) {
+			*tab = '\0';
+			tab++;
+		}
+		if (strstr(line, "Pid:") == line) {
+			proc->pid = atoi(tab);
+		} else if (strstr(line, "PPid:") == line) {
+			proc->ppid = atoi(tab);
+		} else if (strstr(line, "Tgid:") == line) {
+			proc->tgid = atoi(tab);
+		} else if (strstr(line, "Name:") == line) {
+			strcpy(proc->name, tab);
+		} else if (strstr(line, "Path:") == line) {
+			strcpy(proc->path, tab);
+		}
+	}
+
+	if (strstr(proc->name,"python") == proc->name) {
+		char * name = proc->path + strlen(proc->path) - 1;
+
+		while (1) {
+			if (*name == '/') {
+				name++;
+				break;
+			}
+			if (name == proc->name) break;
+			name--;
+		}
+
+		memcpy(proc->name, name, strlen(name)+1);
+	}
+
+	if (proc->tgid != proc->pid) {
+		char tmp[100] = {0};
+		sprintf(tmp, "{%s}", proc->name);
+		memcpy(proc->name, tmp, strlen(tmp)+1);
+	}
+
+	fclose(f);
+
+	return proc;
+}
+
+void show_usage(int argc, char * argv[]) {
+	printf(
+			"killall - send signal to processes with given name\n"
+			"\n"
+			"usage: %s [-s SIG] name\n"
+			"\n"
+			" -s     \033[3msignal to send\033[0m\n"
+			" -?     \033[3mshow this help text\033[0m\n"
+			"\n", argv[0]);
+}
+
+struct sig_def {
+	int sig;
+	const char * name;
+};
+
+struct sig_def signals[] = {
+	{SIGHUP,"HUP"},
+	{SIGINT,"INT"},
+	{SIGQUIT,"QUIT"},
+	{SIGILL,"ILL"},
+	{SIGTRAP,"TRAP"},
+	{SIGABRT,"ABRT"},
+	{SIGEMT,"EMT"},
+	{SIGFPE,"FPE"},
+	{SIGKILL,"KILL"},
+	{SIGBUS,"BUS"},
+	{SIGSEGV,"SEGV"},
+	{SIGSYS,"SYS"},
+	{SIGPIPE,"PIPE"},
+	{SIGALRM,"ALRM"},
+	{SIGTERM,"TERM"},
+	{SIGUSR1,"USR1"},
+	{SIGUSR2,"USR2"},
+	{SIGCHLD,"CHLD"},
+	{SIGPWR,"PWR"},
+	{SIGWINCH,"WINCH"},
+	{SIGURG,"URG"},
+	{SIGPOLL,"POLL"},
+	{SIGSTOP,"STOP"},
+	{SIGTSTP,"TSTP"},
+	{SIGCONT,"CONT"},
+	{SIGTTIN,"TTIN"},
+	{SIGTTOUT,"TTOUT"},
+	{SIGVTALRM,"VTALRM"},
+	{SIGPROF,"PROF"},
+	{SIGXCPU,"XCPU"},
+	{SIGXFSZ,"XFSZ"},
+	{SIGWAITING,"WAITING"},
+	{SIGDIAF,"DIAF"},
+	{SIGHATE,"HATE"},
+	{SIGWINEVENT,"WINEVENT"},
+	{SIGCAT,"CAT"},
+	{0,NULL},
+};
+
+int main (int argc, char * argv[]) {
+
+	int signum = SIGTERM;
+
+	/* Open the directory */
+	DIR * dirp = opendir("/proc");
+
+	char c;
+	while ((c = getopt(argc, argv, "s:?")) != -1) {
+		switch (c) {
+			case 's':
+				{
+					signum = -1;
+					if (strlen(optarg) > 3 && strstr(optarg,"SIG") == (optarg)) {
+						struct sig_def * s = signals;
+						while (s->name) {
+							if (!strcmp(optarg+3,s->name)) {
+								signum = s->sig;
+								break;
+							}
+							s++;
+						}
+					} else {
+						if (optarg[0] < '0' || optarg[0] > '9') {
+							struct sig_def * s = signals;
+							while (s->name) {
+								if (!strcmp(optarg,s->name)) {
+									signum = s->sig;
+									break;
+								}
+								s++;
+							}
+						} else {
+							signum = atoi(optarg);
+						}
+					}
+					if (signum == -1) {
+						fprintf(stderr,"%s: %s: invalid signal specification\n",argv[0],optarg);
+						return 1;
+					}
+				}
+				break;
+			case '?':
+				show_usage(argc, argv);
+				return 0;
+		}
+	}
+
+	if (optind >= argc) {
+		show_usage(argc, argv);
+		return 1;
+	}
+
+	int killed_something = 0;
+
+	struct dirent * ent = readdir(dirp);
+	while (ent != NULL) {
+		if (ent->d_name[0] >= '0' && ent->d_name[0] <= '9') {
+			p_t * proc = build_entry(ent);
+
+			if (!strcmp(proc->name, argv[optind])) {
+				kill(proc->pid, signum);
+				killed_something = 1;
+			}
+		}
+		ent = readdir(dirp);
+	}
+	closedir(dirp);
+
+	if (!killed_something) {
+		fprintf(stderr, "%s: no process found\n", argv[optind]);
+		return 1;
+	}
+	return 0;
+}
+

+ 123 - 0
apps/pidof.c

@@ -0,0 +1,123 @@
+/* vim: ts=4 sw=4 noexpandtab
+ * This file is part of ToaruOS and is released under the terms
+ * of the NCSA / University of Illinois License - see LICENSE.md
+ * Copyright (C) 2018 K. Lange
+ *
+ * pidof
+ *
+ * find and print process IDs
+ */
+
+
+#include <sys/stat.h>
+#include <sys/signal.h>
+#include <fcntl.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <dirent.h>
+#include <signal.h>
+
+typedef struct process {
+	int pid;
+	int ppid;
+	int tgid;
+	char name[100];
+	char path[200];
+} p_t;
+
+#define LINE_LEN 4096
+
+p_t * build_entry(struct dirent * dent) {
+	char tmp[256];
+	FILE * f;
+	char line[LINE_LEN];
+
+	sprintf(tmp, "/proc/%s/status", dent->d_name);
+	f = fopen(tmp, "r");
+
+	p_t * proc = malloc(sizeof(p_t));
+
+	while (fgets(line, LINE_LEN, f) != NULL) {
+		char * n = strstr(line,"\n");
+		if (n) { *n = '\0'; }
+		char * tab = strstr(line,"\t");
+		if (tab) {
+			*tab = '\0';
+			tab++;
+		}
+		if (strstr(line, "Pid:") == line) {
+			proc->pid = atoi(tab);
+		} else if (strstr(line, "PPid:") == line) {
+			proc->ppid = atoi(tab);
+		} else if (strstr(line, "Tgid:") == line) {
+			proc->tgid = atoi(tab);
+		} else if (strstr(line, "Name:") == line) {
+			strcpy(proc->name, tab);
+		} else if (strstr(line, "Path:") == line) {
+			strcpy(proc->path, tab);
+		}
+	}
+
+	if (strstr(proc->name,"python") == proc->name) {
+		char * name = proc->path + strlen(proc->path) - 1;
+
+		while (1) {
+			if (*name == '/') {
+				name++;
+				break;
+			}
+			if (name == proc->name) break;
+			name--;
+		}
+
+		memcpy(proc->name, name, strlen(name)+1);
+	}
+
+	if (proc->tgid != proc->pid) {
+		char tmp[100] = {0};
+		sprintf(tmp, "{%s}", proc->name);
+		memcpy(proc->name, tmp, strlen(tmp)+1);
+	}
+
+	fclose(f);
+
+	return proc;
+}
+
+int main (int argc, char * argv[]) {
+
+	if (argc < 2) return 1;
+
+	int space = 0;
+
+	/* Open the directory */
+	DIR * dirp = opendir("/proc");
+
+	int found_something = 0;
+
+	struct dirent * ent = readdir(dirp);
+	while (ent != NULL) {
+		if (ent->d_name[0] >= '0' && ent->d_name[0] <= '9') {
+			p_t * proc = build_entry(ent);
+
+			if (!strcmp(proc->name, argv[optind])) {
+				if (space++) printf(" ");
+				printf("%d", proc->pid);
+				found_something = 1;
+			}
+		}
+		ent = readdir(dirp);
+	}
+	closedir(dirp);
+
+	if (!found_something) {
+		return 1;
+	}
+	printf("\n");
+	return 0;
+}
+
+