]> git.lizzy.rs Git - plan9front.git/blobdiff - sys/src/cmd/sshfs.c
cc: use 7 octal digits for 21 bit runes
[plan9front.git] / sys / src / cmd / sshfs.c
index eaa32cb9d4d195d70f246e014413319e59bbdb7a..d0c21ec74e0c6dc2e6dae93c62172ea67e4bc7c5 100644 (file)
@@ -7,6 +7,7 @@
 
 int readonly;
 int debug;
+char *root = ".";
 #define dprint(...) if(debug) fprint(2, __VA_ARGS__)
 #pragma        varargck        type    "Σ"    int
 
@@ -171,7 +172,7 @@ idlookup(IDEnt **tab, int id)
        
        for(p = tab[(ulong)id % HASH]; p != nil; p = p->next)
                if(p->id == id)
-                       return strdup(p->name);
+                       return estrdup9p(p->name);
        return smprint("%d", id);
 }
 
@@ -366,19 +367,22 @@ recvpkt(void)
        return rxpkt[0];
 }
 
+void
+freedir1(Dir *d)
+{
+       free(d->name);
+       free(d->uid);
+       free(d->gid);
+       free(d->muid);
+}
+
 void
 freedir(SFid *s)
 {
        int i;
-       Dir *d;
 
-       for(i = 0; i < s->ndirent; i++){
-               d = &s->dirent[i];
-               free(d->name);
-               free(d->uid);
-               free(d->gid);
-               free(d->muid);
-       }
+       for(i = 0; i < s->ndirent; i++)
+               freedir1(&s->dirent[i]);
        free(s->dirent);
        s->dirent = nil;
        s->ndirent = 0;
@@ -435,23 +439,13 @@ submitreq(Req *r)
 char *
 pathcat(char *p, char *c)
 {
-       if(strcmp(p, ".") == 0)
-               return strdup(c);
-       return smprint("%s/%s", p, c);
+       return cleanname(smprint("%s/%s", p, c));
 }
 
 char *
 parentdir(char *p)
 {
-       char *q, *r;
-       
-       if(strcmp(p, ".") == 0) return strdup(".");
-       if(strcmp(p, "/") == 0) return strdup("/");
-       q = strdup(p);
-       r = strrchr(q, '/');
-       if(r != nil) *r = 0;
-       else strcpy(q, ".");
-       return q;
+       return pathcat(p, "..");
 }
 
 char *
@@ -460,8 +454,8 @@ finalelem(char *p)
        char *q;
        
        q = strrchr(p, '/');
-       if(q == nil) return strdup(p);
-       return strdup(q+1);
+       if(q == nil) return estrdup9p(p);
+       return estrdup9p(q+1);
 }
 
 u64int
@@ -474,7 +468,7 @@ qidcalc(char *c)
 }
 
 void
-walkprocess(Req *r, int isdir, char *e)
+walkprocess(Req *r, char *e)
 {
        char *p;
        SFid *sf;
@@ -492,8 +486,6 @@ walkprocess(Req *r, int isdir, char *e)
                submitreq(r);
        }else{
                assert(r->ofcall.nwqid > 0);
-               if(!isdir)
-                       r->ofcall.wqid[r->ofcall.nwqid - 1].type = 0;
                wlock(sf);
                free(sf->fn);
                sf->fn = r->aux;
@@ -523,10 +515,10 @@ attrib2dir(uchar *p0, uchar *ep, Dir *d)
                d->uid = idlookup(uidtab, uid);
                d->gid = idlookup(gidtab, gid);
        }else{
-               d->uid = strdup("sshfs");
-               d->gid = strdup("sshfs");
+               d->uid = estrdup9p("sshfs");
+               d->gid = estrdup9p("sshfs");
        }
