--- libraries/liblber/decode.c 2000/10/02 17:26:31 1.34.6.5 +++ libraries/liblber/decode.c 2009/08/07 14:04:25 1.124 @@ -1,11 +1,19 @@ /* decode.c - ber input decoding routines */ -/* $OpenLDAP: pkg/ldap/libraries/liblber/decode.c,v 1.34.6.4 2000/07/29 01:53:07 kurt Exp $ */ -/* - * Copyright 1998-2000 The OpenLDAP Foundation, All Rights Reserved. - * COPYING RESTRICTIONS APPLY, see COPYRIGHT file +/* $OpenLDAP: pkg/ldap/libraries/liblber/decode.c,v 1.123 2009/08/04 19:11:38 hallvard Exp $ */ +/* This work is part of OpenLDAP Software . + * + * Copyright 1998-2009 The OpenLDAP Foundation. + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted only as authorized by the OpenLDAP + * Public License. + * + * A copy of this license is available in the file LICENSE in the + * top-level directory of the distribution or, alternatively, at + * . */ -/* Portions - * Copyright (c) 1990 Regents of the University of Michigan. +/* Portions Copyright (c) 1990 Regents of the University of Michigan. * All rights reserved. * * Redistribution and use in source and binary forms are permitted @@ -15,23 +23,68 @@ * software without specific prior written permission. This software * is provided ``as is'' without express or implied warranty. */ +/* ACKNOWLEDGEMENTS: + * This work was originally developed by the University of Michigan + * (as part of U-MICH LDAP). + */ #include "portable.h" #include #include - #include #include #include #include "lber-int.h" -static ber_len_t ber_getnint LDAP_P(( - BerElement *ber, - ber_int_t *num, - ber_len_t len )); + +/* out->bv_len should be the buffer size on input */ +int +ber_decode_oid( BerValue *in, BerValue *out ) +{ + const unsigned char *der; + unsigned long val; + unsigned val1; + ber_len_t i; + char *ptr; + + assert( in != NULL ); + assert( out != NULL ); + + /* need 4 chars/inbyte + \0 for input={7f 7f 7f...} */ + if ( !out->bv_val || (out->bv_len+3)/4 <= in->bv_len ) + return -1; + + ptr = NULL; + der = (unsigned char *) in->bv_val; + val = 0; + for ( i=0; i < in->bv_len; i++ ) { + val |= der[i] & 0x7f; + if ( !( der[i] & 0x80 )) { + if ( ptr == NULL ) { + /* Initial "x.y": val=x*40+y, x<=2, y<40 if x<2 */ + ptr = out->bv_val; + val1 = (val < 80 ? val/40 : 2); + val -= val1*40; + ptr += sprintf( ptr, "%u", val1 ); + } + ptr += sprintf( ptr, ".%lu", val ); + val = 0; + } else if ( val - 1UL < LBER_OID_COMPONENT_MAX >> 7 ) { + val <<= 7; + } else { + /* val would overflow, or is 0 from invalid initial 0x80 octet */ + return -1; + } + } + if ( ptr == NULL || val != 0 ) + return -1; + + out->bv_len = ptr - out->bv_val; + return 0; +} /* return the tag - LBER_DEFAULT returned means trouble */ ber_tag_t @@ -39,99 +92,113 @@ ber_get_tag( BerElement *ber ) { unsigned char xbyte; ber_tag_t tag; - unsigned int i; assert( ber != NULL ); - assert( BER_VALID( ber ) ); + assert( LBER_VALID( ber ) ); - if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 ) - return( LBER_DEFAULT ); + if ( ber_pvt_ber_remaining( ber ) < 1 ) { + return LBER_DEFAULT; + } - tag = xbyte; + if ( ber->ber_ptr == ber->ber_buf ) { + tag = *(unsigned char *)ber->ber_ptr; + } else { + tag = ber->ber_tag; + } + ber->ber_ptr++; - if ( (xbyte & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK ) + if ( (tag & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK ) { return tag; + } - for ( i = 1; i < sizeof(ber_tag_t); i++ ) { - if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 ) - return( LBER_DEFAULT ); + do { + if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 ) { + break; + } tag <<= 8; tag |= 0x00ffUL & (ber_tag_t) xbyte; - if ( ! (xbyte & LBER_MORE_TAG_MASK) ) - break; - } - - /* tag too big! */ - if ( i == sizeof(ber_tag_t) ) - return( LBER_DEFAULT ); + if ( ! (xbyte & LBER_MORE_TAG_MASK) ) { + return tag; + } + } while ( tag <= (ber_tag_t)-1 / 256 ); - return tag; + return LBER_DEFAULT; /* error or tag too big */ } ber_tag_t -ber_skip_tag( BerElement *ber, ber_len_t *len ) +ber_skip_tag( BerElement *ber, ber_len_t *lenp ) { ber_tag_t tag; + ber_len_t len; + unsigned i, noctets; unsigned char lc; - ber_len_t i, noctets; unsigned char netlen[sizeof(ber_len_t)]; - assert( ber != NULL ); - assert( len != NULL ); - assert( BER_VALID( ber ) ); + assert( lenp != NULL ); /* * Any ber element looks like this: tag length contents. * Assuming everything's ok, we return the tag byte (we - * can assume a single byte), and return the length in len. + * can assume a single byte), and return the length in lenp. * * Assumptions: * 1) definite lengths * 2) primitive encodings used whenever possible */ - *len = 0; + *lenp = 0; /* * First, we read the tag. */ - if ( (tag = ber_get_tag( ber )) == LBER_DEFAULT ) - return( LBER_DEFAULT ); + if ( (tag = ber_get_tag( ber )) == LBER_DEFAULT ) { + return LBER_DEFAULT; + } /* - * Next, read the length. The first byte contains the length of - * the length. If bit 8 is set, the length is the long form, - * otherwise it's the short form. We don't allow a length that's - * greater than what we can hold in a ber_len_t. + * Next, read the length. The first octet determines the length + * of the length. If bit 8 is 0, the length is the short form, + * otherwise if the octet != 0x80 it's the long form, otherwise + * the ber element has the unsupported indefinite-length format. + * Lengths that do not fit in a ber_len_t are not accepted. */ - if ( ber_read( ber, (char *) &lc, 1 ) != 1 ) - return( LBER_DEFAULT ); + if ( ber_read( ber, (char *) &lc, 1 ) != 1 ) { + return LBER_DEFAULT; + } + len = lc; if ( lc & 0x80U ) { noctets = (lc & 0x7fU); - if ( noctets > sizeof(ber_len_t) ) { - return( LBER_DEFAULT ); + if ( noctets - 1U > sizeof(ber_len_t) - 1U ) { + /* Indefinite-length or too long length */ + return LBER_DEFAULT; } - if( (unsigned) ber_read( ber, netlen, noctets ) != noctets ) { - return( LBER_DEFAULT ); + if( (unsigned) ber_read( ber, (char *) netlen, noctets ) != noctets ) { + return LBER_DEFAULT; } - for( i = 0; i < noctets; i++ ) { - *len <<= 8; - *len |= netlen[i]; + len = netlen[0]; + for( i = 1; i < noctets; i++ ) { + len <<= 8; + len |= netlen[i]; } - } else { - *len = lc; } + *lenp = len; + + /* BER element should have enough data left */ + if( len > (ber_len_t) ber_pvt_ber_remaining( ber ) ) { + return LBER_DEFAULT; + } + ber->ber_tag = *(unsigned char *)ber->ber_ptr; - return( tag ); + return tag; } ber_tag_t @@ -145,50 +212,51 @@ ber_peek_tag( */ char *save; - ber_tag_t tag; + ber_tag_t tag, old; + old = ber->ber_tag; save = ber->ber_ptr; tag = ber_skip_tag( ber, len ); ber->ber_ptr = save; + ber->ber_tag = old; return tag; } -static ber_len_t -ber_getnint( +ber_tag_t +ber_get_int( BerElement *ber, - ber_int_t *num, - ber_len_t len ) + ber_int_t *num ) { + ber_tag_t tag; + ber_len_t len; unsigned char buf[sizeof(ber_int_t)]; - assert( ber != NULL ); assert( num != NULL ); - assert( BER_VALID( ber ) ); - - /* - * The tag and length have already been stripped off. We should - * be sitting right before len bytes of 2's complement integer, - * ready to be read straight into an int. We may have to sign - * extend after we read it in. - */ + if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) { + return LBER_DEFAULT; + } - if ( len > sizeof(ber_int_t) ) - return( -1 ); + if ( len > sizeof(ber_int_t) ) { + return LBER_DEFAULT; + } /* read into the low-order bytes of our buffer */ if ( (ber_len_t) ber_read( ber, (char *) buf, len ) != len ) { - return( -1 ); + return LBER_DEFAULT; } + /* parse two's complement integer */ if( len ) { - /* sign extend if necessary */ ber_len_t i; - ber_int_t netnum = 0x80 & buf[0] ? -1 : 0; + ber_int_t netnum = buf[0] & 0xff; + + /* sign extend */ + netnum -= (netnum & 0x80) << 1; /* shift in the bytes */ - for( i=0 ; iber_tag = *(unsigned char *)ber->ber_ptr; -ber_tag_t -ber_get_int( - BerElement *ber, - ber_int_t *num ) -{ - ber_tag_t tag; - ber_len_t len; - - assert( ber != NULL ); - assert( BER_VALID( ber ) ); - - if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) - return( LBER_DEFAULT ); - - if ( ber_getnint( ber, num, len ) != len ) - return( LBER_DEFAULT ); - else - return( tag ); + return tag; } ber_tag_t @@ -238,96 +288,286 @@ ber_get_stringb( ber_len_t datalen; ber_tag_t tag; - assert( ber != NULL ); - assert( BER_VALID( ber ) ); + if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) { + return LBER_DEFAULT; + } - if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) - return( LBER_DEFAULT ); - if ( datalen > (*len - 1) ) - return( LBER_DEFAULT ); + /* must fit within allocated space with termination */ + if ( datalen >= *len ) { + return LBER_DEFAULT; + } - if ( (ber_len_t) ber_read( ber, buf, datalen ) != datalen ) - return( LBER_DEFAULT ); + if ( (ber_len_t) ber_read( ber, buf, datalen ) != datalen ) { + return LBER_DEFAULT; + } + ber->ber_tag = *(unsigned char *)ber->ber_ptr; buf[datalen] = '\0'; *len = datalen; - return( tag ); + return tag; +} + +/* Definitions for get_string vector + * + * ChArray, BvArray, and BvVec are self-explanatory. + * BvOff is a struct berval embedded in an array of larger structures + * of siz bytes at off bytes from the beginning of the struct. + */ +enum bgbvc { ChArray, BvArray, BvVec, BvOff }; + +/* Use this single cookie for state, to keep actual + * stack use to the absolute minimum. + */ +typedef struct bgbvr { + const enum bgbvc choice; + const int alloc; /* choice == BvOff ? 0 : LBER_ALLOC */ + ber_len_t siz; /* input array element size, output count */ + ber_len_t off; /* BvOff offset to the struct berval */ + void *result; +} bgbvr; + +static ber_tag_t +ber_get_stringbvl( BerElement *ber, bgbvr *b ) +{ + int i = 0, n; + ber_tag_t tag; + ber_len_t len, tot_size = 0, siz = b->siz; + char *last, *orig; + struct berval bv, *bvp = NULL; + union stringbvl_u { + char **ca; /* ChArray */ + BerVarray ba; /* BvArray */ + struct berval **bv; /* BvVec */ + char *bo; /* BvOff */ + } res; + + /* For rewinding, just like ber_peek_tag() */ + orig = ber->ber_ptr; + tag = ber->ber_tag; + + if ( ber_first_element( ber, &len, &last ) != LBER_DEFAULT ) { + for ( ; ber->ber_ptr < last; i++, tot_size += siz ) { + if ( ber_skip_tag( ber, &len ) == LBER_DEFAULT ) break; + ber->ber_ptr += len; + ber->ber_tag = *(unsigned char *) ber->ber_ptr; + } + } + + b->siz = i; + if ( i == 0 ) { + return 0; + } + + /* Allocate and NULL-terminate the result vector */ + b->result = ber_memalloc_x( tot_size + siz, ber->ber_memctx ); + if ( b->result == NULL ) { + return LBER_DEFAULT; + } + switch (b->choice) { + case ChArray: + res.ca = b->result; + res.ca[i] = NULL; + break; + case BvArray: + res.ba = b->result; + res.ba[i].bv_val = NULL; + break; + case BvVec: + res.bv = b->result; + res.bv[i] = NULL; + break; + case BvOff: + res.bo = (char *) b->result + b->off; + ((struct berval *) (res.bo + tot_size))->bv_val = NULL; + break; + } + ber->ber_ptr = orig; + ber->ber_tag = tag; + ber_skip_tag( ber, &len ); + + tot_size = 0; + n = 0; + do { + tag = ber_next_element( ber, &len, last ); + if ( ber_get_stringbv( ber, &bv, b->alloc ) == LBER_DEFAULT ) { + goto nomem; + } + + /* store my result */ + switch (b->choice) { + case ChArray: + res.ca[n] = bv.bv_val; + break; + case BvArray: + res.ba[n] = bv; + break; + case BvVec: + bvp = ber_memalloc_x( sizeof( struct berval ), + ber->ber_memctx ); + if ( !bvp ) { + ber_memfree_x( bv.bv_val, ber->ber_memctx ); + goto nomem; + } + res.bv[n] = bvp; + *bvp = bv; + break; + case BvOff: + *(struct berval *)(res.bo + tot_size) = bv; + tot_size += siz; + break; + } + } while (++n < i); + return tag; + +nomem: + if (b->choice != BvOff) { /* BvOff does not have b->alloc set */ + while (--n >= 0) { + switch(b->choice) { + case ChArray: + ber_memfree_x(res.ca[n], ber->ber_memctx); + break; + case BvArray: + ber_memfree_x(res.ba[n].bv_val, ber->ber_memctx); + break; + case BvVec: + ber_memfree_x(res.bv[n]->bv_val, ber->ber_memctx); + ber_memfree_x(res.bv[n], ber->ber_memctx); + break; + default: + break; + } + } + } + ber_memfree_x(b->result, ber->ber_memctx); + b->result = NULL; + return LBER_DEFAULT; } ber_tag_t -ber_get_stringa( BerElement *ber, char **buf ) +ber_get_stringbv( BerElement *ber, struct berval *bv, int option ) { - ber_len_t datalen; ber_tag_t tag; - assert( ber != NULL ); - assert( buf != NULL ); - - assert( BER_VALID( ber ) ); + assert( bv != NULL ); - if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) { - *buf = NULL; - return( LBER_DEFAULT ); + tag = ber_skip_tag( ber, &bv->bv_len ); + if ( tag == LBER_DEFAULT ) { + bv->bv_val = NULL; + return tag; } - if ( (*buf = (char *) LBER_MALLOC( datalen + 1 )) == NULL ) - return( LBER_DEFAULT ); + if ( option & LBER_BV_ALLOC ) { + bv->bv_val = (char *) ber_memalloc_x( bv->bv_len + 1, + ber->ber_memctx ); + if ( bv->bv_val == NULL ) { + return LBER_DEFAULT; + } - if ( (ber_len_t) ber_read( ber, *buf, datalen ) != datalen ) { - LBER_FREE( *buf ); - *buf = NULL; - return( LBER_DEFAULT ); + if ( bv->bv_len > 0 && (ber_len_t) ber_read( ber, bv->bv_val, + bv->bv_len ) != bv->bv_len ) + { + ber_memfree_x( bv->bv_val, ber->ber_memctx ); + bv->bv_val = NULL; + return LBER_DEFAULT; + } + } else { + bv->bv_val = ber->ber_ptr; + ber->ber_ptr += bv->bv_len; } - (*buf)[datalen] = '\0'; + ber->ber_tag = *(unsigned char *)ber->ber_ptr; + if ( !( option & LBER_BV_NOTERM )) + bv->bv_val[bv->bv_len] = '\0'; - return( tag ); + return tag; } ber_tag_t -ber_get_stringal( BerElement *ber, struct berval **bv ) +ber_get_stringbv_null( BerElement *ber, struct berval *bv, int option ) { - ber_len_t len; ber_tag_t tag; - assert( ber != NULL ); assert( bv != NULL ); - assert( BER_VALID( ber ) ); - - if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) { - *bv = NULL; - return LBER_DEFAULT; + tag = ber_skip_tag( ber, &bv->bv_len ); + if ( tag == LBER_DEFAULT || bv->bv_len == 0 ) { + bv->bv_val = NULL; + return tag; } - *bv = (struct berval *) LBER_MALLOC( sizeof(struct berval) ); - if ( *bv == NULL ) { - return LBER_DEFAULT; - } + if ( option & LBER_BV_ALLOC ) { + bv->bv_val = (char *) ber_memalloc_x( bv->bv_len + 1, + ber->ber_memctx ); + if ( bv->bv_val == NULL ) { + return LBER_DEFAULT; + } - if( len == 0 ) { - (*bv)->bv_val = NULL; - (*bv)->bv_len = 0; - return tag; + if ( bv->bv_len > 0 && (ber_len_t) ber_read( ber, bv->bv_val, + bv->bv_len ) != bv->bv_len ) + { + ber_memfree_x( bv->bv_val, ber->ber_memctx ); + bv->bv_val = NULL; + return LBER_DEFAULT; + } + } else { + bv->bv_val = ber->ber_ptr; + ber->ber_ptr += bv->bv_len; } + ber->ber_tag = *(unsigned char *)ber->ber_ptr; + if ( !( option & LBER_BV_NOTERM )) + bv->bv_val[bv->bv_len] = '\0'; - (*bv)->bv_val = (char *) LBER_MALLOC( len + 1 ); - if ( (*bv)->bv_val == NULL ) { - LBER_FREE( *bv ); - *bv = NULL; + return tag; +} + +ber_tag_t +ber_get_stringa( BerElement *ber, char **buf ) +{ + BerValue bv; + ber_tag_t tag; + + assert( buf != NULL ); + + tag = ber_get_stringbv( ber, &bv, LBER_BV_ALLOC ); + *buf = bv.bv_val; + + return tag; +} + +ber_tag_t +ber_get_stringa_null( BerElement *ber, char **buf ) +{ + BerValue bv; + ber_tag_t tag; + + assert( buf != NULL ); + + tag = ber_get_stringbv_null( ber, &bv, LBER_BV_ALLOC ); + *buf = bv.bv_val; + + return tag; +} + +ber_tag_t +ber_get_stringal( BerElement *ber, struct berval **bv ) +{ + ber_tag_t tag; + + assert( ber != NULL ); + assert( bv != NULL ); + + *bv = (struct berval *) ber_memalloc_x( sizeof(struct berval), + ber->ber_memctx ); + if ( *bv == NULL ) { return LBER_DEFAULT; } - if ( (ber_len_t) ber_read( ber, (*bv)->bv_val, len ) != len ) { - ber_bvfree( *bv ); + tag = ber_get_stringbv( ber, *bv, LBER_BV_ALLOC ); + if ( tag == LBER_DEFAULT ) { + ber_memfree_x( *bv, ber->ber_memctx ); *bv = NULL; - return LBER_DEFAULT; } - - ((*bv)->bv_val)[len] = '\0'; - (*bv)->bv_len = len; - - return( tag ); + return tag; } ber_tag_t @@ -340,53 +580,50 @@ ber_get_bitstringa( ber_tag_t tag; unsigned char unusedbits; - assert( ber != NULL ); assert( buf != NULL ); assert( blen != NULL ); - assert( BER_VALID( ber ) ); - if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) { - *buf = NULL; - return( LBER_DEFAULT ); + goto fail; } - --datalen; - - if ( (*buf = (char *) LBER_MALLOC( datalen )) == NULL ) - return( LBER_DEFAULT ); + if ( --datalen > (ber_len_t)-1 / 8 ) { + goto fail; + } if ( ber_read( ber, (char *)&unusedbits, 1 ) != 1 ) { - LBER_FREE( buf ); - *buf = NULL; - return( LBER_DEFAULT ); + goto fail; + } + if ( unusedbits > 7 ) { + goto fail; + } + + *buf = (char *) ber_memalloc_x( datalen, ber->ber_memctx ); + if ( *buf == NULL ) { + return LBER_DEFAULT; } if ( (ber_len_t) ber_read( ber, *buf, datalen ) != datalen ) { - LBER_FREE( buf ); + ber_memfree_x( buf, ber->ber_memctx ); *buf = NULL; - return( LBER_DEFAULT ); + return LBER_DEFAULT; } + ber->ber_tag = *(unsigned char *)ber->ber_ptr; *blen = datalen * 8 - unusedbits; - return( tag ); + return tag; + + fail: + *buf = NULL; + return LBER_DEFAULT; } ber_tag_t ber_get_null( BerElement *ber ) { ber_len_t len; - ber_tag_t tag; - - assert( ber != NULL ); - assert( BER_VALID( ber ) ); - - if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) - return( LBER_DEFAULT ); - - if ( len != 0 ) - return( LBER_DEFAULT ); + ber_tag_t tag = ber_skip_tag( ber, &len ); - return( tag ); + return( len == 0 ? tag : LBER_DEFAULT ); } ber_tag_t @@ -394,18 +631,7 @@ ber_get_boolean( BerElement *ber, ber_int_t *boolval ) { - ber_int_t longbool; - ber_tag_t rc; - - assert( ber != NULL ); - assert( boolval != NULL ); - - assert( BER_VALID( ber ) ); - - rc = ber_get_int( ber, &longbool ); - *boolval = longbool; - - return( rc ); + return ber_get_int( ber, boolval ); } ber_tag_t @@ -414,23 +640,21 @@ ber_first_element( ber_len_t *len, char **last ) { - assert( ber != NULL ); - assert( len != NULL ); assert( last != NULL ); /* skip the sequence header, use the len to mark where to stop */ if ( ber_skip_tag( ber, len ) == LBER_DEFAULT ) { *last = NULL; - return( LBER_DEFAULT ); + return LBER_DEFAULT; } *last = ber->ber_ptr + *len; - if ( *last == ber->ber_ptr ) { - return( LBER_DEFAULT ); + if ( *len == 0 ) { + return LBER_DEFAULT; } - return( ber_peek_tag( ber, len ) ); + return ber_peek_tag( ber, len ); } ber_tag_t @@ -440,16 +664,14 @@ ber_next_element( LDAP_CONST char *last ) { assert( ber != NULL ); - assert( len != NULL ); assert( last != NULL ); + assert( LBER_VALID( ber ) ); - assert( BER_VALID( ber ) ); - - if ( ber->ber_ptr == last ) { - return( LBER_DEFAULT ); + if ( ber->ber_ptr >= last ) { + return LBER_DEFAULT; } - return( ber_peek_tag( ber, len ) ); + return ber_peek_tag( ber, len ); } /* VARARGS */ @@ -460,28 +682,27 @@ ber_scanf ( BerElement *ber, { va_list ap; LDAP_CONST char *fmt_reset; - char *last; char *s, **ss, ***sss; - struct berval ***bv, **bvp, *bval; + struct berval *bval, **bvp, ***bvpp; ber_int_t *i; - int j; ber_len_t *l; ber_tag_t *t; - ber_tag_t rc, tag; + ber_tag_t rc; ber_len_t len; va_start( ap, fmt ); assert( ber != NULL ); assert( fmt != NULL ); - - assert( BER_VALID( ber ) ); + assert( LBER_VALID( ber ) ); fmt_reset = fmt; - ber_log_printf( LDAP_DEBUG_TRACE, ber->ber_debug, - "ber_scanf fmt (%s) ber:\n", fmt ); - ber_log_dump( LDAP_DEBUG_BER, ber->ber_debug, ber, 1 ); + if ( ber->ber_debug & (LDAP_DEBUG_TRACE|LDAP_DEBUG_BER)) { + ber_log_printf( LDAP_DEBUG_TRACE, ber->ber_debug, + "ber_scanf fmt (%s) ber:\n", fmt ); + ber_log_dump( LDAP_DEBUG_BER, ber->ber_debug, ber, 1 ); + } for ( rc = 0; *fmt && rc != LBER_DEFAULT; fmt++ ) { /* When this is modified, remember to update @@ -502,13 +723,25 @@ ber_scanf ( BerElement *ber, rc = ber_get_stringa( ber, ss ); break; + case 'A': /* octet string - allocate storage as needed, + * but return NULL if len == 0 */ + ss = va_arg( ap, char ** ); + rc = ber_get_stringa_null( ber, ss ); + break; + case 'b': /* boolean */ i = va_arg( ap, ber_int_t * ); rc = ber_get_boolean( ber, i ); break; + case 'B': /* bit string - allocate storage as needed */ + ss = va_arg( ap, char ** ); + l = va_arg( ap, ber_len_t * ); /* for length, in bits */ + rc = ber_get_bitstringa( ber, ss, l ); + break; + case 'e': /* enumerated */ - case 'i': /* int */ + case 'i': /* integer */ i = va_arg( ap, ber_int_t * ); rc = ber_get_int( ber, i ); break; @@ -518,20 +751,35 @@ ber_scanf ( BerElement *ber, rc = ber_peek_tag( ber, l ); break; - case 'n': /* null */ - rc = ber_get_null( ber ); + case 'm': /* octet string in berval, in-place */ + bval = va_arg( ap, struct berval * ); + rc = ber_get_stringbv( ber, bval, 0 ); break; - case 's': /* octet string - in a buffer */ - s = va_arg( ap, char * ); + case 'M': /* bvoffarray - must include address of + * a record len, and record offset. + * number of records will be returned thru + * len ptr on finish. parsed in-place. + */ + { + bgbvr cookie = { BvOff, 0 }; + bvp = va_arg( ap, struct berval ** ); l = va_arg( ap, ber_len_t * ); - rc = ber_get_stringb( ber, s, l ); + cookie.siz = *l; + cookie.off = va_arg( ap, ber_len_t ); + rc = ber_get_stringbvl( ber, &cookie ); + *bvp = cookie.result; + *l = cookie.siz; + break; + } + + case 'n': /* null */ + rc = ber_get_null( ber ); break; case 'o': /* octet string in a supplied berval */ bval = va_arg( ap, struct berval * ); - ber_peek_tag( ber, &bval->bv_len ); - rc = ber_get_stringa( ber, &bval->bv_val ); + rc = ber_get_stringbv( ber, bval, LBER_BV_ALLOC ); break; case 'O': /* octet string - allocate & include length */ @@ -539,10 +787,10 @@ ber_scanf ( BerElement *ber, rc = ber_get_stringal( ber, bvp ); break; - case 'B': /* bit string - allocate storage as needed */ - ss = va_arg( ap, char ** ); - l = va_arg( ap, ber_len_t * ); /* for length, in bits */ - rc = ber_get_bitstringa( ber, ss, l ); + case 's': /* octet string - in a buffer */ + s = va_arg( ap, char * ); + l = va_arg( ap, ber_len_t * ); + rc = ber_get_stringb( ber, s, l ); break; case 't': /* tag of next item */ @@ -556,51 +804,51 @@ ber_scanf ( BerElement *ber, break; case 'v': /* sequence of strings */ - sss = va_arg( ap, char *** ); - *sss = NULL; - j = 0; - for ( tag = ber_first_element( ber, &len, &last ); - tag != LBER_DEFAULT && rc != LBER_DEFAULT; - tag = ber_next_element( ber, &len, last ) ) - { - *sss = (char **) LBER_REALLOC( *sss, - (j + 2) * sizeof(char *) ); - - rc = ber_get_stringa( ber, &((*sss)[j]) ); - j++; - } - if ( j > 0 ) - (*sss)[j] = NULL; + { + bgbvr cookie = { + ChArray, LBER_BV_ALLOC, sizeof( char * ) + }; + rc = ber_get_stringbvl( ber, &cookie ); + *(va_arg( ap, char *** )) = cookie.result; break; + } case 'V': /* sequence of strings + lengths */ - bv = va_arg( ap, struct berval *** ); - *bv = NULL; - j = 0; - for ( tag = ber_first_element( ber, &len, &last ); - tag != LBER_DEFAULT && rc != LBER_DEFAULT; - tag = ber_next_element( ber, &len, last ) ) - { - *bv = (struct berval **) LBER_REALLOC( *bv, - (j + 2) * sizeof(struct berval *) ); - - rc = ber_get_stringal( ber, &((*bv)[j]) ); - j++; - } - if ( j > 0 ) - (*bv)[j] = NULL; + { + bgbvr cookie = { + BvVec, LBER_BV_ALLOC, sizeof( struct berval * ) + }; + rc = ber_get_stringbvl( ber, &cookie ); + *(va_arg( ap, struct berval *** )) = cookie.result; + break; + } + + case 'W': /* bvarray */ + { + bgbvr cookie = { + BvArray, LBER_BV_ALLOC, sizeof( struct berval ) + }; + rc = ber_get_stringbvl( ber, &cookie ); + *(va_arg( ap, struct berval ** )) = cookie.result; break; + } case 'x': /* skip the next element - whatever it is */ if ( (rc = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) break; ber->ber_ptr += len; + ber->ber_tag = *(unsigned char *)ber->ber_ptr; break; case '{': /* begin sequence */ case '[': /* begin set */ - if ( *(fmt + 1) != 'v' && *(fmt + 1) != 'V' ) + switch ( fmt[1] ) { + case 'v': case 'V': case 'W': case 'M': + break; + default: rc = ber_skip_tag( ber, &len ); + break; + } break; case '}': /* end sequence */ @@ -620,13 +868,13 @@ ber_scanf ( BerElement *ber, va_end( ap ); if ( rc == LBER_DEFAULT ) { - /* - * Error. Reclaim malloced memory that was given to the caller. - * Set allocated pointers to NULL, "data length" outvalues to 0. - */ - va_start( ap, fmt ); + /* + * Error. Reclaim malloced memory that was given to the caller. + * Set allocated pointers to NULL, "data length" outvalues to 0. + */ + va_start( ap, fmt ); - for ( ; fmt_reset < fmt; fmt_reset++ ) { + for ( ; fmt_reset < fmt; fmt_reset++ ) { switch ( *fmt_reset ) { case '!': { /* Hook */ BERDecodeCallback *f; @@ -639,77 +887,80 @@ ber_scanf ( BerElement *ber, } break; case 'a': /* octet string - allocate storage as needed */ + case 'A': ss = va_arg( ap, char ** ); - if ( *ss ) { - LBER_FREE( *ss ); - *ss = NULL; - } + ber_memfree_x( *ss, ber->ber_memctx ); + *ss = NULL; break; case 'b': /* boolean */ case 'e': /* enumerated */ - case 'i': /* int */ - (void) va_arg( ap, int * ); + case 'i': /* integer */ + (void) va_arg( ap, ber_int_t * ); break; - case 's': /* octet string - in a buffer */ - (void) va_arg( ap, char * ); - (void) va_arg( ap, ber_len_t * ); + case 'l': /* length of next item */ + *(va_arg( ap, ber_len_t * )) = 0; break; - case 'l': /* length of next item */ - (void) va_arg( ap, ber_len_t * ); + case 'm': /* berval in-place */ + bval = va_arg( ap, struct berval * ); + BER_BVZERO( bval ); break; - case 't': /* tag of next item */ - case 'T': /* skip tag of next item */ - (void) va_arg( ap, ber_tag_t * ); + case 'M': /* BVoff array in-place */ + bvp = va_arg( ap, struct berval ** ); + ber_memfree_x( *bvp, ber->ber_memctx ); + *bvp = NULL; + *(va_arg( ap, ber_len_t * )) = 0; + (void) va_arg( ap, ber_len_t ); break; case 'o': /* octet string in a supplied berval */ bval = va_arg( ap, struct berval * ); - if ( bval->bv_val != NULL ) { - LBER_FREE( bval->bv_val ); - bval->bv_val = NULL; - } - bval->bv_len = 0; + ber_memfree_x( bval->bv_val, ber->ber_memctx ); + BER_BVZERO( bval ); break; case 'O': /* octet string - allocate & include length */ bvp = va_arg( ap, struct berval ** ); - if ( *bvp ) { - ber_bvfree( *bvp ); - *bvp = NULL; - } + ber_bvfree_x( *bvp, ber->ber_memctx ); + *bvp = NULL; + break; + + case 's': /* octet string - in a buffer */ + (void) va_arg( ap, char * ); + *(va_arg( ap, ber_len_t * )) = 0; + break; + + case 't': /* tag of next item */ + case 'T': /* skip tag of next item */ + (void) va_arg( ap, ber_tag_t * ); break; case 'B': /* bit string - allocate storage as needed */ ss = va_arg( ap, char ** ); - if ( *ss ) { - LBER_FREE( *ss ); - *ss = NULL; - } + ber_memfree_x( *ss, ber->ber_memctx ); + *ss = NULL; *(va_arg( ap, ber_len_t * )) = 0; /* for length, in bits */ break; case 'v': /* sequence of strings */ sss = va_arg( ap, char *** ); - if ( *sss ) { - for (j = 0; (*sss)[j]; j++) { - LBER_FREE( (*sss)[j] ); - (*sss)[j] = NULL; - } - LBER_FREE( *sss ); - *sss = NULL; - } + ber_memvfree_x( (void **) *sss, ber->ber_memctx ); + *sss = NULL; break; case 'V': /* sequence of strings + lengths */ - bv = va_arg( ap, struct berval *** ); - if ( *bv ) { - ber_bvecfree( *bv ); - *bv = NULL; - } + bvpp = va_arg( ap, struct berval *** ); + ber_bvecfree_x( *bvpp, ber->ber_memctx ); + *bvpp = NULL; + break; + + case 'W': /* BerVarray */ + bvp = va_arg( ap, struct berval ** ); + ber_bvarray_free_x( *bvp, ber->ber_memctx ); + *bvp = NULL; break; case 'n': /* null */ @@ -724,10 +975,10 @@ ber_scanf ( BerElement *ber, /* format should be good */ assert( 0 ); } - } + } - va_end( ap ); + va_end( ap ); } - return( rc ); + return rc; }