index/zrpn.c

   1 /*
   2  * Copyright (C) 1994-1995, Index Data I/S
   3  * All rights reserved.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Log: zrpn.c,v $
   7  * Revision 1.36  1995-12-06 12:41:27  adam
   8  * New command 'stat' for the index program.
   9  * Filenames can be read from stdin by specifying '-'.
  10  * Bug fix/enhancement of the transformation from terms to regular
  11  * expressons in the search engine.
  12  *
  13  * Revision 1.35  1995/11/27  09:29:00  adam
  14  * Bug fixes regarding conversion to regular expressions.
  15  *
  16  * Revision 1.34  1995/11/16  17:00:56  adam
  17  * Better logging of rpn query.
  18  *
  19  * Revision 1.33  1995/11/01  13:58:28  quinn
  20  * Moving data1 to yaz/retrieval
  21  *
  22  * Revision 1.32  1995/10/27  14:00:11  adam
  23  * Implemented detection of database availability.
  24  *
  25  * Revision 1.31  1995/10/17  18:02:10  adam
  26  * New feature: databases. Implemented as prefix to words in dictionary.
  27  *
  28  * Revision 1.30  1995/10/16  09:32:38  adam
  29  * More work on relational op.
  30  *
  31  * Revision 1.29  1995/10/13  16:01:49  adam
  32  * Work on relations.
  33  *
  34  * Revision 1.28  1995/10/13  12:26:43  adam
  35  * Optimization of truncation.
  36  *
  37  * Revision 1.27  1995/10/12  17:07:22  adam
  38  * Truncation works.
  39  *
  40  * Revision 1.26  1995/10/12  12:40:54  adam
  41  * Bug fixes in rpn_prox.
  42  *
  43  * Revision 1.25  1995/10/10  13:59:24  adam
  44  * Function rset_open changed its wflag parameter to general flags.
  45  *
  46  * Revision 1.24  1995/10/09  16:18:37  adam
  47  * Function dict_lookup_grep got extra client data parameter.
  48  *
  49  * Revision 1.23  1995/10/06  16:33:37  adam
  50  * Use attribute mappings.
  51  *
  52  * Revision 1.22  1995/10/06  15:07:39  adam
  53  * Structure 'local-number' handled.
  54  *
  55  * Revision 1.21  1995/10/06  13:52:06  adam
  56  * Bug fixes. Handler may abort further scanning.
  57  *
  58  * Revision 1.20  1995/10/06  11:06:33  adam
  59  * Scan entries include 'occurrences' now.
  60  *
  61  * Revision 1.19  1995/10/06  10:43:56  adam
  62  * Scan added. 'occurrences' in scan entries not set yet.
  63  *
  64  * Revision 1.18  1995/10/04  16:57:20  adam
  65  * Key input and merge sort in one pass.
  66  *
  67  * Revision 1.17  1995/10/04  12:55:17  adam
  68  * Bug fix in ranked search. Use=Any keys inserted.
  69  *
  70  * Revision 1.16  1995/10/02  16:24:40  adam
  71  * Use attribute actually used in search requests.
  72  *
  73  * Revision 1.15  1995/10/02  15:18:52  adam
  74  * New member in recRetrieveCtrl: diagnostic.
  75  *
  76  * Revision 1.14  1995/09/28  12:10:32  adam
  77  * Bug fixes. Field prefix used in queries.
  78  *
  79  * Revision 1.13  1995/09/18  14:17:50  adam
  80  * Minor changes.
  81  *
  82  * Revision 1.12  1995/09/15  14:45:21  adam
  83  * Retrieve control.
  84  * Work on truncation.
  85  *
  86  * Revision 1.11  1995/09/14  11:53:27  adam
  87  * First work on regular expressions/truncations.
  88  *
  89  * Revision 1.10  1995/09/11  15:23:26  adam
  90  * More work on relevance search.
  91  *
  92  * Revision 1.9  1995/09/11  13:09:35  adam
  93  * More work on relevance feedback.
  94  *
  95  * Revision 1.8  1995/09/08  14:52:27  adam
  96  * Minor changes. Dictionary is lower case now.
  97  *
  98  * Revision 1.7  1995/09/07  13:58:36  adam
  99  * New parameter: result-set file descriptor (RSFD) to support multiple
 100  * positions within the same result-set.
 101  * Boolean operators: and, or, not implemented.
 102  * Result-set references.
 103  *
 104  * Revision 1.6  1995/09/06  16:11:18  adam
 105  * Option: only one word key per file.
 106  *
 107  * Revision 1.5  1995/09/06  10:33:04  adam
 108  * More work on present. Some log messages removed.
 109  *
 110  * Revision 1.4  1995/09/05  15:28:40  adam
 111  * More work on search engine.
 112  *
 113  * Revision 1.3  1995/09/04  15:20:22  adam
 114  * Minor changes.
 115  *
 116  * Revision 1.2  1995/09/04  12:33:43  adam
 117  * Various cleanup. YAZ util used instead.
 118  *
 119  * Revision 1.1  1995/09/04  09:10:40  adam
 120  * More work on index add/del/update.
 121  * Merge sort implemented.
 122  * Initial work on z39 server.
 123  *
 124  */
 125 #include <stdio.h>
 126 #include <assert.h>
 127 #include <unistd.h>
 128 #include <ctype.h>
 129
 130 #include "zserver.h"
 131 #include "attribute.h"
 132
 133 #include <rsisam.h>
 134 #include <rstemp.h>
 135 #include <rsnull.h>
 136 #include <rsbool.h>
 137 #include <rsrel.h>
 138
 139 int index_word_prefix_map (char *string, oid_value attrSet, int attrUse,
 140                            char *basename)
 141 {
 142     attent *attp;
 143
 144     logf (LOG_DEBUG, "oid_value attrSet = %d, attrUse = %d", attrSet, attrUse);
 145     attp = att_getentbyatt (attrSet, attrUse);
 146     if (!attp)
 147         return -1;
 148     logf (LOG_DEBUG, "ord=%d", attp->attset_ordinal);
 149     return index_word_prefix (string, attp->attset_ordinal,
 150                               attp->local_attribute, basename);
 151 }
 152
 153 typedef struct {
 154     int type;
 155     int major;
 156     int minor;
 157     Z_AttributesPlusTerm *zapt;
 158 } AttrType;
 159
 160 static int attr_find (AttrType *src, oid_value *attributeSetP)
 161 {
 162     while (src->major < src->zapt->num_attributes)
 163     {
 164         Z_AttributeElement *element;
 165
 166         element = src->zapt->attributeList[src->major];
 167         if (src->type == *element->attributeType)
 168         {
 169             switch (element->which)
 170             {
 171             case Z_AttributeValue_numeric:
 172                 ++(src->major);
 173                 if (element->attributeSet && attributeSetP)
 174                 {
 175                     oident *attrset;
 176
 177                     attrset = oid_getentbyoid (element->attributeSet);
 178                     *attributeSetP = attrset->value;
 179                 }
 180                 return *element->value.numeric;
 181                 break;
 182             case Z_AttributeValue_complex:
 183                 if (src->minor >= element->value.complex->num_list ||
 184                     element->value.complex->list[src->minor]->which !=
 185                     Z_StringOrNumeric_numeric)
 186                     break;
 187                 ++(src->minor);
 188                 if (element->attributeSet && attributeSetP)
 189                 {
 190                     oident *attrset;
 191
 192                     attrset = oid_getentbyoid (element->attributeSet);
 193                     *attributeSetP = attrset->value;
 194                 }
 195                 return *element->value.complex->list[src->minor-1]->u.numeric;
 196             default:
 197                 assert (0);
 198             }
 199         }
 200         ++(src->major);
 201     }
 202     return -1;
 203 }
 204
 205 static void attr_init (AttrType *src, Z_AttributesPlusTerm *zapt,
 206                        int type)
 207 {
 208     src->zapt = zapt;
 209     src->type = type;
 210     src->major = 0;
 211     src->minor = 0;
 212 }
 213
 214 struct trunc_info {
 215     int  *ptr;
 216     int  *indx;
 217     char **heap;
 218     int  heapnum;
 219     int  (*cmp)(const void *p1, const void *p2);
 220     int  keysize;
 221     char *swapbuf;
 222     char *tmpbuf;
 223     char *buf;
 224 };
 225
 226 static void heap_swap (struct trunc_info *ti, int i1, int i2)
 227 {
 228     int swap;
 229
 230     swap = ti->ptr[i1];
 231     ti->ptr[i1] = ti->ptr[i2];
 232     ti->ptr[i2] = swap;
 233 }
 234
 235 static void heap_delete (struct trunc_info *ti)
 236 {
 237     int cur = 1, child = 2;
 238
 239     heap_swap (ti, 1, ti->heapnum--);
 240     while (child <= ti->heapnum) {
 241         if (child < ti->heapnum &&
 242             (*ti->cmp)(ti->heap[ti->ptr[child]],
 243                        ti->heap[ti->ptr[1+child]]) > 0)
 244             child++;
 245         if ((*ti->cmp)(ti->heap[ti->ptr[cur]],
 246                        ti->heap[ti->ptr[child]]) > 0)
 247         {
 248             heap_swap (ti, cur, child);
 249             cur = child;
 250             child = 2*cur;
 251         }
 252         else
 253             break;
 254     }
 255 }
 256
 257 static void heap_insert (struct trunc_info *ti, const char *buf, int indx)
 258 {
 259     int cur, parent;
 260
 261     cur = ++(ti->heapnum);
 262     memcpy (ti->heap[ti->ptr[cur]], buf, ti->keysize);
 263     ti->indx[ti->ptr[cur]] = indx;
 264     parent = cur/2;
 265     while (parent && (*ti->cmp)(ti->heap[ti->ptr[parent]],
 266                                 ti->heap[ti->ptr[cur]]) > 0)
 267     {
 268         heap_swap (ti, cur, parent);
 269         cur = parent;
 270         parent = cur/2;
 271     }
 272 }
 273
 274 static
 275 struct trunc_info *heap_init (int size, int key_size,
 276                               int (*cmp)(const void *p1, const void *p2))
 277 {
 278     struct trunc_info *ti = xmalloc (sizeof(*ti));
 279     int i;
 280
 281     ++size;
 282     ti->heapnum = 0;
 283     ti->keysize = key_size;
 284     ti->cmp = cmp;
 285     ti->indx = xmalloc (size * sizeof(*ti->indx));
 286     ti->heap = xmalloc (size * sizeof(*ti->heap));
 287     ti->ptr = xmalloc (size * sizeof(*ti->ptr));
 288     ti->swapbuf = xmalloc (ti->keysize);
 289     ti->tmpbuf = xmalloc (ti->keysize);
 290     ti->buf = xmalloc (size * ti->keysize);
 291     for (i = size; --i >= 0; )
 292     {
 293         ti->ptr[i] = i;
 294         ti->heap[i] = ti->buf + ti->keysize * i;
 295     }
 296     return ti;
 297 }
 298
 299 static void heap_close (struct trunc_info *ti)
 300 {
 301     xfree (ti->ptr);
 302     xfree (ti->indx);
 303     xfree (ti->heap);
 304     xfree (ti->swapbuf);
 305     xfree (ti->tmpbuf);
 306     xfree (ti);
 307 }
 308
 309 static RSET rset_trunc_r (ISAM isam, ISAM_P *isam_p, int from, int to,
 310                          int merge_chunk)
 311 {
 312     RSET result;
 313     RSFD result_rsfd;
 314     rset_temp_parms parms;
 315
 316     parms.key_size = sizeof(struct it_key);
 317     result = rset_create (rset_kind_temp, &parms);
 318     result_rsfd = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
 319
 320     if (to - from > merge_chunk)
 321     {
 322         RSFD *rsfd;
 323         RSET *rset;
 324         int i, i_add = (to-from)/merge_chunk + 1;
 325         struct trunc_info *ti;
 326         int rscur = 0;
 327         int rsmax = (to-from)/i_add + 1;
 328
 329         rset = xmalloc (sizeof(*rset) * rsmax);
 330         rsfd = xmalloc (sizeof(*rsfd) * rsmax);
 331
 332         for (i = from; i < to; i += i_add)
 333         {
 334             if (i_add <= to - i)
 335                 rset[rscur] = rset_trunc_r (isam, isam_p, i, i+i_add,
 336                                             merge_chunk);
 337             else
 338                 rset[rscur] = rset_trunc_r (isam, isam_p, i, to,
 339                                             merge_chunk);
 340             rscur++;
 341         }
 342         ti = heap_init (rscur, sizeof(struct it_key), key_compare);
 343         for (i = rscur; --i >= 0; )
 344         {
 345             rsfd[i] = rset_open (rset[i], RSETF_READ|RSETF_SORT_SYSNO);
 346             if (rset_read (rset[i], rsfd[i], ti->tmpbuf))
 347                 heap_insert (ti, ti->tmpbuf, i);
 348             else
 349             {
 350                 rset_close (rset[i], rsfd[i]);
 351                 rset_delete (rset[i]);
 352             }
 353         }
 354         while (ti->heapnum)
 355         {
 356             int n = ti->indx[ti->ptr[1]];
 357
 358             rset_write (result, result_rsfd, ti->heap[ti->ptr[1]]);
 359
 360             while (1)
 361             {
 362                 if (!rset_read (rset[n], rsfd[n], ti->tmpbuf))
 363                 {
 364                     heap_delete (ti);
 365                     rset_close (rset[n], rsfd[n]);
 366                     rset_delete (rset[n]);
 367                     break;
 368                 }
 369                 if ((*ti->cmp)(ti->tmpbuf, ti->heap[ti->ptr[1]]) > 1)
 370                 {
 371                     heap_delete (ti);
 372                     heap_insert (ti, ti->tmpbuf, n);
 373                     break;
 374                 }
 375             }
 376         }
 377         xfree (rset);
 378         xfree (rsfd);
 379         heap_close (ti);
 380     }
 381     else
 382     {
 383         ISPT *ispt;
 384         int i;
 385         struct trunc_info *ti;
 386
 387         ispt = xmalloc (sizeof(*ispt) * (to-from));
 388
 389         ti = heap_init (to-from, sizeof(struct it_key),
 390                         key_compare);
 391         for (i = to-from; --i >= 0; )
 392         {
 393             ispt[i] = is_position (isam, isam_p[from+i]);
 394             if (is_readkey (ispt[i], ti->tmpbuf))
 395                 heap_insert (ti, ti->tmpbuf, i);
 396             else
 397                 is_pt_free (ispt[i]);
 398         }
 399         while (ti->heapnum)
 400         {
 401             int n = ti->indx[ti->ptr[1]];
 402
 403             rset_write (result, result_rsfd, ti->heap[ti->ptr[1]]);
 404 #if 0
 405 /* section that preserve all keys */
 406             heap_delete (ti);
 407             if (is_readkey (ispt[n], ti->tmpbuf))
 408                 heap_insert (ti, ti->tmpbuf, n);
 409             else
 410                 is_pt_free (ispt[n]);
 411 #else
 412 /* section that preserve all keys with unique sysnos */
 413             while (1)
 414             {
 415                 if (!is_readkey (ispt[n], ti->tmpbuf))
 416                 {
 417                     heap_delete (ti);
 418                     is_pt_free (ispt[n]);
 419                     break;
 420                 }
 421                 if ((*ti->cmp)(ti->tmpbuf, ti->heap[ti->ptr[1]]) > 1)
 422                 {
 423                     heap_delete (ti);
 424                     heap_insert (ti, ti->tmpbuf, n);
 425                     break;
 426                 }
 427             }
 428 #endif
 429         }
 430         heap_close (ti);
 431         xfree (ispt);
 432     }
 433     rset_close (result, result_rsfd);
 434     return result;
 435 }
 436
 437 static int isam_trunc_cmp (const void *p1, const void *p2)
 438 {
 439     ISAM_P i1 = *(ISAM_P*) p1;
 440     ISAM_P i2 = *(ISAM_P*) p2;
 441     int d;
 442
 443     d = is_type (i1) - is_type (i2);
 444     if (d)
 445         return d;
 446     return is_block (i1) - is_block (i2);
 447 }
 448
 449 static RSET rset_trunc (ISAM isam, ISAM_P *isam_p, int no)
 450 {
 451
 452     qsort (isam_p, no, sizeof(*isam_p), isam_trunc_cmp);
 453     return rset_trunc_r (isam, isam_p, 0, no, 100);
 454 }
 455
 456 struct grep_info {
 457     ISAM_P *isam_p_buf;
 458     int isam_p_size;
 459     int isam_p_indx;
 460 };
 461
 462 static void add_isam_p (const char *info, struct grep_info *p)
 463 {
 464     if (p->isam_p_indx == p->isam_p_size)
 465     {
 466         ISAM_P *new_isam_p_buf;
 467
 468         p->isam_p_size = 2*p->isam_p_size + 100;
 469         new_isam_p_buf = xmalloc (sizeof(*new_isam_p_buf) *
 470                                   p->isam_p_size);
 471         if (p->isam_p_buf)
 472         {
 473             memcpy (new_isam_p_buf, p->isam_p_buf,
 474                     p->isam_p_indx * sizeof(*p->isam_p_buf));
 475             xfree (p->isam_p_buf);
 476         }
 477         p->isam_p_buf = new_isam_p_buf;
 478     }
 479     assert (*info == sizeof(*p->isam_p_buf));
 480     memcpy (p->isam_p_buf + p->isam_p_indx, info+1, sizeof(*p->isam_p_buf));
 481     (p->isam_p_indx)++;
 482 }
 483
 484 static int grep_handle (Dict_char *name, const char *info, void *p)
 485 {
 486     logf (LOG_DEBUG, "dict name: %s", name);
 487     add_isam_p (info, p);
 488     return 0;
 489 }
 490
 491 static void gen_regular_rel (char *dst, int val, int islt)
 492 {
 493     int dst_p = 1;
 494     int w, d, i;
 495     int pos = 0;
 496     char numstr[20];
 497
 498     *dst = '(';
 499     sprintf (numstr, "%d", val);
 500     for (w = strlen(numstr); --w >= 0; pos++)
 501     {
 502         d = numstr[w];
 503         if (pos > 0)
 504         {
 505             if (islt)
 506             {
 507                 if (d == '0')
 508                     continue;
 509                 d--;
 510             }
 511             else
 512             {
 513                 if (d == '9')
 514                     continue;
 515                 d++;
 516             }
 517         }
 518
 519         strcpy (dst + dst_p, numstr);
 520         dst_p = strlen(dst) - pos - 1;
 521
 522         if (islt)
 523         {
 524             if (d != '0')
 525             {
 526                 dst[dst_p++] = '[';
 527                 dst[dst_p++] = '0';
 528                 dst[dst_p++] = '-';
 529                 dst[dst_p++] = d;
 530                 dst[dst_p++] = ']';
 531             }
 532             else
 533                 dst[dst_p++] = d;
 534         }
 535         else
 536         {
 537             if (d != '9')
 538             {
 539                 dst[dst_p++] = '[';
 540                 dst[dst_p++] = d;
 541                 dst[dst_p++] = '-';
 542                 dst[dst_p++] = '9';
 543                 dst[dst_p++] = ']';
 544             }
 545             else
 546                 dst[dst_p++] = d;
 547         }
 548         for (i = 0; i<pos; i++)
 549         {
 550             dst[dst_p++] = '[';
 551             dst[dst_p++] = '0';
 552             dst[dst_p++] = '-';
 553             dst[dst_p++] = '9';
 554             dst[dst_p++] = ']';
 555         }
 556         dst[dst_p++] = '|';
 557     }
 558     dst[dst_p] = '\0';
 559     if (islt)
 560     {
 561         for (i=1; i<pos; i++)
 562             strcat (dst, "[0-9]?");
 563     }
 564     else
 565     {
 566         for (i = 0; i <= pos; i++)
 567             strcat (dst, "[0-9]");
 568         strcat (dst, "[0-9]*");
 569     }
 570     strcat (dst, ")");
 571 }
 572
 573 static int relational_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
 574                             const char *term_sub,
 575                             char *term_dict,
 576                             oid_value attributeSet,
 577                             struct grep_info *grep_info,
 578                             int *max_pos)
 579 {
 580     AttrType relation;
 581     int relation_value;
 582     int term_value;
 583     int r;
 584
 585     attr_init (&relation, zapt, 2);
 586     relation_value = attr_find (&relation, NULL);
 587     term_value = atoi (term_sub);
 588
 589     switch (relation_value)
 590     {
 591     case 1:
 592         if (term_value <= 0)
 593             return 1;
 594         logf (LOG_DEBUG, "Relation <");
 595         gen_regular_rel (term_dict + strlen(term_dict), term_value-1, 1);
 596         break;
 597     case 2:
 598         if (term_value < 0)
 599             return 1;
 600         logf (LOG_DEBUG, "Relation <=");
 601         gen_regular_rel (term_dict + strlen(term_dict), term_value, 1);
 602         break;
 603     case 4:
 604         if (term_value < 0)
 605             term_value = 0;
 606         logf (LOG_DEBUG, "Relation >=");
 607         gen_regular_rel (term_dict + strlen(term_dict), term_value, 0);
 608         break;
 609     case 5:
 610         if (term_value < 0)
 611             term_value = 0;
 612         logf (LOG_DEBUG, "Relation >");
 613         gen_regular_rel (term_dict + strlen(term_dict), term_value+1, 0);
 614         break;
 615     default:
 616         return 0;
 617     }
 618     logf (LOG_DEBUG, "dict_lookup_grep: %s", term_dict);
 619     r = dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, max_pos,
 620                           grep_handle);
 621     if (r)
 622         logf (LOG_WARN, "dict_lookup_grep fail, rel=gt: %d", r);
 623     logf (LOG_DEBUG, "%d positions", grep_info->isam_p_indx);
 624     return 1;
 625 }
 626
 627 static void verbatim_char (int ch, int *indx, char *dst)
 628 {
 629     if (!isalnum (ch))
 630         dst[(*indx)++] = '\\';
 631     dst[(*indx)++] = ch;
 632 }
 633
 634 static int trunc_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
 635                        const char *term_sub,
 636                        oid_value attributeSet, struct grep_info *grep_info,
 637                        int num_bases, char **basenames)
 638 {
 639     char term_dict[2*IT_MAX_WORD+2];
 640     int i, j, r, base_no;
 641     AttrType truncation;
 642     int truncation_value;
 643     AttrType use;
 644     int use_value;
 645     oid_value curAttributeSet = attributeSet;
 646
 647     attr_init (&use, zapt, 1);
 648     use_value = attr_find (&use, &curAttributeSet);
 649     logf (LOG_DEBUG, "use value %d", use_value);
 650     attr_init (&truncation, zapt, 5);
 651     truncation_value = attr_find (&truncation, NULL);
 652     logf (LOG_DEBUG, "truncation value %d", truncation_value);
 653
 654     if (use_value == -1)
 655         use_value = 1016;
 656
 657     for (base_no = 0; base_no < num_bases; base_no++)
 658     {
 659         int max_pos;
 660         int prefix_len = index_word_prefix_map (term_dict, curAttributeSet,
 661                                                 use_value,
 662                                                 basenames[base_no]);
 663         if (prefix_len < 0)
 664         {
 665             zi->errCode = 114;
 666             return -1;
 667         }
 668         if (!relational_term (zi, zapt, term_sub, term_dict,
 669                               attributeSet, grep_info, &max_pos))
 670         {
 671             switch (truncation_value)
 672             {
 673             case -1:         /* not specified */
 674             case 100:        /* do not truncate */
 675                 j = strlen(term_dict);
 676                 term_dict[j++] = '(';
 677                 for (i = 0; term_sub[i]; i++)
 678                     verbatim_char (term_sub[i], &j, term_dict);
 679                 strcpy (term_dict+j, ")");
 680                 r = dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info,
 681                                       &max_pos, grep_handle);
 682                 if (r)
 683                     logf (LOG_WARN, "dict_lookup_grep err, trunc=none:%d", r);
 684                 break;
 685             case 1:          /* right truncation */
 686                 j = strlen(term_dict);
 687                 term_dict[j++] = '(';
 688                 for (i = 0; term_sub[i]; i++)
 689                     verbatim_char (term_sub[i], &j, term_dict);
 690                 strcpy (term_dict+j, ".*)");
 691                 dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info,
 692                                   &max_pos, grep_handle);
 693                 break;
 694             case 2:          /* left truncation */
 695             case 3:          /* left&right truncation */
 696                 zi->errCode = 120;
 697                 return -1;
 698             case 101:        /* process # in term */
 699                 j = strlen(term_dict);
 700                 term_dict[j++] = '(';
 701                 for (i=0; term_sub[i]; i++)
 702                     if (term_sub[i] == '#' && i > 2)
 703                     {
 704                         term_dict[j++] = '.';
 705                         term_dict[j++] = '*';
 706                     }
 707                     else
 708                         verbatim_char (term_sub[i], &j, term_dict);
 709                 strcpy (term_dict+j, ")");
 710                 r = dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info,
 711                                       &max_pos, grep_handle);
 712                 if (r)
 713                     logf (LOG_WARN, "dict_lookup_grep err, trunc=#: %d",
 714                           r);
 715                 break;
 716             case 102:        /* regular expression */
 717                 sprintf (term_dict + strlen(term_dict), "(%s)", term_sub);
 718                 r = dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info,
 719                                       &max_pos, grep_handle);
 720                 if (r)
 721                     logf (LOG_WARN, "dict_lookup_grep err, trunc=regular: %d",
 722                           r);
 723                 break;
 724             }
 725         }
 726         if (max_pos <= strlen(basenames[base_no]))
 727         {
 728             zi->errCode = 109; /* Database unavailable */
 729             zi->errString = basenames[base_no];
 730             return -1;
 731         }
 732     }
 733     logf (LOG_DEBUG, "%d positions", grep_info->isam_p_indx);
 734     return 0;
 735 }
 736
 737 static void trans_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
 738                         char *termz)
 739 {
 740     size_t i, sizez;
 741     Z_Term *term = zapt->term;
 742
 743     sizez = term->u.general->len;
 744     if (sizez > IT_MAX_WORD)
 745         sizez = IT_MAX_WORD;
 746     for (i = 0; i < sizez; i++)
 747         termz[i] = index_char_cvt (term->u.general->buf[i]);
 748     termz[i] = '\0';
 749 }
 750
 751 static RSET rpn_search_APT_relevance (ZServerInfo *zi,
 752                                       Z_AttributesPlusTerm *zapt,
 753                                       oid_value attributeSet,
 754                                       int num_bases, char **basenames)
 755 {
 756     rset_relevance_parms parms;
 757     char termz[IT_MAX_WORD+1];
 758     char term_sub[IT_MAX_WORD+1];
 759     struct grep_info grep_info;
 760     char *p0 = termz, *p1 = NULL;
 761     RSET result;
 762
 763     parms.key_size = sizeof(struct it_key);
 764     parms.max_rec = 100;
 765     parms.cmp = key_compare;
 766     parms.is = zi->wordIsam;
 767
 768     if (zapt->term->which != Z_Term_general)
 769     {
 770         zi->errCode = 124;
 771         return NULL;
 772     }
 773     trans_term (zi, zapt, termz);
 774     grep_info.isam_p_indx = 0;
 775     grep_info.isam_p_size = 0;
 776     grep_info.isam_p_buf = NULL;
 777     while (1)
 778     {
 779         if ((p1 = strchr (p0, ' ')))
 780         {
 781             memcpy (term_sub, p0, p1-p0);
 782             term_sub[p1-p0] = '\0';
 783         }
 784         else
 785             strcpy (term_sub, p0);
 786         if (trunc_term (zi, zapt, term_sub, attributeSet, &grep_info,
 787                         num_bases, basenames))
 788             return NULL;
 789         if (!p1)
 790             break;
 791         p0 = p1;
 792         while (*++p0 == ' ')
 793             ;
 794     }
 795     parms.isam_positions = grep_info.isam_p_buf;
 796     parms.no_isam_positions = grep_info.isam_p_indx;
 797     if (grep_info.isam_p_indx > 0)
 798         result = rset_create (rset_kind_relevance, &parms);
 799     else
 800         result = rset_create (rset_kind_null, NULL);
 801     xfree (grep_info.isam_p_buf);
 802     return result;
 803 }
 804
 805 static RSET rpn_search_APT_word (ZServerInfo *zi,
 806                                  Z_AttributesPlusTerm *zapt,
 807                                  oid_value attributeSet,
 808                                  int num_bases, char **basenames)
 809 {
 810     rset_isam_parms parms;
 811     char termz[IT_MAX_WORD+1];
 812     struct grep_info grep_info;
 813     RSET result;
 814
 815     if (zapt->term->which != Z_Term_general)
 816     {
 817         zi->errCode = 124;
 818         return NULL;
 819     }
 820     trans_term (zi, zapt, termz);
 821
 822     grep_info.isam_p_indx = 0;
 823     grep_info.isam_p_size = 0;
 824     grep_info.isam_p_buf = NULL;
 825
 826     if (trunc_term (zi, zapt, termz, attributeSet, &grep_info,
 827                     num_bases, basenames))
 828         return NULL;
 829     if (grep_info.isam_p_indx < 1)
 830         result = rset_create (rset_kind_null, NULL);
 831     else if (grep_info.isam_p_indx == 1)
 832     {
 833         parms.is = zi->wordIsam;
 834         parms.pos = *grep_info.isam_p_buf;
 835         result = rset_create (rset_kind_isam, &parms);
 836     }
 837     else
 838         result = rset_trunc (zi->wordIsam, grep_info.isam_p_buf,
 839                              grep_info.isam_p_indx);
 840     xfree (grep_info.isam_p_buf);
 841     return result;
 842 }
 843
 844 static RSET rpn_prox (RSET *rset, int rset_no)
 845 {
 846     int i;
 847     RSFD *rsfd;
 848     int  *more;
 849     struct it_key **buf;
 850     RSFD rsfd_result;
 851     RSET result;
 852     rset_temp_parms parms;
 853
 854     rsfd = xmalloc (sizeof(*rsfd)*rset_no);
 855     more = xmalloc (sizeof(*more)*rset_no);
 856     buf = xmalloc (sizeof(*buf)*rset_no);
 857
 858     for (i = 0; i<rset_no; i++)
 859     {
 860         buf[i] = xmalloc (sizeof(**buf));
 861         rsfd[i] = rset_open (rset[i], RSETF_READ|RSETF_SORT_SYSNO);
 862         if (!(more[i] = rset_read (rset[i], rsfd[i], buf[i])))
 863         {
 864             while (i >= 0)
 865             {
 866                 rset_close (rset[i], rsfd[i]);
 867                 xfree (buf[i]);
 868                 --i;
 869             }
 870             xfree (rsfd);
 871             xfree (more);
 872             xfree (buf);
 873             return rset_create (rset_kind_null, NULL);
 874         }
 875     }
 876     parms.key_size = sizeof (struct it_key);
 877     result = rset_create (rset_kind_temp, &parms);
 878     rsfd_result = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
 879
 880     while (*more)
 881     {
 882         for (i = 1; i<rset_no; i++)
 883         {
 884             int cmp;
 885
 886             if (!more[i])
 887             {
 888                 *more = 0;
 889                 break;
 890             }
 891             cmp = key_compare (buf[i], buf[i-1]);
 892             if (cmp > 1)
 893             {
 894                 more[i-1] = rset_read (rset[i-1], rsfd[i-1], buf[i-1]);
 895                 break;
 896             }
 897             else if (cmp == 1)
 898             {
 899                 if (buf[i-1]->seqno+1 != buf[i]->seqno)
 900                 {
 901                     more[i-1] = rset_read (rset[i-1], rsfd[i-1], buf[i-1]);
 902                     break;
 903                 }
 904             }
 905             else
 906             {
 907                 more[i] = rset_read (rset[i], rsfd[i], buf[i]);
 908                 break;
 909             }
 910         }
 911         if (i == rset_no)
 912         {
 913             rset_write (result, rsfd_result, buf[0]);
 914             more[0] = rset_read (*rset, *rsfd, *buf);
 915         }
 916     }
 917
 918     for (i = 0; i<rset_no; i++)
 919     {
 920         rset_close (rset[i], rsfd[i]);
 921         xfree (buf[i]);
 922     }
 923     rset_close (result, rsfd_result);
 924     xfree (buf);
 925     xfree (more);
 926     xfree (rsfd);
 927     return result;
 928 }
 929
 930 static RSET rpn_search_APT_phrase (ZServerInfo *zi,
 931                                    Z_AttributesPlusTerm *zapt,
 932                                    oid_value attributeSet,
 933                                    int num_bases, char **basenames)
 934 {
 935     char termz[IT_MAX_WORD+1];
 936     char term_sub[IT_MAX_WORD+1];
 937     char *p0 = termz, *p1 = NULL;
 938     RSET rset[60], result;
 939     int i, rset_no = 0;
 940     struct grep_info grep_info;
 941
 942     if (zapt->term->which != Z_Term_general)
 943     {
 944         zi->errCode = 124;
 945         return NULL;
 946     }
 947     trans_term (zi, zapt, termz);
 948
 949     grep_info.isam_p_size = 0;
 950     grep_info.isam_p_buf = NULL;
 951
 952     while (1)
 953     {
 954         if ((p1 = strchr (p0, ' ')))
 955         {
 956             memcpy (term_sub, p0, p1-p0);
 957             term_sub[p1-p0] = '\0';
 958         }
 959         else
 960             strcpy (term_sub, p0);
 961
 962         grep_info.isam_p_indx = 0;
 963         if (trunc_term (zi, zapt, term_sub, attributeSet, &grep_info,
 964                         num_bases, basenames))
 965             return NULL;
 966         if (grep_info.isam_p_indx == 0)
 967             rset[rset_no] = rset_create (rset_kind_null, NULL);
 968         else if (grep_info.isam_p_indx > 1)
 969             rset[rset_no] = rset_trunc (zi->wordIsam,
 970                                         grep_info.isam_p_buf,
 971                                         grep_info.isam_p_indx);
 972         else
 973         {
 974             rset_isam_parms parms;
 975
 976             parms.is = zi->wordIsam;
 977             parms.pos = *grep_info.isam_p_buf;
 978             rset[rset_no] = rset_create (rset_kind_isam, &parms);
 979         }
 980         assert (rset[rset_no]);
 981         if (++rset_no >= sizeof(rset)/sizeof(*rset))
 982             break;
 983         if (!p1)
 984             break;
 985         p0 = p1;
 986         while (*++p0 == ' ')
 987             ;
 988     }
 989     xfree (grep_info.isam_p_buf);
 990     if (rset_no == 0)
 991         return rset_create (rset_kind_null, NULL);
 992     else if (rset_no == 1)
 993         return (rset[0]);
 994
 995     result = rpn_prox (rset, rset_no);
 996     for (i = 0; i<rset_no; i++)
 997         rset_delete (rset[i]);
 998     return result;
 999 }
