rsa2any.c (5774B)
1 #include <u.h> 2 #include <libc.h> 3 #include <bio.h> 4 #include <auth.h> 5 #include <mp.h> 6 #include <libsec.h> 7 #include "rsa2any.h" 8 9 RSApriv* 10 getkey(int argc, char **argv, int needprivate, Attr **pa) 11 { 12 char *file, *s, *p; 13 int sz; 14 RSApriv *key; 15 Biobuf *b; 16 int regen; 17 Attr *a; 18 19 if(argc == 0) 20 file = "/dev/stdin"; 21 else 22 file = argv[0]; 23 24 key = mallocz(sizeof(RSApriv), 1); 25 if(key == nil) 26 return nil; 27 28 if((b = Bopen(file, OREAD)) == nil){ 29 werrstr("open %s: %r", file); 30 return nil; 31 } 32 s = Brdstr(b, '\n', 1); 33 if(s == nil){ 34 werrstr("read %s: %r", file); 35 return nil; 36 } 37 if(strncmp(s, "key ", 4) != 0){ 38 werrstr("bad key format"); 39 return nil; 40 } 41 42 regen = 0; 43 a = _parseattr(s+4); 44 if(a == nil){ 45 werrstr("empty key"); 46 return nil; 47 } 48 if((p = _strfindattr(a, "proto")) == nil){ 49 werrstr("no proto"); 50 return nil; 51 } 52 if(strcmp(p, "rsa") != 0){ 53 werrstr("proto not rsa"); 54 return nil; 55 } 56 if((p = _strfindattr(a, "ek")) == nil){ 57 werrstr("no ek"); 58 return nil; 59 } 60 if((key->pub.ek = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 61 werrstr("bad ek"); 62 return nil; 63 } 64 if((p = _strfindattr(a, "n")) == nil){ 65 werrstr("no n"); 66 return nil; 67 } 68 if((key->pub.n = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 69 werrstr("bad n"); 70 return nil; 71 } 72 if((p = _strfindattr(a, "size")) == nil) 73 fprint(2, "warning: missing size; will add\n"); 74 else if((sz = strtol(p, &p, 10)) == 0 || *p != 0) 75 fprint(2, "warning: bad size; will correct\n"); 76 else if(sz != mpsignif(key->pub.n)) 77 fprint(2, "warning: wrong size (got %d, expected %d); will correct\n", 78 sz, mpsignif(key->pub.n)); 79 if(!needprivate) 80 goto call; 81 if((p = _strfindattr(a, "!dk")) == nil){ 82 werrstr("no !dk"); 83 return nil; 84 } 85 if((key->dk = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 86 werrstr("bad !dk"); 87 return nil; 88 } 89 if((p = _strfindattr(a, "!p")) == nil){ 90 werrstr("no !p"); 91 return nil; 92 } 93 if((key->p = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 94 werrstr("bad !p"); 95 return nil; 96 } 97 if((p = _strfindattr(a, "!q")) == nil){ 98 werrstr("no !q"); 99 return nil; 100 } 101 if((key->q = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 102 werrstr("bad !q"); 103 return nil; 104 } 105 if((p = _strfindattr(a, "!kp")) == nil){ 106 fprint(2, "warning: no !kp\n"); 107 regen = 1; 108 goto regen; 109 } 110 if((key->kp = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 111 fprint(2, "warning: bad !kp\n"); 112 regen = 1; 113 goto regen; 114 } 115 if((p = _strfindattr(a, "!kq")) == nil){ 116 fprint(2, "warning: no !kq\n"); 117 regen = 1; 118 goto regen; 119 } 120 if((key->kq = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 121 fprint(2, "warning: bad !kq\n"); 122 regen = 1; 123 goto regen; 124 } 125 if((p = _strfindattr(a, "!c2")) == nil){ 126 fprint(2, "warning: no !c2\n"); 127 regen = 1; 128 goto regen; 129 } 130 if((key->c2 = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 131 fprint(2, "warning: bad !c2\n"); 132 regen = 1; 133 goto regen; 134 } 135 regen: 136 if(regen){ 137 RSApriv *k2; 138 139 k2 = rsafill(key->pub.n, key->pub.ek, key->dk, key->p, key->q); 140 if(k2 == nil){ 141 werrstr("regenerating chinese-remainder parts failed: %r"); 142 return nil; 143 } 144 key = k2; 145 } 146 call: 147 a = _delattr(a, "ek"); 148 a = _delattr(a, "n"); 149 a = _delattr(a, "size"); 150 a = _delattr(a, "!dk"); 151 a = _delattr(a, "!p"); 152 a = _delattr(a, "!q"); 153 a = _delattr(a, "!c2"); 154 a = _delattr(a, "!kp"); 155 a = _delattr(a, "!kq"); 156 if(pa) 157 *pa = a; 158 return key; 159 } 160 161 DSApriv* 162 getdsakey(int argc, char **argv, int needprivate, Attr **pa) 163 { 164 char *file, *s, *p; 165 DSApriv *key; 166 Biobuf *b; 167 Attr *a; 168 169 if(argc == 0) 170 file = "/dev/stdin"; 171 else 172 file = argv[0]; 173 174 key = mallocz(sizeof(RSApriv), 1); 175 if(key == nil) 176 return nil; 177 178 if((b = Bopen(file, OREAD)) == nil){ 179 werrstr("open %s: %r", file); 180 return nil; 181 } 182 s = Brdstr(b, '\n', 1); 183 if(s == nil){ 184 werrstr("read %s: %r", file); 185 return nil; 186 } 187 if(strncmp(s, "key ", 4) != 0){ 188 werrstr("bad key format"); 189 return nil; 190 } 191 192 a = _parseattr(s+4); 193 if(a == nil){ 194 werrstr("empty key"); 195 return nil; 196 } 197 if((p = _strfindattr(a, "proto")) == nil){ 198 werrstr("no proto"); 199 return nil; 200 } 201 if(strcmp(p, "dsa") != 0){ 202 werrstr("proto not dsa"); 203 return nil; 204 } 205 if((p = _strfindattr(a, "p")) == nil){ 206 werrstr("no p"); 207 return nil; 208 } 209 if((key->pub.p = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 210 werrstr("bad p"); 211 return nil; 212 } 213 if((p = _strfindattr(a, "q")) == nil){ 214 werrstr("no q"); 215 return nil; 216 } 217 if((key->pub.q = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 218 werrstr("bad q"); 219 return nil; 220 } 221 if((p = _strfindattr(a, "alpha")) == nil){ 222 werrstr("no alpha"); 223 return nil; 224 } 225 if((key->pub.alpha = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 226 werrstr("bad alpha"); 227 return nil; 228 } 229 if((p = _strfindattr(a, "key")) == nil){ 230 werrstr("no key="); 231 return nil; 232 } 233 if((key->pub.key = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 234 werrstr("bad key="); 235 return nil; 236 } 237 if(!needprivate) 238 goto call; 239 if((p = _strfindattr(a, "!secret")) == nil){ 240 werrstr("no !secret"); 241 return nil; 242 } 243 if((key->secret = strtomp(p, &p, 16, nil)) == nil || *p != 0){ 244 werrstr("bad !secret"); 245 return nil; 246 } 247 call: 248 a = _delattr(a, "p"); 249 a = _delattr(a, "q"); 250 a = _delattr(a, "alpha"); 251 a = _delattr(a, "key"); 252 a = _delattr(a, "!secret"); 253 if(pa) 254 *pa = a; 255 return key; 256 } 257 258 uchar* 259 put4(uchar *p, uint n) 260 { 261 p[0] = (n>>24)&0xFF; 262 p[1] = (n>>16)&0xFF; 263 p[2] = (n>>8)&0xFF; 264 p[3] = n&0xFF; 265 return p+4; 266 } 267 268 uchar* 269 putn(uchar *p, void *v, uint n) 270 { 271 memmove(p, v, n); 272 p += n; 273 return p; 274 } 275 276 uchar* 277 putstr(uchar *p, char *s) 278 { 279 p = put4(p, strlen(s)); 280 p = putn(p, s, strlen(s)); 281 return p; 282 } 283 284 uchar* 285 putmp2(uchar *p, mpint *b) 286 { 287 int bits, n; 288 289 if(mpcmp(b, mpzero) == 0) 290 return put4(p, 0); 291 bits = mpsignif(b); 292 n = (bits+7)/8; 293 if(bits%8 == 0){ 294 p = put4(p, n+1); 295 *p++ = 0; 296 }else 297 p = put4(p, n); 298 mptobe(b, p, n, nil); 299 p += n; 300 return p; 301 }