index/retrieve.c

   1 /* $Id: retrieve.c,v 1.60 2006-11-29 09:01:53 marc Exp $
   2    Copyright (C) 1995-2006
   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 this program; if not, write to the Free Software
  19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  20
  21 */
  22
  23 #include <stdio.h>
  24 #include <assert.h>
  25
  26 #include <fcntl.h>
  27 #ifdef WIN32
  28 #include <io.h>
  29 #include <process.h>
  30 #endif
  31 #if HAVE_UNISTD_H
  32 #include <unistd.h>
  33 #endif
  34
  35 #include "index.h"
  36 #include <yaz/diagbib1.h>
  37 #include <direntz.h>
  38
  39
  40 #define ZEBRA_XML_HEADER_STR "<record xmlns=\"http://www.indexdata.com/zebra/\""
  41
  42 static int zebra_create_record_stream(ZebraHandle zh,
  43                                Record *rec,
  44                                struct ZebraRecStream *stream){
  45
  46     RecordAttr *recordAttr = rec_init_attr(zh->reg->zei, *rec);
  47
  48     if ((*rec)->size[recInfo_storeData] > 0)
  49         zebra_create_stream_mem(stream, (*rec)->info[recInfo_storeData],
  50                                 (*rec)->size[recInfo_storeData]);
  51     else
  52     {
  53         char full_rep[1024];
  54         int fd;
  55
  56         if (zh->path_reg && !yaz_is_abspath((*rec)->info[recInfo_filename])){
  57             strcpy(full_rep, zh->path_reg);
  58             strcat(full_rep, "/");
  59             strcat(full_rep, (*rec)->info[recInfo_filename]);
  60         }
  61         else
  62             strcpy(full_rep, (*rec)->info[recInfo_filename]);
  63
  64         if ((fd = open (full_rep, O_BINARY|O_RDONLY)) == -1){
  65             yaz_log (YLOG_WARN|YLOG_ERRNO, "Retrieve fail; missing file: %s",
  66                      full_rep);
  67             rec_free(rec);
  68             return 14;
  69         }
  70         zebra_create_stream_fd(stream, fd, recordAttr->recordOffset);
  71     }
  72     return 0;
  73 }
  74
  75
  76
  77 static int parse_zebra_elem(const char *elem,
  78                              const char **index, size_t *index_len,
  79                              const char **type, size_t *type_len)
  80 {
  81     *index = 0;
  82     *index_len = 0;
  83
  84     *type = 0;
  85     *type_len = 0;
  86
  87     if (elem && *elem)
  88     {
  89         char *cp;
  90         /* verify that '::' is in the beginning of *elem
  91            and something more follows */
  92         if (':' != *elem
  93             || !(elem +1) || ':' != *(elem +1)
  94             || !(elem +2) || '\0' == *(elem +2))
  95             return 0;
  96
  97         /* pick out info from string after '::' */
  98         elem = elem + 2;
  99         cp = strchr(elem, ':');
 100
 101         if (!cp) /* index, no colon, no type */
 102         {
 103             *index = elem;
 104             *index_len = strlen(elem);
 105         }
 106         else if (cp[1] == '\0') /* colon, but no following type */
 107         {
 108             return 0;
 109         }
 110         else  /* index, colon and type */
 111         {
 112             *index = elem;
 113             *index_len = cp - elem;
 114             *type = cp+1;
 115             *type_len = strlen(cp+1);
 116         }
 117     }
 118     return 1;
 119 }
 120
 121
 122 int zebra_special_index_fetch(ZebraHandle zh, zint sysno, ODR odr,
 123                               Record rec,
 124                               const char *elemsetname,
 125                               oid_value input_format,
 126                               oid_value *output_format,
 127                               char **rec_bufp, int *rec_lenp)
 128 {
 129     const char *retrieval_index;
 130     size_t retrieval_index_len;
 131     const char *retrieval_type;
 132     size_t retrieval_type_len;
 133     WRBUF wrbuf = 0;
 134     zebra_rec_keys_t keys;
 135
 136     /* set output variables before processing possible error states */
 137     /* *rec_lenp = 0; */
 138
 139     /* only accept XML and SUTRS requests */
 140     if (input_format != VAL_TEXT_XML
 141         && input_format != VAL_SUTRS){
 142         yaz_log(YLOG_WARN, "unsupported format for element set zebra::%s",
 143                 elemsetname);
 144         *output_format = VAL_NONE;
 145         return YAZ_BIB1_NO_SYNTAXES_AVAILABLE_FOR_THIS_REQUEST;
 146     }
 147
 148     if (!parse_zebra_elem(elemsetname,
 149                      &retrieval_index, &retrieval_index_len,
 150                      &retrieval_type,  &retrieval_type_len))
 151         return YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_;
 152
 153     if (retrieval_type_len != 0 && retrieval_type_len != 1)
 154     {
 155         return YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_;
 156     }
 157
 158     if (retrieval_index_len)
 159     {
 160         char retrieval_index_cstr[256];
 161
 162         if (retrieval_index_len  < sizeof(retrieval_index_cstr) -1)
 163         {
 164             memcpy(retrieval_index_cstr, retrieval_index, retrieval_index_len);
 165             retrieval_index_cstr[retrieval_index_len] = '\0';
 166
 167             if (zebraExplain_lookup_attr_str(zh->reg->zei,
 168                                              zinfo_index_category_index,
 169                                              (retrieval_type_len == 0 ? -1 :
 170                                               retrieval_type[0]),
 171                                              retrieval_index_cstr) == -1)
 172                 return YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_;
 173         }
 174     }
 175
 176     keys = zebra_rec_keys_open();
 177     zebra_rec_keys_set_buf(keys, rec->info[recInfo_delKeys],
 178                            rec->size[recInfo_delKeys], 0);
 179
 180     wrbuf = wrbuf_alloc();
 181     if (zebra_rec_keys_rewind(keys)){
 182         size_t slen;
 183         const char *str;
 184         struct it_key key_in;
 185
 186         if (input_format == VAL_TEXT_XML)
 187         {
 188             *output_format = VAL_TEXT_XML;
 189             wrbuf_printf(wrbuf, ZEBRA_XML_HEADER_STR
 190                          " sysno=\"" ZINT_FORMAT "\""
 191                          " set=\"zebra::index%s/\">\n",
 192                          sysno, elemsetname);
 193         }
 194         else if (input_format == VAL_SUTRS)
 195             *output_format = VAL_SUTRS;
 196
 197         while(zebra_rec_keys_read(keys, &str, &slen, &key_in)){
 198             int i;
 199             int ord = CAST_ZINT_TO_INT(key_in.mem[0]);
 200             int index_type;
 201             const char *db = 0;
 202             const char *string_index = 0;
 203             size_t string_index_len;
 204             char dst_buf[IT_MAX_WORD];
 205
 206             zebraExplain_lookup_ord(zh->reg->zei, ord, &index_type, &db,
 207                                     &string_index);
 208             string_index_len = strlen(string_index);
 209
 210             /* process only if index is not defined,
 211                or if defined and matching */
 212             if (retrieval_index == 0
 213                 || (string_index_len == retrieval_index_len
 214                     && !memcmp(string_index, retrieval_index,
 215                                string_index_len))){
 216
 217                 /* process only if type is not defined, or is matching */
 218                 if (retrieval_type == 0
 219                     || (retrieval_type_len == 1
 220                         && retrieval_type[0] == index_type)){
 221
 222
 223                     zebra_term_untrans(zh, index_type, dst_buf, str);
 224                     if (strlen(dst_buf)){
 225
 226                         if (input_format == VAL_TEXT_XML){
 227                             wrbuf_printf(wrbuf, "  <index name=\"%s\"",
 228                                          string_index);
 229
 230                             wrbuf_printf(wrbuf, " type=\"%c\"", index_type);
 231
 232                             wrbuf_printf(wrbuf, " seq=\"" ZINT_FORMAT "\">",
 233                                          key_in.mem[key_in.len -1]);
 234
 235                             wrbuf_xmlputs(wrbuf, dst_buf);
 236                             wrbuf_printf(wrbuf, "</index>\n");
 237                         }
 238                         else if (input_format == VAL_SUTRS){
 239                             wrbuf_printf(wrbuf, "%s ", string_index);
 240
 241                             wrbuf_printf(wrbuf, "%c", index_type);
 242
 243                             for (i = 1; i < key_in.len; i++)
 244                                 wrbuf_printf(wrbuf, " " ZINT_FORMAT,
 245                                              key_in.mem[i]);
 246
 247                         /* zebra_term_untrans(zh, index_type, dst_buf, str); */
 248                             wrbuf_printf(wrbuf, " %s", dst_buf);
 249
 250                             wrbuf_printf(wrbuf, "\n");
 251                         }
 252                     }
 253
 254                 }
 255             }
 256         }
 257         if (input_format == VAL_TEXT_XML)
 258             wrbuf_printf(wrbuf, "</record>\n");
 259      }
 260     *rec_lenp = wrbuf_len(wrbuf);
 261     *rec_bufp = odr_malloc(odr, *rec_lenp);
 262     memcpy(*rec_bufp, wrbuf_buf(wrbuf), *rec_lenp);
 263     wrbuf_free(wrbuf, 1);
 264     zebra_rec_keys_close(keys);
 265     return 0;
 266 }
 267
 268
 269 static void retrieve_puts_attr(WRBUF wrbuf, const char *name,
 270                                const char *value)
 271 {
 272     if (value)
 273     {
 274         wrbuf_printf(wrbuf, " %s=\"", name);
 275         wrbuf_xmlputs(wrbuf, value);
 276         wrbuf_printf(wrbuf, "\"");
 277     }
 278 }
 279
 280 static void retrieve_puts_attr_int(WRBUF wrbuf, const char *name,
 281                                const int value)
 282 {
 283     wrbuf_printf(wrbuf, " %s=\"%i\"", name, value);
 284 }
 285
 286 static void retrieve_puts_str(WRBUF wrbuf, const char *name,
 287                                const char *value)
 288 {
 289     if (value)
 290         wrbuf_printf(wrbuf, "%s %s\n", name, value);
 291 }
 292
 293 static void retrieve_puts_int(WRBUF wrbuf, const char *name,
 294                                const int value)
 295 {
 296     wrbuf_printf(wrbuf, "%s %i\n", name, value);
 297 }
 298
 299 int zebra_special_fetch(ZebraHandle zh, zint sysno, int score, ODR odr,
 300                            const char *elemsetname,
 301                            oid_value input_format,
 302                            oid_value *output_format,
 303                            char **rec_bufp, int *rec_lenp)
 304 {
 305     Record rec;
 306
 307     /* set output variables before processing possible error states */
 308     /* *rec_lenp = 0; */
 309
 310
 311
 312     /* processing zebra::meta::sysno elemset without fetching binary data */
 313     if (elemsetname && 0 == strcmp(elemsetname, "meta::sysno"))
 314     {
 315         int ret = 0;
 316         WRBUF wrbuf = wrbuf_alloc();
 317         if (input_format == VAL_SUTRS)
 318         {
 319             wrbuf_printf(wrbuf, ZINT_FORMAT, sysno);
 320             *output_format = VAL_SUTRS;
 321         }
 322         else if (input_format == VAL_TEXT_XML)
 323         {
 324             wrbuf_printf(wrbuf, ZEBRA_XML_HEADER_STR
 325                          " sysno=\"" ZINT_FORMAT "\"/>\n",
 326                          sysno);
 327             *output_format = VAL_TEXT_XML;
 328         }
 329         *rec_lenp = wrbuf_len(wrbuf);
 330         if (*rec_lenp)
 331             *rec_bufp = odr_strdup(odr, wrbuf_buf(wrbuf));
 332         else
 333             ret = YAZ_BIB1_NO_SYNTAXES_AVAILABLE_FOR_THIS_REQUEST;
 334         wrbuf_free(wrbuf, 1);
 335         return ret;
 336     }
 337
 338     /* fetching binary record up for all other display elementsets */
 339     rec = rec_get(zh->reg->records, sysno);
 340     if (!rec)
 341     {
 342         yaz_log(YLOG_WARN, "rec_get fail on sysno=" ZINT_FORMAT, sysno);
 343         return YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 344     }
 345
 346     /* processing special elementsetnames zebra::data */
 347     if (elemsetname && 0 == strcmp(elemsetname, "data"))
 348     {
 349         struct ZebraRecStream stream;
 350         RecordAttr *recordAttr = rec_init_attr(zh->reg->zei, rec);
 351         zebra_create_record_stream(zh, &rec, &stream);
 352         *output_format = input_format;
 353         *rec_lenp = recordAttr->recordSize;
 354         *rec_bufp = (char *) odr_malloc(odr, *rec_lenp);
 355         stream.readf(&stream, *rec_bufp, *rec_lenp);
 356         stream.destroy(&stream);
 357         rec_free(&rec);
 358         return 0;
 359     }
 360
 361     /* only accept XML and SUTRS requests from now */
 362     if (input_format != VAL_TEXT_XML && input_format != VAL_SUTRS)
 363     {
 364         yaz_log(YLOG_WARN, "unsupported format for element set zebra::%s",
 365                 elemsetname);
 366         return YAZ_BIB1_NO_SYNTAXES_AVAILABLE_FOR_THIS_REQUEST;
 367     }
 368
 369
 370     /* processing special elementsetnames zebra::meta:: */
 371     if (elemsetname && 0 == strcmp(elemsetname, "meta"))
 372     {
 373         int ret = 0;
 374         WRBUF wrbuf = wrbuf_alloc();
 375         RecordAttr *recordAttr = rec_init_attr(zh->reg->zei, rec);
 376
 377         if (input_format == VAL_TEXT_XML)
 378         {
 379             *output_format = VAL_TEXT_XML;
 380
 381             wrbuf_printf(wrbuf, ZEBRA_XML_HEADER_STR
 382                          " sysno=\"" ZINT_FORMAT "\"", sysno);
 383             retrieve_puts_attr(wrbuf, "base", rec->info[recInfo_databaseName]);
 384             retrieve_puts_attr(wrbuf, "file", rec->info[recInfo_filename]);
 385             retrieve_puts_attr(wrbuf, "type", rec->info[recInfo_fileType]);
 386             if (score >= 0)
 387                 retrieve_puts_attr_int(wrbuf, "score", score);
 388
 389             wrbuf_printf(wrbuf,
 390                          " rank=\"" ZINT_FORMAT "\""
 391                          " size=\"%i\""
 392                          " set=\"zebra::%s\"/>\n",
 393                          recordAttr->staticrank,
 394                          recordAttr->recordSize,
 395                          elemsetname);
 396         }
 397         else if (input_format == VAL_SUTRS)
 398         {
 399             *output_format = VAL_SUTRS;
 400             wrbuf_printf(wrbuf, "sysno " ZINT_FORMAT "\n", sysno);
 401             retrieve_puts_str(wrbuf, "base", rec->info[recInfo_databaseName]);
 402             retrieve_puts_str(wrbuf, "file", rec->info[recInfo_filename]);
 403             retrieve_puts_str(wrbuf, "type", rec->info[recInfo_fileType]);
 404             if (score >= 0)
 405                 retrieve_puts_int(wrbuf, "score", score);
 406
 407             wrbuf_printf(wrbuf,
 408                          "rank " ZINT_FORMAT "\n"
 409                          "size %i\n"
 410                          "set zebra::%s\n",
 411                          recordAttr->staticrank,
 412                          recordAttr->recordSize,
 413                          elemsetname);
 414         }
 415         *rec_lenp = wrbuf_len(wrbuf);
 416         if (*rec_lenp)
 417             *rec_bufp = odr_strdup(odr, wrbuf_buf(wrbuf));
 418         else
 419             ret = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 420
 421         wrbuf_free(wrbuf, 1);
 422         rec_free(&rec);
 423         return ret;
 424     }
 425
 426     /* processing special elementsetnames zebra::index:: */
 427     if (elemsetname && 0 == strncmp(elemsetname, "index", 5)){
 428
 429         int ret = zebra_special_index_fetch(zh, sysno, odr, rec,
 430                                             elemsetname + 5,
 431                                             input_format, output_format,
 432                                             rec_bufp, rec_lenp);
 433
 434         rec_free(&rec);
 435         return ret;
 436     }
 437
 438     if (rec)
 439         rec_free(&rec);
 440     return YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_;
 441 }
 442
 443
 444 int zebra_record_fetch(ZebraHandle zh, zint sysno, int score,
 445                        zebra_snippets *hit_snippet, ODR odr,
 446                        oid_value input_format, Z_RecordComposition *comp,
 447                        oid_value *output_format,
 448                        char **rec_bufp, int *rec_lenp, char **basenamep,
 449                        char **addinfo)
 450 {
 451     Record rec;
 452     char *fname, *file_type, *basename;
 453     const char *elemsetname;
 454     struct ZebraRecStream stream;
 455     RecordAttr *recordAttr;
 456     void *clientData;
 457     int return_code = 0;
 458
 459     *basenamep = 0;
 460     *addinfo = 0;
 461     elemsetname = yaz_get_esn(comp);
 462
 463     /* processing zebra special elementset names of form 'zebra:: */
 464     if (elemsetname && 0 == strncmp(elemsetname, "zebra::", 7))
 465         return  zebra_special_fetch(zh, sysno, score, odr,
 466                                     elemsetname + 7,
 467                                     input_format, output_format,
 468                                     rec_bufp, rec_lenp);
 469
 470
 471     /* processing all other element set names */
 472     rec = rec_get(zh->reg->records, sysno);
 473     if (!rec)
 474     {
 475         yaz_log(YLOG_WARN, "rec_get fail on sysno=" ZINT_FORMAT, sysno);
 476         *basenamep = 0;
 477         return YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 478     }
 479
 480
 481     recordAttr = rec_init_attr(zh->reg->zei, rec);
 482
 483     file_type = rec->info[recInfo_fileType];
 484     fname = rec->info[recInfo_filename];
 485     basename = rec->info[recInfo_databaseName];
 486     *basenamep = (char *) odr_malloc (odr, strlen(basename)+1);
 487     strcpy (*basenamep, basename);
 488
 489     yaz_log(YLOG_DEBUG, "retrieve localno=" ZINT_FORMAT " score=%d",
 490             sysno, score);
 491
 492     zebra_create_record_stream(zh, &rec, &stream);
 493
 494     {
 495         /* snippets code */
 496         zebra_snippets *snippet;
 497         zebra_rec_keys_t reckeys = zebra_rec_keys_open();
 498         RecType rt;
 499         struct recRetrieveCtrl retrieveCtrl;
 500
 501         retrieveCtrl.stream = &stream;
 502         retrieveCtrl.fname = fname;
 503         retrieveCtrl.localno = sysno;
 504         retrieveCtrl.staticrank = recordAttr->staticrank;
 505         retrieveCtrl.score = score;
 506         retrieveCtrl.recordSize = recordAttr->recordSize;
 507         retrieveCtrl.odr = odr;
 508         retrieveCtrl.input_format = retrieveCtrl.output_format = input_format;
 509         retrieveCtrl.comp = comp;
 510         retrieveCtrl.encoding = zh->record_encoding;
 511         retrieveCtrl.diagnostic = 0;
 512         retrieveCtrl.addinfo = 0;
 513         retrieveCtrl.dh = zh->reg->dh;
 514         retrieveCtrl.res = zh->res;
 515         retrieveCtrl.rec_buf = 0;
 516         retrieveCtrl.rec_len = -1;
 517         retrieveCtrl.hit_snippet = hit_snippet;
 518         retrieveCtrl.doc_snippet = zebra_snippets_create();
 519
 520         zebra_rec_keys_set_buf(reckeys,
 521                                rec->info[recInfo_delKeys],
 522                                rec->size[recInfo_delKeys],
 523                                0);
 524         zebra_rec_keys_to_snippets(zh, reckeys, retrieveCtrl.doc_snippet);
 525         zebra_rec_keys_close(reckeys);
 526
 527 #if 0
 528         /* for debugging purposes */
 529         yaz_log(YLOG_LOG, "DOC SNIPPET:");
 530         zebra_snippets_log(retrieveCtrl.doc_snippet, YLOG_LOG);
 531         yaz_log(YLOG_LOG, "HIT SNIPPET:");
 532         zebra_snippets_log(retrieveCtrl.hit_snippet, YLOG_LOG);
 533 #endif
 534         snippet = zebra_snippets_window(retrieveCtrl.doc_snippet,
 535                                         retrieveCtrl.hit_snippet,
 536                                         10);
 537 #if 0
 538         /* for debugging purposes */
 539         yaz_log(YLOG_LOG, "WINDOW SNIPPET:");
 540         zebra_snippets_log(snippet, YLOG_LOG);
 541 #endif
 542
 543         if (!(rt = recType_byName(zh->reg->recTypes, zh->res,
 544                                   file_type, &clientData)))
 545         {
 546             return_code = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 547         }
 548         else
 549         {
 550             (*rt->retrieve)(clientData, &retrieveCtrl);
 551             return_code = retrieveCtrl.diagnostic;
 552
 553             *output_format = retrieveCtrl.output_format;
 554             *rec_bufp = (char *) retrieveCtrl.rec_buf;
 555             *rec_lenp = retrieveCtrl.rec_len;
 556             *addinfo = retrieveCtrl.addinfo;
 557         }
 558
 559         zebra_snippets_destroy(snippet);
 560         zebra_snippets_destroy(retrieveCtrl.doc_snippet);
 561      }
 562
 563     stream.destroy(&stream);
 564     rec_free(&rec);
 565
 566     return return_code;
 567 }
 568
 569 /*
 570  * Local variables:
 571  * c-basic-offset: 4
 572  * indent-tabs-mode: nil
 573  * End:
 574  * vim: shiftwidth=4 tabstop=8 expandtab
 575  */
 576