123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507 |
- #include <u.h>
- #include <libc.h>
- #include <bio.h>
- #include <auth.h>
- #include <fcall.h>
- #include <disk.h>
- enum {
- LEN = 8*1024,
- HUNKS = 128,
- };
- typedef struct File File;
- struct File{
- char *new;
- char *elem;
- char *old;
- char *uid;
- char *gid;
- ulong mode;
- };
- typedef void Mkfserr(char*, void*);
- typedef void Mkfsenum(char*, char*, Dir*, void*);
- typedef struct Name Name;
- struct Name {
- int n;
- char *s;
- };
- typedef struct Mkaux Mkaux;
- struct Mkaux {
- Mkfserr *warn;
- Mkfsenum *mkenum;
- char *root;
- char *proto;
- jmp_buf jmp;
- Biobuf *b;
- Name oldfile;
- Name fullname;
- int lineno;
- int indent;
- void *a;
- };
- static void domkfs(Mkaux *mkaux, File *me, int level);
- static int copyfile(Mkaux*, File*, Dir*, int);
- static void freefile(File*);
- static File* getfile(Mkaux*, File*);
- static char* getmode(Mkaux*, char*, ulong*);
- static char* getname(Mkaux*, char*, char**);
- static char* getpath(Mkaux*, char*);
- static int mkfile(Mkaux*, File*);
- static char* mkpath(Mkaux*, char*, char*);
- static void mktree(Mkaux*, File*, int);
- static void setnames(Mkaux*, File*);
- static void skipdir(Mkaux*);
- static void warn(Mkaux*, char *, ...);
- //static void
- //mprint(char *new, char *old, Dir *d, void*)
- //{
- // print("%s %s %D\n", new, old, d);
- //}
- int
- rdproto(char *proto, char *root, Mkfsenum *mkenum, Mkfserr *mkerr, void *a)
- {
- Mkaux mx, *m;
- File file;
- int rv;
- m = &mx;
- memset(&mx, 0, sizeof mx);
- if(root == nil)
- root = "/";
- m->root = root;
- m->warn = mkerr;
- m->mkenum = mkenum;
- m->a = a;
- m->proto = proto;
- m->lineno = 0;
- m->indent = 0;
- if((m->b = Bopen(proto, OREAD)) == nil) {
- werrstr("open '%s': %r", proto);
- return -1;
- }
- memset(&file, 0, sizeof file);
- file.new = "";
- file.old = nil;
- rv = 0;
- if(setjmp(m->jmp) == 0)
- domkfs(m, &file, -1);
- else
- rv = -1;
- free(m->oldfile.s);
- free(m->fullname.s);
- return rv;
- }
- static void*
- emalloc(Mkaux *mkaux, ulong n)
- {
- void *v;
- v = malloc(n);
- if(v == nil)
- longjmp(mkaux->jmp, 1); /* memory leak */
- memset(v, 0, n);
- return v;
- }
- static char*
- estrdup(Mkaux *mkaux, char *s)
- {
- s = strdup(s);
- if(s == nil)
- longjmp(mkaux->jmp, 1); /* memory leak */
- return s;
- }
- static void
- domkfs(Mkaux *mkaux, File *me, int level)
- {
- File *child;
- int rec;
- child = getfile(mkaux, me);
- if(!child)
- return;
- if((child->elem[0] == '+' || child->elem[0] == '*') && child->elem[1] == '\0'){
- rec = child->elem[0] == '+';
- free(child->new);
- child->new = estrdup(mkaux, me->new);
- setnames(mkaux, child);
- mktree(mkaux, child, rec);
- freefile(child);
- child = getfile(mkaux, me);
- }
- while(child && mkaux->indent > level){
- if(mkfile(mkaux, child))
- domkfs(mkaux, child, mkaux->indent);
- freefile(child);
- child = getfile(mkaux, me);
- }
- if(child){
- freefile(child);
- Bseek(mkaux->b, -Blinelen(mkaux->b), 1);
- mkaux->lineno--;
- }
- }
- static void
- mktree(Mkaux *mkaux, File *me, int rec)
- {
- File child;
- Dir *d;
- int i, n, fd;
- fd = open(mkaux->oldfile.s, OREAD);
- if(fd < 0){
- warn(mkaux, "can't open %s: %r", mkaux->oldfile.s);
- return;
- }
- child = *me;
- while((n = dirread(fd, &d)) > 0){
- for(i = 0; i < n; i++){
- child.new = mkpath(mkaux, me->new, d[i].name);
- if(me->old)
- child.old = mkpath(mkaux, me->old, d[i].name);
- child.elem = d[i].name;
- setnames(mkaux, &child);
- if((!(d[i].mode&DMDIR) || rec) && copyfile(mkaux, &child, &d[i], 1) && rec)
- mktree(mkaux, &child, rec);
- free(child.new);
- if(child.old)
- free(child.old);
- }
- }
- close(fd);
- }
- static int
- mkfile(Mkaux *mkaux, File *f)
- {
- Dir *d;
- if((d = dirstat(mkaux->oldfile.s)) == nil){
- warn(mkaux, "can't stat file %s: %r", mkaux->oldfile.s);
- skipdir(mkaux);
- return 0;
- }
- return copyfile(mkaux, f, d, 0);
- }
- enum {
- SLOP = 30
- };
- static void
- setname(Mkaux *mkaux, Name *name, char *s1, char *s2)
- {
- int l;
- l = strlen(s1)+strlen(s2)+1;
- if(name->n < l+SLOP/2) {
- free(name->s);
- name->s = emalloc(mkaux, l+SLOP);
- name->n = l+SLOP;
- }
- snprint(name->s, name->n, "%s%s%s", s1, s1[0]==0 || s1[strlen(s1)-1]!='/' ? "/" : "", s2);
- }
- static int
- copyfile(Mkaux *mkaux, File *f, Dir *d, int permonly)
- {
- Dir *nd;
- ulong xmode;
- char *p;
- setname(mkaux, &mkaux->fullname, mkaux->root, f->old ? f->old : f->new);
- /*
- * Extra stat here is inefficient but accounts for binds.
- */
- if((nd = dirstat(mkaux->fullname.s)) != nil)
- d = nd;
- d->name = f->elem;
- if(d->type != 'M'){
- d->uid = "sys";
- d->gid = "sys";
- xmode = (d->mode >> 6) & 7;
- d->mode |= xmode | (xmode << 3);
- }
- if(strcmp(f->uid, "-") != 0)
- d->uid = f->uid;
- if(strcmp(f->gid, "-") != 0)
- d->gid = f->gid;
- if(f->mode != ~0){
- if(permonly)
- d->mode = (d->mode & ~0666) | (f->mode & 0666);
- else if((d->mode&DMDIR) != (f->mode&DMDIR))
- warn(mkaux, "inconsistent mode for %s", f->new);
- else
- d->mode = f->mode;
- }
- if(p = strrchr(f->new, '/'))
- d->name = p+1;
- else
- d->name = f->new;
- mkaux->mkenum(f->new, mkaux->fullname.s, d, mkaux->a);
- xmode = d->mode;
- free(nd);
- return (xmode&DMDIR) != 0;
- }
- static char *
- mkpath(Mkaux *mkaux, char *prefix, char *elem)
- {
- char *p;
- int n;
- n = strlen(prefix) + strlen(elem) + 2;
- p = emalloc(mkaux, n);
- strcpy(p, prefix);
- strcat(p, "/");
- strcat(p, elem);
- return p;
- }
- static void
- setnames(Mkaux *mkaux, File *f)
- {
-
- if(f->old){
- if(f->old[0] == '/')
- setname(mkaux, &mkaux->oldfile, f->old, "");
- else
- setname(mkaux, &mkaux->oldfile, mkaux->root, f->old);
- } else
- setname(mkaux, &mkaux->oldfile, mkaux->root, f->new);
- }
- static void
- freefile(File *f)
- {
- if(f->old)
- free(f->old);
- if(f->new)
- free(f->new);
- free(f);
- }
- /*
- * skip all files in the proto that
- * could be in the current dir
- */
- static void
- skipdir(Mkaux *mkaux)
- {
- char *p, c;
- int level;
- if(mkaux->indent < 0)
- return;
- level = mkaux->indent;
- for(;;){
- mkaux->indent = 0;
- p = Brdline(mkaux->b, '\n');
- mkaux->lineno++;
- if(!p){
- mkaux->indent = -1;
- return;
- }
- while((c = *p++) != '\n')
- if(c == ' ')
- mkaux->indent++;
- else if(c == '\t')
- mkaux->indent += 8;
- else
- break;
- if(mkaux->indent <= level){
- Bseek(mkaux->b, -Blinelen(mkaux->b), 1);
- mkaux->lineno--;
- return;
- }
- }
- }
- static File*
- getfile(Mkaux *mkaux, File *old)
- {
- File *f;
- char *elem;
- char *p;
- int c;
- if(mkaux->indent < 0)
- return 0;
- loop:
- mkaux->indent = 0;
- p = Brdline(mkaux->b, '\n');
- mkaux->lineno++;
- if(!p){
- mkaux->indent = -1;
- return 0;
- }
- while((c = *p++) != '\n')
- if(c == ' ')
- mkaux->indent++;
- else if(c == '\t')
- mkaux->indent += 8;
- else
- break;
- if(c == '\n' || c == '#')
- goto loop;
- p--;
- f = emalloc(mkaux, sizeof *f);
- p = getname(mkaux, p, &elem);
- if(p == nil)
- return nil;
- f->new = mkpath(mkaux, old->new, elem);
- free(elem);
- f->elem = utfrrune(f->new, L'/') + 1;
- p = getmode(mkaux, p, &f->mode);
- p = getname(mkaux, p, &f->uid); /* LEAK */
- if(p == nil)
- return nil;
- if(!*f->uid)
- strcpy(f->uid, "-");
- p = getname(mkaux, p, &f->gid); /* LEAK */
- if(p == nil)
- return nil;
- if(!*f->gid)
- strcpy(f->gid, "-");
- f->old = getpath(mkaux, p);
- if(f->old && strcmp(f->old, "-") == 0){
- free(f->old);
- f->old = 0;
- }
- setnames(mkaux, f);
- return f;
- }
- static char*
- getpath(Mkaux *mkaux, char *p)
- {
- char *q, *new;
- int c, n;
- while((c = *p) == ' ' || c == '\t')
- p++;
- q = p;
- while((c = *q) != '\n' && c != ' ' && c != '\t')
- q++;
- if(q == p)
- return 0;
- n = q - p;
- new = emalloc(mkaux, n + 1);
- memcpy(new, p, n);
- new[n] = 0;
- return new;
- }
- static char*
- getname(Mkaux *mkaux, char *p, char **buf)
- {
- char *s, *start;
- int c;
- while((c = *p) == ' ' || c == '\t')
- p++;
- start = p;
- while((c = *p) != '\n' && c != ' ' && c != '\t')
- p++;
- *buf = malloc(p+2-start); /* +2: need at least 2 bytes; might strcpy "-" into buf */
- if(*buf == nil)
- return nil;
- memmove(*buf, start, p-start);
- (*buf)[p-start] = '\0';
- if(**buf == '$'){
- s = getenv(*buf+1);
- if(s == 0){
- warn(mkaux, "can't read environment variable %s", *buf+1);
- skipdir(mkaux);
- free(*buf);
- return nil;
- }
- free(*buf);
- *buf = s;
- }
- return p;
- }
- static char*
- getmode(Mkaux *mkaux, char *p, ulong *xmode)
- {
- char *buf, *s;
- ulong m;
- *xmode = ~0;
- p = getname(mkaux, p, &buf);
- if(p == nil)
- return nil;
- s = buf;
- if(!*s || strcmp(s, "-") == 0)
- return p;
- m = 0;
- if(*s == 'd'){
- m |= DMDIR;
- s++;
- }
- if(*s == 'a'){
- m |= DMAPPEND;
- s++;
- }
- if(*s == 'l'){
- m |= DMEXCL;
- s++;
- }
- if(s[0] < '0' || s[0] > '7'
- || s[1] < '0' || s[1] > '7'
- || s[2] < '0' || s[2] > '7'
- || s[3]){
- warn(mkaux, "bad mode specification %s", buf);
- free(buf);
- return p;
- }
- *xmode = m | strtoul(s, 0, 8);
- free(buf);
- return p;
- }
- static void
- warn(Mkaux *mkaux, char *fmt, ...)
- {
- char buf[256];
- va_list va;
- va_start(va, fmt);
- vseprint(buf, buf+sizeof(buf), fmt, va);
- va_end(va);
- if(mkaux->warn)
- mkaux->warn(buf, mkaux->a);
- else
- fprint(2, "warning: %s\n", buf);
- }
|