]> git.sur5r.net Git - openldap/blobdiff - libraries/liblutil/utils.c
allow to specify a per-database list of attributes that need to be always collected...
[openldap] / libraries / liblutil / utils.c
index 2c84ac4afd8db0cc535bac9fbe40b9fb7c460071..0fdd00027ca4ddae0a42d9f6ca1ce26203b11e1a 100644 (file)
@@ -1,7 +1,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 1998-2005 The OpenLDAP Foundation.
+ * Copyright 1998-2010 The OpenLDAP Foundation.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
 
 #include <stdio.h>
 #include <ac/stdlib.h>
+#include <ac/stdarg.h>
 #include <ac/string.h>
 #include <ac/ctype.h>
 #include <ac/unistd.h>
 #include <ac/time.h>
+#include <ac/errno.h>
 #ifdef HAVE_IO_H
 #include <io.h>
 #endif
 #ifdef HAVE_FCNTL_H
 #include <fcntl.h>
 #endif
+#ifdef _WIN32
+#include <windows.h>
+#endif
 
-#include <lutil.h>
-#include <ldap_defaults.h>
+#include "lutil.h"
+#include "ldap_defaults.h"
+#include "ldap_pvt.h"
+#include "lber_pvt.h"
 
 #ifdef HAVE_EBCDIC
 int _trans_argv = 1;
@@ -70,6 +77,13 @@ char* lutil_progname( const char* name, int argc, char *argv[] )
        LUTIL_SLASHPATH( argv[0] );
        progname = strrchr ( argv[0], *LDAP_DIRSEP );
        progname = progname ? &progname[1] : argv[0];
+#ifdef _WIN32
+       {
+               size_t len = strlen( progname );
+               if ( len > 4 && strcasecmp( &progname[len - 4], ".exe" ) == 0 )
+                       progname[len - 4] = '\0';
+       }
+#endif
        return progname;
 }
 
@@ -207,7 +221,7 @@ int lutil_parsetime( char *atm, struct lutil_tm *tm )
                unsigned i, fracs;
 
                /* Is the stamp reasonably long? */
-               for (i=0; isdigit(atm[i]); i++);
+               for (i=0; isdigit((unsigned char) atm[i]); i++);
                if (i < sizeof("00000101000000")-1)
                        break;
 
@@ -247,14 +261,17 @@ int lutil_parsetime( char *atm, struct lutil_tm *tm )
                if (tm->tm_sec < 0 || tm->tm_sec > 61) break;
 
                /* Fractions of seconds */