-       d->muid = strdup(d->uid);
+       d->muid = estrdup9p(d->uid);
        if((flags & SSH_FILEXFER_ATTR_PERMISSIONS) != 0){
                rc = unpack(p, ep - p, "u", &perm); if(rc < 0) return -1; p += rc;
                d->mode = perm & 0777;
@@ -535,6 +527,7 @@ attrib2dir(uchar *p0, uchar *ep, Dir *d)
        d->qid.type = d->mode >> 24;
        if((flags & SSH_FILEXFER_ATTR_ACMODTIME) != 0){
                rc = unpack(p, ep - p, "uu", &d->atime, &d->mtime); if(rc < 0) return -1; p += rc;
+               d->qid.vers = d->mtime;
        }
        if((flags & SSH_FILEXFER_ATTR_EXTENDED) != 0){
                rc = unpack(p, ep - p, "u", &next); if(rc < 0) return -1; p += rc;
@@ -553,7 +546,7 @@ dir2attrib(Dir *d, uchar **rp)
        uchar *r, *p, *e;
        u32int fl;
        int uid, gid;
-       
+
        werrstr("phase error");
        r = emalloc9p(MAXATTRIB);
        e = r + MAXATTRIB;
@@ -595,21 +588,27 @@ dir2attrib(Dir *d, uchar **rp)
 }
 
 int
-attribisdir(char *fn)
+attrfixupqid(Qid *qid)
 {
-       u32int code;
+       u32int flags;
        uchar *p;
-       
-       if(unpack(rxpkt, rxlen, "_____u", &code) < 0) return -1;
-       if((code & 4) == 0){
-               fprint(2, "sshfs: can't determine if %s is a directory\n", fn);
-               return 1;
-       }
+
+       if(unpack(rxpkt, rxlen, "_____u", &flags) < 0) return -1;
        p = rxpkt + 9;
-       if(code & 1) p += 8;
-       if(code & 2) p += 8;
-       if(p + 4 > rxpkt + rxlen) return -1;
-       return (GET4(p) & 0170000) == 0040000;
+       if(flags & SSH_FILEXFER_ATTR_SIZE) p += 8;
+       if(flags & SSH_FILEXFER_ATTR_UIDGID) p += 8;
+       if(flags & SSH_FILEXFER_ATTR_PERMISSIONS){
+               if(p + 4 > rxpkt + rxlen) return -1;
+               if((GET4(p) & 0170000) != 0040000) qid->type = 0;
+               else qid->type = QTDIR;
+               p += 4;
+       }
+       if(flags & SSH_FILEXFER_ATTR_ACMODTIME){
+               if(p + 8 > rxpkt + rxlen) return -1;
+               p += 4;
+               qid->vers = GET4(p);    /* mtime for qid.vers */
+       }
+       return 0;
 }
 
 int
@@ -631,17 +630,16 @@ parsedir(SFid *sf)
        p = rxpkt + 9;
        ep = rxpkt + rxlen;
        for(i = 0; i < c; i++){
-               rc = unpack(p, ep - p, "ss", &fn, &fns, &ln, &lns); if(rc < 0) goto err; p += rc;
                memset(d, 0, sizeof(Dir));
+               rc = unpack(p, ep - p, "ss", &fn, &fns, &ln, &lns); if(rc < 0) goto err; p += rc;
                rc = attrib2dir(p, ep, d); if(rc < 0) goto err; p += rc;
                if(fn[0] == '.' && (fns == 1 || fns == 2 && fn[1] == '.')){
-                       free(d->uid);
-                       free(d->gid);
-                       free(d->muid);
+                       freedir1(d);
                        continue;
                }
                d->name = emalloc9p(fns + 1);
                memcpy(d->name, fn, fns);
+               d->name[fns] = 0;
                s = pathcat(sf->fn, d->name);
                d->qid.path = qidcalc(s);
                free(s);
@@ -651,6 +649,7 @@ parsedir(SFid *sf)
        wunlock(sf);
        return 0;
 err:
+       freedir1(d);
        wunlock(sf);
        return -1;
 }
@@ -724,20 +723,40 @@ sshfsattach(Req *r)
 {
        SFid *sf;
 
-       if(r->ifcall.aname != nil && *r->ifcall.aname != 0 && r->aux == nil){
+       if(r->aux == nil){
+               sf = emalloc9p(sizeof(SFid));
+               if(r->ifcall.aname != nil)
+                       switch(*r->ifcall.aname){
+                       case '~':
+                               switch(r->ifcall.aname[1]){
+                               case 0: sf->fn = estrdup9p("."); break;
+                               case '/': sf->fn = estrdup9p(r->ifcall.aname + 2); break;
+                               default:
+                                       free(sf);
+                                       respond(r, "invalid attach name");
+                                       return;
+                               }
+                               break;
+                       case '/':
+                               sf->fn = estrdup9p(r->ifcall.aname);
+                               break;
+                       case 0:
+                               sf->fn = estrdup9p(root);
+                               break;
+                       default:
+                               sf->fn = pathcat(root, r->ifcall.aname);
+                       }
+               else
+                       sf->fn = estrdup9p(root);
+               r->fid->aux = sf;
                submitreq(r);
-               return;
+       }else{
+               sf = r->fid->aux;
+               sf->qid = (Qid){qidcalc(sf->fn), 0, QTDIR};
+               r->ofcall.qid = sf->qid;
+               r->fid->qid = sf->qid;
+               respond(r, nil);
        }
-       sf = emalloc9p(sizeof(SFid));
-       if(r->ifcall.aname != nil && *r->ifcall.aname != 0)
-               sf->fn = strdup(r->ifcall.aname);
-       else
-               sf->fn = strdup(".");
-       sf->qid = (Qid){qidcalc(sf->fn), 0, QTDIR};
-       r->ofcall.qid = sf->qid;
-       r->fid->qid = sf->qid;
-       r->fid->aux = sf;
-       respond(r, nil);
 }
 
 void
@@ -784,7 +803,7 @@ sendproc(void *)
                sf = r->req->fid != nil ? r->req->fid->aux : nil;
                switch(r->req->ifcall.type){
                case Tattach:
-                       sendpkt("bus", SSH_FXP_STAT, r->reqid, r->req->ifcall.aname, strlen(r->req->ifcall.aname));
+                       sendpkt("bus", SSH_FXP_STAT, r->reqid, sf->fn, strlen(sf->fn));
                        break;
                case Twalk:
                        sendpkt("bus", SSH_FXP_STAT, r->reqid, r->req->aux, strlen(r->req->aux));
@@ -853,8 +872,8 @@ sendproc(void *)
                                }else if(r->req->aux == (void*)-2){
                                        sendpkt("bus", SSH_FXP_OPENDIR, r->reqid, sf->fn, strlen(sf->fn));
                                }else{
-                                       sendpkt("bus", SSH_FXP_READDIR, r->reqid, sf->hand, sf->handn);
                                        sf->dirreads++;
+                                       sendpkt("bus", SSH_FXP_READDIR, r->reqid, sf->hand, sf->handn);
                                }
                                wunlock(sf);
                        }else{
@@ -890,9 +909,9 @@ sendproc(void *)
                                rlock(sf);
                                s = parentdir(sf->fn);
                                t = pathcat(s, r->req->d.name);
-                               sendpkt("buss", SSH_FXP_RENAME, r->reqid, sf->fn, strlen(sf->fn), t, strlen(t));
                                free(s);
                                r->req->aux = t;
+                               sendpkt("buss", SSH_FXP_RENAME, r->reqid, sf->fn, strlen(sf->fn), t, strlen(t));
                                runlock(sf);
                                break;
                        }
@@ -908,6 +927,7 @@ sendproc(void *)
                        else
                                sendpkt("bus[", SSH_FXP_SETSTAT, r->reqid, sf->fn, strlen(sf->fn), s, x);
                        runlock(sf);
+                       free(s);
                        break;
                case Tremove:
                        rlock(sf);
@@ -932,9 +952,9 @@ recvproc(void *)
 
        SReq *r;
        SFid *sf;
-       int t, id, rc;
+       int t, id;
        u32int code;
-       char *msg, *lang, *hand;
+       char *msg, *lang, *hand, *s;
        int msgn, langn, handn;
        int okresp;
        char *e;
@@ -983,20 +1003,20 @@ recvproc(void *)
                switch(r->req->ifcall.type){
                case Tattach:
                        if(t != SSH_FXP_ATTRS) goto common;
-                       rc = attribisdir(r->req->ifcall.aname);
-                       r->req->aux = (void*)-1;
-                       if(rc < 0)
+                       if(attrfixupqid(&r->req->ofcall.qid) < 0)
                                goto garbage;
-                       if(rc == 0)
+                       r->req->aux = (void*)-1;
+                       if((r->req->ofcall.qid.type & QTDIR) == 0)
                                respond(r->req, "not a directory");
                        else
                                sshfsattach(r->req);
                        break;
                case Twalk:
                        if(t != SSH_FXP_ATTRS) goto common;
-                       rc = attribisdir(((SFid*)r->req->fid)->fn);
-                       if(rc < 0) goto garbage;
-                       walkprocess(r->req, rc, nil);
+                       if(r->req->ofcall.nwqid <= 0
+                       || attrfixupqid(&r->req->ofcall.wqid[r->req->ofcall.nwqid - 1]) < 0)
+                               goto garbage;
+                       walkprocess(r->req, nil);
                        break;
                case Tcreate:
                        if(okresp && r->req->aux == (void*)-1){
@@ -1011,6 +1031,14 @@ recvproc(void *)
                        sf->handn = handn;
                        sf->hand = emalloc9p(sf->handn);
                        memcpy(sf->hand, hand, sf->handn);
+                       if(r->req->ifcall.type == Tcreate){
+                               s = sf->fn;
+                               sf->fn = pathcat(s, r->req->ifcall.name);
+                               free(s);
+                               sf->qid = (Qid){qidcalc(sf->fn), 0, (r->req->ifcall.perm & DMDIR) != 0 ? QTDIR : 0};
+                               r->req->ofcall.qid = sf->qid;
+                               r->req->fid->qid = sf->qid;
+                       }
                        wunlock(sf);
                        if(r->req->ifcall.type == Tread){
                                r->req->aux = nil;
@@ -1061,6 +1089,10 @@ recvproc(void *)
                        break;
                case Twstat:
                        if(!okresp) goto common;
+                       if(!r->req->d.name[0]){
+                               respond(r->req, nil);
+                               break;
+                       }
                        if(r->req->aux == nil){
                                r->req->aux = (void *) -1;
                                submitreq(r->req);
@@ -1116,7 +1148,7 @@ recvproc(void *)
                        fprint(2, "sshfs: received unexpected packet %Σ for 9p request %F\n", t, &r->req->ifcall);
                }
                if(r->req->ifcall.type == Twalk)
-                       walkprocess(r->req, 0, e);
+                       walkprocess(r->req, e);
                else
                        respond(r->req, e);
                putsreq(r);
@@ -1135,7 +1167,7 @@ sshfswalk(Req *r)
                r->newfid->qid = r->fid->qid;
                s = r->fid->aux;
                t = emalloc9p(sizeof(SFid));
-               t->fn = strdup(s->fn);
+               t->fn = estrdup9p(s->fn);
                t->qid = s->qid;
                r->newfid->aux = t;
        }else
@@ -1144,12 +1176,9 @@ sshfswalk(Req *r)
                respond(r, nil);
                return;
        }
-       p = strdup(t->fn);
+       p = estrdup9p(t->fn);
        for(i = 0; i < r->ifcall.nwname; i++){
-               if(strcmp(r->ifcall.wname[i], "..") == 0)
-                       q = parentdir(p);
-               else
-                       q = pathcat(p, r->ifcall.wname[i]);
+               q = pathcat(p, r->ifcall.wname[i]);
                free(p);
                p = q;
                r->ofcall.wqid[i] = (Qid){qidcalc(p), 0, QTDIR};
@@ -1184,6 +1213,13 @@ sshfsdestroyreq(Req *r)
                free(r->aux);
 }
 
+void
+sshfsstart(Srv *)
+{
+       proccreate(sendproc, nil, mainstacksize);
+       proccreate(recvproc, nil, mainstacksize);
+}
+
 void
 sshfsend(Srv *)
 {
@@ -1192,6 +1228,7 @@ sshfsend(Srv *)
 }
 
 Srv sshfssrv = {
+       .start sshfsstart,
        .attach sshfsattach,
        .walk sshfswalk,
        .open submitreq,
@@ -1203,7 +1240,7 @@ Srv sshfssrv = {
        .remove submitreq,
        .destroyfid sshfsdestroyfid,
        .destroyreq sshfsdestroyreq,
-       .end sshfsend
+       .end sshfsend,
 };
 
 char *
@@ -1269,7 +1306,7 @@ passwdparse(IDEnt **tab, char *s)
                if(p == nil) break;
                p++;
                e = emalloc9p(sizeof(IDEnt));
-               e->name = strdup(n);
+               e->name = estrdup9p(n);
                e->id = id;
                b = &tab[((ulong)e->id) % HASH];
                e->next = *b;
@@ -1303,7 +1340,7 @@ void
 usage(void)
 {
        static char *common = "[-abdRUG] [-s service] [-m mtpt] [-u uidfile] [-g gidfile]";
-       fprint(2, "usage: %s %s [-- ssh-options] host\n", argv0, common);
+       fprint(2, "usage: %s %s [-- ssh-options] [user@]host\n", argv0, common);
        fprint(2, "       %s %s -c cmdline\n", argv0, common);
        fprint(2, "       %s %s -p\n", argv0, common);
        exits("usage");
@@ -1332,10 +1369,12 @@ threadmain(int argc, char **argv)
        case 'a': mflag |= MAFTER; break;
        case 'b': mflag |= MBEFORE; break;
        case 'm': mtpt = EARGF(usage()); break;
+       case 'M': mtpt = nil; break;
        case 'u': uidfile = EARGF(usage()); break;
        case 'U': uidfile = nil; break;
        case 'g': gidfile = EARGF(usage()); break;
        case 'G': gidfile = nil; break;
+       case 'r': root = EARGF(usage()); break;
        default: usage();
        }ARGEND;
        
@@ -1363,7 +1402,7 @@ threadmain(int argc, char **argv)
                }
                pipe(pfd);
                rdfd = wrfd = pfd[0];
-               procrfork(startssh, nil, mainstacksize, RFFDG|RFNOTEG);
+               procrfork(startssh, nil, mainstacksize, RFFDG|RFNOTEG|RFNAMEG);
                close(pfd[1]);
        }
 
@@ -1374,7 +1413,7 @@ threadmain(int argc, char **argv)
        passwdparse(uidtab, readfile(uidfile));
        passwdparse(gidtab, readfile(gidfile));
        
-       procrfork(sendproc, 0, mainstacksize, RFNOTEG);
-       procrfork(recvproc, 0, mainstacksize, RFNOTEG);
        threadpostmountsrv(&sshfssrv, svc, mtpt, MCREATE | mflag);
+
+       exits(nil);
 }