server/seshigh.c

   1 /*
   2  * Copyright (c) 1995-2002, Index Data
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: seshigh.c,v 1.127 2002-03-05 12:45:49 mike 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 #ifdef WIN32
  32 #include <process.h>
  33 #else
  34 #include <unistd.h>
  35 #endif
  36 #include <assert.h>
  37
  38 #include <yaz/yconfig.h>
  39 #include <yaz/xmalloc.h>
  40 #include <yaz/comstack.h>
  41 #include "eventl.h"
  42 #include "session.h"
  43 #include <yaz/proto.h>
  44 #include <yaz/oid.h>
  45 #include <yaz/log.h>
  46 #include <yaz/logrpn.h>
  47 #include <yaz/statserv.h>
  48 #include <yaz/diagbib1.h>
  49
  50 #include <yaz/backend.h>
  51
  52 static int process_request(association *assoc, request *req, char **msg);
  53 void backend_response(IOCHAN i, int event);
  54 static int process_response(association *assoc, request *req, Z_APDU *res);
  55 static Z_APDU *process_initRequest(association *assoc, request *reqb);
  56 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
  57     int *fd);
  58 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
  59     bend_search_rr *bsrr, int *fd);
  60 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
  61     int *fd);
  62 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd);
  63 static Z_APDU *process_sortRequest(association *assoc, request *reqb, int *fd);
  64 static void process_close(association *assoc, request *reqb);
  65 void save_referenceId (request *reqb, Z_ReferenceId *refid);
  66 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
  67     int *fd);
  68 static Z_APDU *process_segmentRequest (association *assoc, request *reqb);
  69
  70 static FILE *apduf = 0; /* for use in static mode */
  71 static statserv_options_block *control_block = 0;
  72
  73 /* Chas: Added in from DALI */
  74 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd);
  75 /* Chas: End of addition from DALI */
  76
  77 /*
  78  * Create and initialize a new association-handle.
  79  *  channel  : iochannel for the current line.
  80  *  link     : communications channel.
  81  * Returns: 0 or a new association handle.
  82  */
  83 association *create_association(IOCHAN channel, COMSTACK link)
  84 {
  85     association *anew;
  86
  87     if (!control_block)
  88         control_block = statserv_getcontrol();
  89     if (!(anew = (association *)xmalloc(sizeof(*anew))))
  90         return 0;
  91     anew->init = 0;
  92     anew->client_chan = channel;
  93     anew->client_link = link;
  94     anew->cs_get_mask = 0;
  95     anew->cs_put_mask = 0;
  96     anew->cs_accept_mask = 0;
  97     if (!(anew->decode = odr_createmem(ODR_DECODE)) ||
  98         !(anew->encode = odr_createmem(ODR_ENCODE)))
  99         return 0;
 100     if (*control_block->apdufile)
 101     {
 102         char filename[256];
 103         FILE *f;
 104
 105         strcpy(filename, control_block->apdufile);
 106         if (!(anew->print = odr_createmem(ODR_PRINT)))
 107             return 0;
 108         if (*control_block->apdufile != '-')
 109         {
 110             strcpy(filename, control_block->apdufile);
 111             if (!control_block->dynamic)
 112             {
 113                 if (!apduf)
 114                 {
 115                     if (!(apduf = fopen(filename, "w")))
 116                     {
 117                         yaz_log(LOG_WARN|LOG_ERRNO, "%s", filename);
 118                         return 0;
 119                     }
 120                     setvbuf(apduf, 0, _IONBF, 0);
 121                 }
 122                 f = apduf;
 123             }
 124             else
 125             {
 126                 sprintf(filename + strlen(filename), ".%d", getpid());
 127                 if (!(f = fopen(filename, "w")))
 128                 {
 129                     yaz_log(LOG_WARN|LOG_ERRNO, "%s", filename);
 130                     return 0;
 131                 }
 132                 setvbuf(f, 0, _IONBF, 0);
 133             }
 134             odr_setprint(anew->print, f);
 135         }
 136     }
 137     else
 138         anew->print = 0;
 139     anew->input_buffer = 0;
 140     anew->input_buffer_len = 0;
 141     anew->backend = 0;
 142     anew->state = ASSOC_NEW;
 143     request_initq(&anew->incoming);
 144     request_initq(&anew->outgoing);
 145     anew->proto = cs_getproto(link);
 146     return anew;
 147 }
 148
 149 /*
 150  * Free association and release resources.
 151  */
 152 void destroy_association(association *h)
 153 {
 154     statserv_options_block *cb = statserv_getcontrol();
 155
 156     xfree(h->init);
 157     odr_destroy(h->decode);
 158     odr_destroy(h->encode);
 159     if (h->print)
 160         odr_destroy(h->print);
 161     if (h->input_buffer)
 162     xfree(h->input_buffer);
 163     if (h->backend)
 164         (*cb->bend_close)(h->backend);
 165     while (request_deq(&h->incoming));
 166     while (request_deq(&h->outgoing));
 167     request_delq(&h->incoming);
 168     request_delq(&h->outgoing);
 169     xfree(h);
 170     if (control_block && control_block->one_shot)
 171         exit (0);
 172 }
 173
 174 static void do_close_req(association *a, int reason, char *message,
 175                          request *req)
 176 {
 177     Z_APDU apdu;
 178     Z_Close *cls = zget_Close(a->encode);
 179
 180     /* Purge request queue */
 181     while (request_deq(&a->incoming));
 182     while (request_deq(&a->outgoing));
 183     if (a->version >= 3)
 184     {
 185         yaz_log(LOG_LOG, "Sending Close PDU, reason=%d, message=%s",
 186             reason, message ? message : "none");
 187         apdu.which = Z_APDU_close;
 188         apdu.u.close = cls;
 189         *cls->closeReason = reason;
 190         cls->diagnosticInformation = message;
 191         process_response(a, req, &apdu);
 192         iochan_settimeout(a->client_chan, 60);
 193     }
 194     else
 195     {
 196         yaz_log(LOG_DEBUG, "v2 client. No Close PDU");
 197         iochan_setevent(a->client_chan, EVENT_TIMEOUT); /* force imm close */
 198     }
 199     a->state = ASSOC_DEAD;
 200 }
 201
 202 static void do_close(association *a, int reason, char *message)
 203 {
 204     do_close_req (a, reason, message, request_get(&a->outgoing));
 205 }
 206
 207 /*
 208  * This is where PDUs from the client are read and the further
 209  * processing is initiated. Flow of control moves down through the
 210  * various process_* functions below, until the encoded result comes back up
 211  * to the output handler in here.
 212  *
 213  *  h     : the I/O channel that has an outstanding event.
 214  *  event : the current outstanding event.
 215  */
 216 void ir_session(IOCHAN h, int event)
 217 {
 218     int res;
 219     association *assoc = (association *)iochan_getdata(h);
 220     COMSTACK conn = assoc->client_link;
 221     request *req;
 222
 223     assert(h && conn && assoc);
 224     if (event == EVENT_TIMEOUT)
 225     {
 226         if (assoc->state != ASSOC_UP)
 227         {
 228             yaz_log(LOG_LOG, "Final timeout - closing connection.");
 229             cs_close(conn);
 230             destroy_association(assoc);
 231             iochan_destroy(h);
 232         }
 233         else
 234         {
 235             yaz_log(LOG_LOG, "Session idle too long. Sending close.");
 236             do_close(assoc, Z_Close_lackOfActivity, 0);
 237         }
 238         return;
 239     }
 240     if (event & assoc->cs_accept_mask)
 241     {
 242         yaz_log (LOG_DEBUG, "ir_session (accept)");
 243         if (!cs_accept (conn))
 244         {
 245             yaz_log (LOG_LOG, "accept failed");
 246             destroy_association(assoc);
 247             iochan_destroy(h);
 248         }
 249         iochan_clearflag (h, EVENT_OUTPUT|EVENT_OUTPUT);
 250         if (conn->io_pending)
 251         {   /* cs_accept didn't complete */
 252             assoc->cs_accept_mask =
 253                 ((conn->io_pending & CS_WANT_WRITE) ? EVENT_OUTPUT : 0) |
 254                 ((conn->io_pending & CS_WANT_READ) ? EVENT_INPUT : 0);
 255
 256             iochan_setflag (h, assoc->cs_accept_mask);
 257         }
 258         else
 259         {   /* cs_accept completed. Prepare for reading (cs_get) */
 260             assoc->cs_accept_mask = 0;
 261             assoc->cs_get_mask = EVENT_INPUT;
 262             iochan_setflag (h, assoc->cs_get_mask);
 263         }
 264         return;
 265     }
 266     if ((event & assoc->cs_get_mask) || (event & EVENT_WORK)) /* input */
 267     {
 268         if ((assoc->cs_put_mask & EVENT_INPUT) == 0 && (event & assoc->cs_get_mask))
 269         {
 270             yaz_log(LOG_DEBUG, "ir_session (input)");
 271             /* We aren't speaking to this fellow */
 272             if (assoc->state == ASSOC_DEAD)
 273             {
 274                 yaz_log(LOG_LOG, "Closed connection after reject");
 275                 cs_close(conn);
 276                 destroy_association(assoc);
 277                 iochan_destroy(h);
 278                 return;
 279             }
 280             assoc->cs_get_mask = EVENT_INPUT;
 281             if ((res = cs_get(conn, &assoc->input_buffer,
 282                 &assoc->input_buffer_len)) <= 0)
 283             {
 284                 yaz_log(LOG_LOG, "Connection closed by client");
 285                 cs_close(conn);
 286                 destroy_association(assoc);
 287                 iochan_destroy(h);
 288                 return;
 289             }
 290             else if (res == 1) /* incomplete read - wait for more  */
 291             {
 292                 if (conn->io_pending & CS_WANT_WRITE)
 293                     assoc->cs_get_mask |= EVENT_OUTPUT;
 294                 iochan_setflag(h, assoc->cs_get_mask);
 295                 return;
 296             }
 297             if (cs_more(conn)) /* more stuff - call us again later, please */
 298                 iochan_setevent(h, EVENT_INPUT);
 299
 300             /* we got a complete PDU. Let's decode it */
 301             yaz_log(LOG_DEBUG, "Got PDU, %d bytes", res);
 302             req = request_get(&assoc->incoming); /* get a new request structure */
 303             odr_reset(assoc->decode);
 304             odr_setbuf(assoc->decode, assoc->input_buffer, res, 0);
 305             if (!z_APDU(assoc->decode, &req->apdu_request, 0, 0))
 306             {
 307                 yaz_log(LOG_LOG, "ODR error on incoming PDU: %s [near byte %d] ",
 308                         odr_errmsg(odr_geterror(assoc->decode)),
 309                         odr_offset(assoc->decode));
 310                 yaz_log(LOG_LOG, "PDU dump:");
 311                 odr_dumpBER(yaz_log_file(), assoc->input_buffer, res);
 312                 do_close(assoc, Z_Close_protocolError, "Malformed package");
 313                 return;
 314             }
 315             req->request_mem = odr_extract_mem(assoc->decode);
 316             if (assoc->print && !z_APDU(assoc->print, &req->apdu_request, 0, 0))
 317             {
 318                 yaz_log(LOG_WARN, "ODR print error: %s",
 319                     odr_errmsg(odr_geterror(assoc->print)));
 320                 odr_reset(assoc->print);
 321             }
 322             request_enq(&assoc->incoming, req);
 323         }
 324
 325         /* can we do something yet? */
 326         req = request_head(&assoc->incoming);
 327         if (req->state == REQUEST_IDLE)
 328         {
 329             char *msg;
 330             request_deq(&assoc->incoming);
 331             if (process_request(assoc, req, &msg) < 0)
 332                 do_close_req(assoc, Z_Close_systemProblem, msg, req);
 333         }
 334     }
 335     if (event & assoc->cs_put_mask)
 336     {
 337         request *req = request_head(&assoc->outgoing);
 338
 339         assoc->cs_put_mask = 0;
 340         yaz_log(LOG_DEBUG, "ir_session (output)");
 341         req->state = REQUEST_PENDING;
 342         switch (res = cs_put(conn, req->response, req->len_response))
 343         {
 344         case -1:
 345             yaz_log(LOG_LOG, "Connection closed by client");
 346             cs_close(conn);
 347             destroy_association(assoc);
 348             iochan_destroy(h);
 349             break;
 350         case 0: /* all sent - release the request structure */
 351             yaz_log(LOG_DEBUG, "Wrote PDU, %d bytes", req->len_response);
 352             nmem_destroy(req->request_mem);
 353             request_deq(&assoc->outgoing);
 354             request_release(req);
 355             if (!request_head(&assoc->outgoing))
 356             {   /* restore mask for cs_get operation ... */
 357                 iochan_clearflag(h, EVENT_OUTPUT|EVENT_INPUT);
 358                 iochan_setflag(h, assoc->cs_get_mask);
 359             }
 360             break;
 361         default:
 362             if (conn->io_pending & CS_WANT_WRITE)
 363                 assoc->cs_put_mask |= EVENT_OUTPUT;
 364             if (conn->io_pending & CS_WANT_READ)
 365                 assoc->cs_put_mask |= EVENT_INPUT;
 366             iochan_setflag(h, assoc->cs_put_mask);
 367         }
 368     }
 369     if (event & EVENT_EXCEPT)
 370     {
 371         yaz_log(LOG_LOG, "ir_session (exception)");
 372         cs_close(conn);
 373         destroy_association(assoc);
 374         iochan_destroy(h);
 375     }
 376 }
 377
 378 /*
 379  * Initiate request processing.
 380  */
 381 static int process_request(association *assoc, request *req, char **msg)
 382 {
 383     int fd = -1;
 384     Z_APDU *res;
 385     int retval;
 386
 387     *msg = "Unknown Error";
 388     assert(req && req->state == REQUEST_IDLE);
 389     if (req->apdu_request->which != Z_APDU_initRequest && !assoc->init)
 390     {
 391         *msg = "Missing InitRequest";
 392         return -1;
 393     }
 394     switch (req->apdu_request->which)
 395     {
 396     case Z_APDU_initRequest:
 397         res = process_initRequest(assoc, req); break;
 398     case Z_APDU_searchRequest:
 399         res = process_searchRequest(assoc, req, &fd); break;
 400     case Z_APDU_presentRequest:
 401         res = process_presentRequest(assoc, req, &fd); break;
 402     case Z_APDU_scanRequest:
 403         if (assoc->init->bend_scan)
 404             res = process_scanRequest(assoc, req, &fd);
 405         else
 406         {
 407             *msg = "Cannot handle Scan APDU";
 408             return -1;
 409         }
 410         break;
 411     case Z_APDU_extendedServicesRequest:
 412         if (assoc->init->bend_esrequest)
 413             res = process_ESRequest(assoc, req, &fd);
 414         else
 415         {
 416             *msg = "Cannot handle Extended Services APDU";
 417             return -1;
 418         }
 419         break;
 420     case Z_APDU_sortRequest:
 421         if (assoc->init->bend_sort)
 422             res = process_sortRequest(assoc, req, &fd);
 423         else
 424         {
 425             *msg = "Cannot handle Sort APDU";
 426             return -1;
 427         }
 428         break;
 429     case Z_APDU_close:
 430         process_close(assoc, req);
 431         return 0;
 432     case Z_APDU_deleteResultSetRequest:
 433         if (assoc->init->bend_delete)
 434             res = process_deleteRequest(assoc, req, &fd);
 435         else
 436         {
 437             *msg = "Cannot handle Delete APDU";
 438             return -1;
 439         }
 440         break;
 441     case Z_APDU_segmentRequest:
 442         if (assoc->init->bend_segment)
 443         {
 444             res = process_segmentRequest (assoc, req);
 445         }
 446         else
 447         {
 448             *msg = "Cannot handle Segment APDU";
 449             return -1;
 450         }
 451         break;
 452     default:
 453         *msg = "Bad APDU received";
 454         return -1;
 455     }
 456     if (res)
 457     {
 458         yaz_log(LOG_DEBUG, "  result immediately available");
 459         retval = process_response(assoc, req, res);
 460     }
 461     else if (fd < 0)
 462     {
 463         yaz_log(LOG_DEBUG, "  result unavailble");
 464         retval = 0;
 465     }
 466     else /* no result yet - one will be provided later */
 467     {
 468         IOCHAN chan;
 469
 470         /* Set up an I/O handler for the fd supplied by the backend */
 471
 472         yaz_log(LOG_DEBUG, "   establishing handler for result");
 473         req->state = REQUEST_PENDING;
 474         if (!(chan = iochan_create(fd, backend_response, EVENT_INPUT)))
 475             abort();
 476         iochan_setdata(chan, assoc);
 477         retval = 0;
 478     }
 479     return retval;
 480 }
 481
 482 /*
 483  * Handle message from the backend.
 484  */
 485 void backend_response(IOCHAN i, int event)
 486 {
 487     association *assoc = (association *)iochan_getdata(i);
 488     request *req = request_head(&assoc->incoming);
 489     Z_APDU *res;
 490     int fd;
 491
 492     yaz_log(LOG_DEBUG, "backend_response");
 493     assert(assoc && req && req->state != REQUEST_IDLE);
 494     /* determine what it is we're waiting for */
 495     switch (req->apdu_request->which)
 496     {
 497         case Z_APDU_searchRequest:
 498             res = response_searchRequest(assoc, req, 0, &fd); break;
 499 #if 0
 500         case Z_APDU_presentRequest:
 501             res = response_presentRequest(assoc, req, 0, &fd); break;
 502         case Z_APDU_scanRequest:
 503             res = response_scanRequest(assoc, req, 0, &fd); break;
 504 #endif
 505         default:
 506             yaz_log(LOG_WARN, "Serious programmer's lapse or bug");
 507             abort();
 508     }
 509     if ((res && process_response(assoc, req, res) < 0) || fd < 0)
 510     {
 511         yaz_log(LOG_LOG, "Fatal error when talking to backend");
 512         do_close(assoc, Z_Close_systemProblem, 0);
 513         iochan_destroy(i);
 514         return;
 515     }
 516     else if (!res) /* no result yet - try again later */
 517     {
 518         yaz_log(LOG_DEBUG, "   no result yet");
 519         iochan_setfd(i, fd); /* in case fd has changed */
 520     }
 521 }
 522
 523 /*
 524  * Encode response, and transfer the request structure to the outgoing queue.
 525  */
 526 static int process_response(association *assoc, request *req, Z_APDU *res)
 527 {
 528     odr_setbuf(assoc->encode, req->response, req->size_response, 1);
 529
 530     if (assoc->print && !z_APDU(assoc->print, &res, 0, 0))
 531     {
 532         yaz_log(LOG_WARN, "ODR print error: %s",
 533             odr_errmsg(odr_geterror(assoc->print)));
 534         odr_reset(assoc->print);
 535     }
 536     if (!z_APDU(assoc->encode, &res, 0, 0))
 537     {
 538         yaz_log(LOG_WARN, "ODR error when encoding response: %s",
 539             odr_errmsg(odr_geterror(assoc->decode)));
 540         return -1;
 541     }
 542     req->response = odr_getbuf(assoc->encode, &req->len_response,
 543         &req->size_response);
 544     odr_setbuf(assoc->encode, 0, 0, 0); /* don'txfree if we abort later */
 545     odr_reset(assoc->encode);
 546     req->state = REQUEST_IDLE;
 547     request_enq(&assoc->outgoing, req);
 548     /* turn the work over to the ir_session handler */
 549     iochan_setflag(assoc->client_chan, EVENT_OUTPUT);
 550     assoc->cs_put_mask = EVENT_OUTPUT;
 551     /* Is there more work to be done? give that to the input handler too */
 552 #if 1
 553     if (request_head(&assoc->incoming))
 554     {
 555         yaz_log (LOG_DEBUG, "more work to be done");
 556         iochan_setevent(assoc->client_chan, EVENT_WORK);
 557     }
 558 #endif
 559     return 0;
 560 }
 561
 562 /*
 563  * Handle init request.
 564  * At the moment, we don't check the options
 565  * anywhere else in the code - we just try not to do anything that would
 566  * break a naive client. We'll toss 'em into the association block when
 567  * we need them there.
 568  */
 569 static Z_APDU *process_initRequest(association *assoc, request *reqb)
 570 {
 571     statserv_options_block *cb = statserv_getcontrol();
 572     Z_InitRequest *req = reqb->apdu_request->u.initRequest;
 573     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_initResponse);
 574     Z_InitResponse *resp = apdu->u.initResponse;
 575     bend_initresult *binitres;
 576     char options[100];
 577
 578     xfree (assoc->init);
 579     assoc->init = (bend_initrequest *) xmalloc (sizeof(*assoc->init));
 580
 581     yaz_log(LOG_LOG, "Got initRequest");
 582     if (req->implementationId)
 583         yaz_log(LOG_LOG, "Id:        %s", req->implementationId);
 584     if (req->implementationName)
 585         yaz_log(LOG_LOG, "Name:      %s", req->implementationName);
 586     if (req->implementationVersion)
 587         yaz_log(LOG_LOG, "Version:   %s", req->implementationVersion);
 588
 589     assoc->init->stream = assoc->encode;
 590     assoc->init->print = assoc->print;
 591     assoc->init->auth = req->idAuthentication;
 592     assoc->init->referenceId = req->referenceId;
 593     assoc->init->implementation_version = 0;
 594     assoc->init->implementation_id = 0;
 595     assoc->init->implementation_name = 0;
 596     assoc->init->bend_sort = NULL;
 597     assoc->init->bend_search = NULL;
 598     assoc->init->bend_present = NULL;
 599     assoc->init->bend_esrequest = NULL;
 600     assoc->init->bend_delete = NULL;
 601     assoc->init->bend_scan = NULL;
 602     assoc->init->bend_segment = NULL;
 603     assoc->init->bend_fetch = NULL;
 604
 605     assoc->init->peer_name =
 606         odr_strdup (assoc->encode, cs_addrstr(assoc->client_link));
 607     if (!(binitres = (*cb->bend_init)(assoc->init)))
 608     {
 609         yaz_log(LOG_WARN, "Bad response from backend.");
 610         return 0;
 611     }
 612
 613     assoc->backend = binitres->handle;
 614     if ((assoc->init->bend_sort))
 615         yaz_log (LOG_DEBUG, "Sort handler installed");
 616     if ((assoc->init->bend_search))
 617         yaz_log (LOG_DEBUG, "Search handler installed");
 618     if ((assoc->init->bend_present))
 619         yaz_log (LOG_DEBUG, "Present handler installed");
 620     if ((assoc->init->bend_esrequest))
 621         yaz_log (LOG_DEBUG, "ESRequest handler installed");
 622     if ((assoc->init->bend_delete))
 623         yaz_log (LOG_DEBUG, "Delete handler installed");
 624     if ((assoc->init->bend_scan))
 625         yaz_log (LOG_DEBUG, "Scan handler installed");
 626     if ((assoc->init->bend_segment))
 627         yaz_log (LOG_DEBUG, "Segment handler installed");
 628
 629     resp->referenceId = req->referenceId;
 630     *options = '\0';
 631     /* let's tell the client what we can do */
 632     if (ODR_MASK_GET(req->options, Z_Options_search))
 633     {
 634         ODR_MASK_SET(resp->options, Z_Options_search);
 635         strcat(options, "srch");
 636     }
 637     if (ODR_MASK_GET(req->options, Z_Options_present))
 638     {
 639         ODR_MASK_SET(resp->options, Z_Options_present);
 640         strcat(options, " prst");
 641     }
 642     if (ODR_MASK_GET(req->options, Z_Options_delSet) &&
 643         assoc->init->bend_delete)
 644     {
 645         ODR_MASK_SET(resp->options, Z_Options_delSet);
 646         strcat(options, " del");
 647     }
 648     if (ODR_MASK_GET(req->options, Z_Options_extendedServices) &&
 649         assoc->init->bend_esrequest)
 650     {
 651         ODR_MASK_SET(resp->options, Z_Options_extendedServices);
 652         strcat (options, " extendedServices");
 653     }
 654     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
 655     {
 656         ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
 657         strcat(options, " namedresults");
 658     }
 659     if (ODR_MASK_GET(req->options, Z_Options_scan) && assoc->init->bend_scan)
 660     {
 661         ODR_MASK_SET(resp->options, Z_Options_scan);
 662         strcat(options, " scan");
 663     }
 664     if (ODR_MASK_GET(req->options, Z_Options_concurrentOperations))
 665     {
 666         ODR_MASK_SET(resp->options, Z_Options_concurrentOperations);
 667         strcat(options, " concurop");
 668     }
 669     if (ODR_MASK_GET(req->options, Z_Options_sort) && assoc->init->bend_sort)
 670     {
 671         ODR_MASK_SET(resp->options, Z_Options_sort);
 672         strcat(options, " sort");
 673     }
 674     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_1))
 675     {
 676         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_1);
 677         assoc->version = 2; /* 1 & 2 are equivalent */
 678     }
 679     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_2))
 680     {
 681         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_2);
 682         assoc->version = 2;
 683     }
 684     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_3))
 685     {
 686         ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_3);
 687         assoc->version = 3;
 688     }
 689     yaz_log(LOG_LOG, "Negotiated to v%d: %s", assoc->version, options);
 690     assoc->maximumRecordSize = *req->maximumRecordSize;
 691     if (assoc->maximumRecordSize > control_block->maxrecordsize)
 692         assoc->maximumRecordSize = control_block->maxrecordsize;
 693     assoc->preferredMessageSize = *req->preferredMessageSize;
 694     if (assoc->preferredMessageSize > assoc->maximumRecordSize)
 695         assoc->preferredMessageSize = assoc->maximumRecordSize;
 696
 697 #if 0
 698     assoc->maximumRecordSize = 3000000;
 699     assoc->preferredMessageSize = 3000000;
 700 #endif
 701
 702     resp->preferredMessageSize = &assoc->preferredMessageSize;
 703     resp->maximumRecordSize = &assoc->maximumRecordSize;
 704
 705     resp->implementationName = "GFS/YAZ";
 706
 707     if (assoc->init->implementation_id)
 708     {
 709         char *nv = (char *)
 710             odr_malloc (assoc->encode,
 711                         strlen(assoc->init->implementation_id) + 10 +
 712                                strlen(resp->implementationId));
 713         sprintf (nv, "%s / %s",
 714                  resp->implementationId, assoc->init->implementation_id);
 715         resp->implementationId = nv;
 716     }
 717     if (assoc->init->implementation_name)
 718     {
 719         char *nv = (char *)
 720             odr_malloc (assoc->encode,
 721                         strlen(assoc->init->implementation_name) + 10 +
 722                                strlen(resp->implementationName));
 723         sprintf (nv, "%s / %s",
 724                  resp->implementationName, assoc->init->implementation_name);
 725         resp->implementationName = nv;
 726     }
 727     if (assoc->init->implementation_version)
 728     {
 729         char *nv = (char *)
 730             odr_malloc (assoc->encode,
 731                         strlen(assoc->init->implementation_version) + 10 +
 732                                strlen(resp->implementationVersion));
 733         sprintf (nv, "YAZ %s / %s",
 734                  resp->implementationVersion,
 735                  assoc->init->implementation_version);
 736         resp->implementationVersion = nv;
 737     }
 738
 739     if (binitres->errcode)
 740     {
 741         yaz_log(LOG_LOG, "Connection rejected by backend.");
 742         *resp->result = 0;
 743         assoc->state = ASSOC_DEAD;
 744     }
 745     else
 746         assoc->state = ASSOC_UP;
 747     return apdu;
 748 }
 749
 750 /*
 751  * These functions should be merged.
 752  */
 753
 754 static void set_addinfo (Z_DefaultDiagFormat *dr, char *addinfo, ODR odr)
 755 {
 756     dr->which = Z_DefaultDiagFormat_v2Addinfo;
 757     dr->u.v2Addinfo = odr_strdup (odr, addinfo ? addinfo : "");
 758 }
 759
 760 /*
 761  * nonsurrogate diagnostic record.
 762  */
 763 static Z_Records *diagrec(association *assoc, int error, char *addinfo)
 764 {
 765     Z_Records *rec = (Z_Records *)
 766         odr_malloc (assoc->encode, sizeof(*rec));
 767     int *err = odr_intdup(assoc->encode, error);
 768     Z_DiagRec *drec = (Z_DiagRec *)
 769         odr_malloc (assoc->encode, sizeof(*drec));
 770     Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *)
 771         odr_malloc (assoc->encode, sizeof(*dr));
 772
 773     yaz_log(LOG_LOG, "[%d] %s %s%s", error, diagbib1_str(error),
 774         addinfo ? " -- " : "", addinfo ? addinfo : "");
 775     rec->which = Z_Records_NSD;
 776     rec->u.nonSurrogateDiagnostic = dr;
 777     dr->diagnosticSetId =
 778         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
 779     dr->condition = err;
 780     set_addinfo (dr, addinfo, assoc->encode);
 781     return rec;
 782 }
 783
 784 /*
 785  * surrogate diagnostic.
 786  */
 787 static Z_NamePlusRecord *surrogatediagrec(association *assoc, char *dbname,
 788                                           int error, char *addinfo)
 789 {
 790     Z_NamePlusRecord *rec = (Z_NamePlusRecord *)
 791         odr_malloc (assoc->encode, sizeof(*rec));
 792     int *err = odr_intdup(assoc->encode, error);
 793     Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (assoc->encode, sizeof(*drec));
 794     Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *)
 795         odr_malloc (assoc->encode, sizeof(*dr));
 796
 797     yaz_log(LOG_DEBUG, "SurrogateDiagnotic: %d -- %s", error, addinfo);
 798     rec->databaseName = dbname;
 799     rec->which = Z_NamePlusRecord_surrogateDiagnostic;
 800     rec->u.surrogateDiagnostic = drec;
 801     drec->which = Z_DiagRec_defaultFormat;
 802     drec->u.defaultFormat = dr;
 803     dr->diagnosticSetId =
 804         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
 805     dr->condition = err;
 806     set_addinfo (dr, addinfo, assoc->encode);
 807
 808     return rec;
 809 }
 810
 811 /*
 812  * multiple nonsurrogate diagnostics.
 813  */
 814 static Z_DiagRecs *diagrecs(association *assoc, int error, char *addinfo)
 815 {
 816     Z_DiagRecs *recs = (Z_DiagRecs *)odr_malloc (assoc->encode, sizeof(*recs));
 817     int *err = odr_intdup(assoc->encode, error);
 818     Z_DiagRec **recp = (Z_DiagRec **)odr_malloc (assoc->encode, sizeof(*recp));
 819     Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (assoc->encode, sizeof(*drec));
 820     Z_DefaultDiagFormat *rec = (Z_DefaultDiagFormat *)
 821         odr_malloc (assoc->encode, sizeof(*rec));
 822
 823     yaz_log(LOG_DEBUG, "DiagRecs: %d -- %s", error, addinfo ? addinfo : "");
 824
 825     recs->num_diagRecs = 1;
 826     recs->diagRecs = recp;
 827     recp[0] = drec;
 828     drec->which = Z_DiagRec_defaultFormat;
 829     drec->u.defaultFormat = rec;
 830
 831     rec->diagnosticSetId =
 832         yaz_oidval_to_z3950oid (assoc->encode, CLASS_DIAGSET, VAL_BIB1);
 833     rec->condition = err;
 834
 835     rec->which = Z_DefaultDiagFormat_v2Addinfo;
 836     rec->u.v2Addinfo = odr_strdup (assoc->encode, addinfo ? addinfo : "");
 837     return recs;
 838 }
 839
 840 static Z_Records *pack_records(association *a, char *setname, int start,
 841                                int *num, Z_RecordComposition *comp,
 842                                int *next, int *pres, oid_value format,
 843                                Z_ReferenceId *referenceId,
 844                                int *oid)
 845 {
 846     int recno, total_length = 0, toget = *num, dumped_records = 0;
 847     Z_Records *records =
 848         (Z_Records *) odr_malloc (a->encode, sizeof(*records));
 849     Z_NamePlusRecordList *reclist =
 850         (Z_NamePlusRecordList *) odr_malloc (a->encode, sizeof(*reclist));
 851     Z_NamePlusRecord **list =
 852         (Z_NamePlusRecord **) odr_malloc (a->encode, sizeof(*list) * toget);
 853
 854     records->which = Z_Records_DBOSD;
 855     records->u.databaseOrSurDiagnostics = reclist;
 856     reclist->num_records = 0;
 857     reclist->records = list;
 858     *pres = Z_PRES_SUCCESS;
 859     *num = 0;
 860     *next = 0;
 861
 862     yaz_log(LOG_LOG, "Request to pack %d+%d+%s", start, toget, setname);
 863     yaz_log(LOG_DEBUG, "pms=%d, mrs=%d", a->preferredMessageSize,
 864         a->maximumRecordSize);
 865     for (recno = start; reclist->num_records < toget; recno++)
 866     {
 867         bend_fetch_rr freq;
 868         Z_NamePlusRecord *thisrec;
 869         int this_length = 0;
 870         /*
 871          * we get the number of bytes allocated on the stream before any
 872          * allocation done by the backend - this should give us a reasonable
 873          * idea of the total size of the data so far.
 874          */
 875         total_length = odr_total(a->encode) - dumped_records;
 876         freq.errcode = 0;
 877         freq.errstring = 0;
 878         freq.basename = 0;
 879         freq.len = 0;
 880         freq.record = 0;
 881         freq.last_in_set = 0;
 882         freq.setname = setname;
 883         freq.surrogate_flag = 0;
 884         freq.number = recno;
 885         freq.comp = comp;
 886         freq.request_format = format;
 887         freq.request_format_raw = oid;
 888         freq.output_format = format;
 889         freq.output_format_raw = 0;
 890         freq.stream = a->encode;
 891         freq.print = a->print;
 892         freq.surrogate_flag = 0;
 893         freq.referenceId = referenceId;
 894         (*a->init->bend_fetch)(a->backend, &freq);
 895         /* backend should be able to signal whether error is system-wide
 896            or only pertaining to current record */
 897         if (freq.errcode)
 898         {
 899             if (!freq.surrogate_flag)
 900             {
 901                 char s[20];
 902                 *pres = Z_PRES_FAILURE;
 903                 /* for 'present request out of range',
 904                    set addinfo to record position if not set */
 905                 if (freq.errcode == 13 && freq.errstring == 0)
 906                 {
 907                     sprintf (s, "%d", recno);
 908                     freq.errstring = s;
 909                 }
 910                 return diagrec(a, freq.errcode, freq.errstring);
 911             }
 912             reclist->records[reclist->num_records] =
 913                 surrogatediagrec(a, freq.basename, freq.errcode,
 914                                  freq.errstring);
 915             reclist->num_records++;
 916             *next = freq.last_in_set ? 0 : recno + 1;
 917             continue;
 918         }
 919         if (freq.len >= 0)
 920             this_length = freq.len;
 921         else
 922             this_length = odr_total(a->encode) - total_length;
 923         yaz_log(LOG_DEBUG, "  fetched record, len=%d, total=%d",
 924             this_length, total_length);
 925         if (this_length + total_length > a->preferredMessageSize)
 926         {
 927             /* record is small enough, really */
 928             if (this_length <= a->preferredMessageSize)
 929             {
 930                 yaz_log(LOG_DEBUG, "  Dropped last normal-sized record");
 931                 *pres = Z_PRES_PARTIAL_2;
 932                 break;
 933             }
 934             /* record can only be fetched by itself */
 935             if (this_length < a->maximumRecordSize)
 936             {
 937                 yaz_log(LOG_DEBUG, "  Record > prefmsgsz");
 938                 if (toget > 1)
 939                 {
 940                     yaz_log(LOG_DEBUG, "  Dropped it");
 941                     reclist->records[reclist->num_records] =
 942                          surrogatediagrec(a, freq.basename, 16, 0);
 943                     reclist->num_records++;
 944                     *next = freq.last_in_set ? 0 : recno + 1;
 945                     dumped_records += this_length;
 946                     continue;
 947                 }
 948             }
 949             else /* too big entirely */
 950             {
 951                 yaz_log(LOG_LOG, "Record > maxrcdsz this=%d max=%d", this_length, a->maximumRecordSize);
 952                 reclist->records[reclist->num_records] =
 953                     surrogatediagrec(a, freq.basename, 17, 0);
 954                 reclist->num_records++;
 955                 *next = freq.last_in_set ? 0 : recno + 1;
 956                 dumped_records += this_length;
 957                 continue;
 958             }
 959         }
 960
 961         if (!(thisrec = (Z_NamePlusRecord *)
 962               odr_malloc(a->encode, sizeof(*thisrec))))
 963             return 0;
 964         if (!(thisrec->databaseName = (char *)odr_malloc(a->encode,
 965             strlen(freq.basename) + 1)))
 966             return 0;
 967         strcpy(thisrec->databaseName, freq.basename);
 968         thisrec->which = Z_NamePlusRecord_databaseRecord;
 969
 970         if (freq.output_format_raw)
 971         {
 972             struct oident *ident = oid_getentbyoid(freq.output_format_raw);
 973             freq.output_format = ident->value;
 974         }
 975         thisrec->u.databaseRecord = z_ext_record(a->encode, freq.output_format,
 976                                                  freq.record, freq.len);
 977         if (!thisrec->u.databaseRecord)
 978             return 0;
 979         reclist->records[reclist->num_records] = thisrec;
 980         reclist->num_records++;
 981         *next = freq.last_in_set ? 0 : recno + 1;
 982     }
 983     *num = reclist->num_records;
 984     return records;
 985 }
 986
 987 static Z_APDU *process_searchRequest(association *assoc, request *reqb,
 988     int *fd)
 989 {
 990     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
 991     bend_search_rr *bsrr =
 992         (bend_search_rr *)nmem_malloc (reqb->request_mem, sizeof(*bsrr));
 993
 994     yaz_log(LOG_LOG, "Got SearchRequest.");
 995     bsrr->fd = fd;
 996     bsrr->request = reqb;
 997     bsrr->association = assoc;
 998     bsrr->referenceId = req->referenceId;
 999     save_referenceId (reqb, bsrr->referenceId);
1000
1001     yaz_log (LOG_LOG, "ResultSet '%s'", req->resultSetName);
1002     if (req->databaseNames)
1003     {
1004         int i;
1005         for (i = 0; i < req->num_databaseNames; i++)
1006             yaz_log (LOG_LOG, "Database '%s'", req->databaseNames[i]);
1007     }
1008     switch (req->query->which)
1009     {
1010     case Z_Query_type_1: case Z_Query_type_101:
1011         log_rpn_query (req->query->u.type_1);
1012     }
1013     if (assoc->init->bend_search)
1014     {
1015         bsrr->setname = req->resultSetName;
1016         bsrr->replace_set = *req->replaceIndicator;
1017         bsrr->num_bases = req->num_databaseNames;
1018         bsrr->basenames = req->databaseNames;
1019         bsrr->query = req->query;
1020         bsrr->stream = assoc->encode;
1021         bsrr->decode = assoc->decode;
1022         bsrr->print = assoc->print;
1023         bsrr->errcode = 0;
1024         bsrr->hits = 0;
1025         bsrr->errstring = NULL;
1026         (assoc->init->bend_search)(assoc->backend, bsrr);
1027         if (!bsrr->request)
1028             return 0;
1029     }
1030     return response_searchRequest(assoc, reqb, bsrr, fd);
1031 }
1032
1033 int bend_searchresponse(void *handle, bend_search_rr *bsrr) {return 0;}
1034
1035 /*
1036  * Prepare a searchresponse based on the backend results. We probably want
1037  * to look at making the fetching of records nonblocking as well, but
1038  * so far, we'll keep things simple.
1039  * If bsrt is null, that means we're called in response to a communications
1040  * event, and we'll have to get the response for ourselves.
1041  */
1042 static Z_APDU *response_searchRequest(association *assoc, request *reqb,
1043     bend_search_rr *bsrt, int *fd)
1044 {
1045     Z_SearchRequest *req = reqb->apdu_request->u.searchRequest;
1046     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1047     Z_SearchResponse *resp = (Z_SearchResponse *)
1048         odr_malloc (assoc->encode, sizeof(*resp));
1049     int *nulint = odr_intdup (assoc->encode, 0);
1050     bool_t *sr = odr_intdup(assoc->encode, 1);
1051     int *next = odr_intdup(assoc->encode, 0);
1052     int *none = odr_intdup(assoc->encode, Z_RES_NONE);
1053
1054     apdu->which = Z_APDU_searchResponse;
1055     apdu->u.searchResponse = resp;
1056     resp->referenceId = req->referenceId;
1057     resp->additionalSearchInfo = 0;
1058     resp->otherInfo = 0;
1059     *fd = -1;
1060     if (!bsrt && !bend_searchresponse(assoc->backend, bsrt))
1061     {
1062         yaz_log(LOG_FATAL, "Bad result from backend");
1063         return 0;
1064     }
1065     else if (bsrt->errcode)
1066     {
1067         resp->records = diagrec(assoc, bsrt->errcode, bsrt->errstring);
1068         resp->resultCount = nulint;
1069         resp->numberOfRecordsReturned = nulint;
1070         resp->nextResultSetPosition = nulint;
1071         resp->searchStatus = nulint;
1072         resp->resultSetStatus = none;
1073         resp->presentStatus = 0;
1074     }
1075     else
1076     {
1077         int *toget = odr_intdup(assoc->encode, 0);
1078         int *presst = odr_intdup(assoc->encode, 0);
1079         Z_RecordComposition comp, *compp = 0;
1080
1081         yaz_log (LOG_LOG, "resultCount: %d", bsrt->hits);
1082
1083         resp->records = 0;
1084         resp->resultCount = &bsrt->hits;
1085
1086         comp.which = Z_RecordComp_simple;
1087         /* how many records does the user agent want, then? */
1088         if (bsrt->hits <= *req->smallSetUpperBound)
1089         {
1090             *toget = bsrt->hits;
1091             if ((comp.u.simple = req->smallSetElementSetNames))
1092                 compp = &comp;
1093         }
1094         else if (bsrt->hits < *req->largeSetLowerBound)
1095         {
1096             *toget = *req->mediumSetPresentNumber;
1097             if (*toget > bsrt->hits)
1098                 *toget = bsrt->hits;
1099             if ((comp.u.simple = req->mediumSetElementSetNames))
1100                 compp = &comp;
1101         }
1102         else
1103             *toget = 0;
1104
1105         if (*toget && !resp->records)
1106         {
1107             oident *prefformat;
1108             oid_value form;
1109
1110             if (!(prefformat = oid_getentbyoid(req->preferredRecordSyntax)))
1111                 form = VAL_NONE;
1112             else
1113                 form = prefformat->value;
1114             resp->records = pack_records(assoc, req->resultSetName, 1,
1115                 toget, compp, next, presst, form, req->referenceId,
1116                                          req->preferredRecordSyntax);
1117             if (!resp->records)
1118                 return 0;
1119             resp->numberOfRecordsReturned = toget;
1120             resp->nextResultSetPosition = next;
1121             resp->searchStatus = sr;
1122             resp->resultSetStatus = 0;
1123             resp->presentStatus = presst;
1124         }
1125         else
1126         {
1127             if (*resp->resultCount)
1128                 *next = 1;
1129             resp->numberOfRecordsReturned = nulint;
1130             resp->nextResultSetPosition = next;
1131             resp->searchStatus = sr;
1132             resp->resultSetStatus = 0;
1133             resp->presentStatus = 0;
1134         }
1135     }
1136     return apdu;
1137 }
1138
1139 /*
1140  * Maybe we got a little over-friendly when we designed bend_fetch to
1141  * get only one record at a time. Some backends can optimise multiple-record
1142  * fetches, and at any rate, there is some overhead involved in
1143  * all that selecting and hopping around. Problem is, of course, that the
1144  * frontend can't know ahead of time how many records it'll need to
1145  * fill the negotiated PDU size. Annoying. Segmentation or not, Z/SR
1146  * is downright lousy as a bulk data transfer protocol.
1147  *
1148  * To start with, we'll do the fetching of records from the backend
1149  * in one operation: To save some trips in and out of the event-handler,
1150  * and to simplify the interface to pack_records. At any rate, asynch
1151  * operation is more fun in operations that have an unpredictable execution
1152  * speed - which is normally more true for search than for present.
1153  */
1154 static Z_APDU *process_presentRequest(association *assoc, request *reqb,
1155                                       int *fd)
1156 {
1157     Z_PresentRequest *req = reqb->apdu_request->u.presentRequest;
1158     oident *prefformat;
1159     oid_value form;
1160     Z_APDU *apdu;
1161     Z_PresentResponse *resp;
1162     int *next;
1163     int *num;
1164
1165     yaz_log(LOG_LOG, "Got PresentRequest.");
1166
1167     if (!(prefformat = oid_getentbyoid(req->preferredRecordSyntax)))
1168         form = VAL_NONE;
1169     else
1170         form = prefformat->value;
1171     resp = (Z_PresentResponse *)odr_malloc (assoc->encode, sizeof(*resp));
1172     resp->records = 0;
1173     resp->presentStatus = odr_intdup(assoc->encode, 0);
1174     if (assoc->init->bend_present)
1175     {
1176         bend_present_rr *bprr = (bend_present_rr *)
1177             nmem_malloc (reqb->request_mem, sizeof(*bprr));
1178         bprr->setname = req->resultSetId;
1179         bprr->start = *req->resultSetStartPoint;
1180         bprr->number = *req->numberOfRecordsRequested;
1181         bprr->format = form;
1182         bprr->comp = req->recordComposition;
1183         bprr->referenceId = req->referenceId;
1184         bprr->stream = assoc->encode;
1185         bprr->print = assoc->print;
1186         bprr->request = reqb;
1187         bprr->association = assoc;
1188         bprr->errcode = 0;
1189         bprr->errstring = NULL;
1190         (*assoc->init->bend_present)(assoc->backend, bprr);
1191
1192         if (!bprr->request)
1193             return 0;
1194         if (bprr->errcode)
1195         {
1196             resp->records = diagrec(assoc, bprr->errcode, bprr->errstring);
1197             *resp->presentStatus = Z_PRES_FAILURE;
1198         }
1199     }
1200     apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1201     next = odr_intdup(assoc->encode, 0);
1202     num = odr_intdup(assoc->encode, 0);
1203
1204     apdu->which = Z_APDU_presentResponse;
1205     apdu->u.presentResponse = resp;
1206     resp->referenceId = req->referenceId;
1207     resp->otherInfo = 0;
1208
1209     if (!resp->records)
1210     {
1211         *num = *req->numberOfRecordsRequested;
1212         resp->records =
1213             pack_records(assoc, req->resultSetId, *req->resultSetStartPoint,
1214                      num, req->recordComposition, next, resp->presentStatus,
1215                          form, req->referenceId, req->preferredRecordSyntax);
1216     }
1217     if (!resp->records)
1218         return 0;
1219     resp->numberOfRecordsReturned = num;
1220     resp->nextResultSetPosition = next;
1221
1222     return apdu;
1223 }
1224
1225 /*
1226  * Scan was implemented rather in a hurry, and with support for only the basic
1227  * elements of the service in the backend API. Suggestions are welcome.
1228  */
1229 static Z_APDU *process_scanRequest(association *assoc, request *reqb, int *fd)
1230 {
1231     Z_ScanRequest *req = reqb->apdu_request->u.scanRequest;
1232     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1233     Z_ScanResponse *res = (Z_ScanResponse *)
1234         odr_malloc (assoc->encode, sizeof(*res));
1235     int *scanStatus = odr_intdup(assoc->encode, Z_Scan_failure);
1236     int *numberOfEntriesReturned = odr_intdup(assoc->encode, 0);
1237     Z_ListEntries *ents = (Z_ListEntries *)
1238         odr_malloc (assoc->encode, sizeof(*ents));
1239     Z_DiagRecs *diagrecs_p = NULL;
1240     oident *attset;
1241     bend_scan_rr *bsrr = (bend_scan_rr *)
1242         odr_malloc (assoc->encode, sizeof(*bsrr));
1243
1244     yaz_log(LOG_LOG, "Got ScanRequest");
1245
1246     apdu->which = Z_APDU_scanResponse;
1247     apdu->u.scanResponse = res;
1248     res->referenceId = req->referenceId;
1249
1250     /* if step is absent, set it to 0 */
1251     res->stepSize = odr_intdup(assoc->encode, 0);
1252     if (req->stepSize)
1253         *res->stepSize = *req->stepSize;
1254
1255     res->scanStatus = scanStatus;
1256     res->numberOfEntriesReturned = numberOfEntriesReturned;
1257     res->positionOfTerm = 0;
1258     res->entries = ents;
1259     ents->num_entries = 0;
1260     ents->entries = NULL;
1261     ents->num_nonsurrogateDiagnostics = 0;
1262     ents->nonsurrogateDiagnostics = NULL;
1263     res->attributeSet = 0;
1264     res->otherInfo = 0;
1265
1266     if (req->databaseNames)
1267     {
1268         int i;
1269         for (i = 0; i < req->num_databaseNames; i++)
1270             yaz_log (LOG_LOG, "Database '%s'", req->databaseNames[i]);
1271     }
1272     bsrr->num_bases = req->num_databaseNames;
1273     bsrr->basenames = req->databaseNames;
1274     bsrr->num_entries = *req->numberOfTermsRequested;
1275     bsrr->term = req->termListAndStartPoint;
1276     bsrr->referenceId = req->referenceId;
1277     bsrr->stream = assoc->encode;
1278     bsrr->print = assoc->print;
1279     bsrr->step_size = res->stepSize;
1280     if (req->attributeSet &&
1281         (attset = oid_getentbyoid(req->attributeSet)) &&
1282         (attset->oclass == CLASS_ATTSET || attset->oclass == CLASS_GENERAL))
1283         bsrr->attributeset = attset->value;
1284     else
1285         bsrr->attributeset = VAL_NONE;
1286     log_scan_term (req->termListAndStartPoint, bsrr->attributeset);
1287     bsrr->term_position = req->preferredPositionInResponse ?
1288         *req->preferredPositionInResponse : 1;
1289     ((int (*)(void *, bend_scan_rr *))
1290      (*assoc->init->bend_scan))(assoc->backend, bsrr);
1291     if (bsrr->errcode)
1292         diagrecs_p = diagrecs(assoc, bsrr->errcode, bsrr->errstring);
1293     else
1294     {
1295         int i;
1296         Z_Entry **tab = (Z_Entry **)
1297             odr_malloc (assoc->encode, sizeof(*tab) * bsrr->num_entries);
1298
1299         if (bsrr->status == BEND_SCAN_PARTIAL)
1300             *scanStatus = Z_Scan_partial_5;
1301         else
1302             *scanStatus = Z_Scan_success;
1303         ents->entries = tab;
1304         ents->num_entries = bsrr->num_entries;
1305         res->numberOfEntriesReturned = &ents->num_entries;
1306         res->positionOfTerm = &bsrr->term_position;
1307         for (i = 0; i < bsrr->num_entries; i++)
1308         {
1309             Z_Entry *e;
1310             Z_TermInfo *t;
1311             Odr_oct *o;
1312
1313             tab[i] = e = (Z_Entry *)odr_malloc(assoc->encode, sizeof(*e));
1314             if (bsrr->entries[i].occurrences >= 0)
1315             {
1316                 e->which = Z_Entry_termInfo;
1317                 e->u.termInfo = t = (Z_TermInfo *)
1318                     odr_malloc(assoc->encode, sizeof(*t));
1319                 t->suggestedAttributes = 0;
1320                 t->displayTerm = 0;
1321                 t->alternativeTerm = 0;
1322                 t->byAttributes = 0;
1323                 t->otherTermInfo = 0;
1324                 t->globalOccurrences = &bsrr->entries[i].occurrences;
1325                 t->term = (Z_Term *)
1326                     odr_malloc(assoc->encode, sizeof(*t->term));
1327                 t->term->which = Z_Term_general;
1328                 t->term->u.general = o =
1329                     (Odr_oct *)odr_malloc(assoc->encode, sizeof(Odr_oct));
1330                 o->buf = (unsigned char *)
1331                     odr_malloc(assoc->encode, o->len = o->size =
1332                                strlen(bsrr->entries[i].term));
1333                 memcpy(o->buf, bsrr->entries[i].term, o->len);
1334                 yaz_log(LOG_DEBUG, "  term #%d: '%s' (%d)", i,
1335                          bsrr->entries[i].term, bsrr->entries[i].occurrences);
1336             }
1337             else
1338             {
1339                 Z_DiagRecs *drecs = diagrecs (assoc,
1340                                               bsrr->entries[i].errcode,
1341                                               bsrr->entries[i].errstring);
1342                 assert (drecs->num_diagRecs == 1);
1343                 e->which = Z_Entry_surrogateDiagnostic;
1344                 assert (drecs->diagRecs[0]);
1345                 e->u.surrogateDiagnostic = drecs->diagRecs[0];
1346             }
1347         }
1348     }
1349     if (diagrecs_p)
1350     {
1351         ents->num_nonsurrogateDiagnostics = diagrecs_p->num_diagRecs;
1352         ents->nonsurrogateDiagnostics = diagrecs_p->diagRecs;
1353     }
1354     return apdu;
1355 }
1356
1357 static Z_APDU *process_sortRequest(association *assoc, request *reqb,
1358     int *fd)
1359 {
1360     Z_SortRequest *req = reqb->apdu_request->u.sortRequest;
1361     Z_SortResponse *res = (Z_SortResponse *)
1362         odr_malloc (assoc->encode, sizeof(*res));
1363     bend_sort_rr *bsrr = (bend_sort_rr *)
1364         odr_malloc (assoc->encode, sizeof(*bsrr));
1365
1366     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1367
1368     yaz_log(LOG_LOG, "Got SortRequest.");
1369
1370     bsrr->num_input_setnames = req->num_inputResultSetNames;
1371     bsrr->input_setnames = req->inputResultSetNames;
1372     bsrr->referenceId = req->referenceId;
1373     bsrr->output_setname = req->sortedResultSetName;
1374     bsrr->sort_sequence = req->sortSequence;
1375     bsrr->stream = assoc->encode;
1376     bsrr->print = assoc->print;
1377
1378     bsrr->sort_status = Z_SortStatus_failure;
1379     bsrr->errcode = 0;
1380     bsrr->errstring = 0;
1381
1382     (*assoc->init->bend_sort)(assoc->backend, bsrr);
1383
1384     res->referenceId = bsrr->referenceId;
1385     res->sortStatus = odr_intdup(assoc->encode, bsrr->sort_status);
1386     res->resultSetStatus = 0;
1387     if (bsrr->errcode)
1388     {
1389         Z_DiagRecs *dr = diagrecs (assoc, bsrr->errcode, bsrr->errstring);
1390         res->diagnostics = dr->diagRecs;
1391         res->num_diagnostics = dr->num_diagRecs;
1392     }
1393     else
1394     {
1395         res->num_diagnostics = 0;
1396         res->diagnostics = 0;
1397     }
1398     res->otherInfo = 0;
1399
1400     apdu->which = Z_APDU_sortResponse;
1401     apdu->u.sortResponse = res;
1402     return apdu;
1403 }
1404
1405 static Z_APDU *process_deleteRequest(association *assoc, request *reqb,
1406     int *fd)
1407 {
1408     Z_DeleteResultSetRequest *req =
1409         reqb->apdu_request->u.deleteResultSetRequest;
1410     Z_DeleteResultSetResponse *res = (Z_DeleteResultSetResponse *)
1411         odr_malloc (assoc->encode, sizeof(*res));
1412     bend_delete_rr *bdrr = (bend_delete_rr *)
1413         odr_malloc (assoc->encode, sizeof(*bdrr));
1414     Z_APDU *apdu = (Z_APDU *)odr_malloc (assoc->encode, sizeof(*apdu));
1415
1416     yaz_log(LOG_LOG, "Got DeleteRequest.");
1417
1418     bdrr->num_setnames = req->num_resultSetList;
1419     bdrr->setnames = req->resultSetList;
1420     bdrr->stream = assoc->encode;
1421     bdrr->print = assoc->print;
1422     bdrr->function = *req->deleteFunction;
1423     bdrr->referenceId = req->referenceId;
1424     bdrr->statuses = 0;
1425     if (bdrr->num_setnames > 0)
1426     {
1427         int i;
1428         bdrr->statuses = (int*)
1429             odr_malloc(assoc->encode, sizeof(*bdrr->statuses) *
1430                        bdrr->num_setnames);
1431         for (i = 0; i < bdrr->num_setnames; i++)
1432             bdrr->statuses[i] = 0;
1433     }
1434     (*assoc->init->bend_delete)(assoc->backend, bdrr);
1435
1436     res->referenceId = req->referenceId;
1437
1438     res->deleteOperationStatus = odr_intdup(assoc->encode,bdrr->delete_status);
1439
1440     res->deleteListStatuses = 0;
1441     if (bdrr->num_setnames > 0)
1442     {
1443         int i;
1444         res->deleteListStatuses = (Z_ListStatuses *)
1445             odr_malloc(assoc->encode, sizeof(*res->deleteListStatuses));
1446         res->deleteListStatuses->num = bdrr->num_setnames;
1447         res->deleteListStatuses->elements =
1448             (Z_ListStatus **)
1449             odr_malloc (assoc->encode,
1450                         sizeof(*res->deleteListStatuses->elements) *
1451                         bdrr->num_setnames);
1452         for (i = 0; i<bdrr->num_setnames; i++)
1453         {
1454             res->deleteListStatuses->elements[i] =
1455                 (Z_ListStatus *)
1456                 odr_malloc (assoc->encode,
1457                             sizeof(**res->deleteListStatuses->elements));
1458             res->deleteListStatuses->elements[i]->status = bdrr->statuses+i;
1459             res->deleteListStatuses->elements[i]->id =
1460                 odr_strdup (assoc->encode, bdrr->setnames[i]);
1461
1462         }
1463     }
1464     res->numberNotDeleted = 0;
1465     res->bulkStatuses = 0;
1466     res->deleteMessage = 0;
1467     res->otherInfo = 0;
1468
1469     apdu->which = Z_APDU_deleteResultSetResponse;
1470     apdu->u.deleteResultSetResponse = res;
1471     return apdu;
1472 }
1473
1474 static void process_close(association *assoc, request *reqb)
1475 {
1476     Z_Close *req = reqb->apdu_request->u.close;
1477     static char *reasons[] =
1478     {
1479         "finished",
1480         "shutdown",
1481         "systemProblem",
1482         "costLimit",
1483         "resources",
1484         "securityViolation",
1485         "protocolError",
1486         "lackOfActivity",
1487         "peerAbort",
1488         "unspecified"
1489     };
1490
1491     yaz_log(LOG_LOG, "Got Close, reason %s, message %s",
1492         reasons[*req->closeReason], req->diagnosticInformation ?
1493         req->diagnosticInformation : "NULL");
1494     if (assoc->version < 3) /* to make do_force respond with close */
1495         assoc->version = 3;
1496     do_close_req(assoc, Z_Close_finished,
1497                  "Association terminated by client", reqb);
1498 }
1499
1500 void save_referenceId (request *reqb, Z_ReferenceId *refid)
1501 {
1502     if (refid)
1503     {
1504         reqb->len_refid = refid->len;
1505         reqb->refid = (char *)nmem_malloc (reqb->request_mem, refid->len);
1506         memcpy (reqb->refid, refid->buf, refid->len);
1507     }
1508     else
1509     {
1510         reqb->len_refid = 0;
1511         reqb->refid = NULL;
1512     }
1513 }
1514
1515 void bend_request_send (bend_association a, bend_request req, Z_APDU *res)
1516 {
1517     process_response (a, req, res);
1518 }
1519
1520 bend_request bend_request_mk (bend_association a)
1521 {
1522     request *nreq = request_get (&a->outgoing);
1523     nreq->request_mem = nmem_create ();
1524     return nreq;
1525 }
1526
1527 Z_ReferenceId *bend_request_getid (ODR odr, bend_request req)
1528 {
1529     Z_ReferenceId *id;
1530     if (!req->refid)
1531         return 0;
1532     id = (Odr_oct *)odr_malloc (odr, sizeof(*odr));
1533     id->buf = (unsigned char *)odr_malloc (odr, req->len_refid);
1534     id->len = id->size = req->len_refid;
1535     memcpy (id->buf, req->refid, req->len_refid);
1536     return id;
1537 }
1538
1539 void bend_request_destroy (bend_request *req)
1540 {
1541     nmem_destroy((*req)->request_mem);
1542     request_release(*req);
1543     *req = NULL;
1544 }
1545
1546 int bend_backend_respond (bend_association a, bend_request req)
1547 {
1548     char *msg;
1549     int r;
1550     r = process_request (a, req, &msg);
1551     if (r < 0)
1552         logf (LOG_WARN, "%s", msg);
1553     return r;
1554 }
1555
1556 void bend_request_setdata(bend_request r, void *p)
1557 {
1558     r->clientData = p;
1559 }
1560
1561 void *bend_request_getdata(bend_request r)
1562 {
1563     return r->clientData;
1564 }
1565
1566 static Z_APDU *process_segmentRequest (association *assoc, request *reqb)
1567 {
1568     bend_segment_rr request;
1569
1570     request.segment = reqb->apdu_request->u.segmentRequest;
1571     request.stream = assoc->encode;
1572     request.decode = assoc->decode;
1573     request.print = assoc->print;
1574     request.association = assoc;
1575
1576     (*assoc->init->bend_segment)(assoc->backend, &request);
1577
1578     return 0;
1579 }
1580
1581 static Z_APDU *process_ESRequest(association *assoc, request *reqb, int *fd)
1582 {
1583     bend_esrequest_rr esrequest;
1584
1585     Z_ExtendedServicesRequest *req = reqb->apdu_request->u.extendedServicesRequest;
1586     Z_APDU *apdu = zget_APDU(assoc->encode, Z_APDU_extendedServicesResponse);
1587
1588     Z_ExtendedServicesResponse *resp = apdu->u.extendedServicesResponse;
1589
1590     yaz_log(LOG_DEBUG,"inside Process esRequest");
1591
1592     esrequest.esr = reqb->apdu_request->u.extendedServicesRequest;
1593     esrequest.stream = assoc->encode;
1594     esrequest.decode = assoc->decode;
1595     esrequest.print = assoc->print;
1596     esrequest.errcode = 0;
1597     esrequest.errstring = NULL;
1598     esrequest.request = reqb;
1599     esrequest.association = assoc;
1600     esrequest.taskPackage = 0;
1601     esrequest.referenceId = req->referenceId;
1602
1603     (*assoc->init->bend_esrequest)(assoc->backend, &esrequest);
1604
1605     /* If the response is being delayed, return NULL */
1606     if (esrequest.request == NULL)
1607         return(NULL);
1608
1609     resp->referenceId = req->referenceId;
1610
1611     if (esrequest.errcode == -1)
1612     {
1613         /* Backend service indicates request will be processed */
1614         yaz_log(LOG_DEBUG,"Request could be processed...Accepted !");
1615         *resp->operationStatus = Z_ExtendedServicesResponse_accepted;
1616     }
1617     else if (esrequest.errcode == 0)
1618     {
1619         /* Backend service indicates request will be processed */
1620         yaz_log(LOG_DEBUG,"Request could be processed...Done !");
1621         *resp->operationStatus = Z_ExtendedServicesResponse_done;
1622     }
1623     else
1624     {
1625         Z_DiagRecs *diagRecs = diagrecs (assoc, esrequest.errcode,
1626                                          esrequest.errstring);
1627
1628         /* Backend indicates error, request will not be processed */
1629         yaz_log(LOG_DEBUG,"Request could not be processed...failure !");
1630         *resp->operationStatus = Z_ExtendedServicesResponse_failure;
1631         resp->num_diagnostics = diagRecs->num_diagRecs;
1632         resp->diagnostics = diagRecs->diagRecs;
1633     }
1634     /* Do something with the members of bend_extendedservice */
1635     if (esrequest.taskPackage)
1636         resp->taskPackage = z_ext_record (assoc->encode, VAL_EXTENDED,
1637                                          (const char *)  esrequest.taskPackage,
1638                                           -1);
1639     yaz_log(LOG_DEBUG,"Send the result apdu");
1640     return apdu;
1641 }