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