ref: 34f5a8333090c5090e1d0b6ddbaf900d64340eea
dir: /sys/src/cmd/atazz/main.c/
#include <u.h> #include <libc.h> #include <fis.h> #include "atazz.h" #include "tabs.h" #pragma varargck argpos eprint 1 #pragma varargck type "π" char** enum { Dontread = -2, }; int interrupted; int rflag; int squelch; int scttrace; uchar issuetr[0x100]; Atatab *idcmd; Atatab *idpktcmd; Atatab *sigcmd; Atatab *sctread; Atatab *sctissue; int πfmt(Fmt *f) { char **p; p = va_arg(f->args, char**); if(p == nil) return fmtstrcpy(f, "<nil**>"); for(; *p; p++){ fmtstrcpy(f, *p); if(p[1] != nil) fmtstrcpy(f, " "); } return 0; } int eprint(char *fmt, ...) { int n; va_list args; if(squelch) return 0; // Bflush(&out); va_start(args, fmt); n = vfprint(2, fmt, args); va_end(args); return n; } void fisset(Req *r, uint i, uint v) { if(r->fisbits & 1<<i) return; r->fisbits |= 1<<i; r->cmd.fis[i] = v; } void prreq(Req *r) { uchar *u; print("%.2ux %.2ux\n", r->cmd.sdcmd, r->cmd.ataproto); u = r->cmd.fis; print("%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux ", u[0], u[1], u[2], u[3], u[4], u[5], u[6], u[7]); u += 8; print("%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux\n", u[0], u[1], u[2], u[3], u[4], u[5], u[6], u[7]); } char* protostr(char *p, char *e, int pr) { char *s; *p = 0; if(pr & P28) p = seprint(p, e, "28:"); else p = seprint(p, e, "28:"); switch(pr & Pprotom){ default: s = "unk"; break; case Ppkt: s = "pkt"; break; case Pdiag: s = "dig"; break; case Preset: s = "rst"; break; case Pdmq: s = "dmq"; break; case Pdma: s = "dma"; break; case Ppio: s = "pio"; break; } p = seprint(p, e, "%s:", s); switch(pr & Pdatam){ default: s = "nd"; break; case Pin: s = "in"; break; case Pout: s = "out"; break; } p = seprint(p, e, "%s", s); return p; } void displaycmd(Req *r, Atatab *a) { char buf[32]; int i; if(a->cc > nelem(issuetr) || !issuetr[a->cc]) return; protostr(buf, buf + sizeof buf, a->protocol); fprint(2, "cmd %s:%2ux ", buf, a->cc); for(i = 0; i < 16; i++) fprint(2, "%.2ux", r->cmd.fis[i]); fprint(2, "\n"); } int issueata(Req *r, Atatab *a, Dev *d) { uchar u; int n, ok, pr, rv; r->haverfis = 0; pr = a->protocol & Pdatam; r->data = realloc(r->data, r->count); if(r->data == nil && r->count > 0) sysfatal("realloc: %r"); if(r->data == nil && pr != Pnd) sysfatal("no data for cmd %.2ux", a->cc); if(0 && r->fisbits) print("fisbits %.16b\n", r->fisbits); r->cmd.sdcmd = 0xff; r->cmd.ataproto = a->protocol; if(a->cc & 0xff00) fisset(r, 0, a->cc >> 8); else fisset(r, 0, H2dev); fisset(r, 1, Fiscmd); fisset(r, 2, a->cc); switch(pr){ case Pout: if(r->rfd != Dontread){ n = readn(r->rfd, r->data, r->count); if(n != r->count){ if(n == -1) eprint("!short src read %r\n"); else eprint("!short src read %d wanted %lld\n", n, r->count); return -1; } } case Pnd: case Pin: if(0 && (d->feat & Dlba) == 0 && (d->c | d->h | d->s)){ int c, h, s; c = r->lba / (d->s * d->h); h = (r->lba / d->s) % d->h; s = (r->lba % d->s) + 1; print("%d %d %d\n", c, h, s); fisset(r, 4, s); fisset(r, 5, c); fisset(r, 6, c >> 8); fisset(r, 7, Ataobs | h); }else{ fisset(r, 4, r->lba); fisset(r, 5, r->lba >> 8); fisset(r, 6, r->lba >> 16); u = Ataobs; if(pr == Pin || pr == Pout) u |= Atalba; if((d->feat & Dllba) == 0) u |= (r->lba >> 24) & 7; fisset(r, 7, u); fisset(r, 8, r->lba >> 24); fisset(r, 9, r->lba >> 32); fisset(r, 10, r->lba >> 48); } fisset(r, 12, r->nsect); fisset(r, 13, r->nsect >> 8); break; } fisset(r, 7, Ataobs); displaycmd(r, a); if(write(d->fd, &r->cmd, Cmdsz) != Cmdsz){ eprint("fis write error: %r\n"); return -1; } werrstr(""); switch(pr){ default: ok = read(d->fd, "", 0) == 0; break; case Pin: ok = read(d->fd, r->data, r->count) == r->count; r->lba += r->nsect; break; case Pout: ok = write(d->fd, r->data, r->count) == r->count; r->lba += r->nsect; break; } rv = 0; if(ok == 0){ eprint("xfer error: %.2ux %r\n", a->cc); rv = -1; } switch(n = read(d->fd, &r->reply, Replysz)){ case Replysz: r->haverfis = 1; return rv; case -1: eprint("status fis read error: %r\n"); return -1; default: eprint("status fis read error: short read: %d of %d\n", n, Replysz); return -1; } } /* * cheezy code; just issue a inquiry. use scuzz * for real work with atapi devices */ int issuepkt(Req *r, Atatab *a, Dev *d) { char *p; uchar *u; int n, rv; r->haverfis = 0; r->count = 128; r->data = realloc(r->data, r->count); if(r->data == nil && r->count > 0) sysfatal("realloc: %r"); r->cmd.sdcmd = 0xff; r->cmd.ataproto = a->protocol; memset(r->cmd.fis, 0, Fissize); u = r->cmd.fis; u[0] = 0x12; u[4] = 128-1; displaycmd(r, a); if(write(d->fd, &r->cmd, 6 + 2) != 6 + 2){ eprint("fis write error: %r\n"); return -1; } n = read(d->fd, r->data, r->count); rv = 0; if(n == -1){ eprint("xfer error: %.2ux %r\n", a->cc); rv = -1; } print("n is %d (%lld)\n", n, r->count); if(n > 32){ p = (char*)r->data; print("%.8s %.16s\n", p + 8, p + 16); } u = (uchar*)&r->reply; n = read(d->fd, u, Replysz); if(n < 0){ eprint("status fis read error (%d): %r\n", n); return -1; } if(n < Replysz) memset(u + n, 0, Replysz - n); r->haverfis = 1; return rv; } /* * silly protocol: * 1. use write log ext 0xe0 to fill out the command * 2. use write log ext 0xe1 to write or data (if any) * 3. use read log ext 0xe0 to nab status. polled */ void sctreq(Req *r) { memset(r, 0, sizeof *r); r->rfd = Dontread; } char* sctrsp(Req *r) { uint i; static char buf[32]; if(!r->haverfis) return "no rfis"; if((r->reply.fis[Frerror] & (Eidnf | Eabrt)) == 0) return nil; i = r->reply.fis[Fsc] | r->reply.fis[Flba0]<<8; if(i == 0xffff) return "in progress"; else if(i == 0){ snprint(buf, sizeof buf, "unknown %.2ux", r->reply.fis[Frerror]); return buf; }else if(i < nelem(sctetab)) return sctetab[i]; else return "<bad>"; } char* sctready(Dev *d, int sec) { char *s; int i; Req r; static char e[ERRMAX]; for(;;){ if(interrupted){ s = "interrupted"; break; } sctreq(&r); fisset(&r, Fsc, 1); fisset(&r, Flba0, 0xe0); r.count = 512; i = issueata(&r, sctread, d); free(r.data); if(i == -1){ rerrstr(e, ERRMAX); s = e; break; } if((r.cmd.fis[Fsc] | r.cmd.fis[Fsc8]<<8) != 0xffff){ s = sctrsp(&r); break; } if(sec == 0){ s = "timeout"; break; } sleep(1000); sec--; } return s; } typedef struct Sttab Sttab; struct Sttab { int o; int sz; char *name; }; Sttab sctt[] = { 0, 2, "version", 2, 2, "period", 4, 2, "intval", 6, 1, "max op", 7, 1, "max", 8, 1, "min op", 9, 1, "min", }; void sctttab(Req *r) { char c, buf[10]; int i, n, l, d; uchar *u; u = r->data; for(i = 0; i < nelem(sctt); i++){ switch(sctt[i].sz){ case 1: c = u[sctt[i].o]; print("%s\t%d\n", sctt[i].name, c); break; case 2: d = w(u + sctt[i].o); print("%s\t%ud\n", sctt[i].name, d); break; } } n = w(u + 30); l = w(u + 32); for(i = 0; i < n; i++){ c = u[34 + (l + i) % n]; if((uchar)c == 0x80) snprint(buf, sizeof buf, "xx"); else snprint(buf, sizeof buf, "%d", c); d = i%10; if(d == 0) print("\nt%d\t%d", i, c); else print("% .2d", c); } if(i%10) print("\n"); } static struct { uint code; char *s; char *ms; } fxtab[] = { 0x00010001, "set features", 0, 0x00010002, "enabled", 0, 0x00010003, "disabled", 0, 0x00020001, "enabled", 0, 0x00020002, "disabled", 0, 0x0003ffff, "minute", "minutes", }; void sctfcout(ushort *u, Req *r) { uchar *f; ushort v; uint c, m, i; f = r->reply.fis; switch(u[1]){ case 1: case 2: v = f[Fsc] | f[Flba0]<<8; c = u[2]<<16 | v; m = u[2]<<16 | 0xffff; for(i = 0; i < nelem(fxtab); i++) if(fxtab[i].code == c) print("%s\n", fxtab[i].s); else if(fxtab[i].code == m) print("%d %s\n", v, v>1? fxtab[i].ms: fxtab[i].s); break; case 3: v = f[Fsc] | f[Flba0]<<8; if(v & 1) print("preserve\n"); else print("volatile\n"); break; } } void scterout(ushort *u, Req *r) { uchar *f; uint v; f = r->reply.fis; switch(u[1]){ case 2: v = f[Fsc] | f[Flba0]<<8; v *= 100; print("%dms\n", v); } } void sctout(ushort *u, Req *r) { switch(u[0]){ case 5: sctttab(r); break; case 4: sctfcout(u, r); break; case 3: scterout(u, r); break; } } int issuesct0(Req *r0, Atatab *a, Dev *d) { char *s; uchar proto; Atatab *txa; Req r; if((d->feat & Dsct) == 0){ eprint("sct not supported\n"); return -1; } /* 1. issue command */ sctreq(&r); r.data = malloc(r0->count); memcpy(r.data, r0->data, r0->count); r.count = r0->count; fisset(&r, Fsc, 1); fisset(&r, Flba0, 0xe0); if(issueata(&r, sctissue, d) == -1) return -1; if(s = sctrsp(&r)){ eprint("sct error: %s\n", s); return -1; } /* 1a. check response */ if((s = sctready(d, 1)) != nil){ eprint("sct cmd: %s\n", s); return -1; } /* 2. transfer data */ proto = a->protocol; if(r0->fisbits & 1 << 16){ proto &= ~Pdatam; proto |= r0->cmd.ataproto; } switch(proto & Pdatam){ default: txa = nil; break; case Pin: txa = sctread; break; /* case Pout: txa = sctout; break; */ } if(txa != nil){ sctreq(&r); r.count = 512; fisset(&r, Fsc, 1); fisset(&r, Flba0, 0xe1); if(issueata(&r, txa, d) == -1) return -1; /* 2a. check response */ if((s = sctready(d, 1)) != nil){ eprint("sct cmd: %s\n", s); return -1; } } sctout((ushort*)r0->data, &r); free(r.data); return 0; } static void* pushtrace(int i) { void *tr0; tr0 = malloc(sizeof issuetr); if(tr0 == 0) return 0; memcpy(tr0, issuetr, sizeof issuetr); memset(issuetr, i, sizeof issuetr); return tr0; } static void poptrace(void *tr0) { if(tr0 == nil) return; memcpy(issuetr, tr0, sizeof issuetr); free(tr0); } int issuesct(Req *r0, Atatab *a, Dev *d) { int r; void *t; t = nil; if(scttrace) t = pushtrace(1); r = issuesct0(r0, a, d); if(scttrace) poptrace(t); return r; } int issue(Req *r, Atatab *a, Dev *d) { int rv; int (*f)(Req*, Atatab*, Dev*); if(a->protocol & Psct) f = issuesct; else if((a->protocol & Pprotom) == Ppkt) f = issuepkt; else f = issueata; rv = f(r, a, d); if(r->haverfis) if(r->reply.fis[Fstatus] & ASerr){ werrstr("ata error"); rv = -1; } return rv; } void sigfmt(Req *r) { print("%.8ux\n", fistosig(r->reply.fis)); } int opendev(char *dev, Dev *d) { char buf[128]; int rv; ushort *u; Req r; if(d->fd != -1) close(d->fd); memset(d, 0, sizeof *d); snprint(buf, sizeof buf, "%s/raw", dev); d->fd = open(buf, ORDWR); if(d->fd == -1) return -1; memset(&r, 0, sizeof r); if(issue(&r, sigcmd, d) == -1){ lose: close(d->fd); return -1; } setfissig(d, fistosig(r.reply.fis)); memset(&r, 0, sizeof r); r.count = 512; r.nsect = 1; if(d->sig>>16 == 0xeb14) rv = issue(&r, idpktcmd, d); else rv = issue(&r, idcmd, d); if(rv == -1) goto lose; u = (ushort*)r.data; d->nsect = idfeat(d, u); d->secsize = idss(d, u); d->wwn = idwwn(d, u); return 0; } void rawout(Req *r) { int n; n = write(r->wfd, r->data, r->count); if(n != r->count) eprint("!short write %ud wanted %lld\n", n, r->count); } static ushort gbit16(void *a) { ushort j; uchar *i; i = a; j = i[1] << 8; j |= i[0]; return j; } static Btab extra[] = { 12, "ncqpri", 11, "ncqunload", 10, "phyevent", 9, "hpwrctl", 3, "6.0gbit", 2, "3.0gbit", 1, "1.5gbit", }; static Btab suptab[] = { 8, "wwn", 5, "mediaserial", 1, "smartst", 0, "smartlog" }; char* pextraid(char *p, char *e, ushort *id, uint *medserial) { char *p0; ushort u; *p = 0; *medserial = 0; p0 = p; p = sebtab(p, e, extra, nelem(extra), gbit16(id + 76)); if(p != p0) p = seprint(p, e, " "); u = gbit16(id + 83); if(u & 1<<5) p = seprint(p, e, "gpl "); p0 = p; p = sebtab(p, e, suptab, nelem(suptab), gbit16(id + 84)); if(p != p0) p = seprint(p, e, " "); u = gbit16(id + 120); if(u & 1<<2) p = seprint(p, e, "wunc "); return p; } static char *patatab[] = { "ata8-apt", "ata/atapi-7", }; static char *satatab[] = { "ata8-ast", "sata1.0a", "sataiiext", "sata2.5", "sata2.6", "sata3.0", }; char* ptransport(char *p, char *e, ushort *id) { char *s; ushort u, i; u = gbit16(id + 222); if(u == 0 || u == 0xffff) return seprint(p, e, "unreported "); i = (u>>5) & 0x7f; switch(u & 7<<12){ default: s = "unktransport"; break; case 0: s = "unkparallel"; if(i < nelem(patatab)) s = patatab[i]; break; case 1<<12: s = "unkserial"; if(i < nelem(satatab)) s = satatab[i]; break; } return seprint(p, e, "%s ", s); } Btab entab[] = { 10, "hpa", 9, "reset", 8, "service", 7, "release", 6, "rdlookahd", 5, "vwc", 4, "packet", 3, "pm", 2, "security", 1, "smart", }; Btab addlen[] = { 15, "cfast", // 14, "trim", /* check 169 */ 13, "lpsalignerr", 12, "iddma", 11, "rbufdma", 10, "wbufdma", 9, "pwddma", 8, "dlmcdma", }; char* penabled(char *p, char *e, ushort *id) { char *p0; ushort u; p0 = p; p = sebtab(p, e, addlen, nelem(addlen), gbit16(id + 69)); u = gbit16(id + 87); if(u>>14 == 1){ if(p != p0) p = seprint(p, e, " "); p = sebtab(p, e, entab, nelem(entab), gbit16(id + 85)); } return p; } static char *fftab[] = { nil, "5¼", "3½", "2½", "1.8", "<1.8", }; char* pff(char *p, char *e, ushort *id) { char *p0; ushort u; p0 = p; u = gbit16(id + 168); if(u < nelem(fftab) && fftab[u] != nil) p = seprint(p, e, "%s ", fftab[u]); u = gbit16(id + 217); if(u == 1) p = seprint(p, e, "solid-state "); else if(u != 0 && u != 0xfffe) p = seprint(p, e, "%udrpm ", u); if(p != p0) p--; *p = 0; return p; } Btab scttab[] = { 5, "tables", 4, "feactl", 3, "errctl", 2, "wsame", 1, "rwlong", 0, "sct", }; char* psct(char *p, char *e, ushort *id) { return sebtab(p, e, scttab, nelem(scttab), gbit16(id + 206)); } void idfmt(Req *r) { char buf[100]; uint ss, i; ushort *id; uvlong nsect; Sfis f; if(r->fmtrw == 0){ rawout(r); return; } id = (ushort*)r->data; nsect = idfeat(&f, id); ss = idss(&f, id); idmove(buf, id+10, 20); print("serial\t%s\n", buf); idmove(buf, id+23, 8); print("firm\t%s\n", buf); idmove(buf, id+27, 40); print("model\t%s\n", buf); print("wwn\t%ullx\n", idwwn(&f, id)); pflag(buf, buf + sizeof buf, &f); print("flags\t%s", buf); print("geometry %llud %ud", nsect, ss); if(f.c | f.h | f.s) print(" %ud %ud %ud", f.c, f.h, f.s); print("\n"); penabled(buf, buf + sizeof buf, id); print("enabled\t%s\n", buf); pextraid(buf, buf + sizeof buf, id, &i); print("extra\t%s\n", buf); if(i){ idmove(buf, id + 176, 60); if(buf[0] != 0) print("medias\t%s\n", buf); } psct(buf, buf + sizeof buf, id); if(buf[0]) print("sct\t%s\n", buf); ptransport(buf, buf + sizeof buf, id); print("trans\t%s\n", buf); pff(buf, buf + sizeof buf, id); if(buf[0]) print("ff\t%s\n", buf); } void smfmt(Req *r) { uchar *fis; if(r->cmd.fis[Ffeat] == 0xda){ fis = r->reply.fis; if(fis[5] == 0x4f && fis[6] == 0xc2) eprint("normal\n"); else eprint("threshold exceeded\n"); return; } } void iofmt(Req *r) { uchar *u; int i; if(r->fmtrw == 0){ rawout(r); return; } u = r->data; for(i = 0; i < r->count; i += 16) fprint(2, "%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux" "%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux\n", u[i + 0], u[i + 1], u[i + 2], u[i + 3], u[i + 4], u[i + 5], u[i + 6], u[i + 7], u[i + 8], u[i + 9], u[i +10], u[i +11], u[i +12], u[i +13], u[i +14], u[i +15]); } static char *csbyte[] = { "never started", nil, "competed without error", "in progress", "suspended by cmd from host", "aborted by cmd from host", "aborted by device with fatal error", }; static char *exe[] = { "no error or never run", "aborted by host", "interrupted by host", "fatal error; unable to complete", "failed", "failed: electricial", "failed: servo", "failed: read", "failed: shipping damage", [0xf] "in progress", }; char* tabtr(uint u, char **tab, int ntab) { char *s; if(u >= ntab || (s = tab[u]) == nil) s = "reserved"; return s; } void sdfmt(Req *r) { char *s; uchar *b; ushort u; if(r->fmtrw == 0){ rawout(r); return; } b = r->data; u = b[362]; if((u & 0xf0) == 0x80 && u != 0x81 && u != 0x83) u &= 0xf; s = tabtr(u, csbyte, nelem(csbyte)); print("col status: %.2ux %s\n", b[362], s); u = b[363]; s = tabtr(u>>4, exe, nelem(exe)); if(u & 0xf) print("exe status: %.2ux %s, %d0%% left\n", u, s, u & 0xf); else print("exe status: %.2ux %s\n", u, s); u = b[364] | b[365]<<8; print("time left: %uds\n", u); print("shrt poll: %udm\n", b[373]); u = b[374]; if(u == 0xff) u = b[375] | b[376]<<8; print("ext poll: %udm\n", u); } void pagemapfmt(Req *r) { int i; ushort *u; u = (ushort*)r->data; if(u[0] != 1){ print("unsupported\n"); return; } for(i = 1; i < 128; i++) if(u[i] > 0) print("page %d: %d\n", i, u[i]); } void slfmt(Req *r) { switch(r->cmd.fis[Flba0]){ default: iofmt(r); break; case 0: pagemapfmt(r); break; } } enum{ Physz = 7<<12, }; static char *phyec[] = { "no event", "icrc", "err data", "err d2h data", "err h2d data", [0x05] "err nd", "err d2h nd", "err h2d nd", "retry d2h nd", "nready", [0x0a] "comreset", "h2d crc", nil, "bad h2d", nil, "err h2d data crc", [0x10] "err h2d data", nil, "err h2d nd crc", "err h2d nd", }; void phyfmt(Req *r) { char *ec; uchar *p; ushort *u, *e, id, sz; u = (ushort*)r->data; e = u + 510/sizeof *u; for(u += 2; u < e; u += sz){ id = w((uchar*)u); sz = (id & Physz) >> 12; id &= ~Physz; if(sz == 0) break; ec = "unk"; if(id < nelem(phyec) && phyec[id] != nil) ec = phyec[id]; print("%.4ux\t%-15s\t", id, ec); p = (uchar*)u + 2; switch(sz<<1){ default: print("\n"); break; case 2: print("%.4ux\n", w(p)); break; case 4: print("%.8ux\n", dw(p)); break; case 8: print("%.16llux\n", qw(p)); break; } sz += 1; } } typedef struct Gltab Gltab; struct Gltab{ int offset; char *name; }; Gltab page3[] = { 8, "power-on hrs", 16, "head flying hrs", 24, "head loads", 32, "realloc'd sec", 40, "read recovery att", 48, "start failures" }; void qpfmt(Req *r, Gltab *t, int ntab) { uchar *u; int i; uvlong v; u = r->data; for(i = 0; i < ntab; i++){ v = qw(u + t[i].offset); if((v & 3ll<<63) != 3ll<<63) continue; print("%lud\t%s\n", (ulong)v, t[i].name); } } static char *sctsttab[] = { "active waiting", "standby", "sleep", "dst bgnd", "smart bgnd", "sct bgnd", }; void sctstatfmt(Req *r) { char *s; uchar *id, c; id = r->data; print("version\t%d\n", gbit16(id + 0)); print("vnd ver\t%2ux\n", gbit16(id + 2)); print("flags\t%.8ux\n", dw(id + 6)); c = id[10]; s = "unk"; if(c < nelem(sctsttab)) s = sctsttab[c]; print("state\t%s\n", s); print("ext stat\t%.4ux\n", gbit16(id + 14)); print("act code\t%.4ux\n", gbit16(id + 16)); print("fn code\t%.4ux\n", gbit16(id + 18)); print("lba\t%llud\n", qw(id + 40)); print("temp\t%d\n", id[200]); print("min t\t%d %d\n", id[201], id[203]); print("max t\t%d %d\n", id[202], id[204]); print("ot\t%d\n", dw(id + 206)); print("ut\t%d\n", dw(id + 210)); } void glfmt(Req *r) { switch(r->cmd.fis[Flba0]){ case 0: pagemapfmt(r); break; case 3: qpfmt(r, page3, nelem(page3)); break; case 17: phyfmt(r); break; case 0xe0: sctstatfmt(r); break; default: iofmt(r); break; } } char* readline(char *prompt, char *line, int len) { char *p, *e, *q; int n, dump; e = line + len; retry: dump = 0; if(interrupted) eprint("\n%s", prompt); else eprint("%s", prompt); interrupted = 0; for(p = line;; p += n){ if(p == e){ dump = 1; p = line; } n = read(0, p, e - p); if(n < 0){ if(interrupted) goto retry; return nil; } if(n == 0) return nil; if(q = memchr(p, '\n', n)){ if(dump){ eprint("!line too long\n"); goto retry; } p = q; break; } } *p = 0; return line; } void suggesttab(char *cmd, Atatab *a, int n) { int i, l; l = strlen(cmd); for(i = 0; i < n; i++) if(cistrncmp(cmd, a[i].name, l) == 0) eprint("%s\n", a[i].name); } Atatab* findtab(char **cmd, Atatab *a, int n) { char *p, *c; int i, cc, max, l; cc = strtoul(*cmd, &p, 0); if(p != *cmd && (*p == 0 || *p == ' ')){ for(i = 0; i < n; i++) if(a[i].cc == cc){ *cmd = p + 1; return a + cc; } return 0; } max = 0; cc = 0; c = *cmd; for(i = 0; i < n; i++){ l = strlen(a[i].name); if(l > max && cistrncmp(*cmd, a[i].name, l) == 0) if(c[l] == ' ' || c[l] == 0){ max = l + (c[l] == ' '); cc = i; } } if(max > 0){ *cmd = *cmd + max; return a + cc; } return 0; } int catch(void*, char *note) { if(strstr(note, "interrupt") != nil) return interrupted = 1; return 0; } char** ndargs(Atatab*, Req *, char **p) { return p; } char** ioargs(Atatab *, Req *r, char **p) { if(r->nsect == 0) r->nsect = 1; if(p[0] == 0) return p; r->lba = strtoull(p[0], 0, 0); p++; if(p[0] == 0) return p; r->nsect = strtoul(p[0], 0, 0); return p + 1; } char** stdargs(Atatab *, Req *r, char **p) { char *s; Rune x; for(; p[0] && p[0][0] == '-' && p[0][1]; p++){ s = p[0] + 1; if(*s == '-'){ p++; break; } while(*s && (s += chartorune(&x, s))) switch(x){ case 'r': r->raw = 1; break; default: return p; } } return p; } static void chopoff(char *s, char *extra) { char *p; int l, ls; l = strlen(extra); ls = strlen(s); if(l >= ls) return; p = s + ls - l; if(strcmp(p, extra) == 0) *p = 0; } char* trim(char *s) { char *p; while(*s && (*s == ' ' || *s == '\t')) s++; if(*s == 0) return nil; p = s + strlen(s) - 1; while(*p == ' ' || *p == '\t') p--; p[1] = 0; return s; } int doredir(Req *r, char **f, int nf, int mode, int *fd1, int *fd2) { int fd; if(nf != 1 && nf != 2){ eprint("!args\n"); return -1; } fd = -1; if(nf == 2){ fd = open(f[1], mode); if(mode != OREAD){ if(fd == -1) fd = create(f[1], mode, 0660); else seek(fd, 0, 2); } } if(fd1){ close(*fd1); *fd1 = fd; } if(fd2){ r->fmtrw = fd == -1; close(*fd2); *fd2 = fd; } return fd; } int special(char *s, Dev *d, Req *r) { char buf[512], path[128], *f[20], sbuf[512], s2[512], *p, *e, *t; uchar *u; int i, j, nf; Atatab *a; p = buf; e = buf + sizeof buf; if(!strcmp(s, "close")){ r->haverfis = 0; close(d->fd); d->fd = -1; return 0; } if(!strcmp(s, "scttrace")){ scttrace = 1; return 0; } if(!strcmp(s, "dev")){ if(d->fd == -1){ eprint("!bad cmd (device closed)\n"); return 0; } if(fd2path(d->fd, path, sizeof path) == -1) sysfatal("fd2path: %r"); chopoff(path, "/raw"); p = seprint(p, e, "dev\t%s\n", path); p = seprint(p, e, "flags\t"); p = pflag(p, e, d); p = seprint(p, e, "lsectsz\t" "%ud ptol %ud\n", d->lsectsz, 1<<d->physshift); p = seprint(p, e, "geometry %llud %ud\n", d->nsect, d->secsize); if(d->c | d->h | d->s) seprint(p, e, "chs\t%d %d %d\n", d->c, d->h, d->s); print("%s", buf); return 0; } if(!strcmp(s, "help")){ suggesttab(buf, atatab, nelem(atatab)); return 0; } if(!strcmp(s, "probe")){ probe(); return 0; } if(!strcmp(s, "rfis")){ if(r->haverfis == 0){ eprint("!no rfis\n"); return 0; } p = seprint(p, e, "%.2x\n", r->reply.sdcmd); u = r->reply.fis; for(i = 0; i < 16; i++) p = seprint(p, e, "%.2ux", u[i]); seprint(p, e, "\n"); print("%s", buf); return 0; } for(t = s; *t == '<' || *t == '>'; t++) ; if(t != s) snprint(sbuf, sizeof buf, "%.*s %s", (int)(t - s), s, t); else snprint(sbuf, sizeof sbuf, "%s", s); nf = tokenize(sbuf, f, nelem(f)); if(!strcmp(f[0], "issuetr")){ if(nf == 1) for(i = 0; i < nelem(issuetr); i++) issuetr[i] ^= 1; else{ p = s2; e = s2 + sizeof s2; for(i = 1; i < nf - 1; i++) p = seprint(p, e, "%s ", f[i]); p = seprint(p, e, "%s", f[i]); e = s2; for(i = 1; i < nf; i++){ j = strtoul(f[i], &p, 0); if(*p == 0 && j < nelem(issuetr)) issuetr[i] ^= 1; else if(a = findtab(&e, atatab, nelem(atatab))) issuetr[a->cc & 0xff] ^= 1; } } return 0; } if(!strcmp(f[0], "open")){ r->lba = 0; if(nf == 2) opendev(f[1], d); else eprint("!bad args to open\n"); return 0; } if(!strcmp(f[0], ">")){ doredir(r, f, nf, OWRITE, 0, &r->wfd); return 0; } if(!strcmp(f[0], "<")){ doredir(r, f, nf, OREAD, &r->rfd, 0); return 0; } if(!strcmp(f[0], "<>")){ doredir(r, f, nf, OWRITE, &r->rfd, &r->wfd); return 0; } return -1; } void setreg(Req *r, uint reg, uvlong v) { uchar *o; int x; switch(reg & (Sbase | Pbase)){ case 0: r->fisbits |= 1 << reg; r->cmd.fis[reg] = v; break; case Sbase: case Sbase | Pbase: x = reg & ~(Sbase | Ssz); o = r->data + x*2; assert(x < r->count); switch(reg & Ssz){ default: print("reg & Ssz %ux\n", reg & Ssz); _assert("bad table"); case Sw: pw(o, v); break; case Sdw: pdw(o, v); break; case Sqw: pqw(o, v); break; } break; case Pbase: /* fix me please: this is teh suck */ r->fisbits |= 1 << 16; r->cmd.ataproto = v; break; } } int setfis0(Req *r, Txtab *t, char *p) { char *e; uvlong v; v = strtoull(p, &e, 0); setreg(r, t->val, v); return *e != 0; } char** setfis(Atatab*, Req *r, char **p) { char *s; int i; loop: if((s = p[0]) == 0) return p; for(i = 0; i < nelem(regtx); i++) if(strcmp(s, regtx[i].name) == 0 && p[1] != nil){ // print("setfis0 %s %s\n", p[0], p[1]); setfis0(r, regtx + i, p[1]); p += 2; goto loop; } return p; } char* rname(char *buf, int n, int r) { int i; for(i = 0; i < nelem(regtx); i++) if(regtx[i].val == r){ snprint(buf, n, "%s", regtx[i].name); return buf; } snprint(buf, n, "%.2ux", r); return buf; } int mwcmp(char *a, char ***l) { char buf[128], *f[20], **p; int nf, i; if(*a == 0) return 0; p = *l; if(p[0] == 0) return -1; snprint(buf, sizeof buf, "%s", a); nf = tokenize(buf, f, nelem(f)); for(i = 0; i < nf; i++) if(p[i] == nil || cistrcmp(p[i], f[i]) != 0) return -1; *l = p + i - 1; return 0; } char **dofetab(Fetab*, Req*, char**); static char hexdig[] = "ABCDEFabcdef0123456789"; static char hexonly[] = "ABCDEFabcdef"; static char Enum[] = "expecting number"; int fenum(Fetab *, int v, char ***p) { char *e, *s, *r; int base; if(v >= 0) return v; s = *(*p + 1); e = nil; if(s == nil || *s == 0) e = Enum; else{ base = 0; if(strspn(s, hexdig) == strlen(s) && strpbrk(s, hexonly) != nil) base = 0x10; v = strtoul(s, &r, base); if(*r) e = Enum; } if(e == nil) (*p)++; else print("error: %s [%s]\n", e, s); return v; } char** dofetab0(Fetab *t, Req *r, char **p) { int i, v; Txtab *tab; if(t == nil) return p; tab = t->tab; loop: for(i = 0; i < t->ntab; i++) if(mwcmp(tab[i].name, &p) == 0){ v = fenum(t, tab[i].val, &p); setreg(r, t->reg, v); if(tab[i].name[0] != 0){ p = dofetab(tab[i].fe, r, p + 1); goto loop; } } return p; } char** dofetab(Fetab *t, Req *r, char **p) { for(; t != nil && t->ntab > 0; t++) p = dofetab0(t, r, p); return p; } char** dotab(Atatab *a, Req *r, char **p) { if(a->tab == nil) return p; return dofetab(a->tab, r, p); } void initreq(Req *r) { memset(r, 0, sizeof *r); // r->wfd = open("/dev/null", OWRITE); r->wfd = dup(1, -1); if(rflag == 0) r->fmtrw = 1; r->rfd = open("/dev/zero", OREAD); } void setup(void) { int i; for(i = 0; i < nelem(atatab); i++) if(atatab[i].cc == 0x2f){ sctread = atatab + i; break; } for(; i < nelem(atatab); i++) if(atatab[i].cc == 0x3f){ sctissue = atatab + i; break; } for(; i < nelem(atatab); i++) if(atatab[i].cc == 0xa1){ idpktcmd = atatab + i; break; } for(; i < nelem(atatab); i++) if(atatab[i].cc == 0xec){ idcmd = atatab + i; break; } for(; i < nelem(atatab); i++) if(atatab[i].cc == 0xf000){ sigcmd = atatab + i; break; } } typedef struct Htab Htab; struct Htab { ulong bit; char *name; }; Htab ertab[] = { Eicrc, "icrc", Ewp, "wp", Emc, "mc", Eidnf, "idnf", Emcr, "mcr", Eabrt, "abrt", Enm, "nm", Emed, "med", Eunc, "unc", }; Htab sttab[] = { ASbsy, "bsy", ASdrdy, "drdy", ASdf, "df", ASdrq, "drq", ASerr, "err", }; static char* htabfmt(char *p, char *e, Htab *t, int n, ulong u) { char *p0; uint i; p0 = p; for(i = 0; i < n; i++) if(u & t[i].bit) p = seprint(p, e, "%s | ", t[i].name); if(p - 3 >= p0) p -= 3; if(p < e) p[0] = 0; return p; } void prerror(Req *r) { char st[64], er[64]; uchar *u; u = r->reply.fis; if(r->haverfis == 0 || (u[Fstatus] & ASerr) == 0) return; htabfmt(er, er + sizeof er, ertab, nelem(ertab), u[Frerror]); htabfmt(st, st + sizeof st, sttab, nelem(sttab), u[Fstatus] & ~ASobs); fprint(2, "err %.2ux %.2ux (%s, %s)\n", u[Frerror], u[Fstatus], er, st); } void usage(void) { eprint("usage: atazz dev\n"); eprint(" or -c cmd\n"); exits("usage"); } void main(int argc, char **argv) { char buf[1024], *p, *f[20], **fp; int nf, cflag, i; Atatab *a; Req r; Dev d; cflag = 0; ARGBEGIN{ case 'c': cflag = atoi(EARGF(usage())); break; case 'r': rflag = 1; break; default: usage(); }ARGEND if(cflag){ for(i = 0; i < nelem(atatab); i++) if(atatab[i].cc == cflag) print("%s\n", atatab[i].name); exits(""); } setup(); fmtinstall(L'π', πfmt); if(argc > 1) usage(); initreq(&r); d.fd = -1; if(argc == 1 && opendev(*argv, &d) == -1) sysfatal("opendev: %r"); atnotify(catch, 1); for(;;){ memset(&r.cmd, 0, sizeof r.cmd); r.fisbits = 0; if(readline("az> ", buf, sizeof buf-1) == nil) break; if((p = trim(buf)) == nil) continue; if(special(buf, &d, &r) == 0) continue; if(d.fd == -1){ eprint("!bad cmd (device closed)\n"); continue; } a = findtab(&p, atatab, nelem(atatab)); if(!a){ suggesttab(buf, atatab, nelem(atatab)); eprint("!unknown cmd\n"); continue; } nf = tokenize(p, f, nelem(f) - 1); f[nf] = 0; fp = stdargs(a, &r, f); fp = setfis(a, &r, fp); if(a->protocol & Psct){ r.count = 1 * 512; r.data = realloc(r.data, r.count); memset(r.data, 0, r.count); } fp = dotab(a, &r, fp); switch(a->protocol & Pprotom){ default: eprint("!bad proto1 %.2ux\n", a->protocol & Pprotom); continue; case Pnd: fp = ndargs(a, &r, fp); case Preset: case Pdiag: r.count = 0; r.lba = 0; r.nsect = 0; break; case Ppio: case Pdma: case Pdmq: case Ppkt: if(a->flags & Cmd5sc){ r.nsect = r.cmd.fis[Fsc]; if(r.nsect == 0) r.nsect = 1; r.cmd.fis[Fsc] = r.nsect; r.count = r.nsect * 0x200; }else if((a->protocol & Pssm) == P512){ r.lba = 0; r.nsect = 0; r.count = 512; }else{ fp = ioargs(a, &r, fp); r.count = d.secsize * r.nsect; } break; } if(fp[0]){ eprint("!extra args %π\n", fp); continue; } if(issue(&r, a, &d) == -1){ prerror(&r); continue; } if(a->fmt) a->fmt(&r); } exits(""); }