summaryrefslogtreecommitdiff
path: root/sys/src/cmd/snap/snap.c
blob: 16912be34325de1df30c88a5d4a361c841b9dccf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#include <u.h>
#include <libc.h>
#include <bio.h>
#include "snap.h"

void
usage(void)
{
	fprint(2, "usage: %s [-o snapfile] pid...\n", argv0);
	exits("usage");
}

void
main(int argc, char **argv)
{
	char *user, *sys, *arch, *term, *ofile;
	int i;
	long pid, me;
	Biobuf *b;
	Dir *d;
	Proc *p;

	ofile = "/fd/1";
	ARGBEGIN{
	case 'o':
		ofile = ARGF();
		break;
	default:
		usage();
	}ARGEND;

	if(argc < 1)
		usage();

	/* get kernel compilation time */
	if((d = dirstat("#/")) == nil) {
		fprint(2, "cannot stat #/ ???\n");
		exits("stat");
	}

	if((b = Bopen(ofile, OWRITE)) == nil) {
		fprint(2, "cannot write to \"%s\"\n", ofile);
		exits("Bopen");
	}

	if((user = getuser()) == nil)
		user = "gre";
	if((sys = sysname()) == nil)
		sys = "gnot";
	if((arch = getenv("cputype")) == nil)
		arch = "unknown";
	if((term = getenv("terminal")) == nil)
		term = "unknown terminal type";

	Bprint(b, "process snapshot %ld %s@%s %s %ld \"%s\"\n",
		time(0), user, sys, arch, d->mtime, term);
	me = getpid();
	for(i=0; i<argc; i++) {
		if((pid = atol(argv[i])) == me)
			fprint(2, "warning: will not snapshot self\n");
		else if(p = snap(pid, 1))
			writesnap(b, p);
	}
	exits(0);
}