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 57fcfc2a 2005-02-13 devnull 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 57fcfc2a 2005-02-13 devnull 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 57fcfc2a 2005-02-13 devnull 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 57fcfc2a 2005-02-13 devnull 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 57fcfc2a 2005-02-13 devnull 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 int regen;
168 57fcfc2a 2005-02-13 devnull Attr *a;
170 57fcfc2a 2005-02-13 devnull if(argc == 0)
171 57fcfc2a 2005-02-13 devnull file = "/dev/stdin";
173 57fcfc2a 2005-02-13 devnull file = argv[0];
175 57fcfc2a 2005-02-13 devnull key = mallocz(sizeof(RSApriv), 1);
176 57fcfc2a 2005-02-13 devnull if(key == nil)
177 57fcfc2a 2005-02-13 devnull return nil;
179 57fcfc2a 2005-02-13 devnull if((b = Bopen(file, OREAD)) == nil){
180 57fcfc2a 2005-02-13 devnull werrstr("open %s: %r", file);
181 57fcfc2a 2005-02-13 devnull return nil;
183 57fcfc2a 2005-02-13 devnull s = Brdstr(b, '\n', 1);
184 57fcfc2a 2005-02-13 devnull if(s == nil){
185 57fcfc2a 2005-02-13 devnull werrstr("read %s: %r", file);
186 57fcfc2a 2005-02-13 devnull return nil;
188 57fcfc2a 2005-02-13 devnull if(strncmp(s, "key ", 4) != 0){
189 57fcfc2a 2005-02-13 devnull werrstr("bad key format");
190 57fcfc2a 2005-02-13 devnull return nil;
193 57fcfc2a 2005-02-13 devnull regen = 0;
194 57fcfc2a 2005-02-13 devnull a = _parseattr(s+4);
195 57fcfc2a 2005-02-13 devnull if(a == nil){
196 57fcfc2a 2005-02-13 devnull werrstr("empty key");
197 57fcfc2a 2005-02-13 devnull return nil;
199 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "proto")) == nil){
200 57fcfc2a 2005-02-13 devnull werrstr("no proto");
201 57fcfc2a 2005-02-13 devnull return nil;
203 57fcfc2a 2005-02-13 devnull if(strcmp(p, "dsa") != 0){
204 57fcfc2a 2005-02-13 devnull werrstr("proto not dsa");
205 57fcfc2a 2005-02-13 devnull return nil;
207 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "p")) == nil){
208 57fcfc2a 2005-02-13 devnull werrstr("no p");
209 57fcfc2a 2005-02-13 devnull return nil;
211 57fcfc2a 2005-02-13 devnull if((key->pub.p = strtomp(p, &p, 16, nil)) == nil || *p != 0){
212 57fcfc2a 2005-02-13 devnull werrstr("bad p");
213 57fcfc2a 2005-02-13 devnull return nil;
215 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "q")) == nil){
216 57fcfc2a 2005-02-13 devnull werrstr("no q");
217 57fcfc2a 2005-02-13 devnull return nil;
219 57fcfc2a 2005-02-13 devnull if((key->pub.q = strtomp(p, &p, 16, nil)) == nil || *p != 0){
220 57fcfc2a 2005-02-13 devnull werrstr("bad q");
221 57fcfc2a 2005-02-13 devnull return nil;
223 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "alpha")) == nil){
224 57fcfc2a 2005-02-13 devnull werrstr("no alpha");
225 57fcfc2a 2005-02-13 devnull return nil;
227 57fcfc2a 2005-02-13 devnull if((key->pub.alpha = strtomp(p, &p, 16, nil)) == nil || *p != 0){
228 57fcfc2a 2005-02-13 devnull werrstr("bad alpha");
229 57fcfc2a 2005-02-13 devnull return nil;
231 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "key")) == nil){
232 57fcfc2a 2005-02-13 devnull werrstr("no key=");
233 57fcfc2a 2005-02-13 devnull return nil;
235 57fcfc2a 2005-02-13 devnull if((key->pub.key = strtomp(p, &p, 16, nil)) == nil || *p != 0){
236 57fcfc2a 2005-02-13 devnull werrstr("bad key=");
237 57fcfc2a 2005-02-13 devnull return nil;
239 57fcfc2a 2005-02-13 devnull if(!needprivate)
240 57fcfc2a 2005-02-13 devnull goto call;
241 57fcfc2a 2005-02-13 devnull if((p = _strfindattr(a, "!secret")) == nil){
242 57fcfc2a 2005-02-13 devnull werrstr("no !secret");
243 57fcfc2a 2005-02-13 devnull return nil;
245 57fcfc2a 2005-02-13 devnull if((key->secret = strtomp(p, &p, 16, nil)) == nil || *p != 0){
246 57fcfc2a 2005-02-13 devnull werrstr("bad !secret");
247 57fcfc2a 2005-02-13 devnull return nil;
250 57fcfc2a 2005-02-13 devnull a = _delattr(a, "p");
251 57fcfc2a 2005-02-13 devnull a = _delattr(a, "q");
252 57fcfc2a 2005-02-13 devnull a = _delattr(a, "alpha");
253 57fcfc2a 2005-02-13 devnull a = _delattr(a, "key");
254 57fcfc2a 2005-02-13 devnull a = _delattr(a, "!secret");
256 57fcfc2a 2005-02-13 devnull *pa = a;
257 57fcfc2a 2005-02-13 devnull return key;
261 57fcfc2a 2005-02-13 devnull put4(uchar *p, uint n)
263 57fcfc2a 2005-02-13 devnull p[0] = (n>>24)&0xFF;
264 57fcfc2a 2005-02-13 devnull p[1] = (n>>16)&0xFF;
265 57fcfc2a 2005-02-13 devnull p[2] = (n>>8)&0xFF;
266 57fcfc2a 2005-02-13 devnull p[3] = n&0xFF;
267 57fcfc2a 2005-02-13 devnull return p+4;
271 57fcfc2a 2005-02-13 devnull putn(uchar *p, void *v, uint n)
273 57fcfc2a 2005-02-13 devnull memmove(p, v, n);
275 57fcfc2a 2005-02-13 devnull return p;
279 57fcfc2a 2005-02-13 devnull putstr(uchar *p, char *s)
281 57fcfc2a 2005-02-13 devnull p = put4(p, strlen(s));
282 57fcfc2a 2005-02-13 devnull p = putn(p, s, strlen(s));
283 57fcfc2a 2005-02-13 devnull return p;
287 57fcfc2a 2005-02-13 devnull putmp2(uchar *p, mpint *b)
289 57fcfc2a 2005-02-13 devnull int bits, n;
291 57fcfc2a 2005-02-13 devnull if(mpcmp(b, mpzero) == 0)
292 57fcfc2a 2005-02-13 devnull return put4(p, 0);
293 57fcfc2a 2005-02-13 devnull bits = mpsignif(b);
294 57fcfc2a 2005-02-13 devnull n = (bits+7)/8;
295 57fcfc2a 2005-02-13 devnull if(bits%8 == 0){
296 57fcfc2a 2005-02-13 devnull p = put4(p, n+1);
297 57fcfc2a 2005-02-13 devnull *p++ = 0;
299 57fcfc2a 2005-02-13 devnull p = put4(p, n);
300 57fcfc2a 2005-02-13 devnull mptobe(b, p, n, nil);
302 57fcfc2a 2005-02-13 devnull return p;