git.sur5r.net Git - openldap/blob - libraries/liblber/decode.c

   1 /* decode.c - ber input decoding routines */
   2 /* $OpenLDAP$ */
   3 /*
   4  * Copyright 1998-2000 The OpenLDAP Foundation, All Rights Reserved.
   5  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
   6  */
   7 /* Portions
   8  * Copyright (c) 1990 Regents of the University of Michigan.
   9  * All rights reserved.
  10  *
  11  * Redistribution and use in source and binary forms are permitted
  12  * provided that this notice is preserved and that due credit is given
  13  * to the University of Michigan at Ann Arbor. The name of the University
  14  * may not be used to endorse or promote products derived from this
  15  * software without specific prior written permission. This software
  16  * is provided ``as is'' without express or implied warranty.
  17  */
  18
  19 #include "portable.h"
  20
  21 #include <stdio.h>
  22
  23 #include <ac/stdlib.h>
  24
  25 #include <ac/stdarg.h>
  26 #include <ac/string.h>
  27 #include <ac/socket.h>
  28
  29 #include "lber-int.h"
  30
  31 static ber_len_t ber_getnint LDAP_P((
  32         BerElement *ber,
  33         ber_int_t *num,
  34         ber_len_t len ));
  35
  36 /* return the tag - LBER_DEFAULT returned means trouble */
  37 ber_tag_t
  38 ber_get_tag( BerElement *ber )
  39 {
  40         unsigned char   xbyte;
  41         ber_tag_t       tag;
  42         unsigned int    i;
  43
  44         assert( ber != NULL );
  45         assert( BER_VALID( ber ) );
  46
  47         if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 )
  48                 return( LBER_DEFAULT );
  49
  50         tag = xbyte;
  51
  52         if ( (xbyte & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK )
  53                 return tag;
  54
  55         for ( i = 1; i < sizeof(ber_tag_t); i++ ) {
  56                 if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 )
  57                         return( LBER_DEFAULT );
  58
  59                 tag <<= 8;
  60                 tag |= 0x00ffUL & (ber_tag_t) xbyte;
  61
  62                 if ( ! (xbyte & LBER_MORE_TAG_MASK) )
  63                         break;
  64         }
  65
  66         /* tag too big! */
  67         if ( i == sizeof(ber_tag_t) )
  68                 return( LBER_DEFAULT );
  69
  70         return tag;
  71 }
  72
  73 ber_tag_t
  74 ber_skip_tag( BerElement *ber, ber_len_t *len )
  75 {
  76         ber_tag_t       tag;
  77         unsigned char   lc;
  78         ber_len_t       i, noctets;
  79         unsigned char netlen[sizeof(ber_len_t)];
  80
  81         assert( ber != NULL );
  82         assert( len != NULL );
  83         assert( BER_VALID( ber ) );
  84
  85         /*
  86          * Any ber element looks like this: tag length contents.
  87          * Assuming everything's ok, we return the tag byte (we
  88          * can assume a single byte), and return the length in len.
  89          *
  90          * Assumptions:
  91          *      1) definite lengths
  92          *      2) primitive encodings used whenever possible
  93          */
  94
  95         *len = 0;
  96
  97         /*
  98          * First, we read the tag.
  99          */
 100
 101         if ( (tag = ber_get_tag( ber )) == LBER_DEFAULT )
 102                 return( LBER_DEFAULT );
 103
 104         /*
 105          * Next, read the length.  The first byte contains the length of
 106          * the length.  If bit 8 is set, the length is the long form,
 107          * otherwise it's the short form.  We don't allow a length that's
 108          * greater than what we can hold in a ber_len_t.
 109          */
 110
 111         if ( ber_read( ber, (char *) &lc, 1 ) != 1 )
 112                 return( LBER_DEFAULT );
 113
 114         if ( lc & 0x80U ) {
 115                 noctets = (lc & 0x7fU);
 116
 117                 if ( noctets > sizeof(ber_len_t) ) {
 118                         return( LBER_DEFAULT );
 119                 }
 120
 121                 if( (unsigned) ber_read( ber, netlen, noctets ) != noctets ) {
 122                         return( LBER_DEFAULT );
 123                 }
 124
 125                 for( i = 0; i < noctets; i++ ) {
 126                         *len <<= 8;
 127                         *len |= netlen[i];
 128                 }
 129
 130         } else {
 131                 *len = lc;
 132         }
 133
 134         return( tag );
 135 }
 136
 137 ber_tag_t
 138 ber_peek_tag(
 139         BerElement *ber,
 140         ber_len_t *len )
 141 {
 142         /*
 143          * This implementation assumes ber_skip_tag() only
 144          * modifies ber_ptr field of the BerElement.
 145          */
 146
 147         char *save;
 148         ber_tag_t       tag;
 149
 150         save = ber->ber_ptr;
 151         tag = ber_skip_tag( ber, len );
 152         ber->ber_ptr = save;
 153
 154         return tag;
 155 }
 156
 157 static ber_len_t
 158 ber_getnint(
 159         BerElement *ber,
 160         ber_int_t *num,
 161         ber_len_t len )
 162 {
 163         unsigned char buf[sizeof(ber_int_t)];
 164
 165         assert( ber != NULL );
 166         assert( num != NULL );
 167
 168         assert( BER_VALID( ber ) );
 169
 170         /*
 171          * The tag and length have already been stripped off.  We should
 172          * be sitting right before len bytes of 2's complement integer,
 173          * ready to be read straight into an int.  We may have to sign
 174          * extend after we read it in.
 175          */
 176
 177         if ( len > sizeof(ber_int_t) )
 178                 return( -1 );
 179
 180         /* read into the low-order bytes of our buffer */
 181         if ( (ber_len_t) ber_read( ber, (char *) buf, len ) != len ) {
 182                 return( -1 );
 183         }
 184
 185         if( len ) {
 186                 /* sign extend if necessary */
 187                 ber_len_t i;
 188                 ber_int_t netnum = 0x80 & buf[0] ? -1 : 0;
 189
 190                 /* shift in the bytes */
 191                 for( i=0 ; i<len; i++ ) {
 192                         netnum = (netnum << 8 ) | buf[i];
 193                 }
 194
 195                 *num = netnum;
 196
 197         } else {
 198                 *num = 0;
 199         }
 200
 201         return( len );
 202 }
 203
 204 ber_tag_t
 205 ber_get_int(
 206         BerElement *ber,
 207         ber_int_t *num )
 208 {
 209         ber_tag_t       tag;
 210         ber_len_t       len;
 211
 212         assert( ber != NULL );
 213         assert( BER_VALID( ber ) );
 214
 215         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
 216                 return( LBER_DEFAULT );
 217
 218         if ( ber_getnint( ber, num, len ) != len )
 219                 return( LBER_DEFAULT );
 220         else
 221                 return( tag );
 222 }
 223
 224 ber_tag_t
 225 ber_get_enum(
 226         BerElement *ber,
 227         ber_int_t *num )
 228 {
 229         return ber_get_int( ber, num );
 230 }
 231
 232 ber_tag_t
 233 ber_get_stringb(
 234         BerElement *ber,
 235         char *buf,
 236         ber_len_t *len )
 237 {
 238         ber_len_t       datalen;
 239         ber_tag_t       tag;
 240
 241         assert( ber != NULL );
 242         assert( BER_VALID( ber ) );
 243
 244         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT )
 245                 return( LBER_DEFAULT );
 246         if ( datalen > (*len - 1) )
 247                 return( LBER_DEFAULT );
 248
 249         if ( (ber_len_t) ber_read( ber, buf, datalen ) != datalen )
 250                 return( LBER_DEFAULT );
 251
 252         buf[datalen] = '\0';
 253
 254         *len = datalen;
 255         return( tag );
 256 }
 257
 258 ber_tag_t
 259 ber_get_stringa( BerElement *ber, char **buf )
 260 {
 261         ber_len_t       datalen;
 262         ber_tag_t       tag;
 263
 264         assert( ber != NULL );
 265         assert( buf != NULL );
 266
 267         assert( BER_VALID( ber ) );
 268
 269         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) {
 270                 *buf = NULL;
 271                 return( LBER_DEFAULT );
 272         }
 273
 274         if ( (*buf = (char *) LBER_MALLOC( datalen + 1 )) == NULL )
 275                 return( LBER_DEFAULT );
 276
 277         if ( (ber_len_t) ber_read( ber, *buf, datalen ) != datalen ) {
 278                 LBER_FREE( *buf );
 279                 *buf = NULL;
 280                 return( LBER_DEFAULT );
 281         }
 282         (*buf)[datalen] = '\0';
 283
 284         return( tag );
 285 }
 286
 287 ber_tag_t
 288 ber_get_stringal( BerElement *ber, struct berval **bv )
 289 {
 290         ber_len_t       len;
 291         ber_tag_t       tag;
 292
 293         assert( ber != NULL );
 294         assert( bv != NULL );
 295
 296         assert( BER_VALID( ber ) );
 297
 298         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) {
 299                 *bv = NULL;
 300                 return LBER_DEFAULT;
 301         }
 302
 303         *bv = (struct berval *) LBER_MALLOC( sizeof(struct berval) );
 304         if ( *bv == NULL ) {
 305                 return LBER_DEFAULT;
 306         }
 307
 308         if( len == 0 ) {
 309                 (*bv)->bv_val = NULL;
 310                 (*bv)->bv_len = 0;
 311                 return tag;
 312         }
 313
 314         (*bv)->bv_val = (char *) LBER_MALLOC( len + 1 );
 315         if ( (*bv)->bv_val == NULL ) {
 316                 LBER_FREE( *bv );
 317                 *bv = NULL;
 318                 return LBER_DEFAULT;
 319         }
 320
 321         if ( (ber_len_t) ber_read( ber, (*bv)->bv_val, len ) != len ) {
 322                 ber_bvfree( *bv );
 323                 *bv = NULL;
 324                 return LBER_DEFAULT;
 325         }
 326
 327         ((*bv)->bv_val)[len] = '\0';
 328         (*bv)->bv_len = len;
 329
 330         return( tag );
 331 }
 332
 333 ber_tag_t
 334 ber_get_bitstringa(
 335         BerElement *ber,
 336         char **buf,
 337         ber_len_t *blen )
 338 {
 339         ber_len_t       datalen;
 340         ber_tag_t       tag;
 341         unsigned char   unusedbits;
 342
 343         assert( ber != NULL );
 344         assert( buf != NULL );
 345         assert( blen != NULL );
 346
 347         assert( BER_VALID( ber ) );
 348
 349         if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) {
 350                 *buf = NULL;
 351                 return( LBER_DEFAULT );
 352         }
 353         --datalen;
 354
 355         if ( (*buf = (char *) LBER_MALLOC( datalen )) == NULL )
 356                 return( LBER_DEFAULT );
 357
 358         if ( ber_read( ber, (char *)&unusedbits, 1 ) != 1 ) {
 359                 LBER_FREE( buf );
 360                 *buf = NULL;
 361                 return( LBER_DEFAULT );
 362         }
 363
 364         if ( (ber_len_t) ber_read( ber, *buf, datalen ) != datalen ) {
 365                 LBER_FREE( buf );
 366                 *buf = NULL;
 367                 return( LBER_DEFAULT );
 368         }
 369
 370         *blen = datalen * 8 - unusedbits;
 371         return( tag );
 372 }
 373
 374 ber_tag_t
 375 ber_get_null( BerElement *ber )
 376 {
 377         ber_len_t       len;
 378         ber_tag_t       tag;
 379
 380         assert( ber != NULL );
 381         assert( BER_VALID( ber ) );
 382
 383         if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
 384                 return( LBER_DEFAULT );
 385
 386         if ( len != 0 )
 387                 return( LBER_DEFAULT );
 388
 389         return( tag );
 390 }
 391
 392 ber_tag_t
 393 ber_get_boolean(
 394         BerElement *ber,
 395         ber_int_t *boolval )
 396 {
 397         ber_int_t       longbool;
 398         ber_tag_t       rc;
 399
 400         assert( ber != NULL );
 401         assert( boolval != NULL );
 402
 403         assert( BER_VALID( ber ) );
 404
 405         rc = ber_get_int( ber, &longbool );
 406         *boolval = longbool;
 407
 408         return( rc );
 409 }
 410
 411 ber_tag_t
 412 ber_first_element(
 413         BerElement *ber,
 414         ber_len_t *len,
 415         char **last )
 416 {
 417         assert( ber != NULL );
 418         assert( len != NULL );
 419         assert( last != NULL );
 420
 421         /* skip the sequence header, use the len to mark where to stop */
 422         if ( ber_skip_tag( ber, len ) == LBER_DEFAULT ) {
 423                 *last = NULL;
 424                 return( LBER_DEFAULT );
 425         }
 426
 427         *last = ber->ber_ptr + *len;
 428
 429         if ( *last == ber->ber_ptr ) {
 430                 return( LBER_DEFAULT );
 431         }
 432
 433         return( ber_peek_tag( ber, len ) );
 434 }
 435
 436 ber_tag_t
 437 ber_next_element(
 438         BerElement *ber,
 439         ber_len_t *len,
 440         LDAP_CONST char *last )
 441 {
 442         assert( ber != NULL );
 443         assert( len != NULL );
 444         assert( last != NULL );
 445
 446         assert( BER_VALID( ber ) );
 447
 448         if ( ber->ber_ptr == last ) {
 449                 return( LBER_DEFAULT );
 450         }
 451
 452         return( ber_peek_tag( ber, len ) );
 453 }
 454
 455 /* VARARGS */
 456 ber_tag_t
 457 ber_scanf ( BerElement *ber,
 458         LDAP_CONST char *fmt,
 459         ... )
 460 {
 461         va_list         ap;
 462         LDAP_CONST char         *fmt_reset;
 463         char            *last;
 464         char            *s, **ss, ***sss;
 465         struct berval   ***bv, **bvp, *bval;
 466         ber_int_t       *i;
 467         int j;
 468         ber_len_t       *l;
 469         ber_tag_t       *t;
 470         ber_tag_t       rc, tag;
 471         ber_len_t       len;
 472
 473         va_start( ap, fmt );
 474
 475         assert( ber != NULL );
 476         assert( fmt != NULL );
 477
 478         assert( BER_VALID( ber ) );
 479
 480         fmt_reset = fmt;
 481
 482         ber_log_printf( LDAP_DEBUG_TRACE, ber->ber_debug,
 483                 "ber_scanf fmt (%s) ber:\n", fmt );
 484         ber_log_dump( LDAP_DEBUG_BER, ber->ber_debug, ber, 1 );
 485
 486         for ( rc = 0; *fmt && rc != LBER_DEFAULT; fmt++ ) {
 487                 /* When this is modified, remember to update
 488                  * the error-cleanup code below accordingly. */
 489                 switch ( *fmt ) {
 490                 case '!': { /* Hook */
 491                                 BERDecodeCallback *f;
 492                                 void *p;
 493
 494                                 f = va_arg( ap, BERDecodeCallback * );
 495                                 p = va_arg( ap, void * );
 496
 497                                 rc = (*f)( ber, p, 0 );
 498                         } break;
 499
 500                 case 'a':       /* octet string - allocate storage as needed */
 501                         ss = va_arg( ap, char ** );
 502                         rc = ber_get_stringa( ber, ss );
 503                         break;
 504
 505                 case 'b':       /* boolean */
 506                         i = va_arg( ap, ber_int_t * );
 507                         rc = ber_get_boolean( ber, i );
 508                         break;
 509
 510                 case 'e':       /* enumerated */
 511                 case 'i':       /* int */
 512                         i = va_arg( ap, ber_int_t * );
 513                         rc = ber_get_int( ber, i );
 514                         break;
 515
 516                 case 'l':       /* length of next item */
 517                         l = va_arg( ap, ber_len_t * );
 518                         rc = ber_peek_tag( ber, l );
 519                         break;
 520
 521                 case 'n':       /* null */
 522                         rc = ber_get_null( ber );
 523                         break;
 524
 525                 case 's':       /* octet string - in a buffer */
 526                         s = va_arg( ap, char * );
 527                         l = va_arg( ap, ber_len_t * );
 528                         rc = ber_get_stringb( ber, s, l );
 529                         break;
 530
 531                 case 'o':       /* octet string in a supplied berval */
 532                         bval = va_arg( ap, struct berval * );
 533                         ber_peek_tag( ber, &bval->bv_len );
 534                         rc = ber_get_stringa( ber, &bval->bv_val );
 535                         break;
 536
 537                 case 'O':       /* octet string - allocate & include length */
 538                         bvp = va_arg( ap, struct berval ** );
 539                         rc = ber_get_stringal( ber, bvp );
 540                         break;
 541
 542                 case 'B':       /* bit string - allocate storage as needed */
 543                         ss = va_arg( ap, char ** );
 544                         l = va_arg( ap, ber_len_t * ); /* for length, in bits */
 545                         rc = ber_get_bitstringa( ber, ss, l );
 546                         break;
 547
 548                 case 't':       /* tag of next item */
 549                         t = va_arg( ap, ber_tag_t * );
 550                         *t = rc = ber_peek_tag( ber, &len );
 551                         break;
 552
 553                 case 'T':       /* skip tag of next item */
 554                         t = va_arg( ap, ber_tag_t * );
 555                         *t = rc = ber_skip_tag( ber, &len );
 556                         break;
 557
 558                 case 'v':       /* sequence of strings */
 559                         sss = va_arg( ap, char *** );
 560                         *sss = NULL;
 561                         j = 0;
 562                         for ( tag = ber_first_element( ber, &len, &last );
 563                             tag != LBER_DEFAULT && rc != LBER_DEFAULT;
 564                             tag = ber_next_element( ber, &len, last ) )
 565                         {
 566                                 *sss = (char **) LBER_REALLOC( *sss,
 567                                         (j + 2) * sizeof(char *) );
 568
 569                                 rc = ber_get_stringa( ber, &((*sss)[j]) );
 570                                 j++;
 571                         }
 572                         if ( j > 0 )
 573                                 (*sss)[j] = NULL;
 574                         break;
 575
 576                 case 'V':       /* sequence of strings + lengths */
 577                         bv = va_arg( ap, struct berval *** );
 578                         *bv = NULL;
 579                         j = 0;
 580                         for ( tag = ber_first_element( ber, &len, &last );
 581                             tag != LBER_DEFAULT && rc != LBER_DEFAULT;
 582                             tag = ber_next_element( ber, &len, last ) )
 583                         {
 584                                 *bv = (struct berval **) LBER_REALLOC( *bv,
 585                                         (j + 2) * sizeof(struct berval *) );
 586
 587                                 rc = ber_get_stringal( ber, &((*bv)[j]) );
 588                                 j++;
 589                         }
 590                         if ( j > 0 )
 591                                 (*bv)[j] = NULL;
 592                         break;
 593
 594                 case 'x':       /* skip the next element - whatever it is */
 595                         if ( (rc = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
 596                                 break;
 597                         ber->ber_ptr += len;
 598                         break;
 599
 600                 case '{':       /* begin sequence */
 601                 case '[':       /* begin set */
 602                         if ( *(fmt + 1) != 'v' && *(fmt + 1) != 'V' )
 603                                 rc = ber_skip_tag( ber, &len );
 604                         break;
 605
 606                 case '}':       /* end sequence */
 607                 case ']':       /* end set */
 608                         break;
 609
 610                 default:
 611                         if( ber->ber_debug ) {
 612                                 ber_log_printf( LDAP_DEBUG_ANY, ber->ber_debug,
 613                                         "ber_scanf: unknown fmt %c\n", *fmt );
 614                         }
 615                         rc = LBER_DEFAULT;
 616                         break;
 617                 }
 618         }
 619
 620         va_end( ap );
 621
 622         if ( rc == LBER_DEFAULT ) {
 623             /*
 624              * Error.  Reclaim malloced memory that was given to the caller.
 625              * Set allocated pointers to NULL, "data length" outvalues to 0.
 626              */
 627             va_start( ap, fmt );
 628
 629             for ( ; fmt_reset < fmt; fmt_reset++ ) {
 630                 switch ( *fmt_reset ) {
 631                 case '!': { /* Hook */
 632                                 BERDecodeCallback *f;
 633                                 void *p;
 634
 635                                 f = va_arg( ap, BERDecodeCallback * );
 636                                 p = va_arg( ap, void * );
 637
 638                                 (void) (*f)( ber, p, 1 );
 639                         } break;
 640
 641                 case 'a':       /* octet string - allocate storage as needed */
 642                         ss = va_arg( ap, char ** );
 643                         if ( *ss ) {
 644                                 LBER_FREE( *ss );
 645                                 *ss = NULL;
 646                         }
 647                         break;
 648
 649                 case 'b':       /* boolean */
 650                 case 'e':       /* enumerated */
 651                 case 'i':       /* int */
 652                         (void) va_arg( ap, int * );
 653                         break;
 654
 655                 case 's':       /* octet string - in a buffer */
 656                         (void) va_arg( ap, char * );
 657                         (void) va_arg( ap, ber_len_t * );
 658                         break;
 659
 660                 case 'l':       /* length of next item */
 661                         (void) va_arg( ap, ber_len_t * );
 662                         break;
 663
 664                 case 't':       /* tag of next item */
 665                 case 'T':       /* skip tag of next item */
 666                         (void) va_arg( ap, ber_tag_t * );
 667                         break;
 668
 669                 case 'o':       /* octet string in a supplied berval */
 670                         bval = va_arg( ap, struct berval * );
 671                         if ( bval->bv_val != NULL ) {
 672                                 LBER_FREE( bval->bv_val );
 673                                 bval->bv_val = NULL;
 674                         }
 675                         bval->bv_len = 0;
 676                         break;
 677
 678                 case 'O':       /* octet string - allocate & include length */
 679                         bvp = va_arg( ap, struct berval ** );
 680                         if ( *bvp ) {
 681                                 ber_bvfree( *bvp );
 682                                 *bvp = NULL;
 683                         }
 684                         break;
 685
 686                 case 'B':       /* bit string - allocate storage as needed */
 687                         ss = va_arg( ap, char ** );
 688                         if ( *ss ) {
 689                                 LBER_FREE( *ss );
 690                                 *ss = NULL;
 691                         }
 692                         *(va_arg( ap, ber_len_t * )) = 0; /* for length, in bits */
 693                         break;
 694
 695                 case 'v':       /* sequence of strings */
 696                         sss = va_arg( ap, char *** );
 697                         if ( *sss ) {
 698                                 for (j = 0;  (*sss)[j];  j++) {
 699                                         LBER_FREE( (*sss)[j] );
 700                                         (*sss)[j] = NULL;
 701                                 }
 702                                 LBER_FREE( *sss );
 703                                 *sss = NULL;
 704                         }
 705                         break;
 706
 707                 case 'V':       /* sequence of strings + lengths */
 708                         bv = va_arg( ap, struct berval *** );
 709                         if ( *bv ) {
 710                                 ber_bvecfree( *bv );
 711                                 *bv = NULL;
 712                         }
 713                         break;
 714
 715                 case 'n':       /* null */
 716                 case 'x':       /* skip the next element - whatever it is */
 717                 case '{':       /* begin sequence */
 718                 case '[':       /* begin set */
 719                 case '}':       /* end sequence */
 720                 case ']':       /* end set */
 721                         break;
 722
 723                 default:
 724                         /* format should be good */
 725                         assert( 0 );
 726                 }
 727             }
 728
 729             va_end( ap );
 730         }
 731
 732         return( rc );
 733 }