]> git.sur5r.net Git - openldap/blob - libraries/liblber/decode.c
4a15d47a38b111f326ce57f37defedb8b97a4934
[openldap] / libraries / liblber / decode.c
1 /* decode.c - ber input decoding routines */
2 /*
3  * Copyright (c) 1990 Regents of the University of Michigan.
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms are permitted
7  * provided that this notice is preserved and that due credit is given
8  * to the University of Michigan at Ann Arbor. The name of the University
9  * may not be used to endorse or promote products derived from this
10  * software without specific prior written permission. This software
11  * is provided ``as is'' without express or implied warranty.
12  */
13
14 #include "portable.h"
15
16 #include <stdio.h>
17 #include <stdlib.h>
18
19 #include <ac/stdarg.h>
20
21 #include <ac/string.h>
22 #include <ac/socket.h>
23
24 #include "lber-int.h"
25
26 #ifdef LDAP_DEBUG
27 int     lber_debug;
28 #endif
29
30 static int ber_getnint LDAP_P(( BerElement *ber, long *num, int len ));
31
32 /* return the tag - LBER_DEFAULT returned means trouble */
33 unsigned long
34 ber_get_tag( BerElement *ber )
35 {
36         unsigned char   xbyte;
37         unsigned long   tag;
38         char            *tagp;
39         unsigned int    i;
40
41         if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 )
42                 return( LBER_DEFAULT );
43
44         if ( (xbyte & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK )
45                 return( (unsigned long) xbyte );
46
47         tagp = (char *) &tag;
48         tagp[0] = xbyte;
49         for ( i = 1; i < sizeof(long); i++ ) {
50                 if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 )
51                         return( LBER_DEFAULT );
52
53                 tagp[i] = xbyte;
54
55                 if ( ! (xbyte & LBER_MORE_TAG_MASK) )
56                         break;
57         }
58
59         /* tag too big! */
60         if ( i == sizeof(long) )
61                 return( LBER_DEFAULT );
62
63         /* want leading, not trailing 0's */
64         return( tag >> (sizeof(long) - i - 1) );
65 }
66
67 unsigned long
68 ber_skip_tag( BerElement *ber, unsigned long *len )
69 {
70         unsigned long   tag;
71         unsigned char   lc;
72         int             noctets, diff;
73         unsigned long   netlen;
74
75         /*
76          * Any ber element looks like this: tag length contents.
77          * Assuming everything's ok, we return the tag byte (we
78          * can assume a single byte), and return the length in len.
79          *
80          * Assumptions:
81          *      1) definite lengths
82          *      2) primitive encodings used whenever possible
83          */
84
85         /*
86          * First, we read the tag.
87          */
88
89         if ( (tag = ber_get_tag( ber )) == LBER_DEFAULT )
90                 return( LBER_DEFAULT );
91
92         /*
93          * Next, read the length.  The first byte contains the length of
94          * the length.  If bit 8 is set, the length is the long form,
95          * otherwise it's the short form.  We don't allow a length that's
96          * greater than what we can hold in an unsigned long.
97          */
98
99         *len = netlen = 0;
100         if ( ber_read( ber, (char *) &lc, 1 ) != 1 )
101                 return( LBER_DEFAULT );
102         if ( lc & 0x80 ) {
103                 noctets = (lc & 0x7f);
104                 if ( noctets > sizeof(unsigned long) )
105                         return( LBER_DEFAULT );
106                 diff = sizeof(unsigned long) - noctets;
107                 if ( ber_read( ber, (char *) &netlen + diff, noctets )
108                     != noctets )
109                         return( LBER_DEFAULT );
110                 *len = AC_NTOHL( netlen );
111         } else {
112                 *len = lc;
113         }
114
115         return( tag );
116 }
117
118 unsigned long
119 ber_peek_tag( BerElement *ber, unsigned long *len )
120 {
121         char            *save;
122         unsigned long   tag;
123
124         save = ber->ber_ptr;
125         tag = ber_skip_tag( ber, len );
126         ber->ber_ptr = save;
127
128         return( tag );
129 }
130
131 static int
132 ber_getnint( BerElement *ber, long *num, int len )
133 {
134         int     diff, sign, i;
135         long    netnum;
136         char    *p;
137
138         /*
139          * The tag and length have already been stripped off.  We should
140          * be sitting right before len bytes of 2's complement integer,
141          * ready to be read straight into an int.  We may have to sign
142          * extend after we read it in.
143          */
144
145         if ( len > sizeof(long) )
146                 return( -1 );
147
148         netnum = 0;
149         diff = sizeof(long) - len;
150         /* read into the low-order bytes of netnum */
151         if ( ber_read( ber, ((char *) &netnum) + diff, len ) != len )
152                 return( -1 );
153
154         /* sign extend if necessary */
155         p = (char *) &netnum;
156         sign = (0x80 & *(p+diff) );
157         if ( sign && len < sizeof(long) ) {
158                 for ( i = 0; i < diff; i++ ) {
159                         *(p+i) = (unsigned char) 0xff;
160                 }
161         }
162         *num = AC_NTOHL( netnum );
163
164         return( len );
165 }
166
167 unsigned long
168 ber_get_int( BerElement *ber, long *num )
169 {
170         unsigned long   tag, len;
171
172         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
173                 return( LBER_DEFAULT );
174
175         if ( (unsigned long) ber_getnint( ber, num, (int)len ) != len )
176                 return( LBER_DEFAULT );
177         else
178                 return( tag );
179 }
180
181 unsigned long
182 ber_get_stringb( BerElement *ber, char *buf, unsigned long *len )
183 {
184         unsigned long   datalen, tag;
185 #ifdef STR_TRANSLATION
186         char            *transbuf;
187 #endif /* STR_TRANSLATION */
188
189         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT )
190                 return( LBER_DEFAULT );
191         if ( datalen > (*len - 1) )
192                 return( LBER_DEFAULT );
193
194         if ( (unsigned long) ber_read( ber, buf, datalen ) != datalen )
195                 return( LBER_DEFAULT );
196
197         buf[datalen] = '\0';
198
199 #ifdef STR_TRANSLATION
200         if ( datalen > 0 && ( ber->ber_options & LBER_TRANSLATE_STRINGS ) != 0
201             && ber->ber_decode_translate_proc != NULL ) {
202                 transbuf = buf;
203                 ++datalen;
204                 if ( (*(ber->ber_decode_translate_proc))( &transbuf, &datalen,
205                     0 ) != 0 ) {
206                         return( LBER_DEFAULT );
207                 }
208                 if ( datalen > *len ) {
209                         free( transbuf );
210                         return( LBER_DEFAULT );
211                 }
212                 SAFEMEMCPY( buf, transbuf, datalen );
213                 free( transbuf );
214                 --datalen;
215         }
216 #endif /* STR_TRANSLATION */
217
218         *len = datalen;
219         return( tag );
220 }
221
222 unsigned long
223 ber_get_stringa( BerElement *ber, char **buf )
224 {
225         unsigned long   datalen, tag;
226
227         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT )
228                 return( LBER_DEFAULT );
229
230         if ( (*buf = (char *) malloc( (size_t)datalen + 1 )) == NULL )
231                 return( LBER_DEFAULT );
232
233         if ( (unsigned long) ber_read( ber, *buf, datalen ) != datalen )
234                 return( LBER_DEFAULT );
235         (*buf)[datalen] = '\0';
236
237 #ifdef STR_TRANSLATION
238         if ( datalen > 0 && ( ber->ber_options & LBER_TRANSLATE_STRINGS ) != 0
239             && ber->ber_decode_translate_proc != NULL ) {
240                 ++datalen;
241                 if ( (*(ber->ber_decode_translate_proc))( buf, &datalen, 1 )
242                     != 0 ) {
243                         free( *buf );
244                         return( LBER_DEFAULT );
245                 }
246         }
247 #endif /* STR_TRANSLATION */
248
249         return( tag );
250 }
251
252 unsigned long
253 ber_get_stringal( BerElement *ber, struct berval **bv )
254 {
255         unsigned long   len, tag;
256
257         if ( (*bv = (struct berval *) malloc( sizeof(struct berval) )) == NULL )
258                 return( LBER_DEFAULT );
259
260         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
261                 return( LBER_DEFAULT );
262
263         if ( ((*bv)->bv_val = (char *) malloc( (size_t)len + 1 )) == NULL )
264                 return( LBER_DEFAULT );
265
266         if ( (unsigned long) ber_read( ber, (*bv)->bv_val, len ) != len )
267                 return( LBER_DEFAULT );
268         ((*bv)->bv_val)[len] = '\0';
269         (*bv)->bv_len = len;
270
271 #ifdef STR_TRANSLATION
272         if ( len > 0 && ( ber->ber_options & LBER_TRANSLATE_STRINGS ) != 0
273             && ber->ber_decode_translate_proc != NULL ) {
274                 ++len;
275                 if ( (*(ber->ber_decode_translate_proc))( &((*bv)->bv_val),
276                     &len, 1 ) != 0 ) {
277                         free( (*bv)->bv_val );
278                         return( LBER_DEFAULT );
279                 }
280                 (*bv)->bv_len = len - 1;
281         }
282 #endif /* STR_TRANSLATION */
283
284         return( tag );
285 }
286
287 unsigned long
288 ber_get_bitstringa( BerElement *ber, char **buf, unsigned long *blen )
289 {
290         unsigned long   datalen, tag;
291         unsigned char   unusedbits;
292
293         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT )
294                 return( LBER_DEFAULT );
295         --datalen;
296
297         if ( (*buf = (char *) malloc( (size_t)datalen )) == NULL )
298                 return( LBER_DEFAULT );
299
300         if ( ber_read( ber, (char *)&unusedbits, 1 ) != 1 )
301                 return( LBER_DEFAULT );
302
303         if ( (unsigned long) ber_read( ber, *buf, datalen ) != datalen )
304                 return( LBER_DEFAULT );
305
306         *blen = datalen * 8 - unusedbits;
307         return( tag );
308 }
309
310 unsigned long
311 ber_get_null( BerElement *ber )
312 {
313         unsigned long   len, tag;
314
315         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
316                 return( LBER_DEFAULT );
317
318         if ( len != 0 )
319                 return( LBER_DEFAULT );
320
321         return( tag );
322 }
323
324 unsigned long
325 ber_get_boolean( BerElement *ber, int *boolval )
326 {
327         long    longbool;
328         int     rc;
329
330         rc = ber_get_int( ber, &longbool );
331         *boolval = longbool;
332
333         return( rc );
334 }
335
336 unsigned long
337 ber_first_element( BerElement *ber, unsigned long *len, char **last )
338 {
339         /* skip the sequence header, use the len to mark where to stop */
340         if ( ber_skip_tag( ber, len ) == LBER_DEFAULT ) {
341                 return( LBER_DEFAULT );
342         }
343
344         *last = ber->ber_ptr + *len;
345
346         if ( *last == ber->ber_ptr ) {
347 #ifdef LBER_END_SEQORSET 
348                 return( LBER_END_SEQORSET );
349 #else
350                 return( LBER_DEFAULT );
351 #endif
352         }
353
354         return( ber_peek_tag( ber, len ) );
355 }
356
357 unsigned long
358 ber_next_element( BerElement *ber, unsigned long *len, char *last )
359 {
360         if ( ber->ber_ptr == last ) {
361 #ifdef LBER_END_SEQORSET 
362                 return( LBER_END_SEQORSET );
363 #else
364                 return( LBER_DEFAULT );
365 #endif
366         }
367
368         return( ber_peek_tag( ber, len ) );
369 }
370
371 /* VARARGS */
372 unsigned long
373 ber_scanf
374 #if HAVE_STDARG
375         ( BerElement *ber, char *fmt, ... )
376 #else
377         ( va_alist )
378 va_dcl
379 #endif
380 {
381         va_list         ap;
382 #ifndef HAVE_STDARG
383         BerElement      *ber;
384         char            *fmt;
385 #endif
386         char            *last;
387         char            *s, **ss, ***sss;
388         struct berval   ***bv, **bvp, *bval;
389         int             *i, j;
390         long            *l;
391         unsigned long   rc, tag, len;
392
393 #ifdef HAVE_STDARG
394         va_start( ap, fmt );
395 #else
396         va_start( ap );
397         ber = va_arg( ap, BerElement * );
398         fmt = va_arg( ap, char * );
399 #endif
400
401 #ifdef LDAP_DEBUG
402         if ( lber_debug & 64 ) {
403                 fprintf( stderr, "ber_scanf fmt (%s) ber:\n", fmt );
404                 ber_dump( ber, 1 );
405         }
406 #endif
407
408         for ( rc = 0; *fmt && rc != LBER_DEFAULT; fmt++ ) {
409                 switch ( *fmt ) {
410                 case 'a':       /* octet string - allocate storage as needed */
411                         ss = va_arg( ap, char ** );
412                         rc = ber_get_stringa( ber, ss );
413                         break;
414
415                 case 'b':       /* boolean */
416                         i = va_arg( ap, int * );
417                         rc = ber_get_boolean( ber, i );
418                         break;
419
420                 case 'e':       /* enumerated */
421                 case 'i':       /* int */
422                         l = va_arg( ap, long * );
423                         rc = ber_get_int( ber, l );
424                         break;
425
426                 case 'l':       /* length of next item */
427                         l = va_arg( ap, long * );
428                         rc = ber_peek_tag( ber, (unsigned long *)l );
429                         break;
430
431                 case 'n':       /* null */
432                         rc = ber_get_null( ber );
433                         break;
434
435                 case 's':       /* octet string - in a buffer */
436                         s = va_arg( ap, char * );
437                         l = va_arg( ap, long * );
438                         rc = ber_get_stringb( ber, s, (unsigned long *)l );
439                         break;
440
441                 case 'o':       /* octet string in a supplied berval */
442                         bval = va_arg( ap, struct berval * );
443                         ber_peek_tag( ber, &bval->bv_len );
444                         rc = ber_get_stringa( ber, &bval->bv_val );
445                         break;
446
447                 case 'O':       /* octet string - allocate & include length */
448                         bvp = va_arg( ap, struct berval ** );
449                         rc = ber_get_stringal( ber, bvp );
450                         break;
451
452                 case 'B':       /* bit string - allocate storage as needed */
453                         ss = va_arg( ap, char ** );
454                         l = va_arg( ap, long * ); /* for length, in bits */
455                         rc = ber_get_bitstringa( ber, ss, (unsigned long *)l );
456                         break;
457
458                 case 't':       /* tag of next item */
459                         i = va_arg( ap, int * );
460                         *i = rc = ber_peek_tag( ber, &len );
461                         break;
462
463                 case 'T':       /* skip tag of next item */
464                         i = va_arg( ap, int * );
465                         *i = rc = ber_skip_tag( ber, &len );
466                         break;
467
468                 case 'v':       /* sequence of strings */
469                         sss = va_arg( ap, char *** );
470                         *sss = NULL;
471                         j = 0;
472                         for ( tag = ber_first_element( ber, &len, &last );
473                             tag != LBER_DEFAULT && 
474 #ifdef LDAP_END_SEQORSET
475                                         tag != LBER_END_SEQORSET &&
476 #endif
477                                         rc != LBER_DEFAULT;
478                             tag = ber_next_element( ber, &len, last ) )
479                         {
480                                 if ( *sss == NULL ) {
481                                         *sss = (char **) malloc(
482                                             2 * sizeof(char *) );
483                                 } else {
484                                         *sss = (char **) realloc( *sss,
485                                             (j + 2) * sizeof(char *) );
486                                 }
487                                 rc = ber_get_stringa( ber, &((*sss)[j]) );
488                                 j++;
489                         }
490 #ifdef LDAP_END_SEQORSET
491                         if (rc != LBER_DEFAULT && 
492                                 tag != LBER_END_SEQORSET )
493                         {
494                                 rc = LBER_DEFAULT;
495                         }
496 #endif
497                         if ( j > 0 )
498                                 (*sss)[j] = NULL;
499                         break;
500
501                 case 'V':       /* sequence of strings + lengths */
502                         bv = va_arg( ap, struct berval *** );
503                         *bv = NULL;
504                         j = 0;
505                         for ( tag = ber_first_element( ber, &len, &last );
506                             tag != LBER_DEFAULT && 
507 #ifdef LDAP_END_SEQORSET
508                                         tag != LBER_END_SEQORSET &&
509 #endif
510                                         rc != LBER_DEFAULT;
511                             tag = ber_next_element( ber, &len, last ) )
512                         {
513                                 if ( *bv == NULL ) {
514                                         *bv = (struct berval **) malloc(
515                                             2 * sizeof(struct berval *) );
516                                 } else {
517                                         *bv = (struct berval **) realloc( *bv,
518                                             (j + 2) * sizeof(struct berval *) );
519                                 }
520                                 rc = ber_get_stringal( ber, &((*bv)[j]) );
521                                 j++;
522                         }
523 #ifdef LDAP_END_SEQORSET
524                         if (rc != LBER_DEFAULT && 
525                                 tag != LBER_END_SEQORSET )
526                         {
527                                 rc = LBER_DEFAULT;
528                         }
529 #endif
530                         if ( j > 0 )
531                                 (*bv)[j] = NULL;
532                         break;
533
534                 case 'x':       /* skip the next element - whatever it is */
535                         if ( (rc = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
536                                 break;
537                         ber->ber_ptr += len;
538                         break;
539
540                 case '{':       /* begin sequence */
541                 case '[':       /* begin set */
542                         if ( *(fmt + 1) != 'v' && *(fmt + 1) != 'V' )
543                                 rc = ber_skip_tag( ber, &len );
544                         break;
545
546                 case '}':       /* end sequence */
547                 case ']':       /* end set */
548                         break;
549
550                 default:
551 #ifdef LDAP_LIBUI
552                         fprintf( stderr, "unknown fmt %c\n", *fmt );
553 #endif /* LDAP_LIBUI */
554                         rc = LBER_DEFAULT;
555                         break;
556                 }
557         }
558
559         va_end( ap );
560
561         return( rc );
562 }
563
564 void
565 ber_bvfree( struct berval *bv )
566 {
567 #ifdef LBER_ASSERT
568         assert(bv != NULL);                     /* bv damn better point to something */
569 #endif
570         if ( bv->bv_val != NULL )
571                 free( bv->bv_val );
572         free( (char *) bv );
573 }
574
575 void
576 ber_bvecfree( struct berval **bv )
577 {
578         int     i;
579
580 #ifdef LBER_ASSERT
581         assert(bv != NULL);                     /* bv damn better point to something */
582 #endif
583         for ( i = 0; bv[i] != NULL; i++ )
584                 ber_bvfree( bv[i] );
585         free( (char *) bv );
586 }
587
588 struct berval *
589 ber_bvdup( struct berval *bv )
590 {
591         struct berval   *new;
592
593         if ( (new = (struct berval *) malloc( sizeof(struct berval) ))
594             == NULL ) {
595                 return( NULL );
596         }
597
598         if ( bv->bv_val == NULL ) {
599                 new->bv_val = NULL;
600                 new->bv_len = 0;
601                 return ( new );
602         }
603
604         if ( (new->bv_val = (char *) malloc( bv->bv_len + 1 )) == NULL ) {
605                 free( new );
606                 return( NULL );
607         }
608
609         SAFEMEMCPY( new->bv_val, bv->bv_val, (size_t) bv->bv_len );
610         new->bv_val[bv->bv_len] = '\0';
611         new->bv_len = bv->bv_len;
612
613         return( new );
614 }
615
616
617 #ifdef STR_TRANSLATION
618 void
619 ber_set_string_translators( BerElement *ber, BERTranslateProc encode_proc,
620         BERTranslateProc decode_proc )
621 {
622     ber->ber_encode_translate_proc = encode_proc;
623     ber->ber_decode_translate_proc = decode_proc;
624 }
625 #endif /* STR_TRANSLATION */