]> git.sur5r.net Git - openldap/blobdiff - libraries/liblber/options.c
Happy new year
[openldap] / libraries / liblber / options.c
index c6e976c80ed73e1bbd7a89e6737a1844c56133ac..4ecee6485b60ba336591558a61820f76ad39647f 100644 (file)
@@ -1,48 +1,94 @@
+/* $OpenLDAP$ */
+/*
+ * Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved.
+ * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
+ */
 #include "portable.h"
 
-#include <stdlib.h>
-
+#include <ac/stdlib.h>
+#include <ac/string.h>
+#include <ac/stdarg.h>
 #include "lber-int.h"
 
-int lber_int_debug = 0;
+struct lber_options ber_int_options = {
+       LBER_UNINITIALIZED, 0, 0 };
 
 int
-lber_get_option(
+ber_get_option(
        void    *item,
        int             option,
        void    *outvalue)
 {
-       BerElement *ber;
-       Sockbuf *sb;
+       const BerElement *ber;
+       const Sockbuf *sb;
+
+       ber_int_options.lbo_valid = LBER_INITIALIZED;
 
        if(outvalue == NULL) {
                /* no place to get to */
+               ber_errno = LBER_ERROR_PARAM;
                return LBER_OPT_ERROR;
        }
 
        if(item == NULL) {
                if(option == LBER_OPT_BER_DEBUG) {
-                       * (int *) outvalue = lber_int_debug;
+                       * (int *) outvalue = ber_int_debug;
+                       return LBER_OPT_SUCCESS;
+               } else if(option == LBER_OPT_MEMORY_INUSE) {
+                       /* The memory inuse is a global variable on kernal implementations.
+                        * This means that memory debug is shared by all LDAP processes
+                        * so for this variable to have much meaning, only one LDAP process
+                        * should be running and memory inuse should be initialized to zero
+                        * using the lber_set_option() function during startup.
+                        * The counter is not accurate for multithreaded ldap applications.
+                        */
+#ifdef LDAP_MEMORY_DEBUG
+                       * (int *) outvalue = ber_int_options.lbo_meminuse;
+                       return LBER_OPT_SUCCESS;
+#else
+                       return LBER_OPT_ERROR;
+#endif
+               } else if(option == LBER_OPT_LOG_PRINT_FILE) {
+                       *((FILE**)outvalue) = (FILE*)ber_pvt_err_file;
                        return LBER_OPT_SUCCESS;
                }
 
+               ber_errno = LBER_ERROR_PARAM;
                return LBER_OPT_ERROR;
        }
 
-       ber = (BerElement *) item;
-       sb = (Sockbuf *) item;
+       ber = item;
+       sb = item;
 
        switch(option) {
        case LBER_OPT_BER_OPTIONS:
+               assert( LBER_VALID( ber ) );
                * (int *) outvalue = ber->ber_options;
                return LBER_OPT_SUCCESS;
 
        case LBER_OPT_BER_DEBUG:
+               assert( LBER_VALID( ber ) );
                * (int *) outvalue = ber->ber_debug;
                return LBER_OPT_SUCCESS;
 
+       case LBER_OPT_BER_REMAINING_BYTES:
+               assert( LBER_VALID( ber ) );
+               *((ber_len_t *) outvalue) = ber_pvt_ber_remaining(ber);
+               return LBER_OPT_SUCCESS;
+
+       case LBER_OPT_BER_TOTAL_BYTES:
+               assert( LBER_VALID( ber ) );
+               *((ber_len_t *) outvalue) = ber_pvt_ber_total(ber);
+               return LBER_OPT_SUCCESS;
+
+       case LBER_OPT_BER_BYTES_TO_WRITE:
+               assert( LBER_VALID( ber ) );
+               *((ber_len_t *) outvalue) = ber_pvt_ber_write(ber);
+               return LBER_OPT_SUCCESS;
+
        default:
                /* bad param */
+               ber_errno = LBER_ERROR_PARAM;
                break;
        }
 
@@ -50,42 +96,117 @@ lber_get_option(
 }
 
 int
-lber_set_option(
+ber_set_option(
        void    *item,
        int             option,
-       void    *invalue)
+       LDAP_CONST void *invalue)
 {
        BerElement *ber;
        Sockbuf *sb;
 
+       if( (ber_int_options.lbo_valid == LBER_UNINITIALIZED)
+               && ( ber_int_memory_fns == NULL )
+               && ( option == LBER_OPT_MEMORY_FNS )
+               && ( invalue != NULL ))
+       {
+               const BerMemoryFunctions *f =
+                       (const BerMemoryFunctions *) invalue;
+
+               /* make sure all functions are provided */
+               if(!( f->bmf_malloc && f->bmf_calloc
+                       && f->bmf_realloc && f->bmf_free ))
+               {
+                       ber_errno = LBER_ERROR_PARAM;
+                       return LBER_OPT_ERROR;
+               }
+
+               ber_int_memory_fns = (BerMemoryFunctions *)
+                       (*(f->bmf_malloc))(sizeof(BerMemoryFunctions));
+
+               if ( ber_int_memory_fns == NULL ) {
+                       ber_errno = LBER_ERROR_MEMORY;
+                       return LBER_OPT_ERROR;
+               }
+
+               AC_MEMCPY(ber_int_memory_fns, f, sizeof(BerMemoryFunctions));
+
+               ber_int_options.lbo_valid = LBER_INITIALIZED;
+               return LBER_OPT_SUCCESS;
+       }
+
+       ber_int_options.lbo_valid = LBER_INITIALIZED;
+
        if(invalue == NULL) {
                /* no place to set from */
+               ber_errno = LBER_ERROR_PARAM;
                return LBER_OPT_ERROR;
        }
 
        if(item == NULL) {
                if(option == LBER_OPT_BER_DEBUG) {
-                       lber_int_debug = * (int *) invalue;
+                       ber_int_debug = * (const int *) invalue;
+                       return LBER_OPT_SUCCESS;
+
+               } else if(option == LBER_OPT_LOG_PRINT_FN) {
+                       ber_pvt_log_print = (BER_LOG_PRINT_FN) invalue;
                        return LBER_OPT_SUCCESS;
+               } else if(option == LBER_OPT_LOG_PRINT_FILE) {
+                       ber_pvt_err_file = (void *) invalue;
+                       return LBER_OPT_SUCCESS;
+               } else if(option == LBER_OPT_MEMORY_INUSE) {
+                       /* The memory inuse is a global variable on kernal implementations.
+                        * This means that memory debug is shared by all LDAP processes
+                        * so for this variable to have much meaning, only one LDAP process
+                        * should be running and memory inuse should be initialized to zero
+                        * using the lber_set_option() function during startup.
+                        * The counter is not accurate for multithreaded applications.
+                        */
+#ifdef LDAP_MEMORY_DEBUG
+                       ber_int_options.lbo_meminuse = * (int *) invalue;
+                       return LBER_OPT_SUCCESS;
+#else
+                       return LBER_OPT_ERROR;
+#endif
+               } else if(option == LBER_OPT_LOG_PROC) {
+                       ber_int_log_proc = (BER_LOG_FN)invalue;
                }
 
+               ber_errno = LBER_ERROR_PARAM;
                return LBER_OPT_ERROR;
        }
 
-       ber = (BerElement *) item;
-       sb = (Sockbuf *) item;
+       ber = item;
+       sb = item;
 
        switch(option) {
        case LBER_OPT_BER_OPTIONS:
-               ber->ber_options = * (int *) invalue;
+               assert( LBER_VALID( ber ) );
+               ber->ber_options = * (const int *) invalue;
                return LBER_OPT_SUCCESS;
 
        case LBER_OPT_BER_DEBUG:
-               ber->ber_debug = * (int *) invalue;
+               assert( LBER_VALID( ber ) );
+               ber->ber_debug = * (const int *) invalue;
+               return LBER_OPT_SUCCESS;
+
+       case LBER_OPT_BER_REMAINING_BYTES:
+               assert( LBER_VALID( ber ) );
+               ber->ber_end = &ber->ber_ptr[* (const ber_len_t *) invalue];
+               return LBER_OPT_SUCCESS;
+
+       case LBER_OPT_BER_TOTAL_BYTES:
+               assert( LBER_VALID( ber ) );
+               ber->ber_end = &ber->ber_buf[* (const ber_len_t *) invalue];
+               return LBER_OPT_SUCCESS;
+
+       case LBER_OPT_BER_BYTES_TO_WRITE:
+               assert( LBER_VALID( ber ) );
+               ber->ber_ptr = &ber->ber_buf[* (const ber_len_t *) invalue];
                return LBER_OPT_SUCCESS;
 
        default:
                /* bad param */
+               ber_errno = LBER_ERROR_PARAM;
                break;
        }