git.sur5r.net Git - openldap/blob - libraries/liblmdb/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-2013 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( MDB_IDL ids, MDB_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 = 1;
  43         int val = 0;
  44         unsigned n = ids[0];
  45
  46         while( 0 < n ) {
  47                 unsigned pivot = n >> 1;
  48                 cursor = base + pivot + 1;
  49                 val = CMP( ids[cursor], id );
  50
  51                 if( val < 0 ) {
  52                         n = pivot;
  53
  54                 } else if ( val > 0 ) {
  55                         base = cursor;
  56                         n -= pivot + 1;
  57
  58                 } else {
  59                         return cursor;
  60                 }
  61         }
  62
  63         if( val > 0 ) {
  64                 ++cursor;
  65         }
  66         return cursor;
  67 }
  68
  69 int mdb_midl_insert( MDB_IDL ids, MDB_ID id )
  70 {
  71         unsigned x, i;
  72
  73         if (MDB_IDL_IS_RANGE( ids )) {
  74                 /* if already in range, treat as a dup */
  75                 if (id >= MDB_IDL_RANGE_FIRST(ids) && id <= MDB_IDL_RANGE_LAST(ids))
  76                         return -1;
  77                 if (id < MDB_IDL_RANGE_FIRST(ids))
  78                         ids[1] = id;
  79                 else if (id > MDB_IDL_RANGE_LAST(ids))
  80                         ids[2] = id;
  81                 return 0;
  82         }
  83
  84         x = mdb_midl_search( ids, id );
  85         assert( x > 0 );
  86
  87         if( x < 1 ) {
  88                 /* internal error */
  89                 return -2;
  90         }
  91
  92         if ( x <= ids[0] && ids[x] == id ) {
  93                 /* duplicate */
  94                 assert(0);
  95                 return -1;
  96         }
  97
  98         if ( ++ids[0] >= MDB_IDL_DB_MAX ) {
  99                 if( id < ids[1] ) {
 100                         ids[1] = id;
 101                         ids[2] = ids[ids[0]-1];
 102                 } else if ( ids[ids[0]-1] < id ) {
 103                         ids[2] = id;
 104                 } else {
 105                         ids[2] = ids[ids[0]-1];
 106                 }
 107                 ids[0] = MDB_NOID;
 108
 109         } else {
 110                 /* insert id */
 111                 for (i=ids[0]; i>x; i--)
 112                         ids[i] = ids[i-1];
 113                 ids[x] = id;
 114         }
 115
 116         return 0;
 117 }
 118 #endif
 119
 120 MDB_IDL mdb_midl_alloc(int num)
 121 {
 122         MDB_IDL ids = malloc((num+2) * sizeof(MDB_ID));
 123         if (ids)
 124                 *ids++ = num;
 125         return ids;
 126 }
 127
 128 void mdb_midl_free(MDB_IDL ids)
 129 {
 130         if (ids)
 131                 free(ids-1);
 132 }
 133
 134 int mdb_midl_shrink( MDB_IDL *idp )
 135 {
 136         MDB_IDL ids = *idp;
 137         if (*(--ids) > MDB_IDL_UM_MAX) {
 138                 ids = realloc(ids, (MDB_IDL_UM_MAX+1) * sizeof(MDB_ID));
 139                 *ids++ = MDB_IDL_UM_MAX;
 140                 *idp = ids;
 141                 return 1;
 142         }
 143         return 0;
 144 }
 145
 146 int mdb_midl_grow( MDB_IDL *idp, int num )
 147 {
 148         MDB_IDL idn = *idp-1;
 149         /* grow it */
 150         idn = realloc(idn, (*idn + num + 2) * sizeof(MDB_ID));
 151         if (!idn)
 152                 return ENOMEM;
 153         *idn++ += num;
 154         *idp = idn;
 155         return 0;
 156 }
 157
 158 int mdb_midl_append( MDB_IDL *idp, MDB_ID id )
 159 {
 160         MDB_IDL ids = *idp;
 161         /* Too big? */
 162         if (ids[0] >= ids[-1]) {
 163                 if (mdb_midl_grow(idp, MDB_IDL_UM_MAX))
 164                         return ENOMEM;
 165                 ids = *idp;
 166         }
 167         ids[0]++;
 168         ids[ids[0]] = id;
 169         return 0;
 170 }
 171
 172 int mdb_midl_append_list( MDB_IDL *idp, MDB_IDL app )
 173 {
 174         MDB_IDL ids = *idp;
 175         /* Too big? */
 176         if (ids[0] + app[0] >= ids[-1]) {
 177                 if (mdb_midl_grow(idp, app[0]))
 178                         return ENOMEM;
 179                 ids = *idp;
 180         }
 181         memcpy(&ids[ids[0]+1], &app[1], app[0] * sizeof(MDB_ID));
 182         ids[0] += app[0];
 183         return 0;
 184 }
 185
 186 /* Quicksort + Insertion sort for small arrays */
 187
 188 #define SMALL   8
 189 #define SWAP(a,b)       { itmp=(a); (a)=(b); (b)=itmp; }
 190
 191 void
 192 mdb_midl_sort( MDB_IDL ids )
 193 {
 194         /* Max possible depth of int-indexed tree * 2 items/level */
 195         int istack[sizeof(int)*CHAR_BIT * 2];
 196         int i,j,k,l,ir,jstack;
 197         MDB_ID a, itmp;
 198
 199         ir = (int)ids[0];
 200         l = 1;
 201         jstack = 0;
 202         for(;;) {
 203                 if (ir - l < SMALL) {   /* Insertion sort */
 204                         for (j=l+1;j<=ir;j++) {
 205                                 a = ids[j];
 206                                 for (i=j-1;i>=1;i--) {
 207                                         if (ids[i] >= a) break;
 208                                         ids[i+1] = ids[i];
 209                                 }
 210                                 ids[i+1] = a;
 211                         }
 212                         if (jstack == 0) break;
 213                         ir = istack[jstack--];
 214                         l = istack[jstack--];
 215                 } else {
 216                         k = (l + ir) >> 1;      /* Choose median of left, center, right */
 217                         SWAP(ids[k], ids[l+1]);
 218                         if (ids[l] < ids[ir]) {
 219                                 SWAP(ids[l], ids[ir]);
 220                         }
 221                         if (ids[l+1] < ids[ir]) {
 222                                 SWAP(ids[l+1], ids[ir]);
 223                         }
 224                         if (ids[l] < ids[l+1]) {
 225                                 SWAP(ids[l], ids[l+1]);
 226                         }
 227                         i = l+1;
 228                         j = ir;
 229                         a = ids[l+1];
 230                         for(;;) {
 231                                 do i++; while(ids[i] > a);
 232                                 do j--; while(ids[j] < a);
 233                                 if (j < i) break;
 234                                 SWAP(ids[i],ids[j]);
 235                         }
 236                         ids[l+1] = ids[j];
 237                         ids[j] = a;
 238                         jstack += 2;
 239                         if (ir-i+1 >= j-l) {
 240                                 istack[jstack] = ir;
 241                                 istack[jstack-1] = i;
 242                                 ir = j-1;
 243                         } else {
 244                                 istack[jstack] = j-1;
 245                                 istack[jstack-1] = l;
 246                                 l = i;
 247                         }
 248                 }
 249         }
 250 }
 251
 252 unsigned mdb_mid2l_search( MDB_ID2L ids, MDB_ID id )
 253 {
 254         /*
 255          * binary search of id in ids
 256          * if found, returns position of id
 257          * if not found, returns first position greater than id
 258          */
 259         unsigned base = 0;
 260         unsigned cursor = 1;
 261         int val = 0;
 262         unsigned n = (unsigned)ids[0].mid;
 263
 264         while( 0 < n ) {
 265                 unsigned pivot = n >> 1;
 266                 cursor = base + pivot + 1;
 267                 val = CMP( id, ids[cursor].mid );
 268
 269                 if( val < 0 ) {
 270                         n = pivot;
 271
 272                 } else if ( val > 0 ) {
 273                         base = cursor;
 274                         n -= pivot + 1;
 275
 276                 } else {
 277                         return cursor;
 278                 }
 279         }
 280
 281         if( val > 0 ) {
 282                 ++cursor;
 283         }
 284         return cursor;
 285 }
 286
 287 int mdb_mid2l_insert( MDB_ID2L ids, MDB_ID2 *id )
 288 {
 289         unsigned x, i;
 290
 291         x = mdb_mid2l_search( ids, id->mid );
 292         assert( x > 0 );
 293
 294         if( x < 1 ) {
 295                 /* internal error */
 296                 return -2;
 297         }
 298
 299         if ( x <= ids[0].mid && ids[x].mid == id->mid ) {
 300                 /* duplicate */
 301                 return -1;
 302         }
 303
 304         if ( ids[0].mid >= MDB_IDL_UM_MAX ) {
 305                 /* too big */
 306                 return -2;
 307
 308         } else {
 309                 /* insert id */
 310                 ids[0].mid++;
 311                 for (i=(unsigned)ids[0].mid; i>x; i--)
 312                         ids[i] = ids[i-1];
 313                 ids[x] = *id;
 314         }
 315
 316         return 0;
 317 }
 318
 319 int mdb_mid2l_append( MDB_ID2L ids, MDB_ID2 *id )
 320 {
 321         /* Too big? */
 322         if (ids[0].mid >= MDB_IDL_UM_MAX) {
 323                 return -2;
 324         }
 325         ids[0].mid++;
 326         ids[ids[0].mid] = *id;
 327         return 0;
 328 }
 329
 330 /** @} */
 331 /** @} */