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"
  8 57fcfc2a 2005-02-13 devnull 
  9 57fcfc2a 2005-02-13 devnull RSApriv*
 10 57fcfc2a 2005-02-13 devnull getkey(int argc, char **argv, int needprivate, Attr **pa)
 11 57fcfc2a 2005-02-13 devnull {
 12 57fcfc2a 2005-02-13 devnull 	char *file, *s, *p;
 13 57fcfc2a 2005-02-13 devnull 	int sz;
 14 57fcfc2a 2005-02-13 devnull 	RSApriv *key;
 15 57fcfc2a 2005-02-13 devnull 	Biobuf *b;
 16 57fcfc2a 2005-02-13 devnull 	int regen;
 17 57fcfc2a 2005-02-13 devnull 	Attr *a;
 18 57fcfc2a 2005-02-13 devnull 
 19 57fcfc2a 2005-02-13 devnull 	if(argc == 0)
 20 57fcfc2a 2005-02-13 devnull 		file = "/dev/stdin";
 21 57fcfc2a 2005-02-13 devnull 	else
 22 57fcfc2a 2005-02-13 devnull 		file = argv[0];
 23 57fcfc2a 2005-02-13 devnull 
 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;
 27 57fcfc2a 2005-02-13 devnull 
 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;
 31 57fcfc2a 2005-02-13 devnull 	}
 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;
 36 57fcfc2a 2005-02-13 devnull 	}
 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;
 40 57fcfc2a 2005-02-13 devnull 	}
 41 57fcfc2a 2005-02-13 devnull 
 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;
 47 57fcfc2a 2005-02-13 devnull 	}
 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;
 51 57fcfc2a 2005-02-13 devnull 	}
 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;
 55 57fcfc2a 2005-02-13 devnull 	}
 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;
 59 57fcfc2a 2005-02-13 devnull 	}
 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;
 63 57fcfc2a 2005-02-13 devnull 	}
 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;
 67 57fcfc2a 2005-02-13 devnull 	}
 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;
 71 57fcfc2a 2005-02-13 devnull 	}
 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;
 84 57fcfc2a 2005-02-13 devnull 	}
 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;
 88 57fcfc2a 2005-02-13 devnull 	}
 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;
 92 57fcfc2a 2005-02-13 devnull 	}
 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;
 96 57fcfc2a 2005-02-13 devnull 	}
 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;
