Add killall, pidof
This commit is contained in:
K. Lange
parent
ebd877624f
commit
d80c893c38
222
apps/killall.c
Normal file
222
apps/killall.c
Normal file
@ -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
apps/pidof.c
Normal file
123
apps/pidof.c
Normal file
@ -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;
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user