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