git.sur5r.net Git - openldap/blob - libraries/libmdb/midl.c

   1 /**     @file midl.c
   2  *      @brief ldap bdb back-end ID List functions */
   3 /* $OpenLDAP$ */
   4 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   5  *
   6  * Copyright 2000-2011 The OpenLDAP Foundation.
   7  * All rights reserved.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted only as authorized by the OpenLDAP
  11  * Public License.
  12  *
  13  * A copy of this license is available in the file LICENSE in the
  14  * top-level directory of the distribution or, alternatively, at
  15  * <http://www.OpenLDAP.org/license.html>.
  16  */
  17
  18 #include <limits.h>
  19 #include <string.h>
  20 #include <stdlib.h>
  21 #include <sys/types.h>
  22 #include <assert.h>
  23 #include "midl.h"
  24
  25 /** @defgroup internal  MDB Internals
  26  *      @{
  27  */
  28 /** @defgroup idls      ID List Management
  29  *      @{
  30  */
  31 #define CMP(x,y)         ( (x) < (y) ? -1 : (x) > (y) )
  32
  33 #if 0   /* superseded by append/sort */
  34 static unsigned mdb_midl_search( IDL ids, ID id )
  35 {
  36         /*
  37          * binary search of id in ids
  38          * if found, returns position of id
  39          * if not found, returns first position greater than id
  40          */
  41         unsigned base = 0;
  42         unsigned cursor = 0;
  43         int val = 0;
  44         unsigned n = ids[0];
  45
  46         while( 0 < n ) {
  47                 int pivot = n >> 1;
  48                 cursor = base + pivot;
  49                 val = CMP( ids[cursor + 1], id );
  50
  51                 if( val < 0 ) {
  52                         n = pivot;
  53
  54                 } else if ( val > 0 ) {
  55                         base = cursor + 1;
  56                         n -= pivot + 1;
  57
  58                 } else {
  59                         return cursor + 1;
  60                 }
  61         }
  62
  63         if( val > 0 ) {
  64                 return cursor + 2;
  65         } else {
  66                 return cursor + 1;
  67         }
  68 }
  69
  70 int mdb_midl_insert( IDL ids, ID id )
  71 {
  72         unsigned x, i;
  73
  74         if (MDB_IDL_IS_RANGE( ids )) {
  75                 /* if already in range, treat as a dup */
  76                 if (id >= MDB_IDL_FIRST(ids) && id <= MDB_IDL_LAST(ids))
  77                         return -1;
  78                 if (id < MDB_IDL_FIRST(ids))
  79                         ids[1] = id;
  80                 else if (id > MDB_IDL_LAST(ids))
  81                         ids[2] = id;
  82                 return 0;
  83         }
  84
  85         x = mdb_midl_search( ids, id );
  86         assert( x > 0 );
  87
  88         if( x < 1 ) {
  89                 /* internal error */
  90                 return -2;
  91         }
  92
  93         if ( x <= ids[0] && ids[x] == id ) {
  94                 /* duplicate */
  95                 assert(0);
  96                 return -1;
  97         }
  98
  99         if ( ++ids[0] >= MDB_IDL_DB_MAX ) {
 100                 if( id < ids[1] ) {
 101                         ids[1] = id;
 102                         ids[2] = ids[ids[0]-1];
 103                 } else if ( ids[ids[0]-1] < id ) {
 104                         ids[2] = id;
 105                 } else {
 106                         ids[2] = ids[ids[0]-1];
 107                 }
 108                 ids[0] = NOID;
 109
 110         } else {
 111                 /* insert id */
 112                 for (i=ids[0]; i>x; i--)
 113                         ids[i] = ids[i-1];
 114                 ids[x] = id;
 115         }
 116
 117         return 0;
 118 }
 119 #endif
 120
 121 IDL mdb_midl_alloc()
 122 {
 123         IDL ids = malloc((MDB_IDL_UM_MAX+1) * sizeof(ID));
 124         *ids++ = MDB_IDL_UM_MAX;
 125         return ids;
 126 }
 127
 128 void mdb_midl_free(IDL ids)
 129 {
 130         free(ids-1);
 131 }
 132
 133 int mdb_midl_shrink( IDL *idp )
 134 {
 135         IDL ids = *idp;
 136         if (ids[-1] > MDB_IDL_UM_MAX) {
 137                 ids = realloc(ids, (MDB_IDL_UM_MAX+1) * sizeof(ID));
 138                 *ids++ = MDB_IDL_UM_MAX;
 139                 *idp = ids;
 140                 return 1;
 141         }
 142         return 0;
 143 }
 144
 145 int mdb_midl_append( IDL *idp, ID id )
 146 {
 147         IDL ids = *idp;
 148         /* Too big? */
 149         if (ids[0] >= ids[-1]) {
 150                 IDL idn = ids-1;
 151                 /* grow it */
 152                 idn = realloc(idn, (*idn + MDB_IDL_UM_MAX + 1) * sizeof(ID));
 153                 if (!idn)
 154                         return -1;
 155                 *idn++ += MDB_IDL_UM_MAX;
 156                 ids = idn;
 157                 *idp = ids;
 158         }
 159         ids[0]++;
 160         ids[ids[0]] = id;
 161         return 0;
 162 }
 163
 164 /* Quicksort + Insertion sort for small arrays */
 165
 166 #define SMALL   8
 167 #define SWAP(a,b)       { itmp=(a); (a)=(b); (b)=itmp; }
 168
 169 void
 170 mdb_midl_sort( ID *ids )
 171 {
 172         /* Max possible depth of int-indexed tree * 2 items/level */
 173         int istack[sizeof(int)*CHAR_BIT * 2];
 174         int i,j,k,l,ir,jstack;
 175         ID a, itmp;
 176
 177         ir = ids[0];
 178         l = 1;
 179         jstack = 0;
 180         for(;;) {
 181                 if (ir - l < SMALL) {   /* Insertion sort */
 182                         for (j=l+1;j<=ir;j++) {
 183                                 a = ids[j];
 184                                 for (i=j-1;i>=1;i--) {
 185                                         if (ids[i] >= a) break;
 186                                         ids[i+1] = ids[i];
 187                                 }
 188                                 ids[i+1] = a;
 189                         }
 190                         if (jstack == 0) break;
 191                         ir = istack[jstack--];
 192                         l = istack[jstack--];
 193                 } else {
 194                         k = (l + ir) >> 1;      /* Choose median of left, center, right */
 195                         SWAP(ids[k], ids[l+1]);
 196                         if (ids[l] < ids[ir]) {
 197                                 SWAP(ids[l], ids[ir]);
 198                         }
 199                         if (ids[l+1] < ids[ir]) {
 200                                 SWAP(ids[l+1], ids[ir]);
 201                         }
 202                         if (ids[l] < ids[l+1]) {
 203                                 SWAP(ids[l], ids[l+1]);
 204                         }
 205                         i = l+1;
 206                         j = ir;
 207                         a = ids[l+1];
 208                         for(;;) {
 209                                 do i++; while(ids[i] > a);
 210                                 do j--; while(ids[j] < a);
 211                                 if (j < i) break;
 212                                 SWAP(ids[i],ids[j]);
 213                         }
 214                         ids[l+1] = ids[j];
 215                         ids[j] = a;
 216                         jstack += 2;
 217                         if (ir-i+1 >= j-1) {
 218                                 istack[jstack] = ir;
 219                                 istack[jstack-1] = i;
 220                                 ir = j-1;
 221                         } else {
 222                                 istack[jstack] = j-1;
 223                                 istack[jstack-1] = l;
 224                                 l = i;
 225                         }
 226                 }
 227         }
 228 }
 229
 230 unsigned mdb_mid2l_search( ID2L ids, ID id )
 231 {
 232         /*
 233          * binary search of id in ids
 234          * if found, returns position of id
 235          * if not found, returns first position greater than id
 236          */
 237         unsigned base = 0;
 238         unsigned cursor = 0;
 239         int val = 0;
 240         unsigned n = ids[0].mid;
 241
 242         while( 0 < n ) {
 243                 int pivot = n >> 1;
 244                 cursor = base + pivot;
 245                 val = CMP( id, ids[cursor + 1].mid );
 246
 247                 if( val < 0 ) {
 248                         n = pivot;
 249
 250                 } else if ( val > 0 ) {
 251                         base = cursor + 1;
 252                         n -= pivot + 1;
 253
 254                 } else {
 255                         return cursor + 1;
 256                 }
 257         }
 258
 259         if( val > 0 ) {
 260                 return cursor + 2;
 261         } else {
 262                 return cursor + 1;
 263         }
 264 }
 265
 266 int mdb_mid2l_insert( ID2L ids, ID2 *id )
 267 {
 268         unsigned x, i;
 269
 270         x = mdb_mid2l_search( ids, id->mid );
 271         assert( x > 0 );
 272
 273         if( x < 1 ) {
 274                 /* internal error */
 275                 return -2;
 276         }
 277
 278         if ( x <= ids[0].mid && ids[x].mid == id->mid ) {
 279                 /* duplicate */
 280                 return -1;
 281         }
 282
 283         if ( ids[0].mid >= MDB_IDL_UM_MAX ) {
 284                 /* too big */
 285                 return -2;
 286
 287         } else {
 288                 /* insert id */
 289                 ids[0].mid++;
 290                 for (i=ids[0].mid; i>x; i--)
 291                         ids[i] = ids[i-1];
 292                 ids[x] = *id;
 293         }
 294
 295         return 0;
 296 }
 297 /** @} */
 298 /** @} */