X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=libraries%2Fliblber%2Fio.c;h=b54f828cdc88b93b4ffefd2fff060526280cc84e;hb=729fe5669895890b25ddbbfeec4deb545b0e1df3;hp=897378399b48f9683be573b89286e82ee9d6a7f6;hpb=4e5ed2dffc534d0ff0200a03e856838753fa74af;p=openldap diff --git a/libraries/liblber/io.c b/libraries/liblber/io.c index 897378399b..b54f828cdc 100644 --- a/libraries/liblber/io.c +++ b/libraries/liblber/io.c @@ -1,6 +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 @@ -18,7 +19,8 @@ #include "portable.h" #include -#include + +#include #include #include @@ -32,24 +34,14 @@ #include "lber-int.h" -static long BerRead LDAP_P(( Sockbuf *sb, char *buf, long len )); -static int ber_realloc LDAP_P(( BerElement *ber, unsigned long 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 long -BerRead( Sockbuf *sb, char *buf, long len ) +static ber_slen_t +BerRead( + Sockbuf *sb, + unsigned char *buf, + ber_len_t len ) { - int c; - long nread = 0; + ber_slen_t c; + ber_slen_t nread = 0; assert( sb != NULL ); assert( buf != NULL ); @@ -57,12 +49,12 @@ BerRead( Sockbuf *sb, char *buf, long len ) 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; } @@ -70,96 +62,113 @@ BerRead( Sockbuf *sb, char *buf, long len ) return( nread ); } -long -ber_read( BerElement *ber, char *buf, unsigned long len ) +ber_slen_t +ber_read( + BerElement *ber, + char *buf, + ber_len_t len ) { - unsigned long actuallen, nleft; + ber_len_t actuallen, nleft; 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, (size_t)actuallen ); + AC_MEMCPY( buf, ber->ber_ptr, actuallen ); ber->ber_ptr += actuallen; - return( (long)actuallen ); + return( (ber_slen_t) actuallen ); } -long +ber_slen_t ber_write( BerElement *ber, LDAP_CONST char *buf, - unsigned long len, + ber_len_t len, int nosos ) { 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( len ); + return( (ber_slen_t) len ); + } else { if ( ber->ber_sos->sos_ptr + len > ber->ber_end ) { 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( len ); + return( (ber_slen_t) len ); } } -static int -ber_realloc( BerElement *ber, unsigned long len ) +int +ber_realloc( BerElement *ber, ber_len_t len ) { - unsigned long 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( oldbuf, total ); + if ( ber->ber_buf == NULL ) { - if ( (ber->ber_buf = (char *) malloc( (size_t)total )) == NULL ) - return( -1 ); - } else if ( (ber->ber_buf = (char *) realloc( ber->ber_buf, - (size_t)total )) == NULL ) + ber->ber_buf = oldbuf; return( -1 ); + } ber->ber_end = ber->ber_buf + total; /* * 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. */ if ( ber->ber_buf != oldbuf ) { ber->ber_ptr = ber->ber_buf + (ber->ber_ptr - oldbuf); - for ( s = ber->ber_sos; s != NULLSEQORSET; s = s->sos_next ) { + for ( s = ber->ber_sos; s != NULL; s = s->sos_next ) { off = s->sos_first - oldbuf; s->sos_first = ber->ber_buf + off; @@ -174,26 +183,44 @@ ber_realloc( BerElement *ber, unsigned long len ) void ber_free( BerElement *ber, int freebuf ) { +#ifdef LDAP_MEMORY_DEBUG assert( ber != NULL ); - assert( BER_VALID( ber ) ); +#endif + + if( ber == NULL ) { + return; + } + + assert( LBER_VALID( ber ) ); + + if ( freebuf ) { + Seqorset *s, *next; + LBER_FREE( ber->ber_buf ); + + for( s = ber->ber_sos ; s != NULL ; s = next ) { + next = s->sos_next; + LBER_FREE( s ); + } + } - if ( freebuf && ber->ber_buf != NULL ) - free( ber->ber_buf ); ber->ber_buf = NULL; + ber->ber_sos = NULL; ber->ber_valid = LBER_UNINITIALIZED; - free( (char *) ber ); + + LBER_FREE( (char *) ber ); } int ber_flush( Sockbuf *sb, BerElement *ber, int freeit ) { - long nwritten, towrite, rc; + ber_len_t nwritten, towrite; + ber_slen_t rc; 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; @@ -201,26 +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 !defined(MACOS) && !defined(DOS) - 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 ) { - 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; } @@ -240,47 +266,50 @@ ber_alloc_t( int options ) { BerElement *ber; - ber = (BerElement *) calloc( 1, sizeof(BerElement) ); + ber_int_options.lbo_valid = LBER_INITIALIZED; + + ber = (BerElement *) LBER_CALLOC( 1, sizeof(BerElement) ); - if ( ber == NULLBER ) - return( NULLBER ); + 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 * -ber_dup( LDAP_CONST BerElement *ber ) +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 ); } @@ -291,13 +320,15 @@ ber_init_w_nullc( BerElement *ber, int options ) { assert( ber != NULL ); + ber_int_options.lbo_valid = LBER_INITIALIZED; + (void) memset( (char *)ber, '\0', sizeof( BerElement )); ber->ber_valid = LBER_VALID_BERELEMENT; ber->ber_tag = LBER_DEFAULT; ber->ber_options = (char) options; ber->ber_debug = ber_int_debug; - assert( BER_VALID( ber ) ); + assert( LBER_VALID( ber ) ); } /* New C-API ber_init() */ @@ -311,27 +342,30 @@ ber_init( struct berval *bv ) assert( bv != NULL ); + ber_int_options.lbo_valid = LBER_INITIALIZED; + if ( bv == NULL ) { return NULL; } ber = ber_alloc_t( 0 ); - if( ber == NULLBER ) { + if( ber == NULL ) { /* allocation failed */ - return ( NULL ); + return NULL; } /* copy the data */ - if ( ( (unsigned int) 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 */ @@ -340,19 +374,22 @@ ber_init( struct berval *bv ) ** the returned berval. */ int ber_flatten( - LDAP_CONST BerElement *ber, + BerElement *ber, struct berval **bvPtr) { struct berval *bv; assert( bvPtr != NULL ); + ber_int_options.lbo_valid = LBER_INITIALIZED; + if(bvPtr == NULL) { - return( -1 ); + return -1; } - if ( (bv = malloc( sizeof(struct berval))) == NULL ) { - return( -1 ); + bv = LBER_MALLOC( sizeof(struct berval) ); + if ( bv == NULL ) { + return -1; } if ( ber == NULL ) { @@ -362,31 +399,33 @@ int ber_flatten( } else { /* copy the berval */ - ptrdiff_t len = ber->ber_ptr - ber->ber_buf; + ber_len_t len = ber_pvt_ber_write( ber ); - if ( (bv->bv_val = (char *) 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, (size_t)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; } @@ -394,64 +433,31 @@ ber_reset( BerElement *ber, int was_writing ) ber->ber_rwptr = NULL; } -#if 0 -/* return the tag - LBER_DEFAULT returned means trouble */ -static unsigned long -get_tag( Sockbuf *sb ) -{ - unsigned char xbyte; - unsigned long 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( (unsigned long) xbyte ); - - tagp = (char *) &tag; - tagp[0] = xbyte; - for ( i = 1; i < sizeof(long); 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(long) ) - return( LBER_DEFAULT ); - - /* want leading, not trailing 0's */ - return( tag >> (sizeof(long) - 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. */ -unsigned long -ber_get_next( Sockbuf *sb, unsigned long *len, BerElement *ber ) +ber_tag_t +ber_get_next( + Sockbuf *sb, + ber_len_t *len, + BerElement *ber ) { assert( sb != NULL ); assert( len != NULL ); 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. @@ -474,15 +480,18 @@ ber_get_next( Sockbuf *sb, unsigned long *len, BerElement *ber ) ber->ber_tag = 0; } -#define PTR_IN_VAR( ptr, var )\ -(((ptr)>=(char *) &(var)) && ((ptr)< (char *) &(var)+sizeof(var))) +#undef PTR_IN_VAR +#define PTR_IN_VAR( ptr, var ) \ + (((ptr)>=(char *) &(var)) && ((ptr)< (char *) &(var)+sizeof(var))) 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; @@ -491,82 +500,101 @@ ber_get_next( Sockbuf *sb, unsigned long *len, BerElement *ber ) } 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); ber->ber_rwptr = (char *) &ber->ber_usertag; goto get_lenbyte; } - } while (PTR_IN_VAR(ber->ber_rwptr,ber->ber_tag)); + } while( PTR_IN_VAR(ber->ber_rwptr, ber->ber_tag )); + errno = ERANGE; /* this is a serious error. */ return LBER_DEFAULT; } + 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) || ((unsigned) len>sizeof( ber->ber_len ) ) ) { + 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; } } + if (PTR_IN_VAR(ber->ber_rwptr, ber->ber_len)) { - int res; - int to_go; + unsigned char netlen[sizeof(ber_len_t)]; + + ber_slen_t res; + ber_slen_t to_go; to_go = (char *) &ber->ber_len + sizeof( ber->ber_len ) - ber->ber_rwptr; assert( to_go > 0 ); - res = ber_pvt_sb_read( sb, ber->ber_rwptr, to_go ); - if (res <=0) + + 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 = AC_NTOHL( ber->ber_len ); - 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; + ber->ber_buf = (char *) LBER_MALLOC( ber->ber_len ); + if (ber->ber_buf==NULL) { return LBER_DEFAULT; } - ber->ber_buf = (char *) malloc( ber->ber_len ); - 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; } + if ((ber->ber_rwptr>=ber->ber_buf) && (ber->ber_rwptrber_end)) { - int res; - int to_go; + ber_slen_t res; + ber_slen_t to_go; 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; @@ -583,28 +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; } - -void ber_clear( BerElement *ber, int freebuf ) -{ - assert( ber != NULL ); - assert( BER_VALID( ber ) ); - - if ((freebuf) && (ber->ber_buf)) - free( ber->ber_buf ); - ber->ber_buf = NULL; - ber->ber_rwptr = NULL; - ber->ber_end = NULL; - - ber->ber_valid = LBER_UNINITIALIZED; -} -