]> git.sur5r.net Git - openldap/blobdiff - libraries/liblber/decode.c
happy new year
[openldap] / libraries / liblber / decode.c
index accbc3a44f5307377cee828dc5f0073f72a53fd6..7e1343f9267e94dfc8e4a96c010bc6d3c6b34dbd 100644 (file)
@@ -1,10 +1,19 @@
 /* decode.c - ber input decoding routines */
-/*
- * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
- * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
+/* $OpenLDAP$ */
+/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
+ *
+ * Copyright 1998-2007 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
+ * <http://www.OpenLDAP.org/license.html>.
  */
-/* 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
  * 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 <stdio.h>
-#include <stdlib.h>
 
-#include <ac/stdarg.h>
+#include <ac/stdlib.h>
 
+#include <ac/stdarg.h>
 #include <ac/string.h>
 #include <ac/socket.h>
 
 #include "lber-int.h"
 
-static int ber_getnint LDAP_P(( BerElement *ber, long *num, int len ));
+static ber_len_t ber_getnint LDAP_P((
+       BerElement *ber,
+       ber_int_t *num,
+       ber_len_t len ));
 
 /* return the tag - LBER_DEFAULT returned means trouble */
-unsigned long
+ber_tag_t
 ber_get_tag( BerElement *ber )
 {
        unsigned char   xbyte;
-       unsigned long   tag;
-       char            *tagp;
+       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;
+       }
 
-       if ( (xbyte & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK )
-               return( (unsigned long) xbyte );
+       if ( ber->ber_ptr == ber->ber_buf ) {
+               tag = *(unsigned char *)ber->ber_ptr;
+       } else {
+               tag = ber->ber_tag;
+       }
+       ber->ber_ptr++;
 
-       tagp = (char *) &tag;
-       tagp[0] = xbyte;
-       for ( i = 1; i < sizeof(long); i++ ) {
-               if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 )
-                       return( LBER_DEFAULT );
+       if ( (tag & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK ) {
+               return tag;
+       }
 
-               tagp[i] = xbyte;
+       for ( i = 1; i < sizeof(ber_tag_t); i++ ) {
+               if ( ber_read( ber, (char *) &xbyte, 1 ) != 1 ) {
+                       return LBER_DEFAULT;
+               }
 
-               if ( ! (xbyte & LBER_MORE_TAG_MASK) )
+               tag <<= 8;
+               tag |= 0x00ffUL & (ber_tag_t) xbyte;
+
+               if ( ! (xbyte & LBER_MORE_TAG_MASK) ) {
                        break;
+               }
        }
 
        /* tag too big! */
-       if ( i == sizeof(long) )
-               return( LBER_DEFAULT );
+       if ( i == sizeof(ber_tag_t) ) {
+               return LBER_DEFAULT;
+       }
 
-       /* want leading, not trailing 0's */
-       return( tag >> (sizeof(long) - i - 1) );
+       return tag;
 }
 
-unsigned long
-ber_skip_tag( BerElement *ber, unsigned long *len )
+ber_tag_t
+ber_skip_tag( BerElement *ber, ber_len_t *len )
 {
-       unsigned long   tag;
+       ber_tag_t       tag;
        unsigned char   lc;
-       unsigned int    noctets;
-       int             diff;
-       unsigned long   netlen;
+       ber_len_t       i, noctets;
+       unsigned char netlen[sizeof(ber_len_t)];
 
        assert( ber != NULL );
        assert( len != NULL );
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        /*
         * Any ber element looks like this: tag length contents.
@@ -90,75 +114,89 @@ ber_skip_tag( BerElement *ber, unsigned long *len )
         *      2) primitive encodings used whenever possible
         */
 
+       *len = 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,
+        * 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 an unsigned long.
+        * greater than what we can hold in a ber_len_t.
         */
 
-       *len = netlen = 0;
-       if ( ber_read( ber, (char *) &lc, 1 ) != 1 )
-               return( LBER_DEFAULT );
+       if ( ber_read( ber, (char *) &lc, 1 ) != 1 ) {
+               return LBER_DEFAULT;
+       }
+
        if ( lc & 0x80U ) {
                noctets = (lc & 0x7fU);
-               if ( (unsigned) noctets > sizeof(unsigned long) )
-                       return( LBER_DEFAULT );
-               diff = sizeof(unsigned long) - noctets;
-               if ( (unsigned) ber_read( ber, (char *) &netlen + diff, noctets )
-                   != noctets )
-                       return( LBER_DEFAULT );
-               *len = AC_NTOHL( netlen );
+
+               if ( noctets > sizeof(ber_len_t) ) {
+                       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];
+               }
+
        } else {
                *len = lc;
        }
 
-       return( tag );
+       /* 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;
 }
 
-unsigned long
+ber_tag_t
 ber_peek_tag(
-       LDAP_CONST BerElement *ber_in, /* not const per c-api-02 */
-       unsigned long *len )
+       BerElement *ber,
+       ber_len_t *len )
 {
-       unsigned long   tag;
-       BerElement *ber;
-
-       assert( ber_in != NULL );
-       assert( BER_VALID( ber_in ) );
-
-       ber = ber_dup( ber_in );
-
-       if( ber == NULL ) {
-               return LBER_ERROR;
-       }
+       /*
+        * This implementation assumes ber_skip_tag() only
+        * modifies ber_ptr field of the BerElement.
+        */
 
-       assert( BER_VALID( ber ) );
+       char *save;
+       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;
 
-       ber_free( ber, 0 );
-       return( tag );
+       return tag;
 }
 
