src/seshigh.c

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