#include <fcall.h>

#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;
}