server/seshigh.c

   1 /*
   2  * Copyright (c) 1995-2003, Index Data
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: seshigh.c,v 1.149 2003-03-18 13:34:36 adam Exp $
   6  */
   7
   8 /*
   9  * Frontend server logic.
  10  *
  11  * This code receives incoming APDUs, and handles client requests by means
  12  * of the backend API.
  13  *
  14  * Some of the code is getting quite involved, compared to simpler servers -
  15  * primarily because it is asynchronous both in the communication with
  16  * the user and the backend. We think the complexity will pay off in
  17  * the form of greater flexibility when more asynchronous facilities
  18  * are implemented.
  19  *
  20  * Memory management has become somewhat involved. In the simple case, where
  21  * only one PDU is pending at a time, it will simply reuse the same memory,
  22  * once it has found its working size. When we enable multiple concurrent
  23  * operations, perhaps even with multiple parallel calls to the backend, it
  24  * will maintain a pool of buffers for encoding and decoding, trying to
  25  * minimize memory allocation/deallocation during normal operation.
  26  *
  27  */
  28
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include <sys/types.h>
  32 #ifdef WIN32
  33 #include <io.h>
  34 #define S_ISREG(x) (x & _S_IFREG)
  35 #include <process.h>
  36 #include <sys/stat.h>
  37 #else
  38 #include <sys/stat.h>
  39 #include <unistd.h>
  40 #endif
  41 #include <assert.h>
  42 #include <ctype.h>
  43
  44 #include <yaz/yconfig.h>
  45 #include <yaz/xmalloc.h>
  46 #include <yaz/comstack.h>
  47 #include "eventl.h"
  48 #include "session.h"
  49 #include <yaz/proto.h>
  50 #include <yaz/oid.h>
  51 #include <yaz/log.h>
  52 #include <yaz/logrpn.h>
  53 #include <yaz/statserv.h>
  54 #include <yaz/diagbib1.h>
  55 #include <yaz/charneg.h>
  56 #include <yaz/otherinfo.h>
  57 #include <yaz/yaz-util.h>
  58 #include <yaz/pquery.h>
  59
  60 #include <yaz/srw.h>
  61 #include <yaz/backend.h>
  62
  63 static void process_gdu_request(association *assoc, request *req);
  64 static int process_z_request(association *assoc, request *req, char **msg);
  65 void backend_response(IOCHAN i, int event);
  66 static int process_gdu_response(association *assoc, request *req, Z_GDU *res);
  67 static int process_z_response(association *assoc, request *req, Z_APDU *res);
  68 static Z_APDU *process_initRequest(association *assoc, request *reqb);
  69 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
  70     int *fd);
  71 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
  72     bend_search_rr *bsrr, int *fd);
  73 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
  74     int *fd);
  75 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd);
  76 static Z_APDU *process_sortRequest(association *assoc, request *reqb, int *fd);
  77 static void process_close(association *assoc, request *reqb);
  78 void save_referenceId (request *reqb, Z_ReferenceId *refid);
  79 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
  80     int *fd);
  81 static Z_APDU *process_segmentRequest (association *assoc, request *reqb);
  82
  83 static FILE *apduf = 0; /* for use in static mode */
  84 static statserv_options_block *control_block = 0;
  85
  86 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd);
  87
  88 /*
  89  * Create and initialize a new association-handle.
  90  *  channel  : iochannel for the current line.
  91  *  link     : communications channel.
  92  * Returns: 0 or a new association handle.
  93  */
  94 association *create_association(IOCHAN channel, COMSTACK link)
  95 {
  96     association *anew;
  97
  98     if (!control_block)
  99         control_block = statserv_getcontrol();
 100     if (!(anew = (association *)xmalloc(sizeof(*anew))))
 101         return 0;
 102     anew->init = 0;
 103     anew->version = 0;
 104     anew->client_chan = channel;
 105     anew->client_link = link;
 106     anew->cs_get_mask = 0;
 107     anew->cs_put_mask = 0;
 108     anew->cs_accept_mask = 0;
 109     if (!(anew->decode = odr_createmem(ODR_DECODE)) ||
 110         !(anew->encode = odr_createmem(ODR_ENCODE)))
 111         return 0;
 112     if (*control_block->apdufile)
 113     {
 114         char filename[256];
 115         FILE *f;
 116
 117         strcpy(filename, control_block->apdufile);
 118         if (!(anew->print = odr_createmem(ODR_PRINT)))
 119             return 0;
 120         if (*control_block->apdufile == '@')
 121         {
 122             odr_setprint(anew->print, yaz_log_file());
 123         }
 124         else if (*control_block->apdufile != '-')
 125         {
 126             strcpy(filename, control_block->apdufile);
 127             if (!control_block->dynamic)
 128             {
 129                 if (!apduf)
 130                 {
 131                     if (!(apduf = fopen(filename, "w")))
 132                     {
 133                         yaz_log(LOG_WARN|LOG_ERRNO, "%s", filename);
 134                         return 0;
 135                     }
 136                     setvbuf(apduf, 0, _IONBF, 0);
 137                 }
 138                 f = apduf;
 139             }
 140             else
 141             {
 142                 sprintf(filename + strlen(filename), ".%d", getpid());
 143                 if (!(f = fopen(filename, "w")))
 144                 {
 145                     yaz_log(LOG_WARN|LOG_ERRNO, "%s", filename);
 146                     return 0;
 147                 }
 148                 setvbuf(f, 0, _IONBF, 0);
 149             }
 150             odr_setprint(anew->print, f);
 151         }
 152     }
 153     else
 154         anew->print = 0;
 155     anew->input_buffer = 0;
 156     anew->input_buffer_len = 0;
 157     anew->backend = 0;
 158     anew->state = ASSOC_NEW;
 159     request_initq(&anew->incoming);
 160     request_initq(&anew->outgoing);
 161     anew->proto = cs_getproto(link);
 162     return anew;
 163 }
 164
 165 /*
 166  * Free association and release resources.
 167  */
 168 void destroy_association(association *h)
 169 {
 170     statserv_options_block *cb = statserv_getcontrol();
 171
 172     xfree(h->init);
 173     odr_destroy(h->decode);
 174     odr_destroy(h->encode);
 175     if (h->print)
 176         odr_destroy(h->print);
 177     if (h->input_buffer)
 178     xfree(h->input_buffer);
 179     if (h->backend)
 180         (*cb->bend_close)(h->backend);
 181     while (request_deq(&h->incoming));
 182     while (request_deq(&h->outgoing));
 183     request_delq(&h->incoming);
 184     request_delq(&h->outgoing);
 185     xfree(h);
 186     xmalloc_trav("session closed");
 187     if (control_block && control_block->one_shot)
 188     {
 189         exit (0);
 190     }
 191 }
 192
 193 static void do_close_req(association *a, int reason, char *message,
 194                          request *req)
 195 {
 196     Z_APDU apdu;
 197     Z_Close *cls = zget_Close(a->encode);
 198
 199     /* Purge request queue */
 200     while (request_deq(&a->incoming));
 201     while (request_deq(&a->outgoing));
 202     if (a->version >= 3)
 203     {
 204         yaz_log(LOG_LOG, "Sending Close PDU, reason=%d, message=%s",
 205             reason, message ? message : "none");
 206         apdu.which = Z_APDU_close;
 207         apdu.u.close = cls;
 208         *cls->closeReason = reason;
 209         cls->diagnosticInformation = message;
 210         process_z_response(a, req, &apdu);
 211         iochan_settimeout(a->client_chan, 20);
 212     }
 213     else
 214     {
 215         yaz_log(LOG_DEBUG, "v2 client. No Close PDU");
 216         iochan_setevent(a->client_chan, EVENT_TIMEOUT); /* force imm close */
 217     }
 218     a->state = ASSOC_DEAD;
 219 }
 220
 221 static void do_close(association *a, int reason, char *message)
 222 {
 223     do_close_req (a, reason, message, request_get(&a->outgoing));
 224 }
 225
 226 /*
 227  * This is where PDUs from the client are read and the further
 228  * processing is initiated. Flow of control moves down through the
 229  * various process_* functions below, until the encoded result comes back up
 230  * to the output handler in here.
 231  *
 232  *  h     : the I/O channel that has an outstanding event.
 233  *  event : the current outstanding event.
 234  */
 235 void ir_session(IOCHAN h, int event)
 236 {
 237     int res;
 238     association *assoc = (association *)iochan_getdata(h);
 239     COMSTACK conn = assoc->client_link;
 240     request *req;
 241
 242     assert(h && conn && assoc);
 243     if (event == EVENT_TIMEOUT)
 244     {
 245         if (assoc->state != ASSOC_UP)
 246         {
 247             yaz_log(LOG_LOG, "Final timeout - closing connection.");
 248             cs_close(conn);
 249             destroy_association(assoc);
 250             iochan_destroy(h);
 251         }
 252         else
 253         {
 254             yaz_log(LOG_LOG, "Session idle too long. Sending close.");
 255             do_close(assoc, Z_Close_lackOfActivity, 0);
 256         }
 257         return;
 258     }
 259     if (event & assoc->cs_accept_mask)
 260     {
 261         yaz_log (LOG_DEBUG, "ir_session (accept)");
 262         if (!cs_accept (conn))
 263         {
 264             yaz_log (LOG_LOG, "accept failed");
 265             destroy_association(assoc);
 266             iochan_destroy(h);
 267         }
 268         iochan_clearflag (h, EVENT_OUTPUT|EVENT_OUTPUT);
 269         if (conn->io_pending)
 270         {   /* cs_accept didn't complete */
 271             assoc->cs_accept_mask =
 272                 ((conn->io_pending & CS_WANT_WRITE) ? EVENT_OUTPUT : 0) |
 273                 ((conn->io_pending & CS_WANT_READ) ? EVENT_INPUT : 0);
 274
 275             iochan_setflag (h, assoc->cs_accept_mask);
 276         }
 277         else
 278         {   /* cs_accept completed. Prepare for reading (cs_get) */
 279             assoc->cs_accept_mask = 0;
 280             assoc->cs_get_mask = EVENT_INPUT;
 281             iochan_setflag (h, assoc->cs_get_mask);
 282         }
 283         return;
 284     }
 285     if ((event & assoc->cs_get_mask) || (event & EVENT_WORK)) /* input */
 286     {
 287         if ((assoc->cs_put_mask & EVENT_INPUT) == 0 && (event & assoc->cs_get_mask))
 288         {
 289             yaz_log(LOG_DEBUG, "ir_session (input)");
 290             /* We aren't speaking to this fellow */
 291             if (assoc->state == ASSOC_DEAD)
 292             {
 293                 yaz_log(LOG_LOG, "Connection closed - end of session");
 294                 cs_close(conn);
 295                 destroy_association(assoc);
 296                 iochan_destroy(h);
 297                 return;
 298             }
 299             assoc->cs_get_mask = EVENT_INPUT;
 300             if ((res = cs_get(conn, &assoc->input_buffer,
 301                 &assoc->input_buffer_len)) <= 0)
 302             {
 303                 yaz_log(LOG_LOG, "Connection closed by client");
 304                 cs_close(conn);
 305                 destroy_association(assoc);
 306                 iochan_destroy(h);
 307                 return;
 308             }
 309             else if (res == 1) /* incomplete read - wait for more  */
 310             {
 311                 if (conn->io_pending & CS_WANT_WRITE)
 312                     assoc->cs_get_mask |= EVENT_OUTPUT;
 313                 iochan_setflag(h, assoc->cs_get_mask);
 314                 return;
 315             }
 316             if (cs_more(conn)) /* more stuff - call us again later, please */
 317                 iochan_setevent(h, EVENT_INPUT);
 318
 319             /* we got a complete PDU. Let's decode it */
 320             yaz_log(LOG_DEBUG, "Got PDU, %d bytes: lead=%02X %02X %02X", res,
 321                             assoc->input_buffer[0] & 0xff,
 322                             assoc->input_buffer[1] & 0xff,
 323                             assoc->input_buffer[2] & 0xff);
 324             req = request_get(&assoc->incoming); /* get a new request */
 325             odr_reset(assoc->decode);
 326             odr_setbuf(assoc->decode, assoc->input_buffer, res, 0);
 327             if (!z_GDU(assoc->decode, &req->gdu_request, 0, 0))
 328             {
 329                 yaz_log(LOG_LOG, "ODR error on incoming PDU: %s [near byte %d] ",
 330                         odr_errmsg(odr_geterror(assoc->decode)),
 331                         odr_offset(assoc->decode));
 332                 if (assoc->decode->error != OHTTP)
 333                 {
 334                     yaz_log(LOG_LOG, "PDU dump:");
 335                     odr_dumpBER(yaz_log_file(), assoc->input_buffer, res);
 336                     do_close(assoc, Z_Close_protocolError, "Malformed package");
 337                 }
 338                 else
 339                 {
 340                     Z_GDU *p = z_get_HTTP_Response(assoc->encode, 400);
 341                     assoc->state = ASSOC_DEAD;
 342                     process_gdu_response(assoc, req, p);
 343                 }
 344                 return;
 345             }
 346             req->request_mem = odr_extract_mem(assoc->decode);
 347             if (assoc->print && !z_GDU(assoc->print, &req->gdu_request, 0, 0))
 348             {
 349                 yaz_log(LOG_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
 356         /* can we do something yet? */
 357         req = request_head(&assoc->incoming);
 358         if (req->state == REQUEST_IDLE)
 359         {
 360             request_deq(&assoc->incoming);
 361             process_gdu_request(assoc, req);
 362         }
 363     }
 364     if (event & assoc->cs_put_mask)
 365     {
 366         request *req = request_head(&assoc->outgoing);
 367
 368         assoc->cs_put_mask = 0;
 369         yaz_log(LOG_DEBUG, "ir_session (output)");
 370         req->state = REQUEST_PENDING;
 371         switch (res = cs_put(conn, req->response, req->len_response))
 372         {
 373         case -1:
 374             yaz_log(LOG_LOG, "Connection closed by client");
 375             cs_close(conn);
 376             destroy_association(assoc);
 377             iochan_destroy(h);
 378             break;
 379         case 0: /* all sent - release the request structure */
 380             yaz_log(LOG_DEBUG, "Wrote PDU, %d bytes", req->len_response);
 381 #if 0
 382             yaz_log(LOG_DEBUG, "HTTP out:\n%.*s", req->len_response,
 383                     req->response);
 384 #endif
 385             nmem_destroy(req->request_mem);
 386             request_deq(&assoc->outgoing);
 387             request_release(req);
 388             if (!request_head(&assoc->outgoing))
 389             {   /* restore mask for cs_get operation ... */
 390                 iochan_clearflag(h, EVENT_OUTPUT|EVENT_INPUT);
 391                 iochan_setflag(h, assoc->cs_get_mask);
 392                 if (assoc->state == ASSOC_DEAD)
 393                     iochan_setevent(assoc->client_chan, EVENT_TIMEOUT);
 394             }
 395             else
 396             {
 397                 assoc->cs_put_mask = EVENT_OUTPUT;
 398             }
 399             break;
 400         default:
 401             if (conn->io_pending & CS_WANT_WRITE)
 402                 assoc->cs_put_mask |= EVENT_OUTPUT;
 403             if (conn->io_pending & CS_WANT_READ)
 404                 assoc->cs_put_mask |= EVENT_INPUT;
 405             iochan_setflag(h, assoc->cs_put_mask);
 406         }
 407     }
 408     if (event & EVENT_EXCEPT)
 409     {
 410         yaz_log(LOG_LOG, "ir_session (exception)");
 411         cs_close(conn);
 412         destroy_association(assoc);
 413         iochan_destroy(h);
 414     }
 415 }
 416
 417 static int process_z_request(association *assoc, request *req, char **msg);
 418
 419 static void assoc_init_reset(association *assoc)
 420 {
 421     xfree (assoc->init);
 422     assoc->init = (bend_initrequest *) xmalloc (sizeof(*assoc->init));
 423
 424     assoc->init->stream = assoc->encode;
 425     assoc->init->print = assoc->print;
 426     assoc->init->auth = 0;
 427     assoc->init->referenceId = 0;
 428     assoc->init->implementation_version = 0;
 429     assoc->init->implementation_id = 0;
 430     assoc->init->implementation_name = 0;
 431     assoc->init->bend_sort = NULL;
 432     assoc->init->bend_search = NULL;
 433     assoc->init->bend_present = NULL;
 434     assoc->init->bend_esrequest = NULL;
 435     assoc->init->bend_delete = NULL;
 436     assoc->init->bend_scan = NULL;
 437     assoc->init->bend_segment = NULL;
 438     assoc->init->bend_fetch = NULL;
 439     assoc->init->charneg_request = NULL;
 440     assoc->init->charneg_response = NULL;
 441     assoc->init->decode = assoc->decode;
 442     assoc->init->peer_name =
 443         odr_strdup (assoc->encode, cs_addrstr(assoc->client_link));
 444 }
 445
 446 static int srw_bend_init(association *assoc)
 447 {
 448     const char *encoding = "UTF-8";
 449     Z_External *ce;
 450     bend_initresult *binitres;
 451     statserv_options_block *cb = statserv_getcontrol();
 452
 453     assoc_init_reset(assoc);
 454
 455     assoc->maximumRecordSize = 3000000;
 456     assoc->preferredMessageSize = 3000000;
 457 #if 1
 458     ce = yaz_set_proposal_charneg(assoc->decode, &encoding, 1, 0, 0, 1);
 459     assoc->init->charneg_request = ce->u.charNeg3;
 460 #endif
 461     if (!(binitres = (*cb->bend_init)(assoc->init)))
 462     {
 463         yaz_log(LOG_WARN, "Bad response from backend.");
 464         return 0;
 465     }
 466     assoc->backend = binitres->handle;
 467     return 1;
 468 }
 469
 470 static int srw_bend_fetch(association *assoc, int pos,
 471                           Z_SRW_searchRetrieveRequest *srw_req,
 472                           Z_SRW_record *record)
 473 {
 474     bend_fetch_rr rr;
 475     ODR o = assoc->encode;
 476
 477     rr.setname = "default";
 478     rr.number = pos;
 479     rr.referenceId = 0;
 480     rr.request_format = VAL_TEXT_XML;
 481     rr.request_format_raw = yaz_oidval_to_z3950oid(assoc->decode,
 482                                                    CLASS_TRANSYN,
 483                                                    VAL_TEXT_XML);
 484     rr.comp = (Z_RecordComposition *)
 485             odr_malloc(assoc->decode, sizeof(*rr.comp));
 486     rr.comp->which = Z_RecordComp_complex;
 487     rr.comp->u.complex = (Z_CompSpec *)
 488             odr_malloc(assoc->decode, sizeof(Z_CompSpec));
 489     rr.comp->u.complex->selectAlternativeSyntax = (bool_t *)
 490         odr_malloc(assoc->encode, sizeof(bool_t));
 491     *rr.comp->u.complex->selectAlternativeSyntax = 0;
 492     rr.comp->u.complex->num_dbSpecific = 0;
 493     rr.comp->u.complex->dbSpecific = 0;
 494     rr.comp->u.complex->num_recordSyntax = 0;
 495     rr.comp->u.complex->recordSyntax = 0;
 496
 497     rr.comp->u.complex->generic = (Z_Specification *)
 498             odr_malloc(assoc->decode, sizeof(Z_Specification));
 499     rr.comp->u.complex->generic->which = Z_Schema_uri;
 500     rr.comp->u.complex->generic->schema.uri = srw_req->recordSchema;
 501     rr.comp->u.complex->generic->elementSpec = 0;
 502
 503     rr.stream = assoc->encode;
 504     rr.print = assoc->print;
 505
 506     rr.basename = 0;
 507     rr.len = 0;
 508     rr.record = 0;
 509     rr.last_in_set = 0;
 510     rr.output_format = VAL_TEXT_XML;
 511     rr.output_format_raw = 0;
 512     rr.errcode = 0;
 513     rr.errstring = 0;
 514     rr.surrogate_flag = 0;
 515
 516     if (!assoc->init->bend_fetch)
 517         return 1;
 518
 519     (*assoc->init->bend_fetch)(assoc->backend, &rr);
 520
 521     if (rr.len >= 0)
 522     {
 523         record->recordData_buf = rr.record;
 524         record->recordData_len = rr.len;
 525         record->recordPosition = odr_intdup(o, pos);
 526         record->recordSchema = 0;
 527         if (srw_req->recordSchema)
 528             record->recordSchema = odr_strdup(o, srw_req->recordSchema);
 529     }
 530     return rr.errcode;
 531 }
 532
 533 static void srw_bend_search(association *assoc, request *req,
 534                             Z_SRW_searchRetrieveRequest *srw_req,
 535                             Z_SRW_searchRetrieveResponse *srw_res)
 536 {
 537     int srw_error = 0;
 538     bend_search_rr rr;
 539     Z_External *ext;
 540
 541     yaz_log(LOG_LOG, "Got SRW SearchRetrieveRequest");
 542     yaz_log(LOG_DEBUG, "srw_bend_search");
 543     if (!assoc->init)
 544     {
 545         yaz_log(LOG_DEBUG, "srw_bend_init");
 546         if (!srw_bend_init(assoc))
 547         {
 548             srw_error = 3;  /* assume Authentication error */
 549
 550             srw_res->num_diagnostics = 1;
 551             srw_res->diagnostics = (Z_SRW_diagnostic *)
 552                 odr_malloc(assoc->encode, sizeof(*srw_res->diagnostics));
 553             srw_res->diagnostics[0].code =
 554                 odr_intdup(assoc->encode, srw_error);
 555             srw_res->diagnostics[0].details = 0;
 556             return;
 557         }
 558     }
 559
 560     rr.setname = "default";
 561     rr.replace_set = 1;
 562     rr.num_bases = 1;
 563     rr.basenames = &srw_req->database;
 564     rr.referenceId = 0;
 565
 566     rr.query = (Z_Query *) odr_malloc (assoc->decode, sizeof(*rr.query));
 567
 568     if (srw_req->query_type == Z_SRW_query_type_cql)
 569     {
 570         ext = (Z_External *) odr_malloc(assoc->decode, sizeof(*ext));
 571         ext->direct_reference = odr_getoidbystr(assoc->decode,
 572                                                 "1.2.840.10003.16.2");
 573         ext->indirect_reference = 0;
 574         ext->descriptor = 0;
 575         ext->which = Z_External_CQL;
 576         ext->u.cql = srw_req->query.cql;
 577
 578         rr.query->which = Z_Query_type_104;
 579         rr.query->u.type_104 =  ext;
 580     }
 581     else if (srw_req->query_type == Z_SRW_query_type_pqf)
 582     {
 583         Z_RPNQuery *RPNquery;
 584         YAZ_PQF_Parser pqf_parser;
 585
 586         pqf_parser = yaz_pqf_create ();
 587
 588         RPNquery = yaz_pqf_parse (pqf_parser, assoc->decode,
 589                                   srw_req->query.pqf);
 590         if (!RPNquery)
 591         {
 592             const char *pqf_msg;
 593             size_t off;
 594             int code = yaz_pqf_error (pqf_parser, &pqf_msg, &off);
 595             yaz_log(LOG_LOG, "%*s^\n", off+4, "");
 596             yaz_log(LOG_LOG, "Bad PQF: %s (code %d)\n", pqf_msg, code);
 597
 598             srw_error = 10;
 599         }
 600
 601         rr.query->which = Z_Query_type_1;
 602         rr.query->u.type_1 =  RPNquery;
 603
 604         yaz_pqf_destroy (pqf_parser);
 605     }
 606     else
 607         srw_error = 11;
 608
 609     if (!srw_error && srw_req->sort_type != Z_SRW_sort_type_none)
 610         srw_error = 80;
 611
 612     if (!srw_error && !assoc->init->bend_search)
 613         srw_error = 1;
 614
 615     if (srw_error)
 616     {
 617         yaz_log(LOG_DEBUG, "srw_bend_search returned SRW error %d", srw_error);
 618         srw_res->num_diagnostics = 1;
 619         srw_res->diagnostics = (Z_SRW_diagnostic *)
 620             odr_malloc(assoc->encode, sizeof(*srw_res->diagnostics));
 621         srw_res->diagnostics[0].code =
 622             odr_intdup(assoc->encode, srw_error);
 623         srw_res->diagnostics[0].details = 0;
 624         return;
 625     }
 626
 627     rr.stream = assoc->encode;
 628     rr.decode = assoc->decode;
 629     rr.print = assoc->print;
 630     rr.request = req;
 631     rr.association = assoc;
 632     rr.fd = 0;
 633     rr.hits = 0;
 634     rr.errcode = 0;
 635     rr.errstring = 0;
 636     rr.search_info = 0;
 637     yaz_log_zquery(rr.query);
 638     (assoc->init->bend_search)(assoc->backend, &rr);
 639     srw_res->numberOfRecords = odr_intdup(assoc->encode, rr.hits);
 640     if (rr.errcode)
 641     {
 642         yaz_log(LOG_DEBUG, "bend_search returned Bib-1 code %d", rr.errcode);
 643         srw_res->num_diagnostics = 1;
 644         srw_res->diagnostics = (Z_SRW_diagnostic *)
 645             odr_malloc(assoc->encode, sizeof(*srw_res->diagnostics));
 646         srw_res->diagnostics[0].code =
 647             odr_intdup(assoc->encode,
 648                        yaz_diag_bib1_to_srw (rr.errcode));
 649         srw_res->diagnostics[0].details = rr.errstring;
 650         yaz_log(LOG_DEBUG, "srw_bend_search returned SRW error %d",
 651                 srw_res->diagnostics[0].code);
 652
 653     }
 654     else
 655     {
 656         srw_res->numberOfRecords = odr_intdup(assoc->encode, rr.hits);
 657         if (srw_req->maximumRecords && *srw_req->maximumRecords > 0)
 658         {
 659             int number = *srw_req->maximumRecords;
 660             int start = 1;
 661             int i;
 662
 663             if (srw_req->startRecord)
 664                 start = *srw_req->startRecord;
 665
 666             yaz_log(LOG_DEBUG, "srw_bend_search. start=%d max=%d",
 667                     start, *srw_req->maximumRecords);
 668
 669             if (start <= rr.hits)
 670             {
 671                 int j = 0;
 672                 if (start + number > rr.hits)
 673                     number = rr.hits - start + 1;
 674                 srw_res->records = (Z_SRW_record *)
 675                     odr_malloc(assoc->encode,
 676                                number * sizeof(*srw_res->records));
 677                 for (i = 0; i<number; i++)
 678                 {
 679                     int errcode;
 680                     srw_res->records[j].recordData_buf = 0;
 681                     yaz_log(LOG_DEBUG, "srw_bend_fetch %d", i+start);
 682                     errcode = srw_bend_fetch(assoc, i+start, srw_req,
 683                                              srw_res->records + j);
 684                     if (errcode)
 685                     {
 686                         srw_res->num_diagnostics = 1;
 687                         srw_res->diagnostics = (Z_SRW_diagnostic *)
 688                             odr_malloc(assoc->encode,
 689                                        sizeof(*srw_res->diagnostics));
 690                         srw_res->diagnostics[0].code =
 691                             odr_intdup(assoc->encode,
 692                                        yaz_diag_bib1_to_srw (errcode));
 693                         srw_res->diagnostics[0].details = rr.errstring;
 694                         break;
 695                     }
 696                     if (srw_res->records[j].recordData_buf)
 697                         j++;
 698                 }
 699                 srw_res->num_records = j;
 700                 if (!j)
 701                     srw_res->records = 0;
 702             }
 703         }
 704     }
 705 }
 706
 707 static int hex_digit (int ch)
 708 {
 709     if (ch >= '0' && ch <= '9')
 710         return ch - '0';
 711     else if (ch >= 'a' && ch <= 'f')
 712         return ch - 'a'+10;
 713     else if (ch >= 'A' && ch <= 'F')
 714         return ch - 'A'+10;
 715     return 0;
 716 }
 717
 718 static char *uri_val(const char *path, const char *name, ODR o)
 719 {
 720     size_t nlen = strlen(name);
 721     if (*path != '?')
 722         return 0;
 723     path++;
 724     while (path && *path)
 725     {
 726         const char *p1 = strchr(path, '=');
 727         if (!p1)
 728             break;
 729         if (p1 - path == nlen && !memcmp(path, name, nlen))
 730         {
 731             size_t i = 0;
 732             char *ret;
 733
 734             path = p1 + 1;
 735             p1 = strchr(path, '&');
 736             if (!p1)
 737                 p1 = strlen(path) + path;
 738             ret = odr_malloc(o, p1 - path + 1);
 739             while (*path && *path != '&')
 740             {
 741                 if (*path == '+')
 742                 {
 743                     ret[i++] = ' ';
 744                     path++;
 745                 }
 746                 else if (*path == '%' && path[1] && path[2])
 747                 {
 748                     ret[i++] = hex_digit (path[1])*16 + hex_digit (path[2]);
 749                     path = path + 3;
 750                 }
 751                 else
 752                     ret[i++] = *path++;
 753             }
 754             ret[i] = '\0';
 755             return ret;
 756         }
 757         path = strchr(p1, '&');
 758         if (path)
 759             path++;
 760     }
 761     return 0;
 762 }
 763
 764 void uri_val_int(const char *path, const char *name, ODR o, int **intp)
 765 {
 766     const char *v = uri_val(path, name, o);
 767     if (v)
 768         *intp = odr_intdup(o, atoi(v));
 769 }
 770
 771 static void process_http_request(association *assoc, request *req)
 772 {
 773     Z_HTTP_Request *hreq = req->gdu_request->u.HTTP_Request;
 774     ODR o = assoc->encode;
 775     Z_GDU *p = 0;
 776     Z_HTTP_Response *hres = 0;
 777     int keepalive = 1;
 778
 779     if (!strcmp(hreq->method, "GET"))
 780     {
 781         char *charset = 0;
 782         int ret = -1;
 783         Z_SOAP *soap_package = 0;
 784         char *db = "Default";
 785         const char *p0 = hreq->path, *p1;
 786         static Z_SOAP_Handler soap_handlers[2] = {
 787 #if HAVE_XML2
 788             {"http://www.loc.gov/zing/srw/v1.0/", 0,
 789              (Z_SOAP_fun) yaz_srw_codec},
 790 #endif
 791             {0, 0, 0}
 792         };
 793
 794         if (*p0 == '/')
 795             p0++;
 796         p1 = strchr(p0, '?');
 797         if (!p1)
 798             p1 = p0 + strlen(p0);
 799         if (p1 != p0)
 800         {
 801             db = odr_malloc(assoc->decode, p1 - p0 + 1);
 802             memcpy (db, p0, p1 - p0);
 803             db[p1 - p0] = '\0';
 804         }
 805
 806         if (p1 && *p1 == '?' && p1[1])
 807         {
 808             Z_SRW_PDU *res = yaz_srw_get(o, Z_SRW_searchRetrieve_response);
 809             Z_SRW_PDU *sr = yaz_srw_get(o, Z_SRW_searchRetrieve_request);
 810             char *query = uri_val(p1, "query", o);
 811             char *pQuery = uri_val(p1, "pQuery", o);
 812             char *sortKeys = uri_val(p1, "sortKeys", o);
 813
 814             if (query)
 815             {
 816                 sr->u.request->query_type = Z_SRW_query_type_cql;
 817                 sr->u.request->query.cql = query;
 818             }
 819             if (pQuery)
 820             {
 821                 sr->u.request->query_type = Z_SRW_query_type_pqf;
 822                 sr->u.request->query.pqf = pQuery;
 823             }
 824             if (sortKeys)
 825             {
 826                 sr->u.request->sort_type = Z_SRW_sort_type_sort;
 827                 sr->u.request->sort.sortKeys = sortKeys;
 828             }
 829             sr->u.request->recordSchema = uri_val(p1, "recordSchema", o);
 830             uri_val_int(p1, "maximumRecords", o,
 831                         &sr->u.request->maximumRecords);
 832             uri_val_int(p1, "startRecord", o,
 833                         &sr->u.request->startRecord);
 834             if (sr->u.request->startRecord)
 835                 yaz_log(LOG_LOG, "startRecord=%d", *sr->u.request->startRecord);
 836             sr->u.request->database = db;
 837             srw_bend_search(assoc, req, sr->u.request,
 838                             res->u.response);
 839
 840             soap_package = odr_malloc(o, sizeof(*soap_package));
 841             soap_package->which = Z_SOAP_generic;
 842
 843             soap_package->u.generic =
 844                 odr_malloc(o, sizeof(*soap_package->u.generic));
 845
 846             soap_package->u.generic->p = res;
 847             soap_package->u.generic->ns = soap_handlers[0].ns;
 848             soap_package->u.generic->no = 0;
 849
 850             soap_package->ns = "SRU";
 851
 852             p = z_get_HTTP_Response(o, 200);
 853             hres = p->u.HTTP_Response;
 854
 855             ret = z_soap_codec_enc(assoc->encode, &soap_package,
 856                                    &hres->content_buf, &hres->content_len,
 857                                    soap_handlers, charset);
 858             if (!charset)
 859                 z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/xml");
 860             else
 861             {
 862                 char ctype[60];
 863                 strcpy(ctype, "text/xml; charset=");
 864                 strcat(ctype, charset);
 865                 z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
 866             }
 867
 868         }
 869 #ifdef DOCDIR
 870         if (strlen(hreq->path) >= 5 && strlen(hreq->path) < 80 &&
 871                          !memcmp(hreq->path, "/doc/", 5))
 872         {
 873             FILE *f;
 874             char fpath[120];
 875
 876             strcpy(fpath, DOCDIR);
 877             strcat(fpath, hreq->path+4);
 878             f = fopen(fpath, "rb");
 879             if (f) {
 880                 struct stat sbuf;
 881                 if (fstat(fileno(f), &sbuf) || !S_ISREG(sbuf.st_mode))
 882                 {
 883                     fclose(f);
 884                     f = 0;
 885                 }
 886             }
 887             if (f)
 888             {
 889                 long sz;
 890                 fseek(f, 0L, SEEK_END);
 891                 sz = ftell(f);
 892                 if (sz >= 0 && sz < 500000)
 893                 {
 894                     const char *ctype = "application/octet-stream";
 895                     const char *cp;
 896
 897                     p = z_get_HTTP_Response(o, 200);
 898                     hres = p->u.HTTP_Response;
 899                     hres->content_buf = (char *) odr_malloc(o, sz + 1);
 900                     hres->content_len = sz;
 901                     fseek(f, 0L, SEEK_SET);
 902                     fread(hres->content_buf, 1, sz, f);
 903                     if ((cp = strrchr(fpath, '.'))) {
 904                         cp++;
 905                         if (!strcmp(cp, "png"))
 906                             ctype = "image/png";
 907                         else if (!strcmp(cp, "gif"))
 908                             ctype = "image/gif";
 909                         else if (!strcmp(cp, "xml"))
 910                             ctype = "text/xml";
 911                         else if (!strcmp(cp, "html"))
 912                             ctype = "text/html";
 913                     }
 914                     z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
 915                 }
 916                 fclose(f);
 917             }
 918         }
 919 #endif
 920         if (!strcmp(hreq->path, "/"))
 921         {
 922 #ifdef DOCDIR
 923             struct stat sbuf;
 924 #endif
 925             const char *doclink = "";
 926             p = z_get_HTTP_Response(o, 200);
 927             hres = p->u.HTTP_Response;
 928             hres->content_buf = (char *) odr_malloc(o, 400);
 929 #ifdef DOCDIR
 930             if (stat(DOCDIR "/yaz.html", &sbuf) == 0 && S_ISREG(sbuf.st_mode))
 931                 doclink = "<P><A HREF=\"/doc/yaz.html\">Documentation</A></P>";
 932 #endif
 933             sprintf (hres->content_buf,
 934                      "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\">\n"
 935                      "<HTML>\n"
 936                      " <HEAD>\n"
 937                      "  <TITLE>YAZ " YAZ_VERSION "</TITLE>\n"
 938                      " </HEAD>\n"
 939                      " <BODY>\n"
 940                      "  <P><A HREF=\"http://www.indexdata.dk/yaz/\">YAZ</A> "
 941                      YAZ_VERSION "</P>\n"
 942                      "%s"
 943                      " </BODY>\n"
 944                      "</HTML>\n", doclink);
 945             hres->content_len = strlen(hres->content_buf);
 946             z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/html");
 947         }
 948         if (!p)
 949         {
 950             p = z_get_HTTP_Response(o, 404);
 951         }
 952     }
 953     else if (!strcmp(hreq->method, "POST"))
 954     {
 955         const char *content_type = z_HTTP_header_lookup(hreq->headers,
 956                                                         "Content-Type");
 957         const char *soap_action = z_HTTP_header_lookup(hreq->headers,
 958                                                        "SOAPAction");
 959         if (content_type && soap_action &&
 960             !yaz_strcmp_del("text/xml", content_type, "; "))
 961         {
 962             Z_SOAP *soap_package = 0;
 963             int ret = -1;
 964             int http_code = 500;
 965             const char *charset_p = 0;
 966             char *charset = 0;
 967
 968             static Z_SOAP_Handler soap_handlers[2] = {
 969 #if HAVE_XML2
 970                 {"http://www.loc.gov/zing/srw/v1.0/", 0,
 971                                          (Z_SOAP_fun) yaz_srw_codec},
 972 #endif
 973                 {0, 0, 0}
 974             };
 975             if ((charset_p = strstr(content_type, "; charset=")))
 976             {
 977                 int i = 0;
 978                 charset_p += 10;
 979                 while (i < 20 && charset_p[i] &&
 980                        !strchr("; \n\r", charset_p[i]))
 981                     i++;
 982                 charset = odr_malloc(assoc->encode, i+1);
 983                 memcpy(charset, charset_p, i);
 984                 charset[i] = '\0';
 985                 yaz_log(LOG_LOG, "SOAP encoding %s", charset);
 986             }
 987             ret = z_soap_codec(assoc->decode, &soap_package,
 988                                &hreq->content_buf, &hreq->content_len,
 989                                soap_handlers);
 990
 991 #if HAVE_XML2
 992             if (!ret && soap_package->which == Z_SOAP_generic &&
 993                 soap_package->u.generic->no == 0)
 994             {
 995                 /* SRW package */
 996                 Z_SRW_PDU *sr = soap_package->u.generic->p;
 997
 998                 if (sr->which == Z_SRW_searchRetrieve_request)
 999                 {
1000                     Z_SRW_PDU *res =
1001                         yaz_srw_get(assoc->encode,
1002                                     Z_SRW_searchRetrieve_response);
1003
1004                     if (!sr->u.request->database)
1005                     {
1006                         const char *p0 = hreq->path, *p1;
1007                         if (*p0 == '/')
1008                             p0++;
1009                         p1 = strchr(p0, '?');
1010                         if (!p1)
1011                             p1 = p0 + strlen(p0);
1012                         if (p1 != p0)
1013                         {
1014                             sr->u.request->database =
1015                                 odr_malloc(assoc->decode, p1 - p0 + 1);
1016                             memcpy (sr->u.request->database, p0, p1 - p0);
1017                             sr->u.request->database[p1 - p0] = '\0';
1018                         }
1019                         else
1020                             sr->u.request->database = "Default";
1021                     }
1022                     srw_bend_search(assoc, req, sr->u.request,
1023                                     res->u.response);
1024
1025                     soap_package->u.generic->p = res;
1026                     http_code = 200;
1027                 }
1028             }
1029 #endif
1030             p = z_get_HTTP_Response(o, 200);
1031             hres = p->u.HTTP_Response;
1032             ret = z_soap_codec_enc(assoc->encode, &soap_package,
1033                                    &hres->content_buf, &hres->content_len,
1034                                    soap_handlers, charset);
1035             hres->code = http_code;
1036             if (!charset)
1037                 z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/xml");
1038             else
1039             {
1040                 char ctype[60];
1041                 strcpy(ctype, "text/xml; charset=");
1042                 strcat(ctype, charset);
1043                 z_HTTP_header_add(o, &hres->headers, "Content-Type", ctype);
1044             }
1045         }
1046         if (!p) /* still no response ? */
1047             p = z_get_HTTP_Response(o, 500);
1048     }
1049     else
1050     {
1051         p = z_get_HTTP_Response(o, 405);
1052         hres = p->u.HTTP_Response;
1053
1054         z_HTTP_header_add(o, &hres->headers, "Allow", "GET, POST");
1055     }
1056     hres = p->u.HTTP_Response;
1057     if (!strcmp(hreq->version, "1.0"))
1058     {
1059         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1060         if (v && !strcmp(v, "Keep-Alive"))
1061             keepalive = 1;
1062         else
1063             keepalive = 0;
1064         hres->version = "1.0";
1065     }
1066     else
1067     {
1068         const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
1069         if (v && !strcmp(v, "close"))
1070             keepalive = 0;
1071         else
1072             keepalive = 1;
1073         hres->version = "1.1";
1074     }
1075     if (!keepalive)
1076     {
1077         z_HTTP_header_add(o, &hres->headers, "Connection", "close");
1078         assoc->state = ASSOC_DEAD;
1079     }
1080     else
1081     {
1082         int t;
1083         const char *alive = z_HTTP_header_lookup(hreq->headers, "Keep-Alive");
1084
1085         if (alive && isdigit(*alive))
1086             t = atoi(alive);
1087         else
1088             t = 15;
1089         if (t < 0 || t > 3600)
1090             t = 3600;
1091         iochan_settimeout(assoc->client_chan,t);
1092         z_HTTP_header_add(o, &hres->headers, "Connection", "Keep-Alive");
1093     }
1094     process_gdu_response(assoc, req, p);
1095 }
1096
1097 static void process_gdu_request(association *assoc, request *req)
1098 {
1099     if (req->gdu_request->which == Z_GDU_Z3950)
1100     {
1101         char *msg = 0;
1102         req->apdu_request = req->gdu_request->u.z3950;
1103         if (process_z_request(assoc, req, &msg) < 0)
1104             do_close_req(assoc, Z_Close_systemProblem, msg, req);
1105     }
1106     else if (req->gdu_request->which == Z_GDU_HTTP_Request)
1107         process_http_request(assoc, req);
1108     else
1109     {
1110         do_close_req(assoc, Z_Close_systemProblem, "bad protocol packet", req);
1111     }
1112 }
1113
1114 /*
1115  * Initiate request processing.
1116  */
1117 static int process_z_request(association *assoc, request *req, char **msg)
1118 {
1119     int fd = -1;
1120     Z_APDU *res;
1121     int retval;
1122
1123     *msg = "Unknown Error";
1124     assert(req && req->state == REQUEST_IDLE);
1125     if (req->apdu_request->which != Z_APDU_initRequest && !assoc->init)
1126     {
1127         *msg = "Missing InitRequest";
1128         return -1;
1129     }
1130     switch (req->apdu_request->which)
1131     {
1132     case Z_APDU_initRequest:
1133         iochan_settimeout(assoc->client_chan,
1134                           statserv_getcontrol()->idle_timeout * 60);
1135         res = process_initRequest(assoc, req); break;
1136     case Z_APDU_searchRequest:
1137         res = process_searchRequest(assoc, req, &fd); break;
1138     case Z_APDU_presentRequest:
1139         res = process_presentRequest(assoc, req, &fd); break;
1140     case Z_APDU_scanRequest:
1141         if (assoc->init->bend_scan)
1142             res = process_scanRequest(assoc, req, &fd);
1143         else
1144         {
1145             *msg = "Cannot handle Scan APDU";
1146             return -1;
1147         }
1148         break;
1149     case Z_APDU_extendedServicesRequest:
1150         if (assoc->init->bend_esrequest)
1151             res = process_ESRequest(assoc, req, &fd);
1152         else
1153         {
1154             *msg = "Cannot handle Extended Services APDU";
1155             return -1;
1156         }
1157         break;
1158     case Z_APDU_sortRequest:
1159         if (assoc->init->bend_sort)
1160             res = process_sortRequest(assoc, req, &fd);
1161         else
1162         {
1163             *msg = "Cannot handle Sort APDU";
1164             return -1;
1165         }
1166         break;
1167     case Z_APDU_close:
1168         process_close(assoc, req);
1169         return 0;
1170     case Z_APDU_deleteResultSetRequest:
1171         if (assoc->init->bend_delete)
1172             res = process_deleteRequest(assoc, req, &fd);
1173         else
1174         {
1175             *msg = "Cannot handle Delete APDU";
1176             return -1;
1177         }
1178         break;
1179     case Z_APDU_segmentRequest:
1180         if (assoc->init->bend_segment)
1181         {
1182             res = process_segmentRequest (assoc, req);
1183         }
1184         else
1185         {
1186             *msg = "Cannot handle Segment APDU";
1187             return -1;
1188         }
1189         break;
1190     default:
1191         *msg = "Bad APDU received";
1192         return -1;
1193     }
1194     if (res)
1195     {
1196         yaz_log(LOG_DEBUG, "  result immediately available");
1197         retval = process_z_response(assoc, req, res);
1198     }
1199     else if (fd < 0)
1200     {
1201         yaz_log(LOG_DEBUG, "  result unavailble");
1202         retval = 0;
1203     }
1204     else /* no result yet - one will be provided later */
1205     {
1206         IOCHAN chan;
1207
1208         /* Set up an I/O handler for the fd supplied by the backend */
1209
1210         yaz_log(LOG_DEBUG, "   establishing handler for result");
1211         req->state = REQUEST_PENDING;
1212         if (!(chan = iochan_create(fd, backend_response, EVENT_INPUT)))
1213             abort();
1214         iochan_setdata(chan, assoc);
1215         retval = 0;
1216     }
1217     return retval;
1218 }
1219
1220 /*
1221  * Handle message from the backend.
1222  */
1223 void backend_response(IOCHAN i, int event)
1224 {
1225     association *assoc = (association *)iochan_getdata(i);
1226     request *req = request_head(&assoc->incoming);
1227     Z_APDU *res;
1228     int fd;
1229
1230     yaz_log(LOG_DEBUG, "backend_response");
1231     assert(assoc && req && req->state != REQUEST_IDLE);
1232     /* determine what it is we're waiting for */
1233     switch (req->apdu_request->which)
1234     {
1235         case Z_APDU_searchRequest:
1236             res = response_searchRequest(assoc, req, 0, &fd); break;
1237 #if 0
1238         case Z_APDU_presentRequest:
1239             res = response_presentRequest(assoc, req, 0, &fd); break;
1240         case Z_APDU_scanRequest:
1241             res = response_scanRequest(assoc, req, 0, &fd); break;
1242 #endif
1243         default:
1244             yaz_log(LOG_WARN, "Serious programmer's lapse or bug");
1245             abort();
1246     }
1247     if ((res && process_z_response(assoc, req, res) < 0) || fd < 0)
1248     {
1249         yaz_log(LOG_LOG, "Fatal error when talking to backend");
1250         do_close(assoc, Z_Close_systemProblem, 0);
1251         iochan_destroy(i);
1252         return;
1253     }
1254     else if (!res) /* no result yet - try again later */
1255     {
1256         yaz_log(LOG_DEBUG, "   no result yet");
1257         iochan_setfd(i, fd); /* in case fd has changed */
1258     }
1259 }
1260
1261 /*
1262  * Encode response, and transfer the request structure to the outgoing queue.
1263  */
1264 static int process_gdu_response(association *assoc, request *req, Z_GDU *res)
1265 {
1266     odr_setbuf(assoc->encode, req->response, req->size_response, 1);
1267
1268     if (assoc->print && !z_GDU(assoc->print, &res, 0, 0))
1269     {
1270         yaz_log(LOG_WARN, "ODR print error: %s",
1271             odr_errmsg(odr_geterror(assoc->print)));
1272         odr_reset(assoc->print);
1273     }
1274     if (!z_GDU(assoc->encode, &res, 0, 0))
1275     {
1276         yaz_log(LOG_WARN, "ODR error when encoding response: %s",
1277             odr_errmsg(odr_geterror(assoc->decode)));
1278         return -1;
1279     }
1280     req->response = odr_getbuf(assoc->encode, &req->len_response,
1281         &req->size_response);
1282     odr_setbuf(assoc->encode, 0, 0, 0); /* don'txfree if we abort later */
1283     odr_reset(assoc->encode);
1284     req->state = REQUEST_IDLE;
1285     request_enq(&assoc->outgoing, req);
1286     /* turn the work over to the ir_session handler */
1287     iochan_setflag(assoc->client_chan, EVENT_OUTPUT);
1288     assoc->cs_put_mask = EVENT_OUTPUT;
1289     /* Is there more work to be done? give that to the input handler too */
1290 #if 1
1291     if (request_head(&assoc->incoming))
1292     {
1293         yaz_log (LOG_DEBUG, "more work to be done");
1294         iochan_setevent(assoc->client_chan, EVENT_WORK);
1295     }
1296 #endif
1297     return 0;
1298 }
1299
1300 /*
1301  * Encode response, and transfer the request structure to the outgoing queue.
1302  */
1303 static int process_z_response(association *assoc, request *req, Z_APDU *res)
1304 {
1305     Z_GDU *gres = (Z_GDU *) odr_malloc(assoc->encode, sizeof(*res));
1306     gres->which = Z_GDU_Z3950;
1307     gres->u.z3950 = res;
1308
1309     return process_gdu_response(assoc, req, gres);
1310 }
1311
1312
1313 /*
1314  * Handle init request.
1315  * At the moment, we don't check the options
1316  * anywhere else in the code - we just try not to do anything that would
1317  * break a naive client. We'll toss 'em into the association block when
1318  * we need them there.
1319  */
1320 static Z_APDU *process_initRequest(association *assoc, request *reqb)
1321 {
1322     statserv_options_block *cb = statserv_getcontrol();
1323     Z_InitRequest *req = reqb->apdu_request->u.initRequest;
1324     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_initResponse);
1325     Z_InitResponse *resp = apdu->u.initResponse;
1326     bend_initresult *binitres;
1327
1328     char options[140];
1329
1330     yaz_log(LOG_LOG, "Got initRequest");
1331     if (req->implementationId)
1332         yaz_log(LOG_LOG, "Id:        %s", req->implementationId);
1333     if (req->implementationName)
1334         yaz_log(LOG_LOG, "Name:      %s", req->implementationName);
1335     if (req->implementationVersion)
1336         yaz_log(LOG_LOG, "Version:   %s", req->implementationVersion);
1337
1338     assoc_init_reset(assoc);
1339
1340     assoc->init->auth = req->idAuthentication;
1341     assoc->init->referenceId = req->referenceId;
1342
1343     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel))
1344     {
1345         Z_CharSetandLanguageNegotiation *negotiation =
1346             yaz_get_charneg_record (req->otherInfo);
1347         if (negotiation->which == Z_CharSetandLanguageNegotiation_proposal)
1348             assoc->init->charneg_request = negotiation;
1349     }
1350
1351     if (!(binitres = (*cb->bend_init)(assoc->init)))
1352     {
1353         yaz_log(LOG_WARN, "Bad response from backend.");
1354         return 0;
1355     }
1356
1357     assoc->backend = binitres->handle;
1358     if ((assoc->init->bend_sort))
1359         yaz_log (LOG_DEBUG, "Sort handler installed");
1360     if ((assoc->init->bend_search))
1361         yaz_log (LOG_DEBUG, "Search handler installed");
1362     if ((assoc->init->bend_present))
1363         yaz_log (LOG_DEBUG, "Present handler installed");
1364     if ((assoc->init->bend_esrequest))
1365         yaz_log (LOG_DEBUG, "ESRequest handler installed");
1366     if ((assoc->init->bend_delete))
1367         yaz_log (LOG_DEBUG, "Delete handler installed");
1368     if ((assoc->init->bend_scan))
1369         yaz_log (LOG_DEBUG, "Scan handler installed");
1370     if ((assoc->init->bend_segment))
1371         yaz_log (LOG_DEBUG, "Segment handler installed");
1372
1373     resp->referenceId = req->referenceId;
1374     *options = '\0';
1375     /* let's tell the client what we can do */
1376     if (ODR_MASK_GET(req->options, Z_Options_search))
1377     {
1378         ODR_MASK_SET(resp->options, Z_Options_search);
1379         strcat(options, "srch");
1380     }
1381     if (ODR_MASK_GET(req->options, Z_Options_present))
1382     {
1383         ODR_MASK_SET(resp->options, Z_Options_present);
1384         strcat(options, " prst");
1385     }
1386     if (ODR_MASK_GET(req->options, Z_Options_delSet) &&
1387         assoc->init->bend_delete)
1388     {
1389         ODR_MASK_SET(resp->options, Z_Options_delSet);
1390         strcat(options, " del");
1391     }
1392     if (ODR_MASK_GET(req->options, Z_Options_extendedServices) &&
1393         assoc->init->bend_esrequest)
1394     {
1395         ODR_MASK_SET(resp->options, Z_Options_extendedServices);
1396         strcat (options, " extendedServices");
1397     }
1398     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
1399     {
1400         ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
1401         strcat(options, " namedresults");
1402     }
1403     if (ODR_MASK_GET(req->options, Z_Options_scan) && assoc->init->bend_scan)
1404     {
1405         ODR_MASK_SET(resp->options, Z_Options_scan);
1406         strcat(options, " scan");
1407     }
1408     if (ODR_MASK_GET(req->options, Z_Options_concurrentOperations))
1409     {
1410         ODR_MASK_SET(resp->options, Z_Options_concurrentOperations);
1411         strcat(options, " concurrop");
1412     }
1413     if (ODR_MASK_GET(req->options, Z_Options_sort) && assoc->init->bend_sort)
1414     {
1415         ODR_MASK_SET(resp->options, Z_Options_sort);
1416         strcat(options, " sort");
1417     }
1418
1419     if (ODR_MASK_GET(req->options, Z_Options_negotiationModel)
1420         && assoc->init->charneg_response)
1421     {
1422         Z_OtherInformation **p;
1423         Z_OtherInformationUnit *p0;
1424
1425         yaz_oi_APDU(apdu, &p);
1426
1427         if ((p0=yaz_oi_update(p, assoc->encode, NULL, 0, 0))) {
1428             ODR_MASK_SET(resp->options, Z_Options_negotiationModel);
1429
1430             p0->which = Z_OtherInfo_externallyDefinedInfo;
1431             p0->information.externallyDefinedInfo =
1432                 assoc->init->charneg_response;
1433         }
1434         ODR_MASK_SET(resp->options, Z_Options_negotiationModel);
1435         strcat(options, " negotiation");
1436     }
1437
1438     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_1))
1439     {
1440         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_1);
1441         assoc->version = 2; /* 1 & 2 are equivalent */
1442     }
1443     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_2))
1444     {
1445         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_2);
1446         assoc->version = 2;
1447     }
1448     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_3))
1449     {
1450         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_3);
1451         assoc->version = 3;
1452     }
1453
1454     yaz_log(LOG_LOG, "Negotiated to v%d: %s", assoc->version, options);
1455     assoc->maximumRecordSize = *req->maximumRecordSize;
1456     if (assoc->maximumRecordSize > control_block->maxrecordsize)
1457         assoc->maximumRecordSize = control_block->maxrecordsize;
1458     assoc->preferredMessageSize = *req->preferredMessageSize;
1459     if (assoc->preferredMessageSize > assoc->maximumRecordSize)
1460         assoc->preferredMessageSize = assoc->maximumRecordSize;
1461
1462     resp->preferredMessageSize = &assoc->preferredMessageSize;
1463     resp->maximumRecordSize = &assoc->maximumRecordSize;
1464
1465     resp->implementationName = "GFS/YAZ";
1466
1467     if (assoc->init->implementation_id)
1468     {
1469         char *nv = (char *)
1470             odr_malloc (assoc->encode,
1471                         strlen(assoc->init->implementation_id) + 10 +
1472                                strlen(resp->implementationId));
1473         sprintf (nv, "%s / %s",
1474                  resp->implementationId, assoc->init->implementation_id);
1475         resp->implementationId = nv;
1476     }
1477     if (assoc->init->implementation_name)
1478     {
1479         char *nv = (char *)
1480             odr_malloc (assoc->encode,
1481                         strlen(assoc->init->implementation_name) + 10 +
1482                                strlen(resp->implementationName));
1483         sprintf (nv, "%s / %s",
1484                  resp->implementationName, assoc->init->implementation_name);
1485         resp->implementationName = nv;
1486     }
1487     if (assoc->init->implementation_version)
1488     {
1489         char *nv = (char *)
1490             odr_malloc (assoc->encode,
1491                         strlen(assoc->init->implementation_version) + 10 +
1492                                strlen(resp->implementationVersion));
1493         sprintf (nv, "YAZ %s / %s",
1494                  resp->implementationVersion,
1495                  assoc->init->implementation_version);
1496         resp->implementationVersion = nv;
1497     }
1498
1499     if (binitres->errcode)
1500     {
1501         yaz_log(LOG_LOG, "Connection rejected by backend.");
1502         *resp->result = 0;
1503         assoc->state = ASSOC_DEAD;
1504     }
1505     else
1506         assoc->state = ASSOC_UP;
1507     return apdu;
1508 }
1509
1510 /*
1511  * These functions should be merged.
1512  */
1513
1514 static void set_addinfo (Z_DefaultDiagFormat *dr, char *addinfo, ODR odr)
1515 {
1516     dr->which = Z_DefaultDiagFormat_v2Addinfo;
1517     dr->u.v2Addinfo = odr_strdup (odr, addinfo ? addinfo : "");
1518 }
1519
1520 /*
1521  * nonsurrogate diagnostic record.
1522  */
1523 static Z_Records *diagrec(association *assoc, int error, char *addinfo)
1524 {
1525     Z_Records *rec = (Z_Records *)
1526         odr_malloc (assoc->encode, sizeof(*rec));
1527     int *err = odr_intdup(assoc->encode, error);
1528     Z_DiagRec *drec = (Z_DiagRec *)
1529         odr_malloc (assoc->encode, sizeof(*drec));
1530     Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *)
1531         odr_malloc (assoc->encode, sizeof(*dr));
1532
1533     yaz_log(LOG_LOG, "[%d] %s %s%s", error, diagbib1_str(error),
1534         addinfo ? " -- " : "", addinfo ? addinfo : "");
1535     rec->which = Z_Records_NSD;
1536     rec->u.nonSurrogateDiagnostic = dr;
1537     dr->diagnosticSetId =
1538         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
1539     dr->condition = err;
1540     set_addinfo (dr, addinfo, assoc->encode);
1541     return rec;
1542 }
1543
1544 /*
1545  * surrogate diagnostic.
1546  */
1547 static Z_NamePlusRecord *surrogatediagrec(association *assoc, char *dbname,
1548                                           int error, char *addinfo)
1549 {
1550     Z_NamePlusRecord *rec = (Z_NamePlusRecord *)
1551         odr_malloc (assoc->encode, sizeof(*rec));
1552     int *err = odr_intdup(assoc->encode, error);
1553     Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (assoc->encode, sizeof(*drec));
1554     Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *)
1555         odr_malloc (assoc->encode, sizeof(*dr));
1556
1557     yaz_log(LOG_DEBUG, "SurrogateDiagnotic: %d -- %s", error, addinfo);
1558     rec->databaseName = dbname;
1559     rec->which = Z_NamePlusRecord_surrogateDiagnostic;
1560     rec->u.surrogateDiagnostic = drec;
1561     drec->which = Z_DiagRec_defaultFormat;
1562     drec->u.defaultFormat = dr;
1563     dr->diagnosticSetId =
1564         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
1565     dr->condition = err;
1566     set_addinfo (dr, addinfo, assoc->encode);
1567
1568     return rec;
1569 }
1570
1571 /*
1572  * multiple nonsurrogate diagnostics.
1573  */
1574 static Z_DiagRecs *diagrecs(association *assoc, int error, char *addinfo)
1575 {
1576     Z_DiagRecs *recs = (Z_DiagRecs *)odr_malloc (assoc->encode, sizeof(*recs));
1577     int *err = odr_intdup(assoc->encode, error);
1578     Z_DiagRec **recp = (Z_DiagRec **)odr_malloc (assoc->encode, sizeof(*recp));
1579     Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (assoc->encode, sizeof(*drec));
1580     Z_DefaultDiagFormat *rec = (Z_DefaultDiagFormat *)
1581         odr_malloc (assoc->encode, sizeof(*rec));
1582
1583     yaz_log(LOG_DEBUG, "DiagRecs: %d -- %s", error, addinfo ? addinfo : "");
1584
1585     recs->num_diagRecs = 1;
1586     recs->diagRecs = recp;
1587     recp[0] = drec;
1588     drec->which = Z_DiagRec_defaultFormat;
1589     drec->u.defaultFormat = rec;
1590
1591     rec->diagnosticSetId =
1592         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
1593     rec->condition = err;
1594
1595     rec->which = Z_DefaultDiagFormat_v2Addinfo;
1596     rec->u.v2Addinfo = odr_strdup (assoc->encode, addinfo ? addinfo : "");
1597     return recs;
1598 }
1599
1600 static Z_Records *pack_records(association *a, char *setname, int start,
1601                                int *num, Z_RecordComposition *comp,
1602                                int *next, int *pres, oid_value format,
1603                                Z_ReferenceId *referenceId,
1604                                int *oid)
1605 {
1606     int recno, total_length = 0, toget = *num, dumped_records = 0;
1607     Z_Records *records =
1608         (Z_Records *) odr_malloc (a->encode, sizeof(*records));
1609     Z_NamePlusRecordList *reclist =
1610         (Z_NamePlusRecordList *) odr_malloc (a->encode, sizeof(*reclist));
1611     Z_NamePlusRecord **list =
1612         (Z_NamePlusRecord **) odr_malloc (a->encode, sizeof(*list) * toget);
1613
1614     records->which = Z_Records_DBOSD;
1615     records->u.databaseOrSurDiagnostics = reclist;
1616     reclist->num_records = 0;
1617     reclist->records = list;
1618     *pres = Z_PRES_SUCCESS;
1619     *num = 0;
1620     *next = 0;
1621
1622     yaz_log(LOG_LOG, "Request to pack %d+%d+%s", start, toget, setname);
1623     yaz_log(LOG_DEBUG, "pms=%d, mrs=%d", a->preferredMessageSize,
1624         a->maximumRecordSize);
1625     for (recno = start; reclist->num_records < toget; recno++)
1626     {
1627         bend_fetch_rr freq;
1628         Z_NamePlusRecord *thisrec;
1629         int this_length = 0;
1630         /*
1631          * we get the number of bytes allocated on the stream before any
1632          * allocation done by the backend - this should give us a reasonable
1633          * idea of the total size of the data so far.
1634          */
1635         total_length = odr_total(a->encode) - dumped_records;
1636         freq.errcode = 0;
1637         freq.errstring = 0;
1638         freq.basename = 0;
1639         freq.len = 0;
1640         freq.record = 0;
1641         freq.last_in_set = 0;
1642         freq.setname = setname;
1643         freq.surrogate_flag = 0;
1644         freq.number = recno;
1645         freq.comp = comp;
1646         freq.request_format = format;
1647         freq.request_format_raw = oid;
1648         freq.output_format = format;
1649         freq.output_format_raw = 0;
1650         freq.stream = a->encode;
1651         freq.print = a->print;
1652         freq.surrogate_flag = 0;
1653         freq.referenceId = referenceId;
1654         (*a->init->bend_fetch)(a->backend, &freq);
1655         /* backend should be able to signal whether error is system-wide
1656            or only pertaining to current record */
1657         if (freq.errcode)
1658         {
1659             if (!freq.surrogate_flag)
1660             {
1661                 char s[20];
1662                 *pres = Z_PRES_FAILURE;
1663                 /* for 'present request out of range',
1664                    set addinfo to record position if not set */
1665                 if (freq.errcode == 13 && freq.errstring == 0)
1666                 {
1667                     sprintf (s, "%d", recno);
1668                     freq.errstring = s;
1669                 }
1670                 return diagrec(a, freq.errcode, freq.errstring);
1671             }
1672             reclist->records[reclist->num_records] =
1673                 surrogatediagrec(a, freq.basename, freq.errcode,
1674                                  freq.errstring);
1675             reclist->num_records++;
1676             *next = freq.last_in_set ? 0 : recno + 1;
1677             continue;
1678         }
1679         if (freq.len >= 0)
1680             this_length = freq.len;
1681         else
1682             this_length = odr_total(a->encode) - total_length;
1683         yaz_log(LOG_DEBUG, "  fetched record, len=%d, total=%d",
1684             this_length, total_length);
1685         if (this_length + total_length > a->preferredMessageSize)
1686         {
1687             /* record is small enough, really */
1688             if (this_length <= a->preferredMessageSize)
1689             {
1690                 yaz_log(LOG_DEBUG, "  Dropped last normal-sized record");
1691                 *pres = Z_PRES_PARTIAL_2;
1692                 break;
1693             }
1694             /* record can only be fetched by itself */
1695             if (this_length < a->maximumRecordSize)
1696             {
1697                 yaz_log(LOG_DEBUG, "  Record > prefmsgsz");
1698                 if (toget > 1)
1699                 {
1700                     yaz_log(LOG_DEBUG, "  Dropped it");
1701                     reclist->records[reclist->num_records] =
1702                          surrogatediagrec(a, freq.basename, 16, 0);
1703                     reclist->num_records++;
1704                     *next = freq.last_in_set ? 0 : recno + 1;
1705                     dumped_records += this_length;
1706                     continue;
1707                 }
1708             }
1709             else /* too big entirely */
1710             {
1711                 yaz_log(LOG_LOG, "Record > maxrcdsz this=%d max=%d", this_length, a->maximumRecordSize);
1712                 reclist->records[reclist->num_records] =
1713                     surrogatediagrec(a, freq.basename, 17, 0);
1714                 reclist->num_records++;
1715                 *next = freq.last_in_set ? 0 : recno + 1;
1716                 dumped_records += this_length;
1717                 continue;
1718             }
1719         }
1720
1721         if (!(thisrec = (Z_NamePlusRecord *)
1722               odr_malloc(a->encode, sizeof(*thisrec))))
1723             return 0;
1724         if (!(thisrec->databaseName = (char *)odr_malloc(a->encode,
1725             strlen(freq.basename) + 1)))
1726             return 0;
1727         strcpy(thisrec->databaseName, freq.basename);
1728         thisrec->which = Z_NamePlusRecord_databaseRecord;
1729
1730         if (freq.output_format_raw)
1731         {
1732             struct oident *ident = oid_getentbyoid(freq.output_format_raw);
1733             freq.output_format = ident->value;
1734         }
1735         thisrec->u.databaseRecord = z_ext_record(a->encode, freq.output_format,
1736                                                  freq.record, freq.len);
1737         if (!thisrec->u.databaseRecord)
1738             return 0;
1739         reclist->records[reclist->num_records] = thisrec;
1740         reclist->num_records++;
1741         *next = freq.last_in_set ? 0 : recno + 1;
1742     }
1743     *num = reclist->num_records;
1744     return records;
1745 }
1746
1747 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
1748     int *fd)
1749 {
1750     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
1751     bend_search_rr *bsrr =
1752         (bend_search_rr *)nmem_malloc (reqb->request_mem, sizeof(*bsrr));
1753
1754     yaz_log(LOG_LOG, "Got SearchRequest.");
1755     bsrr->fd = fd;
1756     bsrr->request = reqb;
1757     bsrr->association = assoc;
1758     bsrr->referenceId = req->referenceId;
1759     save_referenceId (reqb, bsrr->referenceId);
1760
1761     yaz_log (LOG_LOG, "ResultSet '%s'", req->resultSetName);
1762     if (req->databaseNames)
1763     {
1764         int i;
1765         for (i = 0; i < req->num_databaseNames; i++)
1766             yaz_log (LOG_LOG, "Database '%s'", req->databaseNames[i]);
1767     }
1768     yaz_log_zquery(req->query);
1769
1770     if (assoc->init->bend_search)
1771     {
1772         bsrr->setname = req->resultSetName;
1773         bsrr->replace_set = *req->replaceIndicator;
1774         bsrr->num_bases = req->num_databaseNames;
1775         bsrr->basenames = req->databaseNames;
1776         bsrr->query = req->query;
1777         bsrr->stream = assoc->encode;
1778         bsrr->decode = assoc->decode;
1779         bsrr->print = assoc->print;
1780         bsrr->errcode = 0;
1781         bsrr->hits = 0;
1782         bsrr->errstring = NULL;
1783         bsrr->search_info = NULL;
1784         (assoc->init->bend_search)(assoc->backend, bsrr);
1785         if (!bsrr->request)
1786             return 0;
1787     }
1788     return response_searchRequest(assoc, reqb, bsrr, fd);
1789 }
1790
1791 int bend_searchresponse(void *handle, bend_search_rr *bsrr) {return 0;}
1792
1793 /*
1794  * Prepare a searchresponse based on the backend results. We probably want
1795  * to look at making the fetching of records nonblocking as well, but
1796  * so far, we'll keep things simple.
1797  * If bsrt is null, that means we're called in response to a communications
1798  * event, and we'll have to get the response for ourselves.
1799  */
1800 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
1801     bend_search_rr *bsrt, int *fd)
1802 {
1803     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
1804     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1805     Z_SearchResponse *resp = (Z_SearchResponse *)
1806         odr_malloc (assoc->encode, sizeof(*resp));
1807     int *nulint = odr_intdup (assoc->encode, 0);
1808     bool_t *sr = odr_intdup(assoc->encode, 1);
1809     int *next = odr_intdup(assoc->encode, 0);
1810     int *none = odr_intdup(assoc->encode, Z_RES_NONE);
1811
1812     apdu->which = Z_APDU_searchResponse;
1813     apdu->u.searchResponse = resp;
1814     resp->referenceId = req->referenceId;
1815     resp->additionalSearchInfo = 0;
1816     resp->otherInfo = 0;
1817     *fd = -1;
1818     if (!bsrt && !bend_searchresponse(assoc->backend, bsrt))
1819     {
1820         yaz_log(LOG_FATAL, "Bad result from backend");
1821         return 0;
1822     }
1823     else if (bsrt->errcode)
1824     {
1825         resp->records = diagrec(assoc, bsrt->errcode, bsrt->errstring);
1826         resp->resultCount = nulint;
1827         resp->numberOfRecordsReturned = nulint;
1828         resp->nextResultSetPosition = nulint;
1829         resp->searchStatus = nulint;
1830         resp->resultSetStatus = none;
1831         resp->presentStatus = 0;
1832     }
1833     else
1834     {
1835         int *toget = odr_intdup(assoc->encode, 0);
1836         int *presst = odr_intdup(assoc->encode, 0);
1837         Z_RecordComposition comp, *compp = 0;
1838
1839         yaz_log (LOG_LOG, "resultCount: %d", bsrt->hits);
1840
1841         resp->records = 0;
1842         resp->resultCount = &bsrt->hits;
1843
1844         comp.which = Z_RecordComp_simple;
1845         /* how many records does the user agent want, then? */
1846         if (bsrt->hits <= *req->smallSetUpperBound)
1847         {
1848             *toget = bsrt->hits;
1849             if ((comp.u.simple = req->smallSetElementSetNames))
1850                 compp = &comp;
1851         }
1852         else if (bsrt->hits < *req->largeSetLowerBound)
1853         {
1854             *toget = *req->mediumSetPresentNumber;
1855             if (*toget > bsrt->hits)
1856                 *toget = bsrt->hits;
1857             if ((comp.u.simple = req->mediumSetElementSetNames))
1858                 compp = &comp;
1859         }
1860         else
1861             *toget = 0;
1862
1863         if (*toget && !resp->records)
1864         {
1865             oident *prefformat;
1866             oid_value form;
1867
1868             if (!(prefformat = oid_getentbyoid(req->preferredRecordSyntax)))
1869                 form = VAL_NONE;
1870             else
1871                 form = prefformat->value;
1872             resp->records = pack_records(assoc, req->resultSetName, 1,
1873                 toget, compp, next, presst, form, req->referenceId,
1874                                          req->preferredRecordSyntax);
1875             if (!resp->records)
1876                 return 0;
1877             resp->numberOfRecordsReturned = toget;
1878             resp->nextResultSetPosition = next;
1879             resp->searchStatus = sr;
1880             resp->resultSetStatus = 0;
1881             resp->presentStatus = presst;
1882         }
1883         else
1884         {
1885             if (*resp->resultCount)
1886                 *next = 1;
1887             resp->numberOfRecordsReturned = nulint;
1888             resp->nextResultSetPosition = next;
1889             resp->searchStatus = sr;
1890             resp->resultSetStatus = 0;
1891             resp->presentStatus = 0;
1892         }
1893     }
1894     resp->additionalSearchInfo = bsrt->search_info;
1895     return apdu;
1896 }
1897
1898 /*
1899  * Maybe we got a little over-friendly when we designed bend_fetch to
1900  * get only one record at a time. Some backends can optimise multiple-record
1901  * fetches, and at any rate, there is some overhead involved in
1902  * all that selecting and hopping around. Problem is, of course, that the
1903  * frontend can't know ahead of time how many records it'll need to
1904  * fill the negotiated PDU size. Annoying. Segmentation or not, Z/SR
1905  * is downright lousy as a bulk data transfer protocol.
1906  *
1907  * To start with, we'll do the fetching of records from the backend
1908  * in one operation: To save some trips in and out of the event-handler,
1909  * and to simplify the interface to pack_records. At any rate, asynch
1910  * operation is more fun in operations that have an unpredictable execution
1911  * speed - which is normally more true for search than for present.
1912  */
1913 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
1914                                       int *fd)
1915 {
1916     Z_PresentRequest *req = reqb->apdu_request->u.presentRequest;
1917     oident *prefformat;
1918     oid_value form;
1919     Z_APDU *apdu;
1920     Z_PresentResponse *resp;
1921     int *next;
1922     int *num;
1923
1924     yaz_log(LOG_LOG, "Got PresentRequest.");
1925
1926     if (!(prefformat = oid_getentbyoid(req->preferredRecordSyntax)))
1927         form = VAL_NONE;
1928     else
1929         form = prefformat->value;
1930     resp = (Z_PresentResponse *)odr_malloc (assoc->encode, sizeof(*resp));
1931     resp->records = 0;
1932     resp->presentStatus = odr_intdup(assoc->encode, 0);
1933     if (assoc->init->bend_present)
1934     {
1935         bend_present_rr *bprr = (bend_present_rr *)
1936             nmem_malloc (reqb->request_mem, sizeof(*bprr));
1937         bprr->setname = req->resultSetId;
1938         bprr->start = *req->resultSetStartPoint;
1939         bprr->number = *req->numberOfRecordsRequested;
1940         bprr->format = form;
1941         bprr->comp = req->recordComposition;
1942         bprr->referenceId = req->referenceId;
1943         bprr->stream = assoc->encode;
1944         bprr->print = assoc->print;
1945         bprr->request = reqb;
1946         bprr->association = assoc;
1947         bprr->errcode = 0;
1948         bprr->errstring = NULL;
1949         (*assoc->init->bend_present)(assoc->backend, bprr);
1950
1951         if (!bprr->request)
1952             return 0;
1953         if (bprr->errcode)
1954         {
1955             resp->records = diagrec(assoc, bprr->errcode, bprr->errstring);
1956             *resp->presentStatus = Z_PRES_FAILURE;
1957         }
1958     }
1959     apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1960     next = odr_intdup(assoc->encode, 0);
1961     num = odr_intdup(assoc->encode, 0);
1962
1963     apdu->which = Z_APDU_presentResponse;
1964     apdu->u.presentResponse = resp;
1965     resp->referenceId = req->referenceId;
1966     resp->otherInfo = 0;
1967
1968     if (!resp->records)
1969     {
1970         *num = *req->numberOfRecordsRequested;
1971         resp->records =
1972             pack_records(assoc, req->resultSetId, *req->resultSetStartPoint,
1973                      num, req->recordComposition, next, resp->presentStatus,
1974                          form, req->referenceId, req->preferredRecordSyntax);
1975     }
1976     if (!resp->records)
1977         return 0;
1978     resp->numberOfRecordsReturned = num;
1979     resp->nextResultSetPosition = next;
1980
1981     return apdu;
1982 }
1983
1984 /*
1985  * Scan was implemented rather in a hurry, and with support for only the basic
1986  * elements of the service in the backend API. Suggestions are welcome.
1987  */
1988 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd)
1989 {
1990     Z_ScanRequest *req = reqb->apdu_request->u.scanRequest;
1991     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1992     Z_ScanResponse *res = (Z_ScanResponse *)
1993         odr_malloc (assoc->encode, sizeof(*res));
1994     int *scanStatus = odr_intdup(assoc->encode, Z_Scan_failure);
1995     int *numberOfEntriesReturned = odr_intdup(assoc->encode, 0);
1996     Z_ListEntries *ents = (Z_ListEntries *)
1997         odr_malloc (assoc->encode, sizeof(*ents));
1998     Z_DiagRecs *diagrecs_p = NULL;
1999     oident *attset;
2000     bend_scan_rr *bsrr = (bend_scan_rr *)
2001         odr_malloc (assoc->encode, sizeof(*bsrr));
2002
2003     yaz_log(LOG_LOG, "Got ScanRequest");
2004
2005     apdu->which = Z_APDU_scanResponse;
2006     apdu->u.scanResponse = res;
2007     res->referenceId = req->referenceId;
2008
2009     /* if step is absent, set it to 0 */
2010     res->stepSize = odr_intdup(assoc->encode, 0);
2011     if (req->stepSize)
2012         *res->stepSize = *req->stepSize;
2013
2014     res->scanStatus = scanStatus;
2015     res->numberOfEntriesReturned = numberOfEntriesReturned;
2016     res->positionOfTerm = 0;
2017     res->entries = ents;
2018     ents->num_entries = 0;
2019     ents->entries = NULL;
2020     ents->num_nonsurrogateDiagnostics = 0;
2021     ents->nonsurrogateDiagnostics = NULL;
2022     res->attributeSet = 0;
2023     res->otherInfo = 0;
2024
2025     if (req->databaseNames)
2026     {
2027         int i;
2028         for (i = 0; i < req->num_databaseNames; i++)
2029             yaz_log (LOG_LOG, "Database '%s'", req->databaseNames[i]);
2030     }
2031     bsrr->num_bases = req->num_databaseNames;
2032     bsrr->basenames = req->databaseNames;
2033     bsrr->num_entries = *req->numberOfTermsRequested;
2034     bsrr->term = req->termListAndStartPoint;
2035     bsrr->referenceId = req->referenceId;
2036     bsrr->stream = assoc->encode;
2037     bsrr->print = assoc->print;
2038     bsrr->step_size = res->stepSize;
2039     if (req->attributeSet &&
2040         (attset = oid_getentbyoid(req->attributeSet)) &&
2041         (attset->oclass == CLASS_ATTSET || attset->oclass == CLASS_GENERAL))
2042         bsrr->attributeset = attset->value;
2043     else
2044         bsrr->attributeset = VAL_NONE;
2045     log_scan_term (req->termListAndStartPoint, bsrr->attributeset);
2046     bsrr->term_position = req->preferredPositionInResponse ?
2047         *req->preferredPositionInResponse : 1;
2048     ((int (*)(void *, bend_scan_rr *))
2049      (*assoc->init->bend_scan))(assoc->backend, bsrr);
2050     if (bsrr->errcode)
2051         diagrecs_p = diagrecs(assoc, bsrr->errcode, bsrr->errstring);
2052     else
2053     {
2054         int i;
2055         Z_Entry **tab = (Z_Entry **)
2056             odr_malloc (assoc->encode, sizeof(*tab) * bsrr->num_entries);
2057
2058         if (bsrr->status == BEND_SCAN_PARTIAL)
2059             *scanStatus = Z_Scan_partial_5;
2060         else
2061             *scanStatus = Z_Scan_success;
2062         ents->entries = tab;
2063         ents->num_entries = bsrr->num_entries;
2064         res->numberOfEntriesReturned = &ents->num_entries;
2065         res->positionOfTerm = &bsrr->term_position;
2066         for (i = 0; i < bsrr->num_entries; i++)
2067         {
2068             Z_Entry *e;
2069             Z_TermInfo *t;
2070             Odr_oct *o;
2071
2072             tab[i] = e = (Z_Entry *)odr_malloc(assoc->encode, sizeof(*e));
2073             if (bsrr->entries[i].occurrences >= 0)
2074             {
2075                 e->which = Z_Entry_termInfo;
2076                 e->u.termInfo = t = (Z_TermInfo *)
2077                     odr_malloc(assoc->encode, sizeof(*t));
2078                 t->suggestedAttributes = 0;
2079                 t->displayTerm = 0;
2080                 t->alternativeTerm = 0;
2081                 t->byAttributes = 0;
2082                 t->otherTermInfo = 0;
2083                 t->globalOccurrences = &bsrr->entries[i].occurrences;
2084                 t->term = (Z_Term *)
2085                     odr_malloc(assoc->encode, sizeof(*t->term));
2086                 t->term->which = Z_Term_general;
2087                 t->term->u.general = o =
2088                     (Odr_oct *)odr_malloc(assoc->encode, sizeof(Odr_oct));
2089                 o->buf = (unsigned char *)
2090                     odr_malloc(assoc->encode, o->len = o->size =
2091                                strlen(bsrr->entries[i].term));
2092                 memcpy(o->buf, bsrr->entries[i].term, o->len);
2093                 yaz_log(LOG_DEBUG, "  term #%d: '%s' (%d)", i,
2094                          bsrr->entries[i].term, bsrr->entries[i].occurrences);
2095             }
2096             else
2097             {
2098                 Z_DiagRecs *drecs = diagrecs (assoc,
2099                                               bsrr->entries[i].errcode,
2100                                               bsrr->entries[i].errstring);
2101                 assert (drecs->num_diagRecs == 1);
2102                 e->which = Z_Entry_surrogateDiagnostic;
2103                 assert (drecs->diagRecs[0]);
2104                 e->u.surrogateDiagnostic = drecs->diagRecs[0];
2105             }
2106         }
2107     }
2108     if (diagrecs_p)
2109     {
2110         ents->num_nonsurrogateDiagnostics = diagrecs_p->num_diagRecs;
2111         ents->nonsurrogateDiagnostics = diagrecs_p->diagRecs;
2112     }
2113     return apdu;
2114 }
2115
2116 static Z_APDU *process_sortRequest(association *assoc, request *reqb,
2117     int *fd)
2118 {
2119     Z_SortRequest *req = reqb->apdu_request->u.sortRequest;
2120     Z_SortResponse *res = (Z_SortResponse *)
2121         odr_malloc (assoc->encode, sizeof(*res));
2122     bend_sort_rr *bsrr = (bend_sort_rr *)
2123         odr_malloc (assoc->encode, sizeof(*bsrr));
2124
2125     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
2126
2127     yaz_log(LOG_LOG, "Got SortRequest.");
2128
2129     bsrr->num_input_setnames = req->num_inputResultSetNames;
2130     bsrr->input_setnames = req->inputResultSetNames;
2131     bsrr->referenceId = req->referenceId;
2132     bsrr->output_setname = req->sortedResultSetName;
2133     bsrr->sort_sequence = req->sortSequence;
2134     bsrr->stream = assoc->encode;
2135     bsrr->print = assoc->print;
2136
2137     bsrr->sort_status = Z_SortStatus_failure;
2138     bsrr->errcode = 0;
2139     bsrr->errstring = 0;
2140
2141     (*assoc->init->bend_sort)(assoc->backend, bsrr);
2142
2143     res->referenceId = bsrr->referenceId;
2144     res->sortStatus = odr_intdup(assoc->encode, bsrr->sort_status);
2145     res->resultSetStatus = 0;
2146     if (bsrr->errcode)
2147     {
2148         Z_DiagRecs *dr = diagrecs (assoc, bsrr->errcode, bsrr->errstring);
2149         res->diagnostics = dr->diagRecs;
2150         res->num_diagnostics = dr->num_diagRecs;
2151     }
2152     else
2153     {
2154         res->num_diagnostics = 0;
2155         res->diagnostics = 0;
2156     }
2157     res->otherInfo = 0;
2158
2159     apdu->which = Z_APDU_sortResponse;
2160     apdu->u.sortResponse = res;
2161     return apdu;
2162 }
2163
2164 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
2165     int *fd)
2166 {
2167     Z_DeleteResultSetRequest *req =
2168         reqb->apdu_request->u.deleteResultSetRequest;
2169     Z_DeleteResultSetResponse *res = (Z_DeleteResultSetResponse *)
2170         odr_malloc (assoc->encode, sizeof(*res));
2171     bend_delete_rr *bdrr = (bend_delete_rr *)
2172         odr_malloc (assoc->encode, sizeof(*bdrr));
2173     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
2174
2175     yaz_log(LOG_LOG, "Got DeleteRequest.");
2176
2177     bdrr->num_setnames = req->num_resultSetList;
2178     bdrr->setnames = req->resultSetList;
2179     bdrr->stream = assoc->encode;
2180     bdrr->print = assoc->print;
2181     bdrr->function = *req->deleteFunction;
2182     bdrr->referenceId = req->referenceId;
2183     bdrr->statuses = 0;
2184     if (bdrr->num_setnames > 0)
2185     {
2186         int i;
2187         bdrr->statuses = (int*)
2188             odr_malloc(assoc->encode, sizeof(*bdrr->statuses) *
2189                        bdrr->num_setnames);
2190         for (i = 0; i < bdrr->num_setnames; i++)
2191             bdrr->statuses[i] = 0;
2192     }
2193     (*assoc->init->bend_delete)(assoc->backend, bdrr);
2194
2195     res->referenceId = req->referenceId;
2196
2197     res->deleteOperationStatus = odr_intdup(assoc->encode,bdrr->delete_status);
2198
2199     res->deleteListStatuses = 0;
2200     if (bdrr->num_setnames > 0)
2201     {
2202         int i;
2203         res->deleteListStatuses = (Z_ListStatuses *)
2204             odr_malloc(assoc->encode, sizeof(*res->deleteListStatuses));
2205         res->deleteListStatuses->num = bdrr->num_setnames;
2206         res->deleteListStatuses->elements =
2207             (Z_ListStatus **)
2208             odr_malloc (assoc->encode,
2209                         sizeof(*res->deleteListStatuses->elements) *
2210                         bdrr->num_setnames);
2211         for (i = 0; i<bdrr->num_setnames; i++)
2212         {
2213             res->deleteListStatuses->elements[i] =
2214                 (Z_ListStatus *)
2215                 odr_malloc (assoc->encode,
2216                             sizeof(**res->deleteListStatuses->elements));
2217             res->deleteListStatuses->elements[i]->status = bdrr->statuses+i;
2218             res->deleteListStatuses->elements[i]->id =
2219                 odr_strdup (assoc->encode, bdrr->setnames[i]);
2220
2221         }
2222     }
2223     res->numberNotDeleted = 0;
2224     res->bulkStatuses = 0;
2225     res->deleteMessage = 0;
2226     res->otherInfo = 0;
2227
2228     apdu->which = Z_APDU_deleteResultSetResponse;
2229     apdu->u.deleteResultSetResponse = res;
2230     return apdu;
2231 }
2232
2233 static void process_close(association *assoc, request *reqb)
2234 {
2235     Z_Close *req = reqb->apdu_request->u.close;
2236     static char *reasons[] =
2237     {
2238         "finished",
2239         "shutdown",
2240         "systemProblem",
2241         "costLimit",
2242         "resources",
2243         "securityViolation",
2244         "protocolError",
2245         "lackOfActivity",
2246         "peerAbort",
2247         "unspecified"
2248     };
2249
2250     yaz_log(LOG_LOG, "Got Close, reason %s, message %s",
2251         reasons[*req->closeReason], req->diagnosticInformation ?
2252         req->diagnosticInformation : "NULL");
2253     if (assoc->version < 3) /* to make do_force respond with close */
2254         assoc->version = 3;
2255     do_close_req(assoc, Z_Close_finished,
2256                  "Association terminated by client", reqb);
2257 }
2258
2259 void save_referenceId (request *reqb, Z_ReferenceId *refid)
2260 {
2261     if (refid)
2262     {
2263         reqb->len_refid = refid->len;
2264         reqb->refid = (char *)nmem_malloc (reqb->request_mem, refid->len);
2265         memcpy (reqb->refid, refid->buf, refid->len);
2266     }
2267     else
2268     {
2269         reqb->len_refid = 0;
2270         reqb->refid = NULL;
2271     }
2272 }
2273
2274 void bend_request_send (bend_association a, bend_request req, Z_APDU *res)
2275 {
2276     process_z_response (a, req, res);
2277 }
2278
2279 bend_request bend_request_mk (bend_association a)
2280 {
2281     request *nreq = request_get (&a->outgoing);
2282     nreq->request_mem = nmem_create ();
2283     return nreq;
2284 }
2285
2286 Z_ReferenceId *bend_request_getid (ODR odr, bend_request req)
2287 {
2288     Z_ReferenceId *id;
2289     if (!req->refid)
2290         return 0;
2291     id = (Odr_oct *)odr_malloc (odr, sizeof(*odr));
2292     id->buf = (unsigned char *)odr_malloc (odr, req->len_refid);
2293     id->len = id->size = req->len_refid;
2294     memcpy (id->buf, req->refid, req->len_refid);
2295     return id;
2296 }
2297
2298 void bend_request_destroy (bend_request *req)
2299 {
2300     nmem_destroy((*req)->request_mem);
2301     request_release(*req);
2302     *req = NULL;
2303 }
2304
2305 int bend_backend_respond (bend_association a, bend_request req)
2306 {
2307     char *msg;
2308     int r;
2309     r = process_z_request (a, req, &msg);
2310     if (r < 0)
2311         yaz_log (LOG_WARN, "%s", msg);
2312     return r;
2313 }
2314
2315 void bend_request_setdata(bend_request r, void *p)
2316 {
2317     r->clientData = p;
2318 }
2319
2320 void *bend_request_getdata(bend_request r)
2321 {
2322     return r->clientData;
2323 }
2324
2325 static Z_APDU *process_segmentRequest (association *assoc, request *reqb)
2326 {
2327     bend_segment_rr req;
2328
2329     req.segment = reqb->apdu_request->u.segmentRequest;
2330     req.stream = assoc->encode;
2331     req.decode = assoc->decode;
2332     req.print = assoc->print;
2333     req.association = assoc;
2334
2335     (*assoc->init->bend_segment)(assoc->backend, &req);
2336
2337     return 0;
2338 }
2339
2340 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd)
2341 {
2342     bend_esrequest_rr esrequest;
2343
2344     Z_ExtendedServicesRequest *req =
2345         reqb->apdu_request->u.extendedServicesRequest;
2346     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_extendedServicesResponse);
2347
2348     Z_ExtendedServicesResponse *resp = apdu->u.extendedServicesResponse;
2349
2350     yaz_log(LOG_DEBUG,"inside Process esRequest");
2351
2352     esrequest.esr = reqb->apdu_request->u.extendedServicesRequest;
2353     esrequest.stream = assoc->encode;
2354     esrequest.decode = assoc->decode;
2355     esrequest.print = assoc->print;
2356     esrequest.errcode = 0;
2357     esrequest.errstring = NULL;
2358     esrequest.request = reqb;
2359     esrequest.association = assoc;
2360     esrequest.taskPackage = 0;
2361     esrequest.referenceId = req->referenceId;
2362
2363     (*assoc->init->bend_esrequest)(assoc->backend, &esrequest);
2364
2365     /* If the response is being delayed, return NULL */
2366     if (esrequest.request == NULL)
2367         return(NULL);
2368
2369     resp->referenceId = req->referenceId;
2370
2371     if (esrequest.errcode == -1)
2372     {
2373         /* Backend service indicates request will be processed */
2374         yaz_log(LOG_DEBUG,"Request could be processed...Accepted !");
2375         *resp->operationStatus = Z_ExtendedServicesResponse_accepted;
2376     }
2377     else if (esrequest.errcode == 0)
2378     {
2379         /* Backend service indicates request will be processed */
2380         yaz_log(LOG_DEBUG,"Request could be processed...Done !");
2381         *resp->operationStatus = Z_ExtendedServicesResponse_done;
2382     }
2383     else
2384     {
2385         Z_DiagRecs *diagRecs = diagrecs (assoc, esrequest.errcode,
2386                                          esrequest.errstring);
2387
2388         /* Backend indicates error, request will not be processed */
2389         yaz_log(LOG_DEBUG,"Request could not be processed...failure !");
2390         *resp->operationStatus = Z_ExtendedServicesResponse_failure;
2391         resp->num_diagnostics = diagRecs->num_diagRecs;
2392         resp->diagnostics = diagRecs->diagRecs;
2393     }
2394     /* Do something with the members of bend_extendedservice */
2395     if (esrequest.taskPackage)
2396         resp->taskPackage = z_ext_record (assoc->encode, VAL_EXTENDED,
2397                                          (const char *)  esrequest.taskPackage,
2398                                           -1);
2399     yaz_log(LOG_DEBUG,"Send the result apdu");
2400     return apdu;
2401 }
2402