2 * Copyright (C) 1995-2000, Index Data
5 * $Id: zserver.c,v 1.85 2002-04-04 14:14:13 adam Exp $
18 #include <yaz/data1.h>
25 static int bend_sort (void *handle, bend_sort_rr *rr);
26 static int bend_delete (void *handle, bend_delete_rr *rr);
27 static int bend_esrequest (void *handle, bend_esrequest_rr *rr);
28 static int bend_segment (void *handle, bend_segment_rr *rr);
29 static int bend_search (void *handle, bend_search_rr *r);
30 static int bend_fetch (void *handle, bend_fetch_rr *r);
31 static int bend_scan (void *handle, bend_scan_rr *r);
33 bend_initresult *bend_init (bend_initrequest *q)
35 bend_initresult *r = (bend_initresult *)
36 odr_malloc (q->stream, sizeof(*r));
38 struct statserv_options_block *sob;
44 q->bend_sort = bend_sort;
45 q->bend_delete = bend_delete;
46 q->bend_esrequest = bend_esrequest;
47 q->bend_segment = bend_segment;
48 q->bend_search = bend_search;
49 q->bend_fetch = bend_fetch;
50 q->bend_scan = bend_scan;
53 q->implementation_name = "Z'mbol Information Server";
54 q->implementation_version = "Z'mbol " ZEBRAVER;
56 q->implementation_name = "Zebra Information Server";
57 q->implementation_version = "Zebra " ZEBRAVER;
60 logf (LOG_DEBUG, "bend_init");
62 sob = statserv_getcontrol ();
63 if (!(zh = zebra_open (sob->handle)))
65 logf (LOG_FATAL, "Failed to open Zebra `%s'", sob->configname);
71 if (q->auth->which == Z_IdAuthentication_open)
73 char *openpass = xstrdup (q->auth->u.open);
74 char *cp = strchr (openpass, '/');
78 user = nmem_strdup (odr_getmem (q->stream), openpass);
79 passwd = nmem_strdup (odr_getmem (q->stream), cp+1);
84 if (zebra_auth (zh, user, passwd))
95 static void search_terms (ZebraHandle zh, bend_search_rr *r)
100 struct Z_External *ext;
101 Z_SearchInfoReport *sr;
103 /* get no of terms for result set */
104 zebra_resultSetTerms (zh, r->setname, -1, &count, &no_terms);
108 r->search_info = odr_malloc (r->stream, sizeof(*r->search_info));
110 r->search_info->num_elements = 1;
111 r->search_info->list =
112 odr_malloc (r->stream, sizeof(*r->search_info->list));
113 r->search_info->list[0] =
114 odr_malloc (r->stream, sizeof(**r->search_info->list));
115 r->search_info->list[0]->category = 0;
116 r->search_info->list[0]->which = Z_OtherInfo_externallyDefinedInfo;
117 ext = odr_malloc (r->stream, sizeof(*ext));
118 r->search_info->list[0]->information.externallyDefinedInfo = ext;
119 ext->direct_reference =
120 yaz_oidval_to_z3950oid (r->stream, CLASS_USERINFO, VAL_SEARCHRES1);
121 ext->indirect_reference = 0;
123 ext->which = Z_External_searchResult1;
124 sr = odr_malloc (r->stream, sizeof(Z_SearchInfoReport));
125 ext->u.searchResult1 = sr;
127 sr->elements = odr_malloc (r->stream, sr->num *
128 sizeof(*sr->elements));
129 for (i = 0; i<no_terms; i++)
132 const char *termz = zebra_resultSetTerms (zh, r->setname, i,
135 sr->elements[i] = odr_malloc (r->stream, sizeof(**sr->elements));
136 sr->elements[i]->subqueryId = 0;
137 sr->elements[i]->fullQuery = odr_malloc (r->stream,
139 *sr->elements[i]->fullQuery = 0;
140 sr->elements[i]->subqueryExpression =
141 odr_malloc (r->stream, sizeof(Z_QueryExpression));
142 sr->elements[i]->subqueryExpression->which =
143 Z_QueryExpression_term;
144 sr->elements[i]->subqueryExpression->u.term =
145 odr_malloc (r->stream, sizeof(Z_QueryExpressionTerm));
146 term = odr_malloc (r->stream, sizeof(Z_Term));
147 sr->elements[i]->subqueryExpression->u.term->queryTerm = term;
149 term->which = Z_Term_general;
150 term->u.general = odr_malloc (r->stream, sizeof(Odr_oct));
151 term->u.general->buf = odr_strdup (r->stream, termz);
153 term->u.general->len = strlen (termz);
154 term->u.general->size = strlen (termz);
156 sr->elements[i]->subqueryExpression->u.term->termComment = 0;
157 sr->elements[i]->subqueryInterpretation = 0;
158 sr->elements[i]->subqueryRecommendation = 0;
159 sr->elements[i]->subqueryCount = odr_intdup (r->stream, count);
160 sr->elements[i]->subqueryWeight = 0;
161 sr->elements[i]->resultsByDB = 0;
165 int bend_search (void *handle, bend_search_rr *r)
167 ZebraHandle zh = (ZebraHandle) handle;
173 if (zebra_select_databases (zh, r->num_bases,
174 (const char **) r->basenames))
176 zebra_result (zh, &r->errcode, &r->errstring);
179 logf (LOG_LOG, "ResultSet '%s'", r->setname);
180 switch (r->query->which)
182 case Z_Query_type_1: case Z_Query_type_101:
183 zebra_search_rpn (zh, r->decode, r->stream, r->query->u.type_1,
184 r->setname, &r->hits);
185 zebra_result (zh, &r->errcode, &r->errstring);
187 search_terms (zh, r);
191 r->errstring = "type-2";
200 int bend_fetch (void *handle, bend_fetch_rr *r)
202 ZebraHandle zh = (ZebraHandle) handle;
203 ZebraRetrievalRecord retrievalRecord;
205 retrievalRecord.position = r->number;
208 zebra_records_retrieve (zh, r->stream, r->setname, r->comp,
209 r->request_format, 1, &retrievalRecord);
210 zebra_result (zh, &r->errcode, &r->errstring);
211 /* non Surrogate Diagnostic OR Surrogate Diagnostic */
212 if (r->errcode == 0 && retrievalRecord.errCode)
214 r->surrogate_flag = 1;
215 r->errcode = retrievalRecord.errCode;
216 r->errstring = retrievalRecord.errString;
217 r->basename = retrievalRecord.base;
219 else if (r->errcode == 0) /* Database Record */
222 r->basename = retrievalRecord.base;
223 r->record = retrievalRecord.buf;
224 r->len = retrievalRecord.len;
225 r->output_format = retrievalRecord.format;
230 static int bend_scan (void *handle, bend_scan_rr *r)
232 ZebraScanEntry *entries;
233 ZebraHandle zh = (ZebraHandle) handle;
236 zebra_select_databases (zh, r->num_bases, (const char **) r->basenames);
238 r->entries = (struct scan_entry *)
239 odr_malloc (r->stream, sizeof(*r->entries) * r->num_entries);
240 zebra_scan (zh, r->stream, r->term,
243 &r->num_entries, &entries, &is_partial);
245 r->status = BEND_SCAN_PARTIAL;
247 r->status = BEND_SCAN_SUCCESS;
248 for (i = 0; i < r->num_entries; i++)
250 r->entries[i].term = entries[i].term;
251 r->entries[i].occurrences = entries[i].occurrences;
253 zebra_result (zh, &r->errcode, &r->errstring);
257 void bend_close (void *handle)
259 zebra_close ((ZebraHandle) handle);
260 xmalloc_trav("bend_close");
264 int bend_sort (void *handle, bend_sort_rr *rr)
266 ZebraHandle zh = (ZebraHandle) handle;
268 zebra_sort (zh, rr->stream,
269 rr->num_input_setnames, (const char **) rr->input_setnames,
270 rr->output_setname, rr->sort_sequence, &rr->sort_status);
271 zebra_result (zh, &rr->errcode, &rr->errstring);
275 int bend_delete (void *handle, bend_delete_rr *rr)
277 ZebraHandle zh = (ZebraHandle) handle;
279 rr->delete_status = zebra_deleleResultSet(zh, rr->function,
280 rr->num_setnames, rr->setnames,
285 static int es_admin_request (ZebraHandle zh, Z_AdminEsRequest *r)
287 switch (r->toKeep->which)
289 case Z_ESAdminOriginPartToKeep_reIndex:
290 yaz_log(LOG_LOG, "adm-reindex");
292 case Z_ESAdminOriginPartToKeep_truncate:
293 yaz_log(LOG_LOG, "adm-truncate");
295 case Z_ESAdminOriginPartToKeep_drop:
296 yaz_log(LOG_LOG, "adm-drop");
298 case Z_ESAdminOriginPartToKeep_create:
299 yaz_log(LOG_LOG, "adm-create");
300 zebra_admin_create (zh, r->toKeep->databaseName);
302 case Z_ESAdminOriginPartToKeep_import:
303 yaz_log(LOG_LOG, "adm-import");
304 zebra_admin_import_begin (zh, r->toKeep->databaseName);
306 case Z_ESAdminOriginPartToKeep_refresh:
307 yaz_log(LOG_LOG, "adm-refresh");
309 case Z_ESAdminOriginPartToKeep_commit:
310 yaz_log(LOG_LOG, "adm-commit");
312 case Z_ESAdminOriginPartToKeep_shutdown:
313 yaz_log(LOG_LOG, "shutdown");
314 zebra_admin_shutdown(zh);
316 case Z_ESAdminOriginPartToKeep_start:
317 yaz_log(LOG_LOG, "start");
318 zebra_admin_start(zh);
321 yaz_log(LOG_LOG, "unknown admin");
323 if (r->toKeep->databaseName)
325 yaz_log(LOG_LOG, "database %s", r->toKeep->databaseName);
330 static int es_admin (ZebraHandle zh, Z_Admin *r)
334 case Z_Admin_esRequest:
335 es_admin_request (zh, r->u.esRequest);
337 case Z_Admin_taskPackage:
338 yaz_log (LOG_LOG, "adm taskpackage (unhandled)");
347 int bend_segment (void *handle, bend_segment_rr *rr)
349 ZebraHandle zh = (ZebraHandle) handle;
350 Z_Segment *segment = rr->segment;
352 if (segment->num_segmentRecords)
353 zebra_admin_import_segment (zh, rr->segment);
355 zebra_admin_import_end (zh);
359 int bend_esrequest (void *handle, bend_esrequest_rr *rr)
361 ZebraHandle zh = (ZebraHandle) handle;
363 yaz_log(LOG_LOG, "function: %d", *rr->esr->function);
364 if (rr->esr->packageName)
365 yaz_log(LOG_LOG, "packagename: %s", rr->esr->packageName);
366 yaz_log(LOG_LOG, "Waitaction: %d", *rr->esr->waitAction);
368 if (!rr->esr->taskSpecificParameters)
370 yaz_log (LOG_WARN, "No task specific parameters");
372 else if (rr->esr->taskSpecificParameters->which == Z_External_ESAdmin)
374 es_admin (zh, rr->esr->taskSpecificParameters->u.adminService);
376 zebra_result (zh, &rr->errcode, &rr->errstring);
378 else if (rr->esr->taskSpecificParameters->which == Z_External_itemOrder)
380 Z_ItemOrder *it = rr->esr->taskSpecificParameters->u.itemOrder;
381 yaz_log (LOG_LOG, "Received ItemOrder");
385 case Z_IOItemOrder_esRequest:
387 case Z_ItemOrder_esRequest:
390 Z_IORequest *ir = it->u.esRequest;
391 Z_IOOriginPartToKeep *k = ir->toKeep;
392 Z_IOOriginPartNotToKeep *n = ir->notToKeep;
396 if (k->contact->name)
397 yaz_log(LOG_LOG, "contact name %s", k->contact->name);
398 if (k->contact->phone)
399 yaz_log(LOG_LOG, "contact phone %s", k->contact->phone);
400 if (k->contact->email)
401 yaz_log(LOG_LOG, "contact email %s", k->contact->email);
405 yaz_log(LOG_LOG, "Billing info (not shown)");
408 if (n->resultSetItem)
410 yaz_log(LOG_LOG, "resultsetItem");
411 yaz_log(LOG_LOG, "setId: %s", n->resultSetItem->resultSetId);
412 yaz_log(LOG_LOG, "item: %d", *n->resultSetItem->item);
417 Z_External *r = (Z_External*) n->itemRequest;
418 ILL_ItemRequest *item_req = 0;
419 ILL_Request *ill_req = 0;
420 if (r->direct_reference)
422 oident *ent = oid_getentbyoid(r->direct_reference);
424 yaz_log(LOG_LOG, "OID %s", ent->desc);
425 if (ent && ent->value == VAL_ISO_ILL_1)
427 yaz_log (LOG_LOG, "ItemRequest");
428 if (r->which == ODR_EXTERNAL_single)
430 odr_setbuf(rr->decode,
431 r->u.single_ASN1_type->buf,
432 r->u.single_ASN1_type->len, 0);
434 if (!ill_ItemRequest (rr->decode, &item_req, 0, 0))
437 "Couldn't decode ItemRequest %s near %d",
438 odr_errmsg(odr_geterror(rr->decode)),
439 odr_offset(rr->decode));
440 yaz_log(LOG_LOG, "PDU dump:");
441 odr_dumpBER(yaz_log_file(),
442 r->u.single_ASN1_type->buf,
443 r->u.single_ASN1_type->len);
447 ill_ItemRequest (rr->print, &item_req, 0,
449 odr_reset (rr->print);
452 if (!item_req && r->which == ODR_EXTERNAL_single)
454 yaz_log (LOG_LOG, "ILLRequest");
455 odr_setbuf(rr->decode,
456 r->u.single_ASN1_type->buf,
457 r->u.single_ASN1_type->len, 0);
459 if (!ill_Request (rr->decode, &ill_req, 0, 0))
462 "Couldn't decode ILLRequest %s near %d",
463 odr_errmsg(odr_geterror(rr->decode)),
464 odr_offset(rr->decode));
465 yaz_log(LOG_LOG, "PDU dump:");
466 odr_dumpBER(yaz_log_file(),
467 r->u.single_ASN1_type->buf,
468 r->u.single_ASN1_type->len);
472 ill_Request (rr->print, &ill_req, 0,
474 odr_reset (rr->print);
481 yaz_log (LOG_LOG, "ILL protocol version = %d",
482 *item_req->protocol_version_num);
490 else if (rr->esr->taskSpecificParameters->which == Z_External_update)
492 Z_IUUpdate *up = rr->esr->taskSpecificParameters->u.update;
493 yaz_log (LOG_LOG, "Received DB Update");
494 if (up->which == Z_IUUpdate_esRequest)
496 Z_IUUpdateEsRequest *esRequest = up->u.esRequest;
497 Z_IUOriginPartToKeep *toKeep = esRequest->toKeep;
498 Z_IUSuppliedRecords *notToKeep = esRequest->notToKeep;
500 yaz_log (LOG_LOG, "action");
503 switch (*toKeep->action)
505 case Z_IUOriginPartToKeep_recordInsert:
506 yaz_log (LOG_LOG, "recordInsert");
508 case Z_IUOriginPartToKeep_recordReplace:
509 yaz_log (LOG_LOG, "recordUpdate");
511 case Z_IUOriginPartToKeep_recordDelete:
512 yaz_log (LOG_LOG, "recordDelete");
514 case Z_IUOriginPartToKeep_elementUpdate:
515 yaz_log (LOG_LOG, "elementUpdate");
517 case Z_IUOriginPartToKeep_specialUpdate:
518 yaz_log (LOG_LOG, "specialUpdate");
520 case Z_ESAdminOriginPartToKeep_shutdown:
521 yaz_log (LOG_LOG, "shutDown");
523 case Z_ESAdminOriginPartToKeep_start:
524 yaz_log (LOG_LOG, "start");
527 yaz_log (LOG_LOG, " unknown (%d)", *toKeep->action);
530 if (toKeep->databaseName)
532 yaz_log (LOG_LOG, "database: %s", toKeep->databaseName);
533 if (!strcmp(toKeep->databaseName, "fault"))
536 rr->errstring = toKeep->databaseName;
538 if (!strcmp(toKeep->databaseName, "accept"))
544 for (i = 0; i < notToKeep->num; i++)
546 Z_External *rec = notToKeep->elements[i]->record;
548 if (rec->direct_reference)
550 struct oident *oident;
551 oident = oid_getentbyoid(rec->direct_reference);
553 yaz_log (LOG_LOG, "record %d type %s", i,
558 case Z_External_sutrs:
559 if (rec->u.octet_aligned->len > 170)
560 yaz_log (LOG_LOG, "%d bytes:\n%.168s ...",
564 yaz_log (LOG_LOG, "%d bytes:\n%s",
568 case Z_External_octet:
569 if (rec->u.octet_aligned->len > 170)
570 yaz_log (LOG_LOG, "%d bytes:\n%.168s ...",
571 rec->u.octet_aligned->len,
572 rec->u.octet_aligned->buf);
574 yaz_log (LOG_LOG, "%d bytes\n%s",
575 rec->u.octet_aligned->len,
576 rec->u.octet_aligned->buf);
584 yaz_log (LOG_WARN, "Unknown Extended Service(%d)",
585 rr->esr->taskSpecificParameters->which);
591 static void bend_start (struct statserv_options_block *sob)
598 char *pidfile = "zebrasrv.pid";
599 int fd = creat (pidfile, 0666);
601 logf (LOG_WARN|LOG_ERRNO, "creat %s", pidfile);
606 sprintf (pidstr, "%ld", (long) getpid ());
607 write (fd, pidstr, strlen(pidstr));
613 zebra_stop((ZebraService) sob->handle);
614 sob->handle = zebra_start(sob->configname);
617 static void bend_stop(struct statserv_options_block *sob)
621 ZebraService service = sob->handle;
626 int main (int argc, char **argv)
628 struct statserv_options_block *sob;
630 sob = statserv_getcontrol ();
631 strcpy (sob->configname, "zebra.cfg");
632 sob->bend_start = bend_start;
633 sob->bend_stop = bend_stop;
635 statserv_setcontrol (sob);
637 return statserv_main (argc, argv, bend_init, bend_close);