-static int
-ber_getnint( BerElement *ber, long *num, int len )
+static ber_len_t
+ber_getnint(
+       BerElement *ber,
+       ber_int_t *num,
+       ber_len_t len )
 {
-       int     diff, sign, i;
-       long    netnum;
-       char    *p;
+       unsigned char buf[sizeof(ber_int_t)];
 
        assert( ber != NULL );
        assert( num != NULL );
-
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        /*
         * The tag and length have already been stripped off.  We should
@@ -167,252 +205,468 @@ ber_getnint( BerElement *ber, long *num, int len )
         * extend after we read it in.
         */
 
-       if ( (unsigned) len > sizeof(long) )
-               return( -1 );
-
-       netnum = 0;
-       diff = sizeof(long) - len;
-       /* read into the low-order bytes of netnum */
-       if ( ber_read( ber, ((char *) &netnum) + diff, len ) != len )
-               return( -1 );
-
-        /* sign extend if necessary */
-        p = (char *) &netnum;
-        sign = (0x80 & *(p+diff) );
-        if ( sign && ((unsigned) len < sizeof(long)) ) {
-                for ( i = 0; i < diff; i++ ) {
-                        *(p+i) = (unsigned char) 0xff;
+       if ( len > sizeof(ber_int_t) ) {
+               return -1;
+       }
+
+       /* read into the low-order bytes of our buffer */
+       if ( (ber_len_t) ber_read( ber, (char *) buf, len ) != len ) {
+               return -1;
+       }
+
+       if( len ) {
+               /* sign extend if necessary */
+               ber_len_t i;
+               ber_int_t netnum = 0x80 & buf[0] ? -1 : 0;
+
+               /* shift in the bytes */
+               for( i=0 ; i<len; i++ ) {
+                       netnum = (netnum << 8 ) | buf[i];
                }
+
+               *num = netnum;
+
+       } else {
+               *num = 0;
        }
-       *num = AC_NTOHL( netnum );
+       ber->ber_tag = *(unsigned char *)ber->ber_ptr;
 
-       return( len );
+       return len;
 }
 
-unsigned long
-ber_get_int( BerElement *ber, long *num )
+ber_tag_t
+ber_get_int(
+       BerElement *ber,
+       ber_int_t *num )
 {
-       unsigned long   tag, len;
+       ber_tag_t       tag;
+       ber_len_t       len;
 
        assert( ber != NULL );
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
-       if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
-               return( LBER_DEFAULT );
+       if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) {
+               return LBER_DEFAULT;
+       }
 
-       if ( (unsigned long) ber_getnint( ber, num, (int)len ) != len )
-               return( LBER_DEFAULT );
-       else
-               return( tag );
+       if ( ber_getnint( ber, num, len ) != len ) {
+               return LBER_DEFAULT;
+       }
+       
+       return tag;
+}
+
+ber_tag_t
+ber_get_enum(
+       BerElement *ber,
+       ber_int_t *num )
+{
+       return ber_get_int( ber, num );
 }
 
-unsigned long
-ber_get_stringb( BerElement *ber, char *buf, unsigned long *len )
+ber_tag_t
+ber_get_stringb(
+       BerElement *ber,
+       char *buf,
+       ber_len_t *len )
 {
-       unsigned long   datalen, tag;
-#ifdef STR_TRANSLATION
-       char            *transbuf;
-#endif /* STR_TRANSLATION */
+       ber_len_t       datalen;
+       ber_tag_t       tag;
 
        assert( ber != NULL );
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
-       if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT )
-               return( LBER_DEFAULT );
-       if ( datalen > (*len - 1) )
-               return( LBER_DEFAULT );
+       if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) {
+               return LBER_DEFAULT;
+       }
 
