Subversion Repositories planix.SVN

#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);
}