1 f0f4401f 2004-04-23 devnull #include <u.h>
  2 f0f4401f 2004-04-23 devnull #include <libc.h>
  3 f0f4401f 2004-04-23 devnull #include <bio.h>
  4 f0f4401f 2004-04-23 devnull #include "dat.h"
  5 f0f4401f 2004-04-23 devnull 
  6 f0f4401f 2004-04-23 devnull Biobuf bout;
  7 f0f4401f 2004-04-23 devnull 
  8 f0f4401f 2004-04-23 devnull void
  9 f0f4401f 2004-04-23 devnull usage(void)
 10 f0f4401f 2004-04-23 devnull {
 11 f0f4401f 2004-04-23 devnull 	fprint(2, "usage: auxstats [system [executable]]\n");
 12 f0f4401f 2004-04-23 devnull 	exits("usage");
 13 f0f4401f 2004-04-23 devnull }
 14 f0f4401f 2004-04-23 devnull 
 15 f0f4401f 2004-04-23 devnull int pid;
 16 f0f4401f 2004-04-23 devnull 
 17 f0f4401f 2004-04-23 devnull void
 18 f0f4401f 2004-04-23 devnull notifyf(void *v, char *msg)
 19 f0f4401f 2004-04-23 devnull {
 20 f0f4401f 2004-04-23 devnull 	USED(v);
 21 f0f4401f 2004-04-23 devnull 
 22 272fd660 2004-04-23 devnull 	if(strstr(msg, "child"))
 23 f0f4401f 2004-04-23 devnull 		noted(NCONT);
 24 fdcf3d70 2021-01-14 rsc 	if(pid)
 25 fdcf3d70 2021-01-14 rsc 		postnote(PNPROC, pid, msg);
 26 f0f4401f 2004-04-23 devnull 	exits(nil);
 27 f0f4401f 2004-04-23 devnull }
 28 f0f4401f 2004-04-23 devnull 
 29 f0f4401f 2004-04-23 devnull void
 30 f0f4401f 2004-04-23 devnull main(int argc, char **argv)
 31 f0f4401f 2004-04-23 devnull {
 32 f0f4401f 2004-04-23 devnull 	char *sys, *exe;
 33 f0f4401f 2004-04-23 devnull 	int i;
 34 f0f4401f 2004-04-23 devnull 	Waitmsg *w;
 35 f0f4401f 2004-04-23 devnull 
 36 f0f4401f 2004-04-23 devnull 	ARGBEGIN{
 37 f0f4401f 2004-04-23 devnull 	default:
 38 f0f4401f 2004-04-23 devnull 		usage();
 39 f0f4401f 2004-04-23 devnull 	}ARGEND
 40 f0f4401f 2004-04-23 devnull 
 41 f0f4401f 2004-04-23 devnull 	notify(notifyf);
 42 f0f4401f 2004-04-23 devnull 
 43 f0f4401f 2004-04-23 devnull 	sys = nil;
 44 f0f4401f 2004-04-23 devnull 	exe = nil;
 45 f0f4401f 2004-04-23 devnull 	if(argc > 0)
 46 f0f4401f 2004-04-23 devnull 		sys = argv[0];
 47 f0f4401f 2004-04-23 devnull 	if(argc > 1)
 48 f0f4401f 2004-04-23 devnull 		exe = argv[1];
 49 f0f4401f 2004-04-23 devnull 	if(argc > 2)
 50 f0f4401f 2004-04-23 devnull 		usage();
 51 f0f4401f 2004-04-23 devnull 
 52 f0f4401f 2004-04-23 devnull 	if(sys){
 53 f0f4401f 2004-04-23 devnull 		if(exe == nil)
 54 f0f4401f 2004-04-23 devnull 			exe = "auxstats";
 55 f0f4401f 2004-04-23 devnull 		for(;;){
 56 f0f4401f 2004-04-23 devnull 			switch(pid = fork()){
 57 f0f4401f 2004-04-23 devnull 			case -1:
 58 f0f4401f 2004-04-23 devnull 				sysfatal("fork: %r");
 59 f0f4401f 2004-04-23 devnull 			case 0:
 60 f0f4401f 2004-04-23 devnull 				rfork(RFNOTEG);
 61 272fd660 2004-04-23 devnull 				execlp("ssh", "ssh", "-nTC", sys, exe, nil);
 62 272fd660 2004-04-23 devnull 				_exit(97);
 63 f0f4401f 2004-04-23 devnull 			default:
 64 f0f4401f 2004-04-23 devnull 				if((w = wait()) == nil)
 65 f0f4401f 2004-04-23 devnull 					sysfatal("wait: %r");
 66 f0f4401f 2004-04-23 devnull 				if(w->pid != pid)
 67 f0f4401f 2004-04-23 devnull 					sysfatal("wait got wrong pid");
 68 272fd660 2004-04-23 devnull 				if(atoi(w->msg) == 97)
 69 f0f4401f 2004-04-23 devnull 					sysfatal("exec ssh failed");
 70 f0f4401f 2004-04-23 devnull 				free(w);
 71 f0f4401f 2004-04-23 devnull 				fprint(2, "stats: %s hung up; sleeping 60\n", sys);
 72 f0f4401f 2004-04-23 devnull 				sleep(60*1000);
 73 f0f4401f 2004-04-23 devnull 				fprint(2, "stats: redialing %s\n", sys);
 74 f0f4401f 2004-04-23 devnull 				break;
 75 f0f4401f 2004-04-23 devnull 			}
 76 f0f4401f 2004-04-23 devnull 		}
 77 f0f4401f 2004-04-23 devnull 	}
 78 f0f4401f 2004-04-23 devnull 
 79 f0f4401f 2004-04-23 devnull 	for(i=0; statfn[i]; i++)
 80 f0f4401f 2004-04-23 devnull 		statfn[i](1);
 81 f0f4401f 2004-04-23 devnull 
 82 f0f4401f 2004-04-23 devnull 	Binit(&bout, 1, OWRITE);
 83 f0f4401f 2004-04-23 devnull 	for(;;){
 84 f0f4401f 2004-04-23 devnull 		Bflush(&bout);
 85 f0f4401f 2004-04-23 devnull 		sleep(900);
 86 f0f4401f 2004-04-23 devnull 		for(i=0; statfn[i]; i++)
 87 f0f4401f 2004-04-23 devnull 			statfn[i](0);
 88 f0f4401f 2004-04-23 devnull 	}
 89 f0f4401f 2004-04-23 devnull }
 90 f0f4401f 2004-04-23 devnull 
 91 f0f4401f 2004-04-23 devnull char buf[16384];
 92 f0f4401f 2004-04-23 devnull char *line[100];
 93 f0f4401f 2004-04-23 devnull char *tok[100];
 94 f0f4401f 2004-04-23 devnull int nline, ntok;
 95 f0f4401f 2004-04-23 devnull 
 96 f0f4401f 2004-04-23 devnull void
 97 f0f4401f 2004-04-23 devnull readfile(int fd)
 98 f0f4401f 2004-04-23 devnull {
 99 f0f4401f 2004-04-23 devnull 	int n;
100 f0f4401f 2004-04-23 devnull 
101 f0f4401f 2004-04-23 devnull 	if(fd == -1){
102 f0f4401f 2004-04-23 devnull 		nline = 0;
103 f0f4401f 2004-04-23 devnull 		return;
104 f0f4401f 2004-04-23 devnull 	}
105 f0f4401f 2004-04-23 devnull 
106 f0f4401f 2004-04-23 devnull 	seek(fd, 0, 0);
107 f0f4401f 2004-04-23 devnull 	n = read(fd, buf, sizeof buf-1);
108 f0f4401f 2004-04-23 devnull 	if(n < 0)
109 f0f4401f 2004-04-23 devnull 		n = 0;
110 f0f4401f 2004-04-23 devnull 	buf[n] = 0;
111 f0f4401f 2004-04-23 devnull 
112 f0f4401f 2004-04-23 devnull 	nline = getfields(buf, line, nelem(line), 0, "\n");
113 f0f4401f 2004-04-23 devnull }
114 f0f4401f 2004-04-23 devnull 
115 f0f4401f 2004-04-23 devnull void
116 f0f4401f 2004-04-23 devnull tokens(int i)
117 f0f4401f 2004-04-23 devnull {
118 f0f4401f 2004-04-23 devnull 	if(i >= nline){
119 f0f4401f 2004-04-23 devnull 		ntok = 0;
120 f0f4401f 2004-04-23 devnull 		return;
121 f0f4401f 2004-04-23 devnull 	}
122 f0f4401f 2004-04-23 devnull 	ntok = tokenize(line[i], tok, nelem(tok));
123 f0f4401f 2004-04-23 devnull }