-       if ( (unsigned long) ber_read( ber, buf, datalen ) != datalen )
-               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;
+       }
+       ber->ber_tag = *(unsigned char *)ber->ber_ptr;
 
        buf[datalen] = '\0';
 
-#ifdef STR_TRANSLATION
-       if ( datalen > 0 && ( ber->ber_options & LBER_TRANSLATE_STRINGS ) != 0
-           && ber->ber_decode_translate_proc ) {
-               transbuf = buf;
-               ++datalen;
-               if ( (*(ber->ber_decode_translate_proc))( &transbuf, &datalen,
-                   0 ) != 0 ) {
-                       return( LBER_DEFAULT );
-               }
-               if ( datalen > *len ) {
-                       free( transbuf );
-                       return( LBER_DEFAULT );
+       *len = datalen;
+       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 {
+       enum bgbvc choice;
+       BerElement *ber;
+       int alloc;
+       ber_len_t siz;
+       ber_len_t off;
+       union {
+               char ***c;
+               BerVarray *ba;
+               struct berval ***bv;
+       } res;
+} bgbvr;
+
+static ber_tag_t
+ber_get_stringbvl( bgbvr *b, ber_len_t *rlen )
+{
+       int i = 0, n;
+       ber_tag_t tag;
+       ber_len_t len;
+       char *last, *orig;
+       struct berval bv, *bvp = NULL;
+
+       /* For rewinding, just like ber_peek_tag() */
+       orig = b->ber->ber_ptr;
+       tag = b->ber->ber_tag;
+
+       if ( ber_first_element( b->ber, &len, &last ) != LBER_DEFAULT ) {
+               for ( ; b->ber->ber_ptr < last; i++ ) {
+                       if (ber_skip_tag( b->ber, &len ) == LBER_DEFAULT) break;
+                       b->ber->ber_ptr += len;
+                       b->ber->ber_tag = *(unsigned char *)b->ber->ber_ptr;
                }
-               SAFEMEMCPY( buf, transbuf, datalen );
-               free( transbuf );
-               --datalen;
        }
-#endif /* STR_TRANSLATION */
 
-       *len = datalen;
-       return( tag );
+       if ( rlen ) *rlen = i;
+
+       if ( i == 0 ) {
+               *b->res.c = NULL;
+               return 0;
+       }
+
+       n = i;
+
+       /* Allocate the result vector */
+       switch (b->choice) {
+       case ChArray:
+               *b->res.c = ber_memalloc_x( (n+1)*sizeof( char * ),
+                       b->ber->ber_memctx);
+               if ( *b->res.c == NULL ) return LBER_DEFAULT;
+               (*b->res.c)[n] = NULL;
+               break;
+       case BvArray:
+               *b->res.ba = ber_memalloc_x( (n+1)*sizeof( struct berval ),
+                       b->ber->ber_memctx);
+               if ( *b->res.ba == NULL ) return LBER_DEFAULT;
+               (*b->res.ba)[n].bv_val = NULL;
+               break;
+       case BvVec:
+               *b->res.bv = ber_memalloc_x( (n+1)*sizeof( struct berval *),
+                       b->ber->ber_memctx);
+               if ( *b->res.bv == NULL ) return LBER_DEFAULT;
+               (*b->res.bv)[n] = NULL;
+               break;
+       case BvOff:
+               *b->res.ba = ber_memalloc_x( (n+1) * b->siz, b->ber->ber_memctx );
+               if ( *b->res.ba == NULL ) return LBER_DEFAULT;
+               ((struct berval *)((long)(*b->res.ba) + n*b->siz +
+                       b->off))->bv_val = NULL;
+               break;
+       }
+       b->ber->ber_ptr = orig;
+       b->ber->ber_tag = tag;
+       ber_skip_tag( b->ber, &len );
+       
+       for (n=0; n<i; n++)
+       {
+               tag = ber_next_element( b->ber, &len, last );
+               if ( ber_get_stringbv( b->ber, &bv, b->alloc ) == LBER_DEFAULT ) {
+                       goto nomem;
+               }
+
+               /* store my result */
+               switch (b->choice) {
+               case ChArray:
+                       (*b->res.c)[n] = bv.bv_val;
+                       break;
+               case BvArray:
+                       (*b->res.ba)[n] = bv;
+                       break;
+               case BvVec:
+                       bvp = ber_memalloc_x( sizeof( struct berval ), b->ber->ber_memctx);
+                       if ( !bvp ) {
+                               LBER_FREE(bv.bv_val);
+                               goto nomem;
+                       }
+                       (*b->res.bv)[n] = bvp;
+                       *bvp = bv;
+                       break;
+               case BvOff:
+                       *(BerVarray)((long)(*b->res.ba)+n*b->siz+b->off) = bv;
+                       break;
+               }
+       }
+       return tag;
+
+nomem:
+       if (b->alloc || b->choice == BvVec) {
+               for (--n; n>=0; n--) {
+                       switch(b->choice) {
+                       case ChArray:
+                               LBER_FREE((*b->res.c)[n]);
+                               break;
+                       case BvArray:
+                               LBER_FREE((*b->res.ba)[n].bv_val);
+                               break;
+                       case BvVec:
+                               LBER_FREE((*b->res.bv)[n]->bv_val);
+                               LBER_FREE((*b->res.bv)[n]);
+                               break;
+                       default:
+                               break;
+                       }
+               }
+       }
+       LBER_FREE(*b->res.c);
+       *b->res.c = NULL;
+       return LBER_DEFAULT;
 }
 
-unsigned long
-ber_get_stringa( BerElement *ber, char **buf )
+ber_tag_t
+ber_get_stringbv( BerElement *ber, struct berval *bv, int alloc )
 {
-       unsigned long   datalen, tag;
+       ber_tag_t       tag;
 
        assert( ber != NULL );
-       assert( buf != NULL );
+       assert( bv != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
-       if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) {
-               *buf = NULL;
-               return( LBER_DEFAULT );
+       if ( (tag = ber_skip_tag( ber, &bv->bv_len )) == LBER_DEFAULT ) {
+               bv->bv_val = NULL;
+               return LBER_DEFAULT;
        }
 
-       if ( (*buf = (char *) malloc( (size_t)datalen + 1 )) == NULL )
-               return( LBER_DEFAULT );
+       if ( (ber_len_t) ber_pvt_ber_remaining( ber ) < bv->bv_len ) {
+               return LBER_DEFAULT;
+       }
 
-       if ( (unsigned long) ber_read( ber, *buf, datalen ) != datalen ) {
-               free( *buf );
-               *buf = NULL;
-               return( LBER_DEFAULT );
-       }
-       (*buf)[datalen] = '\0';
-
-#ifdef STR_TRANSLATION
-       if ( datalen > 0 && ( ber->ber_options & LBER_TRANSLATE_STRINGS ) != 0
-           && ber->ber_decode_translate_proc ) {
-               ++datalen;
-               if ( (*(ber->ber_decode_translate_proc))( buf, &datalen, 1 )
-                   != 0 ) {
-                       free( *buf );
-                       *buf = NULL;
-                       return( LBER_DEFAULT );
+       if ( alloc ) {
+               bv->bv_val = (char *) ber_memalloc_x( bv->bv_len + 1,
+                       ber->ber_memctx );
+               if ( bv->bv_val == NULL ) {
+                       return LBER_DEFAULT;
                }
+
+               if ( bv->bv_len > 0 && (ber_len_t) ber_read( ber, bv->bv_val,
+                       bv->bv_len ) != bv->bv_len )
+               {
+                       LBER_FREE( bv->bv_val );
+                       bv->bv_val = NULL;
+                       return LBER_DEFAULT;
+               }
+       } else {
+               bv->bv_val = ber->ber_ptr;
+               ber->ber_ptr += bv->bv_len;
        }
-#endif /* STR_TRANSLATION */
+       ber->ber_tag = *(unsigned char *)ber->ber_ptr;
+       bv->bv_val[bv->bv_len] = '\0';
 
-       return( tag );
+       return tag;
 }
 
-unsigned long
-ber_get_stringal( BerElement *ber, struct berval **bv )
+ber_tag_t
+ber_get_stringbv_null( BerElement *ber, struct berval *bv, int alloc )
 {
-       unsigned long   len, tag;
+       ber_tag_t       tag;
 
        assert( ber != NULL );
        assert( bv != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
-       if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) {
-               *bv = NULL;
-               return( LBER_DEFAULT );
+       if ( (tag = ber_skip_tag( ber, &bv->bv_len )) == LBER_DEFAULT ) {
+               bv->bv_val = NULL;
+               return LBER_DEFAULT;
        }
 
-       if ( (*bv = (struct berval *) malloc( sizeof(struct berval) )) == NULL )
-               return( LBER_DEFAULT );
+       if ( (ber_len_t) ber_pvt_ber_remaining( ber ) < bv->bv_len ) {
+               return LBER_DEFAULT;
+       }
 
-       if ( ((*bv)->bv_val = (char *) malloc( (size_t)len + 1 )) == NULL ) {
-               free( *bv );
-               *bv = NULL;
-               return( LBER_DEFAULT );
+       if ( bv->bv_len == 0 ) {
+               bv->bv_val = NULL;
+               ber->ber_tag = *(unsigned char *)ber->ber_ptr;
+               return tag;
        }
 
-       if ( (unsigned long) ber_read( ber, (*bv)->bv_val, len ) != len ) {
-               ber_bvfree( *bv );
-               *bv = NULL;
-               return( LBER_DEFAULT );
-       }
-       ((*bv)->bv_val)[len] = '\0';
-       (*bv)->bv_len = len;
-
-#ifdef STR_TRANSLATION
-       if ( len > 0 && ( ber->ber_options & LBER_TRANSLATE_STRINGS ) != 0
-           && ber->ber_decode_translate_proc ) {
-               ++len;
-               if ( (*(ber->ber_decode_translate_proc))( &((*bv)->bv_val),
-                   &len, 1 ) != 0 ) {
-                       ber_bvfree( *bv );
-                       *bv = NULL;
-                       return( LBER_DEFAULT );
+       if ( alloc ) {
+               bv->bv_val = (char *) ber_memalloc_x( bv->bv_len + 1,
+                       ber->ber_memctx );
+               if ( bv->bv_val == NULL ) {
+                       return LBER_DEFAULT;
                }
-               (*bv)->bv_len = len - 1;
+
+               if ( bv->bv_len > 0 && (ber_len_t) ber_read( ber, bv->bv_val,
+                       bv->bv_len ) != bv->bv_len )
+               {
+                       LBER_FREE( bv->bv_val );
+                       bv->bv_val = NULL;
+                       return LBER_DEFAULT;
+               }
+       } else {
+               bv->bv_val = ber->ber_ptr;
+               ber->ber_ptr += bv->bv_len;
        }
-#endif /* STR_TRANSLATION */
+       ber->ber_tag = *(unsigned char *)ber->ber_ptr;
+       bv->bv_val[bv->bv_len] = '\0';
 
-       return( tag );
+       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, 1 );
+       *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, 1 );
+       *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;
+       }
+
+       tag = ber_get_stringbv( ber, *bv, 1 );
+       if ( tag == LBER_DEFAULT ) {
+               LBER_FREE( *bv );
+               *bv = NULL;
+       }
+       return tag;
 }
 
-unsigned long
-ber_get_bitstringa( BerElement *ber, char **buf, unsigned long *blen )
+ber_tag_t
+ber_get_bitstringa(
+       BerElement *ber,
+       char **buf,
+       ber_len_t *blen )
 {
-       unsigned long   datalen, tag;
+       ber_len_t       datalen;
+       ber_tag_t       tag;
        unsigned char   unusedbits;
 
        assert( ber != NULL );
        assert( buf != NULL );
        assert( blen != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        if ( (tag = ber_skip_tag( ber, &datalen )) == LBER_DEFAULT ) {
                *buf = NULL;
-               return( LBER_DEFAULT );
+               return LBER_DEFAULT;
        }
        --datalen;
 
-       if ( (*buf = (char *) malloc( (size_t)datalen )) == NULL )
-               return( LBER_DEFAULT );
+       *buf = (char *) ber_memalloc_x( datalen, ber->ber_memctx );
+       if ( *buf == NULL ) {
+               return LBER_DEFAULT;
+       }
 
        if ( ber_read( ber, (char *)&unusedbits, 1 ) != 1 ) {
-               free( buf );
+               LBER_FREE( buf );
                *buf = NULL;
-               return( LBER_DEFAULT );
+               return LBER_DEFAULT;
        }
 
-       if ( (unsigned long) ber_read( ber, *buf, datalen ) != datalen ) {
-               free( buf );
+       if ( (ber_len_t) ber_read( ber, *buf, datalen ) != datalen ) {
+               LBER_FREE( buf );
                *buf = NULL;
-               return( LBER_DEFAULT );
+               return LBER_DEFAULT;
        }
+       ber->ber_tag = *(unsigned char *)ber->ber_ptr;
 
        *blen = datalen * 8 - unusedbits;
-       return( tag );
+       return tag;
 }
 
-unsigned long
+ber_tag_t
 ber_get_null( BerElement *ber )
 {
-       unsigned long   len, tag;
+       ber_len_t       len;
+       ber_tag_t       tag;
 
        assert( ber != NULL );
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
-       if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT )
-               return( LBER_DEFAULT );
+       if ( (tag = ber_skip_tag( ber, &len )) == LBER_DEFAULT ) {
+               return LBER_DEFAULT;
+       }
 
-       if ( len != 0 )
-               return( LBER_DEFAULT );
+       if ( len != 0 ) {
+               return LBER_DEFAULT;
+       }
+       ber->ber_tag = *(unsigned char *)ber->ber_ptr;
 
        return( tag );
 }
 
-unsigned long
-ber_get_boolean( BerElement *ber, int *boolval )
+ber_tag_t
+ber_get_boolean(
+       BerElement *ber,
+       ber_int_t *boolval )
 {
-       long    longbool;
-       int     rc;
+       ber_int_t       longbool;
+       ber_tag_t       rc;
 
        assert( ber != NULL );
        assert( boolval != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        rc = ber_get_int( ber, &longbool );
        *boolval = longbool;
 
-       return( rc );
+       return rc;
 }
 
-unsigned long
-ber_first_element( BerElement *ber, unsigned long *len, char **last )
+ber_tag_t
+ber_first_element(
+       BerElement *ber,
+       ber_len_t *len,
+       char **last )
 {
        assert( ber != NULL );
        assert( len != NULL );
@@ -421,71 +675,60 @@ ber_first_element( BerElement *ber, unsigned long *len, char **last )
        /* 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;
        }
+       ber->ber_tag = *(unsigned char *)ber->ber_ptr;
 
        *last = ber->ber_ptr + *len;
 
        if ( *last == ber->ber_ptr ) {
-               return( LBER_DEFAULT );
+               return LBER_DEFAULT;
        }
 
-       return( ber_peek_tag( ber, len ) );
+       return ber_peek_tag( ber, len );
 }
 
-unsigned long
-ber_next_element( BerElement *ber, unsigned long *len, char *last )
+ber_tag_t
+ber_next_element(
+       BerElement *ber,
+       ber_len_t *len,
+       LDAP_CONST char *last )
 {
        assert( ber != NULL );
        assert( len != NULL );
        assert( last != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_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 */
-unsigned long
-ber_scanf
-#if HAVE_STDARG
-       ( BerElement *ber,
+ber_tag_t
+ber_scanf ( BerElement *ber,
        LDAP_CONST char *fmt,
        ... )
-#else
-       ( va_alist )
-va_dcl
-#endif
 {
        va_list         ap;
-#ifndef HAVE_STDARG
-       BerElement      *ber;
-       char            *fmt;
-#endif
        LDAP_CONST char         *fmt_reset;
-       char            *last;
-       char            *s, **ss, ***sss;
-       struct berval   ***bv, **bvp, *bval;
-       int             *i, j;
-       long            *l;
-       unsigned long   rc, tag, len;
-
-#ifdef HAVE_STDARG
+       char            *s, **ss;
+       struct berval   **bvp, *bval;
+       ber_int_t       *i;
+       ber_len_t       *l;
+       ber_tag_t       *t;
+       ber_tag_t       rc;
+       ber_len_t       len;
+
        va_start( ap, fmt );
-#else
-       va_start( ap );
-       ber = va_arg( ap, BerElement * );
-       fmt = va_arg( ap, char * );
-#endif
 
        assert( ber != NULL );
        assert( fmt != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        fmt_reset = fmt;
 
@@ -512,36 +755,63 @@ va_dcl
                        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, int * );
+                       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 */
-                       l = va_arg( ap, long * );
-                       rc = ber_get_int( ber, l );
+                       i = va_arg( ap, ber_int_t * );
+                       rc = ber_get_int( ber, i );
                        break;
 
                case 'l':       /* length of next item */
-                       l = va_arg( ap, long * );
+                       l = va_arg( ap, ber_len_t * );
                        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 * );
-                       l = va_arg( ap, long * );
-                       rc = ber_get_stringb( ber, s, l );
+               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 };
+                       cookie.ber = ber;
+                       cookie.res.ba = va_arg( ap, struct berval ** );
+                       cookie.alloc = 0;
+                       l = va_arg( ap, ber_len_t * );
+                       cookie.siz = *l;
+                       cookie.off = va_arg( ap, ber_len_t );
+                       rc = ber_get_stringbvl( &cookie, l );
+                       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, 1 );
                        break;
 
                case 'O':       /* octet string - allocate & include length */
@@ -549,75 +819,63 @@ va_dcl
                        rc = ber_get_stringal( ber, bvp );
                        break;
 
-               case 'B':       /* bit string - allocate storage as needed */
-                       ss = va_arg( ap, char ** );
-                       l = va_arg( ap, long * ); /* 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 */
-                       l = va_arg( ap, long * );
-                       *l = rc = ber_peek_tag( ber, &len );
+                       t = va_arg( ap, ber_tag_t * );
+                       *t = rc = ber_peek_tag( ber, &len );
                        break;
 
                case 'T':       /* skip tag of next item */
-                       l = va_arg( ap, long * );
-                       *l = rc = ber_skip_tag( ber, &len );
+                       t = va_arg( ap, ber_tag_t * );
+                       *t = rc = ber_skip_tag( ber, &len );
                        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 ) )
-                       {
-                               if ( *sss == NULL ) {
-                                       *sss = (char **) malloc(
-                                           2 * sizeof(char *) );
-                               } else {
-                                       *sss = (char **) realloc( *sss,
-                                           (j + 2) * sizeof(char *) );
-                               }
-                               rc = ber_get_stringa( ber, &((*sss)[j]) );
-                               j++;
-                       }
-                       if ( j > 0 )
-                               (*sss)[j] = NULL;
+               {
+                       bgbvr cookie = { ChArray };
+                       cookie.ber = ber;
+                       cookie.res.c = va_arg( ap, char *** );
+                       cookie.alloc = 1;
+                       rc = ber_get_stringbvl( &cookie, NULL );
                        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 ) )
-                       {
-                               if ( *bv == NULL ) {
-                                       *bv = (struct berval **) malloc(
-                                           2 * sizeof(struct berval *) );
-                               } else {
-                                       *bv = (struct berval **) realloc( *bv,
-                                           (j + 2) * sizeof(struct berval *) );
-                               }
-                               rc = ber_get_stringal( ber, &((*bv)[j]) );
-                               j++;
-                       }
-                       if ( j > 0 )
-                               (*bv)[j] = NULL;
+               {
+                       bgbvr cookie = { BvVec };
+                       cookie.ber = ber;
+                       cookie.res.bv = va_arg( ap, struct berval *** );
+                       cookie.alloc = 1;
+                       rc = ber_get_stringbvl( &cookie, NULL );
                        break;
+               }
+
+               case 'W':       /* bvarray */
+               {
+                       bgbvr cookie = { BvArray };
+                       cookie.ber = ber;
+                       cookie.res.ba = va_arg( ap, struct berval ** );
+                       cookie.alloc = 1;
+                       rc = ber_get_stringbvl( &cookie, NULL );
+                       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' )
+                       if ( *(fmt + 1) != 'v' && *(fmt + 1) != 'V'
+                               && *(fmt + 1) != 'W' && *(fmt + 1) != 'M' )
                                rc = ber_skip_tag( ber, &len );
                        break;
 
@@ -636,21 +894,14 @@ va_dcl
        }
 
        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.
-            */
-#ifdef HAVE_STDARG
-           va_start( ap, fmt );
-#else
-           va_start( ap );
-           (void) va_arg( ap, BerElement * );
-           (void) va_arg( ap, char * );
-#endif
-
-           for ( ; fmt_reset < fmt; fmt_reset++ ) {
+               /*
+                * 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++ ) {
                switch ( *fmt_reset ) {
                case '!': { /* Hook */
                                BERDecodeCallback *f;
@@ -663,34 +914,28 @@ va_dcl
                        } break;
 
                case 'a':       /* octet string - allocate storage as needed */
+               case 'A':
                        ss = va_arg( ap, char ** );
                        if ( *ss ) {
-                               free( *ss );
+                               LBER_FREE( *ss );
                                *ss = NULL;
                        }
                        break;
 
                case 'b':       /* boolean */
+               case 'e':       /* enumerated */
+               case 'i':       /* int */
                        (void) va_arg( ap, int * );
                        break;
 
-               case 's':       /* octet string - in a buffer */
-                       (void) va_arg( ap, char * );
-                       (void) va_arg( ap, long * );
-                       break;
-
-               case 'e':       /* enumerated */
-               case 'i':       /* int */
                case 'l':       /* length of next item */
-               case 't':       /* tag of next item */
-               case 'T':       /* skip tag of next item */
-                       (void) va_arg( ap, long * );
+                       (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 ) {
-                               free( bval->bv_val );
+                               LBER_FREE( bval->bv_val );
                                bval->bv_val = NULL;
                        }
                        bval->bv_len = 0;
@@ -704,36 +949,31 @@ va_dcl
                        }
                        break;
 
+               case 's':       /* octet string - in a buffer */
+                       (void) va_arg( ap, char * );
+                       (void) va_arg( ap, ber_len_t * );
+                       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 ) {
-                               free( *ss );
+                               LBER_FREE( *ss );
                                *ss = NULL;
                        }
-                       *(va_arg( ap, long * )) = 0; /* for length, in bits */
+                       *(va_arg( ap, ber_len_t * )) = 0; /* for length, in bits */
                        break;
 
+               case 'm':       /* berval in-place */
+               case 'M':       /* BVoff array in-place */
+               case 'n':       /* null */
                case 'v':       /* sequence of strings */
-                       sss = va_arg( ap, char *** );
-                       if ( *sss ) {
-                               for (j = 0;  (*sss)[j];  j++) {
-                                       free( (*sss)[j] );
-                                       (*sss)[j] = NULL;
-                               }
-                               free( *sss );
-                               *sss = NULL;
-                       }
-                       break;
-
                case 'V':       /* sequence of strings + lengths */
-                       bv = va_arg( ap, struct berval *** );
-                       if ( *bv ) {
-                               ber_bvecfree( *bv );
-                               *bv = NULL;
-                       }
-                       break;
-
-               case 'n':       /* null */
+               case 'W':       /* BerVarray */
                case 'x':       /* skip the next element - whatever it is */
                case '{':       /* begin sequence */
                case '[':       /* begin set */
@@ -745,81 +985,10 @@ va_dcl
                        /* format should be good */
                        assert( 0 );
                }
-           }
-
-           va_end( ap );
-       }
-
-       return( rc );
-}
-
-void
-ber_bvfree( struct berval *bv )
-{
-       assert(bv != NULL);                     /* bv damn better point to something */
-
-       if ( bv->bv_val != NULL )
-               free( bv->bv_val );
-       free( (char *) bv );
-}
-
-void
-ber_bvecfree( struct berval **bv )
-{
-       int     i;
-
-       assert(bv != NULL);                     /* bv damn better point to something */
-
-       for ( i = 0; bv[i] != NULL; i++ )
-               ber_bvfree( bv[i] );
-       free( (char *) bv );
-}
-
-struct berval *
-ber_bvdup(
-       LDAP_CONST struct berval *bv )
-{
-       struct berval   *new;
-
-       assert( bv != NULL );
-
-       if( bv == NULL ) {
-               return NULL;
-       }
-
-       if ( (new = (struct berval *) malloc( sizeof(struct berval) ))
-           == NULL ) {
-               return( NULL );
-       }
-
-       if ( bv->bv_val == NULL ) {
-               new->bv_val = NULL;
-               new->bv_len = 0;
-               return ( new );
-       }
+               }
 
-       if ( (new->bv_val = (char *) malloc( bv->bv_len + 1 )) == NULL ) {
-               free( new );
-               return( NULL );
+               va_end( ap );
        }
 
-       SAFEMEMCPY( new->bv_val, bv->bv_val, (size_t) bv->bv_len );
-       new->bv_val[bv->bv_len] = '\0';
-       new->bv_len = bv->bv_len;
-
-       return( new );
-}
-
-
-#ifdef STR_TRANSLATION
-void
-ber_set_string_translators( BerElement *ber, BERTranslateProc encode_proc,
-       BERTranslateProc decode_proc )
-{
-       assert( ber != NULL );
-       assert( BER_VALID( ber ) );
-
-    ber->ber_encode_translate_proc = encode_proc;
-    ber->ber_decode_translate_proc = decode_proc;
+       return rc;
 }
-#endif /* STR_TRANSLATION */