100 57fcfc2a 2005-02-13 devnull 	}
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;
104 57fcfc2a 2005-02-13 devnull 	}
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;
109 57fcfc2a 2005-02-13 devnull 	}
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;
114 57fcfc2a 2005-02-13 devnull 	}
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;
119 57fcfc2a 2005-02-13 devnull 	}
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;
124 57fcfc2a 2005-02-13 devnull 	}
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;
129 57fcfc2a 2005-02-13 devnull 	}
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;
134 57fcfc2a 2005-02-13 devnull 	}
135 57fcfc2a 2005-02-13 devnull regen:
136 57fcfc2a 2005-02-13 devnull 	if(regen){
137 57fcfc2a 2005-02-13 devnull 		RSApriv *k2;
138 57fcfc2a 2005-02-13 devnull 
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;
143 57fcfc2a 2005-02-13 devnull 		}
144 57fcfc2a 2005-02-13 devnull 		key = k2;
145 57fcfc2a 2005-02-13 devnull 	}
146 57fcfc2a 2005-02-13 devnull call:
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");
156 57fcfc2a 2005-02-13 devnull 	if(pa)
157 57fcfc2a 2005-02-13 devnull 		*pa = a;
158 57fcfc2a 2005-02-13 devnull 	return key;
159 57fcfc2a 2005-02-13 devnull }
160 57fcfc2a 2005-02-13 devnull 
161 57fcfc2a 2005-02-13 devnull DSApriv*
162 57fcfc2a 2005-02-13 devnull getdsakey(int argc, char **argv, int needprivate, Attr **pa)
163 57fcfc2a 2005-02-13 devnull {
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;
169 57fcfc2a 2005-02-13 devnull 
170 57fcfc2a 2005-02-13 devnull 	if(argc == 0)
171 57fcfc2a 2005-02-13 devnull 		file = "/dev/stdin";
172 57fcfc2a 2005-02-13 devnull 	else
173 57fcfc2a 2005-02-13 devnull 		file = argv[0];
174 57fcfc2a 2005-02-13 devnull 
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;
178 57fcfc2a 2005-02-13 devnull 
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;
182 57fcfc2a 2005-02-13 devnull 	}
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;
187 57fcfc2a 2005-02-13 devnull 	}
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;
191 57fcfc2a 2005-02-13 devnull 	}
192 57fcfc2a 2005-02-13 devnull 
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;
198 57fcfc2a 2005-02-13 devnull 	}
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;
202 57fcfc2a 2005-02-13 devnull 	}
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;
206 57fcfc2a 2005-02-13 devnull 	}
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;
210 57fcfc2a 2005-02-13 devnull 	}
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;
214 57fcfc2a 2005-02-13 devnull 	}
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;
218 57fcfc2a 2005-02-13 devnull 	}
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;
222 57fcfc2a 2005-02-13 devnull 	}
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;
226 57fcfc2a 2005-02-13 devnull 	}
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;
230 57fcfc2a 2005-02-13 devnull 	}
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;
234 57fcfc2a 2005-02-13 devnull 	}
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;
238 57fcfc2a 2005-02-13 devnull 	}
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;
244 57fcfc2a 2005-02-13 devnull 	}
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;
248 57fcfc2a 2005-02-13 devnull 	}
249 57fcfc2a 2005-02-13 devnull call:
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");
255 57fcfc2a 2005-02-13 devnull 	if(pa)
256 57fcfc2a 2005-02-13 devnull 		*pa = a;
257 57fcfc2a 2005-02-13 devnull 	return key;
258 57fcfc2a 2005-02-13 devnull }
259 57fcfc2a 2005-02-13 devnull 
260 57fcfc2a 2005-02-13 devnull uchar*
261 57fcfc2a 2005-02-13 devnull put4(uchar *p, uint n)
262 57fcfc2a 2005-02-13 devnull {
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;
268 57fcfc2a 2005-02-13 devnull }
269 57fcfc2a 2005-02-13 devnull 
270 57fcfc2a 2005-02-13 devnull uchar*
271 57fcfc2a 2005-02-13 devnull putn(uchar *p, void *v, uint n)
272 57fcfc2a 2005-02-13 devnull {
273 57fcfc2a 2005-02-13 devnull 	memmove(p, v, n);
274 57fcfc2a 2005-02-13 devnull 	p += n;
275 57fcfc2a 2005-02-13 devnull 	return p;
276 57fcfc2a 2005-02-13 devnull }
277 57fcfc2a 2005-02-13 devnull 
278 57fcfc2a 2005-02-13 devnull uchar*
279 57fcfc2a 2005-02-13 devnull putstr(uchar *p, char *s)
280 57fcfc2a 2005-02-13 devnull {
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;
284 57fcfc2a 2005-02-13 devnull }
285 57fcfc2a 2005-02-13 devnull 
286 57fcfc2a 2005-02-13 devnull uchar*
287 57fcfc2a 2005-02-13 devnull putmp2(uchar *p, mpint *b)
288 57fcfc2a 2005-02-13 devnull {
289 57fcfc2a 2005-02-13 devnull 	int bits, n;
290 57fcfc2a 2005-02-13 devnull 	
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;
298 57fcfc2a 2005-02-13 devnull 	}else
299 57fcfc2a 2005-02-13 devnull 		p = put4(p, n);
300 57fcfc2a 2005-02-13 devnull 	mptobe(b, p, n, nil);
301 57fcfc2a 2005-02-13 devnull 	p += n;
302 57fcfc2a 2005-02-13 devnull 	return p;
303 57fcfc2a 2005-02-13 devnull }