More odr_int casts
[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, request *req,
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(void *handle, 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, request *req,
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(assoc->backend, &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, request *req,
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         struct scan_entry *save_entries;
1249
1250         bend_scan_rr *bsrr = (bend_scan_rr *)
1251             odr_malloc(assoc->encode, sizeof(*bsrr));
1252         bsrr->num_bases = 1;
1253         bsrr->basenames = &srw_req->database;
1254
1255         bsrr->num_entries = srw_req->maximumTerms ?
1256             odr_int_to_int(*srw_req->maximumTerms) : 10;
1257         bsrr->term_position = srw_req->responsePosition ?
1258             *srw_req->responsePosition : 1;
1259
1260         bsrr->errcode = 0;
1261         bsrr->errstring = 0;
1262         bsrr->referenceId = 0;
1263         bsrr->stream = assoc->encode;
1264         bsrr->print = assoc->print;
1265         bsrr->step_size = odr_intdup(assoc->decode, 0);
1266         bsrr->entries = 0;
1267         bsrr->setname = 0;
1268
1269         if (bsrr->num_entries > 0) 
1270         {
1271             int i;
1272             bsrr->entries = (struct scan_entry *) 
1273                 odr_malloc(assoc->decode, sizeof(*bsrr->entries) *
1274                            bsrr->num_entries);
1275             for (i = 0; i<bsrr->num_entries; i++)
1276             {
1277                 bsrr->entries[i].term = 0;
1278                 bsrr->entries[i].occurrences = 0;
1279                 bsrr->entries[i].errcode = 0;
1280                 bsrr->entries[i].errstring = 0;
1281                 bsrr->entries[i].display_term = 0;
1282             }
1283         }
1284         save_entries = bsrr->entries;  /* save it so we can compare later */
1285
1286         if (srw_req->query_type == Z_SRW_query_type_pqf &&
1287             assoc->init->bend_scan)
1288         {
1289             YAZ_PQF_Parser pqf_parser = yaz_pqf_create();
1290             
1291             bsrr->term = yaz_pqf_scan(pqf_parser, assoc->decode,
1292                                       &bsrr->attributeset, 
1293                                       srw_req->scanClause.pqf); 
1294             yaz_pqf_destroy(pqf_parser);
1295             bsrr->scanClause = 0;
1296             ((int (*)(void *, bend_scan_rr *))
1297              (*assoc->init->bend_scan))(assoc->backend, bsrr);
1298         }
1299         else if (srw_req->query_type == Z_SRW_query_type_cql
1300                  && assoc->init->bend_scan && assoc->server
1301                  && assoc->server->cql_transform)
1302         {
1303             int srw_error;
1304             bsrr->scanClause = 0;
1305             bsrr->attributeset = 0;
1306             bsrr->term = (Z_AttributesPlusTerm *)
1307                 odr_malloc(assoc->decode, sizeof(*bsrr->term));
1308             srw_error = cql2pqf_scan(assoc->encode,
1309                                      srw_req->scanClause.cql,
1310                                      assoc->server->cql_transform,
1311                                      bsrr->term);
1312             if (srw_error)
1313                 yaz_add_srw_diagnostic(assoc->encode, &srw_res->diagnostics,
1314                                        &srw_res->num_diagnostics,
1315                                        srw_error, 0);
1316             else
1317             {
1318                 ((int (*)(void *, bend_scan_rr *))
1319                  (*assoc->init->bend_scan))(assoc->backend, bsrr);
1320             }
1321         }
1322         else if (srw_req->query_type == Z_SRW_query_type_cql
1323                  && assoc->init->bend_srw_scan)
1324         {
1325             bsrr->term = 0;
1326             bsrr->attributeset = 0;
1327             bsrr->scanClause = srw_req->scanClause.cql;
1328             ((int (*)(void *, bend_scan_rr *))
1329              (*assoc->init->bend_srw_scan))(assoc->backend, bsrr);
1330         }
1331         else
1332         {
1333             yaz_add_srw_diagnostic(assoc->encode, &srw_res->diagnostics,
1334                                    &srw_res->num_diagnostics,
1335                                    YAZ_SRW_UNSUPP_OPERATION, "scan");
1336         }
1337         if (bsrr->errcode)
1338         {
1339             int srw_error;
1340             if (bsrr->errcode == YAZ_BIB1_DATABASE_UNAVAILABLE)
1341             {
1342                 *http_code = 404;
1343                 return;
1344             }
1345             srw_error = yaz_diag_bib1_to_srw(bsrr->errcode);
1346
1347             yaz_add_srw_diagnostic(assoc->encode, &srw_res->diagnostics,
1348                                    &srw_res->num_diagnostics,
1349                                    srw_error, bsrr->errstring);
1350         }
1351         else if (srw_res->num_diagnostics == 0 && bsrr->num_entries)
1352         {
1353             int i;
1354             srw_res->terms = (Z_SRW_scanTerm*)
1355                 odr_malloc(assoc->encode, sizeof(*srw_res->terms) *
1356                            bsrr->num_entries);
1357
1358             srw_res->num_terms =  bsrr->num_entries;
1359             for (i = 0; i<bsrr->num_entries; i++)
1360             {
1361                 Z_SRW_scanTerm *t = srw_res->terms + i;
1362                 t->value = odr_strdup(assoc->encode, bsrr->entries[i].term);
1363                 t->numberOfRecords =
1364                     odr_intdup(assoc->encode, bsrr->entries[i].occurrences);
1365                 t->displayTerm = 0;
1366                 if (save_entries == bsrr->entries && 
1367                     bsrr->entries[i].display_term)
1368                 {
1369                     /* the entries was _not_ set by the handler. So it's
1370                        safe to test for new member display_term. It is
1371                        NULL'ed by us.
1372                     */
1373                     t->displayTerm = odr_strdup(assoc->encode, 
1374                                                 bsrr->entries[i].display_term);
1375                 }
1376                 t->whereInList = 0;
1377             }
1378         }
1379     }
1380     if (log_request)
1381     {
1382         WRBUF wr = wrbuf_alloc();
1383         const char *querytype = 0;
1384         const char *querystr = 0;
1385
1386         switch(srw_req->query_type)
1387         {
1388         case Z_SRW_query_type_pqf:
1389             querytype = "PQF";
1390             querystr = srw_req->scanClause.pqf;
1391             break;
1392         case Z_SRW_query_type_cql:
1393             querytype = "CQL";
1394             querystr = srw_req->scanClause.cql;
1395             break;
1396         default:
1397             querytype = "UNKNOWN";
1398             querystr = "";
1399         }
1400
1401         wrbuf_printf(wr, "SRWScan %s ", srw_req->database);
1402
1403         if (srw_res->num_diagnostics)
1404             wrbuf_printf(wr, "ERROR %s - ", srw_res->diagnostics[0].uri);
1405         else if (srw_res->num_terms)
1406             wrbuf_printf(wr, "OK %d - ", srw_res->num_terms);
1407         else
1408             wrbuf_printf(wr, "OK - - ");
1409
1410         wrbuf_printf(wr, ODR_INT_PRINTF "+" ODR_INT_PRINTF " ",
1411                      (srw_req->responsePosition ? 
1412                       *srw_req->responsePosition : 1),
1413                      (srw_req->maximumTerms ?
1414                       *srw_req->maximumTerms : 1));
1415         /* there is no step size in SRU/W ??? */
1416         wrbuf_printf(wr, "%s: %s ", querytype, querystr);
1417         yaz_log(log_request, "%s ", wrbuf_cstr(wr) );
1418         wrbuf_destroy(wr);
1419     }
1420
1421 }
1422
1423 static void srw_bend_update(association *assoc, request *req,
1424                             Z_SRW_PDU *sr,
1425                             Z_SRW_updateResponse *srw_res,
1426                             int *http_code)
1427 {
1428     Z_SRW_updateRequest *srw_req = sr->u.update_request;
1429     yaz_log(log_session, "SRWUpdate action=%s", srw_req->operation);
1430     yaz_log(YLOG_DEBUG, "num_diag = %d", srw_res->num_diagnostics );
1431     *http_code = 404;
1432     srw_bend_init(assoc, &srw_res->diagnostics, &srw_res->num_diagnostics, sr);
1433     if (assoc->init)
1434     {
1435         bend_update_rr rr;
1436         Z_SRW_extra_record *extra = srw_req->extra_record;
1437         
1438         rr.stream = assoc->encode;
1439         rr.print = assoc->print;
1440         rr.num_bases = 1;
1441         rr.basenames = &srw_req->database;
1442         rr.operation = srw_req->operation;
1443         rr.operation_status = "failed";
1444         rr.record_id = 0;
1445         rr.record_versions = 0;
1446         rr.num_versions = 0;
1447         rr.record_packing = "string";
1448         rr.record_schema = 0;
1449         rr.record_data = 0;
1450         rr.extra_record_data = 0;
1451         rr.extra_request_data = 0;
1452         rr.extra_response_data = 0;
1453         rr.uri = 0;
1454         rr.message = 0;
1455         rr.details = 0;
1456         
1457         *http_code = 200;
1458         if (rr.operation == 0)
1459         {
1460             yaz_add_sru_update_diagnostic(
1461                 assoc->encode, &srw_res->diagnostics,
1462                 &srw_res->num_diagnostics,
1463                 YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,
1464                 "action" );
1465             return;
1466         }
1467         yaz_log(YLOG_DEBUG, "basename = %s", rr.basenames[0] );
1468         yaz_log(YLOG_DEBUG, "Operation = %s", rr.operation );
1469         if (!strcmp( rr.operation, "delete"))
1470         {
1471             if (srw_req->record && !srw_req->record->recordSchema)
1472             {
1473                 rr.record_schema = odr_strdup(
1474                     assoc->encode,
1475                     srw_req->record->recordSchema);
1476             }
1477             if (srw_req->record)
1478             {
1479                 rr.record_data = odr_strdupn(
1480                     assoc->encode, 
1481                     srw_req->record->recordData_buf,
1482                     srw_req->record->recordData_len );
1483             }
1484             if (extra && extra->extraRecordData_len)
1485             {
1486                 rr.extra_record_data = odr_strdupn(
1487                     assoc->encode, 
1488                     extra->extraRecordData_buf,
1489                     extra->extraRecordData_len );
1490             }
1491             if (srw_req->recordId)
1492                 rr.record_id = srw_req->recordId;
1493             else if (extra && extra->recordIdentifier)
1494                 rr.record_id = extra->recordIdentifier;
1495         }
1496         else if (!strcmp(rr.operation, "replace"))
1497         {
1498             if (srw_req->recordId)
1499                 rr.record_id = srw_req->recordId;
1500             else if (extra && extra->recordIdentifier)
1501                 rr.record_id = extra->recordIdentifier;
1502             else 
1503             {
1504                 yaz_add_sru_update_diagnostic(
1505                     assoc->encode, &srw_res->diagnostics,
1506                     &srw_res->num_diagnostics,
1507                     YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,
1508                     "recordIdentifier");
1509             }
1510             if (!srw_req->record)
1511             {
1512                 yaz_add_sru_update_diagnostic(
1513                     assoc->encode, &srw_res->diagnostics,
1514                     &srw_res->num_diagnostics,
1515                     YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,
1516                     "record");
1517             }
1518             else 
1519             {
1520                 if (srw_req->record->recordSchema)
1521                     rr.record_schema = odr_strdup(
1522                         assoc->encode, srw_req->record->recordSchema);
1523                 if (srw_req->record->recordData_len )
1524                 {
1525                     rr.record_data = odr_strdupn(assoc->encode, 
1526                                                  srw_req->record->recordData_buf,
1527                                                  srw_req->record->recordData_len );
1528                 }
1529                 else 
1530                 {
1531                     yaz_add_sru_update_diagnostic(
1532                         assoc->encode, &srw_res->diagnostics,
1533                         &srw_res->num_diagnostics,
1534                         YAZ_SRU_UPDATE_MISSING_MANDATORY_ELEMENT_RECORD_REJECTED,                                              
1535                         "recordData" );
1536                 }
1537             }
1538             if (extra && extra->extraRecordData_len)
1539             {
1540                 rr.extra_record_data = odr_strdupn(
1541                     assoc->encode, 
1542                     extra->extraRecordData_buf,
1543                     extra->extraRecordData_len );
1544             }
1545         }
1546         else if (!strcmp(rr.operation, "insert"))
1547         {
1548             if (srw_req->recordId)
1549                 rr.record_id = srw_req->recordId; 
1550             else if (extra)
1551                 rr.record_id = extra->recordIdentifier;
1552             
1553             if (srw_req->record)
1554             {
1555                 if (srw_req->record->recordSchema)
1556                     rr.record_schema = odr_strdup(
1557                         assoc->encode, srw_req->record->recordSchema);
1558             
1559                 if (srw_req->record->recordData_len)
1560                     rr.record_data = odr_strdupn(
1561                         assoc->encode, 
1562                         srw_req->record->recordData_buf,
1563                         srw_req->record->recordData_len );
1564             }
1565             if (extra && extra->extraRecordData_len)
1566             {
1567                 rr.extra_record_data = odr_strdupn(
1568                     assoc->encode, 
1569                     extra->extraRecordData_buf,
1570                     extra->extraRecordData_len );
1571             }
1572         }
1573         else 
1574             yaz_add_sru_update_diagnostic(assoc->encode, &srw_res->diagnostics,
1575                                           &srw_res->num_diagnostics,
1576                                           YAZ_SRU_UPDATE_INVALID_ACTION,
1577                                           rr.operation );
1578
1579         if (srw_req->record)
1580         {
1581             const char *pack_str = 
1582                 yaz_srw_pack_to_str(srw_req->record->recordPacking);
1583             if (pack_str)
1584                 rr.record_packing = odr_strdup(assoc->encode, pack_str);
1585         }
1586
1587         if (srw_req->num_recordVersions)
1588         {
1589             rr.record_versions = srw_req->recordVersions;
1590             rr.num_versions = srw_req->num_recordVersions;
1591         }
1592         if (srw_req->extraRequestData_len)
1593         {
1594             rr.extra_request_data = odr_strdupn(assoc->encode,
1595                                                 srw_req->extraRequestData_buf,
1596                                                 srw_req->extraRequestData_len );
1597         }
1598         if (srw_res->num_diagnostics == 0)
1599         {
1600             if ( assoc->init->bend_srw_update)
1601                 (*assoc->init->bend_srw_update)(assoc->backend, &rr);
1602             else 
1603                 yaz_add_sru_update_diagnostic(
1604                     assoc->encode, &srw_res->diagnostics,
1605                     &srw_res->num_diagnostics,
1606                     YAZ_SRU_UPDATE_UNSPECIFIED_DATABASE_ERROR,
1607                     "No Update backend handler");
1608         }
1609
1610         if (rr.uri)
1611             yaz_add_srw_diagnostic_uri(assoc->encode,
1612                                        &srw_res->diagnostics,
1613                                        &srw_res->num_diagnostics,
1614                                        rr.uri, 
1615                                        rr.message,
1616                                        rr.details);
1617         srw_res->recordId = rr.record_id;
1618         srw_res->operationStatus = rr.operation_status;
1619         srw_res->recordVersions = rr.record_versions;
1620         srw_res->num_recordVersions = rr.num_versions;
1621         if (srw_res->extraResponseData_len)
1622         {
1623             srw_res->extraResponseData_buf = rr.extra_response_data;
1624             srw_res->extraResponseData_len = strlen(rr.extra_response_data);
1625         }
1626         if (srw_res->num_diagnostics == 0 && rr.record_data)
1627         {
1628             srw_res->record = yaz_srw_get_record(assoc->encode);
1629             srw_res->record->recordSchema = rr.record_schema;
1630             if (rr.record_packing)
1631             {
1632                 int pack = yaz_srw_str_to_pack(rr.record_packing);
1633
1634                 if (pack == -1)
1635                 {
1636                     pack = Z_SRW_recordPacking_string;
1637                     yaz_log(YLOG_WARN, "Back packing %s from backend",
1638                             rr.record_packing);
1639                 }
1640                 srw_res->record->recordPacking = pack;
1641             }
1642             srw_res->record->recordData_buf = rr.record_data;
1643             srw_res->record->recordData_len = strlen(rr.record_data);
1644             if (rr.extra_record_data)
1645             {
1646                 Z_SRW_extra_record *ex = 
1647                     yaz_srw_get_extra_record(assoc->encode);
1648                 srw_res->extra_record = ex;
1649                 ex->extraRecordData_buf = rr.extra_record_data;
1650                 ex->extraRecordData_len = strlen(rr.extra_record_data);
1651             }
1652         }
1653     }
1654 }
1655
1656 /* check if path is OK (1); BAD (0) */
1657 static int check_path(const char *path)
1658 {
1659     if (*path != '/')
1660         return 0;
1661     if (strstr(path, ".."))
1662         return 0;
1663     return 1;
1664 }
1665
1666 static char *read_file(const char *fname, ODR o, size_t *sz)
1667 {
1668     char *buf;
1669     FILE *inf = fopen(fname, "rb");
1670     if (!inf)
1671         return 0;
1672
1673     fseek(inf, 0L, SEEK_END);
1674     *sz = ftell(inf);
1675     rewind(inf);
1676     buf = (char *) odr_malloc(o, *sz);
1677     if (fread(buf, 1, *sz, inf) != *sz)
1678         yaz_log(YLOG_WARN|YLOG_ERRNO, "short read %s", fname);
1679     fclose(inf);
1680     return buf;     
1681 }
1682
1683 static void process_http_request(association *assoc, request *req)
1684 {
1685     Z_HTTP_Request *hreq = req->gdu_request->u.HTTP_Request;
1686     ODR o = assoc->encode;
1687     int r = 2;  /* 2=NOT TAKEN, 1=TAKEN, 0=SOAP TAKEN */
1688     Z_SRW_PDU *sr = 0;
1689     Z_SOAP *soap_package = 0;
1690     Z_GDU *p = 0;
1691     char *charset = 0;
1692     Z_HTTP_Response *hres = 0;
1693     int keepalive = 1;
1694     const char *stylesheet = 0; /* for now .. set later */
1695     Z_SRW_diagnostic *diagnostic = 0;
1696     int num_diagnostic = 0;
1697     const char *host = z_HTTP_header_lookup(hreq->headers, "Host");
1698
1699     yaz_log(log_request, "%s %s HTTP/%s", hreq->method, hreq->path, hreq->version);
1700     if (!control_association(assoc, host, 0))
1701     {
1702         p = z_get_HTTP_Response(o, 404);
1703         r = 1;
1704     }
1705     if (r == 2 && assoc->server && assoc->server->docpath
1706         && hreq->path[0] == '/' 
1707         && 
1708         /* check if path is a proper prefix of documentroot */
1709         strncmp(hreq->path+1, assoc->server->docpath,
1710                 strlen(assoc->server->docpath))
1711         == 0)
1712     {   
1713         if (!check_path(hreq->path))
1714         {
1715             yaz_log(YLOG_LOG, "File %s access forbidden", hreq->path+1);
1716             p = z_get_HTTP_Response(o, 404);
1717         }
1718         else
1719         {
1720             size_t content_size = 0;
1721             char *content_buf = read_file(hreq->path+1, o, &content_size);
1722             if (!content_buf)
1723             {
1724                 yaz_log(YLOG_LOG, "File %s not found", hreq->path+1);
1725                 p = z_get_HTTP_Response(o, 404);
1726             }
1727             else
1728             {
1729                 const char *ctype = 0;
1730                 yaz_mime_types types = yaz_mime_types_create();
1731                 
1732                 yaz_mime_types_add(types, "xsl", "application/xml");
1733                 yaz_mime_types_add(types, "xml", "application/xml");
1734                 yaz_mime_types_add(types, "css", "text/css");
1735                 yaz_mime_types_add(types, "html", "text/html");
1736                 yaz_mime_types_add(types, "htm", "text/html");
1737                 yaz_mime_types_add(types, "txt", "text/plain");
1738                 yaz_mime_types_add(types, "js", "application/x-javascript");
1739                 
1740                 yaz_mime_types_add(types, "gif", "image/gif");
1741                 yaz_mime_types_add(types, "png", "image/png");
1742                 yaz_mime_types_add(types, "jpg", "image/jpeg");
1743                 yaz_mime_types_add(types, "jpeg", "image/jpeg");
1744                 
1745                 ctype = yaz_mime_lookup_fname(types, hreq->path);
1746                 if (!ctype)
1747                 {
1748                     yaz_log(YLOG_LOG, "No mime type for %s", hreq->path+1);
1749                     p = z_get_HTTP_Response(o, 404);
1750                 }
1751                 else
1752                 {
1753                     p = z_get_HTTP_Response(o, 200);
1754                     hres = p->u.HTTP_Response;
1755                     hres->content_buf = content_buf;
1756                     hres->content_len = content_size;
1757                     z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
1758                 }
1759                 yaz_mime_types_destroy(types);
1760             }
1761         }
1762         r = 1;
1763     }
1764
1765     if (r == 2)
1766     {
1767         r = yaz_srw_decode(hreq, &sr, &soap_package, assoc->decode, &charset);
1768         yaz_log(YLOG_DEBUG, "yaz_srw_decode returned %d", r);
1769     }
1770     if (r == 2)  /* not taken */
1771     {
1772         r = yaz_sru_decode(hreq, &sr, &soap_package, assoc->decode, &charset,
1773                            &diagnostic, &num_diagnostic);
1774         yaz_log(YLOG_DEBUG, "yaz_sru_decode returned %d", r);
1775     }
1776     if (r == 0)  /* decode SRW/SRU OK .. */
1777     {
1778         int http_code = 200;
1779         if (sr->which == Z_SRW_searchRetrieve_request)
1780         {
1781             Z_SRW_PDU *res =
1782                 yaz_srw_get_pdu(assoc->encode, Z_SRW_searchRetrieve_response,
1783                                 sr->srw_version);
1784             stylesheet = sr->u.request->stylesheet;
1785             if (num_diagnostic)
1786             {
1787                 res->u.response->diagnostics = diagnostic;
1788                 res->u.response->num_diagnostics = num_diagnostic;
1789             }
1790             else
1791             {
1792                 srw_bend_search(assoc, req, sr, res, &http_code);
1793             }
1794             if (http_code == 200)
1795                 soap_package->u.generic->p = res;
1796         }
1797         else if (sr->which == Z_SRW_explain_request)
1798         {
1799             Z_SRW_PDU *res = yaz_srw_get_pdu(o, Z_SRW_explain_response,
1800                                              sr->srw_version);
1801             stylesheet = sr->u.explain_request->stylesheet;
1802             if (num_diagnostic)
1803             {   
1804                 res->u.explain_response->diagnostics = diagnostic;
1805                 res->u.explain_response->num_diagnostics = num_diagnostic;
1806             }
1807             srw_bend_explain(assoc, req, sr,
1808                              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, req, sr,
1823                           res->u.scan_response, &http_code);
1824             if (http_code == 200)
1825                 soap_package->u.generic->p = res;
1826         }
1827         else if (sr->which == Z_SRW_update_request)
1828         {
1829             Z_SRW_PDU *res = yaz_srw_get_pdu(o, Z_SRW_update_response,
1830                                              sr->srw_version);
1831             yaz_log(YLOG_DEBUG, "handling SRW UpdateRequest");
1832             if (num_diagnostic)
1833             {   
1834                 res->u.update_response->diagnostics = diagnostic;
1835                 res->u.update_response->num_diagnostics = num_diagnostic;
1836             }
1837             yaz_log(YLOG_DEBUG, "num_diag = %d", res->u.update_response->num_diagnostics );
1838             srw_bend_update(assoc, req, sr,
1839                             res->u.update_response, &http_code);
1840             if (http_code == 200)
1841                 soap_package->u.generic->p = res;
1842         }
1843         else
1844         {
1845             yaz_log(log_request, "SOAP ERROR"); 
1846             /* FIXME - what error, what query */
1847             http_code = 500;
1848             z_soap_error(assoc->encode, soap_package,
1849                          "SOAP-ENV:Client", "Bad method", 0); 
1850         }
1851         if (http_code == 200 || http_code == 500)
1852         {
1853             static Z_SOAP_Handler soap_handlers[4] = {
1854 #if YAZ_HAVE_XML2
1855                 {YAZ_XMLNS_SRU_v1_1, 0, (Z_SOAP_fun) yaz_srw_codec},
1856                 {YAZ_XMLNS_SRU_v1_0, 0, (Z_SOAP_fun) yaz_srw_codec},
1857                 {YAZ_XMLNS_UPDATE_v0_9, 0, (Z_SOAP_fun) yaz_ucp_codec},
1858 #endif
1859                 {0, 0, 0}
1860             };
1861             char ctype[80];
1862             int ret;
1863             p = z_get_HTTP_Response(o, 200);
1864             hres = p->u.HTTP_Response;
1865
1866             if (!stylesheet && assoc->server)
1867                 stylesheet = assoc->server->stylesheet;
1868
1869             /* empty stylesheet means NO stylesheet */
1870             if (stylesheet && *stylesheet == '\0')
1871                 stylesheet = 0;
1872
1873             ret = z_soap_codec_enc_xsl(assoc->encode, &soap_package,
1874                                        &hres->content_buf, &hres->content_len,
1875                                        soap_handlers, charset, stylesheet);
1876             hres->code = http_code;
1877
1878             strcpy(ctype, "text/xml");
1879             if (charset && strlen(charset) < sizeof(ctype)-30)
1880             {
1881                 strcat(ctype, "; charset=");
1882                 strcat(ctype, charset);
1883             }
1884             z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
1885         }
1886         else
1887             p = z_get_HTTP_Response(o, http_code);
1888     }
1889
1890     if (p == 0)
1891         p = z_get_HTTP_Response(o, 500);
1892     hres = p->u.HTTP_Response;
1893     if (!strcmp(hreq->version, "1.0")) 
1894     {
1895         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1896         if (v && !strcmp(v, "Keep-Alive"))
1897             keepalive = 1;
1898         else
1899             keepalive = 0;
1900         hres->version = "1.0";
1901     }
1902     else
1903     {
1904         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1905         if (v && !strcmp(v, "close"))
1906             keepalive = 0;
1907         else
1908             keepalive = 1;
1909         hres->version = "1.1";
1910     }
1911     if (!keepalive)
1912     {
1913         z_HTTP_header_add(o, &hres->headers, "Connection", "close");
1914         assoc->state = ASSOC_DEAD;
1915         assoc->cs_get_mask = 0;
1916     }
1917     else
1918     {
1919         int t;
1920         const char *alive = z_HTTP_header_lookup(hreq->headers, "Keep-Alive");
1921
1922         if (alive && isdigit(*(const unsigned char *) alive))
1923             t = atoi(alive);
1924         else
1925             t = 15;
1926         if (t < 0 || t > 3600)
1927             t = 3600;
1928         iochan_settimeout(assoc->client_chan,t);
1929         z_HTTP_header_add(o, &hres->headers, "Connection", "Keep-Alive");
1930     }
1931     process_gdu_response(assoc, req, p);
1932 }
1933
1934 static void process_gdu_request(association *assoc, request *req)
1935 {
1936     if (req->gdu_request->which == Z_GDU_Z3950)
1937     {
1938         char *msg = 0;
1939         req->apdu_request = req->gdu_request->u.z3950;
1940         if (process_z_request(assoc, req, &msg) < 0)
1941             do_close_req(assoc, Z_Close_systemProblem, msg, req);
1942     }
1943     else if (req->gdu_request->which == Z_GDU_HTTP_Request)
1944         process_http_request(assoc, req);
1945     else
1946     {
1947         do_close_req(assoc, Z_Close_systemProblem, "bad protocol packet", req);
1948     }
1949 }
1950
1951 /*
1952  * Initiate request processing.
1953  */
1954 static int process_z_request(association *assoc, request *req, char **msg)
1955 {
1956     Z_APDU *res;
1957     int retval;
1958     
1959     *msg = "Unknown Error";
1960     assert(req && req->state == REQUEST_IDLE);
1961     if (req->apdu_request->which != Z_APDU_initRequest && !assoc->init)
1962     {
1963         *msg = "Missing InitRequest";
1964         return -1;
1965     }
1966     switch (req->apdu_request->which)
1967     {
1968     case Z_APDU_initRequest:
1969         res = process_initRequest(assoc, req); break;
1970     case Z_APDU_searchRequest:
1971         res = process_searchRequest(assoc, req); break;
1972     case Z_APDU_presentRequest:
1973         res = process_presentRequest(assoc, req); break;
1974     case Z_APDU_scanRequest:
1975         if (assoc->init->bend_scan)
1976             res = process_scanRequest(assoc, req);
1977         else
1978         {
1979             *msg = "Cannot handle Scan APDU";
1980             return -1;
1981         }
1982         break;
1983     case Z_APDU_extendedServicesRequest:
1984         if (assoc->init->bend_esrequest)
1985             res = process_ESRequest(assoc, req);
1986         else
1987         {
1988             *msg = "Cannot handle Extended Services APDU";
1989             return -1;
1990         }
1991         break;
1992     case Z_APDU_sortRequest:
1993         if (assoc->init->bend_sort)
1994             res = process_sortRequest(assoc, req);
1995         else
1996         {
1997             *msg = "Cannot handle Sort APDU";
1998             return -1;
1999         }
2000         break;
2001     case Z_APDU_close:
2002         process_close(assoc, req);
2003         return 0;
2004     case Z_APDU_deleteResultSetRequest:
2005         if (assoc->init->bend_delete)
2006             res = process_deleteRequest(assoc, req);
2007         else
2008         {
2009             *msg = "Cannot handle Delete APDU";
2010             return -1;
2011         }
2012         break;
2013     case Z_APDU_segmentRequest:
2014         if (assoc->init->bend_segment)
2015         {
2016             res = process_segmentRequest(assoc, req);
2017         }
2018         else
2019         {
2020             *msg = "Cannot handle Segment APDU";
2021             return -1;
2022         }
2023         break;
2024     case Z_APDU_triggerResourceControlRequest:
2025         return 0;
2026     default:
2027         *msg = "Bad APDU received";
2028         return -1;
2029     }
2030     if (res)
2031     {
2032         yaz_log(YLOG_DEBUG, "  result immediately available");
2033         retval = process_z_response(assoc, req, res);
2034     }
2035     else
2036     {
2037         yaz_log(YLOG_DEBUG, "  result unavailble");
2038         retval = -1;
2039     }
2040     return retval;
2041 }
2042
2043 /*
2044  * Encode response, and transfer the request structure to the outgoing queue.
2045  */
2046 static int process_gdu_response(association *assoc, request *req, Z_GDU *res)
2047 {
2048     odr_setbuf(assoc->encode, req->response, req->size_response, 1);
2049
2050     if (assoc->print)
2051     {
2052         if (!z_GDU(assoc->print, &res, 0, 0))
2053             yaz_log(YLOG_WARN, "ODR print error: %s", 
2054                 odr_errmsg(odr_geterror(assoc->print)));
2055         odr_reset(assoc->print);
2056     }
2057     if (!z_GDU(assoc->encode, &res, 0, 0))
2058     {
2059         yaz_log(YLOG_WARN, "ODR error when encoding PDU: %s [element %s]",
2060                 odr_errmsg(odr_geterror(assoc->decode)),
2061                 odr_getelement(assoc->decode));
2062         return -1;
2063     }
2064     req->response = odr_getbuf(assoc->encode, &req->len_response,
2065         &req->size_response);
2066     odr_setbuf(assoc->encode, 0, 0, 0); /* don'txfree if we abort later */
2067     odr_reset(assoc->encode);
2068     req->state = REQUEST_IDLE;
2069     request_enq(&assoc->outgoing, req);
2070     /* turn the work over to the ir_session handler */
2071     iochan_setflag(assoc->client_chan, EVENT_OUTPUT);
2072     assoc->cs_put_mask = EVENT_OUTPUT;
2073     /* Is there more work to be done? give that to the input handler too */
2074     for (;;)
2075     {
2076         req = request_head(&assoc->incoming);
2077         if (req && req->state == REQUEST_IDLE)
2078         {
2079             request_deq(&assoc->incoming);
2080             process_gdu_request(assoc, req);
2081         }
2082         else
2083             break;
2084     }
2085     return 0;
2086 }
2087
2088 /*
2089  * Encode response, and transfer the request structure to the outgoing queue.
2090  */
2091 static int process_z_response(association *assoc, request *req, Z_APDU *res)
2092 {
2093     Z_GDU *gres = (Z_GDU *) odr_malloc(assoc->encode, sizeof(*res));
2094     gres->which = Z_GDU_Z3950;
2095     gres->u.z3950 = res;
2096
2097     return process_gdu_response(assoc, req, gres);
2098 }
2099
2100 static char *get_vhost(Z_OtherInformation *otherInfo)
2101 {
2102     return yaz_oi_get_string_oid(&otherInfo, yaz_oid_userinfo_proxy, 1, 0);
2103 }
2104
2105 /*
2106  * Handle init request.
2107  * At the moment, we don't check the options
2108  * anywhere else in the code - we just try not to do anything that would
2109  * break a naive client. We'll toss 'em into the association block when
2110  * we need them there.
2111  */
2112 static Z_APDU *process_initRequest(association *assoc, request *reqb)
2113 {
2114     Z_InitRequest *req = reqb->apdu_request->u.initRequest;
2115     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_initResponse);
2116     Z_InitResponse *resp = apdu->u.initResponse;
2117     bend_initresult *binitres;
2118     char options[140];
2119     statserv_options_block *cb = 0;  /* by default no control for backend */
2120
2121     if (control_association(assoc, get_vhost(req->otherInfo), 1))
2122         cb = statserv_getcontrol();  /* got control block for backend */
2123
2124     if (cb && assoc->backend)
2125         (*cb->bend_close)(assoc->backend);
2126
2127     yaz_log(log_requestdetail, "Got initRequest");
2128     if (req->implementationId)
2129         yaz_log(log_requestdetail, "Id:        %s",
2130                 req->implementationId);
2131     if (req->implementationName)
2132         yaz_log(log_requestdetail, "Name:      %s",
2133                 req->implementationName);
2134     if (req->implementationVersion)
2135         yaz_log(log_requestdetail, "Version:   %s",
2136                 req->implementationVersion);
2137     
2138     assoc_init_reset(assoc);
2139
2140     assoc->init->auth = req->idAuthentication;
2141     assoc->init->referenceId = req->referenceId;
2142
2143     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel))
2144     {
2145         Z_CharSetandLanguageNegotiation *negotiation =
2146             yaz_get_charneg_record (req->otherInfo);
2147         if (negotiation &&
2148             negotiation->which == Z_CharSetandLanguageNegotiation_proposal)
2149             assoc->init->charneg_request = negotiation;
2150     }
2151
2152     /* by default named_result_sets is 0 .. Enable it if client asks for it. */
2153     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
2154         assoc->init->named_result_sets = 1;
2155
2156     assoc->backend = 0;
2157     if (cb)
2158     {
2159         if (req->implementationVersion)
2160             yaz_log(log_requestdetail, "Config:    %s",
2161                     cb->configname);
2162     
2163         iochan_settimeout(assoc->client_chan, cb->idle_timeout * 60);
2164         
2165         /* we have a backend control block, so call that init function */
2166         if (!(binitres = (*cb->bend_init)(assoc->init)))
2167         {
2168             yaz_log(YLOG_WARN, "Bad response from backend.");
2169             return 0;
2170         }
2171         assoc->backend = binitres->handle;
2172     }
2173     else
2174     {
2175         /* no backend. return error */
2176         binitres = (bend_initresult *)
2177             odr_malloc(assoc->encode, sizeof(*binitres));
2178         binitres->errstring = 0;
2179         binitres->errcode = YAZ_BIB1_PERMANENT_SYSTEM_ERROR;
2180         iochan_settimeout(assoc->client_chan, 10);
2181     }
2182     if ((assoc->init->bend_sort))
2183         yaz_log(YLOG_DEBUG, "Sort handler installed");
2184     if ((assoc->init->bend_search))
2185         yaz_log(YLOG_DEBUG, "Search handler installed");
2186     if ((assoc->init->bend_present))
2187         yaz_log(YLOG_DEBUG, "Present handler installed");   
2188     if ((assoc->init->bend_esrequest))
2189         yaz_log(YLOG_DEBUG, "ESRequest handler installed");   
2190     if ((assoc->init->bend_delete))
2191         yaz_log(YLOG_DEBUG, "Delete handler installed");   
2192     if ((assoc->init->bend_scan))
2193         yaz_log(YLOG_DEBUG, "Scan handler installed");   
2194     if ((assoc->init->bend_segment))
2195         yaz_log(YLOG_DEBUG, "Segment handler installed");   
2196     
2197     resp->referenceId = req->referenceId;
2198     *options = '\0';
2199     /* let's tell the client what we can do */
2200     if (ODR_MASK_GET(req->options, Z_Options_search))
2201     {
2202         ODR_MASK_SET(resp->options, Z_Options_search);
2203         strcat(options, "srch");
2204     }
2205     if (ODR_MASK_GET(req->options, Z_Options_present))
2206     {
2207         ODR_MASK_SET(resp->options, Z_Options_present);
2208         strcat(options, " prst");
2209     }
2210     if (ODR_MASK_GET(req->options, Z_Options_delSet) &&
2211         assoc->init->bend_delete)
2212     {
2213         ODR_MASK_SET(resp->options, Z_Options_delSet);
2214         strcat(options, " del");
2215     }
2216     if (ODR_MASK_GET(req->options, Z_Options_extendedServices) &&
2217         assoc->init->bend_esrequest)
2218     {
2219         ODR_MASK_SET(resp->options, Z_Options_extendedServices);
2220         strcat(options, " extendedServices");
2221     }
2222     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets)
2223         && assoc->init->named_result_sets)
2224     {
2225         ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
2226         strcat(options, " namedresults");
2227     }
2228     if (ODR_MASK_GET(req->options, Z_Options_scan) && assoc->init->bend_scan)
2229     {
2230         ODR_MASK_SET(resp->options, Z_Options_scan);
2231         strcat(options, " scan");
2232     }
2233     if (ODR_MASK_GET(req->options, Z_Options_concurrentOperations))
2234     {
2235         ODR_MASK_SET(resp->options, Z_Options_concurrentOperations);
2236         strcat(options, " concurrop");
2237     }
2238     if (ODR_MASK_GET(req->options, Z_Options_sort) && assoc->init->bend_sort)
2239     {
2240         ODR_MASK_SET(resp->options, Z_Options_sort);
2241         strcat(options, " sort");
2242     }
2243     
2244     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel))
2245     {
2246         Z_OtherInformationUnit *p0;
2247
2248         if (!assoc->init->charneg_response)
2249         {
2250             if (assoc->init->query_charset)
2251             {
2252                 assoc->init->charneg_response = yaz_set_response_charneg(
2253                     assoc->encode, assoc->init->query_charset, 0, 
2254                     assoc->init->records_in_same_charset);
2255             }
2256             else
2257             {
2258                 yaz_log(YLOG_WARN, "default query_charset not defined by backend");
2259             }
2260         }
2261         if (assoc->init->charneg_response
2262             && (p0=yaz_oi_update(&resp->otherInfo, assoc->encode, NULL, 0, 0)))
2263         {
2264             p0->which = Z_OtherInfo_externallyDefinedInfo;
2265             p0->information.externallyDefinedInfo =
2266                 assoc->init->charneg_response;
2267             ODR_MASK_SET(resp->options, Z_Options_negotiationModel);
2268             strcat(options, " negotiation");
2269         }
2270     }
2271     if (ODR_MASK_GET(req->options, Z_Options_triggerResourceCtrl))
2272         ODR_MASK_SET(resp->options, Z_Options_triggerResourceCtrl);
2273
2274     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_1))
2275     {
2276         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_1);
2277         assoc->version = 1; /* 1 & 2 are equivalent */
2278     }
2279     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_2))
2280     {
2281         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_2);
2282         assoc->version = 2;
2283     }
2284     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_3))
2285     {
2286         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_3);
2287         assoc->version = 3;
2288     }
2289
2290     yaz_log(log_requestdetail, "Negotiated to v%d: %s", assoc->version, options);
2291
2292     if (*req->maximumRecordSize < assoc->maximumRecordSize)
2293         assoc->maximumRecordSize = odr_int_to_int(*req->maximumRecordSize);
2294
2295     if (*req->preferredMessageSize < assoc->preferredMessageSize)
2296         assoc->preferredMessageSize = odr_int_to_int(*req->preferredMessageSize);
2297
2298     resp->preferredMessageSize =
2299         odr_intdup(assoc->encode, assoc->preferredMessageSize);
2300     resp->maximumRecordSize = 
2301         odr_intdup(assoc->encode, assoc->maximumRecordSize);
2302
2303     resp->implementationId = odr_prepend(assoc->encode,
2304                 assoc->init->implementation_id,
2305                 resp->implementationId);
2306
2307     resp->implementationName = odr_prepend(assoc->encode,
2308                 assoc->init->implementation_name,
2309                 odr_prepend(assoc->encode, "GFS", resp->implementationName));
2310
2311     if (binitres->errcode)
2312     {
2313         assoc->state = ASSOC_DEAD;
2314         resp->userInformationField =
2315             init_diagnostics(assoc->encode, binitres->errcode,
2316                              binitres->errstring);
2317         *resp->result = 0;
2318     }
2319     else
2320         assoc->state = ASSOC_UP;
2321     
2322     if (log_request)
2323     {
2324         if (!req->idAuthentication)
2325             yaz_log(log_request, "Auth none");
2326         else if (req->idAuthentication->which == Z_IdAuthentication_open)
2327         {
2328             const char *open = req->idAuthentication->u.open;
2329             const char *slash = strchr(open, '/');
2330             int len;
2331             if (slash)
2332                 len = slash - open;
2333             else
2334                 len = strlen(open);
2335                 yaz_log(log_request, "Auth open %.*s", len, open);
2336         }
2337         else if (req->idAuthentication->which == Z_IdAuthentication_idPass)
2338         {
2339             const char *user = req->idAuthentication->u.idPass->userId;
2340             const char *group = req->idAuthentication->u.idPass->groupId;
2341             yaz_log(log_request, "Auth idPass %s %s",
2342                     user ? user : "-", group ? group : "-");
2343         }
2344         else if (req->idAuthentication->which 
2345                  == Z_IdAuthentication_anonymous)
2346         {
2347             yaz_log(log_request, "Auth anonymous");
2348         }
2349         else
2350         {
2351             yaz_log(log_request, "Auth other");
2352         }
2353     }
2354     if (log_request)
2355     {
2356         WRBUF wr = wrbuf_alloc();
2357         wrbuf_printf(wr, "Init ");
2358         if (binitres->errcode)
2359             wrbuf_printf(wr, "ERROR %d", binitres->errcode);
2360         else
2361             wrbuf_printf(wr, "OK -");
2362         wrbuf_printf(wr, " ID:%s Name:%s Version:%s",
2363                      (req->implementationId ? req->implementationId :"-"), 
2364                      (req->implementationName ?
2365                       req->implementationName : "-"),
2366                      (req->implementationVersion ?
2367                       req->implementationVersion : "-")
2368             );
2369         yaz_log(log_request, "%s", wrbuf_cstr(wr));
2370         wrbuf_destroy(wr);
2371     }
2372     return apdu;
2373 }
2374
2375 /*
2376  * Set the specified `errcode' and `errstring' into a UserInfo-1
2377  * external to be returned to the client in accordance with Z35.90
2378  * Implementor Agreement 5 (Returning diagnostics in an InitResponse):
2379  *      http://lcweb.loc.gov/z3950/agency/agree/initdiag.html
2380  */
2381 static Z_External *init_diagnostics(ODR odr, int error, const char *addinfo)
2382 {
2383     yaz_log(log_requestdetail, "[%d] %s%s%s", error, diagbib1_str(error),
2384         addinfo ? " -- " : "", addinfo ? addinfo : "");
2385     return zget_init_diagnostics(odr, error, addinfo);
2386 }
2387
2388 /*
2389  * nonsurrogate diagnostic record.
2390  */
2391 static Z_Records *diagrec(association *assoc, int error, char *addinfo)
2392 {
2393     Z_Records *rec = (Z_Records *) odr_malloc(assoc->encode, sizeof(*rec));
2394
2395     yaz_log(log_requestdetail, "[%d] %s%s%s", error, diagbib1_str(error),
2396             addinfo ? " -- " : "", addinfo ? addinfo : "");
2397
2398     rec->which = Z_Records_NSD;
2399     rec->u.nonSurrogateDiagnostic = zget_DefaultDiagFormat(assoc->encode,
2400                                                            error, addinfo);
2401     return rec;
2402 }
2403
2404 /*
2405  * surrogate diagnostic.
2406  */
2407 static Z_NamePlusRecord *surrogatediagrec(association *assoc, 
2408                                           const char *dbname,
2409                                           int error, const char *addinfo)
2410 {
2411     yaz_log(log_requestdetail, "[%d] %s%s%s", error, diagbib1_str(error),
2412             addinfo ? " -- " : "", addinfo ? addinfo : "");
2413     return zget_surrogateDiagRec(assoc->encode, dbname, error, addinfo);
2414 }
2415
2416 static Z_Records *pack_records(association *a, char *setname, Odr_int start,
2417                                Odr_int *num, Z_RecordComposition *comp,
2418                                Odr_int *next, Odr_int *pres,
2419                                Z_ReferenceId *referenceId,
2420                                Odr_oid *oid, int *errcode)
2421 {
2422     int recno, total_length = 0, dumped_records = 0;
2423     int toget = odr_int_to_int(*num);
2424     Z_Records *records =
2425         (Z_Records *) odr_malloc(a->encode, sizeof(*records));
2426     Z_NamePlusRecordList *reclist =
2427         (Z_NamePlusRecordList *) odr_malloc(a->encode, sizeof(*reclist));
2428     Z_NamePlusRecord **list =
2429         (Z_NamePlusRecord **) odr_malloc(a->encode, sizeof(*list) * toget);
2430
2431     records->which = Z_Records_DBOSD;
2432     records->u.databaseOrSurDiagnostics = reclist;
2433     reclist->num_records = 0;
2434     reclist->records = list;
2435     *pres = Z_PresentStatus_success;
2436     *num = 0;
2437     *next = 0;
2438
2439     yaz_log(log_requestdetail, "Request to pack " ODR_INT_PRINTF "+%d %s", start, toget, setname);
2440     yaz_log(log_requestdetail, "pms=%d, mrs=%d", a->preferredMessageSize,
2441         a->maximumRecordSize);
2442     for (recno = odr_int_to_int(start); reclist->num_records < toget; recno++)
2443     {
2444         bend_fetch_rr freq;
2445         Z_NamePlusRecord *thisrec;
2446         int this_length = 0;
2447         /*
2448          * we get the number of bytes allocated on the stream before any
2449          * allocation done by the backend - this should give us a reasonable
2450          * idea of the total size of the data so far.
2451          */
2452         total_length = odr_total(a->encode) - dumped_records;
2453         freq.errcode = 0;
2454         freq.errstring = 0;
2455         freq.basename = 0;
2456         freq.len = 0;
2457         freq.record = 0;
2458         freq.last_in_set = 0;
2459         freq.setname = setname;
2460         freq.surrogate_flag = 0;
2461         freq.number = recno;
2462         freq.comp = comp;
2463         freq.request_format = oid;
2464         freq.output_format = 0;
2465         freq.stream = a->encode;
2466         freq.print = a->print;
2467         freq.referenceId = referenceId;
2468         freq.schema = 0;
2469
2470         retrieve_fetch(a, &freq);
2471
2472         *next = freq.last_in_set ? 0 : recno + 1;
2473
2474         if (freq.errcode)
2475         {
2476             if (!freq.surrogate_flag) /* non-surrogate diagnostic i.e. global */
2477             {
2478                 char s[20];
2479                 *pres = Z_PresentStatus_failure;
2480                 /* for 'present request out of range',
2481                    set addinfo to record position if not set */
2482                 if (freq.errcode == YAZ_BIB1_PRESENT_REQUEST_OUT_OF_RANGE  && 
2483                                 freq.errstring == 0)
2484                 {
2485                     sprintf(s, "%d", recno);
2486                     freq.errstring = s;
2487                 }
2488                 if (errcode)
2489                     *errcode = freq.errcode;
2490                 return diagrec(a, freq.errcode, freq.errstring);
2491             }
2492             reclist->records[reclist->num_records] =
2493                 surrogatediagrec(a, freq.basename, freq.errcode,
2494                                  freq.errstring);
2495             reclist->num_records++;
2496             continue;
2497         }
2498         if (freq.record == 0)  /* no error and no record ? */
2499         {
2500             *next = 0;   /* signal end-of-set and stop */
2501             break;
2502         }
2503         if (freq.len >= 0)
2504             this_length = freq.len;
2505         else
2506             this_length = odr_total(a->encode) - total_length - dumped_records;
2507         yaz_log(YLOG_DEBUG, "  fetched record, len=%d, total=%d dumped=%d",
2508             this_length, total_length, dumped_records);
2509         if (a->preferredMessageSize > 0 &&
2510                 this_length + total_length > a->preferredMessageSize)
2511         {
2512             /* record is small enough, really */
2513             if (this_length <= a->preferredMessageSize && recno > start)
2514             {
2515                 yaz_log(log_requestdetail, "  Dropped last normal-sized record");
2516                 *pres = Z_PresentStatus_partial_2;
2517                 break;
2518             }
2519             /* record can only be fetched by itself */
2520             if (this_length < a->maximumRecordSize)
2521             {
2522                 yaz_log(log_requestdetail, "  Record > prefmsgsz");
2523                 if (toget > 1)
2524                 {
2525                     yaz_log(YLOG_DEBUG, "  Dropped it");
2526                     reclist->records[reclist->num_records] =
2527                          surrogatediagrec(a, freq.basename, 16, 0);
2528                     reclist->num_records++;
2529                     dumped_records += this_length;
2530                     continue;
2531                 }
2532             }
2533             else /* too big entirely */
2534             {
2535                 yaz_log(log_requestdetail, "Record > maxrcdsz "
2536                         "this=%d max=%d",
2537                         this_length, a->maximumRecordSize);
2538                 reclist->records[reclist->num_records] =
2539                     surrogatediagrec(a, freq.basename, 17, 0);
2540                 reclist->num_records++;
2541                 dumped_records += this_length;
2542                 continue;
2543             }
2544         }
2545
2546         if (!(thisrec = (Z_NamePlusRecord *)
2547               odr_malloc(a->encode, sizeof(*thisrec))))
2548             return 0;
2549         thisrec->databaseName = odr_strdup_null(a->encode, freq.basename);
2550         thisrec->which = Z_NamePlusRecord_databaseRecord;
2551
2552         if (!freq.output_format)
2553             freq.output_format = freq.request_format;
2554         thisrec->u.databaseRecord = z_ext_record_oid(
2555             a->encode, freq.output_format, freq.record, freq.len);
2556         if (!thisrec->u.databaseRecord)
2557             return 0;
2558         reclist->records[reclist->num_records] = thisrec;
2559         reclist->num_records++;
2560     }
2561     *num = reclist->num_records;
2562     return records;
2563 }
2564
2565 static Z_APDU *process_searchRequest(association *assoc, request *reqb)
2566 {
2567     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
2568     bend_search_rr *bsrr = 
2569         (bend_search_rr *)nmem_malloc(reqb->request_mem, sizeof(*bsrr));
2570     
2571     yaz_log(log_requestdetail, "Got SearchRequest.");
2572     bsrr->association = assoc;
2573     bsrr->referenceId = req->referenceId;
2574     bsrr->srw_sortKeys = 0;
2575     bsrr->srw_setname = 0;
2576     bsrr->srw_setnameIdleTime = 0;
2577     bsrr->estimated_hit_count = 0;
2578     bsrr->partial_resultset = 0;
2579     bsrr->extra_args = 0;
2580     bsrr->extra_response_data = 0;
2581
2582     yaz_log (log_requestdetail, "ResultSet '%s'", req->resultSetName);
2583     if (req->databaseNames)
2584     {
2585         int i;
2586         for (i = 0; i < req->num_databaseNames; i++)
2587             yaz_log(log_requestdetail, "Database '%s'", req->databaseNames[i]);
2588     }
2589
2590     yaz_log_zquery_level(log_requestdetail,req->query);
2591
2592     if (assoc->init->bend_search)
2593     {
2594         bsrr->setname = req->resultSetName;
2595         bsrr->replace_set = *req->replaceIndicator;
2596         bsrr->num_bases = req->num_databaseNames;
2597         bsrr->basenames = req->databaseNames;
2598         bsrr->query = req->query;
2599         bsrr->stream = assoc->encode;
2600         nmem_transfer(odr_getmem(bsrr->stream), reqb->request_mem);
2601         bsrr->decode = assoc->decode;
2602         bsrr->print = assoc->print;
2603         bsrr->hits = 0;
2604         bsrr->errcode = 0;
2605         bsrr->errstring = NULL;
2606         bsrr->search_info = NULL;
2607
2608         if (assoc->server && assoc->server->cql_transform 
2609             && req->query->which == Z_Query_type_104
2610             && req->query->u.type_104->which == Z_External_CQL)
2611         {
2612             /* have a CQL query and a CQL to PQF transform .. */
2613             int srw_errcode = 
2614                 cql2pqf(bsrr->stream, req->query->u.type_104->u.cql,
2615                         assoc->server->cql_transform, bsrr->query);
2616             if (srw_errcode)
2617                 bsrr->errcode = yaz_diag_srw_to_bib1(srw_errcode);
2618         }
2619
2620         if (assoc->server && assoc->server->ccl_transform 
2621             && req->query->which == Z_Query_type_2) /*CCL*/
2622         {
2623             /* have a CCL query and a CCL to PQF transform .. */
2624             int srw_errcode = 
2625                 ccl2pqf(bsrr->stream, req->query->u.type_2,
2626                         assoc->server->ccl_transform, bsrr);
2627             if (srw_errcode)
2628                 bsrr->errcode = yaz_diag_srw_to_bib1(srw_errcode);
2629         }
2630
2631         if (!bsrr->errcode)
2632             (assoc->init->bend_search)(assoc->backend, bsrr);
2633     }
2634     else
2635     { 
2636         /* FIXME - make a diagnostic for it */
2637         yaz_log(YLOG_WARN,"Search not supported ?!?!");
2638     }
2639     return response_searchRequest(assoc, reqb, bsrr);
2640 }
2641
2642 /*
2643  * Prepare a searchresponse based on the backend results. We probably want
2644  * to look at making the fetching of records nonblocking as well, but
2645  * so far, we'll keep things simple.
2646  * If bsrt is null, that means we're called in response to a communications
2647  * event, and we'll have to get the response for ourselves.
2648  */
2649 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
2650                                       bend_search_rr *bsrt)
2651 {
2652     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
2653     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
2654     Z_SearchResponse *resp = (Z_SearchResponse *)
2655         odr_malloc(assoc->encode, sizeof(*resp));
2656     Odr_int *nulint = odr_intdup(assoc->encode, 0);
2657     Odr_int *next = odr_intdup(assoc->encode, 0);
2658     Odr_int *none = odr_intdup(assoc->encode, Z_SearchResponse_none);
2659     Odr_int returnedrecs = 0;
2660
2661     apdu->which = Z_APDU_searchResponse;
2662     apdu->u.searchResponse = resp;
2663     resp->referenceId = req->referenceId;
2664     resp->additionalSearchInfo = 0;
2665     resp->otherInfo = 0;
2666     if (!bsrt)
2667     {
2668         yaz_log(YLOG_FATAL, "Bad result from backend");
2669         return 0;
2670     }
2671     else if (bsrt->errcode)
2672     {
2673         resp->records = diagrec(assoc, bsrt->errcode, bsrt->errstring);
2674         resp->resultCount = nulint;
2675         resp->numberOfRecordsReturned = nulint;
2676         resp->nextResultSetPosition = nulint;
2677         resp->searchStatus = odr_booldup(assoc->encode, 0);
2678         resp->resultSetStatus = none;
2679         resp->presentStatus = 0;
2680     }
2681     else
2682     {
2683         bool_t *sr = odr_booldup(assoc->encode, 1);
2684         Odr_int *toget = odr_intdup(assoc->encode, 0);
2685         Z_RecordComposition comp, *compp = 0;
2686
2687         yaz_log(log_requestdetail, "resultCount: " ODR_INT_PRINTF, bsrt->hits);
2688
2689         resp->records = 0;
2690         resp->resultCount = &bsrt->hits;
2691
2692         comp.which = Z_RecordComp_simple;
2693         /* how many records does the user agent want, then? */
2694         if (bsrt->hits < 0)
2695             *toget = 0;
2696         else if (bsrt->hits <= *req->smallSetUpperBound)
2697         {
2698             *toget = bsrt->hits;
2699             if ((comp.u.simple = req->smallSetElementSetNames))
2700                 compp = &comp;
2701         }
2702         else if (bsrt->hits < *req->largeSetLowerBound)
2703         {
2704             *toget = *req->mediumSetPresentNumber;
2705             if (*toget > bsrt->hits)
2706                 *toget = bsrt->hits;
2707             if ((comp.u.simple = req->mediumSetElementSetNames))
2708                 compp = &comp;
2709         }
2710         else
2711             *toget = 0;
2712
2713         if (*toget && !resp->records)
2714         {
2715             Odr_int *presst = odr_intdup(assoc->encode, 0);
2716             /* Call bend_present if defined */
2717             if (assoc->init->bend_present)
2718             {
2719                 bend_present_rr *bprr = (bend_present_rr *)
2720                     nmem_malloc(reqb->request_mem, sizeof(*bprr));
2721                 bprr->setname = req->resultSetName;
2722                 bprr->start = 1;
2723                 bprr->number = odr_int_to_int(*toget);
2724                 bprr->format = req->preferredRecordSyntax;
2725                 bprr->comp = compp;
2726                 bprr->referenceId = req->referenceId;
2727                 bprr->stream = assoc->encode;
2728                 bprr->print = assoc->print;
2729                 bprr->association = assoc;
2730                 bprr->errcode = 0;
2731                 bprr->errstring = NULL;
2732                 (*assoc->init->bend_present)(assoc->backend, bprr);
2733
2734                 if (bprr->errcode)
2735                 {
2736                     resp->records = diagrec(assoc, bprr->errcode, bprr->errstring);
2737                     *resp->presentStatus = Z_PresentStatus_failure;
2738                 }
2739             }
2740
2741             if (!resp->records)
2742                 resp->records = pack_records(
2743                     assoc, req->resultSetName, 1,
2744                     toget, compp, next, presst, req->referenceId,
2745                     req->preferredRecordSyntax, NULL);
2746             if (!resp->records)
2747                 return 0;
2748             resp->numberOfRecordsReturned = toget;
2749             returnedrecs = *toget;
2750             resp->presentStatus = presst;
2751         }
2752         else
2753         {
2754             if (*resp->resultCount)
2755                 *next = 1;
2756             resp->numberOfRecordsReturned = nulint;
2757             resp->presentStatus = 0;
2758         }
2759         resp->nextResultSetPosition = next;
2760         resp->searchStatus = sr;
2761         resp->resultSetStatus = 0;
2762         if (bsrt->estimated_hit_count)
2763         {
2764             resp->resultSetStatus = odr_intdup(assoc->encode, 
2765                                                Z_SearchResponse_estimate);
2766         }
2767         else if (bsrt->partial_resultset)
2768         {
2769             resp->resultSetStatus = odr_intdup(assoc->encode, 
2770                                                Z_SearchResponse_subset);
2771         }
2772     }
2773     resp->additionalSearchInfo = bsrt->search_info;
2774
2775     if (log_request)
2776     {
2777         int i;
2778         WRBUF wr = wrbuf_alloc();
2779
2780         for (i = 0 ; i < req->num_databaseNames; i++)
2781         {
2782             if (i)
2783                 wrbuf_printf(wr, "+");
2784             wrbuf_puts(wr, req->databaseNames[i]);
2785         }
2786         wrbuf_printf(wr, " ");
2787         
2788         if (bsrt->errcode)
2789             wrbuf_printf(wr, "ERROR %d", bsrt->errcode);
2790         else
2791             wrbuf_printf(wr, "OK " ODR_INT_PRINTF, bsrt->hits);
2792         wrbuf_printf(wr, " %s 1+" ODR_INT_PRINTF " ",
2793                      req->resultSetName, returnedrecs);
2794         yaz_query_to_wrbuf(wr, req->query);
2795         
2796         yaz_log(log_request, "Search %s", wrbuf_cstr(wr));
2797         wrbuf_destroy(wr);
2798     }
2799     return apdu;
2800 }
2801
2802 /*
2803  * Maybe we got a little over-friendly when we designed bend_fetch to
2804  * get only one record at a time. Some backends can optimise multiple-record
2805  * fetches, and at any rate, there is some overhead involved in
2806  * all that selecting and hopping around. Problem is, of course, that the
2807  * frontend can't know ahead of time how many records it'll need to
2808  * fill the negotiated PDU size. Annoying. Segmentation or not, Z/SR
2809  * is downright lousy as a bulk data transfer protocol.
2810  *
2811  * To start with, we'll do the fetching of records from the backend
2812  * in one operation: To save some trips in and out of the event-handler,
2813  * and to simplify the interface to pack_records. At any rate, asynch
2814  * operation is more fun in operations that have an unpredictable execution
2815  * speed - which is normally more true for search than for present.
2816  */
2817 static Z_APDU *process_presentRequest(association *assoc, request *reqb)
2818 {
2819     Z_PresentRequest *req = reqb->apdu_request->u.presentRequest;
2820     Z_APDU *apdu;
2821     Z_PresentResponse *resp;
2822     Odr_int *next;
2823     Odr_int *num;
2824     int errcode = 0;
2825     const char *errstring = 0;
2826
2827     yaz_log(log_requestdetail, "Got PresentRequest.");
2828
2829     resp = (Z_PresentResponse *)odr_malloc(assoc->encode, sizeof(*resp));
2830     resp->records = 0;
2831     resp->presentStatus = odr_intdup(assoc->encode, 0);
2832     if (assoc->init->bend_present)
2833     {
2834         bend_present_rr *bprr = (bend_present_rr *)
2835             nmem_malloc(reqb->request_mem, sizeof(*bprr));
2836         bprr->setname = req->resultSetId;
2837         bprr->start = odr_int_to_int(*req->resultSetStartPoint);
2838         bprr->number = odr_int_to_int(*req->numberOfRecordsRequested);
2839         bprr->format = req->preferredRecordSyntax;
2840         bprr->comp = req->recordComposition;
2841         bprr->referenceId = req->referenceId;
2842         bprr->stream = assoc->encode;
2843         bprr->print = assoc->print;
2844         bprr->association = assoc;
2845         bprr->errcode = 0;
2846         bprr->errstring = NULL;
2847         (*assoc->init->bend_present)(assoc->backend, bprr);
2848         
2849         if (bprr->errcode)
2850         {
2851             resp->records = diagrec(assoc, bprr->errcode, bprr->errstring);
2852             *resp->presentStatus = Z_PresentStatus_failure;
2853             errcode = bprr->errcode;
2854             errstring = bprr->errstring;
2855         }
2856     }
2857     apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
2858     next = odr_intdup(assoc->encode, 0);
2859     num = odr_intdup(assoc->encode, 0);
2860     
2861     apdu->which = Z_APDU_presentResponse;
2862     apdu->u.presentResponse = resp;
2863     resp->referenceId = req->referenceId;
2864     resp->otherInfo = 0;
2865     
2866     if (!resp->records)
2867     {
2868         *num = *req->numberOfRecordsRequested;
2869         resp->records =
2870             pack_records(assoc, req->resultSetId, *req->resultSetStartPoint,
2871                          num, req->recordComposition, next,
2872                          resp->presentStatus,
2873                          req->referenceId, req->preferredRecordSyntax, 
2874                          &errcode);
2875     }
2876     if (log_request)
2877     {
2878         WRBUF wr = wrbuf_alloc();
2879         wrbuf_printf(wr, "Present ");
2880
2881         if (*resp->presentStatus == Z_PresentStatus_failure)
2882             wrbuf_printf(wr, "ERROR %d ", errcode);
2883         else if (*resp->presentStatus == Z_PresentStatus_success)
2884             wrbuf_printf(wr, "OK -  ");
2885         else
2886             wrbuf_printf(wr, "Partial " ODR_INT_PRINTF " - ",
2887                          *resp->presentStatus);
2888
2889         wrbuf_printf(wr, " %s " ODR_INT_PRINTF "+" ODR_INT_PRINTF " ",
2890                 req->resultSetId, *req->resultSetStartPoint,
2891                 *req->numberOfRecordsRequested);
2892         yaz_log(log_request, "%s", wrbuf_cstr(wr) );
2893         wrbuf_destroy(wr);
2894     }
2895     if (!resp->records)
2896         return 0;
2897     resp->numberOfRecordsReturned = num;
2898     resp->nextResultSetPosition = next;
2899     
2900     return apdu;
2901 }
2902
2903 /*
2904  * Scan was implemented rather in a hurry, and with support for only the basic
2905  * elements of the service in the backend API. Suggestions are welcome.
2906  */
2907 static Z_APDU *process_scanRequest(association *assoc, request *reqb)
2908 {
2909     Z_ScanRequest *req = reqb->apdu_request->u.scanRequest;
2910     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
2911     Z_ScanResponse *res = (Z_ScanResponse *)
2912         odr_malloc(assoc->encode, sizeof(*res));
2913     Odr_int *scanStatus = odr_intdup(assoc->encode, Z_Scan_failure);
2914     Odr_int *numberOfEntriesReturned = odr_intdup(assoc->encode, 0);
2915     Z_ListEntries *ents = (Z_ListEntries *)
2916         odr_malloc(assoc->encode, sizeof(*ents));
2917     Z_DiagRecs *diagrecs_p = NULL;
2918     bend_scan_rr *bsrr = (bend_scan_rr *)
2919         odr_malloc(assoc->encode, sizeof(*bsrr));
2920     struct scan_entry *save_entries;
2921
2922     yaz_log(log_requestdetail, "Got ScanRequest");
2923
2924     apdu->which = Z_APDU_scanResponse;
2925     apdu->u.scanResponse = res;
2926     res->referenceId = req->referenceId;
2927
2928     /* if step is absent, set it to 0 */
2929     res->stepSize = odr_intdup(assoc->encode, 0);
2930     if (req->stepSize)
2931         *res->stepSize = *req->stepSize;
2932
2933     res->scanStatus = scanStatus;
2934     res->numberOfEntriesReturned = numberOfEntriesReturned;
2935     res->positionOfTerm = 0;
2936     res->entries = ents;
2937     ents->num_entries = 0;
2938     ents->entries = NULL;
2939     ents->num_nonsurrogateDiagnostics = 0;
2940     ents->nonsurrogateDiagnostics = NULL;
2941     res->attributeSet = 0;
2942     res->otherInfo = 0;
2943
2944     if (req->databaseNames)
2945     {
2946         int i;
2947         for (i = 0; i < req->num_databaseNames; i++)
2948             yaz_log(log_requestdetail, "Database '%s'", req->databaseNames[i]);
2949     }
2950     bsrr->scanClause = 0;
2951     bsrr->errcode = 0;
2952     bsrr->errstring = 0;
2953     bsrr->num_bases = req->num_databaseNames;
2954     bsrr->basenames = req->databaseNames;
2955     bsrr->num_entries = odr_int_to_int(*req->numberOfTermsRequested);
2956     bsrr->term = req->termListAndStartPoint;
2957     bsrr->referenceId = req->referenceId;
2958     bsrr->stream = assoc->encode;
2959     bsrr->print = assoc->print;
2960     bsrr->step_size = res->stepSize;
2961     bsrr->setname = yaz_oi_get_string_oid(&req->otherInfo, 
2962                                           yaz_oid_userinfo_scan_set, 1, 0);
2963     bsrr->entries = 0;
2964     /* For YAZ 2.0 and earlier it was the backend handler that
2965        initialized entries (member display_term did not exist)
2966        YAZ 2.0 and later sets 'entries'  and initialize all members
2967        including 'display_term'. If YAZ 2.0 or later sees that
2968        entries was modified - we assume that it is an old handler and
2969        that 'display_term' is _not_ set.
2970     */
2971     if (bsrr->num_entries > 0) 
2972     {
2973         int i;
2974         bsrr->entries = (struct scan_entry *)
2975             odr_malloc(assoc->decode, sizeof(*bsrr->entries) *
2976                        bsrr->num_entries);
2977         for (i = 0; i<bsrr->num_entries; i++)
2978         {
2979             bsrr->entries[i].term = 0;
2980             bsrr->entries[i].occurrences = 0;
2981             bsrr->entries[i].errcode = 0;
2982             bsrr->entries[i].errstring = 0;
2983             bsrr->entries[i].display_term = 0;
2984         }
2985     }
2986     save_entries = bsrr->entries;  /* save it so we can compare later */
2987
2988     bsrr->attributeset = req->attributeSet;
2989     log_scan_term_level(log_requestdetail, req->termListAndStartPoint, 
2990                         bsrr->attributeset);
2991     bsrr->term_position = req->preferredPositionInResponse ?
2992         *req->preferredPositionInResponse : 1;
2993
2994     ((int (*)(void *, bend_scan_rr *))
2995      (*assoc->init->bend_scan))(assoc->backend, bsrr);
2996
2997     if (bsrr->errcode)
2998         diagrecs_p = zget_DiagRecs(assoc->encode,
2999                                    bsrr->errcode, bsrr->errstring);
3000     else
3001     {
3002         int i;
3003         Z_Entry **tab = (Z_Entry **)
3004             odr_malloc(assoc->encode, sizeof(*tab) * bsrr->num_entries);
3005         
3006         if (bsrr->status == BEND_SCAN_PARTIAL)
3007             *scanStatus = Z_Scan_partial_5;
3008         else
3009             *scanStatus = Z_Scan_success;
3010         ents->entries = tab;
3011         ents->num_entries = bsrr->num_entries;
3012         res->numberOfEntriesReturned = odr_intdup(assoc->encode, 
3013                                                    ents->num_entries);
3014         res->positionOfTerm = &bsrr->term_position;
3015         for (i = 0; i < bsrr->num_entries; i++)
3016         {
3017             Z_Entry *e;
3018             Z_TermInfo *t;
3019             Odr_oct *o;
3020             
3021             tab[i] = e = (Z_Entry *)odr_malloc(assoc->encode, sizeof(*e));
3022             if (bsrr->entries[i].occurrences >= 0)
3023             {
3024                 e->which = Z_Entry_termInfo;
3025                 e->u.termInfo = t = (Z_TermInfo *)
3026                     odr_malloc(assoc->encode, sizeof(*t));
3027                 t->suggestedAttributes = 0;
3028                 t->displayTerm = 0;
3029                 if (save_entries == bsrr->entries && 
3030                     bsrr->entries[i].display_term)
3031                 {
3032                     /* the entries was _not_ set by the handler. So it's
3033                        safe to test for new member display_term. It is
3034                        NULL'ed by us.
3035                     */
3036                     t->displayTerm = odr_strdup(assoc->encode,
3037                                                 bsrr->entries[i].display_term);
3038                 }
3039                 t->alternativeTerm = 0;
3040                 t->byAttributes = 0;
3041                 t->otherTermInfo = 0;
3042                 t->globalOccurrences = &bsrr->entries[i].occurrences;
3043                 t->term = (Z_Term *)
3044                     odr_malloc(assoc->encode, sizeof(*t->term));
3045                 t->term->which = Z_Term_general;
3046                 t->term->u.general = o =
3047                     (Odr_oct *)odr_malloc(assoc->encode, sizeof(Odr_oct));
3048                 o->buf = (unsigned char *)
3049                     odr_malloc(assoc->encode, o->len = o->size =
3050                                strlen(bsrr->entries[i].term));
3051                 memcpy(o->buf, bsrr->entries[i].term, o->len);
3052                 yaz_log(YLOG_DEBUG, "  term #%d: '%s' (" ODR_INT_PRINTF ")", i,
3053                          bsrr->entries[i].term, bsrr->entries[i].occurrences);
3054             }
3055             else
3056             {
3057                 Z_DiagRecs *drecs = zget_DiagRecs(assoc->encode,
3058                                                   bsrr->entries[i].errcode,
3059                                                   bsrr->entries[i].errstring);
3060                 assert(drecs->num_diagRecs == 1);
3061                 e->which = Z_Entry_surrogateDiagnostic;
3062                 assert(drecs->diagRecs[0]);
3063                 e->u.surrogateDiagnostic = drecs->diagRecs[0];
3064             }
3065         }
3066     }
3067     if (diagrecs_p)
3068     {
3069         ents->num_nonsurrogateDiagnostics = diagrecs_p->num_diagRecs;
3070         ents->nonsurrogateDiagnostics = diagrecs_p->diagRecs;
3071     }
3072     if (log_request)
3073     {
3074         int i;
3075         WRBUF wr = wrbuf_alloc();
3076         wrbuf_printf(wr, "Scan ");
3077         for (i = 0 ; i < req->num_databaseNames; i++)
3078         {
3079             if (i)
3080                 wrbuf_printf(wr, "+");
3081             wrbuf_puts(wr, req->databaseNames[i]);
3082         }
3083
3084         wrbuf_printf(wr, " ");
3085         
3086         if (bsrr->errcode)
3087             wr_diag(wr, bsrr->errcode, bsrr->errstring);
3088         else
3089             wrbuf_printf(wr, "OK"); 
3090
3091         wrbuf_printf(wr, " " ODR_INT_PRINTF " - " ODR_INT_PRINTF "+" 
3092                      ODR_INT_PRINTF "+" ODR_INT_PRINTF,
3093                      res->numberOfEntriesReturned ?
3094                      *res->numberOfEntriesReturned : 0,
3095                      (req->preferredPositionInResponse ?
3096                       *req->preferredPositionInResponse : 1),
3097                      *req->numberOfTermsRequested,
3098                      (res->stepSize ? *res->stepSize : 1));
3099         
3100         if (bsrr->setname)
3101             wrbuf_printf(wr, "+%s", bsrr->setname);
3102
3103         wrbuf_printf(wr, " ");
3104         yaz_scan_to_wrbuf(wr, req->termListAndStartPoint, 
3105                           bsrr->attributeset);
3106         yaz_log(log_request, "%s", wrbuf_cstr(wr) );
3107         wrbuf_destroy(wr);
3108     }
3109     return apdu;
3110 }
3111
3112 static Z_APDU *process_sortRequest(association *assoc, request *reqb)
3113 {
3114     int i;
3115     Z_SortRequest *req = reqb->apdu_request->u.sortRequest;
3116     Z_SortResponse *res = (Z_SortResponse *)
3117         odr_malloc(assoc->encode, sizeof(*res));
3118     bend_sort_rr *bsrr = (bend_sort_rr *)
3119         odr_malloc(assoc->encode, sizeof(*bsrr));
3120
3121     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
3122
3123     yaz_log(log_requestdetail, "Got SortRequest.");
3124
3125     bsrr->num_input_setnames = req->num_inputResultSetNames;
3126     for (i=0;i<req->num_inputResultSetNames;i++)
3127         yaz_log(log_requestdetail, "Input resultset: '%s'",
3128                 req->inputResultSetNames[i]);
3129     bsrr->input_setnames = req->inputResultSetNames;
3130     bsrr->referenceId = req->referenceId;
3131     bsrr->output_setname = req->sortedResultSetName;
3132     yaz_log(log_requestdetail, "Output resultset: '%s'",
3133                 req->sortedResultSetName);
3134     bsrr->sort_sequence = req->sortSequence;
3135        /*FIXME - dump those sequences too */
3136     bsrr->stream = assoc->encode;
3137     bsrr->print = assoc->print;
3138
3139     bsrr->sort_status = Z_SortResponse_failure;
3140     bsrr->errcode = 0;
3141     bsrr->errstring = 0;
3142     
3143     (*assoc->init->bend_sort)(assoc->backend, bsrr);
3144     
3145     res->referenceId = bsrr->referenceId;
3146     res->sortStatus = odr_intdup(assoc->encode, bsrr->sort_status);
3147     res->resultSetStatus = 0;
3148     if (bsrr->errcode)
3149     {
3150         Z_DiagRecs *dr = zget_DiagRecs(assoc->encode,
3151                                        bsrr->errcode, bsrr->errstring);
3152         res->diagnostics = dr->diagRecs;
3153         res->num_diagnostics = dr->num_diagRecs;
3154     }
3155     else
3156     {
3157         res->num_diagnostics = 0;
3158         res->diagnostics = 0;
3159     }
3160     res->resultCount = 0;
3161     res->otherInfo = 0;
3162
3163     apdu->which = Z_APDU_sortResponse;
3164     apdu->u.sortResponse = res;
3165     if (log_request)
3166     {
3167         WRBUF wr = wrbuf_alloc();
3168         wrbuf_printf(wr, "Sort ");
3169         if (bsrr->errcode)
3170             wrbuf_printf(wr, " ERROR %d", bsrr->errcode);
3171         else
3172             wrbuf_printf(wr,  "OK -");
3173         wrbuf_printf(wr, " (");
3174         for (i = 0; i<req->num_inputResultSetNames; i++)
3175         {
3176             if (i)
3177                 wrbuf_printf(wr, "+");
3178             wrbuf_puts(wr, req->inputResultSetNames[i]);
3179         }
3180         wrbuf_printf(wr, ")->%s ",req->sortedResultSetName);
3181
3182         yaz_log(log_request, "%s", wrbuf_cstr(wr) );
3183         wrbuf_destroy(wr);
3184     }
3185     return apdu;
3186 }
3187
3188 static Z_APDU *process_deleteRequest(association *assoc, request *reqb)
3189 {
3190     int i;
3191     Z_DeleteResultSetRequest *req =
3192         reqb->apdu_request->u.deleteResultSetRequest;
3193     Z_DeleteResultSetResponse *res = (Z_DeleteResultSetResponse *)
3194         odr_malloc(assoc->encode, sizeof(*res));
3195     bend_delete_rr *bdrr = (bend_delete_rr *)
3196         odr_malloc(assoc->encode, sizeof(*bdrr));
3197     Z_APDU *apdu = (Z_APDU *)odr_malloc(assoc->encode, sizeof(*apdu));
3198
3199     yaz_log(log_requestdetail, "Got DeleteRequest.");
3200
3201     bdrr->num_setnames = req->num_resultSetList;
3202     bdrr->setnames = req->resultSetList;
3203     for (i = 0; i<req->num_resultSetList; i++)
3204         yaz_log(log_requestdetail, "resultset: '%s'",
3205                 req->resultSetList[i]);
3206     bdrr->stream = assoc->encode;
3207     bdrr->print = assoc->print;
3208     bdrr->function = odr_int_to_int(*req->deleteFunction);
3209     bdrr->referenceId = req->referenceId;
3210     bdrr->statuses = 0;
3211     if (bdrr->num_setnames > 0)
3212     {
3213         bdrr->statuses = (Odr_int*) 
3214             odr_malloc(assoc->encode, sizeof(*bdrr->statuses) *
3215                        bdrr->num_setnames);
3216         for (i = 0; i < bdrr->num_setnames; i++)
3217             bdrr->statuses[i] = 0;
3218     }
3219     (*assoc->init->bend_delete)(assoc->backend, bdrr);
3220     
3221     res->referenceId = req->referenceId;
3222
3223     res->deleteOperationStatus = odr_intdup(assoc->encode,bdrr->delete_status);
3224
3225     res->deleteListStatuses = 0;
3226     if (bdrr->num_setnames > 0)
3227     {
3228         int i;
3229         res->deleteListStatuses = (Z_ListStatuses *)
3230             odr_malloc(assoc->encode, sizeof(*res->deleteListStatuses));
3231         res->deleteListStatuses->num = bdrr->num_setnames;
3232         res->deleteListStatuses->elements =
3233             (Z_ListStatus **)
3234             odr_malloc(assoc->encode, 
3235                         sizeof(*res->deleteListStatuses->elements) *
3236                         bdrr->num_setnames);
3237         for (i = 0; i<bdrr->num_setnames; i++)
3238         {
3239             res->deleteListStatuses->elements[i] =
3240                 (Z_ListStatus *)
3241                 odr_malloc(assoc->encode,
3242                             sizeof(**res->deleteListStatuses->elements));
3243             res->deleteListStatuses->elements[i]->status = 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