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