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