r4088: Get medieval on our ass about malloc.... :-). Take control of all our allocation
[tprouty/samba.git] / source / libsmb / asn1.c
1 /* 
2    Unix SMB/CIFS implementation.
3    simple SPNEGO routines
4    Copyright (C) Andrew Tridgell 2001
5    
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 2 of the License, or
9    (at your option) any later version.
10    
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15    
16    You should have received a copy of the GNU General Public License
17    along with this program; if not, write to the Free Software
18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19 */
20
21 #include "includes.h"
22
23 /* free an asn1 structure */
24 void asn1_free(ASN1_DATA *data)
25 {
26         SAFE_FREE(data->data);
27 }
28
29 /* write to the ASN1 buffer, advancing the buffer pointer */
30 BOOL asn1_write(ASN1_DATA *data, const void *p, int len)
31 {
32         if (data->has_error) return False;
33         if (data->length < data->ofs+len) {
34                 uint8 *newp;
35                 newp = SMB_REALLOC(data->data, data->ofs+len);
36                 if (!newp) {
37                         SAFE_FREE(data->data);
38                         data->has_error = True;
39                         return False;
40                 }
41                 data->data = newp;
42                 data->length = data->ofs+len;
43         }
44         memcpy(data->data + data->ofs, p, len);
45         data->ofs += len;
46         return True;
47 }
48
49 /* useful fn for writing a uint8 */
50 BOOL asn1_write_uint8(ASN1_DATA *data, uint8 v)
51 {
52         return asn1_write(data, &v, 1);
53 }
54
55 /* push a tag onto the asn1 data buffer. Used for nested structures */
56 BOOL asn1_push_tag(ASN1_DATA *data, uint8 tag)
57 {
58         struct nesting *nesting;
59
60         asn1_write_uint8(data, tag);
61         nesting = SMB_MALLOC_P(struct nesting);
62         if (!nesting) {
63                 data->has_error = True;
64                 return False;
65         }
66
67         nesting->start = data->ofs;
68         nesting->next = data->nesting;
69         data->nesting = nesting;
70         return asn1_write_uint8(data, 0xff);
71 }
72
73 /* pop a tag */
74 BOOL asn1_pop_tag(ASN1_DATA *data)
75 {
76         struct nesting *nesting;
77         size_t len;
78
79         nesting = data->nesting;
80
81         if (!nesting) {
82                 data->has_error = True;
83                 return False;
84         }
85         len = data->ofs - (nesting->start+1);
86         /* yes, this is ugly. We don't know in advance how many bytes the length
87            of a tag will take, so we assumed 1 byte. If we were wrong then we 
88            need to correct our mistake */
89         if (len > 255) {
90                 data->data[nesting->start] = 0x82;
91                 if (!asn1_write_uint8(data, 0)) return False;
92                 if (!asn1_write_uint8(data, 0)) return False;
93                 memmove(data->data+nesting->start+3, data->data+nesting->start+1, len);
94                 data->data[nesting->start+1] = len>>8;
95                 data->data[nesting->start+2] = len&0xff;
96         } else if (len > 127) {
97                 data->data[nesting->start] = 0x81;
98                 if (!asn1_write_uint8(data, 0)) return False;
99                 memmove(data->data+nesting->start+2, data->data+nesting->start+1, len);
100                 data->data[nesting->start+1] = len;
101         } else {
102                 data->data[nesting->start] = len;
103         }
104
105         data->nesting = nesting->next;
106         free(nesting);
107         return True;
108 }
109
110
111 /* write an integer */
112 BOOL asn1_write_Integer(ASN1_DATA *data, int i)
113 {
114         if (!asn1_push_tag(data, ASN1_INTEGER)) return False;
115         do {
116                 asn1_write_uint8(data, i);
117                 i = i >> 8;
118         } while (i);
119         return asn1_pop_tag(data);
120 }
121
122 /* write an object ID to a ASN1 buffer */
123 BOOL asn1_write_OID(ASN1_DATA *data, const char *OID)
124 {
125         unsigned v, v2;
126         const char *p = (const char *)OID;
127         char *newp;
128
129         if (!asn1_push_tag(data, ASN1_OID))
130                 return False;
131         v = strtol(p, &newp, 10);
132         p = newp;
133         v2 = strtol(p, &newp, 10);
134         p = newp;
135         if (!asn1_write_uint8(data, 40*v + v2))
136                 return False;
137
138         while (*p) {
139                 v = strtol(p, &newp, 10);
140                 p = newp;
141                 if (v >= (1<<28)) asn1_write_uint8(data, 0x80 | ((v>>28)&0xff));
142                 if (v >= (1<<21)) asn1_write_uint8(data, 0x80 | ((v>>21)&0xff));
143                 if (v >= (1<<14)) asn1_write_uint8(data, 0x80 | ((v>>14)&0xff));
144                 if (v >= (1<<7)) asn1_write_uint8(data, 0x80 | ((v>>7)&0xff));
145                 if (!asn1_write_uint8(data, v&0x7f))
146                         return False;
147         }
148         return asn1_pop_tag(data);
149 }
150
151 /* write an octet string */
152 BOOL asn1_write_OctetString(ASN1_DATA *data, const void *p, size_t length)
153 {
154         asn1_push_tag(data, ASN1_OCTET_STRING);
155         asn1_write(data, p, length);
156         asn1_pop_tag(data);
157         return !data->has_error;
158 }
159
160 /* write a general string */
161 BOOL asn1_write_GeneralString(ASN1_DATA *data, const char *s)
162 {
163         asn1_push_tag(data, ASN1_GENERAL_STRING);
164         asn1_write(data, s, strlen(s));
165         asn1_pop_tag(data);
166         return !data->has_error;
167 }
168
169 /* write a BOOLEAN */
170 BOOL asn1_write_BOOLEAN(ASN1_DATA *data, BOOL v)
171 {
172         asn1_write_uint8(data, ASN1_BOOLEAN);
173         asn1_write_uint8(data, v);
174         return !data->has_error;
175 }
176
177 /* write a BOOLEAN - hmm, I suspect this one is the correct one, and the 
178    above boolean is bogus. Need to check */
179 BOOL asn1_write_BOOLEAN2(ASN1_DATA *data, BOOL v)
180 {
181         asn1_push_tag(data, ASN1_BOOLEAN);
182         asn1_write_uint8(data, v);
183         asn1_pop_tag(data);
184         return !data->has_error;
185 }
186
187 /* check a BOOLEAN */
188 BOOL asn1_check_BOOLEAN(ASN1_DATA *data, BOOL v)
189 {
190         uint8 b = 0;
191
192         asn1_read_uint8(data, &b);
193         if (b != ASN1_BOOLEAN) {
194                 data->has_error = True;
195                 return False;
196         }
197         asn1_read_uint8(data, &b);
198         if (b != v) {
199                 data->has_error = True;
200                 return False;
201         }
202         return !data->has_error;
203 }
204
205
206 /* load a ASN1_DATA structure with a lump of data, ready to be parsed */
207 BOOL asn1_load(ASN1_DATA *data, DATA_BLOB blob)
208 {
209         ZERO_STRUCTP(data);
210         data->data = memdup(blob.data, blob.length);
211         if (!data->data) {
212                 data->has_error = True;
213                 return False;
214         }
215         data->length = blob.length;
216         return True;
217 }
218
219 /* read from a ASN1 buffer, advancing the buffer pointer */
220 BOOL asn1_read(ASN1_DATA *data, void *p, int len)
221 {
222         if (data->has_error)
223                 return False;
224
225         if (len < 0 || data->ofs + len < data->ofs || data->ofs + len < len) {
226                 data->has_error = True;
227                 return False;
228         }
229
230         if (data->ofs + len > data->length) {
231                 data->has_error = True;
232                 return False;
233         }
234         memcpy(p, data->data + data->ofs, len);
235         data->ofs += len;
236         return True;
237 }
238
239 /* read a uint8 from a ASN1 buffer */
240 BOOL asn1_read_uint8(ASN1_DATA *data, uint8 *v)
241 {
242         return asn1_read(data, v, 1);
243 }
244
245 /* start reading a nested asn1 structure */
246 BOOL asn1_start_tag(ASN1_DATA *data, uint8 tag)
247 {
248         uint8 b;
249         struct nesting *nesting;
250         
251         if (!asn1_read_uint8(data, &b))
252                 return False;
253
254         if (b != tag) {
255                 data->has_error = True;
256                 return False;
257         }
258         nesting = SMB_MALLOC_P(struct nesting);
259         if (!nesting) {
260                 data->has_error = True;
261                 return False;
262         }
263
264         if (!asn1_read_uint8(data, &b)) {
265                 return False;
266         }
267
268         if (b & 0x80) {
269                 int n = b & 0x7f;
270                 if (!asn1_read_uint8(data, &b))
271                         return False;
272                 nesting->taglen = b;
273                 while (n > 1) {
274                         if (!asn1_read_uint8(data, &b)) 
275                                 return False;
276                         nesting->taglen = (nesting->taglen << 8) | b;
277                         n--;
278                 }
279         } else {
280                 nesting->taglen = b;
281         }
282         nesting->start = data->ofs;
283         nesting->next = data->nesting;
284         data->nesting = nesting;
285         return !data->has_error;
286 }
287
288
289 /* stop reading a tag */
290 BOOL asn1_end_tag(ASN1_DATA *data)
291 {
292         struct nesting *nesting;
293
294         /* make sure we read it all */
295         if (asn1_tag_remaining(data) != 0) {
296                 data->has_error = True;
297                 return False;
298         }
299
300         nesting = data->nesting;
301
302         if (!nesting) {
303                 data->has_error = True;
304                 return False;
305         }
306
307         data->nesting = nesting->next;
308         free(nesting);
309         return True;
310 }
311
312 /* work out how many bytes are left in this nested tag */
313 int asn1_tag_remaining(ASN1_DATA *data)
314 {
315         if (data->has_error)
316                 return 0;
317
318         if (!data->nesting) {
319                 data->has_error = True;
320                 return -1;
321         }
322         return data->nesting->taglen - (data->ofs - data->nesting->start);
323 }
324
325 /* read an object ID from a ASN1 buffer */
326 BOOL asn1_read_OID(ASN1_DATA *data, char **OID)
327 {
328         uint8 b;
329         pstring oid_str;
330         fstring el;
331
332         if (!asn1_start_tag(data, ASN1_OID)) return False;
333         asn1_read_uint8(data, &b);
334
335         oid_str[0] = 0;
336         fstr_sprintf(el, "%u",  b/40);
337         pstrcat(oid_str, el);
338         fstr_sprintf(el, " %u",  b%40);
339         pstrcat(oid_str, el);
340
341         while (asn1_tag_remaining(data) > 0) {
342                 unsigned v = 0;
343                 do {
344                         asn1_read_uint8(data, &b);
345                         v = (v<<7) | (b&0x7f);
346                 } while (!data->has_error && b & 0x80);
347                 fstr_sprintf(el, " %u",  v);
348                 pstrcat(oid_str, el);
349         }
350
351         asn1_end_tag(data);
352
353         *OID = SMB_STRDUP(oid_str);
354
355         return !data->has_error;
356 }
357
358 /* check that the next object ID is correct */
359 BOOL asn1_check_OID(ASN1_DATA *data, const char *OID)
360 {
361         char *id;
362
363         if (!asn1_read_OID(data, &id)) return False;
364
365         if (strcmp(id, OID) != 0) {
366                 data->has_error = True;
367                 return False;
368         }
369         free(id);
370         return True;
371 }
372
373 /* read a GeneralString from a ASN1 buffer */
374 BOOL asn1_read_GeneralString(ASN1_DATA *data, char **s)
375 {
376         int len;
377         if (!asn1_start_tag(data, ASN1_GENERAL_STRING)) return False;
378         len = asn1_tag_remaining(data);
379         if (len < 0) {
380                 data->has_error = True;
381                 return False;
382         }
383         *s = SMB_MALLOC(len+1);
384         if (! *s) {
385                 data->has_error = True;
386                 return False;
387         }
388         asn1_read(data, *s, len);
389         (*s)[len] = 0;
390         asn1_end_tag(data);
391         return !data->has_error;
392 }
393
394 /* read a octet string blob */
395 BOOL asn1_read_OctetString(ASN1_DATA *data, DATA_BLOB *blob)
396 {
397         int len;
398         ZERO_STRUCTP(blob);
399         if (!asn1_start_tag(data, ASN1_OCTET_STRING)) return False;
400         len = asn1_tag_remaining(data);
401         if (len < 0) {
402                 data->has_error = True;
403                 return False;
404         }
405         *blob = data_blob(NULL, len);
406         asn1_read(data, blob->data, len);
407         asn1_end_tag(data);
408         return !data->has_error;
409 }
410
411 /* read an interger */
412 BOOL asn1_read_Integer(ASN1_DATA *data, int *i)
413 {
414         uint8 b;
415         *i = 0;
416         
417         if (!asn1_start_tag(data, ASN1_INTEGER)) return False;
418         while (asn1_tag_remaining(data)>0) {
419                 asn1_read_uint8(data, &b);
420                 *i = (*i << 8) + b;
421         }
422         return asn1_end_tag(data);      
423         
424 }
425
426 /* check a enumarted value is correct */
427 BOOL asn1_check_enumerated(ASN1_DATA *data, int v)
428 {
429         uint8 b;
430         if (!asn1_start_tag(data, ASN1_ENUMERATED)) return False;
431         asn1_read_uint8(data, &b);
432         asn1_end_tag(data);
433
434         if (v != b)
435                 data->has_error = False;
436
437         return !data->has_error;
438 }
439
440 /* write an enumarted value to the stream */
441 BOOL asn1_write_enumerated(ASN1_DATA *data, uint8 v)
442 {
443         if (!asn1_push_tag(data, ASN1_ENUMERATED)) return False;
444         asn1_write_uint8(data, v);
445         asn1_pop_tag(data);
446         return !data->has_error;
447 }