ffed67f5c8d8394334dbf185b96a50d0d660ed50
[yaz-moved-to-github.git] / server / seshigh.c
1 /*
2  * Copyright (c) 1995-2003, Index Data
3  * See the file LICENSE for details.
4  *
5  * $Id: seshigh.c,v 1.151 2003-03-24 22:26:50 adam Exp $
6  */
7
8 /*
9  * Frontend server logic.
10  *
11  * This code receives incoming APDUs, and handles client requests by means
12  * of the backend API.
13  *
14  * Some of the code is getting quite involved, compared to simpler servers -
15  * primarily because it is asynchronous both in the communication with
16  * the user and the backend. We think the complexity will pay off in
17  * the form of greater flexibility when more asynchronous facilities
18  * are implemented.
19  *
20  * Memory management has become somewhat involved. In the simple case, where
21  * only one PDU is pending at a time, it will simply reuse the same memory,
22  * once it has found its working size. When we enable multiple concurrent
23  * operations, perhaps even with multiple parallel calls to the backend, it
24  * will maintain a pool of buffers for encoding and decoding, trying to
25  * minimize memory allocation/deallocation during normal operation.
26  *
27  */
28
29 #include <stdlib.h>
30 #include <stdio.h>
31 #include <sys/types.h>
32 #ifdef WIN32
33 #include <io.h>
34 #define S_ISREG(x) (x & _S_IFREG)
35 #include <process.h>
36 #include <sys/stat.h>
37 #else
38 #include <sys/stat.h>
39 #include <unistd.h>
40 #endif
41 #include <assert.h>
42 #include <ctype.h>
43
44 #include <yaz/yconfig.h>
45 #include <yaz/xmalloc.h>
46 #include <yaz/comstack.h>
47 #include "eventl.h"
48 #include "session.h"
49 #include <yaz/proto.h>
50 #include <yaz/oid.h>
51 #include <yaz/log.h>
52 #include <yaz/logrpn.h>
53 #include <yaz/statserv.h>
54 #include <yaz/diagbib1.h>
55 #include <yaz/charneg.h>
56 #include <yaz/otherinfo.h>
57 #include <yaz/yaz-util.h>
58 #include <yaz/pquery.h>
59
60 #include <yaz/srw.h>
61 #include <yaz/backend.h>
62
63 static void process_gdu_request(association *assoc, request *req);
64 static int process_z_request(association *assoc, request *req, char **msg);
65 void backend_response(IOCHAN i, int event);
66 static int process_gdu_response(association *assoc, request *req, Z_GDU *res);
67 static int process_z_response(association *assoc, request *req, Z_APDU *res);
68 static Z_APDU *process_initRequest(association *assoc, request *reqb);
69 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
70     int *fd);
71 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
72     bend_search_rr *bsrr, int *fd);
73 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
74     int *fd);
75 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd);
76 static Z_APDU *process_sortRequest(association *assoc, request *reqb, int *fd);
77 static void process_close(association *assoc, request *reqb);
78 void save_referenceId (request *reqb, Z_ReferenceId *refid);
79 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
80     int *fd);
81 static Z_APDU *process_segmentRequest (association *assoc, request *reqb);
82
83 static FILE *apduf = 0; /* for use in static mode */
84 static statserv_options_block *control_block = 0;
85
86 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd);
87
88 /*
89  * Create and initialize a new association-handle.
90  *  channel  : iochannel for the current line.
91  *  link     : communications channel.
92  * Returns: 0 or a new association handle.
93  */
94 association *create_association(IOCHAN channel, COMSTACK link)
95 {
96     association *anew;
97
98     if (!control_block)
99         control_block = statserv_getcontrol();
100     if (!(anew = (association *)xmalloc(sizeof(*anew))))
101         return 0;
102     anew->init = 0;
103     anew->version = 0;
104     anew->client_chan = channel;
105     anew->client_link = link;
106     anew->cs_get_mask = 0;
107     anew->cs_put_mask = 0;
108     anew->cs_accept_mask = 0;
109     if (!(anew->decode = odr_createmem(ODR_DECODE)) ||
110         !(anew->encode = odr_createmem(ODR_ENCODE)))
111         return 0;
112     if (*control_block->apdufile)
113     {
114         char filename[256];
115         FILE *f;
116
117         strcpy(filename, control_block->apdufile);
118         if (!(anew->print = odr_createmem(ODR_PRINT)))
119             return 0;
120         if (*control_block->apdufile == '@')
121         {
122             odr_setprint(anew->print, yaz_log_file());
123         }       
124         else if (*control_block->apdufile != '-')
125         {
126             strcpy(filename, control_block->apdufile);
127             if (!control_block->dynamic)
128             {
129                 if (!apduf)
130                 {
131                     if (!(apduf = fopen(filename, "w")))
132                     {
133                         yaz_log(LOG_WARN|LOG_ERRNO, "%s", filename);
134                         return 0;
135                     }
136                     setvbuf(apduf, 0, _IONBF, 0);
137                 }
138                 f = apduf;
139             }
140             else 
141             {
142                 sprintf(filename + strlen(filename), ".%d", getpid());
143                 if (!(f = fopen(filename, "w")))
144                 {
145                     yaz_log(LOG_WARN|LOG_ERRNO, "%s", filename);
146                     return 0;
147                 }
148                 setvbuf(f, 0, _IONBF, 0);
149             }
150             odr_setprint(anew->print, f);
151         }
152     }
153     else
154         anew->print = 0;
155     anew->input_buffer = 0;
156     anew->input_buffer_len = 0;
157     anew->backend = 0;
158     anew->state = ASSOC_NEW;
159     request_initq(&anew->incoming);
160     request_initq(&anew->outgoing);
161     anew->proto = cs_getproto(link);
162     return anew;
163 }
164
165 /*
166  * Free association and release resources.
167  */
168 void destroy_association(association *h)
169 {
170     statserv_options_block *cb = statserv_getcontrol();
171
172     xfree(h->init);
173     odr_destroy(h->decode);
174     odr_destroy(h->encode);
175     if (h->print)
176         odr_destroy(h->print);
177     if (h->input_buffer)
178     xfree(h->input_buffer);
179     if (h->backend)
180         (*cb->bend_close)(h->backend);
181     while (request_deq(&h->incoming));
182     while (request_deq(&h->outgoing));
183     request_delq(&h->incoming);
184     request_delq(&h->outgoing);
185     xfree(h);
186     xmalloc_trav("session closed");
187     if (control_block && control_block->one_shot)
188     {
189         exit (0);
190     }
191 }
192
193 static void do_close_req(association *a, int reason, char *message,
194                          request *req)
195 {
196     Z_APDU apdu;
197     Z_Close *cls = zget_Close(a->encode);
198     
199     /* Purge request queue */
200     while (request_deq(&a->incoming));
201     while (request_deq(&a->outgoing));
202     if (a->version >= 3)
203     {
204         yaz_log(LOG_LOG, "Sending Close PDU, reason=%d, message=%s",
205             reason, message ? message : "none");
206         apdu.which = Z_APDU_close;
207         apdu.u.close = cls;
208         *cls->closeReason = reason;
209         cls->diagnosticInformation = message;
210         process_z_response(a, req, &apdu);
211         iochan_settimeout(a->client_chan, 20);
212     }
213     else
214     {
215         yaz_log(LOG_DEBUG, "v2 client. No Close PDU");
216         iochan_setevent(a->client_chan, EVENT_TIMEOUT); /* force imm close */
217     }
218     a->state = ASSOC_DEAD;
219 }
220
221 static void do_close(association *a, int reason, char *message)
222 {
223     do_close_req (a, reason, message, request_get(&a->outgoing));
224 }
225
226 /*
227  * This is where PDUs from the client are read and the further
228  * processing is initiated. Flow of control moves down through the
229  * various process_* functions below, until the encoded result comes back up
230  * to the output handler in here.
231  * 
232  *  h     : the I/O channel that has an outstanding event.
233  *  event : the current outstanding event.
234  */
235 void ir_session(IOCHAN h, int event)
236 {
237     int res;
238     association *assoc = (association *)iochan_getdata(h);
239     COMSTACK conn = assoc->client_link;
240     request *req;
241
242     assert(h && conn && assoc);
243     if (event == EVENT_TIMEOUT)
244     {
245         if (assoc->state != ASSOC_UP)
246         {
247             yaz_log(LOG_LOG, "Final timeout - closing connection.");
248             cs_close(conn);
249             destroy_association(assoc);
250             iochan_destroy(h);
251         }
252         else
253         {
254             yaz_log(LOG_LOG, "Session idle too long. Sending close.");
255             do_close(assoc, Z_Close_lackOfActivity, 0);
256         }
257         return;
258     }
259     if (event & assoc->cs_accept_mask)
260     {
261         yaz_log (LOG_DEBUG, "ir_session (accept)");
262         if (!cs_accept (conn))
263         {
264             yaz_log (LOG_LOG, "accept failed");
265             destroy_association(assoc);
266             iochan_destroy(h);
267         }
268         iochan_clearflag (h, EVENT_OUTPUT|EVENT_OUTPUT);
269         if (conn->io_pending) 
270         {   /* cs_accept didn't complete */
271             assoc->cs_accept_mask = 
272                 ((conn->io_pending & CS_WANT_WRITE) ? EVENT_OUTPUT : 0) |
273                 ((conn->io_pending & CS_WANT_READ) ? EVENT_INPUT : 0);
274
275             iochan_setflag (h, assoc->cs_accept_mask);
276         }
277         else
278         {   /* cs_accept completed. Prepare for reading (cs_get) */
279             assoc->cs_accept_mask = 0;
280             assoc->cs_get_mask = EVENT_INPUT;
281             iochan_setflag (h, assoc->cs_get_mask);
282         }
283         return;
284     }
285     if ((event & assoc->cs_get_mask) || (event & EVENT_WORK)) /* input */
286     {
287         if ((assoc->cs_put_mask & EVENT_INPUT) == 0 && (event & assoc->cs_get_mask))
288         {
289             yaz_log(LOG_DEBUG, "ir_session (input)");
290             /* We aren't speaking to this fellow */
291             if (assoc->state == ASSOC_DEAD)
292             {
293                 yaz_log(LOG_LOG, "Connection closed - end of session");
294                 cs_close(conn);
295                 destroy_association(assoc);
296                 iochan_destroy(h);
297                 return;
298             }
299             assoc->cs_get_mask = EVENT_INPUT;
300             if ((res = cs_get(conn, &assoc->input_buffer,
301                 &assoc->input_buffer_len)) <= 0)
302             {
303                 yaz_log(LOG_LOG, "Connection closed by client");
304                 cs_close(conn);
305                 destroy_association(assoc);
306                 iochan_destroy(h);
307                 return;
308             }
309             else if (res == 1) /* incomplete read - wait for more  */
310             {
311                 if (conn->io_pending & CS_WANT_WRITE)
312                     assoc->cs_get_mask |= EVENT_OUTPUT;
313                 iochan_setflag(h, assoc->cs_get_mask);
314                 return;
315             }
316             if (cs_more(conn)) /* more stuff - call us again later, please */
317                 iochan_setevent(h, EVENT_INPUT);
318                 
319             /* we got a complete PDU. Let's decode it */
320             yaz_log(LOG_DEBUG, "Got PDU, %d bytes: lead=%02X %02X %02X", res,
321                             assoc->input_buffer[0] & 0xff,
322                             assoc->input_buffer[1] & 0xff,
323                             assoc->input_buffer[2] & 0xff);
324             req = request_get(&assoc->incoming); /* get a new request */
325             odr_reset(assoc->decode);
326             odr_setbuf(assoc->decode, assoc->input_buffer, res, 0);
327             if (!z_GDU(assoc->decode, &req->gdu_request, 0, 0))
328             {
329                 yaz_log(LOG_LOG, "ODR error on incoming PDU: %s [near byte %d] ",
330                         odr_errmsg(odr_geterror(assoc->decode)),
331                         odr_offset(assoc->decode));
332                 if (assoc->decode->error != OHTTP)
333                 {
334                     yaz_log(LOG_LOG, "PDU dump:");
335                     odr_dumpBER(yaz_log_file(), assoc->input_buffer, res);
336                     do_close(assoc, Z_Close_protocolError, "Malformed package");
337                 }
338                 else
339                 {
340                     Z_GDU *p = z_get_HTTP_Response(assoc->encode, 400);
341                     assoc->state = ASSOC_DEAD;
342                     process_gdu_response(assoc, req, p);
343                 }
344                 return;
345             }
346             req->request_mem = odr_extract_mem(assoc->decode);
347             if (assoc->print && !z_GDU(assoc->print, &req->gdu_request, 0, 0))
348             {
349                 yaz_log(LOG_WARN, "ODR print error: %s", 
350                     odr_errmsg(odr_geterror(assoc->print)));
351                 odr_reset(assoc->print);
352             }
353             request_enq(&assoc->incoming, req);
354         }
355
356         /* can we do something yet? */
357         req = request_head(&assoc->incoming);
358         if (req->state == REQUEST_IDLE)
359         {
360             request_deq(&assoc->incoming);
361             process_gdu_request(assoc, req);
362         }
363     }
364     if (event & assoc->cs_put_mask)
365     {
366         request *req = request_head(&assoc->outgoing);
367
368         assoc->cs_put_mask = 0;
369         yaz_log(LOG_DEBUG, "ir_session (output)");
370         req->state = REQUEST_PENDING;
371         switch (res = cs_put(conn, req->response, req->len_response))
372         {
373         case -1:
374             yaz_log(LOG_LOG, "Connection closed by client");
375             cs_close(conn);
376             destroy_association(assoc);
377             iochan_destroy(h);
378             break;
379         case 0: /* all sent - release the request structure */
380             yaz_log(LOG_DEBUG, "Wrote PDU, %d bytes", req->len_response);
381 #if 0
382             yaz_log(LOG_DEBUG, "HTTP out:\n%.*s", req->len_response,
383                     req->response);
384 #endif
385             nmem_destroy(req->request_mem);
386             request_deq(&assoc->outgoing);
387             request_release(req);
388             if (!request_head(&assoc->outgoing))
389             {   /* restore mask for cs_get operation ... */
390                 iochan_clearflag(h, EVENT_OUTPUT|EVENT_INPUT);
391                 iochan_setflag(h, assoc->cs_get_mask);
392                 if (assoc->state == ASSOC_DEAD)
393                     iochan_setevent(assoc->client_chan, EVENT_TIMEOUT);
394             }
395             else
396             {
397                 assoc->cs_put_mask = EVENT_OUTPUT;
398             }
399             break;
400         default:
401             if (conn->io_pending & CS_WANT_WRITE)
402                 assoc->cs_put_mask |= EVENT_OUTPUT;
403             if (conn->io_pending & CS_WANT_READ)
404                 assoc->cs_put_mask |= EVENT_INPUT;
405             iochan_setflag(h, assoc->cs_put_mask);
406         }
407     }
408     if (event & EVENT_EXCEPT)
409     {
410         yaz_log(LOG_LOG, "ir_session (exception)");
411         cs_close(conn);
412         destroy_association(assoc);
413         iochan_destroy(h);
414     }
415 }
416
417 static int process_z_request(association *assoc, request *req, char **msg);
418
419 static void assoc_init_reset(association *assoc)
420 {
421     xfree (assoc->init);
422     assoc->init = (bend_initrequest *) xmalloc (sizeof(*assoc->init));
423
424     assoc->init->stream = assoc->encode;
425     assoc->init->print = assoc->print;
426     assoc->init->auth = 0;
427     assoc->init->referenceId = 0;
428     assoc->init->implementation_version = 0;
429     assoc->init->implementation_id = 0;
430     assoc->init->implementation_name = 0;
431     assoc->init->bend_sort = NULL;
432     assoc->init->bend_search = NULL;
433     assoc->init->bend_present = NULL;
434     assoc->init->bend_esrequest = NULL;
435     assoc->init->bend_delete = NULL;
436     assoc->init->bend_scan = NULL;
437     assoc->init->bend_segment = NULL;
438     assoc->init->bend_fetch = NULL;
439     assoc->init->bend_explain = NULL;
440
441     assoc->init->charneg_request = NULL;
442     assoc->init->charneg_response = NULL;
443
444     assoc->init->decode = assoc->decode;
445     assoc->init->peer_name = 
446         odr_strdup (assoc->encode, cs_addrstr(assoc->client_link));
447 }
448
449 static int srw_bend_init(association *assoc)
450 {
451     const char *encoding = "UTF-8";
452     Z_External *ce;
453     bend_initresult *binitres;
454     statserv_options_block *cb = statserv_getcontrol();
455     
456     assoc_init_reset(assoc);
457
458     assoc->maximumRecordSize = 3000000;
459     assoc->preferredMessageSize = 3000000;
460 #if 1
461     ce = yaz_set_proposal_charneg(assoc->decode, &encoding, 1, 0, 0, 1);
462     assoc->init->charneg_request = ce->u.charNeg3;
463 #endif
464     if (!(binitres = (*cb->bend_init)(assoc->init)))
465     {
466         yaz_log(LOG_WARN, "Bad response from backend.");
467         return 0;
468     }
469     assoc->backend = binitres->handle;
470     return 1;
471 }
472
473 static int srw_bend_fetch(association *assoc, int pos,
474                           Z_SRW_searchRetrieveRequest *srw_req,
475                           Z_SRW_record *record)
476 {
477     bend_fetch_rr rr;
478     ODR o = assoc->encode;
479
480     rr.setname = "default";
481     rr.number = pos;
482     rr.referenceId = 0;
483     rr.request_format = VAL_TEXT_XML;
484     rr.request_format_raw = yaz_oidval_to_z3950oid(assoc->decode,
485                                                    CLASS_TRANSYN,
486                                                    VAL_TEXT_XML);
487     rr.comp = (Z_RecordComposition *)
488             odr_malloc(assoc->decode, sizeof(*rr.comp));
489     rr.comp->which = Z_RecordComp_complex;
490     rr.comp->u.complex = (Z_CompSpec *)
491             odr_malloc(assoc->decode, sizeof(Z_CompSpec));
492     rr.comp->u.complex->selectAlternativeSyntax = (bool_t *)
493         odr_malloc(assoc->encode, sizeof(bool_t));
494     *rr.comp->u.complex->selectAlternativeSyntax = 0;    
495     rr.comp->u.complex->num_dbSpecific = 0;
496     rr.comp->u.complex->dbSpecific = 0;
497     rr.comp->u.complex->num_recordSyntax = 0; 
498     rr.comp->u.complex->recordSyntax = 0;
499
500     rr.comp->u.complex->generic = (Z_Specification *) 
501             odr_malloc(assoc->decode, sizeof(Z_Specification));
502     rr.comp->u.complex->generic->which = Z_Schema_uri;
503     rr.comp->u.complex->generic->schema.uri = srw_req->recordSchema;
504     rr.comp->u.complex->generic->elementSpec = 0;
505     
506     rr.stream = assoc->encode;
507     rr.print = assoc->print;
508
509     rr.basename = 0;
510     rr.len = 0;
511     rr.record = 0;
512     rr.last_in_set = 0;
513     rr.output_format = VAL_TEXT_XML;
514     rr.output_format_raw = 0;
515     rr.errcode = 0;
516     rr.errstring = 0;
517     rr.surrogate_flag = 0;
518
519     if (!assoc->init->bend_fetch)
520         return 1;
521
522     (*assoc->init->bend_fetch)(assoc->backend, &rr);
523
524     if (rr.len >= 0)
525     {
526         record->recordData_buf = rr.record;
527         record->recordData_len = rr.len;
528         record->recordPosition = odr_intdup(o, pos);
529         record->recordSchema = 0;
530         if (srw_req->recordSchema)
531             record->recordSchema = odr_strdup(o, srw_req->recordSchema);
532     }
533     return rr.errcode;
534 }
535
536 static void srw_bend_search(association *assoc, request *req,
537                             Z_SRW_searchRetrieveRequest *srw_req,
538                             Z_SRW_searchRetrieveResponse *srw_res)
539 {
540     int srw_error = 0;
541     bend_search_rr rr;
542     Z_External *ext;
543     
544     yaz_log(LOG_LOG, "Got SRW SearchRetrieveRequest");
545     yaz_log(LOG_DEBUG, "srw_bend_search");
546     if (!assoc->init)
547     {
548         yaz_log(LOG_DEBUG, "srw_bend_init");
549         if (!srw_bend_init(assoc))
550         {
551             srw_error = 3;  /* assume Authentication error */
552
553             srw_res->num_diagnostics = 1;
554             srw_res->diagnostics = (Z_SRW_diagnostic *)
555                 odr_malloc(assoc->encode, sizeof(*srw_res->diagnostics));
556             srw_res->diagnostics[0].code = 
557                 odr_intdup(assoc->encode, srw_error);
558             srw_res->diagnostics[0].details = 0;
559             return;
560         }
561     }
562     
563     rr.setname = "default";
564     rr.replace_set = 1;
565     rr.num_bases = 1;
566     rr.basenames = &srw_req->database;
567     rr.referenceId = 0;
568
569     rr.query = (Z_Query *) odr_malloc (assoc->decode, sizeof(*rr.query));
570
571     if (srw_req->query_type == Z_SRW_query_type_cql)
572     {
573         ext = (Z_External *) odr_malloc(assoc->decode, sizeof(*ext));
574         ext->direct_reference = odr_getoidbystr(assoc->decode, 
575                                                 "1.2.840.10003.16.2");
576         ext->indirect_reference = 0;
577         ext->descriptor = 0;
578         ext->which = Z_External_CQL;
579         ext->u.cql = srw_req->query.cql;
580
581         rr.query->which = Z_Query_type_104;
582         rr.query->u.type_104 =  ext;
583     }
584     else if (srw_req->query_type == Z_SRW_query_type_pqf)
585     {
586         Z_RPNQuery *RPNquery;
587         YAZ_PQF_Parser pqf_parser;
588
589         pqf_parser = yaz_pqf_create ();
590
591         RPNquery = yaz_pqf_parse (pqf_parser, assoc->decode,
592                                   srw_req->query.pqf);
593         if (!RPNquery)
594         {
595             const char *pqf_msg;
596             size_t off;
597             int code = yaz_pqf_error (pqf_parser, &pqf_msg, &off);
598             yaz_log(LOG_LOG, "%*s^\n", off+4, "");
599             yaz_log(LOG_LOG, "Bad PQF: %s (code %d)\n", pqf_msg, code);
600             
601             srw_error = 10;
602         }
603
604         rr.query->which = Z_Query_type_1;
605         rr.query->u.type_1 =  RPNquery;
606
607         yaz_pqf_destroy (pqf_parser);
608     }
609     else
610         srw_error = 11;
611
612     if (!srw_error && srw_req->sort_type != Z_SRW_sort_type_none)
613         srw_error = 80;
614
615     if (!srw_error && !assoc->init->bend_search)
616         srw_error = 1;
617
618     if (srw_error)
619     {
620         yaz_log(LOG_DEBUG, "srw_bend_search returned SRW error %d", srw_error);
621         srw_res->num_diagnostics = 1;
622         srw_res->diagnostics = (Z_SRW_diagnostic *)
623             odr_malloc(assoc->encode, sizeof(*srw_res->diagnostics));
624         srw_res->diagnostics[0].code = 
625             odr_intdup(assoc->encode, srw_error);
626         srw_res->diagnostics[0].details = 0;
627         return;
628     }
629     
630     rr.stream = assoc->encode;
631     rr.decode = assoc->decode;
632     rr.print = assoc->print;
633     rr.request = req;
634     rr.association = assoc;
635     rr.fd = 0;
636     rr.hits = 0;
637     rr.errcode = 0;
638     rr.errstring = 0;
639     rr.search_info = 0;
640     yaz_log_zquery(rr.query);
641     (assoc->init->bend_search)(assoc->backend, &rr);
642     srw_res->numberOfRecords = odr_intdup(assoc->encode, rr.hits);
643     if (rr.errcode)
644     {
645         yaz_log(LOG_DEBUG, "bend_search returned Bib-1 code %d", rr.errcode);
646         srw_res->num_diagnostics = 1;
647         srw_res->diagnostics = (Z_SRW_diagnostic *)
648             odr_malloc(assoc->encode, sizeof(*srw_res->diagnostics));
649         srw_res->diagnostics[0].code = 
650             odr_intdup(assoc->encode, 
651                        yaz_diag_bib1_to_srw (rr.errcode));
652         srw_res->diagnostics[0].details = rr.errstring;
653         yaz_log(LOG_DEBUG, "srw_bend_search returned SRW error %d",
654                 srw_res->diagnostics[0].code);
655                 
656     }
657     else
658     {
659         srw_res->numberOfRecords = odr_intdup(assoc->encode, rr.hits);
660         if (srw_req->maximumRecords && *srw_req->maximumRecords > 0)
661         {
662             int number = *srw_req->maximumRecords;
663             int start = 1;
664             int i;
665
666             if (srw_req->startRecord)
667                 start = *srw_req->startRecord;
668
669             yaz_log(LOG_DEBUG, "srw_bend_search. start=%d max=%d",
670                     start, *srw_req->maximumRecords);
671
672             if (start <= rr.hits)
673             {
674                 int j = 0;
675                 int packing = Z_SRW_recordPacking_string;
676                 if (start + number > rr.hits)
677                     number = rr.hits - start + 1;
678                 if (srw_req->recordPacking && 
679                     !strcmp(srw_req->recordPacking, "xml"))
680                     packing = Z_SRW_recordPacking_XML;
681                 srw_res->records = (Z_SRW_record *)
682                     odr_malloc(assoc->encode,
683                                number * sizeof(*srw_res->records));
684                 for (i = 0; i<number; i++)
685                 {
686                     int errcode;
687                     
688                     srw_res->records[j].recordPacking = packing;
689                     srw_res->records[j].recordData_buf = 0;
690                     yaz_log(LOG_DEBUG, "srw_bend_fetch %d", i+start);
691                     errcode = srw_bend_fetch(assoc, i+start, srw_req,
692                                              srw_res->records + j);
693                     if (errcode)
694                     {
695                         srw_res->num_diagnostics = 1;
696                         srw_res->diagnostics = (Z_SRW_diagnostic *)
697                             odr_malloc(assoc->encode, 
698                                        sizeof(*srw_res->diagnostics));
699                         srw_res->diagnostics[0].code = 
700                             odr_intdup(assoc->encode, 
701                                        yaz_diag_bib1_to_srw (errcode));
702                         srw_res->diagnostics[0].details = rr.errstring;
703                         break;
704                     }
705                     if (srw_res->records[j].recordData_buf)
706                         j++;
707                 }
708                 srw_res->num_records = j;
709                 if (!j)
710                     srw_res->records = 0;
711             }
712         }
713     }
714 }
715
716
717 static void srw_bend_explain(association *assoc, request *req,
718                              Z_SRW_explainRequest *srw_req,
719                              Z_SRW_explainResponse *srw_res)
720 {
721     yaz_log(LOG_LOG, "Got SRW ExplainRequest");
722     if (!assoc->init)
723     {
724         yaz_log(LOG_DEBUG, "srw_bend_init");
725         if (!srw_bend_init(assoc))
726             return;
727     }
728     if (assoc->init && assoc->init->bend_explain)
729     {
730         bend_explain_rr rr;
731
732         rr.stream = assoc->encode;
733         rr.decode = assoc->decode;
734         rr.print = assoc->print;
735         rr.explain_buf = 0;
736         (*assoc->init->bend_explain)(assoc->backend, &rr);
737         if (rr.explain_buf)
738         {
739             srw_res->explainData_buf = rr.explain_buf;
740             srw_res->explainData_len = strlen(rr.explain_buf);
741         }
742     }
743 }
744
745 static int hex_digit (int ch)
746 {
747     if (ch >= '0' && ch <= '9')
748         return ch - '0';
749     else if (ch >= 'a' && ch <= 'f')
750         return ch - 'a'+10;
751     else if (ch >= 'A' && ch <= 'F')
752         return ch - 'A'+10;
753     return 0;
754 }
755
756 static char *uri_val(const char *path, const char *name, ODR o)
757 {
758     size_t nlen = strlen(name);
759     if (*path != '?')
760         return 0;
761     path++;
762     while (path && *path)
763     {
764         const char *p1 = strchr(path, '=');
765         if (!p1)
766             break;
767         if (p1 - path == nlen && !memcmp(path, name, nlen))
768         {
769             size_t i = 0;
770             char *ret;
771             
772             path = p1 + 1;
773             p1 = strchr(path, '&');
774             if (!p1)
775                 p1 = strlen(path) + path;
776             ret = odr_malloc(o, p1 - path + 1);
777             while (*path && *path != '&')
778             {
779                 if (*path == '+')
780                 {
781                     ret[i++] = ' ';
782                     path++;
783                 }
784                 else if (*path == '%' && path[1] && path[2])
785                 {
786                     ret[i++] = hex_digit (path[1])*16 + hex_digit (path[2]);
787                     path = path + 3;
788                 }
789                 else
790                     ret[i++] = *path++;
791             }
792             ret[i] = '\0';
793             return ret;
794         }
795         path = strchr(p1, '&');
796         if (path)
797             path++;
798     }
799     return 0;
800 }
801
802 void uri_val_int(const char *path, const char *name, ODR o, int **intp)
803 {
804     const char *v = uri_val(path, name, o);
805     if (v)
806         *intp = odr_intdup(o, atoi(v));
807 }
808
809 static void process_http_request(association *assoc, request *req)
810 {
811     Z_HTTP_Request *hreq = req->gdu_request->u.HTTP_Request;
812     ODR o = assoc->encode;
813     Z_GDU *p = 0;
814     Z_HTTP_Response *hres = 0;
815     int keepalive = 1;
816
817     if (!strcmp(hreq->method, "GET"))
818     {
819         char *charset = 0;
820         int ret = -1;
821         Z_SOAP *soap_package = 0;
822         char *db = "Default";
823         const char *p0 = hreq->path, *p1;
824         static Z_SOAP_Handler soap_handlers[2] = {
825 #if HAVE_XML2
826             {"http://www.loc.gov/zing/srw/v1.0/", 0,
827              (Z_SOAP_fun) yaz_srw_codec},
828 #endif
829             {0, 0, 0}
830         };
831         
832         if (*p0 == '/')
833             p0++;
834         p1 = strchr(p0, '?');
835         if (!p1)
836             p1 = p0 + strlen(p0);
837         if (p1 != p0)
838         {
839             db = odr_malloc(assoc->decode, p1 - p0 + 1);
840             memcpy (db, p0, p1 - p0);
841             db[p1 - p0] = '\0';
842         }
843         if (p1 && *p1 == '?' && p1[1])
844         {
845             Z_SRW_PDU *res = yaz_srw_get(o, Z_SRW_searchRetrieve_response);
846             Z_SRW_PDU *sr = yaz_srw_get(o, Z_SRW_searchRetrieve_request);
847             char *query = uri_val(p1, "query", o);
848             char *pQuery = uri_val(p1, "pQuery", o);
849             char *sortKeys = uri_val(p1, "sortKeys", o);
850             
851             if (query)
852             {
853                 sr->u.request->query_type = Z_SRW_query_type_cql;
854                 sr->u.request->query.cql = query;
855             }
856             if (pQuery)
857             {
858                 sr->u.request->query_type = Z_SRW_query_type_pqf;
859                 sr->u.request->query.pqf = pQuery;
860             }
861             if (sortKeys)
862             {
863                 sr->u.request->sort_type = Z_SRW_sort_type_sort;
864                 sr->u.request->sort.sortKeys = sortKeys;
865             }
866             sr->u.request->recordSchema = uri_val(p1, "recordSchema", o);
867             sr->u.request->recordPacking = uri_val(p1, "recordPacking", o);
868             uri_val_int(p1, "maximumRecords", o, 
869                         &sr->u.request->maximumRecords);
870             uri_val_int(p1, "startRecord", o,
871                         &sr->u.request->startRecord);
872             if (sr->u.request->startRecord)
873                 yaz_log(LOG_LOG, "startRecord=%d", *sr->u.request->startRecord);
874             sr->u.request->database = db;
875             srw_bend_search(assoc, req, sr->u.request, res->u.response);
876             
877             soap_package = odr_malloc(o, sizeof(*soap_package));
878             soap_package->which = Z_SOAP_generic;
879
880             soap_package->u.generic =
881                 odr_malloc(o, sizeof(*soap_package->u.generic));
882
883             soap_package->u.generic->p = res;
884             soap_package->u.generic->ns = soap_handlers[0].ns;
885             soap_package->u.generic->no = 0;
886             
887             soap_package->ns = "SRU";
888
889             p = z_get_HTTP_Response(o, 200);
890             hres = p->u.HTTP_Response;
891
892             ret = z_soap_codec_enc(assoc->encode, &soap_package,
893                                    &hres->content_buf, &hres->content_len,
894                                    soap_handlers, charset);
895             if (!charset)
896                 z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/xml");
897             else
898             {
899                 char ctype[60];
900                 strcpy(ctype, "text/xml; charset=");
901                 strcat(ctype, charset);
902                 z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
903             }
904
905         }
906         else
907         {
908             Z_SRW_PDU *res = yaz_srw_get(o, Z_SRW_explain_response);
909             Z_SRW_PDU *sr = yaz_srw_get(o, Z_SRW_explain_request);
910
911             srw_bend_explain(assoc, req, sr->u.explain_request,
912                             res->u.explain_response);
913
914             if (res->u.explain_response->explainData_buf)
915             {
916                 soap_package = odr_malloc(o, sizeof(*soap_package));
917                 soap_package->which = Z_SOAP_generic;
918                 
919                 soap_package->u.generic =
920                     odr_malloc(o, sizeof(*soap_package->u.generic));
921                 
922                 soap_package->u.generic->p = res;
923                 soap_package->u.generic->ns = soap_handlers[0].ns;
924                 soap_package->u.generic->no = 0;
925                 
926                 soap_package->ns = "SRU";
927                 
928                 p = z_get_HTTP_Response(o, 200);
929                 hres = p->u.HTTP_Response;
930                 
931                 ret = z_soap_codec_enc(assoc->encode, &soap_package,
932                                        &hres->content_buf, &hres->content_len,
933                                        soap_handlers, charset);
934                 if (!charset)
935                     z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/xml");
936                 else
937                 {
938                     char ctype[60];
939                     strcpy(ctype, "text/xml; charset=");
940                     strcat(ctype, charset);
941                     z_HTTP_header_add(o, &hres->headers, "Content-Type",
942                                       ctype);
943                 }
944             }
945         }
946 #ifdef DOCDIR
947         if (strlen(hreq->path) >= 5 && strlen(hreq->path) < 80 &&
948                          !memcmp(hreq->path, "/doc/", 5))
949         {
950             FILE *f;
951             char fpath[120];
952
953             strcpy(fpath, DOCDIR);
954             strcat(fpath, hreq->path+4);
955             f = fopen(fpath, "rb");
956             if (f) {
957                 struct stat sbuf;
958                 if (fstat(fileno(f), &sbuf) || !S_ISREG(sbuf.st_mode))
959                 {
960                     fclose(f);
961                     f = 0;
962                 }
963             }
964             if (f)
965             {
966                 long sz;
967                 fseek(f, 0L, SEEK_END);
968                 sz = ftell(f);
969                 if (sz >= 0 && sz < 500000)
970                 {
971                     const char *ctype = "application/octet-stream";
972                     const char *cp;
973
974                     p = z_get_HTTP_Response(o, 200);
975                     hres = p->u.HTTP_Response;
976                     hres->content_buf = (char *) odr_malloc(o, sz + 1);
977                     hres->content_len = sz;
978                     fseek(f, 0L, SEEK_SET);
979                     fread(hres->content_buf, 1, sz, f);
980                     if ((cp = strrchr(fpath, '.'))) {
981                         cp++;
982                         if (!strcmp(cp, "png"))
983                             ctype = "image/png";
984                         else if (!strcmp(cp, "gif"))
985                             ctype = "image/gif";
986                         else if (!strcmp(cp, "xml"))
987                             ctype = "text/xml";
988                         else if (!strcmp(cp, "html"))
989                             ctype = "text/html";
990                     }
991                     z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
992                 }
993                 fclose(f);
994             }
995         }
996 #endif
997
998 #if 0
999         if (!strcmp(hreq->path, "/")) 
1000         {
1001 #ifdef DOCDIR
1002             struct stat sbuf;
1003 #endif
1004             const char *doclink = "";
1005             p = z_get_HTTP_Response(o, 200);
1006             hres = p->u.HTTP_Response;
1007             hres->content_buf = (char *) odr_malloc(o, 400);
1008 #ifdef DOCDIR
1009             if (stat(DOCDIR "/yaz.html", &sbuf) == 0 && S_ISREG(sbuf.st_mode))
1010                 doclink = "<P><A HREF=\"/doc/yaz.html\">Documentation</A></P>";
1011 #endif
1012             sprintf (hres->content_buf, 
1013                      "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\">\n"
1014                      "<HTML>\n"
1015                      " <HEAD>\n"
1016                      "  <TITLE>YAZ " YAZ_VERSION "</TITLE>\n"
1017                      " </HEAD>\n"
1018                      " <BODY>\n"
1019                      "  <P><A HREF=\"http://www.indexdata.dk/yaz/\">YAZ</A> " 
1020                      YAZ_VERSION "</P>\n"
1021                      "%s"
1022                      " </BODY>\n"
1023                      "</HTML>\n", doclink);
1024             hres->content_len = strlen(hres->content_buf);
1025             z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/html");
1026         }
1027 #endif
1028
1029         if (!p)
1030         {
1031             p = z_get_HTTP_Response(o, 404);
1032         }
1033     }
1034     else if (!strcmp(hreq->method, "POST"))
1035     {
1036         const char *content_type = z_HTTP_header_lookup(hreq->headers,
1037                                                         "Content-Type");
1038         if (content_type && !yaz_strcmp_del("text/xml", content_type, "; "))
1039         {
1040             Z_SOAP *soap_package = 0;
1041             int ret = -1;
1042             int http_code = 500;
1043             const char *charset_p = 0;
1044             char *charset = 0;
1045
1046             static Z_SOAP_Handler soap_handlers[2] = {
1047 #if HAVE_XML2
1048                 {"http://www.loc.gov/zing/srw/v1.0/", 0,
1049                  (Z_SOAP_fun) yaz_srw_codec},
1050 #endif
1051                 {0, 0, 0}
1052             };
1053             if ((charset_p = strstr(content_type, "; charset=")))
1054             {
1055                 int i = 0;
1056                 charset_p += 10;
1057                 while (i < 20 && charset_p[i] &&
1058                        !strchr("; \n\r", charset_p[i]))
1059                     i++;
1060                 charset = odr_malloc(assoc->encode, i+1);
1061                 memcpy(charset, charset_p, i);
1062                 charset[i] = '\0';
1063                 yaz_log(LOG_LOG, "SOAP encoding %s", charset);
1064             }
1065             ret = z_soap_codec(assoc->decode, &soap_package, 
1066                                &hreq->content_buf, &hreq->content_len,
1067                                soap_handlers);
1068 #if HAVE_XML2
1069             if (!ret && soap_package->which == Z_SOAP_generic &&
1070                 soap_package->u.generic->no == 0)
1071             {
1072                 /* SRW package */
1073                 Z_SRW_PDU *sr = soap_package->u.generic->p;
1074                 
1075                 if (sr->which == Z_SRW_searchRetrieve_request)
1076                 {
1077                     Z_SRW_PDU *res =
1078                         yaz_srw_get(assoc->encode,
1079                                     Z_SRW_searchRetrieve_response);
1080
1081                     if (!sr->u.request->database)
1082                     {
1083                         const char *p0 = hreq->path, *p1;
1084                         if (*p0 == '/')
1085                             p0++;
1086                         p1 = strchr(p0, '?');
1087                         if (!p1)
1088                             p1 = p0 + strlen(p0);
1089                         if (p1 != p0)
1090                         {
1091                             sr->u.request->database =
1092                                 odr_malloc(assoc->decode, p1 - p0 + 1);
1093                             memcpy (sr->u.request->database, p0, p1 - p0);
1094                             sr->u.request->database[p1 - p0] = '\0';
1095                         }
1096                         else
1097                             sr->u.request->database = "Default";
1098                     }
1099                     srw_bend_search(assoc, req, sr->u.request,
1100                                     res->u.response);
1101                     
1102                     soap_package->u.generic->p = res;
1103                     http_code = 200;
1104                 }
1105                 else if (sr->which == Z_SRW_explain_request)
1106                 {
1107                     Z_SRW_PDU *res =
1108                         yaz_srw_get(assoc->encode, Z_SRW_explain_response);
1109
1110                     srw_bend_explain(assoc, req, sr->u.explain_request,
1111                                      res->u.explain_response);
1112                     if (!res->u.explain_response->explainData_buf)
1113                     {
1114                         z_soap_error(assoc->encode, soap_package,
1115                                      "SOAP-ENV:Client", "Explain Not Supported", 0);
1116                     }
1117                     else
1118                     {
1119                         soap_package->u.generic->p = res;
1120                         http_code = 200;
1121                     }
1122                 }
1123                 else
1124                 {
1125                     z_soap_error(assoc->encode, soap_package,
1126                                  "SOAP-ENV:Client", "Bad method", 0); 
1127                 }
1128             }
1129 #endif
1130             p = z_get_HTTP_Response(o, 200);
1131             hres = p->u.HTTP_Response;
1132             ret = z_soap_codec_enc(assoc->encode, &soap_package,
1133                                    &hres->content_buf, &hres->content_len,
1134                                    soap_handlers, charset);
1135             hres->code = http_code;
1136             if (!charset)
1137                 z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/xml");
1138             else
1139             {
1140                 char ctype[60];
1141                 strcpy(ctype, "text/xml; charset=");
1142                 strcat(ctype, charset);
1143                 z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
1144             }
1145         }
1146         if (!p) /* still no response ? */
1147             p = z_get_HTTP_Response(o, 500);
1148     }
1149     else
1150     {
1151         p = z_get_HTTP_Response(o, 405);
1152         hres = p->u.HTTP_Response;
1153
1154         z_HTTP_header_add(o, &hres->headers, "Allow", "GET, POST");
1155     }
1156     hres = p->u.HTTP_Response;
1157     if (!strcmp(hreq->version, "1.0")) 
1158     {
1159         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1160         if (v && !strcmp(v, "Keep-Alive"))
1161             keepalive = 1;
1162         else
1163             keepalive = 0;
1164         hres->version = "1.0";
1165     }
1166     else
1167     {
1168         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1169         if (v && !strcmp(v, "close"))
1170             keepalive = 0;
1171         else
1172             keepalive = 1;
1173         hres->version = "1.1";
1174     }
1175     if (!keepalive)
1176     {
1177         z_HTTP_header_add(o, &hres->headers, "Connection", "close");
1178         assoc->state = ASSOC_DEAD;
1179     }
1180     else
1181     {
1182         int t;
1183         const char *alive = z_HTTP_header_lookup(hreq->headers, "Keep-Alive");
1184
1185         if (alive && isdigit(*alive))
1186             t = atoi(alive);
1187         else
1188             t = 15;
1189         if (t < 0 || t > 3600)
1190             t = 3600;
1191         iochan_settimeout(assoc->client_chan,t);
1192         z_HTTP_header_add(o, &hres->headers, "Connection", "Keep-Alive");
1193     }
1194     process_gdu_response(assoc, req, p);
1195 }
1196
1197 static void process_gdu_request(association *assoc, request *req)
1198 {
1199     if (req->gdu_request->which == Z_GDU_Z3950)
1200     {
1201         char *msg = 0;
1202         req->apdu_request = req->gdu_request->u.z3950;
1203         if (process_z_request(assoc, req, &msg) < 0)
1204             do_close_req(assoc, Z_Close_systemProblem, msg, req);
1205     }
1206     else if (req->gdu_request->which == Z_GDU_HTTP_Request)
1207         process_http_request(assoc, req);
1208     else
1209     {
1210         do_close_req(assoc, Z_Close_systemProblem, "bad protocol packet", req);
1211     }
1212 }
1213
1214 /*
1215  * Initiate request processing.
1216  */
1217 static int process_z_request(association *assoc, request *req, char **msg)
1218 {
1219     int fd = -1;
1220     Z_APDU *res;
1221     int retval;
1222     
1223     *msg = "Unknown Error";
1224     assert(req && req->state == REQUEST_IDLE);
1225     if (req->apdu_request->which != Z_APDU_initRequest && !assoc->init)
1226     {
1227         *msg = "Missing InitRequest";
1228         return -1;
1229     }
1230     switch (req->apdu_request->which)
1231     {
1232     case Z_APDU_initRequest:
1233         iochan_settimeout(assoc->client_chan,
1234                           statserv_getcontrol()->idle_timeout * 60);
1235         res = process_initRequest(assoc, req); break;
1236     case Z_APDU_searchRequest:
1237         res = process_searchRequest(assoc, req, &fd); break;
1238     case Z_APDU_presentRequest:
1239         res = process_presentRequest(assoc, req, &fd); break;
1240     case Z_APDU_scanRequest:
1241         if (assoc->init->bend_scan)
1242             res = process_scanRequest(assoc, req, &fd);
1243         else
1244         {
1245             *msg = "Cannot handle Scan APDU";
1246             return -1;
1247         }
1248         break;
1249     case Z_APDU_extendedServicesRequest:
1250         if (assoc->init->bend_esrequest)
1251             res = process_ESRequest(assoc, req, &fd);
1252         else
1253         {
1254             *msg = "Cannot handle Extended Services APDU";
1255             return -1;
1256         }
1257         break;
1258     case Z_APDU_sortRequest:
1259         if (assoc->init->bend_sort)
1260             res = process_sortRequest(assoc, req, &fd);
1261         else
1262         {
1263             *msg = "Cannot handle Sort APDU";
1264             return -1;
1265         }
1266         break;
1267     case Z_APDU_close:
1268         process_close(assoc, req);
1269         return 0;
1270     case Z_APDU_deleteResultSetRequest:
1271         if (assoc->init->bend_delete)
1272             res = process_deleteRequest(assoc, req, &fd);
1273         else
1274         {
1275             *msg = "Cannot handle Delete APDU";
1276             return -1;
1277         }
1278         break;
1279     case Z_APDU_segmentRequest:
1280         if (assoc->init->bend_segment)
1281         {
1282             res = process_segmentRequest (assoc, req);
1283         }
1284         else
1285         {
1286             *msg = "Cannot handle Segment APDU";
1287             return -1;
1288         }
1289         break;
1290     default:
1291         *msg = "Bad APDU received";
1292         return -1;
1293     }
1294     if (res)
1295     {
1296         yaz_log(LOG_DEBUG, "  result immediately available");
1297         retval = process_z_response(assoc, req, res);
1298     }
1299     else if (fd < 0)
1300     {
1301         yaz_log(LOG_DEBUG, "  result unavailble");
1302         retval = 0;
1303     }
1304     else /* no result yet - one will be provided later */
1305     {
1306         IOCHAN chan;
1307
1308         /* Set up an I/O handler for the fd supplied by the backend */
1309
1310         yaz_log(LOG_DEBUG, "   establishing handler for result");
1311         req->state = REQUEST_PENDING;
1312         if (!(chan = iochan_create(fd, backend_response, EVENT_INPUT)))
1313             abort();
1314         iochan_setdata(chan, assoc);
1315         retval = 0;
1316     }
1317     return retval;
1318 }
1319
1320 /*
1321  * Handle message from the backend.
1322  */
1323 void backend_response(IOCHAN i, int event)
1324 {
1325     association *assoc = (association *)iochan_getdata(i);
1326     request *req = request_head(&assoc->incoming);
1327     Z_APDU *res;
1328     int fd;
1329
1330     yaz_log(LOG_DEBUG, "backend_response");
1331     assert(assoc && req && req->state != REQUEST_IDLE);
1332     /* determine what it is we're waiting for */
1333     switch (req->apdu_request->which)
1334     {
1335         case Z_APDU_searchRequest:
1336             res = response_searchRequest(assoc, req, 0, &fd); break;
1337 #if 0
1338         case Z_APDU_presentRequest:
1339             res = response_presentRequest(assoc, req, 0, &fd); break;
1340         case Z_APDU_scanRequest:
1341             res = response_scanRequest(assoc, req, 0, &fd); break;
1342 #endif
1343         default:
1344             yaz_log(LOG_WARN, "Serious programmer's lapse or bug");
1345             abort();
1346     }
1347     if ((res && process_z_response(assoc, req, res) < 0) || fd < 0)
1348     {
1349         yaz_log(LOG_LOG, "Fatal error when talking to backend");
1350         do_close(assoc, Z_Close_systemProblem, 0);
1351         iochan_destroy(i);
1352         return;
1353     }
1354     else if (!res) /* no result yet - try again later */
1355     {
1356         yaz_log(LOG_DEBUG, "   no result yet");
1357         iochan_setfd(i, fd); /* in case fd has changed */
1358     }
1359 }
1360
1361 /*
1362  * Encode response, and transfer the request structure to the outgoing queue.
1363  */
1364 static int process_gdu_response(association *assoc, request *req, Z_GDU *res)
1365 {
1366     odr_setbuf(assoc->encode, req->response, req->size_response, 1);
1367
1368     if (assoc->print && !z_GDU(assoc->print, &res, 0, 0))
1369     {
1370         yaz_log(LOG_WARN, "ODR print error: %s", 
1371             odr_errmsg(odr_geterror(assoc->print)));
1372         odr_reset(assoc->print);
1373     }
1374     if (!z_GDU(assoc->encode, &res, 0, 0))
1375     {
1376         yaz_log(LOG_WARN, "ODR error when encoding response: %s",
1377             odr_errmsg(odr_geterror(assoc->decode)));
1378         return -1;
1379     }
1380     req->response = odr_getbuf(assoc->encode, &req->len_response,
1381         &req->size_response);
1382     odr_setbuf(assoc->encode, 0, 0, 0); /* don'txfree if we abort later */
1383     odr_reset(assoc->encode);
1384     req->state = REQUEST_IDLE;
1385     request_enq(&assoc->outgoing, req);
1386     /* turn the work over to the ir_session handler */
1387     iochan_setflag(assoc->client_chan, EVENT_OUTPUT);
1388     assoc->cs_put_mask = EVENT_OUTPUT;
1389     /* Is there more work to be done? give that to the input handler too */
1390 #if 1
1391     if (request_head(&assoc->incoming))
1392     {
1393         yaz_log (LOG_DEBUG, "more work to be done");
1394         iochan_setevent(assoc->client_chan, EVENT_WORK);
1395     }
1396 #endif
1397     return 0;
1398 }
1399
1400 /*
1401  * Encode response, and transfer the request structure to the outgoing queue.
1402  */
1403 static int process_z_response(association *assoc, request *req, Z_APDU *res)
1404 {
1405     Z_GDU *gres = (Z_GDU *) odr_malloc(assoc->encode, sizeof(*res));
1406     gres->which = Z_GDU_Z3950;
1407     gres->u.z3950 = res;
1408
1409     return process_gdu_response(assoc, req, gres);
1410 }
1411
1412
1413 /*
1414  * Handle init request.
1415  * At the moment, we don't check the options
1416  * anywhere else in the code - we just try not to do anything that would
1417  * break a naive client. We'll toss 'em into the association block when
1418  * we need them there.
1419  */
1420 static Z_APDU *process_initRequest(association *assoc, request *reqb)
1421 {
1422     statserv_options_block *cb = statserv_getcontrol();
1423     Z_InitRequest *req = reqb->apdu_request->u.initRequest;
1424     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_initResponse);
1425     Z_InitResponse *resp = apdu->u.initResponse;
1426     bend_initresult *binitres;
1427
1428     char options[140];
1429
1430     yaz_log(LOG_LOG, "Got initRequest");
1431     if (req->implementationId)
1432         yaz_log(LOG_LOG, "Id:        %s", req->implementationId);
1433     if (req->implementationName)
1434         yaz_log(LOG_LOG, "Name:      %s", req->implementationName);
1435     if (req->implementationVersion)
1436         yaz_log(LOG_LOG, "Version:   %s", req->implementationVersion);
1437
1438     assoc_init_reset(assoc);
1439
1440     assoc->init->auth = req->idAuthentication;
1441     assoc->init->referenceId = req->referenceId;
1442
1443     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel))
1444     {
1445         Z_CharSetandLanguageNegotiation *negotiation =
1446             yaz_get_charneg_record (req->otherInfo);
1447         if (negotiation->which == Z_CharSetandLanguageNegotiation_proposal)
1448             assoc->init->charneg_request = negotiation;
1449     }
1450     
1451     if (!(binitres = (*cb->bend_init)(assoc->init)))
1452     {
1453         yaz_log(LOG_WARN, "Bad response from backend.");
1454         return 0;
1455     }
1456
1457     assoc->backend = binitres->handle;
1458     if ((assoc->init->bend_sort))
1459         yaz_log (LOG_DEBUG, "Sort handler installed");
1460     if ((assoc->init->bend_search))
1461         yaz_log (LOG_DEBUG, "Search handler installed");
1462     if ((assoc->init->bend_present))
1463         yaz_log (LOG_DEBUG, "Present handler installed");   
1464     if ((assoc->init->bend_esrequest))
1465         yaz_log (LOG_DEBUG, "ESRequest handler installed");   
1466     if ((assoc->init->bend_delete))
1467         yaz_log (LOG_DEBUG, "Delete handler installed");   
1468     if ((assoc->init->bend_scan))
1469         yaz_log (LOG_DEBUG, "Scan handler installed");   
1470     if ((assoc->init->bend_segment))
1471         yaz_log (LOG_DEBUG, "Segment handler installed");   
1472     
1473     resp->referenceId = req->referenceId;
1474     *options = '\0';
1475     /* let's tell the client what we can do */
1476     if (ODR_MASK_GET(req->options, Z_Options_search))
1477     {
1478         ODR_MASK_SET(resp->options, Z_Options_search);
1479         strcat(options, "srch");
1480     }
1481     if (ODR_MASK_GET(req->options, Z_Options_present))
1482     {
1483         ODR_MASK_SET(resp->options, Z_Options_present);
1484         strcat(options, " prst");
1485     }
1486     if (ODR_MASK_GET(req->options, Z_Options_delSet) &&
1487         assoc->init->bend_delete)
1488     {
1489         ODR_MASK_SET(resp->options, Z_Options_delSet);
1490         strcat(options, " del");
1491     }
1492     if (ODR_MASK_GET(req->options, Z_Options_extendedServices) &&
1493         assoc->init->bend_esrequest)
1494     {
1495         ODR_MASK_SET(resp->options, Z_Options_extendedServices);
1496         strcat (options, " extendedServices");
1497     }
1498     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
1499     {
1500         ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
1501         strcat(options, " namedresults");
1502     }
1503     if (ODR_MASK_GET(req->options, Z_Options_scan) && assoc->init->bend_scan)
1504     {
1505         ODR_MASK_SET(resp->options, Z_Options_scan);
1506         strcat(options, " scan");
1507     }
1508     if (ODR_MASK_GET(req->options, Z_Options_concurrentOperations))
1509     {
1510         ODR_MASK_SET(resp->options, Z_Options_concurrentOperations);
1511         strcat(options, " concurrop");
1512     }
1513     if (ODR_MASK_GET(req->options, Z_Options_sort) && assoc->init->bend_sort)
1514     {
1515         ODR_MASK_SET(resp->options, Z_Options_sort);
1516         strcat(options, " sort");
1517     }
1518
1519     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel)
1520         && assoc->init->charneg_response)
1521     {
1522         Z_OtherInformation **p;
1523         Z_OtherInformationUnit *p0;
1524         
1525         yaz_oi_APDU(apdu, &p);
1526         
1527         if ((p0=yaz_oi_update(p, assoc->encode, NULL, 0, 0))) {
1528             ODR_MASK_SET(resp->options, Z_Options_negotiationModel);
1529             
1530             p0->which = Z_OtherInfo_externallyDefinedInfo;
1531             p0->information.externallyDefinedInfo =
1532                 assoc->init->charneg_response;
1533         }
1534         ODR_MASK_SET(resp->options, Z_Options_negotiationModel);
1535         strcat(options, " negotiation");
1536     }
1537
1538     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_1))
1539     {
1540         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_1);
1541         assoc->version = 2; /* 1 & 2 are equivalent */
1542     }
1543     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_2))
1544     {
1545         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_2);
1546         assoc->version = 2;
1547     }
1548     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_3))
1549     {
1550         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_3);
1551         assoc->version = 3;
1552     }
1553
1554     yaz_log(LOG_LOG, "Negotiated to v%d: %s", assoc->version, options);
1555     assoc->maximumRecordSize = *req->maximumRecordSize;
1556     if (assoc->maximumRecordSize > control_block->maxrecordsize)
1557         assoc->maximumRecordSize = control_block->maxrecordsize;
1558     assoc->preferredMessageSize = *req->preferredMessageSize;
1559     if (assoc->preferredMessageSize > assoc->maximumRecordSize)
1560         assoc->preferredMessageSize = assoc->maximumRecordSize;
1561
1562     resp->preferredMessageSize = &assoc->preferredMessageSize;
1563     resp->maximumRecordSize = &assoc->maximumRecordSize;
1564
1565     resp->implementationName = "GFS/YAZ";
1566
1567     if (assoc->init->implementation_id)
1568     {
1569         char *nv = (char *)
1570             odr_malloc (assoc->encode,
1571                         strlen(assoc->init->implementation_id) + 10 + 
1572                                strlen(resp->implementationId));
1573         sprintf (nv, "%s / %s",
1574                  resp->implementationId, assoc->init->implementation_id);
1575         resp->implementationId = nv;
1576     }
1577     if (assoc->init->implementation_name)
1578     {
1579         char *nv = (char *)
1580             odr_malloc (assoc->encode,
1581                         strlen(assoc->init->implementation_name) + 10 + 
1582                                strlen(resp->implementationName));
1583         sprintf (nv, "%s / %s",
1584                  resp->implementationName, assoc->init->implementation_name);
1585         resp->implementationName = nv;
1586     }
1587     if (assoc->init->implementation_version)
1588     {
1589         char *nv = (char *)
1590             odr_malloc (assoc->encode,
1591                         strlen(assoc->init->implementation_version) + 10 + 
1592                                strlen(resp->implementationVersion));
1593         sprintf (nv, "YAZ %s / %s",
1594                  resp->implementationVersion,
1595                  assoc->init->implementation_version);
1596         resp->implementationVersion = nv;
1597     }
1598
1599     if (binitres->errcode)
1600     {
1601         yaz_log(LOG_LOG, "Connection rejected by backend.");
1602         *resp->result = 0;
1603         assoc->state = ASSOC_DEAD;
1604     }
1605     else
1606         assoc->state = ASSOC_UP;
1607     return apdu;
1608 }
1609
1610 /*
1611  * These functions should be merged.
1612  */
1613
1614 static void set_addinfo (Z_DefaultDiagFormat *dr, char *addinfo, ODR odr)
1615 {
1616     dr->which = Z_DefaultDiagFormat_v2Addinfo;
1617     dr->u.v2Addinfo = odr_strdup (odr, addinfo ? addinfo : "");
1618 }
1619
1620 /*
1621  * nonsurrogate diagnostic record.
1622  */
1623 static Z_Records *diagrec(association *assoc, int error, char *addinfo)
1624 {
1625     Z_Records *rec = (Z_Records *)
1626         odr_malloc (assoc->encode, sizeof(*rec));
1627     int *err = odr_intdup(assoc->encode, error);
1628     Z_DiagRec *drec = (Z_DiagRec *)
1629         odr_malloc (assoc->encode, sizeof(*drec));
1630     Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *)
1631         odr_malloc (assoc->encode, sizeof(*dr));
1632
1633     yaz_log(LOG_LOG, "[%d] %s %s%s", error, diagbib1_str(error),
1634         addinfo ? " -- " : "", addinfo ? addinfo : "");
1635     rec->which = Z_Records_NSD;
1636     rec->u.nonSurrogateDiagnostic = dr;
1637     dr->diagnosticSetId =
1638         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
1639     dr->condition = err;
1640     set_addinfo (dr, addinfo, assoc->encode);
1641     return rec;
1642 }
1643
1644 /*
1645  * surrogate diagnostic.
1646  */
1647 static Z_NamePlusRecord *surrogatediagrec(association *assoc, char *dbname,
1648                                           int error, char *addinfo)
1649 {
1650     Z_NamePlusRecord *rec = (Z_NamePlusRecord *)
1651         odr_malloc (assoc->encode, sizeof(*rec));
1652     int *err = odr_intdup(assoc->encode, error);
1653     Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (assoc->encode, sizeof(*drec));
1654     Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *)
1655         odr_malloc (assoc->encode, sizeof(*dr));
1656     
1657     yaz_log(LOG_DEBUG, "SurrogateDiagnotic: %d -- %s", error, addinfo);
1658     rec->databaseName = dbname;
1659     rec->which = Z_NamePlusRecord_surrogateDiagnostic;
1660     rec->u.surrogateDiagnostic = drec;
1661     drec->which = Z_DiagRec_defaultFormat;
1662     drec->u.defaultFormat = dr;
1663     dr->diagnosticSetId =
1664         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
1665     dr->condition = err;
1666     set_addinfo (dr, addinfo, assoc->encode);
1667
1668     return rec;
1669 }
1670
1671 /*
1672  * multiple nonsurrogate diagnostics.
1673  */
1674 static Z_DiagRecs *diagrecs(association *assoc, int error, char *addinfo)
1675 {
1676     Z_DiagRecs *recs = (Z_DiagRecs *)odr_malloc (assoc->encode, sizeof(*recs));
1677     int *err = odr_intdup(assoc->encode, error);
1678     Z_DiagRec **recp = (Z_DiagRec **)odr_malloc (assoc->encode, sizeof(*recp));
1679     Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (assoc->encode, sizeof(*drec));
1680     Z_DefaultDiagFormat *rec = (Z_DefaultDiagFormat *)
1681         odr_malloc (assoc->encode, sizeof(*rec));
1682
1683     yaz_log(LOG_DEBUG, "DiagRecs: %d -- %s", error, addinfo ? addinfo : "");
1684
1685     recs->num_diagRecs = 1;
1686     recs->diagRecs = recp;
1687     recp[0] = drec;
1688     drec->which = Z_DiagRec_defaultFormat;
1689     drec->u.defaultFormat = rec;
1690
1691     rec->diagnosticSetId =
1692         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
1693     rec->condition = err;
1694
1695     rec->which = Z_DefaultDiagFormat_v2Addinfo;
1696     rec->u.v2Addinfo = odr_strdup (assoc->encode, addinfo ? addinfo : "");
1697     return recs;
1698 }
1699
1700 static Z_Records *pack_records(association *a, char *setname, int start,
1701                                int *num, Z_RecordComposition *comp,
1702                                int *next, int *pres, oid_value format,
1703                                Z_ReferenceId *referenceId,
1704                                int *oid)
1705 {
1706     int recno, total_length = 0, toget = *num, dumped_records = 0;
1707     Z_Records *records =
1708         (Z_Records *) odr_malloc (a->encode, sizeof(*records));
1709     Z_NamePlusRecordList *reclist =
1710         (Z_NamePlusRecordList *) odr_malloc (a->encode, sizeof(*reclist));
1711     Z_NamePlusRecord **list =
1712         (Z_NamePlusRecord **) odr_malloc (a->encode, sizeof(*list) * toget);
1713
1714     records->which = Z_Records_DBOSD;
1715     records->u.databaseOrSurDiagnostics = reclist;
1716     reclist->num_records = 0;
1717     reclist->records = list;
1718     *pres = Z_PRES_SUCCESS;
1719     *num = 0;
1720     *next = 0;
1721
1722     yaz_log(LOG_LOG, "Request to pack %d+%d+%s", start, toget, setname);
1723     yaz_log(LOG_DEBUG, "pms=%d, mrs=%d", a->preferredMessageSize,
1724         a->maximumRecordSize);
1725     for (recno = start; reclist->num_records < toget; recno++)
1726     {
1727         bend_fetch_rr freq;
1728         Z_NamePlusRecord *thisrec;
1729         int this_length = 0;
1730         /*
1731          * we get the number of bytes allocated on the stream before any
1732          * allocation done by the backend - this should give us a reasonable
1733          * idea of the total size of the data so far.
1734          */
1735         total_length = odr_total(a->encode) - dumped_records;
1736         freq.errcode = 0;
1737         freq.errstring = 0;
1738         freq.basename = 0;
1739         freq.len = 0;
1740         freq.record = 0;
1741         freq.last_in_set = 0;
1742         freq.setname = setname;
1743         freq.surrogate_flag = 0;
1744         freq.number = recno;
1745         freq.comp = comp;
1746         freq.request_format = format;
1747         freq.request_format_raw = oid;
1748         freq.output_format = format;
1749         freq.output_format_raw = 0;
1750         freq.stream = a->encode;
1751         freq.print = a->print;
1752         freq.surrogate_flag = 0;
1753         freq.referenceId = referenceId;
1754         (*a->init->bend_fetch)(a->backend, &freq);
1755         /* backend should be able to signal whether error is system-wide
1756            or only pertaining to current record */
1757         if (freq.errcode)
1758         {
1759             if (!freq.surrogate_flag)
1760             {
1761                 char s[20];
1762                 *pres = Z_PRES_FAILURE;
1763                 /* for 'present request out of range',
1764                    set addinfo to record position if not set */
1765                 if (freq.errcode == 13 && freq.errstring == 0)
1766                 {
1767                     sprintf (s, "%d", recno);
1768                     freq.errstring = s;
1769                 }
1770                 return diagrec(a, freq.errcode, freq.errstring);
1771             }
1772             reclist->records[reclist->num_records] =
1773                 surrogatediagrec(a, freq.basename, freq.errcode,
1774                                  freq.errstring);
1775             reclist->num_records++;
1776             *next = freq.last_in_set ? 0 : recno + 1;
1777             continue;
1778         }
1779         if (freq.len >= 0)
1780             this_length = freq.len;
1781         else
1782             this_length = odr_total(a->encode) - total_length;
1783         yaz_log(LOG_DEBUG, "  fetched record, len=%d, total=%d",
1784             this_length, total_length);
1785         if (this_length + total_length > a->preferredMessageSize)
1786         {
1787             /* record is small enough, really */
1788             if (this_length <= a->preferredMessageSize)
1789             {
1790                 yaz_log(LOG_DEBUG, "  Dropped last normal-sized record");
1791                 *pres = Z_PRES_PARTIAL_2;
1792                 break;
1793             }
1794             /* record can only be fetched by itself */
1795             if (this_length < a->maximumRecordSize)
1796             {
1797                 yaz_log(LOG_DEBUG, "  Record > prefmsgsz");
1798                 if (toget > 1)
1799                 {
1800                     yaz_log(LOG_DEBUG, "  Dropped it");
1801                     reclist->records[reclist->num_records] =
1802                          surrogatediagrec(a, freq.basename, 16, 0);
1803                     reclist->num_records++;
1804                     *next = freq.last_in_set ? 0 : recno + 1;
1805                     dumped_records += this_length;
1806                     continue;
1807                 }
1808             }
1809             else /* too big entirely */
1810             {
1811                 yaz_log(LOG_LOG, "Record > maxrcdsz this=%d max=%d", this_length, a->maximumRecordSize);
1812                 reclist->records[reclist->num_records] =
1813                     surrogatediagrec(a, freq.basename, 17, 0);
1814                 reclist->num_records++;
1815                 *next = freq.last_in_set ? 0 : recno + 1;
1816                 dumped_records += this_length;
1817                 continue;
1818             }
1819         }
1820
1821         if (!(thisrec = (Z_NamePlusRecord *)
1822               odr_malloc(a->encode, sizeof(*thisrec))))
1823             return 0;
1824         if (!(thisrec->databaseName = (char *)odr_malloc(a->encode,
1825             strlen(freq.basename) + 1)))
1826             return 0;
1827         strcpy(thisrec->databaseName, freq.basename);
1828         thisrec->which = Z_NamePlusRecord_databaseRecord;
1829
1830         if (freq.output_format_raw)
1831         {
1832             struct oident *ident = oid_getentbyoid(freq.output_format_raw);
1833             freq.output_format = ident->value;
1834         }
1835         thisrec->u.databaseRecord = z_ext_record(a->encode, freq.output_format,
1836                                                  freq.record, freq.len);
1837         if (!thisrec->u.databaseRecord)
1838             return 0;
1839         reclist->records[reclist->num_records] = thisrec;
1840         reclist->num_records++;
1841         *next = freq.last_in_set ? 0 : recno + 1;
1842     }
1843     *num = reclist->num_records;
1844     return records;
1845 }
1846
1847 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
1848     int *fd)
1849 {
1850     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
1851     bend_search_rr *bsrr = 
1852         (bend_search_rr *)nmem_malloc (reqb->request_mem, sizeof(*bsrr));
1853     
1854     yaz_log(LOG_LOG, "Got SearchRequest.");
1855     bsrr->fd = fd;
1856     bsrr->request = reqb;
1857     bsrr->association = assoc;
1858     bsrr->referenceId = req->referenceId;
1859     save_referenceId (reqb, bsrr->referenceId);
1860
1861     yaz_log (LOG_LOG, "ResultSet '%s'", req->resultSetName);
1862     if (req->databaseNames)
1863     {
1864         int i;
1865         for (i = 0; i < req->num_databaseNames; i++)
1866             yaz_log (LOG_LOG, "Database '%s'", req->databaseNames[i]);
1867     }
1868     yaz_log_zquery(req->query);
1869
1870     if (assoc->init->bend_search)
1871     {
1872         bsrr->setname = req->resultSetName;
1873         bsrr->replace_set = *req->replaceIndicator;
1874         bsrr->num_bases = req->num_databaseNames;
1875         bsrr->basenames = req->databaseNames;
1876         bsrr->query = req->query;
1877         bsrr->stream = assoc->encode;
1878         bsrr->decode = assoc->decode;
1879         bsrr->print = assoc->print;
1880         bsrr->errcode = 0;
1881         bsrr->hits = 0;
1882         bsrr->errstring = NULL;
1883         bsrr->search_info = NULL;
1884         (assoc->init->bend_search)(assoc->backend, bsrr);
1885         if (!bsrr->request)
1886             return 0;
1887     }
1888     return response_searchRequest(assoc, reqb, bsrr, fd);
1889 }
1890
1891 int bend_searchresponse(void *handle, bend_search_rr *bsrr) {return 0;}
1892
1893 /*
1894  * Prepare a searchresponse based on the backend results. We probably want
1895  * to look at making the fetching of records nonblocking as well, but
1896  * so far, we'll keep things simple.
1897  * If bsrt is null, that means we're called in response to a communications
1898  * event, and we'll have to get the response for ourselves.
1899  */
1900 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
1901     bend_search_rr *bsrt, int *fd)
1902 {
1903     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
1904     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1905     Z_SearchResponse *resp = (Z_SearchResponse *)
1906         odr_malloc (assoc->encode, sizeof(*resp));
1907     int *nulint = odr_intdup (assoc->encode, 0);
1908     bool_t *sr = odr_intdup(assoc->encode, 1);
1909     int *next = odr_intdup(assoc->encode, 0);
1910     int *none = odr_intdup(assoc->encode, Z_RES_NONE);
1911
1912     apdu->which = Z_APDU_searchResponse;
1913     apdu->u.searchResponse = resp;
1914     resp->referenceId = req->referenceId;
1915     resp->additionalSearchInfo = 0;
1916     resp->otherInfo = 0;
1917     *fd = -1;
1918     if (!bsrt && !bend_searchresponse(assoc->backend, bsrt))
1919     {
1920         yaz_log(LOG_FATAL, "Bad result from backend");
1921         return 0;
1922     }
1923     else if (bsrt->errcode)
1924     {
1925         resp->records = diagrec(assoc, bsrt->errcode, bsrt->errstring);
1926         resp->resultCount = nulint;
1927         resp->numberOfRecordsReturned = nulint;
1928         resp->nextResultSetPosition = nulint;
1929         resp->searchStatus = nulint;
1930         resp->resultSetStatus = none;
1931         resp->presentStatus = 0;
1932     }
1933     else
1934     {
1935         int *toget = odr_intdup(assoc->encode, 0);
1936         int *presst = odr_intdup(assoc->encode, 0);
1937         Z_RecordComposition comp, *compp = 0;
1938
1939         yaz_log (LOG_LOG, "resultCount: %d", bsrt->hits);
1940
1941         resp->records = 0;
1942         resp->resultCount = &bsrt->hits;
1943
1944         comp.which = Z_RecordComp_simple;
1945         /* how many records does the user agent want, then? */
1946         if (bsrt->hits <= *req->smallSetUpperBound)
1947         {
1948             *toget = bsrt->hits;
1949             if ((comp.u.simple = req->smallSetElementSetNames))
1950                 compp = &comp;
1951         }
1952         else if (bsrt->hits < *req->largeSetLowerBound)
1953         {
1954             *toget = *req->mediumSetPresentNumber;
1955             if (*toget > bsrt->hits)
1956                 *toget = bsrt->hits;
1957             if ((comp.u.simple = req->mediumSetElementSetNames))
1958                 compp = &comp;
1959         }
1960         else
1961             *toget = 0;
1962
1963         if (*toget && !resp->records)
1964         {
1965             oident *prefformat;
1966             oid_value form;
1967
1968             if (!(prefformat = oid_getentbyoid(req->preferredRecordSyntax)))
1969                 form = VAL_NONE;
1970             else
1971                 form = prefformat->value;
1972             resp->records = pack_records(assoc, req->resultSetName, 1,
1973                 toget, compp, next, presst, form, req->referenceId,
1974                                          req->preferredRecordSyntax);
1975             if (!resp->records)
1976                 return 0;
1977             resp->numberOfRecordsReturned = toget;
1978             resp->nextResultSetPosition = next;
1979             resp->searchStatus = sr;
1980             resp->resultSetStatus = 0;
1981             resp->presentStatus = presst;
1982         }
1983         else
1984         {
1985             if (*resp->resultCount)
1986                 *next = 1;
1987             resp->numberOfRecordsReturned = nulint;
1988             resp->nextResultSetPosition = next;
1989             resp->searchStatus = sr;
1990             resp->resultSetStatus = 0;
1991             resp->presentStatus = 0;
1992         }
1993     }
1994     resp->additionalSearchInfo = bsrt->search_info;
1995     return apdu;
1996 }
1997
1998 /*
1999  * Maybe we got a little over-friendly when we designed bend_fetch to
2000  * get only one record at a time. Some backends can optimise multiple-record
2001  * fetches, and at any rate, there is some overhead involved in
2002  * all that selecting and hopping around. Problem is, of course, that the
2003  * frontend can't know ahead of time how many records it'll need to
2004  * fill the negotiated PDU size. Annoying. Segmentation or not, Z/SR
2005  * is downright lousy as a bulk data transfer protocol.
2006  *
2007  * To start with, we'll do the fetching of records from the backend
2008  * in one operation: To save some trips in and out of the event-handler,
2009  * and to simplify the interface to pack_records. At any rate, asynch
2010  * operation is more fun in operations that have an unpredictable execution
2011  * speed - which is normally more true for search than for present.
2012  */
2013 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
2014                                       int *fd)
2015 {
2016     Z_PresentRequest *req = reqb->apdu_request->u.presentRequest;
2017     oident *prefformat;
2018     oid_value form;
2019     Z_APDU *apdu;
2020     Z_PresentResponse *resp;
2021     int *next;
2022     int *num;
2023
2024     yaz_log(LOG_LOG, "Got PresentRequest.");
2025
2026     if (!(prefformat = oid_getentbyoid(req->preferredRecordSyntax)))
2027         form = VAL_NONE;
2028     else
2029         form = prefformat->value;
2030     resp = (Z_PresentResponse *)odr_malloc (assoc->encode, sizeof(*resp));
2031     resp->records = 0;
2032     resp->presentStatus = odr_intdup(assoc->encode, 0);
2033     if (assoc->init->bend_present)
2034     {
2035         bend_present_rr *bprr = (bend_present_rr *)
2036             nmem_malloc (reqb->request_mem, sizeof(*bprr));
2037         bprr->setname = req->resultSetId;
2038         bprr->start = *req->resultSetStartPoint;
2039         bprr->number = *req->numberOfRecordsRequested;
2040         bprr->format = form;
2041         bprr->comp = req->recordComposition;
2042         bprr->referenceId = req->referenceId;
2043         bprr->stream = assoc->encode;
2044         bprr->print = assoc->print;
2045         bprr->request = reqb;
2046         bprr->association = assoc;
2047         bprr->errcode = 0;
2048         bprr->errstring = NULL;
2049         (*assoc->init->bend_present)(assoc->backend, bprr);
2050         
2051         if (!bprr->request)
2052             return 0;
2053         if (bprr->errcode)
2054         {
2055             resp->records = diagrec(assoc, bprr->errcode, bprr->errstring);
2056             *resp->presentStatus = Z_PRES_FAILURE;
2057         }
2058     }
2059     apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
2060     next = odr_intdup(assoc->encode, 0);
2061     num = odr_intdup(assoc->encode, 0);
2062     
2063     apdu->which = Z_APDU_presentResponse;
2064     apdu->u.presentResponse = resp;
2065     resp->referenceId = req->referenceId;
2066     resp->otherInfo = 0;
2067     
2068     if (!resp->records)
2069     {
2070         *num = *req->numberOfRecordsRequested;
2071         resp->records =
2072             pack_records(assoc, req->resultSetId, *req->resultSetStartPoint,
2073                      num, req->recordComposition, next, resp->presentStatus,
2074                          form, req->referenceId, req->preferredRecordSyntax);
2075     }
2076     if (!resp->records)
2077         return 0;
2078     resp->numberOfRecordsReturned = num;
2079     resp->nextResultSetPosition = next;
2080     
2081     return apdu;
2082 }
2083
2084 /*
2085  * Scan was implemented rather in a hurry, and with support for only the basic
2086  * elements of the service in the backend API. Suggestions are welcome.
2087  */
2088 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd)
2089 {
2090     Z_ScanRequest *req = reqb->apdu_request->u.scanRequest;
2091     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
2092     Z_ScanResponse *res = (Z_ScanResponse *)
2093         odr_malloc (assoc->encode, sizeof(*res));
2094     int *scanStatus = odr_intdup(assoc->encode, Z_Scan_failure);
2095     int *numberOfEntriesReturned = odr_intdup(assoc->encode, 0);
2096     Z_ListEntries *ents = (Z_ListEntries *)
2097         odr_malloc (assoc->encode, sizeof(*ents));
2098     Z_DiagRecs *diagrecs_p = NULL;
2099     oident *attset;
2100     bend_scan_rr *bsrr = (bend_scan_rr *)
2101         odr_malloc (assoc->encode, sizeof(*bsrr));
2102
2103     yaz_log(LOG_LOG, "Got ScanRequest");
2104
2105     apdu->which = Z_APDU_scanResponse;
2106     apdu->u.scanResponse = res;
2107     res->referenceId = req->referenceId;
2108
2109     /* if step is absent, set it to 0 */
2110     res->stepSize = odr_intdup(assoc->encode, 0);
2111     if (req->stepSize)
2112         *res->stepSize = *req->stepSize;
2113
2114     res->scanStatus = scanStatus;
2115     res->numberOfEntriesReturned = numberOfEntriesReturned;
2116     res->positionOfTerm = 0;
2117     res->entries = ents;
2118     ents->num_entries = 0;
2119     ents->entries = NULL;
2120     ents->num_nonsurrogateDiagnostics = 0;
2121     ents->nonsurrogateDiagnostics = NULL;
2122     res->attributeSet = 0;
2123     res->otherInfo = 0;
2124
2125     if (req->databaseNames)
2126     {
2127         int i;
2128         for (i = 0; i < req->num_databaseNames; i++)
2129             yaz_log (LOG_LOG, "Database '%s'", req->databaseNames[i]);
2130     }
2131     bsrr->num_bases = req->num_databaseNames;
2132     bsrr->basenames = req->databaseNames;
2133     bsrr->num_entries = *req->numberOfTermsRequested;
2134     bsrr->term = req->termListAndStartPoint;
2135     bsrr->referenceId = req->referenceId;
2136     bsrr->stream = assoc->encode;
2137     bsrr->print = assoc->print;
2138     bsrr->step_size = res->stepSize;
2139     if (req->attributeSet &&
2140         (attset = oid_getentbyoid(req->attributeSet)) &&
2141         (attset->oclass == CLASS_ATTSET || attset->oclass == CLASS_GENERAL))
2142         bsrr->attributeset = attset->value;
2143     else
2144         bsrr->attributeset = VAL_NONE;
2145     log_scan_term (req->termListAndStartPoint, bsrr->attributeset);
2146     bsrr->term_position = req->preferredPositionInResponse ?
2147         *req->preferredPositionInResponse : 1;
2148     ((int (*)(void *, bend_scan_rr *))
2149      (*assoc->init->bend_scan))(assoc->backend, bsrr);
2150     if (bsrr->errcode)
2151         diagrecs_p = diagrecs(assoc, bsrr->errcode, bsrr->errstring);
2152     else
2153     {
2154         int i;
2155         Z_Entry **tab = (Z_Entry **)
2156             odr_malloc (assoc->encode, sizeof(*tab) * bsrr->num_entries);
2157         
2158         if (bsrr->status == BEND_SCAN_PARTIAL)
2159             *scanStatus = Z_Scan_partial_5;
2160         else
2161             *scanStatus = Z_Scan_success;
2162         ents->entries = tab;
2163         ents->num_entries = bsrr->num_entries;
2164         res->numberOfEntriesReturned = &ents->num_entries;          
2165         res->positionOfTerm = &bsrr->term_position;
2166         for (i = 0; i < bsrr->num_entries; i++)
2167         {
2168             Z_Entry *e;
2169             Z_TermInfo *t;
2170             Odr_oct *o;
2171             
2172             tab[i] = e = (Z_Entry *)odr_malloc(assoc->encode, sizeof(*e));
2173             if (bsrr->entries[i].occurrences >= 0)
2174             {
2175                 e->which = Z_Entry_termInfo;
2176                 e->u.termInfo = t = (Z_TermInfo *)
2177                     odr_malloc(assoc->encode, sizeof(*t));
2178                 t->suggestedAttributes = 0;
2179                 t->displayTerm = 0;
2180                 t->alternativeTerm = 0;
2181                 t->byAttributes = 0;
2182                 t->otherTermInfo = 0;
2183                 t->globalOccurrences = &bsrr->entries[i].occurrences;
2184                 t->term = (Z_Term *)
2185                     odr_malloc(assoc->encode, sizeof(*t->term));
2186                 t->term->which = Z_Term_general;
2187                 t->term->u.general = o =
2188                     (Odr_oct *)odr_malloc(assoc->encode, sizeof(Odr_oct));
2189                 o->buf = (unsigned char *)
2190                     odr_malloc(assoc->encode, o->len = o->size =
2191                                strlen(bsrr->entries[i].term));
2192                 memcpy(o->buf, bsrr->entries[i].term, o->len);
2193                 yaz_log(LOG_DEBUG, "  term #%d: '%s' (%d)", i,
2194                          bsrr->entries[i].term, bsrr->entries[i].occurrences);
2195             }
2196             else
2197             {
2198                 Z_DiagRecs *drecs = diagrecs (assoc,
2199                                               bsrr->entries[i].errcode,
2200                                               bsrr->entries[i].errstring);
2201                 assert (drecs->num_diagRecs == 1);
2202                 e->which = Z_Entry_surrogateDiagnostic;
2203                 assert (drecs->diagRecs[0]);
2204                 e->u.surrogateDiagnostic = drecs->diagRecs[0];
2205             }
2206         }
2207     }
2208     if (diagrecs_p)
2209     {
2210         ents->num_nonsurrogateDiagnostics = diagrecs_p->num_diagRecs;
2211         ents->nonsurrogateDiagnostics = diagrecs_p->diagRecs;
2212     }
2213     return apdu;
2214 }
2215
2216 static Z_APDU *process_sortRequest(association *assoc, request *reqb,
2217     int *fd)
2218 {
2219     Z_SortRequest *req = reqb->apdu_request->u.sortRequest;
2220     Z_SortResponse *res = (Z_SortResponse *)
2221         odr_malloc (assoc->encode, sizeof(*res));
2222     bend_sort_rr *bsrr = (bend_sort_rr *)
2223         odr_malloc (assoc->encode, sizeof(*bsrr));
2224
2225     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
2226
2227     yaz_log(LOG_LOG, "Got SortRequest.");
2228
2229     bsrr->num_input_setnames = req->num_inputResultSetNames;
2230     bsrr->input_setnames = req->inputResultSetNames;
2231     bsrr->referenceId = req->referenceId;
2232     bsrr->output_setname = req->sortedResultSetName;
2233     bsrr->sort_sequence = req->sortSequence;
2234     bsrr->stream = assoc->encode;
2235     bsrr->print = assoc->print;
2236
2237     bsrr->sort_status = Z_SortStatus_failure;
2238     bsrr->errcode = 0;
2239     bsrr->errstring = 0;
2240     
2241     (*assoc->init->bend_sort)(assoc->backend, bsrr);
2242     
2243     res->referenceId = bsrr->referenceId;
2244     res->sortStatus = odr_intdup(assoc->encode, bsrr->sort_status);
2245     res->resultSetStatus = 0;
2246     if (bsrr->errcode)
2247     {
2248         Z_DiagRecs *dr = diagrecs (assoc, bsrr->errcode, bsrr->errstring);
2249         res->diagnostics = dr->diagRecs;
2250         res->num_diagnostics = dr->num_diagRecs;
2251     }
2252     else
2253     {
2254         res->num_diagnostics = 0;
2255         res->diagnostics = 0;
2256     }
2257     res->otherInfo = 0;
2258
2259     apdu->which = Z_APDU_sortResponse;
2260     apdu->u.sortResponse = res;
2261     return apdu;
2262 }
2263
2264 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
2265     int *fd)
2266 {
2267     Z_DeleteResultSetRequest *req =
2268         reqb->apdu_request->u.deleteResultSetRequest;
2269     Z_DeleteResultSetResponse *res = (Z_DeleteResultSetResponse *)
2270         odr_malloc (assoc->encode, sizeof(*res));
2271     bend_delete_rr *bdrr = (bend_delete_rr *)
2272         odr_malloc (assoc->encode, sizeof(*bdrr));
2273     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
2274
2275     yaz_log(LOG_LOG, "Got DeleteRequest.");
2276
2277     bdrr->num_setnames = req->num_resultSetList;
2278     bdrr->setnames = req->resultSetList;
2279     bdrr->stream = assoc->encode;
2280     bdrr->print = assoc->print;
2281     bdrr->function = *req->deleteFunction;
2282     bdrr->referenceId = req->referenceId;
2283     bdrr->statuses = 0;
2284     if (bdrr->num_setnames > 0)
2285     {
2286         int i;
2287         bdrr->statuses = (int*) 
2288             odr_malloc(assoc->encode, sizeof(*bdrr->statuses) *
2289                        bdrr->num_setnames);
2290         for (i = 0; i < bdrr->num_setnames; i++)
2291             bdrr->statuses[i] = 0;
2292     }
2293     (*assoc->init->bend_delete)(assoc->backend, bdrr);
2294     
2295     res->referenceId = req->referenceId;
2296
2297     res->deleteOperationStatus = odr_intdup(assoc->encode,bdrr->delete_status);
2298
2299     res->deleteListStatuses = 0;
2300     if (bdrr->num_setnames > 0)
2301     {
2302         int i;
2303         res->deleteListStatuses = (Z_ListStatuses *)
2304             odr_malloc(assoc->encode, sizeof(*res->deleteListStatuses));
2305         res->deleteListStatuses->num = bdrr->num_setnames;
2306         res->deleteListStatuses->elements =
2307             (Z_ListStatus **)
2308             odr_malloc (assoc->encode, 
2309                         sizeof(*res->deleteListStatuses->elements) *
2310                         bdrr->num_setnames);
2311         for (i = 0; i<bdrr->num_setnames; i++)
2312         {
2313             res->deleteListStatuses->elements[i] =
2314                 (Z_ListStatus *)
2315                 odr_malloc (assoc->encode,
2316                             sizeof(**res->deleteListStatuses->elements));
2317             res->deleteListStatuses->elements[i]->status = bdrr->statuses+i;
2318             res->deleteListStatuses->elements[i]->id =
2319                 odr_strdup (assoc->encode, bdrr->setnames[i]);
2320             
2321         }
2322     }
2323     res->numberNotDeleted = 0;
2324     res->bulkStatuses = 0;
2325     res->deleteMessage = 0;
2326     res->otherInfo = 0;
2327
2328     apdu->which = Z_APDU_deleteResultSetResponse;
2329     apdu->u.deleteResultSetResponse = res;
2330     return apdu;
2331 }
2332
2333 static void process_close(association *assoc, request *reqb)
2334 {
2335     Z_Close *req = reqb->apdu_request->u.close;
2336     static char *reasons[] =
2337     {
2338         "finished",
2339         "shutdown",
2340         "systemProblem",
2341         "costLimit",
2342         "resources",
2343         "securityViolation",
2344         "protocolError",
2345         "lackOfActivity",
2346         "peerAbort",
2347         "unspecified"
2348     };
2349
2350     yaz_log(LOG_LOG, "Got Close, reason %s, message %s",
2351         reasons[*req->closeReason], req->diagnosticInformation ?
2352         req->diagnosticInformation : "NULL");
2353     if (assoc->version < 3) /* to make do_force respond with close */
2354         assoc->version = 3;
2355     do_close_req(assoc, Z_Close_finished,
2356                  "Association terminated by client", reqb);
2357 }
2358
2359 void save_referenceId (request *reqb, Z_ReferenceId *refid)
2360 {
2361     if (refid)
2362     {
2363         reqb->len_refid = refid->len;
2364         reqb->refid = (char *)nmem_malloc (reqb->request_mem, refid->len);
2365         memcpy (reqb->refid, refid->buf, refid->len);
2366     }
2367     else
2368     {
2369         reqb->len_refid = 0;
2370         reqb->refid = NULL;
2371     }
2372 }
2373
2374 void bend_request_send (bend_association a, bend_request req, Z_APDU *res)
2375 {
2376     process_z_response (a, req, res);
2377 }
2378
2379 bend_request bend_request_mk (bend_association a)
2380 {
2381     request *nreq = request_get (&a->outgoing);
2382     nreq->request_mem = nmem_create ();
2383     return nreq;
2384 }
2385
2386 Z_ReferenceId *bend_request_getid (ODR odr, bend_request req)
2387 {
2388     Z_ReferenceId *id;
2389     if (!req->refid)
2390         return 0;
2391     id = (Odr_oct *)odr_malloc (odr, sizeof(*odr));
2392     id->buf = (unsigned char *)odr_malloc (odr, req->len_refid);
2393     id->len = id->size = req->len_refid;
2394     memcpy (id->buf, req->refid, req->len_refid);
2395     return id;
2396 }
2397
2398 void bend_request_destroy (bend_request *req)
2399 {
2400     nmem_destroy((*req)->request_mem);
2401     request_release(*req);
2402     *req = NULL;
2403 }
2404
2405 int bend_backend_respond (bend_association a, bend_request req)
2406 {
2407     char *msg;
2408     int r;
2409     r = process_z_request (a, req, &msg);
2410     if (r < 0)
2411         yaz_log (LOG_WARN, "%s", msg);
2412     return r;
2413 }
2414
2415 void bend_request_setdata(bend_request r, void *p)
2416 {
2417     r->clientData = p;
2418 }
2419
2420 void *bend_request_getdata(bend_request r)
2421 {
2422     return r->clientData;
2423 }
2424
2425 static Z_APDU *process_segmentRequest (association *assoc, request *reqb)
2426 {
2427     bend_segment_rr req;
2428
2429     req.segment = reqb->apdu_request->u.segmentRequest;
2430     req.stream = assoc->encode;
2431     req.decode = assoc->decode;
2432     req.print = assoc->print;
2433     req.association = assoc;
2434     
2435     (*assoc->init->bend_segment)(assoc->backend, &req);
2436
2437     return 0;
2438 }
2439
2440 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd)
2441 {
2442     bend_esrequest_rr esrequest;
2443
2444     Z_ExtendedServicesRequest *req =
2445         reqb->apdu_request->u.extendedServicesRequest;
2446     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_extendedServicesResponse);
2447
2448     Z_ExtendedServicesResponse *resp = apdu->u.extendedServicesResponse;
2449
2450     yaz_log(LOG_DEBUG,"inside Process esRequest");
2451
2452     esrequest.esr = reqb->apdu_request->u.extendedServicesRequest;
2453     esrequest.stream = assoc->encode;
2454     esrequest.decode = assoc->decode;
2455     esrequest.print = assoc->print;
2456     esrequest.errcode = 0;
2457     esrequest.errstring = NULL;
2458     esrequest.request = reqb;
2459     esrequest.association = assoc;
2460     esrequest.taskPackage = 0;
2461     esrequest.referenceId = req->referenceId;
2462     
2463     (*assoc->init->bend_esrequest)(assoc->backend, &esrequest);
2464     
2465     /* If the response is being delayed, return NULL */
2466     if (esrequest.request == NULL)
2467         return(NULL);
2468
2469     resp->referenceId = req->referenceId;
2470
2471     if (esrequest.errcode == -1)
2472     {
2473         /* Backend service indicates request will be processed */
2474         yaz_log(LOG_DEBUG,"Request could be processed...Accepted !");
2475         *resp->operationStatus = Z_ExtendedServicesResponse_accepted;
2476     }
2477     else if (esrequest.errcode == 0)
2478     {
2479         /* Backend service indicates request will be processed */
2480         yaz_log(LOG_DEBUG,"Request could be processed...Done !");
2481         *resp->operationStatus = Z_ExtendedServicesResponse_done;
2482     }
2483     else
2484     {
2485         Z_DiagRecs *diagRecs = diagrecs (assoc, esrequest.errcode,
2486                                          esrequest.errstring);
2487
2488         /* Backend indicates error, request will not be processed */
2489         yaz_log(LOG_DEBUG,"Request could not be processed...failure !");
2490         *resp->operationStatus = Z_ExtendedServicesResponse_failure;
2491         resp->num_diagnostics = diagRecs->num_diagRecs;
2492         resp->diagnostics = diagRecs->diagRecs;
2493     }
2494     /* Do something with the members of bend_extendedservice */
2495     if (esrequest.taskPackage)
2496         resp->taskPackage = z_ext_record (assoc->encode, VAL_EXTENDED,
2497                                          (const char *)  esrequest.taskPackage,
2498                                           -1);
2499     yaz_log(LOG_DEBUG,"Send the result apdu");
2500     return apdu;
2501 }
2502