Removed zebra_clearError. The scheme is meant to be .. Return
[idzebra-moved-to-github.git] / index / zserver.c
1 /* $Id: zserver.c,v 1.144 2005-12-09 11:33:32 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 #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 > 2000000000)
255             count = 2000000000;
256         se->subqueryCount = odr_intdup(r->stream, (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                 r->hits = 2147483647;
287             else
288             r->hits = (int) zhits;
289             search_terms (zh, r);
290         }
291         break;
292     case Z_Query_type_2:
293         r->errcode = YAZ_BIB1_QUERY_TYPE_UNSUPP;
294         r->errstring = "type-2";
295         break;
296     default:
297         r->errcode = YAZ_BIB1_QUERY_TYPE_UNSUPP;
298     }
299     return 0;
300 }
301
302
303 int bend_fetch (void *handle, bend_fetch_rr *r)
304 {
305     ZebraHandle zh = (ZebraHandle) handle;
306     ZebraRetrievalRecord retrievalRecord;
307     ZEBRA_RES res;
308
309     retrievalRecord.position = r->number;
310     
311     r->last_in_set = 0;
312     res = zebra_records_retrieve (zh, r->stream, r->setname, r->comp,
313                                   r->request_format, 1, &retrievalRecord);
314     if (res != ZEBRA_OK)
315     {
316         /* non-surrogate diagnostic */
317         zebra_result (zh, &r->errcode, &r->errstring);
318     }
319     else if (retrievalRecord.errCode)
320     {
321         /* surrogate diagnostic (diagnostic per record) */
322         r->surrogate_flag = 1;
323         r->errcode = retrievalRecord.errCode;
324         r->errstring = retrievalRecord.errString;
325         r->basename = retrievalRecord.base;
326     }
327     else
328     {
329         r->basename = retrievalRecord.base;
330         r->record = retrievalRecord.buf;
331         r->len = retrievalRecord.len;
332         r->output_format = retrievalRecord.format;
333     }
334     return 0;
335 }
336
337 static int bend_scan (void *handle, bend_scan_rr *r)
338 {
339     ZebraScanEntry *entries;
340     ZebraHandle zh = (ZebraHandle) handle;
341     int is_partial, i;
342     ZEBRA_RES res;
343
344     res = zebra_select_databases(zh, r->num_bases, 
345                                  (const char **) r->basenames);
346     if (res != ZEBRA_OK)
347     {
348         zebra_result (zh, &r->errcode, &r->errstring);
349         return 0;
350     }
351     if (r->step_size != 0 && *r->step_size != 0) {
352         r->errcode = YAZ_BIB1_ONLY_ZERO_STEP_SIZE_SUPPORTED_FOR_SCAN;
353         r->errstring = 0;
354         return 0;
355     }
356     r->entries = (struct scan_entry *)
357         odr_malloc (r->stream, sizeof(*r->entries) * r->num_entries);
358     res = zebra_scan(zh, r->stream, r->term,
359                      r->attributeset,
360                      &r->term_position,
361                      &r->num_entries, &entries, &is_partial, 
362                      0 /* setname */);
363     if (res == ZEBRA_OK)
364     {
365         if (is_partial)
366             r->status = BEND_SCAN_PARTIAL;
367         else
368             r->status = BEND_SCAN_SUCCESS;
369         for (i = 0; i < r->num_entries; i++)
370         {
371             r->entries[i].term = entries[i].term;
372             r->entries[i].occurrences = 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     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_result (zh, &rr->errcode, &rr->errstring);
398     return 0;
399 }
400
401 int bend_delete (void *handle, bend_delete_rr *rr)
402 {
403     ZebraHandle zh = (ZebraHandle) handle;
404
405     rr->delete_status = zebra_deleteResultSet(zh, rr->function,
406                                               rr->num_setnames, rr->setnames,
407                                               rr->statuses);
408     return 0;
409 }
410
411 static void es_admin_request (bend_esrequest_rr *rr, ZebraHandle zh, Z_AdminEsRequest *r)
412 {
413     ZEBRA_RES res = ZEBRA_OK;
414     if (r->toKeep->databaseName)
415     {
416         yaz_log(YLOG_LOG, "adm request database %s", r->toKeep->databaseName);
417     }
418     switch (r->toKeep->which)
419     {
420     case Z_ESAdminOriginPartToKeep_reIndex:
421         yaz_log(YLOG_LOG, "adm-reindex");
422         rr->errcode = YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
423         rr->errstring = "adm-reindex not implemented yet";
424         break;
425     case Z_ESAdminOriginPartToKeep_truncate:
426         rr->errcode = YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
427         rr->errstring = "adm-reindex not implemented yet";
428         yaz_log(YLOG_LOG, "adm-truncate");
429         break;
430     case Z_ESAdminOriginPartToKeep_drop:
431         yaz_log(YLOG_LOG, "adm-drop");
432         res = zebra_drop_database (zh, r->toKeep->databaseName);
433         break;
434     case Z_ESAdminOriginPartToKeep_create:
435         yaz_log(YLOG_LOG, "adm-create %s", r->toKeep->databaseName);
436         res = zebra_create_database (zh, r->toKeep->databaseName);
437         break;
438     case Z_ESAdminOriginPartToKeep_import:
439         yaz_log(YLOG_LOG, "adm-import");
440         res = zebra_admin_import_begin (zh, r->toKeep->databaseName,
441                                         r->toKeep->u.import->recordType);
442         break;
443     case Z_ESAdminOriginPartToKeep_refresh:
444         yaz_log(YLOG_LOG, "adm-refresh");
445         break;
446     case Z_ESAdminOriginPartToKeep_commit:
447         yaz_log(YLOG_LOG, "adm-commit");
448         if (r->toKeep->databaseName)
449             zebra_select_database(zh, r->toKeep->databaseName);
450         zebra_commit(zh);
451         break;
452     case Z_ESAdminOriginPartToKeep_shutdown:
453         yaz_log(YLOG_LOG, "shutdown");
454         res = zebra_admin_shutdown(zh);
455         break;
456     case Z_ESAdminOriginPartToKeep_start:
457         yaz_log(YLOG_LOG, "start");
458         zebra_admin_start(zh);
459         break;
460     default:
461         yaz_log(YLOG_LOG, "unknown admin");
462         rr->errcode = YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
463         rr->errstring = "adm-reindex not implemented yet";
464     }
465     if (res != ZEBRA_OK)
466         zebra_result(zh, &rr->errcode, &rr->errstring);
467 }
468
469 static void es_admin (bend_esrequest_rr *rr, ZebraHandle zh, Z_Admin *r)
470 {
471     switch (r->which)
472     {
473     case Z_Admin_esRequest:
474         es_admin_request (rr, zh, r->u.esRequest);
475         return;
476     default:
477         break;
478     }
479         yaz_log (YLOG_WARN, "adm taskpackage (unhandled)");
480     rr->errcode = YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
481     rr->errstring = "adm-task package (unhandled)";
482 }
483
484 int bend_segment (void *handle, bend_segment_rr *rr)
485 {
486     ZebraHandle zh = (ZebraHandle) handle;
487     Z_Segment *segment = rr->segment;
488
489     if (segment->num_segmentRecords)
490         zebra_admin_import_segment (zh, rr->segment);
491     else
492         zebra_admin_import_end (zh);
493     return 0;
494 }
495
496 int bend_esrequest (void *handle, bend_esrequest_rr *rr)
497 {
498     ZebraHandle zh = (ZebraHandle) handle;
499     
500     yaz_log(YLOG_LOG, "function: %d", *rr->esr->function);
501     if (rr->esr->packageName)
502         yaz_log(YLOG_LOG, "packagename: %s", rr->esr->packageName);
503     yaz_log(YLOG_LOG, "Waitaction: %d", *rr->esr->waitAction);
504
505     if (!rr->esr->taskSpecificParameters)
506     {
507         yaz_log (YLOG_WARN, "No task specific parameters");
508     }
509     else if (rr->esr->taskSpecificParameters->which == Z_External_ESAdmin)
510     {
511         es_admin(rr, zh, rr->esr->taskSpecificParameters->u.adminService);
512     }
513     else if (rr->esr->taskSpecificParameters->which == Z_External_update)
514     {
515         Z_IUUpdate *up = rr->esr->taskSpecificParameters->u.update;
516         yaz_log (YLOG_LOG, "Received DB Update");
517         if (up->which == Z_IUUpdate_esRequest)
518         {
519             Z_IUUpdateEsRequest *esRequest = up->u.esRequest;
520             Z_IUOriginPartToKeep *toKeep = esRequest->toKeep;
521             Z_IUSuppliedRecords *notToKeep = esRequest->notToKeep;
522             
523             yaz_log (YLOG_LOG, "action");
524             if (toKeep->action)
525             {
526                 switch (*toKeep->action)
527                 {
528                 case Z_IUOriginPartToKeep_recordInsert:
529                     yaz_log (YLOG_LOG, "recordInsert");
530                     break;
531                 case Z_IUOriginPartToKeep_recordReplace:
532                     yaz_log (YLOG_LOG, "recordUpdate");
533                     break;
534                 case Z_IUOriginPartToKeep_recordDelete:
535                     yaz_log (YLOG_LOG, "recordDelete");
536                     break;
537                 case Z_IUOriginPartToKeep_elementUpdate:
538                     yaz_log (YLOG_LOG, "elementUpdate");
539                     break;
540                 case Z_IUOriginPartToKeep_specialUpdate:
541                     yaz_log (YLOG_LOG, "specialUpdate");
542                     break;
543                 case Z_ESAdminOriginPartToKeep_shutdown:
544                     yaz_log (YLOG_LOG, "shutDown");
545                     break;
546                 case Z_ESAdminOriginPartToKeep_start:
547                     yaz_log (YLOG_LOG, "start");
548                     break;
549                 default:
550                     yaz_log (YLOG_LOG, " unknown (%d)", *toKeep->action);
551                 }
552             }
553             if (toKeep->databaseName)
554             {
555                 yaz_log (YLOG_LOG, "database: %s", toKeep->databaseName);
556
557                 if (zebra_select_database(zh, toKeep->databaseName))
558                     return 0;
559             }
560             else
561             {
562                 yaz_log (YLOG_WARN, "no database supplied for ES Update");
563                 rr->errcode =
564                     YAZ_BIB1_ES_MISSING_MANDATORY_PARAMETER_FOR_SPECIFIED_FUNCTION_;
565                 rr->errstring = "database";
566                 return 0;
567             }
568             if (zebra_begin_trans(zh, 1) != ZEBRA_OK)
569             {
570                 zebra_result(zh, &rr->errcode, &rr->errstring);
571             }
572             else
573             {
574                 int i;
575                 for (i = 0; notToKeep && i < notToKeep->num; i++)
576                 {
577                     Z_External *rec = notToKeep->elements[i]->record;
578                     struct oident *oident = 0;
579                     Odr_oct *opaque_recid = 0;
580                     SYSNO *sysno = 0;
581                     SYSNO sysno_tmp;
582
583                     if (notToKeep->elements[i]->u.opaque)
584                     {
585                         switch(notToKeep->elements[i]->which)
586                         {
587                         case Z_IUSuppliedRecords_elem_opaque:
588                             opaque_recid = notToKeep->elements[i]->u.opaque;
589                             break; /* OK, recid already set */
590                         case Z_IUSuppliedRecords_elem_number:
591                             sysno_tmp = *notToKeep->elements[i]->u.number;
592                             sysno = &sysno_tmp;
593                             break;
594                         }
595                     }
596                     if (rec->direct_reference)
597                     {
598                         oident = oid_getentbyoid(rec->direct_reference);
599                         if (oident)
600                             yaz_log (YLOG_LOG, "record %d type %s", i,
601                                      oident->desc);
602                     }
603                     switch (rec->which)
604                     {
605                     case Z_External_sutrs:
606                         if (rec->u.octet_aligned->len > 170)
607                             yaz_log (YLOG_LOG, "%d bytes:\n%.168s ...",
608                                      rec->u.sutrs->len,
609                                      rec->u.sutrs->buf);
610                         else
611                             yaz_log (YLOG_LOG, "%d bytes:\n%s",
612                                      rec->u.sutrs->len,
613                                      rec->u.sutrs->buf);
614                         break;
615                     case Z_External_octet:
616                         if (rec->u.octet_aligned->len > 170)
617                             yaz_log (YLOG_LOG, "%d bytes:\n%.168s ...",
618                                      rec->u.octet_aligned->len,
619                                      rec->u.octet_aligned->buf);
620                         else
621                             yaz_log (YLOG_LOG, "%d bytes\n%s",
622                                      rec->u.octet_aligned->len,
623                                      rec->u.octet_aligned->buf);
624                     }
625                     if (oident && oident->value != VAL_TEXT_XML)
626                     {
627                         rr->errcode = YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
628                         rr->errstring = "only XML update supported";
629                         break;
630                     }
631                     if (rec->which == Z_External_octet)
632                     {
633                         int action = 0;
634
635                         if (*toKeep->action ==
636                             Z_IUOriginPartToKeep_recordInsert)
637                             action = 1;
638                         if (*toKeep->action ==
639                             Z_IUOriginPartToKeep_recordReplace)
640                             action = 2;
641                         if (*toKeep->action ==
642                             Z_IUOriginPartToKeep_recordDelete)
643                             action = 3;
644                         if (*toKeep->action ==
645                             Z_IUOriginPartToKeep_specialUpdate)
646                             action = 4;
647
648                         if (!action)
649                         {
650                             rr->errcode =
651                                 YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
652                             rr->errstring = "unsupported ES Update action";
653                             break;
654                         }
655                         else if (opaque_recid)
656                         {
657                             int r = zebra_admin_exchange_record (
658                                 zh,
659                                 (const char *) rec->u.octet_aligned->buf,
660                                 rec->u.octet_aligned->len,
661                                 (const char *) opaque_recid->buf,
662                                 opaque_recid->len,
663                                 action);
664                             if (r)
665                             {
666                                 zebra_result(zh, &rr->errcode,
667                                              &rr->errstring);
668                                 break;
669                             }
670                         }
671                         else
672                         {
673                             ZEBRA_RES r = ZEBRA_FAIL;
674                             switch(action) {
675                             case 1:
676                                 r = zebra_insert_record(
677                                     zh,
678                                     0, /* recordType */
679                                     sysno,
680                                     0, /* match */
681                                     0, /* fname */
682                                     (const char *) rec->u.octet_aligned->buf,
683                                     rec->u.octet_aligned->len,
684                                     0);
685                                 if (r == ZEBRA_FAIL)
686                                 {
687                                     rr->errcode =
688                                         YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
689                                     rr->errstring = "insert_record failed";
690                                 }
691                                 break;
692                             case 2:
693                             case 4:
694                                 r = zebra_update_record(
695                                     zh,
696                                     0, /* recordType */
697                                     sysno,
698                                     0, /* match */
699                                     0, /* fname */
700                                     (const char *) rec->u.octet_aligned->buf,
701                                     rec->u.octet_aligned->len,
702                                     1);
703                                 if (r == ZEBRA_FAIL)
704                                 {
705                                     rr->errcode =
706                                         YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
707                                     rr->errstring = "update_record failed";
708                                 }
709                                 break;
710                             case 3:
711                                 r = zebra_delete_record(
712                                     zh,
713                                     0, /* recordType */
714                                     sysno,
715                                     0, /* match */
716                                     0, /* fname */
717                                     (const char *) rec->u.octet_aligned->buf,
718                                     rec->u.octet_aligned->len,
719                                     0);
720                                 if (r == ZEBRA_FAIL)
721                                 {
722                                     rr->errcode =
723                                         YAZ_BIB1_ES_IMMEDIATE_EXECUTION_FAILED;
724                                     rr->errstring = "delete_record failed";
725                                 }
726                                 break;
727                             }                           
728                         }
729                     }
730                 }
731                 zebra_end_trans (zh);
732             }
733         }
734     }
735     else
736     {
737         yaz_log (YLOG_WARN, "Unknown Extended Service(%d)",
738                  rr->esr->taskSpecificParameters->which);
739         rr->errcode = YAZ_BIB1_ES_EXTENDED_SERVICE_TYPE_UNSUPP;
740         
741     }
742     return 0;
743 }
744
745 static void bend_start (struct statserv_options_block *sob)
746 {
747     if (sob->handle)
748         zebra_stop((ZebraService) sob->handle);
749     sob->handle = zebra_start(sob->configname);
750     if (!sob->handle)
751     {
752         yaz_log (YLOG_FATAL, "Failed to read config `%s'", sob->configname);
753         exit (1);
754     }
755 #ifdef WIN32
756     
757 #else
758     if (!sob->inetd) 
759     {
760         char pidfname[4096];
761         struct flock area;
762         int fd;
763
764         zebra_pidfname(sob->handle, pidfname);
765
766         fd = open (pidfname, O_EXCL|O_WRONLY|O_CREAT, 0666);
767         if (fd == -1)
768         {
769             if (errno != EEXIST)
770             {
771                 yaz_log(YLOG_FATAL|YLOG_ERRNO, "lock file %s", pidfname);
772                 exit(1);
773             }
774             fd = open(pidfname, O_RDWR, 0666);
775             if (fd == -1)
776             {
777                 yaz_log(YLOG_FATAL|YLOG_ERRNO, "lock file %s", pidfname);
778                 exit(1);
779             }
780         }
781         area.l_type = F_WRLCK;
782         area.l_whence = SEEK_SET;
783         area.l_len = area.l_start = 0L;
784         if (fcntl (fd, F_SETLK, &area) == -1)
785         {
786             yaz_log(YLOG_ERRNO|YLOG_FATAL, "Zebra server already running");
787             exit(1);
788         }
789         else
790         {
791             char pidstr[30];
792         
793             sprintf (pidstr, "%ld", (long) getpid ());
794             write (fd, pidstr, strlen(pidstr));
795         }
796     }
797 #endif
798 }
799
800 static void bend_stop(struct statserv_options_block *sob)
801 {
802 #ifdef WIN32
803
804 #else
805     if (!sob->inetd && sob->handle) 
806     {
807         char pidfname[4096];
808         zebra_pidfname(sob->handle, pidfname);
809         unlink (pidfname);
810     }
811 #endif
812     if (sob->handle)
813     {
814         ZebraService service = sob->handle;
815         zebra_stop(service);
816     }
817 }
818
819 int main (int argc, char **argv)
820 {
821     struct statserv_options_block *sob;
822
823     sob = statserv_getcontrol ();
824     strcpy (sob->configname, "zebra.cfg");
825     sob->bend_start = bend_start;
826     sob->bend_stop = bend_stop;
827 #ifdef WIN32
828     strcpy (sob->service_display_name, "Zebra Server");
829 #endif
830     statserv_setcontrol (sob);
831
832     return statserv_main (argc, argv, bend_init, bend_close);
833 }