#include <u.h>
#include <libc.h>
#include <ip.h>
#include "dat.h"
#include "protos.h"

typedef struct {
       uchar   res;
       uchar   cmd;
       uchar   err;
       uchar   cnt;
} Hdr;

enum {
       Ocmd,
       Oerr,
       Ocnt,

       Hsize   = 4,
};

static Field p_fields[] =
{
       { "cmd",        Fnum,   Ocmd,   "command", },
       { "err",        Fnum,   Oerr,   "error", },
       { "cnt",        Fnum,   Ocnt,   "count", },
       nil
};

static Mux p_mux[] = {
       { "aoemd",      0 },
       { "aoemd",      1 },
       nil
};

static void
p_compile(Filter *f)
{
       Mux *m;

       if(f->op == '='){
               compile_cmp(aoerr.name, f, p_fields);
               return;
       }
       for(m = p_mux; m->name; m++)
               if(strcmp(f->s, m->name) == 0){
                       f->pr = m->pr;
                       f->ulv = m->val;
                       f->subop = Ocmd;
                       return;
               }
       sysfatal("unknown aoemask field: %s", f->s);
}

static int
p_filter(Filter *f, Msg *m)
{
       Hdr *h;

       if(m->pe - m->ps < Hsize)
               return 0;

       h = (Hdr*)m->ps;
       m->ps += Hsize;

       switch(f->subop){
       case Ocmd:
               return h->cmd == f->ulv;
       case Oerr:
               return h->err == f->ulv;
       case Ocnt:
               return h->cnt == f->ulv;
       }
       return 0;
}

static char *ctab[] = {
       "read",
       "edit",
};

static char *etab[] = {
       "",
       "bad",
       "full",
};

static int
p_seprint(Msg *m)
{
       char *s, *t;
       Hdr *h;

       if(m->pe - m->ps < Hsize)
               return 0;

       h = (Hdr*)m->ps;
       m->ps += Hsize;

       demux(p_mux, h->cmd, h->cmd, m, &dump);

       s = "unk";
       if(h->cmd < nelem(ctab))
               s = ctab[h->cmd];
       t = "unk";
       if(h->err < nelem(etab))
               s = etab[h->err];
       m->p = seprint(m->p, m->e, "cmd=%d %s err=%d %s cnt=%d\n",
               h->cmd, s, h->err, t, h->cnt);
       return 0;
}

Proto aoemask = {
       "aoemask",
       p_compile,
       p_filter,
       p_seprint,
       p_mux,
       "%lud",
       p_fields,
       defaultframer,
};