isamb/isamb.c

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