index/zserver.c

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