Subversion Repositories planix.SVN

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