Subversion Repositories planix.SVN

#include "headers.h"

static SmbProcessResult
query(SmbSession *s, char *cmdname, char *filename, ushort infolevel, vlong cbo, Dir *d)
{
        vlong ntatime, ntmtime;
        ushort dosmode;
        ulong fnlfixupoffset;
        vlong allocsize;

        if (d == nil) {
                smbseterror(s, ERRDOS, ERRbadfile);
                return SmbProcessResultError;
        }

        switch (infolevel) {
        case SMB_QUERY_FILE_BASIC_INFO:
                ntatime = smbplan9time2time(d->atime);
                ntmtime = smbplan9time2time(d->mtime);
                dosmode = smbplan9mode2dosattr(d->mode);

                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FILE_BASIC_INFO\n");
                translogprint(s->transaction.in.setup[0], "REPLY:\n");
                translogprint(s->transaction.in.setup[0], "atime=%s", ctime(d->atime));
                translogprint(s->transaction.in.setup[0], "atime=%s", ctime(d->mtime));
                translogprint(s->transaction.in.setup[0], "mode=0%o -> dosmode=0x%x\n", d->mode, dosmode);

                if (!smbbufferputv(s->transaction.out.data, ntmtime)
                        || !smbbufferputv(s->transaction.out.data, ntatime)
                        || !smbbufferputv(s->transaction.out.data, ntmtime)
                        || !smbbufferputv(s->transaction.out.data, ntmtime)
                        || !smbbufferputl(s->transaction.out.data, dosmode))
                        return SmbProcessResultMisc;
                break;
        case SMB_QUERY_FILE_ALL_INFO:
                ntatime = smbplan9time2time(d->atime);
                ntmtime = smbplan9time2time(d->mtime);
                dosmode = smbplan9mode2dosattr(d->mode);
                allocsize = (d->length + (1 << smbglobals.l2allocationsize) - 1) & ~((1 << smbglobals.l2allocationsize) - 1);

                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FILE_ALL_INFO\n");
                translogprint(s->transaction.in.setup[0], "REPLY:\n");
                translogprint(s->transaction.in.setup[0], "atime=%s", ctime(d->atime));
                translogprint(s->transaction.in.setup[0], "atime=%s", ctime(d->mtime));
                translogprint(s->transaction.in.setup[0], "mode=0%o -> dosmode=0x%x\n", d->mode, dosmode);
                translogprint(s->transaction.in.setup[0], "allocsize=%d\n", allocsize);
                translogprint(s->transaction.in.setup[0], "isdir=%d\n", (d->mode & DMDIR) != 0);

                if (!smbbufferputv(s->transaction.out.data, ntmtime)
                        || !smbbufferputv(s->transaction.out.data, ntatime)
                        || !smbbufferputv(s->transaction.out.data, ntmtime)
                        || !smbbufferputv(s->transaction.out.data, ntmtime)
                        || !smbbufferputs(s->transaction.out.data, dosmode)
                        || !smbbufferputbytes(s->transaction.out.data, nil, 6)
                        || !smbbufferputv(s->transaction.out.data, allocsize)
                        || !smbbufferputv(s->transaction.out.data, d->length)
                        || !smbbufferputl(s->transaction.out.data, 0)           // hard links - ha
                        || !smbbufferputb(s->transaction.out.data, 0)           // TODO delete pending
                        || !smbbufferputb(s->transaction.out.data, (d->mode & DMDIR) != 0)
                        || !smbbufferputv(s->transaction.out.data, d->qid.path)
                        || !smbbufferputl(s->transaction.out.data, 0)           // EA size
                        || !smbbufferputl(s->transaction.out.data, (dosmode & SMB_ATTR_READ_ONLY) ? 0xa1 : 0x1a7)
                        || !smbbufferputv(s->transaction.out.data, cbo)
                        || !smbbufferputs(s->transaction.out.data, dosmode)
                        || !smbbufferputl(s->transaction.out.data, 0))  // alignment
                        return SmbProcessResultMisc;
                fnlfixupoffset = smbbufferwriteoffset(s->transaction.out.data);
                if (!smbbufferputl(s->transaction.out.data, 0)
                    || !smbbufferputstring(s->transaction.out.data, &s->peerinfo, SMB_STRING_REVPATH, filename)
                    || !smbbufferfixuprelativel(s->transaction.out.data, fnlfixupoffset))
                        return SmbProcessResultMisc;
                break;
        case SMB_QUERY_FILE_STANDARD_INFO:
                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FILE_STANDARD_INFO\n");
                translogprint(s->transaction.in.setup[0], "REPLY:\n");
                translogprint(s->transaction.in.setup[0], "length=%lld", d->length);
                translogprint(s->transaction.in.setup[0], "isdir=%d\n", (d->qid.type & QTDIR) != 0);

                if (!smbbufferputv(s->transaction.out.data, smbl2roundupvlong(d->length, smbglobals.l2allocationsize))
                        || !smbbufferputv(s->transaction.out.data, d->length)
                        || !smbbufferputl(s->transaction.out.data, 1)
                        || !smbbufferputb(s->transaction.out.data, 0)
                        || !smbbufferputb(s->transaction.out.data, (d->qid.type & QTDIR) != 0))
                        return SmbProcessResultMisc;
                break;
        case SMB_QUERY_FILE_EA_INFO:
                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FILE_EA_INFO\n");
                translogprint(s->transaction.in.setup[0], "REPLY:\n");
                translogprint(s->transaction.in.setup[0], "ea_len=0\n");
                if (!smbbufferputl(s->transaction.out.data, 0))
                        return SmbProcessResultMisc;
                break;
        case SMB_QUERY_FILE_STREAM_INFO:
                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FILE_STREAM_INFO\n");
                translogprint(s->transaction.in.setup[0], "REPLY: failed\n");
                /* don't do it, never will */
                goto unknownlevel;
        default:
                smblogprint(-1, "smbtrans2query%sinformation: infolevel 0x%.4ux not implemented\n", cmdname, infolevel);
        unknownlevel:
                translogprint(s->transaction.in.setup[0], "[not supported]\n");
                smbseterror(s, ERRDOS, ERRunknownlevel);
                return SmbProcessResultError;
        }
        return SmbProcessResultReply;
}

