index/zserver.c

   1 /* $Id: zserver.c,v 1.130 2005-03-08 14:02:12 adam Exp $
   2    Copyright (C) 1995-2005
   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 <assert.h>
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <fcntl.h>
  27 #ifdef WIN32
  28 #include <io.h>
  29 #include <process.h>
  30 #include <sys/locking.h>
  31 #else
  32 #include <unistd.h>
  33 #endif
  34
  35 #include <errno.h>
  36 #include <yaz/log.h>
  37 #include <yaz/ill.h>
  38 #include <yaz/yaz-util.h>
  39
  40 #include <sys/types.h>
  41
  42 #include "zserver.h"
  43
  44 static int bend_sort (void *handle, bend_sort_rr *rr);
  45 static int bend_delete (void *handle, bend_delete_rr *rr);
  46 static int bend_esrequest (void *handle, bend_esrequest_rr *rr);
  47 static int bend_segment (void *handle, bend_segment_rr *rr);
  48 static int bend_search (void *handle, bend_search_rr *r);
  49 static int bend_fetch (void *handle, bend_fetch_rr *r);
  50 static int bend_scan (void *handle, bend_scan_rr *r);
  51
  52 bend_initresult *bend_init (bend_initrequest *q)
  53 {
  54     bend_initresult *r = (bend_initresult *)
  55         odr_malloc (q->stream, sizeof(*r));
  56     ZebraHandle zh;
  57     struct statserv_options_block *sob;
  58     char *user = NULL;
  59     char *passwd = NULL;
  60
  61     r->errcode = 0;
  62     r->errstring = 0;
  63     q->bend_sort = bend_sort;
  64     q->bend_delete = bend_delete;
  65     q->bend_esrequest = bend_esrequest;
  66     q->bend_segment = bend_segment;
  67     q->bend_search = bend_search;
  68     q->bend_fetch = bend_fetch;
  69     q->bend_scan = bend_scan;
  70
  71     q->implementation_name = "Zebra Information Server";
  72     q->implementation_version = "Zebra " ZEBRAVER;
  73
  74     yaz_log (YLOG_DEBUG, "bend_init");
  75
  76     sob = statserv_getcontrol ();
  77     if (!(zh = zebra_open (sob->handle)))
  78     {
  79         yaz_log (YLOG_WARN, "Failed to read config `%s'", sob->configname);
  80         r->errcode = 1;
  81         return r;
  82     }
  83     r->handle = zh;
  84     if (q->auth)
  85     {
  86         if (q->auth->which == Z_IdAuthentication_open)
  87         {
  88             char *openpass = xstrdup (q->auth->u.open);
  89             char *cp = strchr (openpass, '/');
  90             if (cp)
  91             {
  92                 *cp = '\0';
  93                 user = nmem_strdup (odr_getmem (q->stream), openpass);
  94                 passwd = nmem_strdup (odr_getmem (q->stream), cp+1);
  95             }
  96             xfree (openpass);
  97         }
  98         else if (q->auth->which == Z_IdAuthentication_idPass)
  99         {
 100             Z_IdPass *idPass = q->auth->u.idPass;
 101
 102             user = idPass->userId;
 103             passwd = idPass->password;
 104         }
 105     }
 106     if (zebra_auth (zh, user, passwd))
 107     {
 108         r->errcode = 222;
 109         r->errstring = user;
 110         return r;
 111     }
 112     if (q->charneg_request) /* characater set and langauge negotiation? */
 113     {
 114         char **charsets = 0;
 115         int num_charsets;
 116         char **langs = 0;
 117         int num_langs = 0;
 118         int selected = 0;
 119         int i;
 120         NMEM nmem = nmem_create();
 121
 122         yaz_log (YLOG_LOG, "character set and language negotiation");
 123
 124         yaz_get_proposal_charneg (nmem, q->charneg_request,
 125                                   &charsets, &num_charsets,
 126                                   &langs, &num_langs, &selected);
 127
 128         for (i = 0; i < num_charsets; i++)
 129         {
 130             const char *right_name = "";
 131             /*
 132              * FIXME! It is like rudiment :-))
 133              * We have to support this short names of character sets,
 134              * because a lot servers in Russia to use own in during
 135              * character set and language negotiation still.
 136              */
 137
 138             if (!yaz_matchstr(charsets[i], "win")) {
 139                 right_name = "WINDOWS-1251";
 140             } else if (!yaz_matchstr(charsets[i], "koi")) {
 141                 right_name = "KOI8-R";
 142             } else if (!yaz_matchstr(charsets[i], "iso")) {
 143                 right_name = "ISO-8859-5";
 144             } else if (!yaz_matchstr(charsets[i], "dos")) {
 145                 right_name = "CP866";
 146             } else if (!yaz_matchstr(charsets[i], "uni")) {
 147                 right_name = "UTF-8";
 148             } else {
 149                 right_name = charsets[i];
 150             }
 151             if (odr_set_charset (q->decode, "UTF-8", right_name) == 0)
 152             {
 153                 yaz_log (YLOG_LOG, "charset %d %s (proper name %s): OK", i,
 154                          charsets[i], right_name);
 155                 odr_set_charset (q->stream, right_name, "UTF-8");
 156                 if (selected)
 157                     zebra_record_encoding(zh, right_name);
 158                 zebra_octet_term_encoding(zh, right_name);
 159                 q->charneg_response =
 160                     yaz_set_response_charneg (q->stream, charsets[i],
 161                                               0, selected);
 162                 break;
 163             } else {
 164                 yaz_log (YLOG_LOG, "charset %d %s (proper name %s): unsupported", i,
 165                          charsets[i], right_name);
 166             }
 167         }
 168         nmem_destroy(nmem);
 169     }
 170     return r;
 171 }
 172
 173 static void search_terms (ZebraHandle zh, bend_search_rr *r)
 174 {
 175     zint count = 0;
 176     int no_terms;
 177     int i;
 178     int type = Z_Term_general;
 179     struct Z_External *ext;
 180     Z_SearchInfoReport *sr;
 181
 182     /* get no of terms for result set */
 183     no_terms = 0; /* zebra_resultSetTerms (zh, r->setname, 0, 0, 0, 0, 0); */
 184       /* FIXME - Rsets don't know number of terms no more ??? */
 185     if (!no_terms)
 186         return;
 187
 188     r->search_info = odr_malloc (r->stream, sizeof(*r->search_info));
 189
 190     r->search_info->num_elements = 1;
 191     r->search_info->list =
 192         odr_malloc (r->stream, sizeof(*r->search_info->list));
 193     r->search_info->list[0] =
 194         odr_malloc (r->stream, sizeof(**r->search_info->list));
 195     r->search_info->list[0]->category = 0;
 196     r->search_info->list[0]->which = Z_OtherInfo_externallyDefinedInfo;
 197     ext = odr_malloc (r->stream, sizeof(*ext));
 198     r->search_info->list[0]->information.externallyDefinedInfo = ext;
 199     ext->direct_reference =
 200         yaz_oidval_to_z3950oid (r->stream, CLASS_USERINFO, VAL_SEARCHRES1);
 201     ext->indirect_reference = 0;
 202     ext->descriptor = 0;
 203     ext->which = Z_External_searchResult1;
 204     sr = odr_malloc (r->stream, sizeof(Z_SearchInfoReport));
 205     ext->u.searchResult1 = sr;
 206     sr->num = no_terms;
 207     sr->elements = odr_malloc (r->stream, sr->num *
 208                                sizeof(*sr->elements));
 209     for (i = 0; i<no_terms; i++)
 210     {
 211         Z_Term *term;
 212         char outbuf[1024];
 213         size_t len = sizeof(outbuf);
 214         /* FIXME - Can we just skip this ??? */
 215         /*
 216         zebra_resultSetTerms (zh, r->setname, i,
 217                               &count, &type, outbuf, &len);
 218         */
 219         sr->elements[i] = odr_malloc (r->stream, sizeof(**sr->elements));
 220         sr->elements[i]->subqueryId = 0;
 221         sr->elements[i]->fullQuery = odr_malloc (r->stream,
 222                                                  sizeof(bool_t));
 223         *sr->elements[i]->fullQuery = 0;
 224         sr->elements[i]->subqueryExpression =
 225             odr_malloc (r->stream, sizeof(Z_QueryExpression));
 226         sr->elements[i]->subqueryExpression->which =
 227             Z_QueryExpression_term;
 228         sr->elements[i]->subqueryExpression->u.term =
 229             odr_malloc (r->stream, sizeof(Z_QueryExpressionTerm));
 230         term = odr_malloc (r->stream, sizeof(Z_Term));
 231         sr->elements[i]->subqueryExpression->u.term->queryTerm = term;
 232         switch (type)
 233         {
 234         case Z_Term_characterString:
 235             yaz_log (YLOG_DEBUG, "term as characterString");
 236             term->which = Z_Term_characterString;
 237             term->u.characterString = odr_strdup (r->stream, outbuf);
 238             break;
 239         case Z_Term_general:
 240             yaz_log (YLOG_DEBUG, "term as general");
 241             term->which = Z_Term_general;
 242             term->u.general = odr_malloc (r->stream, sizeof(*term->u.general));
 243             term->u.general->size = term->u.general->len = len;
 244             term->u.general->buf = odr_malloc (r->stream, len);
 245             memcpy (term->u.general->buf, outbuf, len);
 246             break;
 247         default:
 248             term->which = Z_Term_general;
 249             term->u.null = odr_nullval();
 250         }
 251         sr->elements[i]->subqueryExpression->u.term->termComment = 0;
 252         sr->elements[i]->subqueryInterpretation = 0;
 253         sr->elements[i]->subqueryRecommendation = 0;
 254         if (count > 2000000000)
 255             count = 2000000000;
 256         sr->elements[i]->subqueryCount = odr_intdup (r->stream, (int) count);
 257         sr->elements[i]->subqueryWeight = 0;
 258         sr->elements[i]->resultsByDB = 0;
 259     }
 260 }
 261
 262 int bend_search (void *handle, bend_search_rr *r)
 263 {
 264     ZebraHandle zh = (ZebraHandle) handle;
 265     zint zhits = 0;
 266
 267     r->hits = 0;
 268     r->errcode = 0;
 269     r->errstring = NULL;
 270
 271     if (zebra_select_databases (zh, r->num_bases,
 272                                 (const char **) r->basenames))
 273     {
 274         zebra_result (zh, &r->errcode, &r->errstring);
 275         return 0;
 276     }
 277     yaz_log (YLOG_DEBUG, "ResultSet '%s'", r->setname);
 278     switch (r->query->which)
 279     {
 280     case Z_Query_type_1: case Z_Query_type_101:
 281         zebra_search_RPN (zh, r->stream, r->query->u.type_1,
 282                           r->setname, &zhits);
 283         if (zhits >   2147483646)
 284             r->hits = 2147483647;
 285         else
 286             r->hits = (int) zhits;
 287         zebra_result (zh, &r->errcode, &r->errstring);
 288         if (!r->errcode)
 289             search_terms (zh, r);
 290         break;
 291     case Z_Query_type_2:
 292         r->errcode = 107;
 293         r->errstring = "type-2";
 294         break;
 295     default:
 296         r->errcode = 107;
 297     }
 298     return 0;
 299 }
 300
 301
 302 int bend_fetch (void *handle, bend_fetch_rr *r)
 303 {
 304     ZebraHandle zh = (ZebraHandle) handle;
 305     ZebraRetrievalRecord retrievalRecord;
 306
 307     retrievalRecord.position = r->number;
 308
 309     r->last_in_set = 0;
 310     zebra_records_retrieve (zh, r->stream, r->setname, r->comp,
 311                             r->request_format, 1, &retrievalRecord);
 312     zebra_result (zh, &r->errcode, &r->errstring);
 313     /*  non Surrogate Diagnostic OR Surrogate Diagnostic */
 314     if (r->errcode == 0 && retrievalRecord.errCode)
 315     {
 316         r->surrogate_flag = 1;
 317         r->errcode = retrievalRecord.errCode;
 318         r->errstring = retrievalRecord.errString;
 319         r->basename = retrievalRecord.base;
 320     }
 321     else if (r->errcode == 0)        /* Database Record */
 322     {
 323         r->errcode = 0;
 324         r->basename = retrievalRecord.base;
 325         r->record = retrievalRecord.buf;
 326         r->len = retrievalRecord.len;
 327         r->output_format = retrievalRecord.format;
 328     }
 329     return 0;
 330 }
 331
 332 static int bend_scan (void *handle, bend_scan_rr *r)
 333 {
 334     ZebraScanEntry *entries;
 335     ZebraHandle zh = (ZebraHandle) handle;
 336     int is_partial, i;
 337
 338     if (zebra_select_databases (zh, r->num_bases,
 339                                 (const char **) r->basenames))
 340     {
 341         zebra_result (zh, &r->errcode, &r->errstring);
 342         return 0;
 343     }
 344     if (r->step_size != 0 && *r->step_size != 0) {
 345         r->errcode = 205; /* "Only zero step size supported for Scan" */
 346         r->errstring = 0;
 347         return 0;
 348     }
 349     r->entries = (struct scan_entry *)
 350         odr_malloc (r->stream, sizeof(*r->entries) * r->num_entries);
 351     zebra_scan (zh, r->stream, r->term,
 352                 r->attributeset,
 353                 &r->term_position,
 354                 &r->num_entries, &entries, &is_partial);
 355     if (is_partial)
 356         r->status = BEND_SCAN_PARTIAL;
 357     else
 358         r->status = BEND_SCAN_SUCCESS;
 359     for (i = 0; i < r->num_entries; i++)
 360     {
 361         r->entries[i].term = entries[i].term;
 362         r->entries[i].occurrences = entries[i].occurrences;
 363     }
 364     zebra_result (zh, &r->errcode, &r->errstring);
 365     return 0;
 366 }
 367
 368 void bend_close (void *handle)
 369 {
 370     zebra_close ((ZebraHandle) handle);
 371     xmalloc_trav("bend_close");
 372     nmem_print_list();
 373 }
 374
 375 int bend_sort (void *handle, bend_sort_rr *rr)
 376 {
 377     ZebraHandle zh = (ZebraHandle) handle;
 378
 379     zebra_sort (zh, rr->stream,
 380                 rr->num_input_setnames, (const char **) rr->input_setnames,
 381                 rr->output_setname, rr->sort_sequence, &rr->sort_status);
 382     zebra_result (zh, &rr->errcode, &rr->errstring);
 383     return 0;
 384 }
 385
 386 int bend_delete (void *handle, bend_delete_rr *rr)
 387 {
 388     ZebraHandle zh = (ZebraHandle) handle;
 389
 390     rr->delete_status = zebra_deleteResultSet(zh, rr->function,
 391                                               rr->num_setnames, rr->setnames,
 392                                               rr->statuses);
 393     return 0;
 394 }
 395
 396 static int es_admin_request (ZebraHandle zh, Z_AdminEsRequest *r)
 397 {
 398     if (r->toKeep->databaseName)
 399     {
 400         yaz_log(YLOG_LOG, "adm request database %s", r->toKeep->databaseName);
 401     }
 402     switch (r->toKeep->which)
 403     {
 404     case Z_ESAdminOriginPartToKeep_reIndex:
 405         yaz_log(YLOG_LOG, "adm-reindex");
 406         break;
 407     case Z_ESAdminOriginPartToKeep_truncate:
 408         yaz_log(YLOG_LOG, "adm-truncate");
 409         break;
 410     case Z_ESAdminOriginPartToKeep_drop:
 411         yaz_log(YLOG_LOG, "adm-drop");
 412         zebra_drop_database (zh, r->toKeep->databaseName);
 413         break;
 414     case Z_ESAdminOriginPartToKeep_create:
 415         yaz_log(YLOG_LOG, "adm-create %s", r->toKeep->databaseName);
 416         zebra_create_database (zh, r->toKeep->databaseName);
 417         break;
 418     case Z_ESAdminOriginPartToKeep_import:
 419         yaz_log(YLOG_LOG, "adm-import");
 420         zebra_admin_import_begin (zh, r->toKeep->databaseName,
 421                         r->toKeep->u.import->recordType);
 422         break;
 423     case Z_ESAdminOriginPartToKeep_refresh:
 424         yaz_log(YLOG_LOG, "adm-refresh");
 425         break;
 426     case Z_ESAdminOriginPartToKeep_commit:
 427         yaz_log(YLOG_LOG, "adm-commit");
 428         if (r->toKeep->databaseName)
 429             zebra_select_database(zh, r->toKeep->databaseName);
 430         zebra_commit(zh);
 431         break;
 432     case Z_ESAdminOriginPartToKeep_shutdown:
 433         yaz_log(YLOG_LOG, "shutdown");
 434         zebra_admin_shutdown(zh);
 435         break;
 436     case Z_ESAdminOriginPartToKeep_start:
 437         yaz_log(YLOG_LOG, "start");
 438         zebra_admin_start(zh);
 439         break;
 440     default:
 441         yaz_log(YLOG_LOG, "unknown admin");
 442     }
 443     return 0;
 444 }
 445
 446 static int es_admin (ZebraHandle zh, Z_Admin *r)
 447 {
 448     switch (r->which)
 449     {
 450     case Z_Admin_esRequest:
 451         es_admin_request (zh, r->u.esRequest);
 452         break;
 453     case Z_Admin_taskPackage:
 454         yaz_log (YLOG_LOG, "adm taskpackage (unhandled)");
 455         break;
 456     default:
 457         break;
 458     }
 459
 460     return 0;
 461 }
 462
 463 int bend_segment (void *handle, bend_segment_rr *rr)
 464 {
 465     ZebraHandle zh = (ZebraHandle) handle;
 466     Z_Segment *segment = rr->segment;
 467
 468     if (segment->num_segmentRecords)
 469         zebra_admin_import_segment (zh, rr->segment);
 470     else
 471         zebra_admin_import_end (zh);
 472     return 0;
 473 }
 474
 475 int bend_esrequest (void *handle, bend_esrequest_rr *rr)
 476 {
 477     ZebraHandle zh = (ZebraHandle) handle;
 478
 479     yaz_log(YLOG_LOG, "function: %d", *rr->esr->function);
 480     if (rr->esr->packageName)
 481         yaz_log(YLOG_LOG, "packagename: %s", rr->esr->packageName);
 482     yaz_log(YLOG_LOG, "Waitaction: %d", *rr->esr->waitAction);
 483
 484     if (!rr->esr->taskSpecificParameters)
 485     {
 486         yaz_log (YLOG_WARN, "No task specific parameters");
 487     }
 488     else if (rr->esr->taskSpecificParameters->which == Z_External_ESAdmin)
 489     {
 490         es_admin (zh, rr->esr->taskSpecificParameters->u.adminService);
 491
 492         zebra_result (zh, &rr->errcode, &rr->errstring);
 493     }
 494     else if (rr->esr->taskSpecificParameters->which == Z_External_update)
 495     {
 496         Z_IUUpdate *up = rr->esr->taskSpecificParameters->u.update;
 497         yaz_log (YLOG_LOG, "Received DB Update");
 498         if (up->which == Z_IUUpdate_esRequest)
 499         {
 500             Z_IUUpdateEsRequest *esRequest = up->u.esRequest;
 501             Z_IUOriginPartToKeep *toKeep = esRequest->toKeep;
 502             Z_IUSuppliedRecords *notToKeep = esRequest->notToKeep;
 503
 504             yaz_log (YLOG_LOG, "action");
 505             if (toKeep->action)
 506             {
 507                 switch (*toKeep->action)
 508                 {
 509                 case Z_IUOriginPartToKeep_recordInsert:
 510                     yaz_log (YLOG_LOG, "recordInsert");
 511                     break;
 512                 case Z_IUOriginPartToKeep_recordReplace:
 513                     yaz_log (YLOG_LOG, "recordUpdate");
 514                     break;
 515                 case Z_IUOriginPartToKeep_recordDelete:
 516                     yaz_log (YLOG_LOG, "recordDelete");
 517                     break;
 518                 case Z_IUOriginPartToKeep_elementUpdate:
 519                     yaz_log (YLOG_LOG, "elementUpdate");
 520                     break;
 521                 case Z_IUOriginPartToKeep_specialUpdate:
 522                     yaz_log (YLOG_LOG, "specialUpdate");
 523                     break;
 524                 case Z_ESAdminOriginPartToKeep_shutdown:
 525                     yaz_log (YLOG_LOG, "shutDown");
 526                     break;
 527                 case Z_ESAdminOriginPartToKeep_start:
 528                     yaz_log (YLOG_LOG, "start");
 529                     break;
 530                 default:
 531                     yaz_log (YLOG_LOG, " unknown (%d)", *toKeep->action);
 532                 }
 533             }
 534             if (toKeep->databaseName)
 535             {
 536                 yaz_log (YLOG_LOG, "database: %s", toKeep->databaseName);
 537
 538                 if (zebra_select_database(zh, toKeep->databaseName))
 539                     return 0;
 540             }
 541             else
 542             {
 543                 yaz_log (YLOG_WARN, "no database supplied for ES Update");
 544                 rr->errcode = 1008;
 545                 rr->errstring = "database";
 546                 return 0;
 547             }
 548             if (notToKeep)
 549             {
 550                 int i;
 551                 zebra_begin_trans (zh, 1);
 552                 for (i = 0; i < notToKeep->num; i++)
 553                 {
 554                     Z_External *rec = notToKeep->elements[i]->record;
 555                     struct oident *oident = 0;
 556                     Odr_oct *opaque_recid = 0;
 557                     SYSNO sysno = 0;
 558
 559                     if (notToKeep->elements[i]->u.opaque)
 560                     {
 561                         switch(notToKeep->elements[i]->which)
 562                         {
 563                         case Z_IUSuppliedRecords_elem_opaque:
 564                             opaque_recid = notToKeep->elements[i]->u.opaque;
 565                             break; /* OK, recid already set */
 566                         case Z_IUSuppliedRecords_elem_number:
 567                             sysno = *notToKeep->elements[i]->u.number;
 568                             break;
 569                         }
 570                     }
 571                     if (rec->direct_reference)
 572                     {
 573                         oident = oid_getentbyoid(rec->direct_reference);
 574                         if (oident)
 575                             yaz_log (YLOG_LOG, "record %d type %s", i,
 576                                      oident->desc);
 577                     }
 578                     switch (rec->which)
 579                     {
 580                     case Z_External_sutrs:
 581                         if (rec->u.octet_aligned->len > 170)
 582                             yaz_log (YLOG_LOG, "%d bytes:\n%.168s ...",
 583                                      rec->u.sutrs->len,
 584                                      rec->u.sutrs->buf);
 585                         else
 586                             yaz_log (YLOG_LOG, "%d bytes:\n%s",
 587                                      rec->u.sutrs->len,
 588                                      rec->u.sutrs->buf);
 589                         break;
 590                     case Z_External_octet:
 591                         if (rec->u.octet_aligned->len > 170)
 592                             yaz_log (YLOG_LOG, "%d bytes:\n%.168s ...",
 593                                      rec->u.octet_aligned->len,
 594                                      rec->u.octet_aligned->buf);
 595                         else
 596                             yaz_log (YLOG_LOG, "%d bytes\n%s",
 597                                      rec->u.octet_aligned->len,
 598                                      rec->u.octet_aligned->buf);
 599                     }
 600                     if (oident && oident->value != VAL_TEXT_XML)
 601                     {
 602                         rr->errcode = 224;
 603                         rr->errstring = "only XML update supported";
 604                         break;
 605                     }
 606                     if (rec->which == Z_External_octet)
 607                     {
 608                         int action = 0;
 609
 610                         if (*toKeep->action ==
 611                             Z_IUOriginPartToKeep_recordInsert)
 612                             action = 1;
 613                         if (*toKeep->action ==
 614                             Z_IUOriginPartToKeep_recordReplace)
 615                             action = 2;
 616                         if (*toKeep->action ==
 617                             Z_IUOriginPartToKeep_recordDelete)
 618                             action = 3;
 619                         if (*toKeep->action ==
 620                             Z_IUOriginPartToKeep_specialUpdate)
 621                             action = 4;
 622
 623                         if (!action)
 624                         {
 625                             rr->errcode = 224;
 626                             rr->errstring = "unsupported ES Update action";
 627                             break;
 628                         }
 629                         else if (opaque_recid)
 630                         {
 631                             int r = zebra_admin_exchange_record (
 632                                 zh,
 633                                 rec->u.octet_aligned->buf,
 634                                 rec->u.octet_aligned->len,
 635                                 opaque_recid->buf, opaque_recid->len,
 636                                 action);
 637                             if (r)
 638                             {
 639                                 rr->errcode = 224;
 640                                 rr->errstring = "record exchange failed";
 641                                 break;
 642                             }
 643                         }
 644                         else
 645                         {
 646                             int r = -1;
 647                             switch(action) {
 648                             case 1:
 649                                 r = zebra_insert_record(
 650                                     zh,
 651                                     0, /* recordType */
 652                                     &sysno,
 653                                     0, /* match */
 654                                     0, /* fname */
 655                                     rec->u.octet_aligned->buf,
 656                                     rec->u.octet_aligned->len,
 657                                     0);
 658                                 if (r)
 659                                 {
 660                                     rr->errcode = 224;
 661                                     rr->errstring = "insert_record failed";
 662                                 }
 663                                 break;
 664                             case 2:
 665                             case 4:
 666                                 r = zebra_update_record(
 667                                     zh,
 668                                     0, /* recordType */
 669                                     &sysno,
 670                                     0, /* match */
 671                                     0, /* fname */
 672                                     rec->u.octet_aligned->buf,
 673                                     rec->u.octet_aligned->len,
 674                                     1);
 675                                 if (r)
 676                                 {
 677                                     rr->errcode = 224;
 678                                     rr->errstring = "update_record failed";
 679                                 }
 680                                 break;
 681                             case 3:
 682                                 r = zebra_delete_record(
 683                                     zh,
 684                                     0, /* recordType */
 685                                     &sysno,
 686                                     0, /* match */
 687                                     0, /* fname */
 688                                     rec->u.octet_aligned->buf,
 689                                     rec->u.octet_aligned->len,
 690                                     0);
 691                                 if (r)
 692                                 {
 693                                     rr->errcode = 224;
 694                                     rr->errstring = "delete_record failed";
 695                                 }
 696                                 break;
 697                             }
 698                         }
 699                     }
 700                 }
 701                 zebra_end_trans (zh);
 702             }
 703         }
 704     }
 705     else
 706     {
 707         yaz_log (YLOG_WARN, "Unknown Extended Service(%d)",
 708                  rr->esr->taskSpecificParameters->which);
 709         rr->errcode = 221;
 710
 711     }
 712     return 0;
 713 }
 714
 715 static void bend_start (struct statserv_options_block *sob)
 716 {
 717     if (sob->handle)
 718         zebra_stop((ZebraService) sob->handle);
 719     sob->handle = zebra_start(sob->configname);
 720     if (!sob->handle)
 721     {
 722         yaz_log (YLOG_FATAL, "Failed to read config `%s'", sob->configname);
 723         exit (1);
 724     }
 725 #ifdef WIN32
 726
 727 #else
 728     if (!sob->inetd)
 729     {
 730         char pidfname[4096];
 731         struct flock area;
 732         int fd;
 733
 734         zebra_pidfname(sob->handle, pidfname);
 735
 736         fd = open (pidfname, O_EXCL|O_WRONLY|O_CREAT, 0666);
 737         if (fd == -1)
 738         {
 739             if (errno != EEXIST)
 740             {
 741                 yaz_log(YLOG_FATAL|YLOG_ERRNO, "lock file %s", pidfname);
 742                 exit(1);
 743             }
 744             fd = open(pidfname, O_RDWR, 0666);
 745             if (fd == -1)
 746             {
 747                 yaz_log(YLOG_FATAL|YLOG_ERRNO, "lock file %s", pidfname);
 748                 exit(1);
 749             }
 750         }
 751         area.l_type = F_WRLCK;
 752         area.l_whence = SEEK_SET;
 753         area.l_len = area.l_start = 0L;
 754         if (fcntl (fd, F_SETLK, &area) == -1)
 755         {
 756             yaz_log(YLOG_ERRNO|YLOG_FATAL, "Zebra server already running");
 757             exit(1);
 758         }
 759         else
 760         {
 761             char pidstr[30];
 762
 763             sprintf (pidstr, "%ld", (long) getpid ());
 764             write (fd, pidstr, strlen(pidstr));
 765         }
 766     }
 767 #endif
 768 }
 769
 770 static void bend_stop(struct statserv_options_block *sob)
 771 {
 772 #ifdef WIN32
 773
 774 #else
 775     if (!sob->inetd && sob->handle)
 776     {
 777         char pidfname[4096];
 778         zebra_pidfname(sob->handle, pidfname);
 779         unlink (pidfname);
 780     }
 781 #endif
 782     if (sob->handle)
 783     {
 784         ZebraService service = sob->handle;
 785         zebra_stop(service);
 786     }
 787 }
 788
 789 int main (int argc, char **argv)
 790 {
 791     struct statserv_options_block *sob;
 792
 793     sob = statserv_getcontrol ();
 794     strcpy (sob->configname, "zebra.cfg");
 795     sob->bend_start = bend_start;
 796     sob->bend_stop = bend_stop;
 797 #ifdef WIN32
 798     strcpy (sob->service_display_name, "Zebra Server");
 799 #endif
 800     statserv_setcontrol (sob);
 801
 802     return statserv_main (argc, argv, bend_init, bend_close);
 803 }