1 57fcfc2a 2005-02-13 devnull #include <u.h>
2 57fcfc2a 2005-02-13 devnull #include <libc.h>
3 57fcfc2a 2005-02-13 devnull #include <bio.h>
4 57fcfc2a 2005-02-13 devnull #include <auth.h>
5 57fcfc2a 2005-02-13 devnull #include <mp.h>
6 57fcfc2a 2005-02-13 devnull #include <libsec.h>
7 57fcfc2a 2005-02-13 devnull #include "rsa2any.h"
10 57fcfc2a 2005-02-13 devnull getkey(int argc, char **argv, int needprivate, Attr **pa)
12 57fcfc2a 2005-02-13 devnull char *file, *s, *p;
14 57fcfc2a 2005-02-13 devnull RSApriv *key;
15 57fcfc2a 2005-02-13 devnull Biobuf *b;
16 57fcfc2a 2005-02-13 devnull int regen;
19 57fcfc2a 2005-02-13 devnull if(argc == 0)
20 57fcfc2a 2005-02-13 devnull file = "/dev/stdin";
22 57fcfc2a 2005-02-13 devnull file = argv[0];
24 57fcfc2a 2005-02-13 devnull key = mallocz(sizeof(RSApriv), 1);
25 57fcfc2a 2005-02-13 devnull if(key == nil)
26 57fcfc2a 2005-02-13 devnull return nil;
28 57fcfc2a 2005-02-13 devnull if((b = Bopen(file, OREAD)) == nil){
29 57fcfc2a 2005-02-13 devnull werrstr("open %s: %r", file);
30 57fcfc2a 2005-02-13 devnull return nil;
32 57fcfc2a 2005-02-13 devnull s = Brdstr(b, '\n', 1);
33 57fcfc2a 2005-02-13 devnull if(s == nil){
34 57fcfc2a 2005-02-13 devnull werrstr("read %s: %r", file);
35 57fcfc2a 2005-02-13 devnull return nil;
37 57fcfc2a 2005-02-13 devnull if(strncmp(s, "key ", 4) != 0){
38 57fcfc2a 2005-02-13 devnull werrstr("bad key format");
39 57fcfc2a 2005-02-13 devnull return nil;
42 57fcfc2a 2005-02-13 devnull regen = 0;
43 57fcfc2a 2005-02-13 devnull a = _parseattr(s+4);
44 57fcfc2a 2005-02-13 devnull if(a == nil){
45 57fcfc2a 2005-02-13 devnull werrstr("empty key");
46 57fcfc2a 2005-02-13 devnull return nil;
48 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "proto")) == nil){
49 57fcfc2a 2005-02-13 devnull werrstr("no proto");
50 57fcfc2a 2005-02-13 devnull return nil;
52 57fcfc2a 2005-02-13 devnull if(strcmp(p, "rsa") != 0){
53 57fcfc2a 2005-02-13 devnull werrstr("proto not rsa");
54 57fcfc2a 2005-02-13 devnull return nil;
56 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "ek")) == nil){
57 57fcfc2a 2005-02-13 devnull werrstr("no ek");
58 57fcfc2a 2005-02-13 devnull return nil;
60 57fcfc2a 2005-02-13 devnull if((key->pub.ek = strtomp(p, &p, 16, nil)) == nil || *p != 0){
61 57fcfc2a 2005-02-13 devnull werrstr("bad ek");
62 57fcfc2a 2005-02-13 devnull return nil;
64 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "n")) == nil){
65 57fcfc2a 2005-02-13 devnull werrstr("no n");
66 57fcfc2a 2005-02-13 devnull return nil;
68 57fcfc2a 2005-02-13 devnull if((key->pub.n = strtomp(p, &p, 16, nil)) == nil || *p != 0){
69 57fcfc2a 2005-02-13 devnull werrstr("bad n");
70 57fcfc2a 2005-02-13 devnull return nil;
72 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "size")) == nil)
73 57fcfc2a 2005-02-13 devnull fprint(2, "warning: missing size; will add\n");
74 57fcfc2a 2005-02-13 devnull else if((sz = strtol(p, &p, 10)) == 0 || *p != 0)
75 57fcfc2a 2005-02-13 devnull fprint(2, "warning: bad size; will correct\n");
76 57fcfc2a 2005-02-13 devnull else if(sz != mpsignif(key->pub.n))
77 57fcfc2a 2005-02-13 devnull fprint(2, "warning: wrong size (got %d, expected %d); will correct\n",
78 57fcfc2a 2005-02-13 devnull sz, mpsignif(key->pub.n));
79 57fcfc2a 2005-02-13 devnull if(!needprivate)
80 57fcfc2a 2005-02-13 devnull goto call;
81 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!dk")) == nil){
82 57fcfc2a 2005-02-13 devnull werrstr("no !dk");
83 57fcfc2a 2005-02-13 devnull return nil;
85 57fcfc2a 2005-02-13 devnull if((key->dk = strtomp(p, &p, 16, nil)) == nil || *p != 0){
86 57fcfc2a 2005-02-13 devnull werrstr("bad !dk");
87 57fcfc2a 2005-02-13 devnull return nil;
89 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!p")) == nil){
90 57fcfc2a 2005-02-13 devnull werrstr("no !p");
91 57fcfc2a 2005-02-13 devnull return nil;
93 57fcfc2a 2005-02-13 devnull if((key->p = strtomp(p, &p, 16, nil)) == nil || *p != 0){
94 57fcfc2a 2005-02-13 devnull werrstr("bad !p");
95 57fcfc2a 2005-02-13 devnull return nil;
97 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!q")) == nil){
98 57fcfc2a 2005-02-13 devnull werrstr("no !q");
99 57fcfc2a 2005-02-13 devnull return nil;
101 57fcfc2a 2005-02-13 devnull if((key->q = strtomp(p, &p, 16, nil)) == nil || *p != 0){
102 57fcfc2a 2005-02-13 devnull werrstr("bad !q");
103 57fcfc2a 2005-02-13 devnull return nil;
105 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!kp")) == nil){
106 57fcfc2a 2005-02-13 devnull fprint(2, "warning: no !kp\n");
107 57fcfc2a 2005-02-13 devnull regen = 1;
108 57fcfc2a 2005-02-13 devnull goto regen;
110 57fcfc2a 2005-02-13 devnull if((key->kp = strtomp(p, &p, 16, nil)) == nil || *p != 0){
111 57fcfc2a 2005-02-13 devnull fprint(2, "warning: bad !kp\n");
112 fa325e9b 2020-01-10 cross regen = 1;
113 57fcfc2a 2005-02-13 devnull goto regen;
115 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!kq")) == nil){
116 57fcfc2a 2005-02-13 devnull fprint(2, "warning: no !kq\n");
117 fa325e9b 2020-01-10 cross regen = 1;
118 57fcfc2a 2005-02-13 devnull goto regen;
120 57fcfc2a 2005-02-13 devnull if((key->kq = strtomp(p, &p, 16, nil)) == nil || *p != 0){
121 57fcfc2a 2005-02-13 devnull fprint(2, "warning: bad !kq\n");
122 fa325e9b 2020-01-10 cross regen = 1;
123 57fcfc2a 2005-02-13 devnull goto regen;
125 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!c2")) == nil){
126 57fcfc2a 2005-02-13 devnull fprint(2, "warning: no !c2\n");
127 fa325e9b 2020-01-10 cross regen = 1;
128 57fcfc2a 2005-02-13 devnull goto regen;
130 57fcfc2a 2005-02-13 devnull if((key->c2 = strtomp(p, &p, 16, nil)) == nil || *p != 0){
131 57fcfc2a 2005-02-13 devnull fprint(2, "warning: bad !c2\n");
132 fa325e9b 2020-01-10 cross regen = 1;
133 57fcfc2a 2005-02-13 devnull goto regen;
136 57fcfc2a 2005-02-13 devnull if(regen){
137 57fcfc2a 2005-02-13 devnull RSApriv *k2;
139 57fcfc2a 2005-02-13 devnull k2 = rsafill(key->pub.n, key->pub.ek, key->dk, key->p, key->q);
140 57fcfc2a 2005-02-13 devnull if(k2 == nil){
141 57fcfc2a 2005-02-13 devnull werrstr("regenerating chinese-remainder parts failed: %r");
142 57fcfc2a 2005-02-13 devnull return nil;
144 57fcfc2a 2005-02-13 devnull key = k2;
147 57fcfc2a 2005-02-13 devnull a = _delattr(a, "ek");
148 57fcfc2a 2005-02-13 devnull a = _delattr(a, "n");
149 57fcfc2a 2005-02-13 devnull a = _delattr(a, "size");
150 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!dk");
151 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!p");
152 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!q");
153 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!c2");
154 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!kp");
155 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!kq");
157 57fcfc2a 2005-02-13 devnull *pa = a;
158 57fcfc2a 2005-02-13 devnull return key;
161 57fcfc2a 2005-02-13 devnull DSApriv*
162 57fcfc2a 2005-02-13 devnull getdsakey(int argc, char **argv, int needprivate, Attr **pa)
164 57fcfc2a 2005-02-13 devnull char *file, *s, *p;
165 57fcfc2a 2005-02-13 devnull DSApriv *key;
166 57fcfc2a 2005-02-13 devnull Biobuf *b;
167 57fcfc2a 2005-02-13 devnull Attr *a;
169 57fcfc2a 2005-02-13 devnull if(argc == 0)
170 57fcfc2a 2005-02-13 devnull file = "/dev/stdin";
172 57fcfc2a 2005-02-13 devnull file = argv[0];
174 57fcfc2a 2005-02-13 devnull key = mallocz(sizeof(RSApriv), 1);
175 57fcfc2a 2005-02-13 devnull if(key == nil)
176 57fcfc2a 2005-02-13 devnull return nil;
178 57fcfc2a 2005-02-13 devnull if((b = Bopen(file, OREAD)) == nil){
179 57fcfc2a 2005-02-13 devnull werrstr("open %s: %r", file);
180 57fcfc2a 2005-02-13 devnull return nil;
182 57fcfc2a 2005-02-13 devnull s = Brdstr(b, '\n', 1);
183 57fcfc2a 2005-02-13 devnull if(s == nil){
184 57fcfc2a 2005-02-13 devnull werrstr("read %s: %r", file);
185 57fcfc2a 2005-02-13 devnull return nil;
187 57fcfc2a 2005-02-13 devnull if(strncmp(s, "key ", 4) != 0){
188 57fcfc2a 2005-02-13 devnull werrstr("bad key format");
189 57fcfc2a 2005-02-13 devnull return nil;
192 57fcfc2a 2005-02-13 devnull a = _parseattr(s+4);
193 57fcfc2a 2005-02-13 devnull if(a == nil){
194 57fcfc2a 2005-02-13 devnull werrstr("empty key");
195 57fcfc2a 2005-02-13 devnull return nil;
197 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "proto")) == nil){
198 57fcfc2a 2005-02-13 devnull werrstr("no proto");
199 57fcfc2a 2005-02-13 devnull return nil;
201 57fcfc2a 2005-02-13 devnull if(strcmp(p, "dsa") != 0){
202 57fcfc2a 2005-02-13 devnull werrstr("proto not dsa");
203 57fcfc2a 2005-02-13 devnull return nil;
205 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "p")) == nil){
206 57fcfc2a 2005-02-13 devnull werrstr("no p");
207 57fcfc2a 2005-02-13 devnull return nil;
209 57fcfc2a 2005-02-13 devnull if((key->pub.p = strtomp(p, &p, 16, nil)) == nil || *p != 0){
210 57fcfc2a 2005-02-13 devnull werrstr("bad p");
211 57fcfc2a 2005-02-13 devnull return nil;
213 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "q")) == nil){
214 57fcfc2a 2005-02-13 devnull werrstr("no q");
215 57fcfc2a 2005-02-13 devnull return nil;
217 57fcfc2a 2005-02-13 devnull if((key->pub.q = strtomp(p, &p, 16, nil)) == nil || *p != 0){
218 57fcfc2a 2005-02-13 devnull werrstr("bad q");
219 57fcfc2a 2005-02-13 devnull return nil;
221 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "alpha")) == nil){
222 57fcfc2a 2005-02-13 devnull werrstr("no alpha");
223 57fcfc2a 2005-02-13 devnull return nil;
225 57fcfc2a 2005-02-13 devnull if((key->pub.alpha = strtomp(p, &p, 16, nil)) == nil || *p != 0){
226 57fcfc2a 2005-02-13 devnull werrstr("bad alpha");
227 57fcfc2a 2005-02-13 devnull return nil;
229 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "key")) == nil){
230 57fcfc2a 2005-02-13 devnull werrstr("no key=");
231 57fcfc2a 2005-02-13 devnull return nil;
233 57fcfc2a 2005-02-13 devnull if((key->pub.key = strtomp(p, &p, 16, nil)) == nil || *p != 0){
234 57fcfc2a 2005-02-13 devnull werrstr("bad key=");
235 57fcfc2a 2005-02-13 devnull return nil;
237 57fcfc2a 2005-02-13 devnull if(!needprivate)
238 57fcfc2a 2005-02-13 devnull goto call;
239 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!secret")) == nil){
240 57fcfc2a 2005-02-13 devnull werrstr("no !secret");
241 57fcfc2a 2005-02-13 devnull return nil;
243 57fcfc2a 2005-02-13 devnull if((key->secret = strtomp(p, &p, 16, nil)) == nil || *p != 0){
244 57fcfc2a 2005-02-13 devnull werrstr("bad !secret");
245 57fcfc2a 2005-02-13 devnull return nil;
248 57fcfc2a 2005-02-13 devnull a = _delattr(a, "p");
249 57fcfc2a 2005-02-13 devnull a = _delattr(a, "q");
250 57fcfc2a 2005-02-13 devnull a = _delattr(a, "alpha");
251 57fcfc2a 2005-02-13 devnull a = _delattr(a, "key");
252 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!secret");
254 57fcfc2a 2005-02-13 devnull *pa = a;
255 57fcfc2a 2005-02-13 devnull return key;
259 57fcfc2a 2005-02-13 devnull put4(uchar *p, uint n)
261 57fcfc2a 2005-02-13 devnull p[0] = (n>>24)&0xFF;
262 57fcfc2a 2005-02-13 devnull p[1] = (n>>16)&0xFF;
263 57fcfc2a 2005-02-13 devnull p[2] = (n>>8)&0xFF;
264 57fcfc2a 2005-02-13 devnull p[3] = n&0xFF;
265 57fcfc2a 2005-02-13 devnull return p+4;
269 57fcfc2a 2005-02-13 devnull putn(uchar *p, void *v, uint n)
271 57fcfc2a 2005-02-13 devnull memmove(p, v, n);
273 57fcfc2a 2005-02-13 devnull return p;
277 57fcfc2a 2005-02-13 devnull putstr(uchar *p, char *s)
279 57fcfc2a 2005-02-13 devnull p = put4(p, strlen(s));
280 57fcfc2a 2005-02-13 devnull p = putn(p, s, strlen(s));
281 57fcfc2a 2005-02-13 devnull return p;
285 57fcfc2a 2005-02-13 devnull putmp2(uchar *p, mpint *b)
287 57fcfc2a 2005-02-13 devnull int bits, n;
289 57fcfc2a 2005-02-13 devnull if(mpcmp(b, mpzero) == 0)
290 57fcfc2a 2005-02-13 devnull return put4(p, 0);
291 57fcfc2a 2005-02-13 devnull bits = mpsignif(b);
292 57fcfc2a 2005-02-13 devnull n = (bits+7)/8;
293 57fcfc2a 2005-02-13 devnull if(bits%8 == 0){
294 57fcfc2a 2005-02-13 devnull p = put4(p, n+1);
295 57fcfc2a 2005-02-13 devnull *p++ = 0;
297 57fcfc2a 2005-02-13 devnull p = put4(p, n);
298 57fcfc2a 2005-02-13 devnull mptobe(b, p, n, nil);
300 57fcfc2a 2005-02-13 devnull return p;