SmbProcessResult
smbtrans2querypathinformation(SmbSession *s, SmbHeader *h)
{
        SmbTree *t;
        SmbBuffer *b = nil;
        SmbProcessResult pr;
        ushort infolevel;
        Dir *d;
        char *path = nil;
        char *fullpath;

        t = smbidmapfind(s->tidmap, h->tid);
        if (t == nil) {
                smbseterror(s, ERRSRV, ERRinvtid);
                pr = SmbProcessResultError;
                goto done;
        }
        b = smbbufferinit(s->transaction.in.parameters, s->transaction.in.parameters, s->transaction.in.tpcount);
        if (!smbbuffergets(b, &infolevel) || !smbbuffergetbytes(b, nil, 4)
                || !smbbuffergetstring(b, h, SMB_STRING_PATH, &path)) {
                pr = SmbProcessResultMisc;
                goto done;
        }
        translogprint(s->transaction.in.setup[0], "infolevel 0x%.4ux\n", infolevel);
        translogprint(s->transaction.in.setup[0], "path %s\n", path);
        fullpath = nil;
        smbstringprint(&fullpath, "%s%s", t->serv->path, path);
        translogprint(s->transaction.in.setup[0], "fullpath %s\n", fullpath);
        d = dirstat(fullpath);
        pr = query(s, "path", path, infolevel, 0, d);
        free(d);
        free(fullpath);
done:
        free(path);
        smbbufferfree(&b);
        return pr;
}

SmbProcessResult
smbtrans2queryfileinformation(SmbSession *s, SmbHeader *h)
{
        SmbTree *t;
        SmbFile *f;
        SmbBuffer *b = nil;
        SmbProcessResult pr;
        ushort fid;
        ushort infolevel;
        vlong o;
        Dir *d;

        t = smbidmapfind(s->tidmap, h->tid);
        if (t == nil) {
                smbseterror(s, ERRSRV, ERRinvtid);
                pr = SmbProcessResultError;
                goto done;
        }
        b = smbbufferinit(s->transaction.in.parameters, s->transaction.in.parameters, s->transaction.in.tpcount);
        if (!smbbuffergets(b, &fid) || !smbbuffergets(b, &infolevel)) {
                pr = SmbProcessResultMisc;
                goto done;
        }
        translogprint(s->transaction.in.setup[0], "fid 0x%.4ux\n", fid);
        translogprint(s->transaction.in.setup[0], "infolevel 0x%.4ux\n", infolevel);
        f = smbidmapfind(s->fidmap, fid);
        if (f == nil) {
                smbseterror(s, ERRDOS, ERRbadfid);
                pr = SmbProcessResultError;
                goto done;
        }
        if(f->fd >= 0){
                o = seek(f->fd, 0, 1);
                d = dirfstat(f->fd);
        } else {
                char *fullpath = nil;

                o = 0;
                smbstringprint(&fullpath, "%s%s", f->t->serv->path, f->name);
                d = dirstat(fullpath);
                free(fullpath);
        }
        pr = query(s, "file", f->name, infolevel, o, d);
        free(d);
done:
        smbbufferfree(&b);
        return pr;
}

