ref: f35d5ee5b036b7b4fa59b1839f52fca15b7f99f9
dir: /sys/src/cmd/auth/rsa2any.c/
#include <u.h> #include <libc.h> #include <bio.h> #include <auth.h> #include <mp.h> #include <libsec.h> #include "rsa2any.h" RSApriv* getrsakey(int argc, char **argv, int needprivate, Attr **pa) { char *file, *s, *p; int sz; RSApriv *key; Biobuf *b; int regen; Attr *a; if(argc == 0) file = "#d/0"; else file = argv[0]; key = mallocz(sizeof(RSApriv), 1); if(key == nil) return nil; if((b = Bopen(file, OREAD)) == nil){ werrstr("open %s: %r", file); return nil; } s = Brdstr(b, '\n', 1); if(s == nil){ werrstr("read %s: %r", file); return nil; } if(strncmp(s, "key ", 4) != 0){ werrstr("bad key format"); return nil; } regen = 0; a = _parseattr(s+4); if(a == nil){ werrstr("empty key"); return nil; } if((p = _strfindattr(a, "proto")) == nil){ werrstr("no proto"); return nil; } if(strcmp(p, "rsa") != 0){ werrstr("proto not rsa"); return nil; } if((p = _strfindattr(a, "ek")) == nil){ werrstr("no ek"); return nil; } if((key->pub.ek = strtomp(p, &p, 16, nil)) == nil || *p != 0){ werrstr("bad ek"); return nil; } if((p = _strfindattr(a, "n")) == nil){ werrstr("no n"); return nil; } if((key->pub.n = strtomp(p, &p, 16, nil)) == nil || *p != 0){ werrstr("bad n"); return nil; } if((p = _strfindattr(a, "size")) == nil) fprint(2, "warning: missing size; will add\n"); else if((sz = strtol(p, &p, 10)) == 0 || *p != 0) fprint(2, "warning: bad size; will correct\n"); else if(sz != mpsignif(key->pub.n)) fprint(2, "warning: wrong size (got %d, expected %d); will correct\n", sz, mpsignif(key->pub.n)); if(!needprivate) goto call; if((p = _strfindattr(a, "!dk")) == nil){ werrstr("no !dk"); return nil; } if((key->dk = strtomp(p, &p, 16, nil)) == nil || *p != 0){ werrstr("bad !dk"); return nil; } if((p = _strfindattr(a, "!p")) == nil){ werrstr("no !p"); return nil; } if((key->p = strtomp(p, &p, 16, nil)) == nil || *p != 0){ werrstr("bad !p"); return nil; } if((p = _strfindattr(a, "!q")) == nil){ werrstr("no !q"); return nil; } if((key->q = strtomp(p, &p, 16, nil)) == nil || *p != 0){ werrstr("bad !q"); return nil; } if((p = _strfindattr(a, "!kp")) == nil){ fprint(2, "warning: no !kp\n"); regen = 1; goto regen; } if((key->kp = strtomp(p, &p, 16, nil)) == nil || *p != 0){ fprint(2, "warning: bad !kp\n"); regen = 1; goto regen; } if((p = _strfindattr(a, "!kq")) == nil){ fprint(2, "warning: no !kq\n"); regen = 1; goto regen; } if((key->kq = strtomp(p, &p, 16, nil)) == nil || *p != 0){ fprint(2, "warning: bad !kq\n"); regen = 1; goto regen; } if((p = _strfindattr(a, "!c2")) == nil){ fprint(2, "warning: no !c2\n"); regen = 1; goto regen; } if((key->c2 = strtomp(p, &p, 16, nil)) == nil || *p != 0){ fprint(2, "warning: bad !c2\n"); regen = 1; goto regen; } regen: if(regen){ RSApriv *k2; k2 = rsafill(key->pub.n, key->pub.ek, key->dk, key->p, key->q); if(k2 == nil){ werrstr("regenerating chinese-remainder parts failed: %r"); return nil; } key = k2; } call: a = _delattr(a, "ek"); a = _delattr(a, "n"); a = _delattr(a, "size"); a = _delattr(a, "!dk"); a = _delattr(a, "!p"); a = _delattr(a, "!q"); a = _delattr(a, "!c2"); a = _delattr(a, "!kp"); a = _delattr(a, "!kq"); if(pa) *pa = a; return key; } uchar* put4(uchar *p, uint n) { p[0] = (n>>24)&0xFF; p[1] = (n>>16)&0xFF; p[2] = (n>>8)&0xFF; p[3] = n&0xFF; return p+4; } uchar* putn(uchar *p, void *v, uint n) { memmove(p, v, n); p += n; return p; } uchar* putstr(uchar *p, char *s) { p = put4(p, strlen(s)); p = putn(p, s, strlen(s)); return p; } uchar* putmp2(uchar *p, mpint *b) { int bits, n; if(mpcmp(b, mpzero) == 0) return put4(p, 0); bits = mpsignif(b); n = (bits+7)/8; if(bits%8 == 0){ p = put4(p, n+1); *p++ = 0; }else p = put4(p, n); mptobe(b, p, n, nil); p += n; return p; }