Subversion Repositories planix.SVN

/*
 * generate a list of files and their metadata
 * using a given proto file.
 */
#include "all.h"

int changesonly;
char *uid;
Db *db;
Biobuf blog;
ulong now;
int n;
char **x;
int nx;
int justlog;
char *root=".";
char **match;
int nmatch;

int
ismatch(char *s)
{
        int i, len;

        if(nmatch == 0)
                return 1;
        for(i=0; i<nmatch; i++){
                if(strcmp(s, match[i]) == 0)
                        return 1;
                len = strlen(match[i]);
                if(strncmp(s, match[i], len) == 0 && s[len]=='/')
                        return 1;
        }
        return 0;
}

void
xlog(int c, char *name, Dir *d)
{
        char *dname;

        dname = d->name;
        if(strcmp(dname, name) == 0)
                dname = "-";
        if(!justlog)
                Bprint(&blog, "%lud %d ", now, n++);
        Bprint(&blog, "%c %q %q %luo %q %q %lud %lld\n",
                c, name, dname, d->mode, uid ? uid : d->uid, d->gid, d->mtime, d->length);
}

void
walk(char *new, char *old, Dir *xd, void*)
{
        int i, change, len;
        Dir od, d;

        new = unroot(new, "/");
        old = unroot(old, root);

        if(!ismatch(new))
                return;
        for(i=0; i<nx; i++){
                if(strcmp(new, x[i]) == 0)
                        return;
                len = strlen(x[i]);
                if(strncmp(new, x[i], len)==0 && new[len]=='/')
                        return;
        }

        d = *xd;
        d.name = old;
        memset(&od, 0, sizeof od);
        change = 0;
        if(markdb(db, new, &od) < 0){
                if(!changesonly){
                        xlog('a', new, &d);
                        change = 1;
                }
        }else{
                if((d.mode&DMDIR)==0 && (od.mtime!=d.mtime || od.length!=d.length)){
                        xlog('c', new, &d);
                        change = 1;
                }
                if((!uid&&strcmp(od.uid,d.uid)!=0)
                || strcmp(od.gid,d.gid)!=0 
                || od.mode!=d.mode){
                        xlog('m', new, &d);
                        change = 1;
                }
        }
        if(!justlog && change){
                if(uid)
                        d.uid = uid;
                d.muid = "mark";        /* mark bit */
                insertdb(db, new, &d);
        }
}

void
warn(char *msg, void*)
{
        char *p;

        fprint(2, "warning: %s\n", msg);

        /* find the %r in "can't open foo: %r" */
        p = strstr(msg, ": ");
        if(p)
                p += 2;

        /*
         * if the error is about a remote server failing,
         * then there's no point in continuing to look
         * for changes -- we'll think everything got deleted!
         *
         * actual errors i see are:
         *      "i/o on hungup channel" for a local hangup
         *      "i/o on hungup channel" for a timeout (yank the network wire)
         *      "'/n/sources/plan9' Hangup" for a remote hangup
         * the rest is paranoia.
         */
        if(p){
                if(cistrstr(p, "hungup") || cistrstr(p, "Hangup")
                || cistrstr(p, "rpc error")
                || cistrstr(p, "shut down")
                || cistrstr(p, "i/o")
                || cistrstr(p, "connection"))
                        sysfatal("suspected network or i/o error - bailing out");
        }
}

void
usage(void)
{
        fprint(2, "usage: replica/updatedb [-c] [-p proto] [-r root] [-t now n] [-u uid] [-x path]... db [paths]\n");
        exits("usage");
}

void
main(int argc, char **argv)
{
        char *proto;
        Avlwalk *w;
        Dir d;
        Entry *e;

        quotefmtinstall();
        proto = "/sys/lib/sysconfig/proto/allproto";
        now = time(0);
        Binit(&blog, 1, OWRITE);
        ARGBEGIN{
        case 'c':
                changesonly = 1;
                break;
        case 'l':
                justlog = 1;
                break;
        case 'p':
                proto = EARGF(usage());
                break;
        case 'r':
                root = EARGF(usage());
                break;
        case 't':
                now = strtoul(EARGF(usage()), 0, 0);
                n = atoi(EARGF(usage()));
                break;
        case 'u':
                uid = EARGF(usage());
                break;
        case 'x':
                if(nx%16 == 0)
                        x = erealloc(x, (nx+16)*sizeof(x[0]));
                x[nx++] = EARGF(usage());
                break;
        default:
                usage();
        }ARGEND

        if(argc <1)
                usage();

        match = argv+1;
        nmatch = argc-1;

        db = opendb(argv[0]);
        if(rdproto(proto, root, walk, warn, nil) < 0)
                sysfatal("rdproto: %r");

        if(!changesonly){
                w = avlwalk(db->avl);
                while(e = (Entry*)avlprev(w)){
                        if(!ismatch(e->name))
                                continue;
                        if(!e->d.mark){         /* not visited during walk */
                                memset(&d, 0, sizeof d);
                                d.name = e->d.name;
                                d.uid = e->d.uid;
                                d.gid = e->d.gid;
                                d.mtime = e->d.mtime;
                                d.mode = e->d.mode;
                                xlog('d', e->name, &d);
                                if(!justlog)
                                        removedb(db, e->name);
                        }
                }
        }

        if(Bterm(&blog) < 0)
                sysfatal("writing output: %r");

        exits(nil);
}