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

You are viewing proxied material from sdf.org. The copyright of proxied material belongs to its original authors. Any comments or complaints in relation to proxied material should be directed to the original authors of the content concerned. Please see the disclaimer for more details.