]> git.sur5r.net Git - openldap/blobdiff - libraries/liblber/io.c
Made LDAP_UTF8_CHARLEN/ldap_utf8_charlen table-driven.
[openldap] / libraries / liblber / io.c
index 96b02ad861e61eada4229ec1a724833b0e95cc16..b54f828cdc88b93b4ffefd2fff060526280cc84e 100644 (file)
@@ -1,7 +1,7 @@
 /* io.c - ber general i/o routines */
 /* $OpenLDAP$ */
 /*
- * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
+ * Copyright 1998-2000 The OpenLDAP Foundation, All Rights Reserved.
  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
  */
 /* Portions
 
 #include "lber-int.h"
 
-static ber_slen_t BerRead LDAP_P((
-       Sockbuf *sb,
-       char *buf,
-       ber_len_t len ));
-
-static int ber_realloc LDAP_P((
-       BerElement *ber,
-       ber_len_t len ));
-
-#define EXBUFSIZ       1024
-
-/* probably far too large... */
-#define MAX_BERBUFSIZE (128*1024)
-
-#if defined( DOS ) && !defined( _WIN32 ) && (MAX_BERBUFSIZE > 65535)
-# undef MAX_BERBUFSIZE
-# define MAX_BERBUFSIZE 65535
-#endif
-
 static ber_slen_t
 BerRead(
        Sockbuf *sb,
-       char *buf,
+       unsigned char *buf,
        ber_len_t len )
 {
        ber_slen_t      c;
@@ -68,12 +49,12 @@ BerRead(
        assert( SOCKBUF_VALID( sb ) );
 
        while ( len > 0 ) {
-               if ( (c = ber_pvt_sb_read( sb, buf, len )) <= 0 ) {
+               if ( (c = ber_int_sb_read( sb, buf, len )) <= 0 ) {
                        if ( nread > 0 )
                                break;
                        return( c );
                }
-               buf+= c;
+               buf+=c;
                nread+=c;
                len-=c;
        }
@@ -92,12 +73,12 @@ ber_read(
        assert( ber != NULL );
        assert( buf != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
-       nleft = ber->ber_end - ber->ber_ptr;
+       nleft = ber_pvt_ber_remaining( ber );
        actuallen = nleft < len ? nleft : len;
 
-       SAFEMEMCPY( buf, ber->ber_ptr, actuallen );
+       AC_MEMCPY( buf, ber->ber_ptr, actuallen );
 
        ber->ber_ptr += actuallen;
 
@@ -114,14 +95,14 @@ ber_write(
        assert( ber != NULL );
        assert( buf != NULL );
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        if ( nosos || ber->ber_sos == NULL ) {
                if ( ber->ber_ptr + len > ber->ber_end ) {
                        if ( ber_realloc( ber, len ) != 0 )
                                return( -1 );
                }
-               SAFEMEMCPY( ber->ber_ptr, buf, (size_t)len );
+               AC_MEMCPY( ber->ber_ptr, buf, (size_t)len );
                ber->ber_ptr += len;
                return( (ber_slen_t) len );
 
@@ -130,33 +111,46 @@ ber_write(
                        if ( ber_realloc( ber, len ) != 0 )
                                return( -1 );
                }
-               SAFEMEMCPY( ber->ber_sos->sos_ptr, buf, (size_t)len );
+               AC_MEMCPY( ber->ber_sos->sos_ptr, buf, (size_t)len );
                ber->ber_sos->sos_ptr += len;
                ber->ber_sos->sos_clen += len;
                return( (ber_slen_t) len );
        }
 }
 
-static int
+int
 ber_realloc( BerElement *ber, ber_len_t len )
 {
-       ber_len_t       need, have, total;
+       ber_len_t       total;
        Seqorset        *s;
        long            off;
        char            *oldbuf;
 
        assert( ber != NULL );
        assert( len > 0 );
-
-       assert( BER_VALID( ber ) );
-
-       have = (ber->ber_end - ber->ber_buf) / EXBUFSIZ;
-       need = (len < EXBUFSIZ ? 1 : (len + (EXBUFSIZ - 1)) / EXBUFSIZ);
-       total = have * EXBUFSIZ + need * EXBUFSIZ;
+       assert( LBER_VALID( ber ) );
+
+       total = ber_pvt_ber_total( ber );
+
+#define LBER_EXBUFSIZ  1000 /* a few words less than 2^N for binary buddy */
+#if defined( LBER_EXBUFSIZ ) && LBER_EXBUFSIZ > 0
+# ifndef notdef
+       /* don't realloc by small amounts */
+       total += len < LBER_EXBUFSIZ ? LBER_EXBUFSIZ : len;
+# else
+       {       /* not sure what value this adds */
+               ber_len_t have = (total + (LBER_EXBUFSIZE - 1)) / LBER_EXBUFSIZ;
+               ber_len_t need = (len + (LBER_EXBUFSIZ - 1)) / LBER_EXBUFSIZ;
+               total = ( have + need ) * LBER_EXBUFSIZ;
+       }
+# endif
+#else
+       total += len;   /* realloc just what's needed */
+#endif
 
        oldbuf = ber->ber_buf;
 
-       ber->ber_buf = (char *) LBER_REALLOC( ber->ber_buf, total );
+       ber->ber_buf = (char *) LBER_REALLOC( oldbuf, total );
        
        if ( ber->ber_buf == NULL ) {
                ber->ber_buf = oldbuf;
@@ -167,7 +161,7 @@ ber_realloc( BerElement *ber, ber_len_t len )
 
        /*
         * If the stinking thing was moved, we need to go through and
-        * reset all the sos and ber pointers.  Offsets would've been
+        * reset all the sos and ber pointers.  Offsets would've been
         * a better idea... oh well.
         */
 
@@ -197,7 +191,7 @@ ber_free( BerElement *ber, int freebuf )
                return;
        }
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        if ( freebuf ) {
                Seqorset *s, *next;
@@ -225,8 +219,8 @@ ber_flush( Sockbuf *sb, BerElement *ber, int freeit )
        assert( sb != NULL );
        assert( ber != NULL );
 
-       assert( SOCKBUF_VALID( ber ) );
-       assert( BER_VALID( ber ) );
+       assert( SOCKBUF_VALID( sb ) );
+       assert( LBER_VALID( ber ) );
 
        if ( ber->ber_rwptr == NULL ) {
                ber->ber_rwptr = ber->ber_buf;
@@ -234,28 +228,25 @@ ber_flush( Sockbuf *sb, BerElement *ber, int freeit )
        towrite = ber->ber_ptr - ber->ber_rwptr;
 
        if ( sb->sb_debug ) {
+#ifdef NEW_LOGGING
+               LDAP_LOG(( "liblber", LDAP_LEVEL_DETAIL1,
+                          "ber_flush: %ld bytes to sd %ld%s\n",
+                          towrite, (long)sb->sb_fd,
+                          ber->ber_rwptr != ber->ber_buf ? " (re-flush)" : "" ));
+               BER_DUMP(( "liblber", LDAP_LEVEL_DETAIL2, ber, 1 ));
+#else
                ber_log_printf( LDAP_DEBUG_ANY, sb->sb_debug,
-                       "ber_flush: %ld bytes to sd %ld%s\n", towrite,
-                   (long) sb->sb_sd, ber->ber_rwptr != ber->ber_buf ? " (re-flush)"
-                   : "" );
+                       "ber_flush: %ld bytes to sd %ld%s\n",
+                       towrite, (long) sb->sb_fd,
+                       ber->ber_rwptr != ber->ber_buf ?  " (re-flush)" : "" );
                ber_log_bprint( LDAP_DEBUG_PACKETS, sb->sb_debug,
                        ber->ber_rwptr, towrite );
+#endif
        }
 
-#if HAVE_WRITE
-       if ( sb->sb_options & (LBER_TO_FILE | LBER_TO_FILE_ONLY) ) {
-               rc = write( sb->sb_fd, ber->ber_rwptr, towrite );
-               if ( sb->sb_options & LBER_TO_FILE_ONLY ) {
-                       if ( freeit )
-                               ber_free( ber, 1 );
-                       return( (int)rc );
-               }
-       }
-#endif
-       
        nwritten = 0;
        do {
-               rc = ber_pvt_sb_write( sb, ber->ber_rwptr, towrite );
+               rc = ber_int_sb_write( sb, ber->ber_rwptr, towrite );
                if (rc<=0) {
                        return -1;
                }
@@ -279,28 +270,29 @@ ber_alloc_t( int options )
 
        ber = (BerElement *) LBER_CALLOC( 1, sizeof(BerElement) );
 
-       if ( ber == NULL )
-               return( NULL );
+       if ( ber == NULL ) {
+               return NULL;
+       }
 
        ber->ber_valid = LBER_VALID_BERELEMENT;
        ber->ber_tag = LBER_DEFAULT;
        ber->ber_options = options;
        ber->ber_debug = ber_int_debug;
 
-       assert( BER_VALID( ber ) );
-       return( ber );
+       assert( LBER_VALID( ber ) );
+       return ber;
 }
 
 BerElement *
 ber_alloc( void )      /* deprecated */
 {
-       return( ber_alloc_t( 0 ) );
+       return ber_alloc_t( 0 );
 }
 
 BerElement *
 der_alloc( void )      /* deprecated */
 {
-       return( ber_alloc_t( LBER_USE_DER ) );
+       return ber_alloc_t( LBER_USE_DER );
 }
 
 BerElement *
@@ -309,15 +301,15 @@ ber_dup( BerElement *ber )
        BerElement      *new;
 
        assert( ber != NULL );
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        if ( (new = ber_alloc_t( ber->ber_options )) == NULL ) {
-               return( NULL );
+               return NULL;
        }
 
        *new = *ber;
 
-       assert( BER_VALID( new ) );
+       assert( LBER_VALID( new ) );
        return( new );
 }
 
@@ -336,7 +328,7 @@ ber_init_w_nullc( BerElement *ber, int options )
        ber->ber_options = (char) options;
        ber->ber_debug = ber_int_debug;
 
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 }
 
 /* New C-API ber_init() */
@@ -360,19 +352,20 @@ ber_init( struct berval *bv )
 
        if( ber == NULL ) {
                /* allocation failed */
-               return ( NULL );
+               return NULL;
        }
 
        /* copy the data */
-       if ( ( (ber_len_t) ber_write ( ber, bv->bv_val, bv->bv_len, 0 )) != bv->bv_len ) {
+       if ( ((ber_len_t) ber_write ( ber, bv->bv_val, bv->bv_len, 0 ))
+               != bv->bv_len )
+       {
                /* write failed, so free and return NULL */
                ber_free( ber, 1 );
-               return( NULL );
+               return NULL;
        }
 
        ber_reset( ber, 1 );    /* reset the pointer to the start of the buffer */
-
-       return ( ber );
+       return ber;
 }
 
 /* New C-API ber_flatten routine */
@@ -391,11 +384,12 @@ int ber_flatten(
     ber_int_options.lbo_valid = LBER_INITIALIZED;
 
        if(bvPtr == NULL) {
-               return( -1 );
+               return -1;
        }
 
-       if ( (bv = LBER_MALLOC( sizeof(struct berval))) == NULL ) {
-               return( -1 );
+       bv = LBER_MALLOC( sizeof(struct berval) );
+       if ( bv == NULL ) {
+               return -1;
        }
 
        if ( ber == NULL ) {
@@ -405,31 +399,33 @@ int ber_flatten(
 
        } else {
                /* copy the berval */
-               ber_len_t len = ber->ber_ptr - ber->ber_buf;
+               ber_len_t len = ber_pvt_ber_write( ber );
 
-               if ( (bv->bv_val = (char *) LBER_MALLOC( len + 1 )) == NULL ) {
-                       ber_bvfree( bv );
-                       return( -1 );
+               bv->bv_val = (char *) LBER_MALLOC( len + 1 );
+               if ( bv->bv_val == NULL ) {
+                       LBER_FREE( bv );
+                       return -1;
                }
 
-               SAFEMEMCPY( bv->bv_val, ber->ber_buf, len );
+               AC_MEMCPY( bv->bv_val, ber->ber_buf, len );
                bv->bv_val[len] = '\0';
                bv->bv_len = len;
        }
     
        *bvPtr = bv;
-       return( 0 );
+       return 0;
 }
 
 void
 ber_reset( BerElement *ber, int was_writing )
 {
        assert( ber != NULL );
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
        if ( was_writing ) {
                ber->ber_end = ber->ber_ptr;
                ber->ber_ptr = ber->ber_buf;
+
        } else {
                ber->ber_ptr = ber->ber_end;
        }
@@ -437,49 +433,9 @@ ber_reset( BerElement *ber, int was_writing )
        ber->ber_rwptr = NULL;
 }
 
-#if 0
-/* return the tag - LBER_DEFAULT returned means trouble */
-static ber_tag_t
-get_tag( Sockbuf *sb )
-{
-       unsigned char   xbyte;
-       ber_tag_t       tag;
-       char            *tagp;
-       unsigned int    i;
-
-       assert( sb != NULL );
-       assert( SOCKBUF_VALID( sb ) );
-
-       if ( ber_pvt_sb_read( sb, (char *) &xbyte, 1 ) != 1 )
-               return( LBER_DEFAULT );
-
-       if ( (xbyte & LBER_BIG_TAG_MASK) != LBER_BIG_TAG_MASK )
-               return( (ber_tag_t) xbyte );
-
-       tagp = (char *) &tag;
-       tagp[0] = xbyte;
-       for ( i = 1; i < sizeof(ber_tag_t); i++ ) {
-               if ( ber_pvt_sb_read( sb, (char *) &xbyte, 1 ) != 1 )
-                       return( LBER_DEFAULT );
-
-               tagp[i] = xbyte;
-
-               if ( ! (xbyte & LBER_MORE_TAG_MASK) )
-                       break;
-       }
-
-       /* tag too big! */
-       if ( i == sizeof(ber_tag_t) )
-               return( LBER_DEFAULT );
-
-       /* want leading, not trailing 0's */
-       return( tag >> (sizeof(ber_tag_t) - i - 1) );
-}
-#endif
-
 /*
  * A rewrite of ber_get_next that can safely be called multiple times 
- * for the same packet. It will simply continue were it stopped until
+ * for the same packet. It will simply continue where it stopped until
  * a full packet is read.
  */
 
@@ -494,10 +450,14 @@ ber_get_next(
        assert( ber != NULL );
 
        assert( SOCKBUF_VALID( sb ) );
-       assert( BER_VALID( ber ) );
+       assert( LBER_VALID( ber ) );
 
+#ifdef NEW_LOGGING
+       LDAP_LOG(( "liblber", LDAP_LEVEL_ENTRY, "ber_get_next: enter\n" ));
+#else
        ber_log_printf( LDAP_DEBUG_TRACE, ber->ber_debug,
                "ber_get_next\n" );
+#endif
 
        /*
         * Any ber element looks like this: tag length contents.
@@ -526,10 +486,12 @@ ber_get_next(
        
        if (PTR_IN_VAR(ber->ber_rwptr, ber->ber_tag)) {
                if (ber->ber_rwptr == (char *) &ber->ber_tag) {
-                       if (ber_pvt_sb_read( sb, ber->ber_rwptr, 1)<=0)
+                       if (ber_int_sb_read( sb, ber->ber_rwptr, 1)<=0)
                                return LBER_DEFAULT;
+
                        if ((ber->ber_rwptr[0] & LBER_BIG_TAG_MASK)
-                               != LBER_BIG_TAG_MASK) {
+                               != LBER_BIG_TAG_MASK)
+                       {
                                ber->ber_tag = ber->ber_rwptr[0];
                                ber->ber_rwptr = (char *) &ber->ber_usertag;
                                goto get_lenbyte;
@@ -538,8 +500,10 @@ ber_get_next(
                }
                do {
                        /* reading the tag... */
-                       if (ber_pvt_sb_read( sb, ber->ber_rwptr, 1)<=0)
+                       if (ber_int_sb_read( sb, ber->ber_rwptr, 1)<=0) {
                                return LBER_DEFAULT;
+                       }
+
                        if (! (ber->ber_rwptr[0] & LBER_MORE_TAG_MASK) ) {
                                ber->ber_tag>>=sizeof(ber->ber_tag) -
                                  ((char *) &ber->ber_tag - ber->ber_rwptr);
@@ -547,6 +511,7 @@ ber_get_next(
                                goto get_lenbyte;
                        }
                } while( PTR_IN_VAR(ber->ber_rwptr, ber->ber_tag ));
+
                errno = ERANGE; /* this is a serious error. */
                return LBER_DEFAULT;
        }
@@ -554,17 +519,21 @@ ber_get_next(
 get_lenbyte:
        if (ber->ber_rwptr==(char *) &ber->ber_usertag) {
                unsigned char c;
-               if (ber_pvt_sb_read( sb, (char *) &c, 1)<=0)
+               if (ber_int_sb_read( sb, (char *) &c, 1)<=0) {
                        return LBER_DEFAULT;
+               }
+
                if (c & 0x80U) {
                        int len = c & 0x7fU;
                        if ( (len==0) || ( len>sizeof( ber->ber_len ) ) ) {
                                errno = ERANGE;
                                return LBER_DEFAULT;
                        }
+
                        ber->ber_rwptr = (char *) &ber->ber_len +
                                sizeof(ber->ber_len) - len;
                        ber->ber_len = 0;
+
                } else {
                        ber->ber_len = c;
                        goto fill_buffer;
@@ -579,40 +548,40 @@ get_lenbyte:
                to_go = (char *) &ber->ber_len + sizeof( ber->ber_len ) -
                        ber->ber_rwptr;
                assert( to_go > 0 );
-               res = ber_pvt_sb_read( sb, netlen, to_go );
+
+               res = BerRead( sb, netlen, to_go );
                if (res <= 0) {
                        return LBER_DEFAULT;
                }
                ber->ber_rwptr += res;
 
-               if (res==to_go) {
-                       /* convert length. */
-                       ber->ber_len = 0;
-                       for( to_go = 0; to_go < res ; to_go++ ) {
-                               ber->ber_len <<= 8;
-                               ber->ber_len |= netlen[to_go];
-                       }
-                       goto fill_buffer;
-               } else {
-#if defined( EWOULDBLOCK )
-                       errno = EWOULDBLOCK;
-#elif defined( EAGAIN )
-                       errno = EAGAIN;
-#endif                 
+               /* convert length. */
+               for( to_go = 0; to_go < res ; to_go++ ) {
+                       ber->ber_len <<= 8;
+                       ber->ber_len |= netlen[to_go];
+               }
+
+               if (PTR_IN_VAR(ber->ber_rwptr, ber->ber_len)) {
                        return LBER_DEFAULT;
                }
        }
 
 fill_buffer:   
        /* now fill the buffer. */
+
+       /* make sure length is reasonable */
+       if ( ber->ber_len == 0 ||
+               ( sb->sb_max_incoming && ber->ber_len > sb->sb_max_incoming ))
+       {
+               errno = ERANGE;
+               return LBER_DEFAULT;
+       }
+
        if (ber->ber_buf==NULL) {
-               if (ber->ber_len > MAX_BERBUFSIZE) {
-                       errno = ERANGE;
-                       return LBER_DEFAULT;
-               }
                ber->ber_buf = (char *) LBER_MALLOC( ber->ber_len );
-               if (ber->ber_buf==NULL)
+               if (ber->ber_buf==NULL) {
                        return LBER_DEFAULT;
+               }
                ber->ber_rwptr = ber->ber_buf;
                ber->ber_ptr = ber->ber_buf;
                ber->ber_end = ber->ber_buf + ber->ber_len;
@@ -625,7 +594,7 @@ fill_buffer:
                to_go = ber->ber_end - ber->ber_rwptr;
                assert( to_go > 0 );
                
-               res = ber_pvt_sb_read( sb, ber->ber_rwptr, to_go );
+               res = ber_int_sb_read( sb, ber->ber_rwptr, to_go );
                if (res<=0)
                        return LBER_DEFAULT;
                ber->ber_rwptr+=res;
@@ -642,13 +611,21 @@ fill_buffer:
                ber->ber_rwptr = NULL;
                *len = ber->ber_len;
                if ( ber->ber_debug ) {
+#ifdef NEW_LOGGING
+                       LDAP_LOG(( "liblber", LDAP_LEVEL_DETAIL1,
+                                  "ber_get_next: tag 0x%lx len %ld\n",
+                                  ber->ber_tag, ber->ber_len ));
+                       BER_DUMP(( "liblber", LDAP_LEVEL_DETAIL2, ber, 1 ));
+#else
                        ber_log_printf( LDAP_DEBUG_TRACE, ber->ber_debug,
                                "ber_get_next: tag 0x%lx len %ld contents:\n",
                                ber->ber_tag, ber->ber_len );
                        ber_log_dump( LDAP_DEBUG_BER, ber->ber_debug, ber, 1 );
+#endif
                }
                return (ber->ber_tag);
        }
+
        assert( 0 ); /* ber structure is messed up ?*/
        return LBER_DEFAULT;
 }