Subversion Repositories planix.SVN

#include <u.h>
#include <libc.h>
#include <fcall.h>
#include <thread.h>
#include <9p.h>
#include "cifs.h"

static Pkt *
tnthdr(Session *s, Share *sp, int cmd)
{
        Pkt *p;

        p = cifshdr(s, sp, SMB_COM_NT_TRANSACT);
        p->tbase = p8(p, 0);            /*  0  Max setup count to return */
        pl16(p, 0);                     /*  1  reserved */
        pl32(p, 0);                     /*  3  Total parameter count */
        pl32(p, 0);                     /*  7  Total data count */
        pl32(p, 64);                    /* 11  Max parameter count to return */
        pl32(p, (MTU - T2HDRLEN)-64);   /* 15  Max data count to return */
        pl32(p, 0);                     /* 19  Parameter count (in this buffer) */
        pl32(p, 0);                     /* 23  Offset to parameters (in this buffer) */
        pl32(p, 0);                     /* 27  Count of data  in this buffer */
        pl32(p, 0);                     /* 31  Offset to data in this buffer */
        p8(p, 1);                       /* 35  Count of setup words */
        pl16(p, cmd);                   /* 37  setup[0] */
        pl16(p, 0);                     /* padding ??!?!? */
        pbytes(p);
        return p;
}

static void
ptntparam(Pkt *p)
{
        uchar *pos = p->pos;
        assert(p->tbase != 0);

        p->pos = p->tbase +23;
        pl32(p, (pos - p->buf) - NBHDRLEN); /* param offset */

        p->tparam = p->pos = pos;
}

static void
ptntdata(Pkt *p)
{
        uchar *pos = p->pos;
        assert(p->tbase != 0);
        assert(p->tparam != 0);

        p->pos = p->tbase +3;
        pl32(p, pos - p->tparam);               /* total param count */

        p->pos = p->tbase +19;
        pl32(p, pos - p->tparam);               /* param count */

        p->pos = p->tbase +31;
        pl32(p, (pos - p->buf) - NBHDRLEN);     /* data offset */
        p->tdata = p->pos = pos;
}

static int
tntrpc(Pkt *p)
{
        int got;
        uchar *pos;
        assert(p->tbase != 0);
        assert(p->tdata != 0);

        pos = p->pos;

        p->pos = p->tbase +7;
        pl32(p, pos - p->tdata);                /* total data count */

        p->pos = p->tbase +27;
        pl32(p, pos - p->tdata);                /* data count */

        p->pos = pos;
        if((got = cifsrpc(p)) == -1)
                return -1;

        g8(p);                          /* Reserved */
        g8(p);                          /* Reserved */
        g8(p);                          /* Reserved */
        gl32(p);                        /* Total parameter count */
        gl32(p);                        /* Total data count */
        gl32(p);                        /* Parameter count in this buffer */
        p->tparam = p->buf +NBHDRLEN +gl32(p); /* Parameter offset */
        gl32(p);                        /* Parameter displacement */
        gl32(p);                        /* Data count (this buffer); */
        p->tdata = p->buf +NBHDRLEN +gl32(p); /* Data offset */
        gl32(p);                        /* Data displacement */
        g8(p);                          /* Setup count */
        gl16(p);                        /* padding ???  */

        return got;
}

static void
gtntparam(Pkt *p)
{
        p->pos = p->tparam;
}

static void
gtntdata(Pkt *p)
{
        p->pos = p->tdata;
}


int
TNTquerysecurity(Session *s, Share *sp, int fh, char **usid, char **gsid)
{
        Pkt *p;
        uchar *base;
        Fmt fmt, *f = &fmt;
        int n, i, off2owner, off2group;

        p = tnthdr(s, sp, NT_TRANSACT_QUERY_SECURITY_DESC);
        ptntparam(p);

        pl16(p, fh);            /* File handle */
        pl16(p, 0);             /* Reserved */
        pl32(p, QUERY_OWNER_SECURITY_INFORMATION |
                QUERY_GROUP_SECURITY_INFORMATION);

        ptntdata(p);

        if(tntrpc(p) == -1){
                free(p);
                return -1;
        }

        gtntdata(p);

        base = p->pos;
        gl16(p);                        /* revision */
        gl16(p);                        /* type */
        off2owner = gl32(p);            /* offset to owner */
        off2group = gl32(p);            /* offset to group */
        gl32(p);
        gl32(p);

        if(off2owner){
                p->pos = base +  off2owner;
                fmtstrinit(f);
                fmtprint(f, "S-%ud", g8(p));    /* revision */
                n = g8(p);                      /* num auth */
                fmtprint(f, "-%llud", gb48(p)); /* authority */
                for(i = 0; i < n; i++)
                        fmtprint(f, "-%ud", gl32(p));   /* sub-authorities */
                *usid = fmtstrflush(f);
        }

        if(off2group){
                p->pos = base +  off2group;
                fmtstrinit(f);
                fmtprint(f, "S-%ud", g8(p));    /* revision */
                n = g8(p);                      /* num auth */
                fmtprint(f, "-%llud", gb48(p)); /* authority */
                for(i = 0; i < n; i++)
                        fmtprint(f, "-%ud", gl32(p));   /* sub-authorities */
                *gsid = fmtstrflush(f);
        }
        free(p);
        return 0;
}