]> git.sur5r.net Git - openldap/blob - libraries/liblber/decode.c
partially addresses an issue with ITS#5931
[openldap] / libraries / liblber / decode.c
1 /* decode.c - ber input decoding routines */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 1998-2009 The OpenLDAP Foundation.
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted only as authorized by the OpenLDAP
10  * Public License.
11  *
12  * A copy of this license is available in the file LICENSE in the
13  * top-level directory of the distribution or, alternatively, at
14  * <http://www.OpenLDAP.org/license.html>.
15  */
16 /* Portions Copyright (c) 1990 Regents of the University of Michigan.
17  * All rights reserved.
18  *
19  * Redistribution and use in source and binary forms are permitted
20  * provided that this notice is preserved and that due credit is given
21  * to the University of Michigan at Ann Arbor. The name of the University
22  * may not be used to endorse or promote products derived from this
23  * software without specific prior written permission. This software
24  * is provided ``as is'' without express or implied warranty.
25  */
26 /* ACKNOWLEDGEMENTS:
27  * This work was originally developed by the University of Michigan
28  * (as part of U-MICH LDAP).
29  */
30
31 #include "portable.h"
32
33 #include <stdio.h>
34
35 #include <ac/stdlib.h>
36
37 #include <ac/stdarg.h>
38 #include <ac/string.h>
39 #include <ac/socket.h>
40
41 #include "lber-int.h"
42
43 static ber_len_t ber_getnint LDAP_P((
44         BerElement *ber,
45         ber_int_t *num,
46         ber_len_t len ));
47
48 /* out->bv_len should be the buffer size on input */
49 int
50 ber_decode_oid( BerValue *in, BerValue *out )
51 {
52         const unsigned char *der;
53         unsigned long val;
54         unsigned val1;
55         ber_len_t i;
56         char *ptr;
57
58         assert( in != NULL );
59         assert( out != NULL );
60
61         /* need 4 chars/inbyte + \0 for input={7f 7f 7f...} */
62         if ( !out->bv_val || (out->bv_len+3)/4 <= in->bv_len )
63                 return -1;
64
65         ptr = NULL;
66         der = (unsigned char *) in->bv_val;
67         val = 0;
68         for ( i=0; i < in->bv_len; i++ ) {
69                 val |= der[i] & 0x7f;
70                 if ( !( der[i] & 0x80 )) {
71                         if ( ptr == NULL ) {
72                                 /* Initial "x.y": val=x*40+y, x<=2, y<40 if x<2 */
73                                 ptr = out->bv_val;
74                                 val1 = (val < 80 ? val/40 : 2);
75                                 val -= val1*40;
76                                 ptr += sprintf( ptr, "%u", val1 );
77                         }
78                         ptr += sprintf( ptr, ".%lu", val );
79                         val = 0;
80                 } else if ( val - 1UL < LBER_OID_COMPONENT_MAX >> 7 ) {
81                         val <<= 7;
82                 } else {
83                         /* val would overflow, or is 0 from invalid initial 0x80 octet */
84                         return -1;
85                 }
86         }
87         if ( ptr == NULL || val != 0 )
88                 return -1;
89
90         out->bv_len = ptr - out->bv_val;
91         return 0;
92 }
93
94 /* return the tag - LBER_DEFAULT returned means trouble */
95 ber_tag_t
96 ber_get_tag( BerElement *ber )
97 {
98         unsigned char   xbyte;
99         ber_tag_t       tag;
100         unsigned int    i;
101
102         assert( ber != NULL );
103         assert( LBER_VALID( ber ) );
104
105         if ( ber_pvt_ber_remaining( ber ) < 1 ) {
106                 return LBER_DEFAULT;
107         }
108
109         if ( ber->ber_ptr == ber->ber_buf ) {
110                 tag = *(unsigned char *)ber->ber_ptr;
111         } else {
112                 tag = ber->ber_tag;
113         }
114         ber->ber_ptr++;
115
116         if ( (tag & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK ) {
117                 return tag;
118         }
119
120         for ( i = 1; i < sizeof(ber_tag_t); i++ ) {
121                 if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 ) {
122                         return LBER_DEFAULT;
123                 }
124
125                 tag <<= 8;
126                 tag |= 0x00ffUL & (ber_tag_t) xbyte;
127
128                 if ( ! (xbyte & LBER_MORE_TAG_MASK) ) {
129                         break;
130                 }
131         }
132
133         /* tag too big! */
134         if ( i == sizeof(ber_tag_t) ) {
135                 return LBER_DEFAULT;
136         }
137
138         return tag;
139 }
140
141 ber_tag_t
142 ber_skip_tag( BerElement *ber, ber_len_t *len )
143 {
144         ber_tag_t       tag;
145         unsigned char   lc;
146         char            *save;
147
148         assert( ber != NULL );
149         assert( len != NULL );
150         assert( LBER_VALID( ber ) );
151
152         save = ber->ber_ptr;
153
154         /*
155          * Any ber element looks like this: tag length contents.
156          * Assuming everything's ok, we return the tag byte (we
157          * can assume a single byte), and return the length in len.
158          *
159          * Assumptions:
160          *      1) definite lengths
161          *      2) primitive encodings used whenever possible
162          */
163
164         *len = 0;
165
166         /*
167          * First, we read the tag.
168          */
169
170         if ( (tag = ber_get_tag( ber )) == LBER_DEFAULT ) {
171                 return LBER_DEFAULT;
172         }
173
174         /*
175          * Next, read the length.  The first byte contains the length of
176          * the length.  If bit 8 is set, the length is the long form,
177          * otherwise it's the short form.  We don't allow a length that's
178          * greater than what we can hold in a ber_len_t.
179          */
180
181         if ( ber_read( ber, (char *) &lc, 1 ) != 1 ) {
182                 return LBER_DEFAULT;
183         }
184
185         if ( lc & 0x80U ) {
186                 ber_len_t       i, noctets;
187                 unsigned char netlen[sizeof(ber_len_t)];
188
189                 noctets = (lc & 0x7fU);
190
191                 if ( noctets > sizeof(ber_len_t) ) {
192                         return LBER_DEFAULT;
193                 }
194
195                 if( (unsigned) ber_read( ber, (char *) netlen, noctets ) != noctets ) {
196                         return LBER_DEFAULT;
197                 }
198
199                 for( i = 0; i < noctets; i++ ) {
200                         *len <<= 8;
201                         *len |= netlen[i];
202                 }
203
204         } else {
205                 *len = lc;
206         }
207
208         /* BER element should have enough data left */
209         if( *len > (ber_len_t) (ber_pvt_ber_remaining( ber ) + ber->ber_ptr - save) ) {
210                 return LBER_DEFAULT;
211         }
212         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
213
214         return tag;
215 }
216
217 ber_tag_t
218 ber_peek_tag(
219         BerElement *ber,
220         ber_len_t *len )
221 {
222         /*
223          * This implementation assumes ber_skip_tag() only
224          * modifies ber_ptr field of the BerElement.
225          */
226
227         char *save;
228         ber_tag_t       tag, old;
229
230         old = ber->ber_tag;
231         save = ber->ber_ptr;
232         tag = ber_skip_tag( ber, len );
233         ber->ber_ptr = save;
234         ber->ber_tag = old;
235
236         return tag;
237 }
238
239 static ber_len_t
240 ber_getnint(
241         BerElement *ber,
242         ber_int_t *num,
243         ber_len_t len )
244 {
245         unsigned char buf[sizeof(ber_int_t)];
246
247         assert( ber != NULL );
248         assert( num != NULL );
249         assert( LBER_VALID( ber ) );
250
251         /*
252          * The tag and length have already been stripped off.  We should
253          * be sitting right before len bytes of 2's complement integer,
254          * ready to be read straight into an int.  We may have to sign
255          * extend after we read it in.
256          */
257
258         if ( len > sizeof(ber_int_t) ) {
259                 return -1;
260         }
261
262         /* read into the low-order bytes of our buffer */
263         if ( (ber_len_t) ber_read( ber, (char *) buf, len ) != len ) {
264                 return -1;
265         }
266
267         if( len ) {
268                 /* sign extend if necessary */
269                 ber_len_t i;
270                 ber_int_t netnum = 0x80 & buf[0] ? -1 : 0;
271
272                 /* shift in the bytes */
273                 for( i=0 ; i<len; i++ ) {
274                         netnum = (netnum << 8 ) | buf[i];
275                 }
276
277                 *num = netnum;
278
279         } else {
280                 *num = 0;
281         }
282         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
283
284         return len;
285 }
286
287 ber_tag_t
288 ber_get_int(
289         BerElement *ber,
290         ber_int_t *num )
291 {
292         ber_tag_t       tag;
293         ber_len_t       len;
294
295         assert( ber != NULL );
296         assert( LBER_VALID( ber ) );
297
298         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) {
299                 return LBER_DEFAULT;
300         }
301
302         if ( ber_getnint( ber, num, len ) != len ) {
303                 return LBER_DEFAULT;
304         }
305         
306         return tag;
307 }
308
309 ber_tag_t
310 ber_get_enum(
311         BerElement *ber,
312         ber_int_t *num )
313 {
314         return ber_get_int( ber, num );
315 }
316
317 ber_tag_t
318 ber_get_stringb(
319         BerElement *ber,
320         char *buf,
321         ber_len_t *len )
322 {
323         ber_len_t       datalen;
324         ber_tag_t       tag;
325
326         assert( ber != NULL );
327         assert( LBER_VALID( ber ) );
328
329         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) {
330                 return LBER_DEFAULT;
331         }
332
333         /* must fit within allocated space with termination */
334         if ( datalen >= *len ) {
335                 return LBER_DEFAULT;
336         }
337
338         if ( (ber_len_t) ber_read( ber, buf, datalen ) != datalen ) {
339                 return LBER_DEFAULT;
340         }
341         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
342
343         buf[datalen] = '\0';
344
345         *len = datalen;
346         return tag;
347 }
348
349 /* Definitions for get_string vector
350  *
351  * ChArray, BvArray, and BvVec are self-explanatory.
352  * BvOff is a struct berval embedded in an array of larger structures
353  * of siz bytes at off bytes from the beginning of the struct.
354  */
355 enum bgbvc { ChArray, BvArray, BvVec, BvOff };
356
357 /* Use this single cookie for state, to keep actual
358  * stack use to the absolute minimum.
359  */
360 typedef struct bgbvr {
361         enum bgbvc choice;
362         BerElement *ber;
363         int alloc;
364         ber_len_t siz;
365         ber_len_t off;
366         union {
367                 char ***c;
368                 BerVarray *ba;
369                 struct berval ***bv;
370         } res;
371 } bgbvr;
372
373 static ber_tag_t
374 ber_get_stringbvl( bgbvr *b, ber_len_t *rlen )
375 {
376         int i = 0, n;
377         ber_tag_t tag;
378         ber_len_t len;
379         char *last, *orig;
380         struct berval bv, *bvp = NULL;
381
382         /* For rewinding, just like ber_peek_tag() */
383         orig = b->ber->ber_ptr;
384         tag = b->ber->ber_tag;
385
386         if ( ber_first_element( b->ber, &len, &last ) != LBER_DEFAULT ) {
387                 for ( ; b->ber->ber_ptr < last; i++ ) {
388                         if (ber_skip_tag( b->ber, &len ) == LBER_DEFAULT) break;
389                         b->ber->ber_ptr += len;
390                         b->ber->ber_tag = *(unsigned char *)b->ber->ber_ptr;
391                 }
392         }
393
394         if ( rlen ) *rlen = i;
395
396         if ( i == 0 ) {
397                 *b->res.c = NULL;
398                 return 0;
399         }
400
401         n = i;
402
403         /* Allocate the result vector */
404         switch (b->choice) {
405         case ChArray:
406                 *b->res.c = ber_memalloc_x( (n+1)*sizeof( char * ),
407                         b->ber->ber_memctx);
408                 if ( *b->res.c == NULL ) return LBER_DEFAULT;
409                 (*b->res.c)[n] = NULL;
410                 break;
411         case BvArray:
412                 *b->res.ba = ber_memalloc_x( (n+1)*sizeof( struct berval ),
413                         b->ber->ber_memctx);
414                 if ( *b->res.ba == NULL ) return LBER_DEFAULT;
415                 (*b->res.ba)[n].bv_val = NULL;
416                 break;
417         case BvVec:
418                 *b->res.bv = ber_memalloc_x( (n+1)*sizeof( struct berval *),
419                         b->ber->ber_memctx);
420                 if ( *b->res.bv == NULL ) return LBER_DEFAULT;
421                 (*b->res.bv)[n] = NULL;
422                 break;
423         case BvOff:
424                 *b->res.ba = ber_memalloc_x( (n+1) * b->siz, b->ber->ber_memctx );
425                 if ( *b->res.ba == NULL ) return LBER_DEFAULT;
426                 ((struct berval *)((char *)(*b->res.ba) + n*b->siz +
427                         b->off))->bv_val = NULL;
428                 break;
429         }
430         b->ber->ber_ptr = orig;
431         b->ber->ber_tag = tag;
432         ber_skip_tag( b->ber, &len );
433         
434         for (n=0; n<i; n++)
435         {
436                 tag = ber_next_element( b->ber, &len, last );
437                 if ( ber_get_stringbv( b->ber, &bv, b->alloc ) == LBER_DEFAULT ) {
438                         goto nomem;
439                 }
440
441                 /* store my result */
442                 switch (b->choice) {
443                 case ChArray:
444                         (*b->res.c)[n] = bv.bv_val;
445                         break;
446                 case BvArray:
447                         (*b->res.ba)[n] = bv;
448                         break;
449                 case BvVec:
450                         bvp = ber_memalloc_x( sizeof( struct berval ), b->ber->ber_memctx);
451                         if ( !bvp ) {
452                                 LBER_FREE(bv.bv_val);
453                                 goto nomem;
454                         }
455                         (*b->res.bv)[n] = bvp;
456                         *bvp = bv;
457                         break;
458                 case BvOff:
459                         *(BerVarray)((char *)(*b->res.ba)+n*b->siz+b->off) = bv;
460                         break;
461                 }
462         }
463         return tag;
464
465 nomem:
466         if (b->alloc || b->choice == BvVec) {
467                 for (--n; n>=0; n--) {
468                         switch(b->choice) {
469                         case ChArray:
470                                 LBER_FREE((*b->res.c)[n]);
471                                 break;
472                         case BvArray:
473                                 LBER_FREE((*b->res.ba)[n].bv_val);
474                                 break;
475                         case BvVec:
476                                 LBER_FREE((*b->res.bv)[n]->bv_val);
477                                 LBER_FREE((*b->res.bv)[n]);
478                                 break;
479                         default:
480                                 break;
481                         }
482                 }
483         }
484         LBER_FREE(*b->res.c);
485         *b->res.c = NULL;
486         return LBER_DEFAULT;
487 }
488
489 ber_tag_t
490 ber_get_stringbv( BerElement *ber, struct berval *bv, int option )
491 {
492         ber_tag_t       tag;
493
494         assert( ber != NULL );
495         assert( bv != NULL );
496
497         assert( LBER_VALID( ber ) );
498
499         if ( (tag = ber_skip_tag( ber, &bv->bv_len )) == LBER_DEFAULT ) {
500                 bv->bv_val = NULL;
501                 return LBER_DEFAULT;
502         }
503
504         if ( (ber_len_t) ber_pvt_ber_remaining( ber ) < bv->bv_len ) {
505                 return LBER_DEFAULT;
506         }
507
508         if ( option & LBER_BV_ALLOC ) {
509                 bv->bv_val = (char *) ber_memalloc_x( bv->bv_len + 1,
510                         ber->ber_memctx );
511                 if ( bv->bv_val == NULL ) {
512                         return LBER_DEFAULT;
513                 }
514
515                 if ( bv->bv_len > 0 && (ber_len_t) ber_read( ber, bv->bv_val,
516                         bv->bv_len ) != bv->bv_len )
517                 {
518                         LBER_FREE( bv->bv_val );
519                         bv->bv_val = NULL;
520                         return LBER_DEFAULT;
521                 }
522         } else {
523                 bv->bv_val = ber->ber_ptr;
524                 ber->ber_ptr += bv->bv_len;
525         }
526         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
527         if ( !( option & LBER_BV_NOTERM ))
528                 bv->bv_val[bv->bv_len] = '\0';
529
530         return tag;
531 }
532
533 ber_tag_t
534 ber_get_stringbv_null( BerElement *ber, struct berval *bv, int option )
535 {
536         ber_tag_t       tag;
537
538         assert( ber != NULL );
539         assert( bv != NULL );
540
541         assert( LBER_VALID( ber ) );
542
543         if ( (tag = ber_skip_tag( ber, &bv->bv_len )) == LBER_DEFAULT ) {
544                 bv->bv_val = NULL;
545                 return LBER_DEFAULT;
546         }
547
548         if ( (ber_len_t) ber_pvt_ber_remaining( ber ) < bv->bv_len ) {
549                 return LBER_DEFAULT;
550         }
551
552         if ( bv->bv_len == 0 ) {
553                 bv->bv_val = NULL;
554                 ber->ber_tag = *(unsigned char *)ber->ber_ptr;
555                 return tag;
556         }
557
558         if ( option & LBER_BV_ALLOC ) {
559                 bv->bv_val = (char *) ber_memalloc_x( bv->bv_len + 1,
560                         ber->ber_memctx );
561                 if ( bv->bv_val == NULL ) {
562                         return LBER_DEFAULT;
563                 }
564
565                 if ( bv->bv_len > 0 && (ber_len_t) ber_read( ber, bv->bv_val,
566                         bv->bv_len ) != bv->bv_len )
567                 {
568                         LBER_FREE( bv->bv_val );
569                         bv->bv_val = NULL;
570                         return LBER_DEFAULT;
571                 }
572         } else {
573                 bv->bv_val = ber->ber_ptr;
574                 ber->ber_ptr += bv->bv_len;
575         }
576         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
577         if ( !( option & LBER_BV_NOTERM ))
578                 bv->bv_val[bv->bv_len] = '\0';
579
580         return tag;
581 }
582
583 ber_tag_t
584 ber_get_stringa( BerElement *ber, char **buf )
585 {
586         BerValue        bv;
587         ber_tag_t       tag;
588
589         assert( buf != NULL );
590
591         tag = ber_get_stringbv( ber, &bv, LBER_BV_ALLOC );
592         *buf = bv.bv_val;
593
594         return tag;
595 }
596
597 ber_tag_t
598 ber_get_stringa_null( BerElement *ber, char **buf )
599 {
600         BerValue        bv;
601         ber_tag_t       tag;
602
603         assert( buf != NULL );
604
605         tag = ber_get_stringbv_null( ber, &bv, LBER_BV_ALLOC );
606         *buf = bv.bv_val;
607
608         return tag;
609 }
610
611 ber_tag_t
612 ber_get_stringal( BerElement *ber, struct berval **bv )
613 {
614         ber_tag_t       tag;
615
616         assert( ber != NULL );
617         assert( bv != NULL );
618
619         *bv = (struct berval *) ber_memalloc_x( sizeof(struct berval),
620                 ber->ber_memctx );
621         if ( *bv == NULL ) {
622                 return LBER_DEFAULT;
623         }
624
625         tag = ber_get_stringbv( ber, *bv, LBER_BV_ALLOC );
626         if ( tag == LBER_DEFAULT ) {
627                 LBER_FREE( *bv );
628                 *bv = NULL;
629         }
630         return tag;
631 }
632
633 ber_tag_t
634 ber_get_bitstringa(
635         BerElement *ber,
636         char **buf,
637         ber_len_t *blen )
638 {
639         ber_len_t       datalen;
640         ber_tag_t       tag;
641         unsigned char   unusedbits;
642
643         assert( ber != NULL );
644         assert( buf != NULL );
645         assert( blen != NULL );
646
647         assert( LBER_VALID( ber ) );
648
649         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) {
650                 *buf = NULL;
651                 return LBER_DEFAULT;
652         }
653         --datalen;
654
655         *buf = (char *) ber_memalloc_x( datalen, ber->ber_memctx );
656         if ( *buf == NULL ) {
657                 return LBER_DEFAULT;
658         }
659
660         if ( ber_read( ber, (char *)&unusedbits, 1 ) != 1 ) {
661                 LBER_FREE( buf );
662                 *buf = NULL;
663                 return LBER_DEFAULT;
664         }
665
666         if ( (ber_len_t) ber_read( ber, *buf, datalen ) != datalen ) {
667                 LBER_FREE( buf );
668                 *buf = NULL;
669                 return LBER_DEFAULT;
670         }
671         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
672
673         *blen = datalen * 8 - unusedbits;
674         return tag;
675 }
676
677 ber_tag_t
678 ber_get_null( BerElement *ber )
679 {
680         ber_len_t       len;
681         ber_tag_t       tag;
682
683         assert( ber != NULL );
684         assert( LBER_VALID( ber ) );
685
686         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) {
687                 return LBER_DEFAULT;
688         }
689
690         if ( len != 0 ) {
691                 return LBER_DEFAULT;
692         }
693         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
694
695         return( tag );
696 }
697
698 ber_tag_t
699 ber_get_boolean(
700         BerElement *ber,
701         ber_int_t *boolval )
702 {
703         ber_int_t       longbool;
704         ber_tag_t       rc;
705
706         assert( ber != NULL );
707         assert( boolval != NULL );
708
709         assert( LBER_VALID( ber ) );
710
711         rc = ber_get_int( ber, &longbool );
712         *boolval = longbool;
713
714         return rc;
715 }
716
717 ber_tag_t
718 ber_first_element(
719         BerElement *ber,
720         ber_len_t *len,
721         char **last )
722 {
723         assert( ber != NULL );
724         assert( len != NULL );
725         assert( last != NULL );
726
727         /* skip the sequence header, use the len to mark where to stop */
728         if ( ber_skip_tag( ber, len ) == LBER_DEFAULT ) {
729                 *last = NULL;
730                 return LBER_DEFAULT;
731         }
732         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
733
734         *last = ber->ber_ptr + *len;
735
736         if ( *last == ber->ber_ptr ) {
737                 return LBER_DEFAULT;
738         }
739
740         return ber_peek_tag( ber, len );
741 }
742
743 ber_tag_t
744 ber_next_element(
745         BerElement *ber,
746         ber_len_t *len,
747         LDAP_CONST char *last )
748 {
749         assert( ber != NULL );
750         assert( len != NULL );
751         assert( last != NULL );
752
753         assert( LBER_VALID( ber ) );
754
755         if ( ber->ber_ptr >= last ) {
756                 return LBER_DEFAULT;
757         }
758
759         return ber_peek_tag( ber, len );
760 }
761
762 /* VARARGS */
763 ber_tag_t
764 ber_scanf ( BerElement *ber,
765         LDAP_CONST char *fmt,
766         ... )
767 {
768         va_list         ap;
769         LDAP_CONST char         *fmt_reset;
770         char            *s, **ss;
771         struct berval   **bvp, *bval;
772         ber_int_t       *i;
773         ber_len_t       *l;
774         ber_tag_t       *t;
775         ber_tag_t       rc;
776         ber_len_t       len;
777
778         va_start( ap, fmt );
779
780         assert( ber != NULL );
781         assert( fmt != NULL );
782
783         assert( LBER_VALID( ber ) );
784
785         fmt_reset = fmt;
786
787         if ( ber->ber_debug & (LDAP_DEBUG_TRACE|LDAP_DEBUG_BER)) {
788                 ber_log_printf( LDAP_DEBUG_TRACE, ber->ber_debug,
789                         "ber_scanf fmt (%s) ber:\n", fmt );
790                 ber_log_dump( LDAP_DEBUG_BER, ber->ber_debug, ber, 1 );
791         }
792
793         for ( rc = 0; *fmt && rc != LBER_DEFAULT; fmt++ ) {
794                 /* When this is modified, remember to update
795                  * the error-cleanup code below accordingly. */
796                 switch ( *fmt ) {
797                 case '!': { /* Hook */
798                                 BERDecodeCallback *f;
799                                 void *p;
800
801                                 f = va_arg( ap, BERDecodeCallback * );
802                                 p = va_arg( ap, void * );
803
804                                 rc = (*f)( ber, p, 0 );
805                         } break;
806
807                 case 'a':       /* octet string - allocate storage as needed */
808                         ss = va_arg( ap, char ** );
809                         rc = ber_get_stringa( ber, ss );
810                         break;
811
812                 case 'A':       /* octet string - allocate storage as needed,
813                                  * but return NULL if len == 0 */
814                         ss = va_arg( ap, char ** );
815                         rc = ber_get_stringa_null( ber, ss );
816                         break;
817
818                 case 'b':       /* boolean */
819                         i = va_arg( ap, ber_int_t * );
820                         rc = ber_get_boolean( ber, i );
821                         break;
822
823                 case 'B':       /* bit string - allocate storage as needed */
824                         ss = va_arg( ap, char ** );
825                         l = va_arg( ap, ber_len_t * ); /* for length, in bits */
826                         rc = ber_get_bitstringa( ber, ss, l );
827                         break;
828
829                 case 'e':       /* enumerated */
830                 case 'i':       /* int */
831                         i = va_arg( ap, ber_int_t * );
832                         rc = ber_get_int( ber, i );
833                         break;
834
835                 case 'l':       /* length of next item */
836                         l = va_arg( ap, ber_len_t * );
837                         rc = ber_peek_tag( ber, l );
838                         break;
839
840                 case 'm':       /* octet string in berval, in-place */
841                         bval = va_arg( ap, struct berval * );
842                         rc = ber_get_stringbv( ber, bval, 0 );
843                         break;
844
845                 case 'M':       /* bvoffarray - must include address of
846                                  * a record len, and record offset.
847                                  * number of records will be returned thru
848                                  * len ptr on finish. parsed in-place.
849                                  */
850                 {
851                         bgbvr cookie = { BvOff };
852                         cookie.ber = ber;
853                         cookie.res.ba = va_arg( ap, struct berval ** );
854                         cookie.alloc = 0;
855                         l = va_arg( ap, ber_len_t * );
856                         cookie.siz = *l;
857                         cookie.off = va_arg( ap, ber_len_t );
858                         rc = ber_get_stringbvl( &cookie, l );
859                         break;
860                 }
861
862                 case 'n':       /* null */
863                         rc = ber_get_null( ber );
864                         break;
865
866                 case 'o':       /* octet string in a supplied berval */
867                         bval = va_arg( ap, struct berval * );
868                         rc = ber_get_stringbv( ber, bval, LBER_BV_ALLOC );
869                         break;
870
871                 case 'O':       /* octet string - allocate & include length */
872                         bvp = va_arg( ap, struct berval ** );
873                         rc = ber_get_stringal( ber, bvp );
874                         break;
875
876                 case 's':       /* octet string - in a buffer */
877                         s = va_arg( ap, char * );
878                         l = va_arg( ap, ber_len_t * );
879                         rc = ber_get_stringb( ber, s, l );
880                         break;
881
882                 case 't':       /* tag of next item */
883                         t = va_arg( ap, ber_tag_t * );
884                         *t = rc = ber_peek_tag( ber, &len );
885                         break;
886
887                 case 'T':       /* skip tag of next item */
888                         t = va_arg( ap, ber_tag_t * );
889                         *t = rc = ber_skip_tag( ber, &len );
890                         break;
891
892                 case 'v':       /* sequence of strings */
893                 {
894                         bgbvr cookie = { ChArray };
895                         cookie.ber = ber;
896                         cookie.res.c = va_arg( ap, char *** );
897                         cookie.alloc = LBER_BV_ALLOC;
898                         rc = ber_get_stringbvl( &cookie, NULL );
899                         break;
900                 }
901
902                 case 'V':       /* sequence of strings + lengths */
903                 {
904                         bgbvr cookie = { BvVec };
905                         cookie.ber = ber;
906                         cookie.res.bv = va_arg( ap, struct berval *** );
907                         cookie.alloc = LBER_BV_ALLOC;
908                         rc = ber_get_stringbvl( &cookie, NULL );
909                         break;
910                 }
911
912                 case 'W':       /* bvarray */
913                 {
914                         bgbvr cookie = { BvArray };
915                         cookie.ber = ber;
916                         cookie.res.ba = va_arg( ap, struct berval ** );
917                         cookie.alloc = LBER_BV_ALLOC;
918                         rc = ber_get_stringbvl( &cookie, NULL );
919                         break;
920                 }
921
922                 case 'x':       /* skip the next element - whatever it is */
923                         if ( (rc = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
924                                 break;
925                         ber->ber_ptr += len;
926                         ber->ber_tag = *(unsigned char *)ber->ber_ptr;
927                         break;
928
929                 case '{':       /* begin sequence */
930                 case '[':       /* begin set */
931                         if ( *(fmt + 1) != 'v' && *(fmt + 1) != 'V'
932                                 && *(fmt + 1) != 'W' && *(fmt + 1) != 'M' )
933                                 rc = ber_skip_tag( ber, &len );
934                         break;
935
936                 case '}':       /* end sequence */
937                 case ']':       /* end set */
938                         break;
939
940                 default:
941                         if( ber->ber_debug ) {
942                                 ber_log_printf( LDAP_DEBUG_ANY, ber->ber_debug,
943                                         "ber_scanf: unknown fmt %c\n", *fmt );
944                         }
945                         rc = LBER_DEFAULT;
946                         break;
947                 }
948         }
949
950         va_end( ap );
951         if ( rc == LBER_DEFAULT ) {
952                 /*
953                  * Error.  Reclaim malloced memory that was given to the caller.
954                  * Set allocated pointers to NULL, "data length" outvalues to 0.
955                  */
956                 va_start( ap, fmt );
957
958                 for ( ; fmt_reset < fmt; fmt_reset++ ) {
959                 switch ( *fmt_reset ) {
960                 case '!': { /* Hook */
961                                 BERDecodeCallback *f;
962                                 void *p;
963
964                                 f = va_arg( ap, BERDecodeCallback * );
965                                 p = va_arg( ap, void * );
966
967                                 (void) (*f)( ber, p, 1 );
968                         } break;
969
970                 case 'a':       /* octet string - allocate storage as needed */
971                 case 'A':
972                         ss = va_arg( ap, char ** );
973                         if ( *ss ) {
974                                 LBER_FREE( *ss );
975                                 *ss = NULL;
976                         }
977                         break;
978
979                 case 'b':       /* boolean */
980                 case 'e':       /* enumerated */
981                 case 'i':       /* int */
982                         (void) va_arg( ap, int * );
983                         break;
984
985                 case 'l':       /* length of next item */
986                         (void) va_arg( ap, ber_len_t * );
987                         break;
988
989                 case 'o':       /* octet string in a supplied berval */
990                         bval = va_arg( ap, struct berval * );
991                         if ( bval->bv_val != NULL ) {
992                                 LBER_FREE( bval->bv_val );
993                                 bval->bv_val = NULL;
994                         }
995                         bval->bv_len = 0;
996                         break;
997
998                 case 'O':       /* octet string - allocate & include length */
999                         bvp = va_arg( ap, struct berval ** );
1000                         if ( *bvp ) {
1001                                 ber_bvfree( *bvp );
1002                                 *bvp = NULL;
1003                         }
1004                         break;
1005
1006                 case 's':       /* octet string - in a buffer */
1007                         (void) va_arg( ap, char * );
1008                         (void) va_arg( ap, ber_len_t * );
1009                         break;
1010
1011                 case 't':       /* tag of next item */
1012                 case 'T':       /* skip tag of next item */
1013                         (void) va_arg( ap, ber_tag_t * );
1014                         break;
1015
1016                 case 'B':       /* bit string - allocate storage as needed */
1017                         ss = va_arg( ap, char ** );
1018                         if ( *ss ) {
1019                                 LBER_FREE( *ss );
1020                                 *ss = NULL;
1021                         }
1022                         *(va_arg( ap, ber_len_t * )) = 0; /* for length, in bits */
1023                         break;
1024
1025                 case 'm':       /* berval in-place */
1026                 case 'M':       /* BVoff array in-place */
1027                 case 'n':       /* null */
1028                 case 'v':       /* sequence of strings */
1029                 case 'V':       /* sequence of strings + lengths */
1030                 case 'W':       /* BerVarray */
1031                 case 'x':       /* skip the next element - whatever it is */
1032                 case '{':       /* begin sequence */
1033                 case '[':       /* begin set */
1034                 case '}':       /* end sequence */
1035                 case ']':       /* end set */
1036                         break;
1037
1038                 default:
1039                         /* format should be good */
1040                         assert( 0 );
1041                 }
1042                 }
1043
1044                 va_end( ap );
1045         }
1046
1047         return rc;
1048 }