git.sur5r.net Git - openldap/blob - 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                 return( LBER_DEFAULT );
 348         }
 349
 350         return( ber_peek_tag( ber, len ) );
 351 }
 352
 353 unsigned long
 354 ber_next_element( BerElement *ber, unsigned long *len, char *last )
 355 {
 356         if ( ber->ber_ptr == last ) {
 357                 return( LBER_DEFAULT );
 358         }
 359
 360         return( ber_peek_tag( ber, len ) );
 361 }
 362
 363 /* VARARGS */
 364 unsigned long
 365 ber_scanf
 366 #if HAVE_STDARG
 367         ( BerElement *ber, char *fmt, ... )
 368 #else
 369         ( va_alist )
 370 va_dcl
 371 #endif
 372 {
 373         va_list         ap;
 374 #ifndef HAVE_STDARG
 375         BerElement      *ber;
 376         char            *fmt;
 377 #endif
 378         char            *last;
 379         char            *s, **ss, ***sss;
 380         struct berval   ***bv, **bvp, *bval;
 381         int             *i, j;
 382         long            *l;
 383         unsigned long   rc, tag, len;
 384
 385 #ifdef HAVE_STDARG
 386         va_start( ap, fmt );
 387 #else
 388         va_start( ap );
 389         ber = va_arg( ap, BerElement * );
 390         fmt = va_arg( ap, char * );
 391 #endif
 392
 393 #ifdef LDAP_DEBUG
 394         if ( lber_debug & 64 ) {
 395                 fprintf( stderr, "ber_scanf fmt (%s) ber:\n", fmt );
 396                 ber_dump( ber, 1 );
 397         }
 398 #endif
 399
 400         for ( rc = 0; *fmt && rc != LBER_DEFAULT; fmt++ ) {
 401                 switch ( *fmt ) {
 402                 case 'a':       /* octet string - allocate storage as needed */
 403                         ss = va_arg( ap, char ** );
 404                         rc = ber_get_stringa( ber, ss );
 405                         break;
 406
 407                 case 'b':       /* boolean */
 408                         i = va_arg( ap, int * );
 409                         rc = ber_get_boolean( ber, i );
 410                         break;
 411
 412                 case 'e':       /* enumerated */
 413                 case 'i':       /* int */
 414                         l = va_arg( ap, long * );
 415                         rc = ber_get_int( ber, l );
 416                         break;
 417
 418                 case 'l':       /* length of next item */
 419                         l = va_arg( ap, long * );
 420                         rc = ber_peek_tag( ber, (unsigned long *)l );
 421                         break;
 422
 423                 case 'n':       /* null */
 424                         rc = ber_get_null( ber );
 425                         break;
 426
 427                 case 's':       /* octet string - in a buffer */
 428                         s = va_arg( ap, char * );
 429                         l = va_arg( ap, long * );
 430                         rc = ber_get_stringb( ber, s, (unsigned long *)l );
 431                         break;
 432
 433                 case 'o':       /* octet string in a supplied berval */
 434                         bval = va_arg( ap, struct berval * );
 435                         ber_peek_tag( ber, &bval->bv_len );
 436                         rc = ber_get_stringa( ber, &bval->bv_val );
 437                         break;
 438
 439                 case 'O':       /* octet string - allocate & include length */
 440                         bvp = va_arg( ap, struct berval ** );
 441                         rc = ber_get_stringal( ber, bvp );
 442                         break;
 443
 444                 case 'B':       /* bit string - allocate storage as needed */
 445                         ss = va_arg( ap, char ** );
 446                         l = va_arg( ap, long * ); /* for length, in bits */
 447                         rc = ber_get_bitstringa( ber, ss, (unsigned long *)l );
 448                         break;
 449
 450                 case 't':       /* tag of next item */
 451                         i = va_arg( ap, int * );
 452                         *i = rc = ber_peek_tag( ber, &len );
 453                         break;
 454
 455                 case 'T':       /* skip tag of next item */
 456                         i = va_arg( ap, int * );
 457                         *i = rc = ber_skip_tag( ber, &len );
 458                         break;
 459
 460                 case 'v':       /* sequence of strings */
 461                         sss = va_arg( ap, char *** );
 462                         *sss = NULL;
 463                         j = 0;
 464                         for ( tag = ber_first_element( ber, &len, &last );
 465                             tag != LBER_DEFAULT && rc != LBER_DEFAULT;
 466                             tag = ber_next_element( ber, &len, last ) ) {
 467                                 if ( *sss == NULL ) {
 468                                         *sss = (char **) malloc(
 469                                             2 * sizeof(char *) );
 470                                 } else {
 471                                         *sss = (char **) realloc( *sss,
 472                                             (j + 2) * sizeof(char *) );
 473                                 }
 474                                 rc = ber_get_stringa( ber, &((*sss)[j]) );
 475                                 j++;
 476                         }
 477                         if ( j > 0 )
 478                                 (*sss)[j] = NULL;
 479                         break;
 480
 481                 case 'V':       /* sequence of strings + lengths */
 482                         bv = va_arg( ap, struct berval *** );
 483                         *bv = NULL;
 484                         j = 0;
 485                         for ( tag = ber_first_element( ber, &len, &last );
 486                             tag != LBER_DEFAULT && rc != LBER_DEFAULT;
 487                             tag = ber_next_element( ber, &len, last ) ) {
 488                                 if ( *bv == NULL ) {
 489                                         *bv = (struct berval **) malloc(
 490                                             2 * sizeof(struct berval *) );
 491                                 } else {
 492                                         *bv = (struct berval **) realloc( *bv,
 493                                             (j + 2) * sizeof(struct berval *) );
 494                                 }
 495                                 rc = ber_get_stringal( ber, &((*bv)[j]) );
 496                                 j++;
 497                         }
 498                         if ( j > 0 )
 499                                 (*bv)[j] = NULL;
 500                         break;
 501
 502                 case 'x':       /* skip the next element - whatever it is */
 503                         if ( (rc = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
 504                                 break;
 505                         ber->ber_ptr += len;
 506                         break;
 507
 508                 case '{':       /* begin sequence */
 509                 case '[':       /* begin set */
 510                         if ( *(fmt + 1) != 'v' && *(fmt + 1) != 'V' )
 511                                 rc = ber_skip_tag( ber, &len );
 512                         break;
 513
 514                 case '}':       /* end sequence */
 515                 case ']':       /* end set */
 516                         break;
 517
 518                 default:
 519 #ifdef LDAP_LIBUI
 520                         fprintf( stderr, "unknown fmt %c\n", *fmt );
 521 #endif /* LDAP_LIBUI */
 522                         rc = LBER_DEFAULT;
 523                         break;
 524                 }
 525         }
 526
 527         va_end( ap );
 528
 529         return( rc );
 530 }
 531
 532 void
 533 ber_bvfree( struct berval *bv )
 534 {
 535         if ( bv->bv_val != NULL )
 536                 free( bv->bv_val );
 537         free( (char *) bv );
 538 }
 539
 540 void
 541 ber_bvecfree( struct berval **bv )
 542 {
 543         int     i;
 544
 545         for ( i = 0; bv[i] != NULL; i++ )
 546                 ber_bvfree( bv[i] );
 547         free( (char *) bv );
 548 }
 549
 550 struct berval *
 551 ber_bvdup( struct berval *bv )
 552 {
 553         struct berval   *new;
 554
 555         if ( (new = (struct berval *) malloc( sizeof(struct berval) ))
 556             == NULL ) {
 557                 return( NULL );
 558         }
 559         if ( (new->bv_val = (char *) malloc( bv->bv_len + 1 )) == NULL ) {
 560                 return( NULL );
 561         }
 562         SAFEMEMCPY( new->bv_val, bv->bv_val, (size_t) bv->bv_len );
 563         new->bv_val[bv->bv_len] = '\0';
 564         new->bv_len = bv->bv_len;
 565
 566         return( new );
 567 }
 568
 569
 570 #ifdef STR_TRANSLATION
 571 void
 572 ber_set_string_translators( BerElement *ber, BERTranslateProc encode_proc,
 573         BERTranslateProc decode_proc )
 574 {
 575     ber->ber_encode_translate_proc = encode_proc;
 576     ber->ber_decode_translate_proc = decode_proc;
 577 }
 578 #endif /* STR_TRANSLATION */