Subversion Repositories planix.SVN

#include "all.h"

static Entry *fe;

static Entry*
allocentry(void)
{
        int i;
        Entry *e;

        if(fe == nil){
                fe = emalloc(128*sizeof(Entry));
                for(i=0; i<128-1; i++)
                        fe[i].name = (char*)&fe[i+1];
                fe[i].name = nil;
        }

        e = fe;
        fe = (Entry*)e->name;
        memset(e, 0, sizeof *e);
        return e;
}

static void
freeentry(Entry *e)
{
        e->name = (char*)fe;
        fe = e;
}

static void
_removedb(Db *db, char *name)
{
        Entry *e, k;

        memset(&k, 0, sizeof k);
        k.name = name;
        e = nil;
        deleteavl(db->avl, (Avl*)&k, (Avl**)&e);
        if(e)
                freeentry(e);
}

static void
_insertdb(Db *db, Entry *e)
{
        Entry *o, *ne;

        ne = allocentry();
        *ne = *e;
        o = nil;
        insertavl(db->avl, (Avl*)ne, (Avl**)&o);
        if(o)
                freeentry(o);
}

static int
entrycmp(Avl *a, Avl *b)
{
        Entry *ea, *eb;
        int r;

        ea = (Entry*)a;
        eb = (Entry*)b;
        r = strcmp(ea->name, eb->name);
        return r > 0 ? 1 : r < 0 ? -1 : 0;
}

Db*
opendb(char *file)
{
        char *f[10], *s, *t;
        int i, fd, nf;
        Biobuf b;
        Db *db;
        Entry e;

        if(file == nil)
                fd = -1;
        else if((fd = open(file, ORDWR)) < 0)
                sysfatal("opendb %s: %r", file);
        db = emalloc(sizeof(Db));
        db->avl = mkavltree(entrycmp);
        db->fd = fd;
        if(fd < 0)
                return db;
        Binit(&b, fd, OREAD);
        i = 0;
        for(; s=Brdstr(&b, '\n', 1); free(s)){
                t = estrdup(s);
                nf = tokenize(s, f, nelem(f));
                if(nf != 7)
                        sysfatal("bad database entry '%s'", t);
                free(t);
                if(strcmp(f[2], "REMOVED") == 0)
                        _removedb(db, f[0]);
                else{
                        memset(&e, 0, sizeof e);
                        e.name = atom(f[0]);
                        e.d.name = atom(f[1]);
                        if(strcmp(e.d.name, "-")==0)
                                e.d.name = e.name;
                        e.d.mode = strtoul(f[2], 0, 8);
                        e.d.uid = atom(f[3]);
                        e.d.gid = atom(f[4]);
                        e.d.mtime = strtoul(f[5], 0, 10);
                        e.d.length = strtoll(f[6], 0, 10);
                        _insertdb(db, &e);
                        i++;
                }
        }
        return db;
}

static int
_finddb(Db *db, char *name, Dir *d, int domark)
{
        Entry *e, k;

        memset(&k, 0, sizeof k);
        k.name = name;

        e = (Entry*)lookupavl(db->avl, (Avl*)&k);
        if(e == nil)
                return -1;
        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;
        d->length = e->d.length;
        if(domark)
                e->d.mark = 1;
        return 0;
}

int
finddb(Db *db, char *name, Dir *d)
{
        return _finddb(db, name, d, 0);
}

int
markdb(Db *db, char *name, Dir *d)
{
        return _finddb(db, name, d, 1);
}

void
removedb(Db *db, char *name)
{
        if(db->fd>=0 && fprint(db->fd, "%q xxx REMOVED xxx xxx 0 0\n", name) < 0)
                sysfatal("appending to db: %r");
        _removedb(db, name);
}

void
insertdb(Db *db, char *name, Dir *d)
{
        char *dname;
        Entry e;

        memset(&e, 0, sizeof e);
        e.name = atom(name);
        e.d.name = atom(d->name);
        e.d.uid = atom(d->uid);
        e.d.gid = atom(d->gid);
        e.d.mtime = d->mtime;
        e.d.mode = d->mode;
        e.d.length = d->length;
        e.d.mark = d->muid!=0;

        dname = d->name;
        if(strcmp(name, dname) == 0)
                dname = "-";
        if(db->fd>=0 && fprint(db->fd, "%q %q %luo %q %q %lud %lld\n", name, dname, d->mode, d->uid, d->gid, d->mtime, d->length) < 0)
                sysfatal("appending to db: %r");
        _insertdb(db, &e);
}