-               for (i = 0, fracs = 0;isdigit(*ptr);) {
-                       i*=10; i+= *ptr++ - '0';
-                       fracs++;
-               }
-               tm->tm_usec = i;
-               if (i) {
-                       for (i = fracs; i<6; i++)
-                               tm->tm_usec *= 10;
+               if ( *ptr == '.' ) {
+                       ptr++;
+                       for (i = 0, fracs = 0; isdigit((unsigned char) *ptr); ) {
+                               i*=10; i+= *ptr++ - '0';
+                               fracs++;
+                       }
+                       tm->tm_usec = i;
+                       if (i) {
+                               for (i = fracs; i<6; i++)
+                                       tm->tm_usec *= 10;
+                       }
                }
 
                /* Must be UTC */
@@ -300,6 +317,21 @@ lutil_strncopy(
        return a-1;
 }
 
+/* memcopy is like memcpy except it returns a pointer to the byte past
+ * the end of the result buffer, set to NULL. This allows fast construction
+ * of catenated buffers.  Provided for API consistency with lutil_str*copy().
+ */
+char *
+lutil_memcopy(
+       char *a,
+       const char *b,
+       size_t n
+)
+{
+       AC_MEMCPY(a, b, n);
+       return a + n;
+}
+
 #ifndef HAVE_MKSTEMP
 int mkstemp( char * template )
 {
@@ -311,11 +343,109 @@ int mkstemp( char * template )
 }
 #endif
 
+#ifdef _MSC_VER
+/* Equivalent of MS CRT's _dosmaperr().
+ * @param lastError[in] Result of GetLastError().
+ */
+static errno_t win2errno(DWORD lastError)
+{
+       const struct { 
+               DWORD   windows_code;
+               errno_t errno_code;
+       } WIN2ERRNO_TABLE[] = {
+               { ERROR_SUCCESS, 0 },
+               { ERROR_FILE_NOT_FOUND, ENOENT },
+               { ERROR_PATH_NOT_FOUND, ENOENT },
+               { ERROR_TOO_MANY_OPEN_FILES, EMFILE },
+               { ERROR_ACCESS_DENIED, EACCES },
+               { ERROR_INVALID_HANDLE, EBADF },
+               { ERROR_NOT_ENOUGH_MEMORY, ENOMEM },
+               { ERROR_LOCK_VIOLATION, EACCES },
+               { ERROR_FILE_EXISTS, EEXIST },
+               { ERROR_INVALID_PARAMETER, EINVAL },
+               { ERROR_FILENAME_EXCED_RANGE, ENAMETOOLONG },
+       };
+       const unsigned int WIN2ERRNO_TABLE_SIZE = sizeof(WIN2ERRNO_TABLE) /
+sizeof(WIN2ERRNO_TABLE[0]);
+       const errno_t DEFAULT_ERRNO_ERROR = -1;
+       unsigned int i;
+
+       for (i = 0; i < WIN2ERRNO_TABLE_SIZE; ++i) {
+               if (WIN2ERRNO_TABLE[i].windows_code == lastError) {
+                       return WIN2ERRNO_TABLE[i].errno_code;
+               }
+       }
+       return DEFAULT_ERRNO_ERROR;
+}
+
+struct dirent {
+       char *d_name;
+};
+typedef struct DIR {
+       HANDLE dir;
+       struct dirent data;
+       int first;
+       char buf[MAX_PATH+1];
+} DIR;
+DIR *opendir( char *path )
+{
+       char tmp[32768];
+       int len = strlen(path);
+       DIR *d;
+       HANDLE h;
+       WIN32_FIND_DATA data;
+       
+       if (len+3 >= sizeof(tmp)) {
+               errno = ENAMETOOLONG;
+               return NULL;
+       }
+
+       strcpy(tmp, path);
+       tmp[len++] = '\\';
+       tmp[len++] = '*';
+       tmp[len] = '\0';
+
+       h = FindFirstFile( tmp, &data );
+
+       if ( h == INVALID_HANDLE_VALUE ) {
+               errno = win2errno( GetLastError());
+               return NULL;
+       }
+
+       d = ber_memalloc( sizeof(DIR) );
+       if ( !d )
+               return NULL;
+       d->dir = h;
+       d->data.d_name = d->buf;
+       d->first = 1;
+       strcpy(d->data.d_name, data.cFileName);
+       return d;
+}
+struct dirent *readdir(DIR *dir)
+{
+       WIN32_FIND_DATA data;
+
+       if (dir->first) {
+               dir->first = 0;
+       } else {
+               if (!FindNextFile(dir->dir, &data))
+                       return NULL;
+               strcpy(dir->data.d_name, data.cFileName);
+       }
+       return &dir->data;
+}
+int closedir(DIR *dir)
+{
+       FindClose(dir->dir);
+       ber_memfree(dir);
+}
+#endif
+
 /*
  * Memory Reverse Search
  */
 void *
-lutil_memrchr(const void *b, int c, size_t n)
+(lutil_memrchr)(const void *b, int c, size_t n)
 {
        if (n != 0) {
                const unsigned char *s, *bb = b, cc = c;
@@ -331,7 +461,7 @@ lutil_memrchr(const void *b, int c, size_t n)
 }
 
 int
-lutil_atoi( int *v, const char *s )
+lutil_atoix( int *v, const char *s, int x )
 {
        char            *next;
        long            i;
@@ -339,7 +469,7 @@ lutil_atoi( int *v, const char *s )
        assert( s != NULL );
        assert( v != NULL );
 
-       i = strtol( s, &next, 10 );
+       i = strtol( s, &next, x );
        if ( next == s || next[ 0 ] != '\0' ) {
                return -1;
        }
@@ -354,7 +484,7 @@ lutil_atoi( int *v, const char *s )
 }
 
 int
-lutil_atou( unsigned *v, const char *s )
+lutil_atoux( unsigned *v, const char *s, int x )
 {
        char            *next;
        unsigned long   u;
@@ -362,7 +492,12 @@ lutil_atou( unsigned *v, const char *s )
        assert( s != NULL );
        assert( v != NULL );
 
-       u = strtoul( s, &next, 10 );
+       /* strtoul() has an odd interface */
+       if ( s[ 0 ] == '-' ) {
+               return -1;
+       }
+
+       u = strtoul( s, &next, x );
        if ( next == s || next[ 0 ] != '\0' ) {
                return -1;
        }
@@ -377,7 +512,7 @@ lutil_atou( unsigned *v, const char *s )
 }
 
 int
-lutil_atol( long *v, const char *s )
+lutil_atolx( long *v, const char *s, int x )
 {
        char            *next;
        long            l;
@@ -385,7 +520,7 @@ lutil_atol( long *v, const char *s )
        assert( s != NULL );
        assert( v != NULL );
 
-       l = strtol( s, &next, 10 );
+       l = strtol( s, &next, x );
        if ( next == s || next[ 0 ] != '\0' ) {
                return -1;
        }
@@ -396,7 +531,7 @@ lutil_atol( long *v, const char *s )
 }
 
 int
-lutil_atoul( unsigned long *v, const char *s )
+lutil_atoulx( unsigned long *v, const char *s, int x )
 {
        char            *next;
        unsigned long   ul;
@@ -404,7 +539,12 @@ lutil_atoul( unsigned long *v, const char *s )
        assert( s != NULL );
        assert( v != NULL );
 
-       ul = strtoul( s, &next, 10 );
+       /* strtoul() has an odd interface */
+       if ( s[ 0 ] == '-' ) {
+               return -1;
+       }
+
+       ul = strtoul( s, &next, x );
        if ( next == s || next[ 0 ] != '\0' ) {
                return -1;
        }
@@ -414,8 +554,207 @@ lutil_atoul( unsigned long *v, const char *s )
        return 0;
 }
 
+/* Multiply an integer by 100000000 and add new */
+typedef struct lutil_int_decnum {
+       unsigned char *buf;
+       int bufsiz;
+       int beg;
+       int len;
+} lutil_int_decnum;
+
+#define        FACTOR1 (100000000&0xffff)
+#define FACTOR2 (100000000>>16)
+
+static void
+scale( int new, lutil_int_decnum *prev, unsigned char *tmp )
+{
+       int i, j;
+       unsigned char *in = prev->buf+prev->beg;
+       unsigned int part;
+       unsigned char *out = tmp + prev->bufsiz - prev->len;
+
+       memset( tmp, 0, prev->bufsiz );
+       if ( prev->len ) {
+               for ( i = prev->len-1; i>=0; i-- ) {
+                       part = in[i] * FACTOR1;
+                       for ( j = i; part; j-- ) {
+                               part += out[j];
+                               out[j] = part & 0xff;
+                               part >>= 8;
+                       }
+                       part = in[i] * FACTOR2;
+                       for ( j = i-2; part; j-- ) {
+                               part += out[j];
+                               out[j] = part & 0xff;
+                               part >>= 8;
+                       }
+               }
+               j++;
+               prev->beg += j;
+               prev->len -= j;
+       }
+
+       out = tmp + prev->bufsiz;
+       i = 0;
+       do {
+               i--;
+               new += out[i];
+               out[i] = new & 0xff;
+               new >>= 8;
+       } while ( new );
+       i = -i;
+       if ( prev->len < i ) {
+               prev->beg = prev->bufsiz - i;
+               prev->len = i;
+       }
+       AC_MEMCPY( prev->buf+prev->beg, tmp+prev->beg, prev->len );
+}
+
+/* Convert unlimited length decimal or hex string to binary.
+ * Output buffer must be provided, bv_len must indicate buffer size
+ * Hex input can be "0x1234" or "'1234'H"
+ *
+ * Temporarily modifies the input string.
+ *
+ * Note: High bit of binary form is always the sign bit. If the number
+ * is supposed to be positive but has the high bit set, a zero byte
+ * is prepended. It is assumed that this has already been handled on
+ * any hex input.
+ */
+int
+lutil_str2bin( struct berval *in, struct berval *out, void *ctx )
+{
+       char *pin, *pout, ctmp;
+       char *end;
+       int i, chunk, len, rc = 0, hex = 0;
+       if ( !out || !out->bv_val || out->bv_len < in->bv_len )
+               return -1;
+
+       pout = out->bv_val;
+       /* Leading "0x" for hex input */
+       if ( in->bv_len > 2 && in->bv_val[0] == '0' &&
+               ( in->bv_val[1] == 'x' || in->bv_val[1] == 'X' ) )
+       {
+               len = in->bv_len - 2;
+               pin = in->bv_val + 2;
+               hex = 1;
+       } else if ( in->bv_len > 3 && in->bv_val[0] == '\'' &&
+               in->bv_val[in->bv_len-2] == '\'' &&
+               in->bv_val[in->bv_len-1] == 'H' )
+       {
+               len = in->bv_len - 3;
+               pin = in->bv_val + 1;
+               hex = 1;
+       }
+       if ( hex ) {
+#define HEXMAX (2 * sizeof(long))
+               unsigned long l;
+               /* Convert a longword at a time, but handle leading
+                * odd bytes first
+                */
+               chunk = len % HEXMAX;
+               if ( !chunk )
+                       chunk = HEXMAX;
+
+               while ( len ) {
+                       int ochunk;
+                       ctmp = pin[chunk];
+                       pin[chunk] = '\0';
+                       errno = 0;
+                       l = strtoul( pin, &end, 16 );
+                       pin[chunk] = ctmp;
+                       if ( errno )
+                               return -1;
+                       ochunk = (chunk + 1)/2;
+                       for ( i = ochunk - 1; i >= 0; i-- ) {
+                               pout[i] = l & 0xff;
+                               l >>= 8;
+                       }
+                       pin += chunk;
+                       pout += ochunk;
+                       len -= chunk;
+                       chunk = HEXMAX;
+               }
+               out->bv_len = pout - out->bv_val;
+       } else {
+       /* Decimal */
+               char tmpbuf[64], *tmp;
+               lutil_int_decnum num;
+               int neg = 0;
+               long l;
+
+               len = in->bv_len;
+               pin = in->bv_val;
+               num.buf = (unsigned char *)out->bv_val;
+               num.bufsiz = out->bv_len;
+               num.beg = num.bufsiz-1;
+               num.len = 0;
+               if ( pin[0] == '-' ) {
+                       neg = 0xff;
+                       len--;
+                       pin++;
+               }
+
+#define        DECMAX  8       /* 8 digits at a time */
+
+               /* tmp must be at least as large as outbuf */
+               if ( out->bv_len > sizeof(tmpbuf)) {
+                       tmp = ber_memalloc_x( out->bv_len, ctx );
+               } else {
+                       tmp = tmpbuf;
+               }
+               chunk = len & (DECMAX-1);
+               if ( !chunk )
+                       chunk = DECMAX;
+
+               while ( len ) {
+                       ctmp = pin[chunk];
+                       pin[chunk] = '\0';
+                       errno = 0;
+                       l = strtol( pin, &end, 10 );
+                       pin[chunk] = ctmp;
+                       if ( errno ) {
+                               rc = -1;
+                               goto decfail;
+                       }
+                       scale( l, &num, (unsigned char *)tmp );
+                       pin += chunk;
+                       len -= chunk;
+                       chunk = DECMAX;
+               }
+               /* Negate the result */
+               if ( neg ) {
+                       unsigned char *ptr;
+
+                       ptr = num.buf+num.beg;
+
+                       /* flip all bits */
+                       for ( i=0; i<num.len; i++ )
+                               ptr[i] ^= 0xff;
+
+                       /* add 1, with carry - overflow handled below */
+                       while ( i-- && ! (ptr[i] = (ptr[i] + 1) & 0xff )) ;
+               }
+               /* Prepend sign byte if wrong sign bit */
+               if (( num.buf[num.beg] ^ neg ) & 0x80 ) {
+                       num.beg--;
+                       num.len++;
+                       num.buf[num.beg] = neg;
+               }
+               if ( num.beg )
+                       AC_MEMCPY( num.buf, num.buf+num.beg, num.len );
+               out->bv_len = num.len;
+decfail:
+               if ( tmp != tmpbuf ) {
+                       ber_memfree_x( tmp, ctx );
+               }
+       }
+       return rc;
+}
+
 static char            time_unit[] = "dhms";
 
+/* Used to parse and unparse time intervals, not timestamps */
 int
 lutil_parse_time(
        const char      *in,
@@ -433,6 +772,11 @@ lutil_parse_time(
                unsigned long   u;
                char            *what;
 
+               /* strtoul() has an odd interface */
+               if ( s[ 0 ] == '-' ) {
+                       return -1;
+               }
+
                u = strtoul( s, &next, 10 );
                if ( next == s ) {
                        return -1;
@@ -471,6 +815,7 @@ lutil_unparse_time(
 {
        int             len, i;
        unsigned long   v[ 4 ];
+       char            *ptr = buf;
 
        v[ 0 ] = t/86400;
        v[ 1 ] = (t%86400)/3600;
@@ -478,14 +823,64 @@ lutil_unparse_time(
        v[ 3 ] = t%60;
 
        for ( i = 0; i < 4; i++ ) {
-               if ( v[i] > 0 || i == 3 ) {
-                       len = snprintf( buf, buflen, "%lu%c", v[ i ], time_unit[ i ] );
+               if ( v[i] > 0 || ( i == 3 && ptr == buf ) ) {
+                       len = snprintf( ptr, buflen, "%lu%c", v[ i ], time_unit[ i ] );
                        if ( len < 0 || (unsigned)len >= buflen ) {
                                return -1;
                        }
                        buflen -= len;
-                       buf += len;
+                       ptr += len;
+               }
+       }
+
+       return 0;
+}
+
+/*
+ * formatted print to string
+ *
+ * - if return code < 0, the error code returned by vsnprintf(3) is returned
+ *
+ * - if return code > 0, the buffer was not long enough;
+ *     - if next is not NULL, *next will be set to buf + bufsize - 1
+ *     - if len is not NULL, *len will contain the required buffer length
+ *
+ * - if return code == 0, the buffer was long enough;
+ *     - if next is not NULL, *next will point to the end of the string printed so far
+ *     - if len is not NULL, *len will contain the length of the string printed so far 
+ */
+int
+lutil_snprintf( char *buf, ber_len_t bufsize, char **next, ber_len_t *len, LDAP_CONST char *fmt, ... )
+{
+       va_list         ap;
+       int             ret;
+
+       assert( buf != NULL );
+       assert( bufsize > 0 );
+       assert( fmt != NULL );
+
+       va_start( ap, fmt );
+       ret = vsnprintf( buf, bufsize, fmt, ap );
+       va_end( ap );
+
+       if ( ret < 0 ) {
+               return ret;
+       }
+
+       if ( len ) {
+               *len = ret;
+       }
+
+       if ( (unsigned) ret >= bufsize ) {
+               if ( next ) {
+                       *next = &buf[ bufsize - 1 ];
                }
+
+               return 1;
+       }
+
+       if ( next ) {
+               *next = &buf[ ret ];
        }
 
        return 0;