s4/asn1: Use explicite TALLOC_CTX in ber_write_OID functions
[ira/wip.git] / lib / util / asn1.c
1 /* 
2    Unix SMB/CIFS implementation.
3    simple ASN1 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 3 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, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include "includes.h"
21 #include "../lib/util/asn1.h"
22
23 /* allocate an asn1 structure */
24 struct asn1_data *asn1_init(TALLOC_CTX *mem_ctx)
25 {
26         struct asn1_data *ret = talloc_zero(mem_ctx, struct asn1_data);
27         if (ret == NULL) {
28                 DEBUG(0,("asn1_init failed! out of memory\n"));
29         }
30         return ret;
31 }
32
33 /* free an asn1 structure */
34 void asn1_free(struct asn1_data *data)
35 {
36         talloc_free(data);
37 }
38
39 /* write to the ASN1 buffer, advancing the buffer pointer */
40 bool asn1_write(struct asn1_data *data, const void *p, int len)
41 {
42         if (data->has_error) return false;
43         if (data->length < data->ofs+len) {
44                 uint8_t *newp;
45                 newp = talloc_realloc(data, data->data, uint8_t, data->ofs+len);
46                 if (!newp) {
47                         asn1_free(data);
48                         data->has_error = true;
49                         return false;
50                 }
51                 data->data = newp;
52                 data->length = data->ofs+len;
53         }
54         memcpy(data->data + data->ofs, p, len);
55         data->ofs += len;
56         return true;
57 }
58
59 /* useful fn for writing a uint8_t */
60 bool asn1_write_uint8(struct asn1_data *data, uint8_t v)
61 {
62         return asn1_write(data, &v, 1);
63 }
64
65 /* push a tag onto the asn1 data buffer. Used for nested structures */
66 bool asn1_push_tag(struct asn1_data *data, uint8_t tag)
67 {
68         struct nesting *nesting;
69
70         asn1_write_uint8(data, tag);
71         nesting = talloc(data, struct nesting);
72         if (!nesting) {
73                 data->has_error = true;
74                 return false;
75         }
76
77         nesting->start = data->ofs;
78         nesting->next = data->nesting;
79         data->nesting = nesting;
80         return asn1_write_uint8(data, 0xff);
81 }
82
83 /* pop a tag */
84 bool asn1_pop_tag(struct asn1_data *data)
85 {
86         struct nesting *nesting;
87         size_t len;
88
89         nesting = data->nesting;
90
91         if (!nesting) {
92                 data->has_error = true;
93                 return false;
94         }
95         len = data->ofs - (nesting->start+1);
96         /* yes, this is ugly. We don't know in advance how many bytes the length
97            of a tag will take, so we assumed 1 byte. If we were wrong then we 
98            need to correct our mistake */
99         if (len > 0xFFFFFF) {
100                 data->data[nesting->start] = 0x84;
101                 if (!asn1_write_uint8(data, 0)) return false;
102                 if (!asn1_write_uint8(data, 0)) return false;
103                 if (!asn1_write_uint8(data, 0)) return false;
104                 if (!asn1_write_uint8(data, 0)) return false;
105                 memmove(data->data+nesting->start+5, data->data+nesting->start+1, len);
106                 data->data[nesting->start+1] = (len>>24) & 0xFF;
107                 data->data[nesting->start+2] = (len>>16) & 0xFF;
108                 data->data[nesting->start+3] = (len>>8) & 0xFF;
109                 data->data[nesting->start+4] = len&0xff;
110         } else if (len > 0xFFFF) {
111                 data->data[nesting->start] = 0x83;
112                 if (!asn1_write_uint8(data, 0)) return false;
113                 if (!asn1_write_uint8(data, 0)) return false;
114                 if (!asn1_write_uint8(data, 0)) return false;
115                 memmove(data->data+nesting->start+4, data->data+nesting->start+1, len);
116                 data->data[nesting->start+1] = (len>>16) & 0xFF;
117                 data->data[nesting->start+2] = (len>>8) & 0xFF;
118                 data->data[nesting->start+3] = len&0xff;
119         } else if (len > 255) {
120                 data->data[nesting->start] = 0x82;
121                 if (!asn1_write_uint8(data, 0)) return false;
122                 if (!asn1_write_uint8(data, 0)) return false;
123                 memmove(data->data+nesting->start+3, data->data+nesting->start+1, len);
124                 data->data[nesting->start+1] = len>>8;
125                 data->data[nesting->start+2] = len&0xff;
126         } else if (len > 127) {
127                 data->data[nesting->start] = 0x81;
128                 if (!asn1_write_uint8(data, 0)) return false;
129                 memmove(data->data+nesting->start+2, data->data+nesting->start+1, len);
130                 data->data[nesting->start+1] = len;
131         } else {
132                 data->data[nesting->start] = len;
133         }
134
135         data->nesting = nesting->next;
136         talloc_free(nesting);
137         return true;
138 }
139
140 /* "i" is the one's complement representation, as is the normal result of an
141  * implicit signed->unsigned conversion */
142
143 static bool push_int_bigendian(struct asn1_data *data, unsigned int i, bool negative)
144 {
145         uint8_t lowest = i & 0xFF;
146
147         i = i >> 8;
148         if (i != 0)
149                 if (!push_int_bigendian(data, i, negative))
150                         return false;
151
152         if (data->nesting->start+1 == data->ofs) {
153
154                 /* We did not write anything yet, looking at the highest
155                  * valued byte */
156
157                 if (negative) {
158                         /* Don't write leading 0xff's */
159                         if (lowest == 0xFF)
160                                 return true;
161
162                         if ((lowest & 0x80) == 0) {
163                                 /* The only exception for a leading 0xff is if
164                                  * the highest bit is 0, which would indicate
165                                  * a positive value */
166                                 if (!asn1_write_uint8(data, 0xff))
167                                         return false;
168                         }
169                 } else {
170                         if (lowest & 0x80) {
171                                 /* The highest bit of a positive integer is 1,
172                                  * this would indicate a negative number. Push
173                                  * a 0 to indicate a positive one */
174                                 if (!asn1_write_uint8(data, 0))
175                                         return false;
176                         }
177                 }
178         }
179
180         return asn1_write_uint8(data, lowest);
181 }
182
183 /* write an Integer without the tag framing. Needed for example for the LDAP
184  * Abandon Operation */
185
186 bool asn1_write_implicit_Integer(struct asn1_data *data, int i)
187 {
188         if (i == -1) {
189                 /* -1 is special as it consists of all-0xff bytes. In
190                     push_int_bigendian this is the only case that is not
191                     properly handled, as all 0xff bytes would be handled as
192                     leading ones to be ignored. */
193                 return asn1_write_uint8(data, 0xff);
194         } else {
195                 return push_int_bigendian(data, i, i<0);
196         }
197 }
198
199
200 /* write an integer */
201 bool asn1_write_Integer(struct asn1_data *data, int i)
202 {
203         if (!asn1_push_tag(data, ASN1_INTEGER)) return false;
204         if (!asn1_write_implicit_Integer(data, i)) return false;
205         return asn1_pop_tag(data);
206 }
207
208 /* write a BIT STRING */
209 bool asn1_write_BitString(struct asn1_data *data, const void *p, size_t length, uint8_t padding)
210 {
211         if (!asn1_push_tag(data, ASN1_BIT_STRING)) return false;
212         if (!asn1_write_uint8(data, padding)) return false;
213         if (!asn1_write(data, p, length)) return false;
214         return asn1_pop_tag(data);
215 }
216
217 bool ber_write_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char *OID)
218 {
219         uint_t v, v2;
220         const char *p = (const char *)OID;
221         char *newp;
222         int i;
223
224         v = strtoul(p, &newp, 10);
225         if (newp[0] != '.') return false;
226         p = newp + 1;
227
228         v2 = strtoul(p, &newp, 10);
229         if (newp[0] != '.') return false;
230         p = newp + 1;
231
232         /*the ber representation can't use more space then the string one */
233         *blob = data_blob_talloc(mem_ctx, NULL, strlen(OID));
234         if (!blob->data) return false;
235
236         blob->data[0] = 40*v + v2;
237
238         i = 1;
239         while (*p) {
240                 v = strtoul(p, &newp, 10);
241                 if (newp[0] == '.') {
242                         p = newp + 1;
243                 } else if (newp[0] == '\0') {
244                         p = newp;
245                 } else {
246                         data_blob_free(blob);
247                         return false;
248                 }
249                 if (v >= (1<<28)) blob->data[i++] = (0x80 | ((v>>28)&0x7f));
250                 if (v >= (1<<21)) blob->data[i++] = (0x80 | ((v>>21)&0x7f));
251                 if (v >= (1<<14)) blob->data[i++] = (0x80 | ((v>>14)&0x7f));
252                 if (v >= (1<<7)) blob->data[i++] = (0x80 | ((v>>7)&0x7f));
253                 blob->data[i++] = (v&0x7f);
254         }
255
256         blob->length = i;
257
258         return true;
259 }
260
261 /**
262  * Serialize partial OID string.
263  * Partial OIDs are in the form:
264  *   1:2.5.6:0x81
265  *   1:2.5.6:0x8182
266  */
267 bool ber_write_partial_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char *partial_oid)
268 {
269         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
270         char *oid = talloc_strdup(tmp_ctx, partial_oid);
271         char *p;
272
273         /* truncate partial part so ber_write_OID_String() works */
274         p = strchr(oid, ':');
275         if (p) {
276                 *p = '\0';
277                 p++;
278         }
279
280         if (!ber_write_OID_String(mem_ctx, blob, oid)) {
281                 talloc_free(tmp_ctx);
282                 return false;
283         }
284
285         /* Add partially endcoded subidentifier */
286         if (p) {
287                 DATA_BLOB tmp_blob = strhex_to_data_blob(tmp_ctx, p);
288                 data_blob_append(mem_ctx, blob, tmp_blob.data, tmp_blob.length);
289         }
290
291         talloc_free(tmp_ctx);
292
293         return true;
294 }
295
296 /* write an object ID to a ASN1 buffer */
297 bool asn1_write_OID(struct asn1_data *data, const char *OID)
298 {
299         DATA_BLOB blob;
300
301         if (!asn1_push_tag(data, ASN1_OID)) return false;
302
303         if (!ber_write_OID_String(NULL, &blob, OID)) {
304                 data->has_error = true;
305                 return false;
306         }
307
308         if (!asn1_write(data, blob.data, blob.length)) {
309                 data_blob_free(&blob);
310                 data->has_error = true;
311                 return false;
312         }
313         data_blob_free(&blob);
314         return asn1_pop_tag(data);
315 }
316
317 /* write an octet string */
318 bool asn1_write_OctetString(struct asn1_data *data, const void *p, size_t length)
319 {
320         asn1_push_tag(data, ASN1_OCTET_STRING);
321         asn1_write(data, p, length);
322         asn1_pop_tag(data);
323         return !data->has_error;
324 }
325
326 /* write a LDAP string */
327 bool asn1_write_LDAPString(struct asn1_data *data, const char *s)
328 {
329         asn1_write(data, s, strlen(s));
330         return !data->has_error;
331 }
332
333 /* write a LDAP string from a DATA_BLOB */
334 bool asn1_write_DATA_BLOB_LDAPString(struct asn1_data *data, const DATA_BLOB *s)
335 {
336         asn1_write(data, s->data, s->length);
337         return !data->has_error;
338 }
339
340 /* write a general string */
341 bool asn1_write_GeneralString(struct asn1_data *data, const char *s)
342 {
343         asn1_push_tag(data, ASN1_GENERAL_STRING);
344         asn1_write_LDAPString(data, s);
345         asn1_pop_tag(data);
346         return !data->has_error;
347 }
348
349 bool asn1_write_ContextSimple(struct asn1_data *data, uint8_t num, DATA_BLOB *blob)
350 {
351         asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(num));
352         asn1_write(data, blob->data, blob->length);
353         asn1_pop_tag(data);
354         return !data->has_error;
355 }
356
357 /* write a BOOLEAN */
358 bool asn1_write_BOOLEAN(struct asn1_data *data, bool v)
359 {
360         asn1_push_tag(data, ASN1_BOOLEAN);
361         asn1_write_uint8(data, v ? 0xFF : 0);
362         asn1_pop_tag(data);
363         return !data->has_error;
364 }
365
366 bool asn1_read_BOOLEAN(struct asn1_data *data, bool *v)
367 {
368         uint8_t tmp = 0;
369         asn1_start_tag(data, ASN1_BOOLEAN);
370         asn1_read_uint8(data, &tmp);
371         if (tmp == 0xFF) {
372                 *v = true;
373         } else {
374                 *v = false;
375         }
376         asn1_end_tag(data);
377         return !data->has_error;
378 }
379
380 /* write a BOOLEAN in a simple context */
381 bool asn1_write_BOOLEAN_context(struct asn1_data *data, bool v, int context)
382 {
383         asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(context));
384         asn1_write_uint8(data, v ? 0xFF : 0);
385         asn1_pop_tag(data);
386         return !data->has_error;
387 }
388
389 bool asn1_read_BOOLEAN_context(struct asn1_data *data, bool *v, int context)
390 {
391         uint8_t tmp = 0;
392         asn1_start_tag(data, ASN1_CONTEXT_SIMPLE(context));
393         asn1_read_uint8(data, &tmp);
394         if (tmp == 0xFF) {
395                 *v = true;
396         } else {
397                 *v = false;
398         }
399         asn1_end_tag(data);
400         return !data->has_error;
401 }
402
403 /* check a BOOLEAN */
404 bool asn1_check_BOOLEAN(struct asn1_data *data, bool v)
405 {
406         uint8_t b = 0;
407
408         asn1_read_uint8(data, &b);
409         if (b != ASN1_BOOLEAN) {
410                 data->has_error = true;
411                 return false;
412         }
413         asn1_read_uint8(data, &b);
414         if (b != v) {
415                 data->has_error = true;
416                 return false;
417         }
418         return !data->has_error;
419 }
420
421
422 /* load a struct asn1_data structure with a lump of data, ready to be parsed */
423 bool asn1_load(struct asn1_data *data, DATA_BLOB blob)
424 {
425         ZERO_STRUCTP(data);
426         data->data = (uint8_t *)talloc_memdup(data, blob.data, blob.length);
427         if (!data->data) {
428                 data->has_error = true;
429                 return false;
430         }
431         data->length = blob.length;
432         return true;
433 }
434
435 /* Peek into an ASN1 buffer, not advancing the pointer */
436 bool asn1_peek(struct asn1_data *data, void *p, int len)
437 {
438         if (data->has_error)
439                 return false;
440
441         if (len < 0 || data->ofs + len < data->ofs || data->ofs + len < len)
442                 return false;
443
444         if (data->ofs + len > data->length) {
445                 /* we need to mark the buffer as consumed, so the caller knows
446                    this was an out of data error, and not a decode error */
447                 data->ofs = data->length;
448                 return false;
449         }
450
451         memcpy(p, data->data + data->ofs, len);
452         return true;
453 }
454
455 /* read from a ASN1 buffer, advancing the buffer pointer */
456 bool asn1_read(struct asn1_data *data, void *p, int len)
457 {
458         if (!asn1_peek(data, p, len)) {
459                 data->has_error = true;
460                 return false;
461         }
462
463         data->ofs += len;
464         return true;
465 }
466
467 /* read a uint8_t from a ASN1 buffer */
468 bool asn1_read_uint8(struct asn1_data *data, uint8_t *v)
469 {
470         return asn1_read(data, v, 1);
471 }
472
473 bool asn1_peek_uint8(struct asn1_data *data, uint8_t *v)
474 {
475         return asn1_peek(data, v, 1);
476 }
477
478 bool asn1_peek_tag(struct asn1_data *data, uint8_t tag)
479 {
480         uint8_t b;
481
482         if (asn1_tag_remaining(data) <= 0) {
483                 return false;
484         }
485
486         if (!asn1_peek_uint8(data, &b))
487                 return false;
488
489         return (b == tag);
490 }
491
492 /* start reading a nested asn1 structure */
493 bool asn1_start_tag(struct asn1_data *data, uint8_t tag)
494 {
495         uint8_t b;
496         struct nesting *nesting;
497         
498         if (!asn1_read_uint8(data, &b))
499                 return false;
500
501         if (b != tag) {
502                 data->has_error = true;
503                 return false;
504         }
505         nesting = talloc(data, struct nesting);
506         if (!nesting) {
507                 data->has_error = true;
508                 return false;
509         }
510
511         if (!asn1_read_uint8(data, &b)) {
512                 return false;
513         }
514
515         if (b & 0x80) {
516                 int n = b & 0x7f;
517                 if (!asn1_read_uint8(data, &b))
518                         return false;
519                 nesting->taglen = b;
520                 while (n > 1) {
521                         if (!asn1_read_uint8(data, &b)) 
522                                 return false;
523                         nesting->taglen = (nesting->taglen << 8) | b;
524                         n--;
525                 }
526         } else {
527                 nesting->taglen = b;
528         }
529         nesting->start = data->ofs;
530         nesting->next = data->nesting;
531         data->nesting = nesting;
532         if (asn1_tag_remaining(data) == -1) {
533                 return false;
534         }
535         return !data->has_error;
536 }
537
538 /* stop reading a tag */
539 bool asn1_end_tag(struct asn1_data *data)
540 {
541         struct nesting *nesting;
542
543         /* make sure we read it all */
544         if (asn1_tag_remaining(data) != 0) {
545                 data->has_error = true;
546                 return false;
547         }
548
549         nesting = data->nesting;
550
551         if (!nesting) {
552                 data->has_error = true;
553                 return false;
554         }
555
556         data->nesting = nesting->next;
557         talloc_free(nesting);
558         return true;
559 }
560
561 /* work out how many bytes are left in this nested tag */
562 int asn1_tag_remaining(struct asn1_data *data)
563 {
564         int remaining;
565         if (data->has_error) {
566                 return -1;
567         }
568
569         if (!data->nesting) {
570                 data->has_error = true;
571                 return -1;
572         }
573         remaining = data->nesting->taglen - (data->ofs - data->nesting->start);
574         if (remaining > (data->length - data->ofs)) {
575                 data->has_error = true;
576                 return -1;
577         }
578         return remaining;
579 }
580
581 /**
582  * Internal implementation for reading binary OIDs
583  * Reading is done as far in the buffer as valid OID
584  * till buffer ends or not valid sub-identifier is found.
585  */
586 static bool _ber_read_OID_String_impl(TALLOC_CTX *mem_ctx, DATA_BLOB blob,
587                                         const char **OID, size_t *bytes_eaten)
588 {
589         int i;
590         uint8_t *b;
591         uint_t v;
592         char *tmp_oid = NULL;
593
594         if (blob.length < 2) return false;
595
596         b = blob.data;
597
598         tmp_oid = talloc_asprintf(mem_ctx, "%u",  b[0]/40);
599         if (!tmp_oid) goto nomem;
600         tmp_oid = talloc_asprintf_append_buffer(tmp_oid, ".%u",  b[0]%40);
601         if (!tmp_oid) goto nomem;
602
603         for(i = 1, v = 0; i < blob.length; i++) {
604                 v = (v<<7) | (b[i]&0x7f);
605                 if ( ! (b[i] & 0x80)) {
606                         tmp_oid = talloc_asprintf_append_buffer(tmp_oid, ".%u",  v);
607                         v = 0;
608                         if (bytes_eaten)
609                                 *bytes_eaten = i+1;
610                 }
611                 if (!tmp_oid) goto nomem;
612         }
613
614         *OID = tmp_oid;
615         return true;
616
617 nomem:
618         return false;
619 }
620
621 /* read an object ID from a data blob */
622 bool ber_read_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB blob, const char **OID)
623 {
624         size_t bytes_eaten;
625
626         if (!_ber_read_OID_String_impl(mem_ctx, blob, OID, &bytes_eaten))
627                 return false;
628
629         return (bytes_eaten == blob.length);
630 }
631
632 /**
633  * Deserialize partial OID string.
634  * Partial OIDs are in the form:
635  *   1:2.5.6:0x81
636  *   1:2.5.6:0x8182
637  */
638 bool ber_read_partial_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB blob, const char **partial_oid)
639 {
640         size_t bytes_left;
641         size_t bytes_eaten;
642         char *identifier = NULL;
643         char *tmp_oid = NULL;
644
645         if (!_ber_read_OID_String_impl(mem_ctx, blob, (const char **)&tmp_oid, &bytes_eaten))
646                 return false;
647
648         if (bytes_eaten < blob.length) {
649                 bytes_left = blob.length - bytes_eaten;
650                 identifier = hex_encode_talloc(mem_ctx, &blob.data[bytes_eaten], bytes_left);
651                 if (!identifier)        goto nomem;
652
653                 *partial_oid = talloc_asprintf_append_buffer(tmp_oid, ":0x%s", identifier);
654                 if (!*partial_oid)      goto nomem;
655                 TALLOC_FREE(identifier);
656         } else {
657                 *partial_oid = tmp_oid;
658         }
659
660         return true;
661
662 nomem:
663         TALLOC_FREE(identifier);
664         TALLOC_FREE(tmp_oid);
665         return false;
666 }
667
668 /* read an object ID from a ASN1 buffer */
669 bool asn1_read_OID(struct asn1_data *data, TALLOC_CTX *mem_ctx, const char **OID)
670 {
671         DATA_BLOB blob;
672         int len;
673
674         if (!asn1_start_tag(data, ASN1_OID)) return false;
675
676         len = asn1_tag_remaining(data);
677         if (len < 0) {
678                 data->has_error = true;
679                 return false;
680         }
681
682         blob = data_blob(NULL, len);
683         if (!blob.data) {
684                 data->has_error = true;
685                 return false;
686         }
687
688         asn1_read(data, blob.data, len);
689         asn1_end_tag(data);
690         if (data->has_error) {
691                 data_blob_free(&blob);
692                 return false;
693         }
694
695         if (!ber_read_OID_String(mem_ctx, blob, OID)) {
696                 data->has_error = true;
697                 data_blob_free(&blob);
698                 return false;
699         }
700
701         data_blob_free(&blob);
702         return true;
703 }
704
705 /* check that the next object ID is correct */
706 bool asn1_check_OID(struct asn1_data *data, const char *OID)
707 {
708         const char *id;
709
710         if (!asn1_read_OID(data, data, &id)) return false;
711
712         if (strcmp(id, OID) != 0) {
713                 talloc_free(discard_const(id));
714                 data->has_error = true;
715                 return false;
716         }
717         talloc_free(discard_const(id));
718         return true;
719 }
720
721 /* read a LDAPString from a ASN1 buffer */
722 bool asn1_read_LDAPString(struct asn1_data *data, TALLOC_CTX *mem_ctx, char **s)
723 {
724         int len;
725         len = asn1_tag_remaining(data);
726         if (len < 0) {
727                 data->has_error = true;
728                 return false;
729         }
730         *s = talloc_array(mem_ctx, char, len+1);
731         if (! *s) {
732                 data->has_error = true;
733                 return false;
734         }
735         asn1_read(data, *s, len);
736         (*s)[len] = 0;
737         return !data->has_error;
738 }
739
740
741 /* read a GeneralString from a ASN1 buffer */
742 bool asn1_read_GeneralString(struct asn1_data *data, TALLOC_CTX *mem_ctx, char **s)
743 {
744         if (!asn1_start_tag(data, ASN1_GENERAL_STRING)) return false;
745         if (!asn1_read_LDAPString(data, mem_ctx, s)) return false;
746         return asn1_end_tag(data);
747 }
748
749
750 /* read a octet string blob */
751 bool asn1_read_OctetString(struct asn1_data *data, TALLOC_CTX *mem_ctx, DATA_BLOB *blob)
752 {
753         int len;
754         ZERO_STRUCTP(blob);
755         if (!asn1_start_tag(data, ASN1_OCTET_STRING)) return false;
756         len = asn1_tag_remaining(data);
757         if (len < 0) {
758                 data->has_error = true;
759                 return false;
760         }
761         *blob = data_blob_talloc(mem_ctx, NULL, len+1);
762         if (!blob->data) {
763                 data->has_error = true;
764                 return false;
765         }
766         asn1_read(data, blob->data, len);
767         asn1_end_tag(data);
768         blob->length--;
769         blob->data[len] = 0;
770         
771         if (data->has_error) {
772                 data_blob_free(blob);
773                 *blob = data_blob_null;
774                 return false;
775         }
776         return true;
777 }
778
779 bool asn1_read_ContextSimple(struct asn1_data *data, uint8_t num, DATA_BLOB *blob)
780 {
781         int len;
782         ZERO_STRUCTP(blob);
783         if (!asn1_start_tag(data, ASN1_CONTEXT_SIMPLE(num))) return false;
784         len = asn1_tag_remaining(data);
785         if (len < 0) {
786                 data->has_error = true;
787                 return false;
788         }
789         *blob = data_blob(NULL, len);
790         if ((len != 0) && (!blob->data)) {
791                 data->has_error = true;
792                 return false;
793         }
794         asn1_read(data, blob->data, len);
795         asn1_end_tag(data);
796         return !data->has_error;
797 }
798
799 /* read an integer without tag*/
800 bool asn1_read_implicit_Integer(struct asn1_data *data, int *i)
801 {
802         uint8_t b;
803         *i = 0;
804
805         while (!data->has_error && asn1_tag_remaining(data)>0) {
806                 if (!asn1_read_uint8(data, &b)) return false;
807                 *i = (*i << 8) + b;
808         }
809         return !data->has_error;        
810         
811 }
812
813 /* read an integer */
814 bool asn1_read_Integer(struct asn1_data *data, int *i)
815 {
816         *i = 0;
817
818         if (!asn1_start_tag(data, ASN1_INTEGER)) return false;
819         if (!asn1_read_implicit_Integer(data, i)) return false;
820         return asn1_end_tag(data);      
821 }
822
823 /* read a BIT STRING */
824 bool asn1_read_BitString(struct asn1_data *data, TALLOC_CTX *mem_ctx, DATA_BLOB *blob, uint8_t *padding)
825 {
826         int len;
827         ZERO_STRUCTP(blob);
828         if (!asn1_start_tag(data, ASN1_BIT_STRING)) return false;
829         len = asn1_tag_remaining(data);
830         if (len < 0) {
831                 data->has_error = true;
832                 return false;
833         }
834         if (!asn1_read_uint8(data, padding)) return false;
835
836         *blob = data_blob_talloc(mem_ctx, NULL, len);
837         if (!blob->data) {
838                 data->has_error = true;
839                 return false;
840         }
841         if (asn1_read(data, blob->data, len - 1)) {
842                 blob->length--;
843                 blob->data[len] = 0;
844                 asn1_end_tag(data);
845         }
846
847         if (data->has_error) {
848                 data_blob_free(blob);
849                 *blob = data_blob_null;
850                 *padding = 0;
851                 return false;
852         }
853         return true;
854 }
855
856 /* read an integer */
857 bool asn1_read_enumerated(struct asn1_data *data, int *v)
858 {
859         *v = 0;
860         
861         if (!asn1_start_tag(data, ASN1_ENUMERATED)) return false;
862         while (!data->has_error && asn1_tag_remaining(data)>0) {
863                 uint8_t b;
864                 asn1_read_uint8(data, &b);
865                 *v = (*v << 8) + b;
866         }
867         return asn1_end_tag(data);      
868 }
869
870 /* check a enumerated value is correct */
871 bool asn1_check_enumerated(struct asn1_data *data, int v)
872 {
873         uint8_t b;
874         if (!asn1_start_tag(data, ASN1_ENUMERATED)) return false;
875         asn1_read_uint8(data, &b);
876         asn1_end_tag(data);
877
878         if (v != b)
879                 data->has_error = false;
880
881         return !data->has_error;
882 }
883
884 /* write an enumerated value to the stream */
885 bool asn1_write_enumerated(struct asn1_data *data, uint8_t v)
886 {
887         if (!asn1_push_tag(data, ASN1_ENUMERATED)) return false;
888         asn1_write_uint8(data, v);
889         asn1_pop_tag(data);
890         return !data->has_error;
891 }
892
893 /*
894   Get us the data just written without copying
895 */
896 bool asn1_blob(const struct asn1_data *asn1, DATA_BLOB *blob)
897 {
898         if (asn1->has_error) {
899                 return false;
900         }
901         if (asn1->nesting != NULL) {
902                 return false;
903         }
904         blob->data = asn1->data;
905         blob->length = asn1->length;
906         return true;
907 }
908
909 /*
910   Fill in an asn1 struct without making a copy
911 */
912 void asn1_load_nocopy(struct asn1_data *data, uint8_t *buf, size_t len)
913 {
914         ZERO_STRUCTP(data);
915         data->data = buf;
916         data->length = len;
917 }
918
919 /*
920   check if a ASN.1 blob is a full tag
921 */
922 NTSTATUS asn1_full_tag(DATA_BLOB blob, uint8_t tag, size_t *packet_size)
923 {
924         struct asn1_data *asn1 = asn1_init(NULL);
925         int size;
926
927         NT_STATUS_HAVE_NO_MEMORY(asn1);
928
929         asn1->data = blob.data;
930         asn1->length = blob.length;
931         asn1_start_tag(asn1, tag);
932         if (asn1->has_error) {
933                 talloc_free(asn1);
934                 return STATUS_MORE_ENTRIES;
935         }
936         size = asn1_tag_remaining(asn1) + asn1->ofs;
937
938         talloc_free(asn1);
939
940         if (size > blob.length) {
941                 return STATUS_MORE_ENTRIES;
942         }               
943
944         *packet_size = size;
945         return NT_STATUS_OK;
946 }