Log session timeout. Reenable ASSOC_UP.
[yaz-moved-to-github.git] / src / seshigh.c
1 /* This file is part of the YAZ toolkit.
2  * Copyright (C) 1995-2009 Index Data
3  * See the file LICENSE for details.
4  */
5 /**
6  * \file seshigh.c
7  * \brief Implements GFS session logic.
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 <assert.h>
32 #include <ctype.h>
33
34 #if HAVE_SYS_TYPES_H
35 #include <sys/types.h>
36 #endif
37 #if HAVE_SYS_STAT_H
38 #include <sys/stat.h>
39 #endif
40
41 #ifdef WIN32
42 #include <io.h>
43 #define S_ISREG(x) (x & _S_IFREG)
44 #include <process.h>
45 #endif
46
47 #if HAVE_UNISTD_H
48 #include <unistd.h>
49 #endif
50
51 #if YAZ_HAVE_XML2
52 #include <libxml/parser.h>
53 #include <libxml/tree.h>
54 #endif
55
56 #include <yaz/yconfig.h>
57 #include <yaz/xmalloc.h>
58 #include <yaz/comstack.h>
59 #include "eventl.h"
60 #include "session.h"
61 #include "mime.h"
62 #include <yaz/proto.h>
63 #include <yaz/oid_db.h>
64 #include <yaz/log.h>
65 #include <yaz/logrpn.h>
66 #include <yaz/querytowrbuf.h>
67 #include <yaz/statserv.h>
68 #include <yaz/diagbib1.h>
69 #include <yaz/charneg.h>
70 #include <yaz/otherinfo.h>
71 #include <yaz/yaz-util.h>
72 #include <yaz/pquery.h>
73 #include <yaz/oid_db.h>
74
75 #include <yaz/srw.h>
76 #include <yaz/backend.h>
77 #include <yaz/yaz-ccl.h>
78
79 static void process_gdu_request(association *assoc, request *req);
80 static int process_z_request(association *assoc, request *req, char **msg);
81 void backend_response(IOCHAN i, int event);
82 static int process_gdu_response(association *assoc, request *req, Z_GDU *res);
83 static int process_z_response(association *assoc, request *req, Z_APDU *res);
84 static Z_APDU *process_initRequest(association *assoc, request *reqb);
85 static Z_External *init_diagnostics(ODR odr, int errcode,
86                                     const char *errstring);
87 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
88     int *fd);
89 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
90     bend_search_rr *bsrr, int *fd);
91 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
92     int *fd);
93 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd);
94 static Z_APDU *process_sortRequest(association *assoc, request *reqb, int *fd);
95 static void process_close(association *assoc, request *reqb);
96 void save_referenceId(request *reqb, Z_ReferenceId *refid);
97 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
98     int *fd);
99 static Z_APDU *process_segmentRequest(association *assoc, request *reqb);
100
101 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd);
102
103 /* dynamic logging levels */
104 static int logbits_set = 0;
105 static int log_session = 0; /* one-line logs for session */
106 static int log_sessiondetail = 0; /* more detailed stuff */
107 static int log_request = 0; /* one-line logs for requests */
108 static int log_requestdetail = 0;  /* more detailed stuff */
109
110 /** get_logbits sets global loglevel bits */
111 static void get_logbits(void)
112 { /* needs to be called after parsing cmd-line args that can set loglevels!*/
113     if (!logbits_set)
114     {
115         logbits_set = 1;
116         log_session = yaz_log_module_level("session"); 
117         log_sessiondetail = yaz_log_module_level("sessiondetail");
118         log_request = yaz_log_module_level("request");
119         log_requestdetail = yaz_log_module_level("requestdetail"); 
120     }
121 }
122
123
124
125 static void wr_diag(WRBUF w, int error, const char *addinfo)
126 {
127     wrbuf_printf(w, "ERROR %d+", error);
128     wrbuf_puts_replace_char(w, diagbib1_str(error), ' ', '_');
129     if (addinfo){
130         wrbuf_puts(w, "+");
131         wrbuf_puts_replace_char(w, addinfo, ' ', '_');
132     }
133     
134     wrbuf_puts(w, " ");    
135 }
136
137
138 /*
139  * Create and initialize a new association-handle.
140  *  channel  : iochannel for the current line.
141  *  link     : communications channel.
142  * Returns: 0 or a new association handle.
143  */
144 association *create_association(IOCHAN channel, COMSTACK link,
145                                 const char *apdufile)
146 {
147     association *anew;
148
149     if (!logbits_set)
150         get_logbits();
151     if (!(anew = (association *)xmalloc(sizeof(*anew))))
152         return 0;
153     anew->init = 0;
154     anew->version = 0;
155     anew->last_control = 0;
156     anew->client_chan = channel;
157     anew->client_link = link;
158     anew->cs_get_mask = 0;
159     anew->cs_put_mask = 0;
160     anew->cs_accept_mask = 0;
161     if (!(anew->decode = odr_createmem(ODR_DECODE)) ||
162         !(anew->encode = odr_createmem(ODR_ENCODE)))
163         return 0;
164     if (apdufile && *apdufile)
165     {
166         FILE *f;
167
168         if (!(anew->print = odr_createmem(ODR_PRINT)))
169             return 0;
170         if (*apdufile == '@')
171         {
172             odr_setprint(anew->print, yaz_log_file());
173         }       
174         else if (*apdufile != '-')
175         {
176             char filename[256];
177             sprintf(filename, "%.200s.%ld", apdufile, (long)getpid());
178             if (!(f = fopen(filename, "w")))
179             {
180                 yaz_log(YLOG_WARN|YLOG_ERRNO, "%s", filename);
181                 return 0;
182             }
183             setvbuf(f, 0, _IONBF, 0);
184             odr_setprint(anew->print, f);
185         }
186     }
187     else
188         anew->print = 0;
189     anew->input_buffer = 0;
190     anew->input_buffer_len = 0;
191     anew->backend = 0;
192     anew->state = ASSOC_NEW;
193     request_initq(&anew->incoming);
194     request_initq(&anew->outgoing);
195     anew->proto = cs_getproto(link);
196     anew->server = 0;
197     return anew;
198 }
199
200 /*
201  * Free association and release resources.
202  */
203 void destroy_association(association *h)
204 {
205     statserv_options_block *cb = statserv_getcontrol();
206     request *req;
207
208     xfree(h->init);
209     odr_destroy(h->decode);
210     odr_destroy(h->encode);
211     if (h->print)
212         odr_destroy(h->print);
213     if (h->input_buffer)
214     xfree(h->input_buffer);
215     if (h->backend)
216         (*cb->bend_close)(h->backend);
217     while ((req = request_deq(&h->incoming)))
218         request_release(req);
219     while ((req = request_deq(&h->outgoing)))
220         request_release(req);
221     request_delq(&h->incoming);
222     request_delq(&h->outgoing);
223     xfree(h);
224     xmalloc_trav("session closed");
225     if (cb && cb->one_shot)
226     {
227         exit(0);
228     }
229 }
230
231 static void do_close_req(association *a, int reason, char *message,
232                          request *req)
233 {
234     Z_APDU apdu;
235     Z_Close *cls = zget_Close(a->encode);
236     
237     /* Purge request queue */
238     while (request_deq(&a->incoming));
239     while (request_deq(&a->outgoing));
240     if (a->version >= 3)
241     {
242         yaz_log(log_requestdetail, "Sending Close PDU, reason=%d, message=%s",
243             reason, message ? message : "none");
244         apdu.which = Z_APDU_close;
245         apdu.u.close = cls;
246         *cls->closeReason = reason;
247         cls->diagnosticInformation = message;
248         process_z_response(a, req, &apdu);
249         iochan_settimeout(a->client_chan, 20);
250     }
251     else
252     {
253         request_release(req);
254         yaz_log(log_requestdetail, "v2 client. No Close PDU");
255         iochan_setevent(a->client_chan, EVENT_TIMEOUT); /* force imm close */
256         a->cs_put_mask = 0;
257     }
258     a->state = ASSOC_DEAD;
259 }
260
261 static void do_close(association *a, int reason, char *message)
262 {
263     request *req = request_get(&a->outgoing);
264     do_close_req (a, reason, message, req);
265 }
266
267
268 int ir_read(IOCHAN h, int event)
269 {
270     association *assoc = (association *)iochan_getdata(h);
271     COMSTACK conn = assoc->client_link;
272     request *req;
273     
274     if ((assoc->cs_put_mask & EVENT_INPUT) == 0 && (event & assoc->cs_get_mask))
275     {
276         yaz_log(YLOG_DEBUG, "ir_session (input)");
277         /* We aren't speaking to this fellow */
278         if (assoc->state == ASSOC_DEAD)
279         {
280             yaz_log(log_sessiondetail, "Connection closed - end of session");
281             cs_close(conn);
282             destroy_association(assoc);
283             iochan_destroy(h);
284             return 0;
285         }
286         assoc->cs_get_mask = EVENT_INPUT;
287
288         do
289         {
290             int res = cs_get(conn, &assoc->input_buffer,
291                              &assoc->input_buffer_len);
292             if (res < 0 && cs_errno(conn) == CSBUFSIZE)
293             {
294                 yaz_log(log_session, "Connection error: %s res=%d",
295                         cs_errmsg(cs_errno(conn)), res);
296                 req = request_get(&assoc->incoming); /* get a new request */
297                 do_close_req(assoc, Z_Close_protocolError, 
298                              "Incoming package too large", req);
299                 return 0;
300             }
301             else if (res <= 0)
302             {
303                 yaz_log(log_session, "Connection closed by client");
304                 assoc->state = ASSOC_DEAD;
305                 return 0;
306             }
307             else if (res == 1) /* incomplete read - wait for more  */
308             {
309                 if (conn->io_pending & CS_WANT_WRITE)
310                     assoc->cs_get_mask |= EVENT_OUTPUT;
311                 iochan_setflag(h, assoc->cs_get_mask);
312                 return 0;
313             }
314             /* we got a complete PDU. Let's decode it */
315             yaz_log(YLOG_DEBUG, "Got PDU, %d bytes: lead=%02X %02X %02X", res,
316                     assoc->input_buffer[0] & 0xff,
317                     assoc->input_buffer[1] & 0xff,
318                     assoc->input_buffer[2] & 0xff);
319             req = request_get(&assoc->incoming); /* get a new request */
320             odr_reset(assoc->decode);
321             odr_setbuf(assoc->decode, assoc->input_buffer, res, 0);
322             if (!z_GDU(assoc->decode, &req->gdu_request, 0, 0))
323             {
324                 yaz_log(YLOG_WARN, "ODR error on incoming PDU: %s [element %s] "
325                         "[near byte %ld] ",
326                         odr_errmsg(odr_geterror(assoc->decode)),
327                         odr_getelement(assoc->decode),
328                         (long) odr_offset(assoc->decode));
329                 if (assoc->decode->error != OHTTP)
330                 {
331                     yaz_log(YLOG_WARN, "PDU dump:");
332                     odr_dumpBER(yaz_log_file(), assoc->input_buffer, res);
333                     request_release(req);
334                     do_close(assoc, Z_Close_protocolError, "Malformed package");
335                 }
336                 else
337                 {
338                     Z_GDU *p = z_get_HTTP_Response(assoc->encode, 400);
339                     assoc->state = ASSOC_DEAD;
340                     process_gdu_response(assoc, req, p);
341                 }
342                 return 0;
343             }
344             req->request_mem = odr_extract_mem(assoc->decode);
345             if (assoc->print) 
346             {
347                 if (!z_GDU(assoc->print, &req->gdu_request, 0, 0))
348                     yaz_log(YLOG_WARN, "ODR print error: %s", 
349                             odr_errmsg(odr_geterror(assoc->print)));
350                 odr_reset(assoc->print);
351             }
352             request_enq(&assoc->incoming, req);
353         }
354         while (cs_more(conn));
355     }
356     return 1;
357 }
358
359 /*
360  * This is where PDUs from the client are read and the further
361  * processing is initiated. Flow of control moves down through the
362  * various process_* functions below, until the encoded result comes back up
363  * to the output handler in here.
364  * 
365  *  h     : the I/O channel that has an outstanding event.
366  *  event : the current outstanding event.
367  */
368 void ir_session(IOCHAN h, int event)
369 {
370     int res;
371     association *assoc = (association *)iochan_getdata(h);
372     COMSTACK conn = assoc->client_link;
373     request *req;
374
375     assert(h && conn && assoc);
376     if (event == EVENT_TIMEOUT)
377     {
378         if (assoc->state != ASSOC_UP)
379         {
380             yaz_log(log_session, "Timeout. Closing connection");
381             /* do we need to lod this at all */
382             cs_close(conn);
383             destroy_association(assoc);
384             iochan_destroy(h);
385         }
386         else
387         {
388             yaz_log(log_sessiondetail, "Timeout. Sending Z39.50 Close");
389             do_close(assoc, Z_Close_lackOfActivity, 0);
390         }
391         return;
392     }
393     if (event & assoc->cs_accept_mask)
394     {
395         if (!cs_accept(conn))
396         {
397             yaz_log(YLOG_WARN, "accept failed");
398             destroy_association(assoc);
399             iochan_destroy(h);
400             return;
401         }
402         iochan_clearflag(h, EVENT_OUTPUT);
403         if (conn->io_pending) 
404         {   /* cs_accept didn't complete */
405             assoc->cs_accept_mask = 
406                 ((conn->io_pending & CS_WANT_WRITE) ? EVENT_OUTPUT : 0) |
407                 ((conn->io_pending & CS_WANT_READ) ? EVENT_INPUT : 0);
408
409             iochan_setflag(h, assoc->cs_accept_mask);
410         }
411         else
412         {   /* cs_accept completed. Prepare for reading (cs_get) */
413             assoc->cs_accept_mask = 0;
414             assoc->cs_get_mask = EVENT_INPUT;
415             iochan_setflag(h, assoc->cs_get_mask);
416         }
417         return;
418     }
419     if (event & assoc->cs_get_mask) /* input */
420     {
421         if (!ir_read(h, event))
422             return;
423         req = request_head(&assoc->incoming);
424         if (req->state == REQUEST_IDLE)
425         {
426             request_deq(&assoc->incoming);
427             process_gdu_request(assoc, req);
428         }
429     }
430     if (event & assoc->cs_put_mask)
431     {
432         request *req = request_head(&assoc->outgoing);
433
434         assoc->cs_put_mask = 0;
435         yaz_log(YLOG_DEBUG, "ir_session (output)");
436         req->state = REQUEST_PENDING;
437         switch (res = cs_put(conn, req->response, req->len_response))
438         {
439         case -1:
440             yaz_log(log_sessiondetail, "Connection closed by client");
441             cs_close(conn);
442             destroy_association(assoc);
443             iochan_destroy(h);
444             break;
445         case 0: /* all sent - release the request structure */
446             yaz_log(YLOG_DEBUG, "Wrote PDU, %d bytes", req->len_response);
447 #if 0
448             yaz_log(YLOG_DEBUG, "HTTP out:\n%.*s", req->len_response,
449                     req->response);
450 #endif
451             nmem_destroy(req->request_mem);
452             request_deq(&assoc->outgoing);
453             request_release(req);
454             if (!request_head(&assoc->outgoing))
455             {   /* restore mask for cs_get operation ... */
456                 iochan_clearflag(h, EVENT_OUTPUT|EVENT_INPUT);
457                 iochan_setflag(h, assoc->cs_get_mask);
458                 if (assoc->state == ASSOC_DEAD)
459                     iochan_setevent(assoc->client_chan, EVENT_TIMEOUT);
460             }
461             else
462             {
463                 assoc->cs_put_mask = EVENT_OUTPUT;
464             }
465             break;
466         default:
467             if (conn->io_pending & CS_WANT_WRITE)
468                 assoc->cs_put_mask |= EVENT_OUTPUT;
469             if (conn->io_pending & CS_WANT_READ)
470                 assoc->cs_put_mask |= EVENT_INPUT;
471             iochan_setflag(h, assoc->cs_put_mask);
472         }
473     }
474     if (event & EVENT_EXCEPT)
475     {
476         yaz_log(YLOG_WARN, "ir_session (exception)");
477         cs_close(conn);
478         destroy_association(assoc);
479         iochan_destroy(h);
480     }
481 }
482
483 static int process_z_request(association *assoc, request *req, char **msg);
484
485
486 static void assoc_init_reset(association *assoc)
487 {
488     xfree (assoc->init);
489     assoc->init = (bend_initrequest *) xmalloc(sizeof(*assoc->init));
490
491     assoc->init->stream = assoc->encode;
492     assoc->init->print = assoc->print;
493     assoc->init->auth = 0;
494     assoc->init->referenceId = 0;
495     assoc->init->implementation_version = 0;
496     assoc->init->implementation_id = 0;
497     assoc->init->implementation_name = 0;
498     assoc->init->query_charset = 0;
499     assoc->init->records_in_same_charset = 0;
500     assoc->init->bend_sort = NULL;
501     assoc->init->bend_search = NULL;
502     assoc->init->bend_present = NULL;
503     assoc->init->bend_esrequest = NULL;
504     assoc->init->bend_delete = NULL;
505     assoc->init->bend_scan = NULL;
506     assoc->init->bend_segment = NULL;
507     assoc->init->bend_fetch = NULL;
508     assoc->init->bend_explain = NULL;
509     assoc->init->bend_srw_scan = NULL;
510     assoc->init->bend_srw_update = NULL;
511     assoc->init->named_result_sets = 0;
512
513     assoc->init->charneg_request = NULL;
514     assoc->init->charneg_response = NULL;
515
516     assoc->init->decode = assoc->decode;
517     assoc->init->peer_name = 
518         odr_strdup(assoc->encode, cs_addrstr(assoc->client_link));
519
520     yaz_log(log_requestdetail, "peer %s", assoc->init->peer_name);
521 }
522
523 static int srw_bend_init(association *assoc, Z_SRW_diagnostic **d, int *num, Z_SRW_PDU *sr)
524 {
525     statserv_options_block *cb = statserv_getcontrol();
526     if (!assoc->init)
527     {
528         const char *encoding = "UTF-8";
529         Z_External *ce;
530         bend_initresult *binitres;
531
532         yaz_log(log_requestdetail, "srw_bend_init config=%s", cb->configname);
533         assoc_init_reset(assoc);
534         
535         if (sr->username)
536         {
537             Z_IdAuthentication *auth = (Z_IdAuthentication *)
538                 odr_malloc(assoc->decode, sizeof(*auth));
539             int len;
540
541             len = strlen(sr->username) + 1;
542             if (sr->password) 
543                 len += strlen(sr->password) + 2;
544             auth->which = Z_IdAuthentication_open;
545             auth->u.open = (char *) odr_malloc(assoc->decode, len);
546             strcpy(auth->u.open, sr->username);
547             if (sr->password && *sr->password)
548             {
549                 strcat(auth->u.open, "/");
550                 strcat(auth->u.open, sr->password);
551             }
552             assoc->init->auth = auth;
553         }
554
555 #if 1
556         ce = yaz_set_proposal_charneg(assoc->decode, &encoding, 1, 0, 0, 1);
557         assoc->init->charneg_request = ce->u.charNeg3;
558 #endif
559         assoc->backend = 0;
560         if (!(binitres = (*cb->bend_init)(assoc->init)))
561         {
562             assoc->state = ASSOC_DEAD;
563             yaz_add_srw_diagnostic(assoc->encode, d, num,
564                             YAZ_SRW_AUTHENTICATION_ERROR, 0);
565             return 0;
566         }
567         assoc->backend = binitres->handle;
568         assoc->init->auth = 0;
569         if (binitres->errcode)
570         {
571             int srw_code = yaz_diag_bib1_to_srw(binitres->errcode);
572             assoc->state = ASSOC_DEAD;
573             yaz_add_srw_diagnostic(assoc->encode, d, num, srw_code,
574                                    binitres->errstring);
575             return 0;
576         }
577         return 1;
578     }
579     return 1;
580 }
581
582 static int retrieve_fetch(association *assoc, bend_fetch_rr *rr)
583 {
584 #if YAZ_HAVE_XML2
585     yaz_record_conv_t rc = 0;
586     const char *match_schema = 0;
587     Odr_oid *match_syntax = 0;
588
589     if (assoc->server)
590     {
591         int r;
592         const char *input_schema = yaz_get_esn(rr->comp);
593         Odr_oid *input_syntax_raw = rr->request_format;
594         
595         const char *backend_schema = 0;
596         Odr_oid *backend_syntax = 0;
597
598         r = yaz_retrieval_request(assoc->server->retrieval,
599                                   input_schema,
600                                   input_syntax_raw,
601                                   &match_schema,
602                                   &match_syntax,
603                                   &rc,
604                                   &backend_schema,
605                                   &backend_syntax);
606         if (r == -1) /* error ? */
607         {
608             const char *details = yaz_retrieval_get_error(
609                 assoc->server->retrieval);
610
611             rr->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
612             if (details)
613                 rr->errstring = odr_strdup(rr->stream, details);
614             return -1;
615         }
616         else if (r == 1 || r == 3)
617         {
618             const char *details = input_schema;
619             rr->errcode =
620                 YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_;
621             if (details)
622                 rr->errstring = odr_strdup(rr->stream, details);
623             return -1;
624         }
625         else if (r == 2)
626         {
627             rr->errcode = YAZ_BIB1_RECORD_SYNTAX_UNSUPP;
628             if (input_syntax_raw)
629             {
630                 char oidbuf[OID_STR_MAX];
631                 oid_oid_to_dotstring(input_syntax_raw, oidbuf);
632                 rr->errstring = odr_strdup(rr->stream, oidbuf);
633             }
634             return -1;
635         }
636         if (backend_schema)
637         {
638             yaz_set_esn(&rr->comp, backend_schema, odr_getmem(rr->stream));
639         }
640         if (backend_syntax)
641             rr->request_format = backend_syntax;
642     }
643     (*assoc->init->bend_fetch)(assoc->backend, rr);
644     if (rc && rr->record && rr->errcode == 0 && rr->len > 0)
645     {   /* post conversion must take place .. */
646         WRBUF output_record = wrbuf_alloc();
647         int r = yaz_record_conv_record(rc, rr->record, rr->len, output_record);
648         if (r)
649         {
650             const char *details = yaz_record_conv_get_error(rc);
651             rr->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
652             if (details)
653                 rr->errstring = odr_strdup(rr->stream, details);
654         }
655         else
656         {
657             rr->len = wrbuf_len(output_record);
658             rr->record = (char *) odr_malloc(rr->stream, rr->len);
659             memcpy(rr->record, wrbuf_buf(output_record), rr->len);
660         }
661         wrbuf_destroy(output_record);
662     }
663     if (match_syntax)
664         rr->output_format = match_syntax;
665     if (match_schema)
666         rr->schema = odr_strdup(rr->stream, match_schema);
667 #else
668     (*assoc->init->bend_fetch)(assoc->backend, rr);
669 #endif
670     return 0;
671 }
672
673 static int srw_bend_fetch(association *assoc, int pos,
674                           Z_SRW_searchRetrieveRequest *srw_req,
675                           Z_SRW_record *record,
676                           const char **addinfo)
677 {
678     bend_fetch_rr rr;
679     ODR o = assoc->encode;
680
681     rr.setname = "default";
682     rr.number = pos;
683     rr.referenceId = 0;
684     rr.request_format = odr_oiddup(assoc->decode, yaz_oid_recsyn_xml);
685
686     rr.comp = (Z_RecordComposition *)
687             odr_malloc(assoc->decode, sizeof(*rr.comp));
688     rr.comp->which = Z_RecordComp_complex;
689     rr.comp->u.complex = (Z_CompSpec *)
690             odr_malloc(assoc->decode, sizeof(Z_CompSpec));
691     rr.comp->u.complex->selectAlternativeSyntax = (bool_t *)
692         odr_malloc(assoc->encode, sizeof(bool_t));
693     *rr.comp->u.complex->selectAlternativeSyntax = 0;    
694     rr.comp->u.complex->num_dbSpecific = 0;
695     rr.comp->u.complex->dbSpecific = 0;
696     rr.comp->u.complex->num_recordSyntax = 0; 
697     rr.comp->u.complex->recordSyntax = 0;
698
699     rr.comp->u.complex->generic = (Z_Specification *) 
700             odr_malloc(assoc->decode, sizeof(Z_Specification));
701
702     /* schema uri = recordSchema (or NULL if recordSchema is not given) */
703     rr.comp->u.complex->generic->which = Z_Schema_uri;
704     rr.comp->u.complex->generic->schema.uri = srw_req->recordSchema;
705
706     /* ESN = recordSchema if recordSchema is present */
707     rr.comp->u.complex->generic->elementSpec = 0;
708     if (srw_req->recordSchema)
709     {
710         rr.comp->u.complex->generic->elementSpec = 
711             (Z_ElementSpec *) odr_malloc(assoc->encode, sizeof(Z_ElementSpec));
712         rr.comp->u.complex->generic->elementSpec->which = 
713             Z_ElementSpec_elementSetName;
714         rr.comp->u.complex->generic->elementSpec->u.elementSetName =
715             srw_req->recordSchema;
716     }
717     
718     rr.stream = assoc->encode;
719     rr.print = assoc->print;
720
721     rr.basename = 0;
722     rr.len = 0;
723     rr.record = 0;
724     rr.last_in_set = 0;
725     rr.errcode = 0;
726     rr.errstring = 0;
727     rr.surrogate_flag = 0;
728     rr.schema = srw_req->recordSchema;
729
730     if (!assoc->init->bend_fetch)
731         return 1;
732
733     retrieve_fetch(assoc, &rr);
734
735     if (rr.errcode && rr.surrogate_flag)
736     {
737         int code = yaz_diag_bib1_to_srw(rr.errcode);
738         yaz_mk_sru_surrogate(o, record, pos, code, rr.errstring);
739         return 0;
740     }
741     else if (rr.len >= 0)
742     {
743         record->recordData_buf = rr.record;
744         record->recordData_len = rr.len;
745         record->recordPosition = odr_intdup(o, pos);
746         record->recordSchema = odr_strdup_null(
747             o, rr.schema ? rr.schema : srw_req->recordSchema);
748     }
749     if (rr.errcode)
750     {
751         *addinfo = rr.errstring;
752         return rr.errcode;
753     }
754     return 0;
755 }
756
757 static int cql2pqf(ODR odr, const char *cql, cql_transform_t ct,
758                    Z_Query *query_result)
759 {
760     /* have a CQL query and  CQL to PQF transform .. */
761     CQL_parser cp = cql_parser_create();
762     int r;
763     int srw_errcode = 0;
764     const char *add = 0;
765     char rpn_buf[5120];
766             
767     r = cql_parser_string(cp, cql);
768     if (r)
769     {
770         srw_errcode = YAZ_SRW_QUERY_SYNTAX_ERROR;
771     }
772     if (!r)
773     {
774         /* Syntax OK */
775         r = cql_transform_buf(ct,
776                               cql_parser_result(cp),
777                               rpn_buf, sizeof(rpn_buf)-1);
778         if (r)
779             srw_errcode = cql_transform_error(ct, &add);
780     }
781     if (!r)
782     {
783         /* Syntax & transform OK. */
784         /* Convert PQF string to Z39.50 to RPN query struct */
785         YAZ_PQF_Parser pp = yaz_pqf_create();
786         Z_RPNQuery *rpnquery = yaz_pqf_parse(pp, odr, rpn_buf);
787         if (!rpnquery)
788         {
789             size_t off;
790             const char *pqf_msg;
791             int code = yaz_pqf_error(pp, &pqf_msg, &off);
792             yaz_log(YLOG_WARN, "PQF Parser Error %s (code %d)",
793                     pqf_msg, code);
794             srw_errcode = YAZ_SRW_QUERY_SYNTAX_ERROR;
795         }
796         else
797         {
798             query_result->which = Z_Query_type_1;
799             query_result->u.type_1 = rpnquery;
800         }
801         yaz_pqf_destroy(pp);
802     }
803     cql_parser_destroy(cp);
804     return srw_errcode;
805 }
806
807 static int cql2pqf_scan(ODR odr, const char *cql, cql_transform_t ct,
808                         Z_AttributesPlusTerm *result)
809 {
810     Z_Query query;
811     Z_RPNQuery *rpn;
812     int srw_error = cql2pqf(odr, cql, ct, &query);
813     if (srw_error)
814         return srw_error;
815     if (query.which != Z_Query_type_1 && query.which != Z_Query_type_101)
816         return YAZ_SRW_QUERY_SYNTAX_ERROR; /* bad query type */
817     rpn = query.u.type_1;
818     if (!rpn->RPNStructure) 
819         return YAZ_SRW_QUERY_SYNTAX_ERROR; /* must be structure */
820     if (rpn->RPNStructure->which != Z_RPNStructure_simple)
821         return YAZ_SRW_QUERY_SYNTAX_ERROR; /* must be simple */
822     if (rpn->RPNStructure->u.simple->which != Z_Operand_APT)
823         return YAZ_SRW_QUERY_SYNTAX_ERROR; /* must be be attributes + term */
824     memcpy(result, rpn->RPNStructure->u.simple->u.attributesPlusTerm,
825            sizeof(*result));
826     return 0;
827 }
828                    
829
830 static int ccl2pqf(ODR odr, const Odr_oct *ccl, CCL_bibset bibset,
831                    bend_search_rr *bsrr) {
832     char *ccl0;
833     struct ccl_rpn_node *node;
834     int errcode, pos;
835
836     ccl0 = odr_strdupn(odr, (char*) ccl->buf, ccl->len);
837     if ((node = ccl_find_str(bibset, ccl0, &errcode, &pos)) == 0) {
838         bsrr->errstring = (char*) ccl_err_msg(errcode);
839         return YAZ_SRW_QUERY_SYNTAX_ERROR;    /* Query syntax error */
840     }
841
842     bsrr->query->which = Z_Query_type_1;
843     bsrr->query->u.type_1 = ccl_rpn_query(odr, node);
844     return 0;
845 }
846
847
848 static void srw_bend_search(association *assoc, request *req,
849                             Z_SRW_PDU *sr,
850                             Z_SRW_searchRetrieveResponse *srw_res,
851                             int *http_code)
852 {
853     int srw_error = 0;
854     Z_External *ext;
855     Z_SRW_searchRetrieveRequest *srw_req = sr->u.request;
856     
857     *http_code = 200;
858     yaz_log(log_requestdetail, "Got SRW SearchRetrieveRequest");
859     srw_bend_init(assoc, &srw_res->diagnostics, &srw_res->num_diagnostics, sr);
860     if (srw_res->num_diagnostics == 0 && assoc->init)
861     {
862         bend_search_rr rr;
863         rr.setname = "default";
864         rr.replace_set = 1;
865         rr.num_bases = 1;
866         rr.basenames = &srw_req->database;
867         rr.referenceId = 0;
868         rr.srw_sortKeys = 0;
869         rr.srw_setname = 0;
870         rr.srw_setnameIdleTime = 0;
871         rr.estimated_hit_count = 0;
872         rr.partial_resultset = 0;
873         rr.query = (Z_Query *) odr_malloc(assoc->decode, sizeof(*rr.query));
874         rr.query->u.type_1 = 0;
875         
876         if (srw_req->query_type == Z_SRW_query_type_cql)
877         {
878             if (assoc->server && assoc->server->cql_transform)
879             {
880                 int srw_errcode = cql2pqf(assoc->encode, srw_req->query.cql,
881                                           assoc->server->cql_transform,
882                                           rr.query);
883                 if (srw_errcode)
884                 {
885                     yaz_add_srw_diagnostic(assoc->encode,
886                                            &srw_res->diagnostics,
887                                            &srw_res->num_diagnostics,
888                                            srw_errcode, 0);
889                 }
890             }
891             else
892             {
893                 /* CQL query to backend. Wrap it - Z39.50 style */
894                 ext = (Z_External *) odr_malloc(assoc->decode, sizeof(*ext));
895                 ext->direct_reference = odr_getoidbystr(assoc->decode, 
896                                                         "1.2.840.10003.16.2");
897                 ext->indirect_reference = 0;
898                 ext->descriptor = 0;
899                 ext->which = Z_External_CQL;
900                 ext->u.cql = srw_req->query.cql;
901                 
902                 rr.query->which = Z_Query_type_104;
903                 rr.query->u.type_104 =  ext;
904             }
905         }
906         else if (srw_req->query_type == Z_SRW_query_type_pqf)
907         {
908             Z_RPNQuery *RPNquery;
909             YAZ_PQF_Parser pqf_parser;
910             
911             pqf_parser = yaz_pqf_create();
912             
913             RPNquery = yaz_pqf_parse(pqf_parser, assoc->decode,
914                                      srw_req->query.pqf);
915             if (!RPNquery)
916             {
917                 const char *pqf_msg;
918                 size_t off;
919                 int code = yaz_pqf_error(pqf_parser, &pqf_msg, &off);
920                 yaz_log(log_requestdetail, "Parse error %d %s near offset %ld",
921                         code, pqf_msg, (long) off);
922                 srw_error = YAZ_SRW_QUERY_SYNTAX_ERROR;
923             }
924             
925             rr.query->which = Z_Query_type_1;
926             rr.query->u.type_1 =  RPNquery;
927             
928             yaz_pqf_destroy(pqf_parser);
929         }
930         else
931         {
932             yaz_add_srw_diagnostic(assoc->encode, &srw_res->diagnostics,
933                                    &srw_res->num_diagnostics,
934                                    YAZ_SRW_UNSUPP_QUERY_TYPE, 0);
935         }
936         if (rr.query->u.type_1)
937         {
938             rr.stream = assoc->encode;
939             rr.decode = assoc->decode;
940             rr.print = assoc->print;
941             rr.request = req;
942             if ( srw_req->sort.sortKeys )
943                 rr.srw_sortKeys = odr_strdup(assoc->encode, 
944                                              srw_req->sort.sortKeys );
945             rr.association = assoc;
946             rr.fd = 0;
947             rr.hits = 0;
948             rr.errcode = 0;
949             rr.errstring = 0;
950             rr.search_info = 0;
951             yaz_log_zquery_level(log_requestdetail,rr.query);
952             
953             (assoc->init->bend_search)(assoc->backend, &rr);
954             if (rr.errcode)
955             {
956                 if (rr.errcode == YAZ_BIB1_DATABASE_UNAVAILABLE)
957                 {
958                     *http_code = 404;
959                 }
960                 else
961                 {
962                     srw_error = yaz_diag_bib1_to_srw(rr.errcode);
963                     yaz_add_srw_diagnostic(assoc->encode,
964                                            &srw_res->diagnostics,
965                                            &srw_res->num_diagnostics,
966                                            srw_error, rr.errstring);
967                 }
968             }
969             else
970             {
971                 int number = srw_req->maximumRecords ? *srw_req->maximumRecords : 0;
972                 int start = srw_req->startRecord ? *srw_req->startRecord : 1;
973                 
974                 yaz_log(log_requestdetail, "Request to pack %d+%d out of "
975                         ODR_INT_PRINTF,
976                         start, number, rr.hits);
977                 
978                 srw_res->numberOfRecords = odr_intdup(assoc->encode, rr.hits);
979                 if (rr.srw_setname)
980                 {
981                     srw_res->resultSetId =
982                         odr_strdup(assoc->encode, rr.srw_setname );
983                     srw_res->resultSetIdleTime =
984                         odr_intdup(assoc->encode, *rr.srw_setnameIdleTime );
985                 }
986                 
987                 if (start > rr.hits || start < 1)
988                 {
989                     /* if hits<=0 and start=1 we don't return a diagnostic */
990                     if (start != 1)
991                         yaz_add_srw_diagnostic(
992                             assoc->encode, 
993                             &srw_res->diagnostics, &srw_res->num_diagnostics,
994                             YAZ_SRW_FIRST_RECORD_POSITION_OUT_OF_RANGE, 0);
995                 }
996                 else if (number > 0)
997                 {
998                     int i;
999                     int ok = 1;
1000                     if (start + number > rr.hits)
1001                         number = rr.hits - start + 1;
1002                     
1003                     /* Call bend_present if defined */
1004                     if (assoc->init->bend_present)
1005                     {
1006                         bend_present_rr *bprr = (bend_present_rr*)
1007                             odr_malloc(assoc->decode, sizeof(*bprr));
1008                         bprr->setname = "default";
1009                         bprr->start = start;
1010                         bprr->number = number;
1011                         if (srw_req->recordSchema)
1012                         {
1013                             bprr->comp = (Z_RecordComposition *) odr_malloc(assoc->decode,
1014                                                                             sizeof(*bprr->comp));
1015                             bprr->comp->which = Z_RecordComp_simple;
1016                             bprr->comp->u.simple = (Z_ElementSetNames *)
1017                                 odr_malloc(assoc->decode, sizeof(Z_ElementSetNames));
1018                             bprr->comp->u.simple->which = Z_ElementSetNames_generic;
1019                             bprr->comp->u.simple->u.generic = srw_req->recordSchema;
1020                         }
1021                         else
1022                         {
1023                             bprr->comp = 0;
1024                         }
1025                         bprr->stream = assoc->encode;
1026                         bprr->referenceId = 0;
1027                         bprr->print = assoc->print;
1028                         bprr->request = req;
1029                         bprr->association = assoc;
1030                         bprr->errcode = 0;
1031                         bprr->errstring = NULL;
1032                         (*assoc->init->bend_present)(assoc->backend, bprr);
1033                         
1034                         if (!bprr->request)
1035                             return;
1036                         if (bprr->errcode)
1037                         {
1038                             srw_error = yaz_diag_bib1_to_srw(bprr->errcode);
1039                             yaz_add_srw_diagnostic(assoc->encode,
1040                                                    &srw_res->diagnostics,
1041                                                    &srw_res->num_diagnostics,
1042                                                    srw_error, bprr->errstring);
1043                             ok = 0;
1044                         }
1045                     }
1046                     
1047                     if (ok)
1048                     {
1049                         int j = 0;
1050                         int packing = Z_SRW_recordPacking_string;
1051                         if (srw_req->recordPacking)
1052                         {
1053                             packing = 
1054                                 yaz_srw_str_to_pack(srw_req->recordPacking);
1055                             if (packing == -1)
1056                                 packing = Z_SRW_recordPacking_string;
1057                         }
1058                         srw_res->records = (Z_SRW_record *)
1059                             odr_malloc(assoc->encode,
1060                                        number * sizeof(*srw_res->records));
1061                         
1062                         srw_res->extra_records = (Z_SRW_extra_record **)
1063                             odr_malloc(assoc->encode,
1064                                        number*sizeof(*srw_res->extra_records));
1065
1066                         for (i = 0; i<number; i++)
1067                         {
1068                             int errcode;
1069                             const char *addinfo = 0;
1070                             
1071                             srw_res->records[j].recordPacking = packing;
1072                             srw_res->records[j].recordData_buf = 0;
1073                             srw_res->extra_records[j] = 0;
1074                             yaz_log(YLOG_DEBUG, "srw_bend_fetch %d", i+start);
1075                             errcode = srw_bend_fetch(assoc, i+start, srw_req,
1076                                                      srw_res->records + j,
1077                                                      &addinfo);
1078                             if (errcode)
1079                             {
1080                                 yaz_add_srw_diagnostic(assoc->encode,
1081                                                        &srw_res->diagnostics,
1082                                                        &srw_res->num_diagnostics,
1083                                                        yaz_diag_bib1_to_srw(errcode),
1084                                                        addinfo);
1085                                 
1086                                 break;
1087                             }
1088                             if (srw_res->records[j].recordData_buf)
1089                                 j++;
1090                         }
1091                         srw_res->num_records = j;
1092                         if (!j)
1093                             srw_res->records = 0;
1094                     }
1095                 }
1096                 if (rr.estimated_hit_count || rr.partial_resultset)
1097                 {
1098                     yaz_add_srw_diagnostic(
1099                         assoc->encode,
1100                         &srw_res->diagnostics,
1101                         &srw_res->num_diagnostics,
1102                         YAZ_SRW_RESULT_SET_CREATED_WITH_VALID_PARTIAL_RESULTS_AVAILABLE,
1103                         0);
1104                 }
1105             }
1106         }
1107     }
1108     if (log_request)
1109     {
1110         const char *querystr = "?";
1111         const char *querytype = "?";
1112         WRBUF wr = wrbuf_alloc();
1113
1114         switch (srw_req->query_type)
1115         {
1116         case Z_SRW_query_type_cql:
1117             querytype = "CQL";
1118             querystr = srw_req->query.cql;
1119             break;
1120         case Z_SRW_query_type_pqf:
1121             querytype = "PQF";
1122             querystr = srw_req->query.pqf;
1123             break;
1124         }
1125         wrbuf_printf(wr, "SRWSearch %s ", srw_req->database);
1126         if (srw_res->num_diagnostics)
1127             wrbuf_printf(wr, "ERROR %s", srw_res->diagnostics[0].uri);
1128         else if (*http_code != 200)
1129             wrbuf_printf(wr, "ERROR info:http/%d", *http_code);
1130         else if (srw_res->numberOfRecords)
1131         {
1132             wrbuf_printf(wr, "OK " ODR_INT_PRINTF,
1133                          (srw_res->numberOfRecords ?
1134                           *srw_res->numberOfRecords : 0));
1135         }
1136         wrbuf_printf(wr, " %s " ODR_INT_PRINTF " +%d", 
1137                      (srw_res->resultSetId ?
1138                       srw_res->resultSetId : "-"),
1139                      (srw_req->startRecord ? *srw_req->startRecord : 1), 
1140                      srw_res->num_records);
1141         yaz_log(log_request, "%s %s: %s", wrbuf_cstr(wr), querytype, querystr);
1142         wrbuf_destroy(wr);
1143     }
1144 }
1145
1146 static char *srw_bend_explain_default(void *handle, bend_explain_rr *rr)
1147 {
1148 #if YAZ_HAVE_XML2
1149     xmlNodePtr ptr = (xmlNode *) rr->server_node_ptr;
1150     if (!ptr)
1151         return 0;
1152     for (ptr = ptr->children; ptr; ptr = ptr->next)
1153     {
1154         if (ptr->type != XML_ELEMENT_NODE)
1155             continue;
1156         if (!strcmp((const char *) ptr->name, "explain"))
1157         {
1158             int len;
1159             xmlDocPtr doc = xmlNewDoc(BAD_CAST "1.0");
1160             xmlChar *buf_out;
1161             char *content;
1162
1163             ptr = xmlCopyNode(ptr, 1);
1164         
1165             xmlDocSetRootElement(doc, ptr);
1166             
1167             xmlDocDumpMemory(doc, &buf_out, &len);
1168             content = (char*) odr_malloc(rr->stream, 1+len);
1169             memcpy(content, buf_out, len);
1170             content[len] = '\0';
1171             
1172             xmlFree(buf_out);
1173             xmlFreeDoc(doc);
1174             rr->explain_buf = content;
1175             return 0;
1176         }
1177     }
1178 #endif
1179     return 0;
1180 }
1181
1182 static void srw_bend_explain(association *assoc, request *req,
1183                              Z_SRW_PDU *sr,
1184                              Z_SRW_explainResponse *srw_res,
1185                              int *http_code)
1186 {
1187     Z_SRW_explainRequest *srw_req = sr->u.explain_request;
1188     yaz_log(log_requestdetail, "Got SRW ExplainRequest");
1189     *http_code = 404;
1190     srw_bend_init(assoc, &srw_res->diagnostics, &srw_res->num_diagnostics, sr);
1191     if (assoc->init)
1192     {
1193         bend_explain_rr rr;
1194         
1195         rr.stream = assoc->encode;
1196         rr.decode = assoc->decode;
1197         rr.print = assoc->print;
1198         rr.explain_buf = 0;
1199         rr.database = srw_req->database;
1200         if (assoc->server)
1201             rr.server_node_ptr = assoc->server->server_node_ptr;
1202         else
1203             rr.server_node_ptr = 0;
1204         rr.schema = "http://explain.z3950.org/dtd/2.0/";
1205         if (assoc->init->bend_explain)
1206             (*assoc->init->bend_explain)(assoc->backend, &rr);
1207         else
1208             srw_bend_explain_default(assoc->backend, &rr);
1209
1210         if (rr.explain_buf)
1211         {
1212             int packing = Z_SRW_recordPacking_string;
1213             if (srw_req->recordPacking)
1214             {
1215                 packing = 
1216                     yaz_srw_str_to_pack(srw_req->recordPacking);
1217                 if (packing == -1)
1218                     packing = Z_SRW_recordPacking_string;
1219             }
1220             srw_res->record.recordSchema = rr.schema;
1221             srw_res->record.recordPacking = packing;
1222             srw_res->record.recordData_buf = rr.explain_buf;
1223             srw_res->record.recordData_len = strlen(rr.explain_buf);
1224             srw_res->record.recordPosition = 0;
1225             *http_code = 200;
1226         }
1227     }
1228 }
1229
1230 static void srw_bend_scan(association *assoc, request *req,
1231                           Z_SRW_PDU *sr,
1232                           Z_SRW_scanResponse *srw_res,
1233                           int *http_code)
1234 {
1235     Z_SRW_scanRequest *srw_req = sr->u.scan_request;
1236     yaz_log(log_requestdetail, "Got SRW ScanRequest");
1237
1238     *http_code = 200;
1239     srw_bend_init(assoc, &srw_res->diagnostics, &srw_res->num_diagnostics, sr);
1240     if (srw_res->num_diagnostics == 0 && assoc->init)
1241     {
1242         struct scan_entry *save_entries;
1243
1244         bend_scan_rr *bsrr = (bend_scan_rr *)
1245             odr_malloc(assoc->encode, sizeof(*bsrr));
1246         bsrr->num_bases = 1;
1247         bsrr->basenames = &srw_req->database;
1248
1249         bsrr->num_entries = srw_req->maximumTerms ?
1250             *srw_req->maximumTerms : 10;
1251         bsrr->term_position = srw_req->responsePosition ?
1252             *srw_req->responsePosition : 1;
1253
1254         bsrr->errcode = 0;
1255         bsrr->errstring = 0;
1256         bsrr->referenceId = 0;
1257         bsrr->stream = assoc->encode;
1258         bsrr->print = assoc->print;
1259         bsrr->step_size = odr_intdup(assoc->decode, 0);
1260         bsrr->entries = 0;
1261         bsrr->setname = 0;
1262
1263         if (bsrr->num_entries > 0) 
1264         {
1265             int i;
1266             bsrr->entries = (struct scan_entry *) 
1267                 odr_malloc(assoc->decode, sizeof(*bsrr->entries) *
1268                            bsrr->num_entries);
1269             for (i = 0; i<bsrr->num_entries; i++)
1270             {
1271                 bsrr->entries[i].term = 0;
1272                 bsrr->entries[i].occurrences = 0;
1273                 bsrr->entries[i].errcode = 0;
1274                 bsrr->entries[i].errstring = 0;
1275                 bsrr->entries[i].display_term = 0;
1276             }
1277         }
1278         save_entries = bsrr->entries;  /* save it so we can compare later */
1279
1280         if (srw_req->query_type == Z_SRW_query_type_pqf &&
1281             assoc->init->bend_scan)
1282         {
1283             YAZ_PQF_Parser pqf_parser = yaz_pqf_create();
1284             
1285             bsrr->term = yaz_pqf_scan(pqf_parser, assoc->decode,
1286                                       &bsrr->attributeset, 
1287                                       srw_req->scanClause.pqf); 
1288             yaz_pqf_destroy(pqf_parser);
1289             bsrr->scanClause = 0;
1290             ((int (*)(void *, bend_scan_rr *))
1291              (*assoc->init->bend_scan))(assoc->backend, bsrr);
1292         }
1293         else if (srw_req->query_type == Z_SRW_query_type_cql
1294                  && assoc->init->bend_scan && assoc->server
1295                  && assoc->server->cql_transform)
1296         {
1297             int srw_error;
1298             bsrr->scanClause = 0;
1299             bsrr->attributeset = 0;
1300             bsrr->term = (Z_AttributesPlusTerm *)
1301                 odr_malloc(assoc->decode, sizeof(*bsrr->term));
1302             srw_error = cql2pqf_scan(assoc->encode,
1303                                      srw_req->scanClause.cql,
1304                                      assoc->server->cql_transform,
1305                                      bsrr->term);
1306             if (srw_error)
1307                 yaz_add_srw_diagnostic(assoc->encode, &srw_res->diagnostics,
1308                                        &srw_res->num_diagnostics,
1309                                        srw_error, 0);
1310             else
1311             {
1312                 ((int (*)(void *, bend_scan_rr *))
1313                  (*assoc->init->bend_scan))(assoc->backend, bsrr);
1314             }
1315         }
1316         else if (srw_req->query_type == Z_SRW_query_type_cql
1317                  && assoc->init->bend_srw_scan)
1318         {
1319             bsrr->term = 0;
1320             bsrr->attributeset = 0;
1321             bsrr->scanClause = srw_req->scanClause.cql;
1322             ((int (*)(void *, bend_scan_rr *))
1323              (*assoc->init->bend_srw_scan))(assoc->backend, bsrr);
1324         }
1325         else
1326         {
1327             yaz_add_srw_diagnostic(assoc->encode, &srw_res->diagnostics,
1328                                    &srw_res->num_diagnostics,
1329                                    YAZ_SRW_UNSUPP_OPERATION, "scan");
1330         }
1331         if (bsrr->errcode)
1332         {
1333             int srw_error;
1334             if (bsrr->errcode == YAZ_BIB1_DATABASE_UNAVAILABLE)
1335             {
1336                 *http_code = 404;
1337                 return;
1338             }
1339             srw_error = yaz_diag_bib1_to_srw(bsrr->errcode);
1340
1341             yaz_add_srw_diagnostic(assoc->encode, &srw_res->diagnostics,
1342                                    &srw_res->num_diagnostics,
1343                                    srw_error, bsrr->errstring);
1344         }
1345         else if (srw_res->num_diagnostics == 0 && bsrr->num_entries)
1346         {
1347             int i;
1348             srw_res->terms = (Z_SRW_scanTerm*)
1349                 odr_malloc(assoc->encode, sizeof(*srw_res->terms) *
1350                            bsrr->num_entries);
1351
1352             srw_res->num_terms =  bsrr->num_entries;
1353             for (i = 0; i<bsrr->num_entries; i++)
1354             {
1355                 Z_SRW_scanTerm *t = srw_res->terms + i;
1356                 t->value = odr_strdup(assoc->encode, bsrr->entries[i].term);
1357                 t->numberOfRecords =
1358                     odr_intdup(assoc->encode, bsrr->entries[i].occurrences);
1359                 t->displayTerm = 0;
1360                 if (save_entries == bsrr->entries && 
1361                     bsrr->entries[i].display_term)
1362                 {
1363                     /* the entries was _not_ set by the handler. So it's
1364                        safe to test for new member display_term. It is
1365                        NULL'ed by us.
1366                     */
1367                     t->displayTerm = odr_strdup(assoc->encode, 
1368                                                 bsrr->entries[i].display_term);
1369                 }
1370                 t->whereInList = 0;
1371             }
1372         }
1373     }
1374     if (log_request)
1375     {
1376         WRBUF wr = wrbuf_alloc();
1377         const char *querytype = 0;
1378         const char *querystr = 0;
1379
1380         switch(srw_req->query_type)
1381         {
1382         case Z_SRW_query_type_pqf:
1383             querytype = "PQF";
1384             querystr = srw_req->scanClause.pqf;
1385             break;
1386         case Z_SRW_query_type_cql:
1387             querytype = "CQL";
1388             querystr = srw_req->scanClause.cql;
1389             break;
1390         default:
1391             querytype = "UNKNOWN";
1392             querystr = "";
1393         }
1394
1395         wrbuf_printf(wr, "SRWScan %s ", srw_req->database);
1396
1397         if (srw_res->num_diagnostics)
1398             wrbuf_printf(wr, "ERROR %s - ", srw_res->diagnostics[0].uri);
1399         else if (srw_res->num_terms)
1400             wrbuf_printf(wr, "OK %d - ", srw_res->num_terms);
1401         else
1402             wrbuf_printf(wr, "OK - - ");
1403
1404         wrbuf_printf(wr, ODR_INT_PRINTF "+" ODR_INT_PRINTF " ",
1405                      (srw_req->responsePosition ? 
1406                       *srw_req->responsePosition : 1),
1407                      (srw_req->maximumTerms ?
1408                       *srw_req->maximumTerms : 1));
1409         /* there is no step size in SRU/W ??? */
1410         wrbuf_printf(wr, "%s: %s ", querytype, querystr);
1411         yaz_log(log_request, "%s ", wrbuf_cstr(wr) );
1412         wrbuf_destroy(wr);
1413     }
1414
1415 }
1416
1417 static void srw_bend_update(association *assoc, request *req,
1418                             Z_SRW_PDU *sr,
1419                             Z_SRW_updateResponse *srw_res,
1420                             int *http_code)
1421 {
1422     Z_SRW_updateRequest *srw_req = sr->u.update_request;
1423     yaz_log(log_session, "SRWUpdate action=%s", srw_req->operation);
1424     yaz_log(YLOG_DEBUG, "num_diag = %d", srw_res->num_diagnostics );
1425     *http_code = 404;
1426     srw_bend_init(assoc, &srw_res->diagnostics, &srw_res->num_diagnostics, sr);
1427     if (assoc->init)
1428     {
1429         bend_update_rr rr;
1430         Z_SRW_extra_record *extra = srw_req->extra_record;
1431         
1432         rr.stream = assoc->encode;
1433         rr.print = assoc->print;
1434         rr.num_bases = 1;
1435         rr.basenames = &srw_req->database;
1436         rr.operation = srw_req->operation;
1437         rr.operation_status = "failed";
1438         rr.record_id = 0;
1439         rr.record_versions = 0;
1440         rr.num_versions = 0;
1441         rr.record_packing = "string";
1442         rr.record_schema = 0;
1443         rr.record_data = 0;
1444         rr.extra_record_data = 0;
1445         rr.extra_request_data = 0;
1446         rr.extra_response_data = 0;
1447         rr.uri = 0;
1448         rr.message = 0;
1449         rr.details = 0;
1450         
1451         *http_code = 200;
1452         if (rr.operation == 0)
1453         {
1454             yaz_add_sru_update_diagnostic(
1455                 assoc->encode, &srw_res->diagnostics,
1456                 &srw_res->num_diagnostics,
1457                 YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,
1458                 "action" );
1459             return;
1460         }
1461         yaz_log(YLOG_DEBUG, "basename = %s", rr.basenames[0] );
1462         yaz_log(YLOG_DEBUG, "Operation = %s", rr.operation );
1463         if (!strcmp( rr.operation, "delete"))
1464         {
1465             if (srw_req->record && !srw_req->record->recordSchema)
1466             {
1467                 rr.record_schema = odr_strdup(
1468                     assoc->encode,
1469                     srw_req->record->recordSchema);
1470             }
1471             if (srw_req->record)
1472             {
1473                 rr.record_data = odr_strdupn(
1474                     assoc->encode, 
1475                     srw_req->record->recordData_buf,
1476                     srw_req->record->recordData_len );
1477             }
1478             if (extra && extra->extraRecordData_len)
1479             {
1480                 rr.extra_record_data = odr_strdupn(
1481                     assoc->encode, 
1482                     extra->extraRecordData_buf,
1483                     extra->extraRecordData_len );
1484             }
1485             if (srw_req->recordId)
1486                 rr.record_id = srw_req->recordId;
1487             else if (extra && extra->recordIdentifier)
1488                 rr.record_id = extra->recordIdentifier;
1489         }
1490         else if (!strcmp(rr.operation, "replace"))
1491         {
1492             if (srw_req->recordId)
1493                 rr.record_id = srw_req->recordId;
1494             else if (extra && extra->recordIdentifier)
1495                 rr.record_id = extra->recordIdentifier;
1496             else 
1497             {
1498                 yaz_add_sru_update_diagnostic(
1499                     assoc->encode, &srw_res->diagnostics,
1500                     &srw_res->num_diagnostics,
1501                     YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,
1502                     "recordIdentifier");
1503             }
1504             if (!srw_req->record)
1505             {
1506                 yaz_add_sru_update_diagnostic(
1507                     assoc->encode, &srw_res->diagnostics,
1508                     &srw_res->num_diagnostics,
1509                     YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,
1510                     "record");
1511             }
1512             else 
1513             {
1514                 if (srw_req->record->recordSchema)
1515                     rr.record_schema = odr_strdup(
1516                         assoc->encode, srw_req->record->recordSchema);
1517                 if (srw_req->record->recordData_len )
1518                 {
1519                     rr.record_data = odr_strdupn(assoc->encode, 
1520                                                  srw_req->record->recordData_buf,
1521                                                  srw_req->record->recordData_len );
1522                 }
1523                 else 
1524                 {
1525                     yaz_add_sru_update_diagnostic(
1526                         assoc->encode, &srw_res->diagnostics,
1527                         &srw_res->num_diagnostics,
1528                         YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,                                              
1529                         "recordData" );
1530                 }
1531             }
1532             if (extra && extra->extraRecordData_len)
1533             {
1534                 rr.extra_record_data = odr_strdupn(
1535                     assoc->encode, 
1536                     extra->extraRecordData_buf,
1537                     extra->extraRecordData_len );
1538             }
1539         }
1540         else if (!strcmp(rr.operation, "insert"))
1541         {
1542             if (srw_req->recordId)
1543                 rr.record_id = srw_req->recordId; 
1544             else if (extra)
1545                 rr.record_id = extra->recordIdentifier;
1546             
1547             if (srw_req->record)
1548             {
1549                 if (srw_req->record->recordSchema)
1550                     rr.record_schema = odr_strdup(
1551                         assoc->encode, srw_req->record->recordSchema);
1552             
1553                 if (srw_req->record->recordData_len)
1554                     rr.record_data = odr_strdupn(
1555                         assoc->encode, 
1556                         srw_req->record->recordData_buf,
1557                         srw_req->record->recordData_len );
1558             }
1559             if (extra && extra->extraRecordData_len)
1560             {
1561                 rr.extra_record_data = odr_strdupn(
1562                     assoc->encode, 
1563                     extra->extraRecordData_buf,
1564                     extra->extraRecordData_len );
1565             }
1566         }
1567         else 
1568             yaz_add_sru_update_diagnostic(assoc->encode, &srw_res->diagnostics,
1569                                           &srw_res->num_diagnostics,
1570                                           YAZ_SRU_UPDATE_INVALID_ACTION,
1571                                           rr.operation );
1572
1573         if (srw_req->record)
1574         {
1575             const char *pack_str = 
1576                 yaz_srw_pack_to_str(srw_req->record->recordPacking);
1577             if (pack_str)
1578                 rr.record_packing = odr_strdup(assoc->encode, pack_str);
1579         }
1580
1581         if (srw_req->num_recordVersions)
1582         {
1583             rr.record_versions = srw_req->recordVersions;
1584             rr.num_versions = srw_req->num_recordVersions;
1585         }
1586         if (srw_req->extraRequestData_len)
1587         {
1588             rr.extra_request_data = odr_strdupn(assoc->encode,
1589                                                 srw_req->extraRequestData_buf,
1590                                                 srw_req->extraRequestData_len );
1591         }
1592         if (srw_res->num_diagnostics == 0)
1593         {
1594             if ( assoc->init->bend_srw_update)
1595                 (*assoc->init->bend_srw_update)(assoc->backend, &rr);
1596             else 
1597                 yaz_add_sru_update_diagnostic(
1598                     assoc->encode, &srw_res->diagnostics,
1599                     &srw_res->num_diagnostics,
1600                     YAZ_SRU_UPDATE_UNSPECIFIED_DATABASE_ERROR,
1601                     "No Update backend handler");
1602         }
1603
1604         if (rr.uri)
1605             yaz_add_srw_diagnostic_uri(assoc->encode,
1606                                        &srw_res->diagnostics,
1607                                        &srw_res->num_diagnostics,
1608                                        rr.uri, 
1609                                        rr.message,
1610                                        rr.details);
1611         srw_res->recordId = rr.record_id;
1612         srw_res->operationStatus = rr.operation_status;
1613         srw_res->recordVersions = rr.record_versions;
1614         srw_res->num_recordVersions = rr.num_versions;
1615         if (srw_res->extraResponseData_len)
1616         {
1617             srw_res->extraResponseData_buf = rr.extra_response_data;
1618             srw_res->extraResponseData_len = strlen(rr.extra_response_data);
1619         }
1620         if (srw_res->num_diagnostics == 0 && rr.record_data)
1621         {
1622             srw_res->record = yaz_srw_get_record(assoc->encode);
1623             srw_res->record->recordSchema = rr.record_schema;
1624             if (rr.record_packing)
1625             {
1626                 int pack = yaz_srw_str_to_pack(rr.record_packing);
1627
1628                 if (pack == -1)
1629                 {
1630                     pack = Z_SRW_recordPacking_string;
1631                     yaz_log(YLOG_WARN, "Back packing %s from backend",
1632                             rr.record_packing);
1633                 }
1634                 srw_res->record->recordPacking = pack;
1635             }
1636             srw_res->record->recordData_buf = rr.record_data;
1637             srw_res->record->recordData_len = strlen(rr.record_data);
1638             if (rr.extra_record_data)
1639             {
1640                 Z_SRW_extra_record *ex = 
1641                     yaz_srw_get_extra_record(assoc->encode);
1642                 srw_res->extra_record = ex;
1643                 ex->extraRecordData_buf = rr.extra_record_data;
1644                 ex->extraRecordData_len = strlen(rr.extra_record_data);
1645             }
1646         }
1647     }
1648 }
1649
1650 /* check if path is OK (1); BAD (0) */
1651 static int check_path(const char *path)
1652 {
1653     if (*path != '/')
1654         return 0;
1655     if (strstr(path, ".."))
1656         return 0;
1657     return 1;
1658 }
1659
1660 static char *read_file(const char *fname, ODR o, size_t *sz)
1661 {
1662     char *buf;
1663     FILE *inf = fopen(fname, "rb");
1664     if (!inf)
1665         return 0;
1666
1667     fseek(inf, 0L, SEEK_END);
1668     *sz = ftell(inf);
1669     rewind(inf);
1670     buf = (char *) odr_malloc(o, *sz);
1671     if (fread(buf, 1, *sz, inf) != *sz)
1672         yaz_log(YLOG_WARN|YLOG_ERRNO, "short read %s", fname);
1673     fclose(inf);
1674     return buf;     
1675 }
1676
1677 static void process_http_request(association *assoc, request *req)
1678 {
1679     Z_HTTP_Request *hreq = req->gdu_request->u.HTTP_Request;
1680     ODR o = assoc->encode;
1681     int r = 2;  /* 2=NOT TAKEN, 1=TAKEN, 0=SOAP TAKEN */
1682     Z_SRW_PDU *sr = 0;
1683     Z_SOAP *soap_package = 0;
1684     Z_GDU *p = 0;
1685     char *charset = 0;
1686     Z_HTTP_Response *hres = 0;
1687     int keepalive = 1;
1688     const char *stylesheet = 0; /* for now .. set later */
1689     Z_SRW_diagnostic *diagnostic = 0;
1690     int num_diagnostic = 0;
1691     const char *host = z_HTTP_header_lookup(hreq->headers, "Host");
1692
1693     if (!control_association(assoc, host, 0))
1694     {
1695         p = z_get_HTTP_Response(o, 404);
1696         r = 1;
1697     }
1698     if (r == 2 && assoc->server && assoc->server->docpath
1699         && hreq->path[0] == '/' 
1700         && 
1701         /* check if path is a proper prefix of documentroot */
1702         strncmp(hreq->path+1, assoc->server->docpath,
1703                 strlen(assoc->server->docpath))
1704         == 0)
1705     {   
1706         if (!check_path(hreq->path))
1707         {
1708             yaz_log(YLOG_LOG, "File %s access forbidden", hreq->path+1);
1709             p = z_get_HTTP_Response(o, 404);
1710         }
1711         else
1712         {
1713             size_t content_size = 0;
1714             char *content_buf = read_file(hreq->path+1, o, &content_size);
1715             if (!content_buf)
1716             {
1717                 yaz_log(YLOG_LOG, "File %s not found", hreq->path+1);
1718                 p = z_get_HTTP_Response(o, 404);
1719             }
1720             else
1721             {
1722                 const char *ctype = 0;
1723                 yaz_mime_types types = yaz_mime_types_create();
1724                 
1725                 yaz_mime_types_add(types, "xsl", "application/xml");
1726                 yaz_mime_types_add(types, "xml", "application/xml");
1727                 yaz_mime_types_add(types, "css", "text/css");
1728                 yaz_mime_types_add(types, "html", "text/html");
1729                 yaz_mime_types_add(types, "htm", "text/html");
1730                 yaz_mime_types_add(types, "txt", "text/plain");
1731                 yaz_mime_types_add(types, "js", "application/x-javascript");
1732                 
1733                 yaz_mime_types_add(types, "gif", "image/gif");
1734                 yaz_mime_types_add(types, "png", "image/png");
1735                 yaz_mime_types_add(types, "jpg", "image/jpeg");
1736                 yaz_mime_types_add(types, "jpeg", "image/jpeg");
1737                 
1738                 ctype = yaz_mime_lookup_fname(types, hreq->path);
1739                 if (!ctype)
1740                 {
1741                     yaz_log(YLOG_LOG, "No mime type for %s", hreq->path+1);
1742                     p = z_get_HTTP_Response(o, 404);
1743                 }
1744                 else
1745                 {
1746                     p = z_get_HTTP_Response(o, 200);
1747                     hres = p->u.HTTP_Response;
1748                     hres->content_buf = content_buf;
1749                     hres->content_len = content_size;
1750                     z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
1751                 }
1752                 yaz_mime_types_destroy(types);
1753             }
1754         }
1755         r = 1;
1756     }
1757
1758     if (r == 2)
1759     {
1760         r = yaz_srw_decode(hreq, &sr, &soap_package, assoc->decode, &charset);
1761         yaz_log(YLOG_DEBUG, "yaz_srw_decode returned %d", r);
1762     }
1763     if (r == 2)  /* not taken */
1764     {
1765         r = yaz_sru_decode(hreq, &sr, &soap_package, assoc->decode, &charset,
1766                            &diagnostic, &num_diagnostic);
1767         yaz_log(YLOG_DEBUG, "yaz_sru_decode returned %d", r);
1768     }
1769     if (r == 0)  /* decode SRW/SRU OK .. */
1770     {
1771         int http_code = 200;
1772         if (sr->which == Z_SRW_searchRetrieve_request)
1773         {
1774             Z_SRW_PDU *res =
1775                 yaz_srw_get_pdu(assoc->encode, Z_SRW_searchRetrieve_response,
1776                                 sr->srw_version);
1777             stylesheet = sr->u.request->stylesheet;
1778             if (num_diagnostic)
1779             {
1780                 res->u.response->diagnostics = diagnostic;
1781                 res->u.response->num_diagnostics = num_diagnostic;
1782             }
1783             else
1784             {
1785                 srw_bend_search(assoc, req, sr, res->u.response, 
1786                                 &http_code);
1787             }
1788             if (http_code == 200)
1789                 soap_package->u.generic->p = res;
1790         }
1791         else if (sr->which == Z_SRW_explain_request)
1792         {
1793             Z_SRW_PDU *res = yaz_srw_get_pdu(o, Z_SRW_explain_response,
1794                                              sr->srw_version);
1795             stylesheet = sr->u.explain_request->stylesheet;
1796             if (num_diagnostic)
1797             {   
1798                 res->u.explain_response->diagnostics = diagnostic;
1799                 res->u.explain_response->num_diagnostics = num_diagnostic;
1800             }
1801             srw_bend_explain(assoc, req, sr,
1802                              res->u.explain_response, &http_code);
1803             if (http_code == 200)
1804                 soap_package->u.generic->p = res;
1805         }
1806         else if (sr->which == Z_SRW_scan_request)
1807         {
1808             Z_SRW_PDU *res = yaz_srw_get_pdu(o, Z_SRW_scan_response,
1809                                              sr->srw_version);
1810             stylesheet = sr->u.scan_request->stylesheet;
1811             if (num_diagnostic)
1812             {   
1813                 res->u.scan_response->diagnostics = diagnostic;
1814                 res->u.scan_response->num_diagnostics = num_diagnostic;
1815             }
1816             srw_bend_scan(assoc, req, sr,
1817                           res->u.scan_response, &http_code);
1818             if (http_code == 200)
1819                 soap_package->u.generic->p = res;
1820         }
1821         else if (sr->which == Z_SRW_update_request)
1822         {
1823             Z_SRW_PDU *res = yaz_srw_get_pdu(o, Z_SRW_update_response,
1824                                              sr->srw_version);
1825             yaz_log(YLOG_DEBUG, "handling SRW UpdateRequest");
1826             if (num_diagnostic)
1827             {   
1828                 res->u.update_response->diagnostics = diagnostic;
1829                 res->u.update_response->num_diagnostics = num_diagnostic;
1830             }
1831             yaz_log(YLOG_DEBUG, "num_diag = %d", res->u.update_response->num_diagnostics );
1832             srw_bend_update(assoc, req, sr,
1833                             res->u.update_response, &http_code);
1834             if (http_code == 200)
1835                 soap_package->u.generic->p = res;
1836         }
1837         else
1838         {
1839             yaz_log(log_request, "SOAP ERROR"); 
1840             /* FIXME - what error, what query */
1841             http_code = 500;
1842             z_soap_error(assoc->encode, soap_package,
1843                          "SOAP-ENV:Client", "Bad method", 0); 
1844         }
1845         if (http_code == 200 || http_code == 500)
1846         {
1847             static Z_SOAP_Handler soap_handlers[4] = {
1848 #if YAZ_HAVE_XML2
1849                 {YAZ_XMLNS_SRU_v1_1, 0, (Z_SOAP_fun) yaz_srw_codec},
1850                 {YAZ_XMLNS_SRU_v1_0, 0, (Z_SOAP_fun) yaz_srw_codec},
1851                 {YAZ_XMLNS_UPDATE_v0_9, 0, (Z_SOAP_fun) yaz_ucp_codec},
1852 #endif
1853                 {0, 0, 0}
1854             };
1855             char ctype[80];
1856             int ret;
1857             p = z_get_HTTP_Response(o, 200);
1858             hres = p->u.HTTP_Response;
1859
1860             if (!stylesheet && assoc->server)
1861                 stylesheet = assoc->server->stylesheet;
1862
1863             /* empty stylesheet means NO stylesheet */
1864             if (stylesheet && *stylesheet == '\0')
1865                 stylesheet = 0;
1866
1867             ret = z_soap_codec_enc_xsl(assoc->encode, &soap_package,
1868                                        &hres->content_buf, &hres->content_len,
1869                                        soap_handlers, charset, stylesheet);
1870             hres->code = http_code;
1871
1872             strcpy(ctype, "text/xml");
1873             if (charset && strlen(charset) < sizeof(ctype)-30)
1874             {
1875                 strcat(ctype, "; charset=");
1876                 strcat(ctype, charset);
1877             }
1878             z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
1879         }
1880         else
1881             p = z_get_HTTP_Response(o, http_code);
1882     }
1883
1884     if (p == 0)
1885         p = z_get_HTTP_Response(o, 500);
1886     hres = p->u.HTTP_Response;
1887     if (!strcmp(hreq->version, "1.0")) 
1888     {
1889         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1890         if (v && !strcmp(v, "Keep-Alive"))
1891             keepalive = 1;
1892         else
1893             keepalive = 0;
1894         hres->version = "1.0";
1895     }
1896     else
1897     {
1898         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1899         if (v && !strcmp(v, "close"))
1900             keepalive = 0;
1901         else
1902             keepalive = 1;
1903         hres->version = "1.1";
1904     }
1905     if (!keepalive)
1906     {
1907         z_HTTP_header_add(o, &hres->headers, "Connection", "close");
1908         assoc->state = ASSOC_DEAD;
1909         assoc->cs_get_mask = 0;
1910     }
1911     else
1912     {
1913         int t;
1914         const char *alive = z_HTTP_header_lookup(hreq->headers, "Keep-Alive");
1915
1916         if (alive && isdigit(*(const unsigned char *) alive))
1917             t = atoi(alive);
1918         else
1919             t = 15;
1920         if (t < 0 || t > 3600)
1921             t = 3600;
1922         iochan_settimeout(assoc->client_chan,t);
1923         z_HTTP_header_add(o, &hres->headers, "Connection", "Keep-Alive");
1924     }
1925     process_gdu_response(assoc, req, p);
1926 }
1927
1928 static void process_gdu_request(association *assoc, request *req)
1929 {
1930     if (req->gdu_request->which == Z_GDU_Z3950)
1931     {
1932         char *msg = 0;
1933         req->apdu_request = req->gdu_request->u.z3950;
1934         if (process_z_request(assoc, req, &msg) < 0)
1935             do_close_req(assoc, Z_Close_systemProblem, msg, req);
1936     }
1937     else if (req->gdu_request->which == Z_GDU_HTTP_Request)
1938         process_http_request(assoc, req);
1939     else
1940     {
1941         do_close_req(assoc, Z_Close_systemProblem, "bad protocol packet", req);
1942     }
1943 }
1944
1945 /*
1946  * Initiate request processing.
1947  */
1948 static int process_z_request(association *assoc, request *req, char **msg)
1949 {
1950     int fd = -1;
1951     Z_APDU *res;
1952     int retval;
1953     
1954     *msg = "Unknown Error";
1955     assert(req && req->state == REQUEST_IDLE);
1956     if (req->apdu_request->which != Z_APDU_initRequest && !assoc->init)
1957     {
1958         *msg = "Missing InitRequest";
1959         return -1;
1960     }
1961     switch (req->apdu_request->which)
1962     {
1963     case Z_APDU_initRequest:
1964         res = process_initRequest(assoc, req); break;
1965     case Z_APDU_searchRequest:
1966         res = process_searchRequest(assoc, req, &fd); break;
1967     case Z_APDU_presentRequest:
1968         res = process_presentRequest(assoc, req, &fd); break;
1969     case Z_APDU_scanRequest:
1970         if (assoc->init->bend_scan)
1971             res = process_scanRequest(assoc, req, &fd);
1972         else
1973         {
1974             *msg = "Cannot handle Scan APDU";
1975             return -1;
1976         }
1977         break;
1978     case Z_APDU_extendedServicesRequest:
1979         if (assoc->init->bend_esrequest)
1980             res = process_ESRequest(assoc, req, &fd);
1981         else
1982         {
1983             *msg = "Cannot handle Extended Services APDU";
1984             return -1;
1985         }
1986         break;
1987     case Z_APDU_sortRequest:
1988         if (assoc->init->bend_sort)
1989             res = process_sortRequest(assoc, req, &fd);
1990         else
1991         {
1992             *msg = "Cannot handle Sort APDU";
1993             return -1;
1994         }
1995         break;
1996     case Z_APDU_close:
1997         process_close(assoc, req);
1998         return 0;
1999     case Z_APDU_deleteResultSetRequest:
2000         if (assoc->init->bend_delete)
2001             res = process_deleteRequest(assoc, req, &fd);
2002         else
2003         {
2004             *msg = "Cannot handle Delete APDU";
2005             return -1;
2006         }
2007         break;
2008     case Z_APDU_segmentRequest:
2009         if (assoc->init->bend_segment)
2010         {
2011             res = process_segmentRequest(assoc, req);
2012         }
2013         else
2014         {
2015             *msg = "Cannot handle Segment APDU";
2016             return -1;
2017         }
2018         break;
2019     case Z_APDU_triggerResourceControlRequest:
2020         return 0;
2021     default:
2022         *msg = "Bad APDU received";
2023         return -1;
2024     }
2025     if (res)
2026     {
2027         yaz_log(YLOG_DEBUG, "  result immediately available");
2028         retval = process_z_response(assoc, req, res);
2029     }
2030     else if (fd < 0)
2031     {
2032         yaz_log(YLOG_DEBUG, "  result unavailble");
2033         retval = 0;
2034     }
2035     else /* no result yet - one will be provided later */
2036     {
2037         IOCHAN chan;
2038
2039         /* Set up an I/O handler for the fd supplied by the backend */
2040
2041         yaz_log(YLOG_DEBUG, "   establishing handler for result");
2042         req->state = REQUEST_PENDING;
2043         if (!(chan = iochan_create(fd, backend_response, EVENT_INPUT, 0)))
2044             abort();
2045         iochan_setdata(chan, assoc);
2046         retval = 0;
2047     }
2048     return retval;
2049 }
2050
2051 /*
2052  * Handle message from the backend.
2053  */
2054 void backend_response(IOCHAN i, int event)
2055 {
2056     association *assoc = (association *)iochan_getdata(i);
2057     request *req = request_head(&assoc->incoming);
2058     Z_APDU *res;
2059     int fd;
2060
2061     yaz_log(YLOG_DEBUG, "backend_response");
2062     assert(assoc && req && req->state != REQUEST_IDLE);
2063     /* determine what it is we're waiting for */
2064     switch (req->apdu_request->which)
2065     {
2066         case Z_APDU_searchRequest:
2067             res = response_searchRequest(assoc, req, 0, &fd); break;
2068 #if 0
2069         case Z_APDU_presentRequest:
2070             res = response_presentRequest(assoc, req, 0, &fd); break;
2071         case Z_APDU_scanRequest:
2072             res = response_scanRequest(assoc, req, 0, &fd); break;
2073 #endif
2074         default:
2075             yaz_log(YLOG_FATAL, "Serious programmer's lapse or bug");
2076             abort();
2077     }
2078     if ((res && process_z_response(assoc, req, res) < 0) || fd < 0)
2079     {
2080         yaz_log(YLOG_WARN, "Fatal error when talking to backend");
2081         do_close(assoc, Z_Close_systemProblem, 0);
2082         iochan_destroy(i);
2083         return;
2084     }
2085     else if (!res) /* no result yet - try again later */
2086     {
2087         yaz_log(YLOG_DEBUG, "   no result yet");
2088         iochan_setfd(i, fd); /* in case fd has changed */
2089     }
2090 }
2091
2092 /*
2093  * Encode response, and transfer the request structure to the outgoing queue.
2094  */
2095 static int process_gdu_response(association *assoc, request *req, Z_GDU *res)
2096 {
2097     odr_setbuf(assoc->encode, req->response, req->size_response, 1);
2098
2099     if (assoc->print)
2100     {
2101         if (!z_GDU(assoc->print, &res, 0, 0))
2102             yaz_log(YLOG_WARN, "ODR print error: %s", 
2103                 odr_errmsg(odr_geterror(assoc->print)));
2104         odr_reset(assoc->print);
2105     }
2106     if (!z_GDU(assoc->encode, &res, 0, 0))
2107     {
2108         yaz_log(YLOG_WARN, "ODR error when encoding PDU: %s [element %s]",
2109                 odr_errmsg(odr_geterror(assoc->decode)),
2110                 odr_getelement(assoc->decode));
2111         return -1;
2112     }
2113     req->response = odr_getbuf(assoc->encode, &req->len_response,
2114         &req->size_response);
2115     odr_setbuf(assoc->encode, 0, 0, 0); /* don'txfree if we abort later */
2116     odr_reset(assoc->encode);
2117     req->state = REQUEST_IDLE;
2118     request_enq(&assoc->outgoing, req);
2119     /* turn the work over to the ir_session handler */
2120     iochan_setflag(assoc->client_chan, EVENT_OUTPUT);
2121     assoc->cs_put_mask = EVENT_OUTPUT;
2122     /* Is there more work to be done? give that to the input handler too */
2123     for (;;)
2124     {
2125         req = request_head(&assoc->incoming);
2126         if (req && req->state == REQUEST_IDLE)
2127         {
2128             request_deq(&assoc->incoming);
2129             process_gdu_request(assoc, req);
2130         }
2131         else
2132             break;
2133     }
2134     return 0;
2135 }
2136
2137 /*
2138  * Encode response, and transfer the request structure to the outgoing queue.
2139  */
2140 static int process_z_response(association *assoc, request *req, Z_APDU *res)
2141 {
2142     Z_GDU *gres = (Z_GDU *) odr_malloc(assoc->encode, sizeof(*res));
2143     gres->which = Z_GDU_Z3950;
2144     gres->u.z3950 = res;
2145
2146     return process_gdu_response(assoc, req, gres);
2147 }
2148
2149 static char *get_vhost(Z_OtherInformation *otherInfo)
2150 {
2151     return yaz_oi_get_string_oid(&otherInfo, yaz_oid_userinfo_proxy, 1, 0);
2152 }
2153
2154 /*
2155  * Handle init request.
2156  * At the moment, we don't check the options
2157  * anywhere else in the code - we just try not to do anything that would
2158  * break a naive client. We'll toss 'em into the association block when
2159  * we need them there.
2160  */
2161 static Z_APDU *process_initRequest(association *assoc, request *reqb)
2162 {
2163     Z_InitRequest *req = reqb->apdu_request->u.initRequest;
2164     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_initResponse);
2165     Z_InitResponse *resp = apdu->u.initResponse;
2166     bend_initresult *binitres;
2167     char options[140];
2168     statserv_options_block *cb = 0;  /* by default no control for backend */
2169
2170     if (control_association(assoc, get_vhost(req->otherInfo), 1))
2171         cb = statserv_getcontrol();  /* got control block for backend */
2172
2173     if (cb && assoc->backend)
2174         (*cb->bend_close)(assoc->backend);
2175
2176     yaz_log(log_requestdetail, "Got initRequest");
2177     if (req->implementationId)
2178         yaz_log(log_requestdetail, "Id:        %s",
2179                 req->implementationId);
2180     if (req->implementationName)
2181         yaz_log(log_requestdetail, "Name:      %s",
2182                 req->implementationName);
2183     if (req->implementationVersion)
2184         yaz_log(log_requestdetail, "Version:   %s",
2185                 req->implementationVersion);
2186     
2187     assoc_init_reset(assoc);
2188
2189     assoc->init->auth = req->idAuthentication;
2190     assoc->init->referenceId = req->referenceId;
2191
2192     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel))
2193     {
2194         Z_CharSetandLanguageNegotiation *negotiation =
2195             yaz_get_charneg_record (req->otherInfo);
2196         if (negotiation &&
2197             negotiation->which == Z_CharSetandLanguageNegotiation_proposal)
2198             assoc->init->charneg_request = negotiation;
2199     }
2200
2201     /* by default named_result_sets is 0 .. Enable it if client asks for it. */
2202     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
2203         assoc->init->named_result_sets = 1;
2204
2205     assoc->backend = 0;
2206     if (cb)
2207     {
2208         if (req->implementationVersion)
2209             yaz_log(log_requestdetail, "Config:    %s",
2210                     cb->configname);
2211     
2212         iochan_settimeout(assoc->client_chan, cb->idle_timeout * 60);
2213         
2214         /* we have a backend control block, so call that init function */
2215         if (!(binitres = (*cb->bend_init)(assoc->init)))
2216         {
2217             yaz_log(YLOG_WARN, "Bad response from backend.");
2218             return 0;
2219         }
2220         assoc->backend = binitres->handle;
2221     }
2222     else
2223     {
2224         /* no backend. return error */
2225         binitres = (bend_initresult *)
2226             odr_malloc(assoc->encode, sizeof(*binitres));
2227         binitres->errstring = 0;
2228         binitres->errcode = YAZ_BIB1_PERMANENT_SYSTEM_ERROR;
2229         iochan_settimeout(assoc->client_chan, 10);
2230     }
2231     if ((assoc->init->bend_sort))
2232         yaz_log(YLOG_DEBUG, "Sort handler installed");
2233     if ((assoc->init->bend_search))
2234         yaz_log(YLOG_DEBUG, "Search handler installed");
2235     if ((assoc->init->bend_present))
2236         yaz_log(YLOG_DEBUG, "Present handler installed");   
2237     if ((assoc->init->bend_esrequest))
2238         yaz_log(YLOG_DEBUG, "ESRequest handler installed");   
2239     if ((assoc->init->bend_delete))
2240         yaz_log(YLOG_DEBUG, "Delete handler installed");   
2241     if ((assoc->init->bend_scan))
2242         yaz_log(YLOG_DEBUG, "Scan handler installed");   
2243     if ((assoc->init->bend_segment))
2244         yaz_log(YLOG_DEBUG, "Segment handler installed");   
2245     
2246     resp->referenceId = req->referenceId;
2247     *options = '\0';
2248     /* let's tell the client what we can do */
2249     if (ODR_MASK_GET(req->options, Z_Options_search))
2250     {
2251         ODR_MASK_SET(resp->options, Z_Options_search);
2252         strcat(options, "srch");
2253     }
2254     if (ODR_MASK_GET(req->options, Z_Options_present))
2255     {
2256         ODR_MASK_SET(resp->options, Z_Options_present);
2257         strcat(options, " prst");
2258     }
2259     if (ODR_MASK_GET(req->options, Z_Options_delSet) &&
2260         assoc->init->bend_delete)
2261     {
2262         ODR_MASK_SET(resp->options, Z_Options_delSet);
2263         strcat(options, " del");
2264     }
2265     if (ODR_MASK_GET(req->options, Z_Options_extendedServices) &&
2266         assoc->init->bend_esrequest)
2267     {
2268         ODR_MASK_SET(resp->options, Z_Options_extendedServices);
2269         strcat(options, " extendedServices");
2270     }
2271     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets)
2272         && assoc->init->named_result_sets)
2273     {
2274         ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
2275         strcat(options, " namedresults");
2276     }
2277     if (ODR_MASK_GET(req->options, Z_Options_scan) && assoc->init->bend_scan)
2278     {
2279         ODR_MASK_SET(resp->options, Z_Options_scan);
2280         strcat(options, " scan");
2281     }
2282     if (ODR_MASK_GET(req->options, Z_Options_concurrentOperations))
2283     {
2284         ODR_MASK_SET(resp->options, Z_Options_concurrentOperations);
2285         strcat(options, " concurrop");
2286     }
2287     if (ODR_MASK_GET(req->options, Z_Options_sort) && assoc->init->bend_sort)
2288     {
2289         ODR_MASK_SET(resp->options, Z_Options_sort);
2290         strcat(options, " sort");
2291     }
2292     
2293     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel))
2294     {
2295         Z_OtherInformationUnit *p0;
2296
2297         if (!assoc->init->charneg_response)
2298         {
2299             if (assoc->init->query_charset)
2300             {
2301                 assoc->init->charneg_response = yaz_set_response_charneg(
2302                     assoc->encode, assoc->init->query_charset, 0, 
2303                     assoc->init->records_in_same_charset);
2304             }
2305             else
2306             {
2307                 yaz_log(YLOG_WARN, "default query_charset not defined by backend");
2308             }
2309         }
2310         if (assoc->init->charneg_response
2311             && (p0=yaz_oi_update(&resp->otherInfo, assoc->encode, NULL, 0, 0)))
2312         {
2313             p0->which = Z_OtherInfo_externallyDefinedInfo;
2314             p0->information.externallyDefinedInfo =
2315                 assoc->init->charneg_response;
2316             ODR_MASK_SET(resp->options, Z_Options_negotiationModel);
2317             strcat(options, " negotiation");
2318         }
2319     }
2320     if (ODR_MASK_GET(req->options, Z_Options_triggerResourceCtrl))
2321         ODR_MASK_SET(resp->options, Z_Options_triggerResourceCtrl);
2322
2323     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_1))
2324     {
2325         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_1);
2326         assoc->version = 1; /* 1 & 2 are equivalent */
2327     }
2328     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_2))
2329     {
2330         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_2);
2331         assoc->version = 2;
2332     }
2333     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_3))
2334     {
2335         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_3);
2336         assoc->version = 3;
2337     }
2338
2339     yaz_log(log_requestdetail, "Negotiated to v%d: %s", assoc->version, options);
2340
2341     if (*req->maximumRecordSize < assoc->maximumRecordSize)
2342         assoc->maximumRecordSize = *req->maximumRecordSize;
2343
2344     if (*req->preferredMessageSize < assoc->preferredMessageSize)
2345         assoc->preferredMessageSize = *req->preferredMessageSize;
2346
2347     resp->preferredMessageSize = &assoc->preferredMessageSize;
2348     resp->maximumRecordSize = &assoc->maximumRecordSize;
2349
2350     resp->implementationId = odr_prepend(assoc->encode,
2351                 assoc->init->implementation_id,
2352                 resp->implementationId);
2353
2354     resp->implementationName = odr_prepend(assoc->encode,
2355                 assoc->init->implementation_name,
2356                 odr_prepend(assoc->encode, "GFS", resp->implementationName));
2357
2358     if (binitres->errcode)
2359     {
2360         assoc->state = ASSOC_DEAD;
2361         resp->userInformationField =
2362             init_diagnostics(assoc->encode, binitres->errcode,
2363                              binitres->errstring);
2364         *resp->result = 0;
2365     }
2366     else
2367         assoc->state = ASSOC_UP;
2368     
2369     if (log_request)
2370     {
2371         if (!req->idAuthentication)
2372             yaz_log(log_request, "Auth none");
2373         else if (req->idAuthentication->which == Z_IdAuthentication_open)
2374         {
2375             const char *open = req->idAuthentication->u.open;
2376             const char *slash = strchr(open, '/');
2377             int len;
2378             if (slash)
2379                 len = slash - open;
2380             else
2381                 len = strlen(open);
2382                 yaz_log(log_request, "Auth open %.*s", len, open);
2383         }
2384         else if (req->idAuthentication->which == Z_IdAuthentication_idPass)
2385         {
2386             const char *user = req->idAuthentication->u.idPass->userId;
2387             const char *group = req->idAuthentication->u.idPass->groupId;
2388             yaz_log(log_request, "Auth idPass %s %s",
2389                     user ? user : "-", group ? group : "-");
2390         }
2391         else if (req->idAuthentication->which 
2392                  == Z_IdAuthentication_anonymous)
2393         {
2394             yaz_log(log_request, "Auth anonymous");
2395         }
2396         else
2397         {
2398             yaz_log(log_request, "Auth other");
2399         }
2400     }
2401     if (log_request)
2402     {
2403         WRBUF wr = wrbuf_alloc();
2404         wrbuf_printf(wr, "Init ");
2405         if (binitres->errcode)
2406             wrbuf_printf(wr, "ERROR %d", binitres->errcode);
2407         else
2408             wrbuf_printf(wr, "OK -");
2409         wrbuf_printf(wr, " ID:%s Name:%s Version:%s",
2410                      (req->implementationId ? req->implementationId :"-"), 
2411                      (req->implementationName ?
2412                       req->implementationName : "-"),
2413                      (req->implementationVersion ?
2414                       req->implementationVersion : "-")
2415             );
2416         yaz_log(log_request, "%s", wrbuf_cstr(wr));
2417         wrbuf_destroy(wr);
2418     }
2419     return apdu;
2420 }
2421
2422 /*
2423  * Set the specified `errcode' and `errstring' into a UserInfo-1
2424  * external to be returned to the client in accordance with Z35.90
2425  * Implementor Agreement 5 (Returning diagnostics in an InitResponse):
2426  *      http://lcweb.loc.gov/z3950/agency/agree/initdiag.html
2427  */
2428 static Z_External *init_diagnostics(ODR odr, int error, const char *addinfo)
2429 {
2430     yaz_log(log_requestdetail, "[%d] %s%s%s", error, diagbib1_str(error),
2431         addinfo ? " -- " : "", addinfo ? addinfo : "");
2432     return zget_init_diagnostics(odr, error, addinfo);
2433 }
2434
2435 /*
2436  * nonsurrogate diagnostic record.
2437  */
2438 static Z_Records *diagrec(association *assoc, int error, char *addinfo)
2439 {
2440     Z_Records *rec = (Z_Records *) odr_malloc(assoc->encode, sizeof(*rec));
2441
2442     yaz_log(log_requestdetail, "[%d] %s%s%s", error, diagbib1_str(error),
2443             addinfo ? " -- " : "", addinfo ? addinfo : "");
2444
2445     rec->which = Z_Records_NSD;
2446     rec->u.nonSurrogateDiagnostic = zget_DefaultDiagFormat(assoc->encode,
2447                                                            error, addinfo);
2448     return rec;
2449 }
2450
2451 /*
2452  * surrogate diagnostic.
2453  */
2454 static Z_NamePlusRecord *surrogatediagrec(association *assoc, 
2455                                           const char *dbname,
2456                                           int error, const char *addinfo)
2457 {
2458     yaz_log(log_requestdetail, "[%d] %s%s%s", error, diagbib1_str(error),
2459             addinfo ? " -- " : "", addinfo ? addinfo : "");
2460     return zget_surrogateDiagRec(assoc->encode, dbname, error, addinfo);
2461 }
2462
2463 static Z_Records *pack_records(association *a, char *setname, Odr_int start,
2464                                Odr_int *num, Z_RecordComposition *comp,
2465                                Odr_int *next, Odr_int *pres,
2466                                Z_ReferenceId *referenceId,
2467                                Odr_oid *oid, int *errcode)
2468 {
2469     int recno, total_length = 0, toget = *num, dumped_records = 0;
2470     Z_Records *records =
2471         (Z_Records *) odr_malloc(a->encode, sizeof(*records));
2472     Z_NamePlusRecordList *reclist =
2473         (Z_NamePlusRecordList *) odr_malloc(a->encode, sizeof(*reclist));
2474     Z_NamePlusRecord **list =
2475         (Z_NamePlusRecord **) odr_malloc(a->encode, sizeof(*list) * toget);
2476
2477     records->which = Z_Records_DBOSD;
2478     records->u.databaseOrSurDiagnostics = reclist;
2479     reclist->num_records = 0;
2480     reclist->records = list;
2481     *pres = Z_PresentStatus_success;
2482     *num = 0;
2483     *next = 0;
2484
2485     yaz_log(log_requestdetail, "Request to pack " ODR_INT_PRINTF "+%d %s", start, toget, setname);
2486     yaz_log(log_requestdetail, "pms=" ODR_INT_PRINTF
2487             ", mrs=" ODR_INT_PRINTF, a->preferredMessageSize,
2488         a->maximumRecordSize);
2489     for (recno = start; reclist->num_records < toget; recno++)
2490     {
2491         bend_fetch_rr freq;
2492         Z_NamePlusRecord *thisrec;
2493         int this_length = 0;
2494         /*
2495          * we get the number of bytes allocated on the stream before any
2496          * allocation done by the backend - this should give us a reasonable
2497          * idea of the total size of the data so far.
2498          */
2499         total_length = odr_total(a->encode) - dumped_records;
2500         freq.errcode = 0;
2501         freq.errstring = 0;
2502         freq.basename = 0;
2503         freq.len = 0;
2504         freq.record = 0;
2505         freq.last_in_set = 0;
2506         freq.setname = setname;
2507         freq.surrogate_flag = 0;
2508         freq.number = recno;
2509         freq.comp = comp;
2510         freq.request_format = oid;
2511         freq.output_format = 0;
2512         freq.stream = a->encode;
2513         freq.print = a->print;
2514         freq.referenceId = referenceId;
2515         freq.schema = 0;
2516
2517         retrieve_fetch(a, &freq);
2518
2519         *next = freq.last_in_set ? 0 : recno + 1;
2520
2521         if (freq.errcode)
2522         {
2523             if (!freq.surrogate_flag) /* non-surrogate diagnostic i.e. global */
2524             {
2525                 char s[20];
2526                 *pres = Z_PresentStatus_failure;
2527                 /* for 'present request out of range',
2528                    set addinfo to record position if not set */
2529                 if (freq.errcode == YAZ_BIB1_PRESENT_REQUEST_OUT_OF_RANGE  && 
2530                                 freq.errstring == 0)
2531                 {
2532                     sprintf (s, "%d", recno);
2533                     freq.errstring = s;
2534                 }
2535                 if (errcode)
2536                     *errcode = freq.errcode;
2537                 return diagrec(a, freq.errcode, freq.errstring);
2538             }
2539             reclist->records[reclist->num_records] =
2540                 surrogatediagrec(a, freq.basename, freq.errcode,
2541                                  freq.errstring);
2542             reclist->num_records++;
2543             continue;
2544         }
2545         if (freq.record == 0)  /* no error and no record ? */
2546         {
2547             *next = 0;   /* signal end-of-set and stop */
2548             break;
2549         }
2550         if (freq.len >= 0)
2551             this_length = freq.len;
2552         else
2553             this_length = odr_total(a->encode) - total_length - dumped_records;
2554         yaz_log(YLOG_DEBUG, "  fetched record, len=%d, total=%d dumped=%d",
2555             this_length, total_length, dumped_records);
2556         if (a->preferredMessageSize > 0 &&
2557                 this_length + total_length > a->preferredMessageSize)
2558         {
2559             /* record is small enough, really */
2560             if (this_length <= a->preferredMessageSize && recno > start)
2561             {
2562                 yaz_log(log_requestdetail, "  Dropped last normal-sized record");
2563                 *pres = Z_PresentStatus_partial_2;
2564                 break;
2565             }
2566             /* record can only be fetched by itself */
2567             if (this_length < a->maximumRecordSize)
2568             {
2569                 yaz_log(log_requestdetail, "  Record > prefmsgsz");
2570                 if (toget > 1)
2571                 {
2572                     yaz_log(YLOG_DEBUG, "  Dropped it");
2573                     reclist->records[reclist->num_records] =
2574                          surrogatediagrec(a, freq.basename, 16, 0);
2575                     reclist->num_records++;
2576                     dumped_records += this_length;
2577                     continue;
2578                 }
2579             }
2580             else /* too big entirely */
2581             {
2582                 yaz_log(log_requestdetail, "Record > maxrcdsz "
2583                         "this=%d max=" ODR_INT_PRINTF,
2584                         this_length, a->maximumRecordSize);
2585                 reclist->records[reclist->num_records] =
2586                     surrogatediagrec(a, freq.basename, 17, 0);
2587                 reclist->num_records++;
2588                 dumped_records += this_length;
2589                 continue;
2590             }
2591         }
2592
2593         if (!(thisrec = (Z_NamePlusRecord *)
2594               odr_malloc(a->encode, sizeof(*thisrec))))
2595             return 0;
2596         thisrec->databaseName = odr_strdup_null(a->encode, freq.basename);
2597         thisrec->which = Z_NamePlusRecord_databaseRecord;
2598
2599         if (!freq.output_format)
2600             freq.output_format = freq.request_format;
2601         thisrec->u.databaseRecord = z_ext_record_oid(
2602             a->encode, freq.output_format, freq.record, freq.len);
2603         if (!thisrec->u.databaseRecord)
2604             return 0;
2605         reclist->records[reclist->num_records] = thisrec;
2606         reclist->num_records++;
2607     }
2608     *num = reclist->num_records;
2609     return records;
2610 }
2611
2612 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
2613     int *fd)
2614 {
2615     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
2616     bend_search_rr *bsrr = 
2617         (bend_search_rr *)nmem_malloc(reqb->request_mem, sizeof(*bsrr));
2618     
2619     yaz_log(log_requestdetail, "Got SearchRequest.");
2620     bsrr->fd = fd;
2621     bsrr->request = reqb;
2622     bsrr->association = assoc;
2623     bsrr->referenceId = req->referenceId;
2624     save_referenceId (reqb, bsrr->referenceId);
2625     bsrr->srw_sortKeys = 0;
2626     bsrr->srw_setname = 0;
2627     bsrr->srw_setnameIdleTime = 0;
2628     bsrr->estimated_hit_count = 0;
2629     bsrr->partial_resultset = 0;
2630
2631     yaz_log (log_requestdetail, "ResultSet '%s'", req->resultSetName);
2632     if (req->databaseNames)
2633     {
2634         int i;
2635         for (i = 0; i < req->num_databaseNames; i++)
2636             yaz_log(log_requestdetail, "Database '%s'", req->databaseNames[i]);
2637     }
2638
2639     yaz_log_zquery_level(log_requestdetail,req->query);
2640
2641     if (assoc->init->bend_search)
2642     {
2643         bsrr->setname = req->resultSetName;
2644         bsrr->replace_set = *req->replaceIndicator;
2645         bsrr->num_bases = req->num_databaseNames;
2646         bsrr->basenames = req->databaseNames;
2647         bsrr->query = req->query;
2648         bsrr->stream = assoc->encode;
2649         nmem_transfer(odr_getmem(bsrr->stream), reqb->request_mem);
2650         bsrr->decode = assoc->decode;
2651         bsrr->print = assoc->print;
2652         bsrr->hits = 0;
2653         bsrr->errcode = 0;
2654         bsrr->errstring = NULL;
2655         bsrr->search_info = NULL;
2656
2657         if (assoc->server && assoc->server->cql_transform 
2658             && req->query->which == Z_Query_type_104
2659             && req->query->u.type_104->which == Z_External_CQL)
2660         {
2661             /* have a CQL query and a CQL to PQF transform .. */
2662             int srw_errcode = 
2663                 cql2pqf(bsrr->stream, req->query->u.type_104->u.cql,
2664                         assoc->server->cql_transform, bsrr->query);
2665             if (srw_errcode)
2666                 bsrr->errcode = yaz_diag_srw_to_bib1(srw_errcode);
2667         }
2668
2669         if (assoc->server && assoc->server->ccl_transform 
2670             && req->query->which == Z_Query_type_2) /*CCL*/
2671         {
2672             /* have a CCL query and a CCL to PQF transform .. */
2673             int srw_errcode = 
2674                 ccl2pqf(bsrr->stream, req->query->u.type_2,
2675                         assoc->server->ccl_transform, bsrr);
2676             if (srw_errcode)
2677                 bsrr->errcode = yaz_diag_srw_to_bib1(srw_errcode);
2678         }
2679
2680         if (!bsrr->errcode)
2681             (assoc->init->bend_search)(assoc->backend, bsrr);
2682         if (!bsrr->request)  /* backend not ready with the search response */
2683             return 0;  /* should not be used any more */
2684     }
2685     else
2686     { 
2687         /* FIXME - make a diagnostic for it */
2688         yaz_log(YLOG_WARN,"Search not supported ?!?!");
2689     }
2690     return response_searchRequest(assoc, reqb, bsrr, fd);
2691 }
2692
2693 int bend_searchresponse(void *handle, bend_search_rr *bsrr) {return 0;}
2694
2695 /*
2696  * Prepare a searchresponse based on the backend results. We probably want
2697  * to look at making the fetching of records nonblocking as well, but
2698  * so far, we'll keep things simple.
2699  * If bsrt is null, that means we're called in response to a communications
2700  * event, and we'll have to get the response for ourselves.
2701  */
2702 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
2703     bend_search_rr *bsrt, int *fd)
2704 {
2705     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
2706     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
2707     Z_SearchResponse *resp = (Z_SearchResponse *)
2708         odr_malloc(assoc->encode, sizeof(*resp));
2709     Odr_int *nulint = odr_intdup(assoc->encode, 0);
2710     Odr_int *next = odr_intdup(assoc->encode, 0);
2711     Odr_int *none = odr_intdup(assoc->encode, Z_SearchResponse_none);
2712     int returnedrecs = 0;
2713
2714     apdu->which = Z_APDU_searchResponse;
2715     apdu->u.searchResponse = resp;
2716     resp->referenceId = req->referenceId;
2717     resp->additionalSearchInfo = 0;
2718     resp->otherInfo = 0;
2719     *fd = -1;
2720     if (!bsrt && !bend_searchresponse(assoc->backend, bsrt))
2721     {
2722         yaz_log(YLOG_FATAL, "Bad result from backend");
2723         return 0;
2724     }
2725     else if (bsrt->errcode)
2726     {
2727         resp->records = diagrec(assoc, bsrt->errcode, bsrt->errstring);
2728         resp->resultCount = nulint;
2729         resp->numberOfRecordsReturned = nulint;
2730         resp->nextResultSetPosition = nulint;
2731         resp->searchStatus = odr_booldup(assoc->encode, 0);
2732         resp->resultSetStatus = none;
2733         resp->presentStatus = 0;
2734     }
2735     else
2736     {
2737         bool_t *sr = odr_booldup(assoc->encode, 1);
2738         Odr_int *toget = odr_intdup(assoc->encode, 0);
2739         Z_RecordComposition comp, *compp = 0;
2740
2741         yaz_log(log_requestdetail, "resultCount: " ODR_INT_PRINTF, bsrt->hits);
2742
2743         resp->records = 0;
2744         resp->resultCount = &bsrt->hits;
2745
2746         comp.which = Z_RecordComp_simple;
2747         /* how many records does the user agent want, then? */
2748         if (bsrt->hits <= *req->smallSetUpperBound)
2749         {
2750             *toget = bsrt->hits;
2751             if ((comp.u.simple = req->smallSetElementSetNames))
2752                 compp = &comp;
2753         }
2754         else if (bsrt->hits < *req->largeSetLowerBound)
2755         {
2756             *toget = *req->mediumSetPresentNumber;
2757             if (*toget > bsrt->hits)
2758                 *toget = bsrt->hits;
2759             if ((comp.u.simple = req->mediumSetElementSetNames))
2760                 compp = &comp;
2761         }
2762         else
2763             *toget = 0;
2764
2765         if (*toget && !resp->records)
2766         {
2767             Odr_int *presst = odr_intdup(assoc->encode, 0);
2768             /* Call bend_present if defined */
2769             if (assoc->init->bend_present)
2770             {
2771                 bend_present_rr *bprr = (bend_present_rr *)
2772                     nmem_malloc(reqb->request_mem, sizeof(*bprr));
2773                 bprr->setname = req->resultSetName;
2774                 bprr->start = 1;
2775                 bprr->number = *toget;
2776                 bprr->format = req->preferredRecordSyntax;
2777                 bprr->comp = compp;
2778                 bprr->referenceId = req->referenceId;
2779                 bprr->stream = assoc->encode;
2780                 bprr->print = assoc->print;
2781                 bprr->request = reqb;
2782                 bprr->association = assoc;
2783                 bprr->errcode = 0;
2784                 bprr->errstring = NULL;
2785                 (*assoc->init->bend_present)(assoc->backend, bprr);
2786
2787                 if (!bprr->request)
2788                     return 0;
2789                 if (bprr->errcode)
2790                 {
2791                     resp->records = diagrec(assoc, bprr->errcode, bprr->errstring);
2792                     *resp->presentStatus = Z_PresentStatus_failure;
2793                 }
2794             }
2795
2796             if (!resp->records)
2797                 resp->records = pack_records(
2798                     assoc, req->resultSetName, 1,
2799                     toget, compp, next, presst, req->referenceId,
2800                     req->preferredRecordSyntax, NULL);
2801             if (!resp->records)
2802                 return 0;
2803             resp->numberOfRecordsReturned = toget;
2804             returnedrecs = *toget;
2805             resp->presentStatus = presst;
2806         }
2807         else
2808         {
2809             if (*resp->resultCount)
2810                 *next = 1;
2811             resp->numberOfRecordsReturned = nulint;
2812             resp->presentStatus = 0;
2813         }
2814         resp->nextResultSetPosition = next;
2815         resp->searchStatus = sr;
2816         resp->resultSetStatus = 0;
2817         if (bsrt->estimated_hit_count)
2818         {
2819             resp->resultSetStatus = odr_intdup(assoc->encode, 
2820                                                Z_SearchResponse_estimate);
2821         }
2822         else if (bsrt->partial_resultset)
2823         {
2824             resp->resultSetStatus = odr_intdup(assoc->encode, 
2825                                                Z_SearchResponse_subset);
2826         }
2827     }
2828     resp->additionalSearchInfo = bsrt->search_info;
2829
2830     if (log_request)
2831     {
2832         int i;
2833         WRBUF wr = wrbuf_alloc();
2834
2835         for (i = 0 ; i < req->num_databaseNames; i++){
2836             if (i)
2837                 wrbuf_printf(wr, "+");
2838             wrbuf_puts(wr, req->databaseNames[i]);
2839         }
2840         wrbuf_printf(wr, " ");
2841         
2842         if (bsrt->errcode)
2843             wrbuf_printf(wr, "ERROR %d", bsrt->errcode);
2844         else
2845             wrbuf_printf(wr, "OK " ODR_INT_PRINTF, bsrt->hits);
2846         wrbuf_printf(wr, " %s 1+%d ",
2847                      req->resultSetName, returnedrecs);
2848         yaz_query_to_wrbuf(wr, req->query);
2849         
2850         yaz_log(log_request, "Search %s", wrbuf_cstr(wr));
2851         wrbuf_destroy(wr);
2852     }
2853     return apdu;
2854 }
2855
2856 /*
2857  * Maybe we got a little over-friendly when we designed bend_fetch to
2858  * get only one record at a time. Some backends can optimise multiple-record
2859  * fetches, and at any rate, there is some overhead involved in
2860  * all that selecting and hopping around. Problem is, of course, that the
2861  * frontend can't know ahead of time how many records it'll need to
2862  * fill the negotiated PDU size. Annoying. Segmentation or not, Z/SR
2863  * is downright lousy as a bulk data transfer protocol.
2864  *
2865  * To start with, we'll do the fetching of records from the backend
2866  * in one operation: To save some trips in and out of the event-handler,
2867  * and to simplify the interface to pack_records. At any rate, asynch
2868  * operation is more fun in operations that have an unpredictable execution
2869  * speed - which is normally more true for search than for present.
2870  */
2871 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
2872                                       int *fd)
2873 {
2874     Z_PresentRequest *req = reqb->apdu_request->u.presentRequest;
2875     Z_APDU *apdu;
2876     Z_PresentResponse *resp;
2877     Odr_int *next;
2878     Odr_int *num;
2879     int errcode = 0;
2880     const char *errstring = 0;
2881
2882     yaz_log(log_requestdetail, "Got PresentRequest.");
2883
2884     resp = (Z_PresentResponse *)odr_malloc(assoc->encode, sizeof(*resp));
2885     resp->records = 0;
2886     resp->presentStatus = odr_intdup(assoc->encode, 0);
2887     if (assoc->init->bend_present)
2888     {
2889         bend_present_rr *bprr = (bend_present_rr *)
2890             nmem_malloc(reqb->request_mem, sizeof(*bprr));
2891         bprr->setname = req->resultSetId;
2892         bprr->start = *req->resultSetStartPoint;
2893         bprr->number = *req->numberOfRecordsRequested;
2894         bprr->format = req->preferredRecordSyntax;
2895         bprr->comp = req->recordComposition;
2896         bprr->referenceId = req->referenceId;
2897         bprr->stream = assoc->encode;
2898         bprr->print = assoc->print;
2899         bprr->request = reqb;
2900         bprr->association = assoc;
2901         bprr->errcode = 0;
2902         bprr->errstring = NULL;
2903         (*assoc->init->bend_present)(assoc->backend, bprr);
2904         
2905         if (!bprr->request)
2906             return 0; /* should not happen */
2907         if (bprr->errcode)
2908         {
2909             resp->records = diagrec(assoc, bprr->errcode, bprr->errstring);
2910             *resp->presentStatus = Z_PresentStatus_failure;
2911             errcode = bprr->errcode;
2912             errstring = bprr->errstring;
2913         }
2914     }
2915     apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
2916     next = odr_intdup(assoc->encode, 0);
2917     num = odr_intdup(assoc->encode, 0);
2918     
2919     apdu->which = Z_APDU_presentResponse;
2920     apdu->u.presentResponse = resp;
2921     resp->referenceId = req->referenceId;
2922     resp->otherInfo = 0;
2923     
2924     if (!resp->records)
2925     {
2926         *num = *req->numberOfRecordsRequested;
2927         resp->records =
2928             pack_records(assoc, req->resultSetId, *req->resultSetStartPoint,
2929                          num, req->recordComposition, next,
2930                          resp->presentStatus,
2931                          req->referenceId, req->preferredRecordSyntax, 
2932                          &errcode);
2933     }
2934     if (log_request)
2935     {
2936         WRBUF wr = wrbuf_alloc();
2937         wrbuf_printf(wr, "Present ");
2938
2939         if (*resp->presentStatus == Z_PresentStatus_failure)
2940             wrbuf_printf(wr, "ERROR %d ", errcode);
2941         else if (*resp->presentStatus == Z_PresentStatus_success)
2942             wrbuf_printf(wr, "OK -  ");
2943         else
2944             wrbuf_printf(wr, "Partial " ODR_INT_PRINTF " - ",
2945                          *resp->presentStatus);
2946
2947         wrbuf_printf(wr, " %s " ODR_INT_PRINTF "+" ODR_INT_PRINTF " ",
2948                 req->resultSetId, *req->resultSetStartPoint,
2949                 *req->numberOfRecordsRequested);
2950         yaz_log(log_request, "%s", wrbuf_cstr(wr) );
2951         wrbuf_destroy(wr);
2952     }
2953     if (!resp->records)
2954         return 0;
2955     resp->numberOfRecordsReturned = num;
2956     resp->nextResultSetPosition = next;
2957     
2958     return apdu;
2959 }
2960
2961 /*
2962  * Scan was implemented rather in a hurry, and with support for only the basic
2963  * elements of the service in the backend API. Suggestions are welcome.
2964  */
2965 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd)
2966 {
2967     Z_ScanRequest *req = reqb->apdu_request->u.scanRequest;
2968     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
2969     Z_ScanResponse *res = (Z_ScanResponse *)
2970         odr_malloc(assoc->encode, sizeof(*res));
2971     Odr_int *scanStatus = odr_intdup(assoc->encode, Z_Scan_failure);
2972     Odr_int *numberOfEntriesReturned = odr_intdup(assoc->encode, 0);
2973     Z_ListEntries *ents = (Z_ListEntries *)
2974         odr_malloc(assoc->encode, sizeof(*ents));
2975     Z_DiagRecs *diagrecs_p = NULL;
2976     bend_scan_rr *bsrr = (bend_scan_rr *)
2977         odr_malloc(assoc->encode, sizeof(*bsrr));
2978     struct scan_entry *save_entries;
2979
2980     yaz_log(log_requestdetail, "Got ScanRequest");
2981
2982     apdu->which = Z_APDU_scanResponse;
2983     apdu->u.scanResponse = res;
2984     res->referenceId = req->referenceId;
2985
2986     /* if step is absent, set it to 0 */
2987     res->stepSize = odr_intdup(assoc->encode, 0);
2988     if (req->stepSize)
2989         *res->stepSize = *req->stepSize;
2990
2991     res->scanStatus = scanStatus;
2992     res->numberOfEntriesReturned = numberOfEntriesReturned;
2993     res->positionOfTerm = 0;
2994     res->entries = ents;
2995     ents->num_entries = 0;
2996     ents->entries = NULL;
2997     ents->num_nonsurrogateDiagnostics = 0;
2998     ents->nonsurrogateDiagnostics = NULL;
2999     res->attributeSet = 0;
3000     res->otherInfo = 0;
3001
3002     if (req->databaseNames)
3003     {
3004         int i;
3005         for (i = 0; i < req->num_databaseNames; i++)
3006             yaz_log(log_requestdetail, "Database '%s'", req->databaseNames[i]);
3007     }
3008     bsrr->scanClause = 0;
3009     bsrr->errcode = 0;
3010     bsrr->errstring = 0;
3011     bsrr->num_bases = req->num_databaseNames;
3012     bsrr->basenames = req->databaseNames;
3013     bsrr->num_entries = *req->numberOfTermsRequested;
3014     bsrr->term = req->termListAndStartPoint;
3015     bsrr->referenceId = req->referenceId;
3016     bsrr->stream = assoc->encode;
3017     bsrr->print = assoc->print;
3018     bsrr->step_size = res->stepSize;
3019     bsrr->setname = yaz_oi_get_string_oid(&req->otherInfo, 
3020                                           yaz_oid_userinfo_scan_set, 1, 0);
3021     bsrr->entries = 0;
3022     /* For YAZ 2.0 and earlier it was the backend handler that
3023        initialized entries (member display_term did not exist)
3024        YAZ 2.0 and later sets 'entries'  and initialize all members
3025        including 'display_term'. If YAZ 2.0 or later sees that
3026        entries was modified - we assume that it is an old handler and
3027        that 'display_term' is _not_ set.
3028     */
3029     if (bsrr->num_entries > 0) 
3030     {
3031         int i;
3032         bsrr->entries = (struct scan_entry *)
3033             odr_malloc(assoc->decode, sizeof(*bsrr->entries) *
3034                        bsrr->num_entries);
3035         for (i = 0; i<bsrr->num_entries; i++)
3036         {
3037             bsrr->entries[i].term = 0;
3038             bsrr->entries[i].occurrences = 0;
3039             bsrr->entries[i].errcode = 0;
3040             bsrr->entries[i].errstring = 0;
3041             bsrr->entries[i].display_term = 0;
3042         }
3043     }
3044     save_entries = bsrr->entries;  /* save it so we can compare later */
3045
3046     bsrr->attributeset = req->attributeSet;
3047     log_scan_term_level(log_requestdetail, req->termListAndStartPoint, 
3048                         bsrr->attributeset);
3049     bsrr->term_position = req->preferredPositionInResponse ?
3050         *req->preferredPositionInResponse : 1;
3051
3052     ((int (*)(void *, bend_scan_rr *))
3053      (*assoc->init->bend_scan))(assoc->backend, bsrr);
3054
3055     if (bsrr->errcode)
3056         diagrecs_p = zget_DiagRecs(assoc->encode,
3057                                    bsrr->errcode, bsrr->errstring);
3058     else
3059     {
3060         int i;
3061         Z_Entry **tab = (Z_Entry **)
3062             odr_malloc(assoc->encode, sizeof(*tab) * bsrr->num_entries);
3063         
3064         if (bsrr->status == BEND_SCAN_PARTIAL)
3065             *scanStatus = Z_Scan_partial_5;
3066         else
3067             *scanStatus = Z_Scan_success;
3068         ents->entries = tab;
3069         ents->num_entries = bsrr->num_entries;
3070         res->numberOfEntriesReturned = odr_intdup(assoc->encode, 
3071                                                    ents->num_entries);
3072         res->positionOfTerm = &bsrr->term_position;
3073         for (i = 0; i < bsrr->num_entries; i++)
3074         {
3075             Z_Entry *e;
3076             Z_TermInfo *t;
3077             Odr_oct *o;
3078             
3079             tab[i] = e = (Z_Entry *)odr_malloc(assoc->encode, sizeof(*e));
3080             if (bsrr->entries[i].occurrences >= 0)
3081             {
3082                 e->which = Z_Entry_termInfo;
3083                 e->u.termInfo = t = (Z_TermInfo *)
3084                     odr_malloc(assoc->encode, sizeof(*t));
3085                 t->suggestedAttributes = 0;
3086                 t->displayTerm = 0;
3087                 if (save_entries == bsrr->entries && 
3088                     bsrr->entries[i].display_term)
3089                 {
3090                     /* the entries was _not_ set by the handler. So it's
3091                        safe to test for new member display_term. It is
3092                        NULL'ed by us.
3093                     */
3094                     t->displayTerm = odr_strdup(assoc->encode,
3095                                                 bsrr->entries[i].display_term);
3096                 }
3097                 t->alternativeTerm = 0;
3098                 t->byAttributes = 0;
3099                 t->otherTermInfo = 0;
3100                 t->globalOccurrences = &bsrr->entries[i].occurrences;
3101                 t->term = (Z_Term *)
3102                     odr_malloc(assoc->encode, sizeof(*t->term));
3103                 t->term->which = Z_Term_general;
3104                 t->term->u.general = o =
3105                     (Odr_oct *)odr_malloc(assoc->encode, sizeof(Odr_oct));
3106                 o->buf = (unsigned char *)
3107                     odr_malloc(assoc->encode, o->len = o->size =
3108                                strlen(bsrr->entries[i].term));
3109                 memcpy(o->buf, bsrr->entries[i].term, o->len);
3110                 yaz_log(YLOG_DEBUG, "  term #%d: '%s' (" ODR_INT_PRINTF ")", i,
3111                          bsrr->entries[i].term, bsrr->entries[i].occurrences);
3112             }
3113             else
3114             {
3115                 Z_DiagRecs *drecs = zget_DiagRecs(assoc->encode,
3116                                                   bsrr->entries[i].errcode,
3117                                                   bsrr->entries[i].errstring);
3118                 assert(drecs->num_diagRecs == 1);
3119                 e->which = Z_Entry_surrogateDiagnostic;
3120                 assert(drecs->diagRecs[0]);
3121                 e->u.surrogateDiagnostic = drecs->diagRecs[0];
3122             }
3123         }
3124     }
3125     if (diagrecs_p)
3126     {
3127         ents->num_nonsurrogateDiagnostics = diagrecs_p->num_diagRecs;
3128         ents->nonsurrogateDiagnostics = diagrecs_p->diagRecs;
3129     }
3130     if (log_request)
3131     {
3132         int i;
3133         WRBUF wr = wrbuf_alloc();
3134         wrbuf_printf(wr, "Scan ");
3135         for (i = 0 ; i < req->num_databaseNames; i++)
3136         {
3137             if (i)
3138                 wrbuf_printf(wr, "+");
3139             wrbuf_puts(wr, req->databaseNames[i]);
3140         }
3141
3142         wrbuf_printf(wr, " ");
3143         
3144         if (bsrr->errcode)
3145             wr_diag(wr, bsrr->errcode, bsrr->errstring);
3146         else
3147             wrbuf_printf(wr, "OK"); 
3148
3149         wrbuf_printf(wr, " " ODR_INT_PRINTF " - " ODR_INT_PRINTF "+" 
3150                      ODR_INT_PRINTF "+" ODR_INT_PRINTF,
3151                      res->numberOfEntriesReturned ?
3152                      *res->numberOfEntriesReturned : 0,
3153                      (req->preferredPositionInResponse ?
3154                       *req->preferredPositionInResponse : 1),
3155                      *req->numberOfTermsRequested,
3156                      (res->stepSize ? *res->stepSize : 1));
3157         
3158         if (bsrr->setname)
3159             wrbuf_printf(wr, "+%s", bsrr->setname);
3160
3161         wrbuf_printf(wr, " ");
3162         yaz_scan_to_wrbuf(wr, req->termListAndStartPoint, 
3163                           bsrr->attributeset);
3164         yaz_log(log_request, "%s", wrbuf_cstr(wr) );
3165         wrbuf_destroy(wr);
3166     }
3167     return apdu;
3168 }
3169
3170 static Z_APDU *process_sortRequest(association *assoc, request *reqb,
3171     int *fd)
3172 {
3173     int i;
3174     Z_SortRequest *req = reqb->apdu_request->u.sortRequest;
3175     Z_SortResponse *res = (Z_SortResponse *)
3176         odr_malloc(assoc->encode, sizeof(*res));
3177     bend_sort_rr *bsrr = (bend_sort_rr *)
3178         odr_malloc(assoc->encode, sizeof(*bsrr));
3179
3180     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
3181
3182     yaz_log(log_requestdetail, "Got SortRequest.");
3183
3184     bsrr->num_input_setnames = req->num_inputResultSetNames;
3185     for (i=0;i<req->num_inputResultSetNames;i++)
3186         yaz_log(log_requestdetail, "Input resultset: '%s'",
3187                 req->inputResultSetNames[i]);
3188     bsrr->input_setnames = req->inputResultSetNames;
3189     bsrr->referenceId = req->referenceId;
3190     bsrr->output_setname = req->sortedResultSetName;
3191     yaz_log(log_requestdetail, "Output resultset: '%s'",
3192                 req->sortedResultSetName);
3193     bsrr->sort_sequence = req->sortSequence;
3194        /*FIXME - dump those sequences too */
3195     bsrr->stream = assoc->encode;
3196     bsrr->print = assoc->print;
3197
3198     bsrr->sort_status = Z_SortResponse_failure;
3199     bsrr->errcode = 0;
3200     bsrr->errstring = 0;
3201     
3202     (*assoc->init->bend_sort)(assoc->backend, bsrr);
3203     
3204     res->referenceId = bsrr->referenceId;
3205     res->sortStatus = odr_intdup(assoc->encode, bsrr->sort_status);
3206     res->resultSetStatus = 0;
3207     if (bsrr->errcode)
3208     {
3209         Z_DiagRecs *dr = zget_DiagRecs(assoc->encode,
3210                                        bsrr->errcode, bsrr->errstring);
3211         res->diagnostics = dr->diagRecs;
3212         res->num_diagnostics = dr->num_diagRecs;
3213     }
3214     else
3215     {
3216         res->num_diagnostics = 0;
3217         res->diagnostics = 0;
3218     }
3219     res->resultCount = 0;
3220     res->otherInfo = 0;
3221
3222     apdu->which = Z_APDU_sortResponse;
3223     apdu->u.sortResponse = res;
3224     if (log_request)
3225     {
3226         WRBUF wr = wrbuf_alloc();
3227         wrbuf_printf(wr, "Sort ");
3228         if (bsrr->errcode)
3229             wrbuf_printf(wr, " ERROR %d", bsrr->errcode);
3230         else
3231             wrbuf_printf(wr,  "OK -");
3232         wrbuf_printf(wr, " (");
3233         for (i = 0; i<req->num_inputResultSetNames; i++)
3234         {
3235             if (i)
3236                 wrbuf_printf(wr, "+");
3237             wrbuf_puts(wr, req->inputResultSetNames[i]);
3238         }
3239         wrbuf_printf(wr, ")->%s ",req->sortedResultSetName);
3240
3241         yaz_log(log_request, "%s", wrbuf_cstr(wr) );
3242         wrbuf_destroy(wr);
3243     }
3244     return apdu;
3245 }
3246
3247 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
3248     int *fd)
3249 {
3250     int i;
3251     Z_DeleteResultSetRequest *req =
3252         reqb->apdu_request->u.deleteResultSetRequest;
3253     Z_DeleteResultSetResponse *res = (Z_DeleteResultSetResponse *)
3254         odr_malloc(assoc->encode, sizeof(*res));
3255     bend_delete_rr *bdrr = (bend_delete_rr *)
3256         odr_malloc(assoc->encode, sizeof(*bdrr));
3257     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
3258
3259     yaz_log(log_requestdetail, "Got DeleteRequest.");
3260
3261     bdrr->num_setnames = req->num_resultSetList;
3262     bdrr->setnames = req->resultSetList;
3263     for (i = 0; i<req->num_resultSetList; i++)
3264         yaz_log(log_requestdetail, "resultset: '%s'",
3265                 req->resultSetList[i]);
3266     bdrr->stream = assoc->encode;
3267     bdrr->print = assoc->print;
3268     bdrr->function = *req->deleteFunction;
3269     bdrr->referenceId = req->referenceId;
3270     bdrr->statuses = 0;
3271     if (bdrr->num_setnames > 0)
3272     {
3273         bdrr->statuses = (Odr_int*) 
3274             odr_malloc(assoc->encode, sizeof(*bdrr->statuses) *
3275                        bdrr->num_setnames);
3276         for (i = 0; i < bdrr->num_setnames; i++)
3277             bdrr->statuses[i] = 0;
3278     }
3279     (*assoc->init->bend_delete)(assoc->backend, bdrr);
3280     
3281     res->referenceId = req->referenceId;
3282
3283     res->deleteOperationStatus = odr_intdup(assoc->encode,bdrr->delete_status);
3284
3285     res->deleteListStatuses = 0;
3286     if (bdrr->num_setnames > 0)
3287     {
3288         int i;
3289         res->deleteListStatuses = (Z_ListStatuses *)
3290             odr_malloc(assoc->encode, sizeof(*res->deleteListStatuses));
3291         res->deleteListStatuses->num = bdrr->num_setnames;
3292         res->deleteListStatuses->elements =
3293             (Z_ListStatus **)
3294             odr_malloc(assoc->encode, 
3295                         sizeof(*res->deleteListStatuses->elements) *
3296                         bdrr->num_setnames);
3297         for (i = 0; i<bdrr->num_setnames; i++)
3298         {
3299             res->deleteListStatuses->elements[i] =
3300                 (Z_ListStatus *)
3301                 odr_malloc(assoc->encode,
3302                             sizeof(**res->deleteListStatuses->elements));
3303             res->deleteListStatuses->elements[i]->status = bdrr->statuses+i;
3304             res->deleteListStatuses->elements[i]->id =
3305                 odr_strdup(assoc->encode, bdrr->setnames[i]);
3306         }
3307     }
3308     res->numberNotDeleted = 0;
3309     res->bulkStatuses = 0;
3310     res->deleteMessage = 0;
3311     res->otherInfo = 0;
3312
3313     apdu->which = Z_APDU_deleteResultSetResponse;
3314     apdu->u.deleteResultSetResponse = res;
3315     if (log_request)
3316     {
3317         WRBUF wr = wrbuf_alloc();
3318         wrbuf_printf(wr, "Delete ");
3319         if (bdrr->delete_status)
3320             wrbuf_printf(wr, "ERROR %d", bdrr->delete_status);
3321         else
3322             wrbuf_printf(wr, "OK -");
3323         for (i = 0; i<req->num_resultSetList; i++)
3324             wrbuf_printf(wr, " %s ", req->resultSetList[i]);
3325         yaz_log(log_request, "%s", wrbuf_cstr(wr) );
3326         wrbuf_destroy(wr);
3327     }
3328     return apdu;
3329 }
3330
3331 static void process_close(association *assoc, request *reqb)
3332 {
3333     Z_Close *req = reqb->apdu_request->u.close;
3334     static char *reasons[] =
3335     {
3336         "finished",
3337         "shutdown",
3338         "systemProblem",
3339         "costLimit",
3340         "resources",
3341         "securityViolation",
3342         "protocolError",
3343         "lackOfActivity",
3344         "peerAbort",
3345         "unspecified"
3346     };
3347
3348     yaz_log(log_requestdetail, "Got Close, reason %s, message %s",
3349         reasons[*req->closeReason], req->diagnosticInformation ?
3350         req->diagnosticInformation : "NULL");
3351     if (assoc->version < 3) /* to make do_force respond with close */
3352         assoc->version = 3;
3353     do_close_req(assoc, Z_Close_finished,
3354                  "Association terminated by client", reqb);
3355     yaz_log(log_request,"Close OK");
3356 }
3357
3358 void save_referenceId(request *reqb, Z_ReferenceId *refid)
3359 {
3360     if (refid)
3361     {
3362         reqb->len_refid = refid->len;
3363         reqb->refid = (char *)nmem_malloc(reqb->request_mem, refid->len);
3364         memcpy(reqb->refid, refid->buf, refid->len);
3365     }
3366     else
3367     {
3368         reqb->len_refid = 0;
3369         reqb->refid = NULL;
3370     }
3371 }
3372
3373 void bend_request_send(bend_association a, bend_request req, Z_APDU *res)
3374 {
3375     process_z_response(a, req, res);
3376 }
3377
3378 bend_request bend_request_mk(bend_association a)
3379 {
3380     request *nreq = request_get(&a->outgoing);
3381     nreq->request_mem = nmem_create();
3382     return nreq;
3383 }
3384
3385 Z_ReferenceId *bend_request_getid(ODR odr, bend_request req)
3386 {
3387     Z_ReferenceId *id;
3388     if (!req->refid)
3389         return 0;
3390     id = (Odr_oct *)odr_malloc(odr, sizeof(*odr));
3391     id->buf = (unsigned char *)odr_malloc(odr, req->len_refid);
3392     id->len = id->size = req->len_refid;
3393     memcpy(id->buf, req->refid, req->len_refid);
3394     return id;
3395 }
3396
3397 void bend_request_destroy(bend_request *req)
3398 {
3399     nmem_destroy((*req)->request_mem);
3400     request_release(*req);
3401     *req = NULL;
3402 }
3403
3404 int bend_backend_respond(bend_association a, bend_request req)
3405 {
3406     char *msg;
3407     int r;
3408     r = process_z_request(a, req, &msg);
3409     if (r < 0)
3410         yaz_log(YLOG_WARN, "%s", msg);
3411     return r;
3412 }
3413
3414 void bend_request_setdata(bend_request r, void *p)
3415 {
3416     r->clientData = p;
3417 }
3418
3419 void *bend_request_getdata(bend_request r)
3420 {
3421     return r->clientData;
3422 }
3423
3424 static Z_APDU *process_segmentRequest(association *assoc, request *reqb)
3425 {
3426     bend_segment_rr req;
3427
3428     req.segment = reqb->apdu_request->u.segmentRequest;
3429     req.stream = assoc->encode;
3430     req.decode = assoc->decode;
3431     req.print = assoc->print;
3432     req.association = assoc;
3433     
3434     (*assoc->init->bend_segment)(assoc->backend, &req);
3435
3436     return 0;
3437 }
3438
3439 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd)
3440 {
3441     bend_esrequest_rr esrequest;
3442     const char *ext_name = "unknown";
3443
3444     Z_ExtendedServicesRequest *req =
3445         reqb->apdu_request->u.extendedServicesRequest;
3446     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_extendedServicesResponse);
3447
3448     Z_ExtendedServicesResponse *resp = apdu->u.extendedServicesResponse;
3449
3450     esrequest.esr = reqb->apdu_request->u.extendedServicesRequest;
3451     esrequest.stream = assoc->encode;
3452     esrequest.decode = assoc->decode;
3453     esrequest.print = assoc->print;
3454     esrequest.errcode = 0;
3455     esrequest.errstring = NULL;
3456     esrequest.request = reqb;
3457     esrequest.association = assoc;
3458     esrequest.taskPackage = 0;
3459     esrequest.referenceId = req->referenceId;
3460
3461     
3462     if (esrequest.esr && esrequest.esr->taskSpecificParameters)
3463     {
3464         switch(esrequest.esr->taskSpecificParameters->which)
3465         {
3466         case Z_External_itemOrder:
3467             ext_name = "ItemOrder"; break;
3468         case Z_External_update:
3469             ext_name = "Update"; break;
3470         case Z_External_update0:
3471             ext_name = "Update0"; break;
3472         case Z_External_ESAdmin:
3473             ext_name = "Admin"; break;
3474
3475         }
3476     }
3477
3478     (*assoc->init->bend_esrequest)(assoc->backend, &esrequest);
3479     
3480     /* If the response is being delayed, return NULL */
3481     if (esrequest.request == NULL)
3482         return(NULL);
3483
3484     resp->referenceId = req->referenceId;
3485
3486     if (esrequest.errcode == -1)
3487     {
3488         /* Backend service indicates request will be processed */
3489         yaz_log(log_request, "Extended Service: %s (accepted)", ext_name);
3490         *resp->operationStatus = Z_ExtendedServicesResponse_accepted;
3491     }
3492     else if (esrequest.errcode == 0)
3493     {
3494         /* Backend service indicates request will be processed */
3495         yaz_log(log_request, "Extended Service: %s (done)", ext_name);
3496         *resp->operationStatus = Z_ExtendedServicesResponse_done;
3497     }
3498     else
3499     {
3500         Z_DiagRecs *diagRecs =
3501             zget_DiagRecs(assoc->encode, esrequest.errcode,
3502                           esrequest.errstring);
3503         /* Backend indicates error, request will not be processed */
3504         yaz_log(log_request, "Extended Service: %s (failed)", ext_name);
3505         *resp->operationStatus = Z_ExtendedServicesResponse_failure;
3506         resp->num_diagnostics = diagRecs->num_diagRecs;
3507         resp->diagnostics = diagRecs->diagRecs;
3508         if (log_request)
3509         {
3510             WRBUF wr = wrbuf_alloc();
3511             wrbuf_diags(wr, resp->num_diagnostics, resp->diagnostics);
3512             yaz_log(log_request, "EsRequest %s", wrbuf_cstr(wr) );
3513             wrbuf_destroy(wr);
3514         }
3515
3516     }
3517     /* Do something with the members of bend_extendedservice */
3518     if (esrequest.taskPackage)
3519     {
3520         resp->taskPackage = z_ext_record_oid(
3521             assoc->encode, yaz_oid_recsyn_extended,
3522             (const char *)  esrequest.taskPackage, -1);
3523     }
3524     yaz_log(YLOG_DEBUG,"Send the result apdu");
3525     return apdu;
3526 }
3527
3528 int bend_assoc_is_alive(bend_association assoc)
3529 {
3530     if (assoc->state == ASSOC_DEAD)
3531         return 0; /* already marked as dead. Don't check I/O chan anymore */
3532
3533     return iochan_is_alive(assoc->client_chan);
3534 }
3535
3536
3537 /*
3538  * Local variables:
3539  * c-basic-offset: 4
3540  * c-file-style: "Stroustrup"
3541  * indent-tabs-mode: nil
3542  * End:
3543  * vim: shiftwidth=4 tabstop=8 expandtab
3544  */
3545