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