isamb/isamb.c

   1 /* $Id: isamb.c,v 1.24 2003-03-17 20:22:54 adam Exp $
   2    Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003
   3    Index Data Aps
   4
   5 This file is part of the Zebra server.
   6
   7 Zebra is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 2, or (at your option) any later
  10 version.
  11
  12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with Zebra; see the file LICENSE.zebra.  If not, write to the
  19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  20 02111-1307, USA.
  21 */
  22
  23 #include <string.h>
  24 #include <yaz/xmalloc.h>
  25 #include <yaz/log.h>
  26 #include <isamb.h>
  27 #include <assert.h>
  28
  29 struct ISAMB_head {
  30     int first_block;
  31     int last_block;
  32     int block_size;
  33     int block_max;
  34     int free_list;
  35 };
  36
  37 #define ISAMB_DATA_OFFSET 3
  38
  39 #define DST_ITEM_MAX 256
  40
  41 /* approx 2*4 K + max size of item */
  42 #define DST_BUF_SIZE 8448
  43
  44 #define ISAMB_CACHE_ENTRY_SIZE 4096
  45
  46 struct ISAMB_cache_entry {
  47     ISAMB_P pos;
  48     unsigned char *buf;
  49     int dirty;
  50     int hits;
  51     struct ISAMB_cache_entry *next;
  52 };
  53
  54
  55 struct ISAMB_file {
  56     BFile bf;
  57     int head_dirty;
  58     struct ISAMB_head head;
  59     struct ISAMB_cache_entry *cache_entries;
  60 };
  61
  62 struct ISAMB_s {
  63     BFiles bfs;
  64     ISAMC_M method;
  65
  66     struct ISAMB_file *file;
  67     int no_cat;
  68     int cache; /* 0=no cache, 1=use cache, -1=dummy isam (for testing only) */
  69     int log_io;        /* log level for bf_read/bf_write calls */
  70     int log_freelist;  /* log level for freelist handling */
  71 };
  72
  73 struct ISAMB_block {
  74     ISAMB_P pos;
  75     int cat;
  76     int size;
  77     int leaf;
  78     int dirty;
  79     int deleted;
  80     int offset;
  81     char *bytes;
  82     unsigned char *buf;
  83     void *decodeClientData;
  84     int log_rw;
  85 };
  86
  87 struct ISAMB_PP_s {
  88     ISAMB isamb;
  89     ISAMB_P pos;
  90     int level;
  91     int total_size;
  92     int no_blocks;
  93     struct ISAMB_block **block;
  94 };
  95
  96 void encode_ptr (char **dst, int pos)
  97 {
  98     memcpy (*dst, &pos, sizeof(pos));
  99     (*dst) += sizeof(pos);
 100 }
 101
 102 void decode_ptr (char **src, int *pos)
 103 {
 104     memcpy (pos, *src, sizeof(*pos));
 105     (*src) += sizeof(*pos);
 106 }
 107
 108 ISAMB isamb_open (BFiles bfs, const char *name, int writeflag, ISAMC_M method,
 109                   int cache)
 110 {
 111     ISAMB isamb = xmalloc (sizeof(*isamb));
 112     int i, b_size = 32;
 113
 114     isamb->bfs = bfs;
 115     isamb->method = (ISAMC_M) xmalloc (sizeof(*method));
 116     memcpy (isamb->method, method, sizeof(*method));
 117     isamb->no_cat = 4;
 118     isamb->log_io = 0;
 119     isamb->cache = cache;
 120
 121     assert (cache == 0);
 122     isamb->file = xmalloc (sizeof(*isamb->file) * isamb->no_cat);
 123     for (i = 0; i<isamb->no_cat; i++)
 124     {
 125         char fname[DST_BUF_SIZE];
 126         isamb->file[i].cache_entries = 0;
 127         isamb->file[i].head_dirty = 0;
 128         sprintf (fname, "%s%c", name, i+'A');
 129         if (cache)
 130             isamb->file[i].bf = bf_open (bfs, fname, ISAMB_CACHE_ENTRY_SIZE,
 131                                          writeflag);
 132         else
 133             isamb->file[i].bf = bf_open (bfs, fname, b_size, writeflag);
 134
 135
 136         if (!bf_read (isamb->file[i].bf, 0, 0, sizeof(struct ISAMB_head),
 137                       &isamb->file[i].head))
 138         {
 139             isamb->file[i].head.first_block = ISAMB_CACHE_ENTRY_SIZE/b_size+1;
 140             isamb->file[i].head.last_block = isamb->file[i].head.first_block;
 141             isamb->file[i].head.block_size = b_size;
 142             isamb->file[i].head.block_max = b_size - ISAMB_DATA_OFFSET;
 143             isamb->file[i].head.free_list = 0;
 144         }
 145         assert (isamb->file[i].head.block_size >= ISAMB_DATA_OFFSET);
 146         isamb->file[i].head_dirty = 0;
 147         assert(isamb->file[i].head.block_size == b_size);
 148         b_size = b_size * 4;
 149     }
 150     return isamb;
 151 }
 152
 153 static void flush_blocks (ISAMB b, int cat)
 154 {
 155     while (b->file[cat].cache_entries)
 156     {
 157         struct ISAMB_cache_entry *ce_this = b->file[cat].cache_entries;
 158         b->file[cat].cache_entries = ce_this->next;
 159
 160         if (ce_this->dirty)
 161         {
 162             yaz_log (b->log_io, "bf_write: flush_blocks");
 163             bf_write (b->file[cat].bf, ce_this->pos, 0, 0, ce_this->buf);
 164         }
 165         xfree (ce_this->buf);
 166         xfree (ce_this);
 167     }
 168 }
 169
 170 static int get_block (ISAMB b, ISAMC_P pos, char *userbuf, int wr)
 171 {
 172     int cat = pos&3;
 173     int off = ((pos/4) &
 174                (ISAMB_CACHE_ENTRY_SIZE / b->file[cat].head.block_size - 1))
 175         * b->file[cat].head.block_size;
 176     int norm = pos / (4*ISAMB_CACHE_ENTRY_SIZE / b->file[cat].head.block_size);
 177     int no = 0;
 178     struct ISAMB_cache_entry **ce, *ce_this = 0, **ce_last = 0;
 179
 180     if (!b->cache)
 181         return 0;
 182
 183     assert (ISAMB_CACHE_ENTRY_SIZE >= b->file[cat].head.block_size);
 184     for (ce = &b->file[cat].cache_entries; *ce; ce = &(*ce)->next, no++)
 185     {
 186         ce_last = ce;
 187         if ((*ce)->pos == norm)
 188         {
 189             ce_this = *ce;
 190             *ce = (*ce)->next;   /* remove from list */
 191
 192             ce_this->next = b->file[cat].cache_entries;  /* move to front */
 193             b->file[cat].cache_entries = ce_this;
 194
 195             if (wr)
 196             {
 197                 memcpy (ce_this->buf + off, userbuf,
 198                         b->file[cat].head.block_size);
 199                 ce_this->dirty = 1;
 200             }
 201             else
 202                 memcpy (userbuf, ce_this->buf + off,
 203                         b->file[cat].head.block_size);
 204             return 1;
 205         }
 206     }
 207     if (no >= 40)
 208     {
 209         assert (no == 40);
 210         assert (ce_last && *ce_last);
 211         ce_this = *ce_last;
 212         *ce_last = 0;  /* remove the last entry from list */
 213         if (ce_this->dirty)
 214         {
 215             yaz_log (b->log_io, "bf_write: get_block");
 216             bf_write (b->file[cat].bf, ce_this->pos, 0, 0, ce_this->buf);
 217         }
 218         xfree (ce_this->buf);
 219         xfree (ce_this);
 220     }
 221     ce_this = xmalloc (sizeof(*ce_this));
 222     ce_this->next = b->file[cat].cache_entries;
 223     b->file[cat].cache_entries = ce_this;
 224     ce_this->buf = xmalloc (ISAMB_CACHE_ENTRY_SIZE);
 225     ce_this->pos = norm;
 226     yaz_log (b->log_io, "bf_read: get_block");
 227     if (!bf_read (b->file[cat].bf, norm, 0, 0, ce_this->buf))
 228         memset (ce_this->buf, 0, ISAMB_CACHE_ENTRY_SIZE);
 229     if (wr)
 230     {
 231         memcpy (ce_this->buf + off, userbuf, b->file[cat].head.block_size);
 232         ce_this->dirty = 1;
 233     }
 234     else
 235     {
 236         ce_this->dirty = 0;
 237         memcpy (userbuf, ce_this->buf + off, b->file[cat].head.block_size);
 238     }
 239     return 1;
 240 }
 241
 242
 243 void isamb_close (ISAMB isamb)
 244 {
 245     int i;
 246     for (i = 0; i<isamb->no_cat; i++)
 247     {
 248         flush_blocks (isamb, i);
 249         if (isamb->file[i].head_dirty)
 250             bf_write (isamb->file[i].bf, 0, 0,
 251                       sizeof(struct ISAMB_head), &isamb->file[i].head);
 252
 253         bf_close (isamb->file[i].bf);
 254     }
 255     xfree (isamb->file);
 256     xfree (isamb->method);
 257     xfree (isamb);
 258 }
 259
 260
 261 struct ISAMB_block *open_block (ISAMB b, ISAMC_P pos)
 262 {
 263     int cat = pos&3;
 264     struct ISAMB_block *p;
 265     if (!pos)
 266         return 0;
 267     p = xmalloc (sizeof(*p));
 268     p->pos = pos;
 269     p->cat = pos & 3;
 270     p->buf = xmalloc (b->file[cat].head.block_size);
 271
 272     if (!get_block (b, pos, p->buf, 0))
 273     {
 274         yaz_log (b->log_io, "bf_read: open_block");
 275         if (!bf_read (b->file[cat].bf, pos/4, 0, 0, p->buf))
 276         {
 277             yaz_log (LOG_FATAL, "read failure for pos=%ld block=%ld",
 278                      (long) pos, (long) pos/4);
 279             abort();
 280         }
 281     }
 282     p->bytes = p->buf + ISAMB_DATA_OFFSET;
 283     p->leaf = p->buf[0];
 284     p->size = (p->buf[1] + 256 * p->buf[2]) - ISAMB_DATA_OFFSET;
 285     if (p->size < 0)
 286     {
 287         fprintf (stderr, "pos=%d\n", pos);
 288     }
 289     assert (p->size >= 0);
 290     p->offset = 0;
 291     p->dirty = 0;
 292     p->deleted = 0;
 293     p->decodeClientData = (*b->method->code_start)(ISAMC_DECODE);
 294     return p;
 295 }
 296
 297 struct ISAMB_block *new_block (ISAMB b, int leaf, int cat)
 298 {
 299     struct ISAMB_block *p;
 300
 301     p = xmalloc (sizeof(*p));
 302     p->buf = xmalloc (b->file[cat].head.block_size);
 303
 304     if (!b->file[cat].head.free_list)
 305     {
 306         int block_no;
 307         block_no = b->file[cat].head.last_block++;
 308         p->pos = block_no * 4 + cat;
 309     }
 310     else
 311     {
 312         p->pos = b->file[cat].head.free_list;
 313         assert((p->pos & 3) == cat);
 314         if (!get_block (b, p->pos, p->buf, 0))
 315         {
 316             yaz_log (b->log_io, "bf_read: new_block");
 317             if (!bf_read (b->file[cat].bf, p->pos/4, 0, 0, p->buf))
 318             {
 319                 yaz_log (LOG_FATAL, "read failure for pos=%ld block=%ld",
 320                          (long) p->pos/4, (long) p->pos/4);
 321                 abort ();
 322             }
 323         }
 324         yaz_log (b->log_freelist, "got block %d from freelist %d:%d", p->pos,
 325                  cat, p->pos/4);
 326         memcpy (&b->file[cat].head.free_list, p->buf, sizeof(int));
 327     }
 328     p->cat = cat;
 329     b->file[cat].head_dirty = 1;
 330     memset (p->buf, 0, b->file[cat].head.block_size);
 331     p->bytes = p->buf + ISAMB_DATA_OFFSET;
 332     p->leaf = leaf;
 333     p->size = 0;
 334     p->dirty = 1;
 335     p->deleted = 0;
 336     p->offset = 0;
 337     p->decodeClientData = (*b->method->code_start)(ISAMC_DECODE);
 338     return p;
 339 }
 340
 341 struct ISAMB_block *new_leaf (ISAMB b, int cat)
 342 {
 343     return new_block (b, 1, cat);
 344 }
 345
 346
 347 struct ISAMB_block *new_int (ISAMB b, int cat)
 348 {
 349     return new_block (b, 0, cat);
 350 }
 351
 352 static void check_block (ISAMB b, struct ISAMB_block *p)
 353 {
 354     if (p->leaf)
 355     {
 356         ;
 357     }
 358     else
 359     {
 360         /* sanity check */
 361         char *startp = p->bytes;
 362         char *src = startp;
 363         char *endp = p->bytes + p->size;
 364         int pos;
 365
 366         decode_ptr (&src, &pos);
 367         assert ((pos&3) == p->cat);
 368         while (src != endp)
 369         {
 370             int item_len;
 371             decode_ptr (&src, &item_len);
 372             assert (item_len > 0 && item_len < 30);
 373             src += item_len;
 374             decode_ptr (&src, &pos);
 375             assert ((pos&3) == p->cat);
 376         }
 377     }
 378 }
 379
 380 void close_block (ISAMB b, struct ISAMB_block *p)
 381 {
 382     if (!p)
 383         return;
 384     if (p->deleted)
 385     {
 386         yaz_log (b->log_freelist, "release block %d from freelist %d:%d",
 387                  p->pos, p->cat, p->pos/4);
 388         memcpy (p->buf, &b->file[p->cat].head.free_list, sizeof(int));
 389         b->file[p->cat].head.free_list = p->pos;
 390         if (!get_block (b, p->pos, p->buf, 1))
 391         {
 392             yaz_log (b->log_io, "bf_write: close_block (deleted)");
 393             bf_write (b->file[p->cat].bf, p->pos/4, 0, 0, p->buf);
 394         }
 395     }
 396     else if (p->dirty)
 397     {
 398         int size = p->size + ISAMB_DATA_OFFSET;
 399         assert (p->size >= 0);
 400         p->buf[0] = p->leaf;
 401         p->buf[1] = size & 255;
 402         p->buf[2] = size >> 8;
 403         check_block(b, p);
 404         if (!get_block (b, p->pos, p->buf, 1))
 405         {
 406             yaz_log (b->log_io, "bf_write: close_block");
 407             bf_write (b->file[p->cat].bf, p->pos/4, 0, 0, p->buf);
 408         }
 409     }
 410     (*b->method->code_stop)(ISAMC_DECODE, p->decodeClientData);
 411     xfree (p->buf);
 412     xfree (p);
 413 }
 414
 415 int insert_sub (ISAMB b, struct ISAMB_block **p,
 416                 void *new_item, int *mode,
 417                 ISAMC_I stream,
 418                 struct ISAMB_block **sp,
 419                 void *sub_item, int *sub_size,
 420                 void *max_item);
 421
 422 int insert_int (ISAMB b, struct ISAMB_block *p, void *lookahead_item,
 423                 int *mode,
 424                 ISAMC_I stream, struct ISAMB_block **sp,
 425                 void *split_item, int *split_size, void *last_max_item)
 426 {
 427     char *startp = p->bytes;
 428     char *src = startp;
 429     char *endp = p->bytes + p->size;
 430     int pos;
 431     struct ISAMB_block *sub_p1 = 0, *sub_p2 = 0;
 432     char sub_item[DST_ITEM_MAX];
 433     int sub_size;
 434     int more;
 435
 436     *sp = 0;
 437
 438     assert(p->size >= 0);
 439     decode_ptr (&src, &pos);
 440     while (src != endp)
 441     {
 442         int item_len;
 443         int d;
 444         char *src0 = src;
 445         decode_ptr (&src, &item_len);
 446         d = (*b->method->compare_item)(src, lookahead_item);
 447         if (d > 0)
 448         {
 449             sub_p1 = open_block (b, pos);
 450             assert (sub_p1);
 451             more = insert_sub (b, &sub_p1, lookahead_item, mode,
 452                                stream, &sub_p2,
 453                                sub_item, &sub_size, src);
 454             src = src0;
 455             break;
 456         }
 457         src += item_len;
 458         decode_ptr (&src, &pos);
 459     }
 460     if (!sub_p1)
 461     {
 462         sub_p1 = open_block (b, pos);
 463         assert (sub_p1);
 464         more = insert_sub (b, &sub_p1, lookahead_item, mode, stream, &sub_p2,
 465                            sub_item, &sub_size, last_max_item);
 466     }
 467     if (sub_p2)
 468     {
 469         /* there was a split - must insert pointer in this one */
 470         char dst_buf[DST_BUF_SIZE];
 471         char *dst = dst_buf;
 472
 473         assert (sub_size < 30 && sub_size > 1);
 474
 475         memcpy (dst, startp, src - startp);
 476
 477         dst += src - startp;
 478
 479         encode_ptr (&dst, sub_size);      /* sub length and item */
 480         memcpy (dst, sub_item, sub_size);
 481         dst += sub_size;
 482
 483         encode_ptr (&dst, sub_p2->pos);   /* pos */
 484
 485         if (endp - src)                   /* remaining data */
 486         {
 487             memcpy (dst, src, endp - src);
 488             dst += endp - src;
 489         }
 490         p->size = dst - dst_buf;
 491         assert (p->size >= 0);
 492         if (p->size <= b->file[p->cat].head.block_max)
 493         {
 494             memcpy (startp, dst_buf, dst - dst_buf);
 495         }
 496         else
 497         {
 498             int p_new_size;
 499             char *half;
 500             src = dst_buf;
 501             endp = dst;
 502
 503             half = src + b->file[p->cat].head.block_size/2;
 504             decode_ptr (&src, &pos);
 505             while (src <= half)
 506             {
 507                 decode_ptr (&src, split_size);
 508                 src += *split_size;
 509                 decode_ptr (&src, &pos);
 510             }
 511             p_new_size = src - dst_buf;
 512             memcpy (p->bytes, dst_buf, p_new_size);
 513
 514             decode_ptr (&src, split_size);
 515             memcpy (split_item, src, *split_size);
 516             src += *split_size;
 517
 518             *sp = new_int (b, p->cat);
 519             (*sp)->size = endp - src;
 520             memcpy ((*sp)->bytes, src, (*sp)->size);
 521
 522             p->size = p_new_size;
 523         }
 524         p->dirty = 1;
 525         close_block (b, sub_p2);
 526     }
 527     close_block (b, sub_p1);
 528     return more;
 529 }
 530
 531
 532 int insert_leaf (ISAMB b, struct ISAMB_block **sp1, void *lookahead_item,
 533                  int *lookahead_mode, ISAMC_I stream, struct ISAMB_block **sp2,
 534                  void *sub_item, int *sub_size,
 535                  void *max_item)
 536 {
 537     struct ISAMB_block *p = *sp1;
 538     char *src = 0, *endp = 0;
 539     char dst_buf[DST_BUF_SIZE], *dst = dst_buf;
 540     int new_size;
 541     void *c1 = (*b->method->code_start)(ISAMC_DECODE);
 542     void *c2 = (*b->method->code_start)(ISAMC_ENCODE);
 543     int more = 1;
 544     int quater = b->file[b->no_cat-1].head.block_max / 4;
 545     char *cut = dst_buf + quater * 2;
 546     char *maxp = dst_buf + b->file[b->no_cat-1].head.block_max;
 547     char *half1 = 0;
 548     char *half2 = 0;
 549     char cut_item_buf[DST_ITEM_MAX];
 550     int cut_item_size = 0;
 551
 552     if (p && p->size)
 553     {
 554         char file_item_buf[DST_ITEM_MAX];
 555         char *file_item = file_item_buf;
 556
 557         src = p->bytes;
 558         endp = p->bytes + p->size;
 559         (*b->method->code_item)(ISAMC_DECODE, c1, &file_item, &src);
 560         while (1)
 561         {
 562             char *dst_item = 0;
 563             char *dst_0 = dst;
 564             char *lookahead_next;
 565             int d = -1;
 566
 567             if (lookahead_item)
 568                 d = (*b->method->compare_item)(file_item_buf, lookahead_item);
 569
 570             if (d > 0)
 571             {
 572                 dst_item = lookahead_item;
 573                 assert (*lookahead_mode);
 574             }
 575             else
 576                 dst_item = file_item_buf;
 577             if (!*lookahead_mode && d == 0)
 578             {
 579                 p->dirty = 1;
 580             }
 581             else if (!half1 && dst > cut)
 582             {
 583                 char *dst_item_0 = dst_item;
 584                 half1 = dst; /* candidate for splitting */
 585
 586                 (*b->method->code_item)(ISAMC_ENCODE, c2, &dst, &dst_item);
 587
 588                 cut_item_size = dst_item - dst_item_0;
 589                 memcpy (cut_item_buf, dst_item_0, cut_item_size);
 590
 591                 half2 = dst;
 592             }
 593             else
 594                 (*b->method->code_item)(ISAMC_ENCODE, c2, &dst, &dst_item);
 595             if (d > 0)
 596             {
 597                 if (dst > maxp)
 598                 {
 599                     dst = dst_0;
 600                     lookahead_item = 0;
 601                 }
 602                 else
 603                 {
 604                     lookahead_next = lookahead_item;
 605                     if (!(*stream->read_item)(stream->clientData,
 606                                               &lookahead_next,
 607                                               lookahead_mode))
 608                     {
 609                         lookahead_item = 0;
 610                         more = 0;
 611                     }
 612                     if (lookahead_item && max_item &&
 613                         (*b->method->compare_item)(max_item, lookahead_item) <= 0)
 614                     {
 615                         /* max_item 1 */
 616                         lookahead_item = 0;
 617                     }
 618
 619                     p->dirty = 1;
 620                 }
 621             }
 622             else if (d == 0)
 623             {
 624                 lookahead_next = lookahead_item;
 625                 if (!(*stream->read_item)(stream->clientData,
 626                                           &lookahead_next, lookahead_mode))
 627                 {
 628                     lookahead_item = 0;
 629                     more = 0;
 630                 }
 631                 if (src == endp)
 632                     break;
 633                 file_item = file_item_buf;
 634                 (*b->method->code_item)(ISAMC_DECODE, c1, &file_item, &src);
 635             }
 636             else
 637             {
 638                 if (src == endp)
 639                     break;
 640                 file_item = file_item_buf;
 641                 (*b->method->code_item)(ISAMC_DECODE, c1, &file_item, &src);
 642             }
 643         }
 644     }
 645     maxp = dst_buf + b->file[b->no_cat-1].head.block_max + quater;
 646     while (lookahead_item)
 647     {
 648         char *dst_item = lookahead_item;
 649         char *dst_0 = dst;
 650
 651         if (max_item &&
 652             (*b->method->compare_item)(max_item, lookahead_item) <= 0)
 653         {
 654             /* max_item 2 */
 655             break;
 656         }
 657         if (!*lookahead_mode)
 658         {
 659             yaz_log (LOG_WARN, "Inconsistent register (2)");
 660             abort();
 661         }
 662         else if (!half1 && dst > cut)
 663         {
 664             char *dst_item_0 = dst_item;
 665             half1 = dst; /* candidate for splitting */
 666
 667             (*b->method->code_item)(ISAMC_ENCODE, c2, &dst, &dst_item);
 668
 669             cut_item_size = dst_item - dst_item_0;
 670             memcpy (cut_item_buf, dst_item_0, cut_item_size);
 671
 672             half2 = dst;
 673         }
 674         else
 675             (*b->method->code_item)(ISAMC_ENCODE, c2, &dst, &dst_item);
 676
 677         if (dst > maxp)
 678         {
 679             dst = dst_0;
 680             break;
 681         }
 682         if (p)
 683             p->dirty = 1;
 684         dst_item = lookahead_item;
 685         if (!(*stream->read_item)(stream->clientData, &dst_item,
 686                                   lookahead_mode))
 687         {
 688             lookahead_item = 0;
 689             more = 0;
 690         }
 691     }
 692     new_size = dst - dst_buf;
 693     if (p && p->cat != b->no_cat-1 &&
 694         new_size > b->file[p->cat].head.block_max)
 695     {
 696         /* non-btree block will be removed */
 697         p->deleted = 1;
 698         close_block (b, p);
 699         /* delete it too!! */
 700         p = 0; /* make a new one anyway */
 701     }
 702     if (!p)
 703     {   /* must create a new one */
 704         int i;
 705         for (i = 0; i < b->no_cat; i++)
 706             if (new_size <= b->file[i].head.block_max)
 707                 break;
 708         if (i == b->no_cat)
 709             i = b->no_cat - 1;
 710         p = new_leaf (b, i);
 711     }
 712     if (new_size > b->file[p->cat].head.block_max)
 713     {
 714         char *first_dst;
 715         char *cut_item = cut_item_buf;
 716
 717         assert (half1);
 718         assert (half2);
 719
 720        /* first half */
 721         p->size = half1 - dst_buf;
 722         memcpy (p->bytes, dst_buf, half1 - dst_buf);
 723
 724         /* second half */
 725         *sp2 = new_leaf (b, p->cat);
 726
 727         (*b->method->code_reset)(c2);
 728
 729         first_dst = (*sp2)->bytes;
 730
 731         (*b->method->code_item)(ISAMC_ENCODE, c2, &first_dst, &cut_item);
 732
 733         memcpy (first_dst, half2, dst - half2);
 734
 735         (*sp2)->size = (first_dst - (*sp2)->bytes) + (dst - half2);
 736         (*sp2)->dirty = 1;
 737         p->dirty = 1;
 738         memcpy (sub_item, cut_item_buf, cut_item_size);
 739         *sub_size = cut_item_size;
 740     }
 741     else
 742     {
 743         memcpy (p->bytes, dst_buf, dst - dst_buf);
 744         p->size = new_size;
 745     }
 746     (*b->method->code_stop)(ISAMC_DECODE, c1);
 747     (*b->method->code_stop)(ISAMC_ENCODE, c2);
 748     *sp1 = p;
 749     return more;
 750 }
 751
 752 int insert_sub (ISAMB b, struct ISAMB_block **p, void *new_item,
 753                 int *mode,
 754                 ISAMC_I stream,
 755                 struct ISAMB_block **sp,
 756                 void *sub_item, int *sub_size,
 757                 void *max_item)
 758 {
 759     if (!*p || (*p)->leaf)
 760         return insert_leaf (b, p, new_item, mode, stream, sp, sub_item,
 761                             sub_size, max_item);
 762     else
 763         return insert_int (b, *p, new_item, mode, stream, sp, sub_item,
 764                            sub_size, max_item);
 765 }
 766
 767 int isamb_merge (ISAMB b, ISAMC_P pos, ISAMC_I stream)
 768 {
 769     char item_buf[DST_ITEM_MAX];
 770     char *item_ptr;
 771     int i_mode;
 772     int more;
 773
 774     if (b->cache < 0)
 775     {
 776         int more = 1;
 777         while (more)
 778         {
 779             item_ptr = item_buf;
 780             more =
 781                 (*stream->read_item)(stream->clientData, &item_ptr, &i_mode);
 782         }
 783         return 1;
 784     }
 785     item_ptr = item_buf;
 786     more = (*stream->read_item)(stream->clientData, &item_ptr, &i_mode);
 787     while (more)
 788     {
 789         struct ISAMB_block *p = 0, *sp = 0;
 790         char sub_item[DST_ITEM_MAX];
 791         int sub_size;
 792
 793         if (pos)
 794             p = open_block (b, pos);
 795         more = insert_sub (b, &p, item_buf, &i_mode, stream, &sp,
 796                             sub_item, &sub_size, 0);
 797         if (sp)
 798         {    /* increase level of tree by one */
 799             struct ISAMB_block *p2 = new_int (b, p->cat);
 800             char *dst = p2->bytes + p2->size;
 801
 802             encode_ptr (&dst, p->pos);
 803             assert (sub_size < 20);
 804             encode_ptr (&dst, sub_size);
 805             memcpy (dst, sub_item, sub_size);
 806             dst += sub_size;
 807             encode_ptr (&dst, sp->pos);
 808
 809             p2->size = dst - p2->bytes;
 810             pos = p2->pos;  /* return new super page */
 811             close_block (b, sp);
 812             close_block (b, p2);
 813         }
 814         else
 815             pos = p->pos;   /* return current one (again) */
 816         close_block (b, p);
 817     }
 818     return pos;
 819 }
 820
 821 ISAMB_PP isamb_pp_open_x (ISAMB isamb, ISAMB_P pos, int *level)
 822 {
 823     ISAMB_PP pp = xmalloc (sizeof(*pp));
 824
 825     pp->isamb = isamb;
 826     pp->block = xmalloc (10 * sizeof(*pp->block));
 827
 828     pp->pos = pos;
 829     pp->level = 0;
 830     pp->total_size = 0;
 831     pp->no_blocks = 0;
 832     while (1)
 833     {
 834         struct ISAMB_block *p = open_block (isamb, pos);
 835         char *src = p->bytes + p->offset;
 836         pp->block[pp->level] = p;
 837
 838         pp->total_size += p->size;
 839         pp->no_blocks++;
 840
 841         if (p->leaf)
 842             break;
 843
 844         decode_ptr (&src, &pos);
 845         p->offset = src - p->bytes;
 846         pp->level++;
 847     }
 848     pp->block[pp->level+1] = 0;
 849     if (level)
 850         *level = pp->level;
 851     return pp;
 852 }
 853
 854 ISAMB_PP isamb_pp_open (ISAMB isamb, ISAMB_P pos)
 855 {
 856     return isamb_pp_open_x (isamb, pos, 0);
 857 }
 858
 859 void isamb_pp_close_x (ISAMB_PP pp, int *size, int *blocks)
 860 {
 861     int i;
 862     if (!pp)
 863         return;
 864     if (size)
 865         *size = pp->total_size;
 866     if (blocks)
 867         *blocks = pp->no_blocks;
 868     for (i = 0; i <= pp->level; i++)
 869         close_block (pp->isamb, pp->block[i]);
 870     xfree (pp->block);
 871     xfree (pp);
 872 }
 873
 874 int isamb_block_info (ISAMB isamb, int cat)
 875 {
 876     if (cat >= 0 && cat < isamb->no_cat)
 877         return isamb->file[cat].head.block_size;
 878     return -1;
 879 }
 880
 881 void isamb_pp_close (ISAMB_PP pp)
 882 {
 883     isamb_pp_close_x (pp, 0, 0);
 884 }
 885
 886 int isamb_pp_read (ISAMB_PP pp, void *buf)
 887 {
 888     char *dst = buf;
 889     char *src;
 890     struct ISAMB_block *p = pp->block[pp->level];
 891     if (!p)
 892         return 0;
 893
 894     while (p->offset == p->size)
 895     {
 896         int pos, item_len;
 897         while (p->offset == p->size)
 898         {
 899             if (pp->level == 0)
 900                 return 0;
 901             close_block (pp->isamb, pp->block[pp->level]);
 902             pp->block[pp->level] = 0;
 903             (pp->level)--;
 904             p = pp->block[pp->level];
 905             assert (!p->leaf);  /* must be int */
 906         }
 907         src = p->bytes + p->offset;
 908
 909         decode_ptr (&src, &item_len);
 910         src += item_len;
 911         decode_ptr (&src, &pos);
 912
 913         p->offset = src - (char*) p->bytes;
 914
 915         ++(pp->level);
 916
 917         while (1)
 918         {
 919             pp->block[pp->level] = p = open_block (pp->isamb, pos);
 920
 921             pp->total_size += p->size;
 922             pp->no_blocks++;
 923
 924             if (p->leaf) /* leaf */
 925             {
 926                 break;
 927             }
 928             src = p->bytes + p->offset;
 929             decode_ptr (&src, &pos);
 930             p->offset = src - (char*) p->bytes;
 931             pp->level++;
 932         }
 933     }
 934     assert (p->offset < p->size);
 935     assert (p->leaf);
 936     src = p->bytes + p->offset;
 937     (*pp->isamb->method->code_item)(ISAMC_DECODE, p->decodeClientData,
 938                                     &dst, &src);
 939     p->offset = src - (char*) p->bytes;
 940     return 1;
 941 }
 942
 943 int isamb_pp_num (ISAMB_PP pp)
 944 {
 945     return 1;
 946 }