1000
1001 static RSET rpn_search_APT_local (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
1002                                   oid_value attributeSet)
1003 {
1004     RSET result;
1005     RSFD rsfd;
1006     struct it_key key;
1007     rset_temp_parms parms;
1008     char termz[IT_MAX_WORD+1];
1009
1010     if (zapt->term->which != Z_Term_general)
1011     {
1012         zi->errCode = 124;
1013         return NULL;
1014     }
1015     parms.key_size = sizeof (struct it_key);
1016     result = rset_create (rset_kind_temp, &parms);
1017     rsfd = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
1018
1019     trans_term (zi, zapt, termz);
1020     key.sysno = atoi (termz);
1021     if (key.sysno <= 0)
1022         key.sysno = 1;
1023     rset_write (result, rsfd, &key);
1024     rset_close (result, rsfd);
1025     return result;
1026 }
1027
1028 static RSET rpn_search_APT (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
1029                             oid_value attributeSet,
1030                             int num_bases, char **basenames)
1031 {
1032     AttrType relation;
1033     AttrType structure;
1034     int relation_value, structure_value;
1035
1036     attr_init (&relation, zapt, 2);
1037     attr_init (&structure, zapt, 4);
1038
1039     relation_value = attr_find (&relation, NULL);
1040     structure_value = attr_find (&structure, NULL);
1041     switch (structure_value)
1042     {
1043     case -1:
1044         if (relation_value == 102) /* relevance relation */
1045             return rpn_search_APT_relevance (zi, zapt, attributeSet,
1046                                              num_bases, basenames);
1047         return rpn_search_APT_phrase (zi, zapt, attributeSet,
1048                                       num_bases, basenames);
1049     case 1: /* phrase */
1050         if (relation_value == 102) /* relevance relation */
1051             return rpn_search_APT_relevance (zi, zapt, attributeSet,
1052                                              num_bases, basenames);
1053         return rpn_search_APT_phrase (zi, zapt, attributeSet,
1054                                       num_bases, basenames);
1055         break;
1056     case 2: /* word */
1057         if (relation_value == 102) /* relevance relation */
1058             return rpn_search_APT_relevance (zi, zapt, attributeSet,
1059                                              num_bases, basenames);
1060         return rpn_search_APT_word (zi, zapt, attributeSet,
1061                                     num_bases, basenames);
1062     case 3: /* key */
1063         break;
1064     case 4: /* year */
1065         break;
1066     case 5: /* date - normalized */
1067         break;
1068     case 6: /* word list */
1069         return rpn_search_APT_relevance (zi, zapt, attributeSet,
1070                                          num_bases, basenames);
1071     case 100: /* date - un-normalized */
1072         break;
1073     case 101: /* name - normalized */
1074         break;
1075     case 102: /* date - un-normalized */
1076         break;
1077     case 103: /* structure */
1078         break;
1079     case 104: /* urx */
1080         break;
1081     case 105: /* free-form-text */
1082         return rpn_search_APT_relevance (zi, zapt, attributeSet,
1083                                          num_bases, basenames);
1084     case 106: /* document-text */
1085         return rpn_search_APT_relevance (zi, zapt, attributeSet,
1086                                          num_bases, basenames);
1087     case 107: /* local-number */
1088         return rpn_search_APT_local (zi, zapt, attributeSet);
1089     case 108: /* string */
1090         return rpn_search_APT_word (zi, zapt, attributeSet,
1091                                     num_bases, basenames);
1092     case 109: /* numeric string */
1093         break;
1094     }
1095     zi->errCode = 118;
1096     return NULL;
1097 }
1098
1099 static RSET rpn_search_ref (ZServerInfo *zi, Z_ResultSetId *resultSetId)
1100 {
1101     ZServerSet *s;
1102
1103     if (!(s = resultSetGet (zi, resultSetId)))
1104         return rset_create (rset_kind_null, NULL);
1105     return s->rset;
1106 }
1107
1108 static RSET rpn_search_structure (ZServerInfo *zi, Z_RPNStructure *zs,
1109                                   oid_value attributeSet,
1110                                   int num_bases, char **basenames)
1111 {
1112     RSET r = NULL;
1113     if (zs->which == Z_RPNStructure_complex)
1114     {
1115         rset_bool_parms bool_parms;
1116
1117         bool_parms.rset_l = rpn_search_structure (zi, zs->u.complex->s1,
1118                                                   attributeSet,
1119                                                   num_bases, basenames);
1120         if (bool_parms.rset_l == NULL)
1121             return NULL;
1122         bool_parms.rset_r = rpn_search_structure (zi, zs->u.complex->s2,
1123                                                   attributeSet,
1124                                                   num_bases, basenames);
1125         if (bool_parms.rset_r == NULL)
1126         {
1127             rset_delete (bool_parms.rset_l);
1128             return NULL;
1129         }
1130         bool_parms.key_size = sizeof(struct it_key);
1131         bool_parms.cmp = key_compare;
1132
1133         switch (zs->u.complex->operator->which)
1134         {
1135         case Z_Operator_and:
1136             r = rset_create (rset_kind_and, &bool_parms);
1137             break;
1138         case Z_Operator_or:
1139             r = rset_create (rset_kind_or, &bool_parms);
1140             break;
1141         case Z_Operator_and_not:
1142             r = rset_create (rset_kind_not, &bool_parms);
1143             break;
1144         default:
1145             assert (0);
1146         }
1147     }
1148     else if (zs->which == Z_RPNStructure_simple)
1149     {
1150         if (zs->u.simple->which == Z_Operand_APT)
1151         {
1152             logf (LOG_DEBUG, "rpn_search_APT");
1153             r = rpn_search_APT (zi, zs->u.simple->u.attributesPlusTerm,
1154                                 attributeSet, num_bases, basenames);
1155         }
1156         else if (zs->u.simple->which == Z_Operand_resultSetId)
1157         {
1158             logf (LOG_DEBUG, "rpn_search_ref");
1159             r = rpn_search_ref (zi, zs->u.simple->u.resultSetId);
1160         }
1161         else
1162         {
1163             assert (0);
1164         }
1165     }
1166     else
1167     {
1168         assert (0);
1169     }
1170     return r;
1171 }
1172
1173 static void count_set (RSET r, int *count)
1174 {
1175     int psysno = 0;
1176     struct it_key key;
1177     RSFD rfd;
1178
1179     logf (LOG_DEBUG, "rpn_save_set");
1180     *count = 0;
1181     rfd = rset_open (r, RSETF_READ|RSETF_SORT_SYSNO);
1182     while (rset_read (r, rfd, &key))
1183     {
1184         if (key.sysno != psysno)
1185         {
1186             psysno = key.sysno;
1187             (*count)++;
1188         }
1189     }
1190     rset_close (r, rfd);
1191     logf (LOG_DEBUG, "%d distinct sysnos", *count);
1192 }
1193
1194 int rpn_search (ZServerInfo *zi,
1195                 Z_RPNQuery *rpn, int num_bases, char **basenames,
1196                 const char *setname, int *hits)
1197 {
1198     RSET rset;
1199     oident *attrset;
1200     oid_value attributeSet;
1201
1202     zlog_rpn (rpn);
1203
1204     zi->errCode = 0;
1205     zi->errString = NULL;
1206
1207     attrset = oid_getentbyoid (rpn->attributeSetId);
1208     attributeSet = attrset->value;
1209     rset = rpn_search_structure (zi, rpn->RPNStructure, attributeSet,
1210                                  num_bases, basenames);
1211     if (!rset)
1212         return zi->errCode;
1213     count_set (rset, hits);
1214     resultSetAdd (zi, setname, 1, rset);
1215     if (zi->errCode)
1216         logf (LOG_DEBUG, "search error: %d", zi->errCode);
1217     return zi->errCode;
1218 }
1219
1220 struct scan_info {
1221     struct scan_entry *list;
1222     ODR odr;
1223     int before, after;
1224     ISAM isam;
1225     char prefix[20];
1226 };
1227
1228 static int scan_handle (Dict_char *name, const char *info, int pos,
1229                         void *client)
1230 {
1231     int len_prefix, idx;
1232     ISAM_P isam_p;
1233     RSET rset;
1234     struct scan_info *scan_info = client;
1235
1236     rset_isam_parms parms;
1237
1238     len_prefix = strlen(scan_info->prefix);
1239     if (memcmp (name, scan_info->prefix, len_prefix))
1240         return 1;
1241     if (pos > 0)
1242         idx = scan_info->after - pos + scan_info->before;
1243     else
1244         idx = - pos - 1;
1245     scan_info->list[idx].term = odr_malloc (scan_info->odr,
1246                                             strlen(name + len_prefix)+1);
1247     strcpy (scan_info->list[idx].term, name + len_prefix);
1248     assert (*info == sizeof(isam_p));
1249     memcpy (&isam_p, info+1, sizeof(isam_p));
1250     parms.is = scan_info->isam;
1251     parms.pos = isam_p;
1252 #if 1
1253     rset = rset_create (rset_kind_isam, &parms);
1254     count_set (rset, &scan_info->list[idx].occurrences);
1255     rset_delete (rset);
1256 #else
1257     scan_info->list[idx].occurrences = 1;
1258 #endif
1259     logf (LOG_DEBUG, "pos=%3d idx=%3d name=%s", pos, idx, name);
1260     return 0;
1261 }
1262
1263
1264 static int dummy_handle (Dict_char *name, const char *info, void *p)
1265 {
1266     return 0;
1267 }
1268
1269 int rpn_scan (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
1270               int num_bases, char **basenames,
1271               int *position, int *num_entries, struct scan_entry **list,
1272               int *status)
1273 {
1274     int i, j, sizez, max_pos;
1275     int pos = *position;
1276     int num = *num_entries;
1277     int before;
1278     int after;
1279     char termz[IT_MAX_WORD+20];
1280     AttrType use;
1281     int use_value;
1282     Z_Term *term = zapt->term;
1283     struct scan_info scan_info;
1284
1285     logf (LOG_DEBUG, "scan, position = %d, num = %d", pos, num);
1286
1287     if (num_bases != 1)
1288         return 111;
1289     scan_info.before = before = pos-1;
1290     scan_info.after = after = 1+num-pos;
1291     scan_info.odr = zi->odr;
1292
1293     logf (LOG_DEBUG, "scan, before = %d, after = %d", before, after);
1294
1295     scan_info.isam = zi->wordIsam;
1296     scan_info.list = odr_malloc (zi->odr, (before+after)*
1297                                  sizeof(*scan_info.list));
1298     for (j = 0; j<before+after; j++)
1299         scan_info.list[j].term = NULL;
1300     attr_init (&use, zapt, 1);
1301     use_value = attr_find (&use, NULL);
1302     logf (LOG_DEBUG, "use value %d", use_value);
1303
1304     if (use_value == -1)
1305         use_value = 1016;
1306     i = index_word_prefix (termz, 1, use_value, *basenames);
1307
1308     dict_lookup_grep (zi->wordDict, termz, 0, NULL, &max_pos,
1309                       dummy_handle);
1310     if (max_pos <= strlen(*basenames))
1311     {
1312         zi->errString = *basenames;
1313         return zi->errCode = 109; /* Database unavailable */
1314     }
1315     strcpy (scan_info.prefix, termz);
1316     sizez = term->u.general->len;
1317     if (sizez > IT_MAX_WORD)
1318         sizez = IT_MAX_WORD;
1319     for (j = 0; j<sizez; j++)
1320         termz[j+i] = index_char_cvt (term->u.general->buf[j]);
1321     termz[j+i] = '\0';
1322
1323     dict_scan (zi->wordDict, termz, &before, &after, &scan_info, scan_handle);
1324
1325     *status = BEND_SCAN_SUCCESS;
1326
1327     for (i = 0; i<scan_info.after; i++)
1328         if (scan_info.list[scan_info.before+scan_info.after-i-1].term)
1329             break;
1330     *num_entries -= i;
1331     if (i)
1332         *status = BEND_SCAN_PARTIAL;
1333
1334     for (i = 0; i<scan_info.before; i++)
1335         if (scan_info.list[i].term)
1336             break;
1337     if (i)
1338         *status = BEND_SCAN_PARTIAL;
1339     *position -= i;
1340     *num_entries -= i;
1341
1342     *list = scan_info.list+i;       /* list is set to first 'real' entry */
1343
1344     if (*num_entries == 0)          /* signal 'unsupported use-attribute' */
1345         zi->errCode = 114;          /* if no entries was found */
1346     logf (LOG_DEBUG, "position = %d, num_entries = %d",
1347           *position, *num_entries);
1348     if (zi->errCode)
1349         logf (LOG_DEBUG, "scan error: %d", zi->errCode);
1350     return 0;
1351 }
1352
1353
1354