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