SmbProcessResult
smbtrans2queryfsinformation(SmbSession *s, SmbHeader *h)
{
        SmbTree *t;
        ushort infolevel;
        SmbBuffer *b;
        SmbProcessResult pr;
        ulong fixup;
        ulong vnbase;

        t = smbidmapfind(s->tidmap, h->tid);
        if (t == nil) {
                smbseterror(s, ERRSRV, ERRinvtid);
                pr = SmbProcessResultError;
                goto done;
        }
        b = smbbufferinit(s->transaction.in.parameters, s->transaction.in.parameters, s->transaction.in.tpcount);
        if (!smbbuffergets(b, &infolevel)) {
        misc:
                pr = SmbProcessResultMisc;
                goto done;
        }
        pr = SmbProcessResultReply;
        switch (infolevel) {
        case SMB_INFO_ALLOCATION:
                translogprint(s->transaction.in.setup[0], "SMB_INFO_ALLOCATION\n");
                if (!smbbufferputl(s->transaction.out.data, 0)
                        || !smbbufferputl(s->transaction.out.data, 1 << (smbglobals.l2allocationsize - smbglobals.l2sectorsize))
                        || !smbbufferputl(s->transaction.out.data, 0xffffffff)
                        || !smbbufferputl(s->transaction.out.data, 0xffffffff)
                        || !smbbufferputs(s->transaction.out.data, 1 << smbglobals.l2sectorsize))
                        goto misc;
                break;
        case SMB_INFO_VOLUME:
                translogprint(s->transaction.in.setup[0], "SMB_INFO_VOLUME\n");
                if (!smbbufferputl(s->transaction.out.data, 0xdeadbeef)
                        || !smbbufferputstring(s->transaction.out.data, &s->peerinfo, 0, t->serv->name))
                        goto misc;
                break;
        case SMB_QUERY_FS_VOLUME_INFO:
                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FS_VOLUME_INFO\n");
                if (!smbbufferputv(s->transaction.out.data, 0)
                        || !smbbufferputl(s->transaction.out.data, 0xdeadbeef))
                        goto misc;
                fixup = smbbufferwriteoffset(s->transaction.out.data);
                if (!smbbufferputl(s->transaction.out.data, 0)
                        || !smbbufferputs(s->transaction.out.data, 0))
                        goto misc;
                vnbase = smbbufferwriteoffset(s->transaction.out.data);
                if (!smbbufferputstring(s->transaction.out.data, &s->peerinfo, 0, t->serv->name)
                        || !smbbufferfixupl(s->transaction.out.data, fixup,
                                smbbufferwriteoffset(s->transaction.out.data) - vnbase))
                        goto misc;
                break;
        case SMB_QUERY_FS_SIZE_INFO:
                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FS_SIZE_INFO\n");
                if (!smbbufferputv(s->transaction.out.data, 0LL)
                        || !smbbufferputv(s->transaction.out.data, 0LL)
                        || !smbbufferputl(s->transaction.out.data, 1 << (smbglobals.l2allocationsize - smbglobals.l2sectorsize))
                        || !smbbufferputl(s->transaction.out.data, 1 << smbglobals.l2sectorsize))
                        goto misc;
                break;
        case SMB_QUERY_FS_ATTRIBUTE_INFO:
                translogprint(s->transaction.in.setup[0], "SMB_QUERY_FS_ATTRIBUTE_INFO\n");
                if (!smbbufferputl(s->transaction.out.data, 3)
                        || !smbbufferputl(s->transaction.out.data, 255))
                        goto misc;
                fixup = smbbufferwriteoffset(s->transaction.out.data);
                if (!smbbufferputl(s->transaction.out.data, 0)
                        || !smbbufferputstring(s->transaction.out.data, &s->peerinfo, SMB_STRING_UNTERMINATED, smbglobals.serverinfo.nativelanman)
                        || !smbbufferfixuprelativel(s->transaction.out.data, fixup))
                        goto misc;
                break;
        default:
                smblogprint(-1, "smbtrans2queryfsinformation: infolevel 0x%.4ux not implemented\n", infolevel);
                smbseterror(s, ERRDOS, ERRunknownlevel);
                pr = SmbProcessResultError;
        }
done:
        smbbufferfree(&b);
        return pr;
}