src/zoom-c.c

   1 /*
   2  * Copyright (C) 1995-2007, Index Data ApS
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: zoom-c.c,v 1.145 2007-09-06 12:40:53 mike Exp $
   6  */
   7 /**
   8  * \file zoom-c.c
   9  * \brief Implements ZOOM C interface.
  10  */
  11
  12 #include <assert.h>
  13 #include <string.h>
  14 #include <errno.h>
  15 #include "zoom-p.h"
  16
  17 #include <yaz/yaz-util.h>
  18 #include <yaz/xmalloc.h>
  19 #include <yaz/otherinfo.h>
  20 #include <yaz/log.h>
  21 #include <yaz/pquery.h>
  22 #include <yaz/marcdisp.h>
  23 #include <yaz/diagbib1.h>
  24 #include <yaz/charneg.h>
  25 #include <yaz/ill.h>
  26 #include <yaz/srw.h>
  27 #include <yaz/cql.h>
  28 #include <yaz/ccl.h>
  29 #include <yaz/query-charset.h>
  30 #include <yaz/copy_types.h>
  31 #include <yaz/snprintf.h>
  32
  33 static int log_api = 0;
  34 static int log_details = 0;
  35
  36 typedef enum {
  37     zoom_pending,
  38     zoom_complete
  39 } zoom_ret;
  40
  41 static void resultset_destroy(ZOOM_resultset r);
  42 static zoom_ret ZOOM_connection_send_init(ZOOM_connection c);
  43 static zoom_ret do_write_ex(ZOOM_connection c, char *buf_out, int len_out);
  44 static char *cql2pqf(ZOOM_connection c, const char *cql);
  45
  46
  47 /*
  48  * This wrapper is just for logging failed lookups.  It would be nicer
  49  * if it could cause failure when a lookup fails, but that's hard.
  50  */
  51 static Odr_oid *zoom_yaz_str_to_z3950oid(ZOOM_connection c,
  52                                      int oid_class, const char *str) {
  53     Odr_oid *res = yaz_string_to_oid_odr(yaz_oid_std(), oid_class, str,
  54                                      c->odr_out);
  55     if (res == 0)
  56         yaz_log(YLOG_WARN, "%p OID lookup (%d, '%s') failed",
  57                 c, (int) oid_class, str);
  58     return res;
  59 }
  60
  61
  62 static void initlog(void)
  63 {
  64     static int log_level_initialized = 0;
  65     if (!log_level_initialized)
  66     {
  67         log_api = yaz_log_module_level("zoom");
  68         log_details = yaz_log_module_level("zoomdetails");
  69         log_level_initialized = 1;
  70     }
  71 }
  72
  73 static ZOOM_Event ZOOM_Event_create(int kind)
  74 {
  75     ZOOM_Event event = (ZOOM_Event) xmalloc(sizeof(*event));
  76     event->kind = kind;
  77     event->next = 0;
  78     event->prev = 0;
  79     yaz_log(log_details, "ZOOM_Event_create(kind=%d)", kind);
  80     return event;
  81 }
  82
  83 static void ZOOM_Event_destroy(ZOOM_Event event)
  84 {
  85     xfree(event);
  86 }
  87
  88 static void ZOOM_connection_put_event(ZOOM_connection c, ZOOM_Event event)
  89 {
  90     if (c->m_queue_back)
  91     {
  92         c->m_queue_back->prev = event;
  93         assert(c->m_queue_front);
  94     }
  95     else
  96     {
  97         assert(!c->m_queue_front);
  98         c->m_queue_front = event;
  99     }
 100     event->next = c->m_queue_back;
 101     event->prev = 0;
 102     c->m_queue_back = event;
 103 }
 104
 105 static ZOOM_Event ZOOM_connection_get_event(ZOOM_connection c)
 106 {
 107     ZOOM_Event event = c->m_queue_front;
 108     if (!event)
 109     {
 110         c->last_event = ZOOM_EVENT_NONE;
 111         return 0;
 112     }
 113     assert(c->m_queue_back);
 114     c->m_queue_front = event->prev;
 115     if (c->m_queue_front)
 116     {
 117         assert(c->m_queue_back);
 118         c->m_queue_front->next = 0;
 119     }
 120     else
 121         c->m_queue_back = 0;
 122     c->last_event = event->kind;
 123     return event;
 124 }
 125
 126 static void ZOOM_connection_remove_events(ZOOM_connection c)
 127 {
 128     ZOOM_Event event;
 129     while ((event = ZOOM_connection_get_event(c)))
 130         ZOOM_Event_destroy(event);
 131 }
 132
 133 ZOOM_API(int) ZOOM_connection_peek_event(ZOOM_connection c)
 134 {
 135     ZOOM_Event event = c->m_queue_front;
 136
 137     return event ? event->kind : ZOOM_EVENT_NONE;
 138 }
 139
 140 void ZOOM_connection_remove_tasks(ZOOM_connection c);
 141
 142 static void set_dset_error(ZOOM_connection c, int error,
 143                            const char *dset,
 144                            const char *addinfo, const char *addinfo2)
 145 {
 146     char *cp;
 147
 148     xfree(c->addinfo);
 149     c->addinfo = 0;
 150     c->error = error;
 151     if (!c->diagset || strcmp(dset, c->diagset))
 152     {
 153         xfree(c->diagset);
 154         c->diagset = xstrdup(dset);
 155         /* remove integer part from SRW diagset .. */
 156         if ((cp = strrchr(c->diagset, '/')))
 157             *cp = '\0';
 158     }
 159     if (addinfo && addinfo2)
 160     {
 161         c->addinfo = (char*) xmalloc(strlen(addinfo) + strlen(addinfo2) + 2);
 162         strcpy(c->addinfo, addinfo);
 163         strcat(c->addinfo, addinfo2);
 164     }
 165     else if (addinfo)
 166         c->addinfo = xstrdup(addinfo);
 167     if (error != ZOOM_ERROR_NONE)
 168     {
 169         yaz_log(log_api, "%p set_dset_error %s %s:%d %s %s",
 170                 c, c->host_port ? c->host_port : "<>", dset, error,
 171                 addinfo ? addinfo : "",
 172                 addinfo2 ? addinfo2 : "");
 173         ZOOM_connection_remove_tasks(c);
 174     }
 175 }
 176
 177 #if YAZ_HAVE_XML2
 178 static void set_HTTP_error(ZOOM_connection c, int error,
 179                            const char *addinfo, const char *addinfo2)
 180 {
 181     set_dset_error(c, error, "HTTP", addinfo, addinfo2);
 182 }
 183
 184 static void set_SRU_error(ZOOM_connection c, Z_SRW_diagnostic *d)
 185 {
 186     const char *uri = d->uri;
 187     if (uri)
 188     {
 189         int code = 0;
 190         const char *cp;
 191         if ((cp = strrchr(uri, '/')))
 192             code = atoi(cp+1);
 193         set_dset_error(c, code, uri, d->details, 0);
 194     }
 195 }
 196
 197 #endif
 198
 199
 200 static void set_ZOOM_error(ZOOM_connection c, int error,
 201                            const char *addinfo)
 202 {
 203     set_dset_error(c, error, "ZOOM", addinfo, 0);
 204 }
 205
 206 static void clear_error(ZOOM_connection c)
 207 {
 208     /*
 209      * If an error is tied to an operation then it's ok to clear: for
 210      * example, a diagnostic returned from a search is cleared by a
 211      * subsequent search.  However, problems such as Connection Lost
 212      * or Init Refused are not cleared, because they are not
 213      * recoverable: doing another search doesn't help.
 214      */
 215
 216     ZOOM_connection_remove_events(c);
 217     switch (c->error)
 218     {
 219     case ZOOM_ERROR_CONNECT:
 220     case ZOOM_ERROR_MEMORY:
 221     case ZOOM_ERROR_DECODE:
 222     case ZOOM_ERROR_CONNECTION_LOST:
 223     case ZOOM_ERROR_INIT:
 224     case ZOOM_ERROR_INTERNAL:
 225     case ZOOM_ERROR_UNSUPPORTED_PROTOCOL:
 226         break;
 227     default:
 228         set_ZOOM_error(c, ZOOM_ERROR_NONE, 0);
 229     }
 230 }
 231
 232 void ZOOM_connection_show_task(ZOOM_task task)
 233 {
 234     switch(task->which)
 235     {
 236     case ZOOM_TASK_SEARCH:
 237         yaz_log(YLOG_LOG, "search p=%p", task);
 238         break;
 239     case ZOOM_TASK_RETRIEVE:
 240         yaz_log(YLOG_LOG, "retrieve p=%p", task);
 241         break;
 242     case ZOOM_TASK_CONNECT:
 243         yaz_log(YLOG_LOG, "connect p=%p", task);
 244         break;
 245     case ZOOM_TASK_SCAN:
 246         yaz_log(YLOG_LOG, "scant p=%p", task);
 247         break;
 248     }
 249 }
 250
 251 void ZOOM_connection_show_tasks(ZOOM_connection c)
 252 {
 253     ZOOM_task task;
 254     yaz_log(YLOG_LOG, "connection p=%p tasks", c);
 255     for (task = c->tasks; task; task = task->next)
 256         ZOOM_connection_show_task(task);
 257 }
 258
 259 ZOOM_task ZOOM_connection_add_task(ZOOM_connection c, int which)
 260 {
 261     ZOOM_task *taskp = &c->tasks;
 262     while (*taskp)
 263         taskp = &(*taskp)->next;
 264     *taskp = (ZOOM_task) xmalloc(sizeof(**taskp));
 265     (*taskp)->running = 0;
 266     (*taskp)->which = which;
 267     (*taskp)->next = 0;
 268     clear_error(c);
 269     return *taskp;
 270 }
 271
 272 ZOOM_API(int) ZOOM_connection_is_idle(ZOOM_connection c)
 273 {
 274     return c->tasks ? 0 : 1;
 275 }
 276
 277 ZOOM_task ZOOM_connection_insert_task(ZOOM_connection c, int which)
 278 {
 279     ZOOM_task task = (ZOOM_task) xmalloc(sizeof(*task));
 280
 281     task->next = c->tasks;
 282     c->tasks = task;
 283
 284     task->running = 0;
 285     task->which = which;
 286     clear_error(c);
 287     return task;
 288 }
 289
 290 void ZOOM_connection_remove_task(ZOOM_connection c)
 291 {
 292     ZOOM_task task = c->tasks;
 293
 294     if (task)
 295     {
 296         c->tasks = task->next;
 297         switch (task->which)
 298         {
 299         case ZOOM_TASK_SEARCH:
 300             resultset_destroy(task->u.search.resultset);
 301             xfree(task->u.search.syntax);
 302             xfree(task->u.search.elementSetName);
 303             break;
 304         case ZOOM_TASK_RETRIEVE:
 305             resultset_destroy(task->u.retrieve.resultset);
 306             xfree(task->u.retrieve.syntax);
 307             xfree(task->u.retrieve.elementSetName);
 308             break;
 309         case ZOOM_TASK_CONNECT:
 310             break;
 311         case ZOOM_TASK_SCAN:
 312             ZOOM_scanset_destroy(task->u.scan.scan);
 313             break;
 314         case ZOOM_TASK_PACKAGE:
 315             ZOOM_package_destroy(task->u.package);
 316             break;
 317         case ZOOM_TASK_SORT:
 318             resultset_destroy(task->u.sort.resultset);
 319             ZOOM_query_destroy(task->u.sort.q);
 320             break;
 321         default:
 322             assert(0);
 323         }
 324         xfree(task);
 325
 326         if (!c->tasks)
 327         {
 328             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_END);
 329             ZOOM_connection_put_event(c, event);
 330         }
 331     }
 332 }
 333
 334 static int ZOOM_connection_exec_task(ZOOM_connection c);
 335
 336 void ZOOM_connection_remove_tasks(ZOOM_connection c)
 337 {
 338     while (c->tasks)
 339         ZOOM_connection_remove_task(c);
 340 }
 341
 342 static ZOOM_record record_cache_lookup(ZOOM_resultset r, int pos,
 343                                        const char *syntax,
 344                                        const char *elementSetName);
 345
 346 ZOOM_API(ZOOM_connection)
 347     ZOOM_connection_create(ZOOM_options options)
 348 {
 349     ZOOM_connection c = (ZOOM_connection) xmalloc(sizeof(*c));
 350
 351     initlog();
 352
 353     yaz_log(log_api, "%p ZOOM_connection_create", c);
 354
 355     c->proto = PROTO_Z3950;
 356     c->cs = 0;
 357     ZOOM_connection_set_mask(c, 0);
 358     c->reconnect_ok = 0;
 359     c->state = STATE_IDLE;
 360     c->addinfo = 0;
 361     c->diagset = 0;
 362     set_ZOOM_error(c, ZOOM_ERROR_NONE, 0);
 363     c->buf_in = 0;
 364     c->len_in = 0;
 365     c->buf_out = 0;
 366     c->len_out = 0;
 367     c->resultsets = 0;
 368
 369     c->options = ZOOM_options_create_with_parent(options);
 370
 371     c->host_port = 0;
 372     c->path = 0;
 373     c->proxy = 0;
 374
 375     c->charset = c->lang = 0;
 376
 377     c->cookie_out = 0;
 378     c->cookie_in = 0;
 379     c->client_IP = 0;
 380     c->tasks = 0;
 381
 382     c->user = 0;
 383     c->group = 0;
 384     c->password = 0;
 385
 386     c->maximum_record_size = 0;
 387     c->preferred_message_size = 0;
 388
 389     c->odr_in = odr_createmem(ODR_DECODE);
 390     c->odr_out = odr_createmem(ODR_ENCODE);
 391
 392     c->async = 0;
 393     c->support_named_resultsets = 0;
 394     c->last_event = ZOOM_EVENT_NONE;
 395
 396     c->m_queue_front = 0;
 397     c->m_queue_back = 0;
 398
 399     c->sru_version = xstrdup("1.1");
 400     return c;
 401 }
 402
 403
 404 /* set database names. Take local databases (if set); otherwise
 405    take databases given in ZURL (if set); otherwise use Default */
 406 static char **set_DatabaseNames(ZOOM_connection con, ZOOM_options options,
 407                                 int *num, ODR odr)
 408 {
 409     char **databaseNames;
 410     const char *cp = ZOOM_options_get(options, "databaseName");
 411
 412     if ((!cp || !*cp) && con->host_port)
 413     {
 414         if (strncmp(con->host_port, "unix:", 5) == 0)
 415             cp = strchr(con->host_port+5, ':');
 416         else
 417             cp = strchr(con->host_port, '/');
 418         if (cp)
 419             cp++;
 420     }
 421     if (!cp)
 422         cp = "Default";
 423     nmem_strsplit(odr_getmem(odr), "+", cp,  &databaseNames, num);
 424     return databaseNames;
 425 }
 426
 427 ZOOM_API(ZOOM_connection)
 428     ZOOM_connection_new(const char *host, int portnum)
 429 {
 430     ZOOM_connection c = ZOOM_connection_create(0);
 431
 432     ZOOM_connection_connect(c, host, portnum);
 433     return c;
 434 }
 435
 436 static zoom_sru_mode get_sru_mode_from_string(const char *s)
 437 {
 438     if (!s || !*s)
 439         return zoom_sru_soap;
 440     if (!yaz_matchstr(s, "soap"))
 441         return zoom_sru_soap;
 442     else if (!yaz_matchstr(s, "get"))
 443         return zoom_sru_get;
 444     else if (!yaz_matchstr(s, "post"))
 445         return zoom_sru_post;
 446     return zoom_sru_error;
 447 }
 448
 449 ZOOM_API(void)
 450     ZOOM_connection_connect(ZOOM_connection c,
 451                             const char *host, int portnum)
 452 {
 453     const char *val;
 454     ZOOM_task task;
 455
 456     initlog();
 457
 458     yaz_log(log_api, "%p ZOOM_connection_connect host=%s portnum=%d",
 459             c, host ? host : "null", portnum);
 460
 461     set_ZOOM_error(c, ZOOM_ERROR_NONE, 0);
 462     ZOOM_connection_remove_tasks(c);
 463
 464     if (c->cs)
 465     {
 466         yaz_log(log_details, "%p ZOOM_connection_connect reconnect ok", c);
 467         c->reconnect_ok = 1;
 468         return;
 469     }
 470     yaz_log(log_details, "%p ZOOM_connection_connect connect", c);
 471     xfree(c->proxy);
 472     c->proxy = 0;
 473     val = ZOOM_options_get(c->options, "proxy");
 474     if (val && *val)
 475     {
 476         yaz_log(log_details, "%p ZOOM_connection_connect proxy=%s", c, val);
 477         c->proxy = xstrdup(val);
 478     }
 479
 480     xfree(c->charset);
 481     c->charset = 0;
 482     val = ZOOM_options_get(c->options, "charset");
 483     if (val && *val)
 484     {
 485         yaz_log(log_details, "%p ZOOM_connection_connect charset=%s", c, val);
 486         c->charset = xstrdup(val);
 487     }
 488
 489     xfree(c->lang);
 490     val = ZOOM_options_get(c->options, "lang");
 491     if (val && *val)
 492     {
 493         yaz_log(log_details, "%p ZOOM_connection_connect lang=%s", c, val);
 494         c->lang = xstrdup(val);
 495     }
 496     else
 497         c->lang = 0;
 498
 499     if (host)
 500     {
 501         xfree(c->host_port);
 502         if (portnum)
 503         {
 504             char hostn[128];
 505             sprintf(hostn, "%.80s:%d", host, portnum);
 506             c->host_port = xstrdup(hostn);
 507         }
 508         else
 509             c->host_port = xstrdup(host);
 510     }
 511
 512     {
 513         /*
 514          * If the "<scheme>:" part of the host string is preceded by one
 515          * or more comma-separated <name>=<value> pairs, these are taken
 516          * to be options to be set on the connection object.  Among other
 517          * applications, this facility can be used to embed authentication
 518          * in a host string:
 519          *          user=admin,password=secret,tcp:localhost:9999
 520          */
 521         char *remainder = c->host_port;
 522         char *pcolon = strchr(remainder, ':');
 523         char *pcomma;
 524         char *pequals;
 525         while ((pcomma = strchr(remainder, ',')) != 0 &&
 526                (pcolon == 0 || pcomma < pcolon)) {
 527             *pcomma = '\0';
 528             if ((pequals = strchr(remainder, '=')) != 0) {
 529                 *pequals = '\0';
 530                 /*printf("# setting '%s'='%s'\n", remainder, pequals+1);*/
 531                 ZOOM_connection_option_set(c, remainder, pequals+1);
 532             }
 533             remainder = pcomma+1;
 534         }
 535
 536         if (remainder != c->host_port) {
 537             xfree(c->host_port);
 538             c->host_port = xstrdup(remainder);
 539             /*printf("# reset hp='%s'\n", remainder);*/
 540         }
 541     }
 542
 543     val = ZOOM_options_get(c->options, "sru");
 544     c->sru_mode = get_sru_mode_from_string(val);
 545
 546     xfree(c->sru_version);
 547     val = ZOOM_options_get(c->options, "sru_version");
 548     c->sru_version = xstrdup(val ? val : "1.1");
 549
 550     ZOOM_options_set(c->options, "host", c->host_port);
 551
 552     xfree(c->cookie_out);
 553     c->cookie_out = 0;
 554     val = ZOOM_options_get(c->options, "cookie");
 555     if (val && *val)
 556     {
 557         yaz_log(log_details, "%p ZOOM_connection_connect cookie=%s", c, val);
 558         c->cookie_out = xstrdup(val);
 559     }
 560
 561     xfree(c->client_IP);
 562     c->client_IP = 0;
 563     val = ZOOM_options_get(c->options, "clientIP");
 564     if (val && *val)
 565     {
 566         yaz_log(log_details, "%p ZOOM_connection_connect clientIP=%s",
 567                 c, val);
 568         c->client_IP = xstrdup(val);
 569     }
 570
 571     xfree(c->group);
 572     c->group = 0;
 573     val = ZOOM_options_get(c->options, "group");
 574     if (val && *val)
 575         c->group = xstrdup(val);
 576
 577     xfree(c->user);
 578     c->user = 0;
 579     val = ZOOM_options_get(c->options, "user");
 580     if (val && *val)
 581         c->user = xstrdup(val);
 582
 583     xfree(c->password);
 584     c->password = 0;
 585     val = ZOOM_options_get(c->options, "password");
 586     if (!val)
 587         val = ZOOM_options_get(c->options, "pass");
 588
 589     if (val && *val)
 590         c->password = xstrdup(val);
 591
 592     c->maximum_record_size =
 593         ZOOM_options_get_int(c->options, "maximumRecordSize", 1024*1024);
 594     c->preferred_message_size =
 595         ZOOM_options_get_int(c->options, "preferredMessageSize", 1024*1024);
 596
 597     c->async = ZOOM_options_get_bool(c->options, "async", 0);
 598     yaz_log(log_details, "%p ZOOM_connection_connect async=%d", c, c->async);
 599
 600     task = ZOOM_connection_add_task(c, ZOOM_TASK_CONNECT);
 601
 602     if (!c->async)
 603     {
 604         while (ZOOM_event(1, &c))
 605             ;
 606     }
 607 }
 608
 609 ZOOM_API(ZOOM_query)
 610     ZOOM_query_create(void)
 611 {
 612     ZOOM_query s = (ZOOM_query) xmalloc(sizeof(*s));
 613
 614     yaz_log(log_details, "%p ZOOM_query_create", s);
 615     s->refcount = 1;
 616     s->z_query = 0;
 617     s->sort_spec = 0;
 618     s->odr = odr_createmem(ODR_ENCODE);
 619     s->query_string = 0;
 620
 621     return s;
 622 }
 623
 624 ZOOM_API(void)
 625     ZOOM_query_destroy(ZOOM_query s)
 626 {
 627     if (!s)
 628         return;
 629
 630     (s->refcount)--;
 631     yaz_log(log_details, "%p ZOOM_query_destroy count=%d", s, s->refcount);
 632     if (s->refcount == 0)
 633     {
 634         odr_destroy(s->odr);
 635         xfree(s);
 636     }
 637 }
 638
 639 ZOOM_API(int)
 640     ZOOM_query_prefix(ZOOM_query s, const char *str)
 641 {
 642     s->query_string = odr_strdup(s->odr, str);
 643     s->z_query = (Z_Query *) odr_malloc(s->odr, sizeof(*s->z_query));
 644     s->z_query->which = Z_Query_type_1;
 645     s->z_query->u.type_1 =  p_query_rpn(s->odr, str);
 646     if (!s->z_query->u.type_1)
 647     {
 648         yaz_log(log_details, "%p ZOOM_query_prefix str=%s failed", s, str);
 649         s->z_query = 0;
 650         return -1;
 651     }
 652     yaz_log(log_details, "%p ZOOM_query_prefix str=%s", s, str);
 653     return 0;
 654 }
 655
 656 ZOOM_API(int)
 657     ZOOM_query_cql(ZOOM_query s, const char *str)
 658 {
 659     Z_External *ext;
 660
 661     s->query_string = odr_strdup(s->odr, str);
 662
 663     ext = (Z_External *) odr_malloc(s->odr, sizeof(*ext));
 664     ext->direct_reference = odr_oiddup(s->odr, yaz_oid_userinfo_cql);
 665     ext->indirect_reference = 0;
 666     ext->descriptor = 0;
 667     ext->which = Z_External_CQL;
 668     ext->u.cql = s->query_string;
 669
 670     s->z_query = (Z_Query *) odr_malloc(s->odr, sizeof(*s->z_query));
 671     s->z_query->which = Z_Query_type_104;
 672     s->z_query->u.type_104 =  ext;
 673
 674     yaz_log(log_details, "%p ZOOM_query_cql str=%s", s, str);
 675
 676     return 0;
 677 }
 678
 679 /*
 680  * Translate the CQL string client-side into RPN which is passed to
 681  * the server.  This is useful for server's that don't themselves
 682  * support CQL, for which ZOOM_query_cql() is useless.  `conn' is used
 683  * only as a place to stash diagnostics if compilation fails; if this
 684  * information is not needed, a null pointer may be used.
 685  */
 686 ZOOM_API(int)
 687     ZOOM_query_cql2rpn(ZOOM_query s, const char *str, ZOOM_connection conn)
 688 {
 689     char *rpn;
 690     int ret;
 691     ZOOM_connection freeme = 0;
 692
 693     yaz_log(log_details, "%p ZOOM_query_cql2rpn str=%s conn=%p", s, str, conn);
 694     if (conn == 0)
 695         conn = freeme = ZOOM_connection_create(0);
 696
 697     rpn = cql2pqf(conn, str);
 698     if (freeme != 0)
 699         ZOOM_connection_destroy(freeme);
 700     if (rpn == 0)
 701         return -1;
 702
 703     ret = ZOOM_query_prefix(s, rpn);
 704     xfree(rpn);
 705     return ret;
 706 }
 707
 708 /*
 709  * Analogous in every way to ZOOM_query_cql2rpn(), except that there
 710  * is no analogous ZOOM_query_ccl() that just sends uninterpreted CCL
 711  * to the server, as the YAZ GFS doesn't know how to handle this.
 712  */
 713 ZOOM_API(int)
 714     ZOOM_query_ccl2rpn(ZOOM_query s, const char *str, const char *config,
 715                        int *ccl_error, const char **error_string,
 716                        int *error_pos)
 717 {
 718     int ret;
 719     struct ccl_rpn_node *rpn;
 720     CCL_bibset bibset = ccl_qual_mk();
 721
 722     if (config)
 723         ccl_qual_buf(bibset, config);
 724
 725     rpn = ccl_find_str(bibset, str, ccl_error, error_pos);
 726     if (!rpn)
 727     {
 728         *error_string = ccl_err_msg(*ccl_error);
 729         ret = -1;
 730     }
 731     else
 732     {
 733         WRBUF wr = wrbuf_alloc();
 734         ccl_pquery(wr, rpn);
 735         ccl_rpn_delete(rpn);
 736         ret = ZOOM_query_prefix(s, wrbuf_cstr(wr));
 737         wrbuf_destroy(wr);
 738     }
 739     ccl_qual_rm(&bibset);
 740     return ret;
 741 }
 742
 743 ZOOM_API(int)
 744     ZOOM_query_sortby(ZOOM_query s, const char *criteria)
 745 {
 746     s->sort_spec = yaz_sort_spec(s->odr, criteria);
 747     if (!s->sort_spec)
 748     {
 749         yaz_log(log_details, "%p ZOOM_query_sortby criteria=%s failed",
 750                 s, criteria);
 751         return -1;
 752     }
 753     yaz_log(log_details, "%p ZOOM_query_sortby criteria=%s", s, criteria);
 754     return 0;
 755 }
 756
 757 static zoom_ret do_write(ZOOM_connection c);
 758
 759 ZOOM_API(void)
 760     ZOOM_connection_destroy(ZOOM_connection c)
 761 {
 762     ZOOM_resultset r;
 763     if (!c)
 764         return;
 765     yaz_log(log_api, "%p ZOOM_connection_destroy", c);
 766     if (c->cs)
 767         cs_close(c->cs);
 768     for (r = c->resultsets; r; r = r->next)
 769         r->connection = 0;
 770
 771     xfree(c->buf_in);
 772     xfree(c->addinfo);
 773     xfree(c->diagset);
 774     odr_destroy(c->odr_in);
 775     odr_destroy(c->odr_out);
 776     ZOOM_options_destroy(c->options);
 777     ZOOM_connection_remove_tasks(c);
 778     ZOOM_connection_remove_events(c);
 779     xfree(c->host_port);
 780     xfree(c->path);
 781     xfree(c->proxy);
 782     xfree(c->charset);
 783     xfree(c->lang);
 784     xfree(c->cookie_out);
 785     xfree(c->cookie_in);
 786     xfree(c->client_IP);
 787     xfree(c->user);
 788     xfree(c->group);
 789     xfree(c->password);
 790     xfree(c->sru_version);
 791     xfree(c);
 792 }
 793
 794 void ZOOM_resultset_addref(ZOOM_resultset r)
 795 {
 796     if (r)
 797     {
 798         (r->refcount)++;
 799         yaz_log(log_details, "%p ZOOM_resultset_addref count=%d",
 800                 r, r->refcount);
 801     }
 802 }
 803
 804 ZOOM_resultset ZOOM_resultset_create(void)
 805 {
 806     int i;
 807     ZOOM_resultset r = (ZOOM_resultset) xmalloc(sizeof(*r));
 808
 809     initlog();
 810
 811     yaz_log(log_details, "%p ZOOM_resultset_create", r);
 812     r->refcount = 1;
 813     r->size = 0;
 814     r->odr = odr_createmem(ODR_ENCODE);
 815     r->piggyback = 1;
 816     r->setname = 0;
 817     r->schema = 0;
 818     r->step = 0;
 819     for (i = 0; i<RECORD_HASH_SIZE; i++)
 820         r->record_hash[i] = 0;
 821     r->r_sort_spec = 0;
 822     r->query = 0;
 823     r->connection = 0;
 824     r->next = 0;
 825     r->databaseNames = 0;
 826     r->num_databaseNames = 0;
 827     return r;
 828 }
 829
 830 ZOOM_API(ZOOM_resultset)
 831     ZOOM_connection_search_pqf(ZOOM_connection c, const char *q)
 832 {
 833     ZOOM_resultset r;
 834     ZOOM_query s = ZOOM_query_create();
 835
 836     ZOOM_query_prefix(s, q);
 837
 838     r = ZOOM_connection_search(c, s);
 839     ZOOM_query_destroy(s);
 840     return r;
 841 }
 842
 843 ZOOM_API(ZOOM_resultset)
 844     ZOOM_connection_search(ZOOM_connection c, ZOOM_query q)
 845 {
 846     ZOOM_resultset r = ZOOM_resultset_create();
 847     ZOOM_task task;
 848     const char *cp;
 849     int start, count;
 850     const char *syntax, *elementSetName;
 851
 852     yaz_log(log_api, "%p ZOOM_connection_search set %p query %p", c, r, q);
 853     r->r_sort_spec = q->sort_spec;
 854     r->query = q;
 855
 856     r->options = ZOOM_options_create_with_parent(c->options);
 857
 858     start = ZOOM_options_get_int(r->options, "start", 0);
 859     count = ZOOM_options_get_int(r->options, "count", 0);
 860     {
 861         /* If "presentChunk" is defined use that; otherwise "step" */
 862         const char *cp = ZOOM_options_get(r->options, "presentChunk");
 863         r->step = ZOOM_options_get_int(r->options,
 864                                        (cp != 0 ? "presentChunk": "step"), 0);
 865     }
 866     r->piggyback = ZOOM_options_get_bool(r->options, "piggyback", 1);
 867     cp = ZOOM_options_get(r->options, "setname");
 868     if (cp)
 869         r->setname = xstrdup(cp);
 870     cp = ZOOM_options_get(r->options, "schema");
 871     if (cp)
 872         r->schema = xstrdup(cp);
 873
 874     r->databaseNames = set_DatabaseNames(c, c->options, &r->num_databaseNames,
 875                                          r->odr);
 876
 877     r->connection = c;
 878
 879     r->next = c->resultsets;
 880     c->resultsets = r;
 881
 882
 883
 884     if (c->host_port && c->proto == PROTO_HTTP)
 885     {
 886         if (!c->cs)
 887         {
 888             yaz_log(log_details, "ZOOM_connection_search: no comstack");
 889             ZOOM_connection_add_task(c, ZOOM_TASK_CONNECT);
 890         }
 891         else
 892         {
 893             yaz_log(log_details, "ZOOM_connection_search: reconnect");
 894             c->reconnect_ok = 1;
 895         }
 896     }
 897
 898     task = ZOOM_connection_add_task(c, ZOOM_TASK_SEARCH);
 899     task->u.search.resultset = r;
 900     task->u.search.start = start;
 901     task->u.search.count = count;
 902
 903     syntax = ZOOM_options_get(r->options, "preferredRecordSyntax");
 904     task->u.search.syntax = syntax ? xstrdup(syntax) : 0;
 905     elementSetName = ZOOM_options_get(r->options, "elementSetName");
 906     task->u.search.elementSetName = elementSetName
 907         ? xstrdup(elementSetName) : 0;
 908
 909     ZOOM_resultset_addref(r);
 910
 911     (q->refcount)++;
 912
 913     if (!c->async)
 914     {
 915         while (ZOOM_event(1, &c))
 916             ;
 917     }
 918     return r;
 919 }
 920
 921 ZOOM_API(void)
 922     ZOOM_resultset_sort(ZOOM_resultset r,
 923                          const char *sort_type, const char *sort_spec)
 924 {
 925     (void) ZOOM_resultset_sort1(r, sort_type, sort_spec);
 926 }
 927
 928 ZOOM_API(int)
 929     ZOOM_resultset_sort1(ZOOM_resultset r,
 930                          const char *sort_type, const char *sort_spec)
 931 {
 932     ZOOM_connection c = r->connection;
 933     ZOOM_task task;
 934     ZOOM_query newq;
 935
 936     newq = ZOOM_query_create();
 937     if (ZOOM_query_sortby(newq, sort_spec) < 0)
 938         return -1;
 939
 940     yaz_log(log_api, "%p ZOOM_resultset_sort r=%p sort_type=%s sort_spec=%s",
 941             r, r, sort_type, sort_spec);
 942     if (!c)
 943         return 0;
 944
 945     if (c->host_port && c->proto == PROTO_HTTP)
 946     {
 947         if (!c->cs)
 948         {
 949             yaz_log(log_details, "%p ZOOM_resultset_sort: no comstack", r);
 950             ZOOM_connection_add_task(c, ZOOM_TASK_CONNECT);
 951         }
 952         else
 953         {
 954             yaz_log(log_details, "%p ZOOM_resultset_sort: prepare reconnect",
 955                     r);
 956             c->reconnect_ok = 1;
 957         }
 958     }
 959
 960     ZOOM_resultset_cache_reset(r);
 961     task = ZOOM_connection_add_task(c, ZOOM_TASK_SORT);
 962     task->u.sort.resultset = r;
 963     task->u.sort.q = newq;
 964
 965     ZOOM_resultset_addref(r);
 966
 967     if (!c->async)
 968     {
 969         while (ZOOM_event(1, &c))
 970             ;
 971     }
 972
 973     return 0;
 974 }
 975
 976 ZOOM_API(void)
 977     ZOOM_resultset_cache_reset(ZOOM_resultset r)
 978 {
 979     int i;
 980     for (i = 0; i<RECORD_HASH_SIZE; i++)
 981     {
 982         ZOOM_record_cache rc;
 983         for (rc = r->record_hash[i]; rc; rc = rc->next)
 984         {
 985             if (rc->rec.wrbuf_marc)
 986                 wrbuf_destroy(rc->rec.wrbuf_marc);
 987             if (rc->rec.wrbuf_iconv)
 988                 wrbuf_destroy(rc->rec.wrbuf_iconv);
 989             if (rc->rec.wrbuf_opac)
 990                 wrbuf_destroy(rc->rec.wrbuf_opac);
 991         }
 992         r->record_hash[i] = 0;
 993     }
 994 }
 995
 996 ZOOM_API(void)
 997     ZOOM_resultset_destroy(ZOOM_resultset r)
 998 {
 999     resultset_destroy(r);
1000 }
1001
1002 static void resultset_destroy(ZOOM_resultset r)
1003 {
1004     if (!r)
1005         return;
1006     (r->refcount)--;
1007     yaz_log(log_details, "%p ZOOM_resultset_destroy r=%p count=%d",
1008             r, r, r->refcount);
1009     if (r->refcount == 0)
1010     {
1011         ZOOM_resultset_cache_reset(r);
1012
1013         if (r->connection)
1014         {
1015             /* remove ourselves from the resultsets in connection */
1016             ZOOM_resultset *rp = &r->connection->resultsets;
1017             while (1)
1018             {
1019                 assert(*rp);   /* we must be in this list!! */
1020                 if (*rp == r)
1021                 {   /* OK, we're here - take us out of it */
1022                     *rp = (*rp)->next;
1023                     break;
1024                 }
1025                 rp = &(*rp)->next;
1026             }
1027         }
1028         ZOOM_query_destroy(r->query);
1029         ZOOM_options_destroy(r->options);
1030         odr_destroy(r->odr);
1031         xfree(r->setname);
1032         xfree(r->schema);
1033         xfree(r);
1034     }
1035 }
1036
1037 ZOOM_API(size_t)
1038     ZOOM_resultset_size(ZOOM_resultset r)
1039 {
1040     yaz_log(log_details, "ZOOM_resultset_size r=%p count=%d",
1041             r, r->size);
1042     return r->size;
1043 }
1044
1045 static void do_close(ZOOM_connection c)
1046 {
1047     if (c->cs)
1048         cs_close(c->cs);
1049     c->cs = 0;
1050     ZOOM_connection_set_mask(c, 0);
1051     c->state = STATE_IDLE;
1052 }
1053
1054 static int ZOOM_test_reconnect(ZOOM_connection c)
1055 {
1056     ZOOM_Event event;
1057
1058     if (!c->reconnect_ok)
1059         return 0;
1060     do_close(c);
1061     c->reconnect_ok = 0;
1062     c->tasks->running = 0;
1063     ZOOM_connection_insert_task(c, ZOOM_TASK_CONNECT);
1064
1065     event = ZOOM_Event_create(ZOOM_EVENT_CONNECT);
1066     ZOOM_connection_put_event(c, event);
1067
1068     return 1;
1069 }
1070
1071 static void ZOOM_resultset_retrieve(ZOOM_resultset r,
1072                                     int force_sync, int start, int count)
1073 {
1074     ZOOM_task task;
1075     ZOOM_connection c;
1076     const char *cp;
1077     const char *syntax, *elementSetName;
1078
1079     if (!r)
1080         return;
1081     yaz_log(log_details, "%p ZOOM_resultset_retrieve force_sync=%d start=%d"
1082             " count=%d", r, force_sync, start, count);
1083     c = r->connection;
1084     if (!c)
1085         return;
1086
1087     if (c->host_port && c->proto == PROTO_HTTP)
1088     {
1089         if (!c->cs)
1090         {
1091             yaz_log(log_details, "%p ZOOM_resultset_retrieve: no comstack", r);
1092             ZOOM_connection_add_task(c, ZOOM_TASK_CONNECT);
1093         }
1094         else
1095         {
1096             yaz_log(log_details, "%p ZOOM_resultset_retrieve: prepare "
1097                     "reconnect", r);
1098             c->reconnect_ok = 1;
1099         }
1100     }
1101     task = ZOOM_connection_add_task(c, ZOOM_TASK_RETRIEVE);
1102     task->u.retrieve.resultset = r;
1103     task->u.retrieve.start = start;
1104     task->u.retrieve.count = count;
1105
1106     syntax = ZOOM_options_get(r->options, "preferredRecordSyntax");
1107     task->u.retrieve.syntax = syntax ? xstrdup(syntax) : 0;
1108     elementSetName = ZOOM_options_get(r->options, "elementSetName");
1109     task->u.retrieve.elementSetName = elementSetName
1110         ? xstrdup(elementSetName) : 0;
1111
1112     cp = ZOOM_options_get(r->options, "schema");
1113     if (cp)
1114     {
1115         if (!r->schema || strcmp(r->schema, cp))
1116         {
1117             xfree(r->schema);
1118             r->schema = xstrdup(cp);
1119         }
1120     }
1121
1122     ZOOM_resultset_addref(r);
1123
1124     if (!r->connection->async || force_sync)
1125         while (r->connection && ZOOM_event(1, &r->connection))
1126             ;
1127 }
1128
1129 ZOOM_API(void)
1130     ZOOM_resultset_records(ZOOM_resultset r, ZOOM_record *recs,
1131                            size_t start, size_t count)
1132 {
1133     int force_present = 0;
1134
1135     if (!r)
1136         return ;
1137     yaz_log(log_api, "%p ZOOM_resultset_records r=%p start=%ld count=%ld",
1138             r, r, (long) start, (long) count);
1139     if (count && recs)
1140         force_present = 1;
1141     ZOOM_resultset_retrieve(r, force_present, start, count);
1142     if (force_present)
1143     {
1144         size_t i;
1145         for (i = 0; i< count; i++)
1146             recs[i] = ZOOM_resultset_record_immediate(r, i+start);
1147     }
1148 }
1149
1150 static void get_cert(ZOOM_connection c)
1151 {
1152     char *cert_buf;
1153     int cert_len;
1154
1155     if (cs_get_peer_certificate_x509(c->cs, &cert_buf, &cert_len))
1156     {
1157         ZOOM_connection_option_setl(c, "sslPeerCert",
1158                                     cert_buf, cert_len);
1159         xfree(cert_buf);
1160     }
1161 }
1162
1163 static zoom_ret do_connect(ZOOM_connection c)
1164 {
1165     void *add;
1166     const char *effective_host;
1167
1168     if (c->proxy)
1169         effective_host = c->proxy;
1170     else
1171         effective_host = c->host_port;
1172
1173     yaz_log(log_details, "%p do_connect effective_host=%s", c, effective_host);
1174
1175     if (c->cs)
1176         cs_close(c->cs);
1177     c->cs = cs_create_host(effective_host, 0, &add);
1178
1179     if (c->cs && c->cs->protocol == PROTO_HTTP)
1180     {
1181 #if YAZ_HAVE_XML2
1182         const char *path = 0;
1183
1184         c->proto = PROTO_HTTP;
1185         cs_get_host_args(c->host_port, &path);
1186         xfree(c->path);
1187         c->path = (char*) xmalloc(strlen(path)+2);
1188         c->path[0] = '/';
1189         strcpy(c->path+1, path);
1190 #else
1191         set_ZOOM_error(c, ZOOM_ERROR_UNSUPPORTED_PROTOCOL, "SRW");
1192         do_close(c);
1193         return zoom_complete;
1194 #endif
1195     }
1196     if (c->cs)
1197     {
1198         int ret = cs_connect(c->cs, add);
1199         if (ret == 0)
1200         {
1201             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_CONNECT);
1202             ZOOM_connection_put_event(c, event);
1203             get_cert(c);
1204             if (c->proto == PROTO_Z3950)
1205                 ZOOM_connection_send_init(c);
1206             else
1207             {
1208                 /* no init request for SRW .. */
1209                 assert(c->tasks->which == ZOOM_TASK_CONNECT);
1210                 ZOOM_connection_remove_task(c);
1211                 ZOOM_connection_set_mask(c, 0);
1212                 ZOOM_connection_exec_task(c);
1213             }
1214             c->state = STATE_ESTABLISHED;
1215             return zoom_pending;
1216         }
1217         else if (ret > 0)
1218         {
1219             int mask = ZOOM_SELECT_EXCEPT;
1220             if (c->cs->io_pending & CS_WANT_WRITE)
1221                 mask += ZOOM_SELECT_WRITE;
1222             if (c->cs->io_pending & CS_WANT_READ)
1223                 mask += ZOOM_SELECT_READ;
1224             ZOOM_connection_set_mask(c, mask);
1225             c->state = STATE_CONNECTING;
1226             return zoom_pending;
1227         }
1228     }
1229     c->state = STATE_IDLE;
1230     set_ZOOM_error(c, ZOOM_ERROR_CONNECT, c->host_port);
1231     return zoom_complete;
1232 }
1233
1234 static void otherInfo_attach(ZOOM_connection c, Z_APDU *a, ODR out)
1235 {
1236     int i;
1237     for (i = 0; i<200; i++)
1238     {
1239         size_t len;
1240         Odr_oid *oid;
1241         Z_OtherInformation **oi;
1242         char buf[80];
1243         const char *val;
1244         const char *cp;
1245
1246         sprintf(buf, "otherInfo%d", i);
1247         val = ZOOM_options_get(c->options, buf);
1248         if (!val)
1249             break;
1250         cp = strchr(val, ':');
1251         if (!cp)
1252             continue;
1253         len = cp - val;
1254         if (len >= sizeof(buf))
1255             len = sizeof(buf)-1;
1256         memcpy(buf, val, len);
1257         buf[len] = '\0';
1258
1259         oid = yaz_string_to_oid_odr(yaz_oid_std(), CLASS_USERINFO,
1260                                     buf, out);
1261         if (!oid)
1262             continue;
1263
1264         yaz_oi_APDU(a, &oi);
1265         yaz_oi_set_string_oid(oi, out, oid, 1, cp+1);
1266     }
1267 }
1268
1269 static int encode_APDU(ZOOM_connection c, Z_APDU *a, ODR out)
1270 {
1271     assert(a);
1272     if (c->cookie_out)
1273     {
1274         Z_OtherInformation **oi;
1275         yaz_oi_APDU(a, &oi);
1276         yaz_oi_set_string_oid(oi, out, yaz_oid_userinfo_cookie,
1277                               1, c->cookie_out);
1278     }
1279     if (c->client_IP)
1280     {
1281         Z_OtherInformation **oi;
1282         yaz_oi_APDU(a, &oi);
1283         yaz_oi_set_string_oid(oi, out, yaz_oid_userinfo_client_ip,
1284                               1, c->client_IP);
1285     }
1286     otherInfo_attach(c, a, out);
1287     if (!z_APDU(out, &a, 0, 0))
1288     {
1289         FILE *outf = fopen("/tmp/apdu.txt", "a");
1290         if (a && outf)
1291         {
1292             ODR odr_pr = odr_createmem(ODR_PRINT);
1293             fprintf(outf, "a=%p\n", a);
1294             odr_setprint(odr_pr, outf);
1295             z_APDU(odr_pr, &a, 0, 0);
1296             odr_destroy(odr_pr);
1297         }
1298         yaz_log(log_api, "%p encoding_APDU: encoding failed", c);
1299         set_ZOOM_error(c, ZOOM_ERROR_ENCODE, 0);
1300         odr_reset(out);
1301         return -1;
1302     }
1303     yaz_log(log_details, "%p encoding_APDU encoding OK", c);
1304     return 0;
1305 }
1306
1307 static zoom_ret send_APDU(ZOOM_connection c, Z_APDU *a)
1308 {
1309     ZOOM_Event event;
1310     assert(a);
1311     if (encode_APDU(c, a, c->odr_out))
1312         return zoom_complete;
1313     yaz_log(log_details, "%p send APDU type=%d", c, a->which);
1314     c->buf_out = odr_getbuf(c->odr_out, &c->len_out, 0);
1315     event = ZOOM_Event_create(ZOOM_EVENT_SEND_APDU);
1316     ZOOM_connection_put_event(c, event);
1317     odr_reset(c->odr_out);
1318     return do_write(c);
1319 }
1320
1321 /* returns 1 if PDU was sent OK (still pending )
1322    0 if PDU was not sent OK (nothing to wait for)
1323 */
1324
1325 static zoom_ret ZOOM_connection_send_init(ZOOM_connection c)
1326 {
1327     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_initRequest);
1328     Z_InitRequest *ireq = apdu->u.initRequest;
1329     Z_IdAuthentication *auth = (Z_IdAuthentication *)
1330         odr_malloc(c->odr_out, sizeof(*auth));
1331     char *version;
1332
1333     ODR_MASK_SET(ireq->options, Z_Options_search);
1334     ODR_MASK_SET(ireq->options, Z_Options_present);
1335     ODR_MASK_SET(ireq->options, Z_Options_scan);
1336     ODR_MASK_SET(ireq->options, Z_Options_sort);
1337     ODR_MASK_SET(ireq->options, Z_Options_extendedServices);
1338     ODR_MASK_SET(ireq->options, Z_Options_namedResultSets);
1339
1340     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_1);
1341     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_2);
1342     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_3);
1343
1344     /* Index Data's Z39.50 Implementor Id is 81 */
1345     ireq->implementationId =
1346         odr_prepend(c->odr_out,
1347                     ZOOM_options_get(c->options, "implementationId"),
1348                     odr_prepend(c->odr_out, "81", ireq->implementationId));
1349
1350     ireq->implementationName =
1351         odr_prepend(c->odr_out,
1352                     ZOOM_options_get(c->options, "implementationName"),
1353                     odr_prepend(c->odr_out, "ZOOM-C",
1354                                 ireq->implementationName));
1355
1356     version = odr_strdup(c->odr_out, "$Revision: 1.145 $");
1357     if (strlen(version) > 10)   /* check for unexpanded CVS strings */
1358         version[strlen(version)-2] = '\0';
1359     ireq->implementationVersion =
1360         odr_prepend(c->odr_out,
1361                     ZOOM_options_get(c->options, "implementationVersion"),
1362                     odr_prepend(c->odr_out, &version[11],
1363                                 ireq->implementationVersion));
1364
1365     *ireq->maximumRecordSize = c->maximum_record_size;
1366     *ireq->preferredMessageSize = c->preferred_message_size;
1367
1368     if (c->group || c->password)
1369     {
1370         Z_IdPass *pass = (Z_IdPass *) odr_malloc(c->odr_out, sizeof(*pass));
1371         pass->groupId = 0;
1372         if (c->group)
1373             pass->groupId = odr_strdup(c->odr_out, c->group);
1374         pass->userId = 0;
1375         if (c->user)
1376             pass->userId = odr_strdup(c->odr_out, c->user);
1377         pass->password = 0;
1378         if (c->password)
1379             pass->password = odr_strdup(c->odr_out, c->password);
1380         auth->which = Z_IdAuthentication_idPass;
1381         auth->u.idPass = pass;
1382         ireq->idAuthentication = auth;
1383     }
1384     else if (c->user)
1385     {
1386         auth->which = Z_IdAuthentication_open;
1387         auth->u.open = odr_strdup(c->odr_out, c->user);
1388         ireq->idAuthentication = auth;
1389     }
1390     if (c->proxy)
1391     {
1392         yaz_oi_set_string_oid(&ireq->otherInfo, c->odr_out,
1393                               yaz_oid_userinfo_proxy, 1, c->host_port);
1394     }
1395     if (c->charset || c->lang)
1396     {
1397         Z_OtherInformation **oi;
1398         Z_OtherInformationUnit *oi_unit;
1399
1400         yaz_oi_APDU(apdu, &oi);
1401
1402         if ((oi_unit = yaz_oi_update(oi, c->odr_out, NULL, 0, 0)))
1403         {
1404             ODR_MASK_SET(ireq->options, Z_Options_negotiationModel);
1405             oi_unit->which = Z_OtherInfo_externallyDefinedInfo;
1406             oi_unit->information.externallyDefinedInfo =
1407                 yaz_set_proposal_charneg_list(c->odr_out, " ",
1408                                               c->charset, c->lang, 1);
1409         }
1410     }
1411     assert(apdu);
1412     return send_APDU(c, apdu);
1413 }
1414
1415 #if YAZ_HAVE_XML2
1416 static zoom_ret send_srw(ZOOM_connection c, Z_SRW_PDU *sr)
1417 {
1418     Z_GDU *gdu;
1419     ZOOM_Event event;
1420
1421     gdu = z_get_HTTP_Request_host_path(c->odr_out, c->host_port, c->path);
1422
1423     if (c->sru_mode == zoom_sru_get)
1424     {
1425         yaz_sru_get_encode(gdu->u.HTTP_Request, sr, c->odr_out, c->charset);
1426     }
1427     else if (c->sru_mode == zoom_sru_post)
1428     {
1429         yaz_sru_post_encode(gdu->u.HTTP_Request, sr, c->odr_out, c->charset);
1430     }
1431     else if (c->sru_mode == zoom_sru_soap)
1432     {
1433         yaz_sru_soap_encode(gdu->u.HTTP_Request, sr, c->odr_out, c->charset);
1434     }
1435     if (!z_GDU(c->odr_out, &gdu, 0, 0))
1436         return zoom_complete;
1437     c->buf_out = odr_getbuf(c->odr_out, &c->len_out, 0);
1438
1439     event = ZOOM_Event_create(ZOOM_EVENT_SEND_APDU);
1440     ZOOM_connection_put_event(c, event);
1441     odr_reset(c->odr_out);
1442     return do_write(c);
1443 }
1444 #endif
1445
1446
1447 static Z_SRW_PDU *ZOOM_srw_get_pdu(ZOOM_connection c, int type) {
1448     Z_SRW_PDU *sr = yaz_srw_get_pdu(c->odr_out, type, c->sru_version);
1449     sr->username = c->user;
1450     sr->password = c->password;
1451     return sr;
1452 }
1453
1454
1455 #if YAZ_HAVE_XML2
1456 static zoom_ret ZOOM_connection_srw_send_search(ZOOM_connection c)
1457 {
1458     int i;
1459     int *start, *count;
1460     ZOOM_resultset resultset = 0;
1461     Z_SRW_PDU *sr = 0;
1462     const char *option_val = 0;
1463
1464     if (c->error)                  /* don't continue on error */
1465         return zoom_complete;
1466     assert(c->tasks);
1467     switch(c->tasks->which)
1468     {
1469     case ZOOM_TASK_SEARCH:
1470         resultset = c->tasks->u.search.resultset;
1471         resultset->setname = xstrdup("default");
1472         ZOOM_options_set(resultset->options, "setname", resultset->setname);
1473         start = &c->tasks->u.search.start;
1474         count = &c->tasks->u.search.count;
1475         break;
1476     case ZOOM_TASK_RETRIEVE:
1477         resultset = c->tasks->u.retrieve.resultset;
1478
1479         start = &c->tasks->u.retrieve.start;
1480         count = &c->tasks->u.retrieve.count;
1481
1482         if (*start >= resultset->size)
1483             return zoom_complete;
1484         if (*start + *count > resultset->size)
1485             *count = resultset->size - *start;
1486
1487         for (i = 0; i < *count; i++)
1488         {
1489             ZOOM_record rec =
1490                 record_cache_lookup(resultset, i + *start,
1491                                     c->tasks->u.retrieve.syntax,
1492                                     c->tasks->u.retrieve.elementSetName);
1493             if (!rec)
1494                 break;
1495             else
1496             {
1497                 ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_RECV_RECORD);
1498                 ZOOM_connection_put_event(c, event);
1499             }
1500         }
1501         *start += i;
1502         *count -= i;
1503
1504         if (*count == 0)
1505             return zoom_complete;
1506         break;
1507     default:
1508         return zoom_complete;
1509     }
1510     assert(resultset->query);
1511
1512     sr = ZOOM_srw_get_pdu(c, Z_SRW_searchRetrieve_request);
1513     if (resultset->query->z_query->which == Z_Query_type_104
1514         && resultset->query->z_query->u.type_104->which == Z_External_CQL)
1515     {
1516         sr->u.request->query_type = Z_SRW_query_type_cql;
1517         sr->u.request->query.cql =resultset->query->z_query->u.type_104->u.cql;
1518     }
1519     else if (resultset->query->z_query->which == Z_Query_type_1 &&
1520              resultset->query->z_query->u.type_1)
1521     {
1522         sr->u.request->query_type = Z_SRW_query_type_pqf;
1523         sr->u.request->query.pqf = resultset->query->query_string;
1524     }
1525     else
1526     {
1527         set_ZOOM_error(c, ZOOM_ERROR_UNSUPPORTED_QUERY, 0);
1528         return zoom_complete;
1529     }
1530     sr->u.request->startRecord = odr_intdup(c->odr_out, *start + 1);
1531     sr->u.request->maximumRecords = odr_intdup(
1532         c->odr_out, resultset->step>0 ? resultset->step : *count);
1533     sr->u.request->recordSchema = resultset->schema;
1534
1535     option_val = ZOOM_resultset_option_get(resultset, "recordPacking");
1536     if (option_val)
1537         sr->u.request->recordPacking = odr_strdup(c->odr_out, option_val);
1538
1539     option_val = ZOOM_resultset_option_get(resultset, "extraArgs");
1540     if (option_val)
1541         sr->extra_args = odr_strdup(c->odr_out, option_val);
1542     return send_srw(c, sr);
1543 }
1544 #else
1545 static zoom_ret ZOOM_connection_srw_send_search(ZOOM_connection c)
1546 {
1547     return zoom_complete;
1548 }
1549 #endif
1550
1551 static zoom_ret ZOOM_connection_send_search(ZOOM_connection c)
1552 {
1553     ZOOM_resultset r;
1554     int lslb, ssub, mspn;
1555     const char *syntax;
1556     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_searchRequest);
1557     Z_SearchRequest *search_req = apdu->u.searchRequest;
1558     const char *elementSetName;
1559     const char *smallSetElementSetName;
1560     const char *mediumSetElementSetName;
1561
1562     assert(c->tasks);
1563     assert(c->tasks->which == ZOOM_TASK_SEARCH);
1564
1565     r = c->tasks->u.search.resultset;
1566
1567     yaz_log(log_details, "%p ZOOM_connection_send_search set=%p", c, r);
1568
1569     elementSetName =
1570         ZOOM_options_get(r->options, "elementSetName");
1571     smallSetElementSetName  =
1572         ZOOM_options_get(r->options, "smallSetElementSetName");
1573     mediumSetElementSetName =
1574         ZOOM_options_get(r->options, "mediumSetElementSetName");
1575
1576     if (!smallSetElementSetName)
1577         smallSetElementSetName = elementSetName;
1578
1579     if (!mediumSetElementSetName)
1580         mediumSetElementSetName = elementSetName;
1581
1582     assert(r);
1583     assert(r->query);
1584
1585     /* prepare query for the search request */
1586     search_req->query = r->query->z_query;
1587     if (!search_req->query)
1588     {
1589         set_ZOOM_error(c, ZOOM_ERROR_INVALID_QUERY, 0);
1590         return zoom_complete;
1591     }
1592     if (r->query->z_query->which == Z_Query_type_1 ||
1593         r->query->z_query->which == Z_Query_type_101)
1594     {
1595         const char *cp = ZOOM_options_get(r->options, "rpnCharset");
1596         if (cp)
1597         {
1598             yaz_iconv_t cd = yaz_iconv_open(cp, "UTF-8");
1599             if (cd)
1600             {
1601                 search_req->query = yaz_copy_Z_Query(search_req->query,
1602                                                      c->odr_out);
1603
1604                 yaz_query_charset_convert_rpnquery(search_req->query->u.type_1,
1605                                                    c->odr_out, cd);
1606                 yaz_iconv_close(cd);
1607             }
1608         }
1609     }
1610     search_req->databaseNames = r->databaseNames;
1611     search_req->num_databaseNames = r->num_databaseNames;
1612
1613     /* get syntax (no need to provide unless piggyback is in effect) */
1614     syntax = c->tasks->u.search.syntax;
1615
1616     lslb = ZOOM_options_get_int(r->options, "largeSetLowerBound", -1);
1617     ssub = ZOOM_options_get_int(r->options, "smallSetUpperBound", -1);
1618     mspn = ZOOM_options_get_int(r->options, "mediumSetPresentNumber", -1);
1619     if (lslb != -1 && ssub != -1 && mspn != -1)
1620     {
1621         /* So're a Z39.50 expert? Let's hope you don't do sort */
1622         *search_req->largeSetLowerBound = lslb;
1623         *search_req->smallSetUpperBound = ssub;
1624         *search_req->mediumSetPresentNumber = mspn;
1625     }
1626     else if (c->tasks->u.search.start == 0 && c->tasks->u.search.count > 0
1627              && r->piggyback && !r->r_sort_spec && !r->schema)
1628     {
1629         /* Regular piggyback - do it unless we're going to do sort */
1630         *search_req->largeSetLowerBound = 2000000000;
1631         *search_req->smallSetUpperBound = 1;
1632         *search_req->mediumSetPresentNumber =
1633             r->step>0 ? r->step : c->tasks->u.search.count;
1634     }
1635     else
1636     {
1637         /* non-piggyback. Need not provide elementsets or syntaxes .. */
1638         smallSetElementSetName = 0;
1639         mediumSetElementSetName = 0;
1640         syntax = 0;
1641     }
1642     if (smallSetElementSetName && *smallSetElementSetName)
1643     {
1644         Z_ElementSetNames *esn = (Z_ElementSetNames *)
1645             odr_malloc(c->odr_out, sizeof(*esn));
1646
1647         esn->which = Z_ElementSetNames_generic;
1648         esn->u.generic = odr_strdup(c->odr_out, smallSetElementSetName);
1649         search_req->smallSetElementSetNames = esn;
1650     }
1651     if (mediumSetElementSetName && *mediumSetElementSetName)
1652     {
1653         Z_ElementSetNames *esn =(Z_ElementSetNames *)
1654             odr_malloc(c->odr_out, sizeof(*esn));
1655
1656         esn->which = Z_ElementSetNames_generic;
1657         esn->u.generic = odr_strdup(c->odr_out, mediumSetElementSetName);
1658         search_req->mediumSetElementSetNames = esn;
1659     }
1660     if (syntax)
1661         search_req->preferredRecordSyntax =
1662             zoom_yaz_str_to_z3950oid(c, CLASS_RECSYN, syntax);
1663
1664     if (!r->setname)
1665     {
1666         if (c->support_named_resultsets)
1667         {
1668             char setname[14];
1669             int ord;
1670             /* find the lowest unused ordinal so that we re-use
1671                result sets on the server. */
1672             for (ord = 1; ; ord++)
1673             {
1674                 ZOOM_resultset rp;
1675                 sprintf(setname, "%d", ord);
1676                 for (rp = c->resultsets; rp; rp = rp->next)
1677                     if (rp->setname && !strcmp(rp->setname, setname))
1678                         break;
1679                 if (!rp)
1680                     break;
1681             }
1682             r->setname = xstrdup(setname);
1683             yaz_log(log_details, "%p ZOOM_connection_send_search: allocating "
1684                     "set %s", c, r->setname);
1685         }
1686         else
1687         {
1688             yaz_log(log_details, "%p ZOOM_connection_send_search: using "
1689                     "default set", c);
1690             r->setname = xstrdup("default");
1691         }
1692         ZOOM_options_set(r->options, "setname", r->setname);
1693     }
1694     search_req->resultSetName = odr_strdup(c->odr_out, r->setname);
1695     return send_APDU(c, apdu);
1696 }
1697
1698 static void response_default_diag(ZOOM_connection c, Z_DefaultDiagFormat *r)
1699 {
1700     char oid_name_buf[OID_STR_MAX];
1701     const char *oid_name;
1702     char *addinfo = 0;
1703
1704     oid_name = yaz_oid_to_string_buf(r->diagnosticSetId, 0, oid_name_buf);
1705     switch (r->which)
1706     {
1707     case Z_DefaultDiagFormat_v2Addinfo:
1708         addinfo = r->u.v2Addinfo;
1709         break;
1710     case Z_DefaultDiagFormat_v3Addinfo:
1711         addinfo = r->u.v3Addinfo;
1712         break;
1713     }
1714     xfree(c->addinfo);
1715     c->addinfo = 0;
1716     set_dset_error(c, *r->condition, oid_name, addinfo, 0);
1717 }
1718
1719 static void response_diag(ZOOM_connection c, Z_DiagRec *p)
1720 {
1721     if (p->which != Z_DiagRec_defaultFormat)
1722         set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1723     else
1724         response_default_diag(c, p->u.defaultFormat);
1725 }
1726
1727 ZOOM_API(ZOOM_record)
1728     ZOOM_record_clone(ZOOM_record srec)
1729 {
1730     char *buf;
1731     int size;
1732     ODR odr_enc;
1733     ZOOM_record nrec;
1734
1735     odr_enc = odr_createmem(ODR_ENCODE);
1736     if (!z_NamePlusRecord(odr_enc, &srec->npr, 0, 0))
1737         return 0;
1738     buf = odr_getbuf(odr_enc, &size, 0);
1739
1740     nrec = (ZOOM_record) xmalloc(sizeof(*nrec));
1741     nrec->odr = odr_createmem(ODR_DECODE);
1742     nrec->wrbuf_marc = 0;
1743     nrec->wrbuf_iconv = 0;
1744     nrec->wrbuf_opac = 0;
1745     odr_setbuf(nrec->odr, buf, size, 0);
1746     z_NamePlusRecord(nrec->odr, &nrec->npr, 0, 0);
1747
1748     odr_destroy(odr_enc);
1749     return nrec;
1750 }
1751
1752 ZOOM_API(ZOOM_record)
1753     ZOOM_resultset_record_immediate(ZOOM_resultset s,size_t pos)
1754 {
1755     const char *syntax =
1756         ZOOM_options_get(s->options, "preferredRecordSyntax");
1757     const char *elementSetName =
1758         ZOOM_options_get(s->options, "elementSetName");
1759
1760     return record_cache_lookup(s, pos, syntax, elementSetName);
1761 }
1762
1763 ZOOM_API(ZOOM_record)
1764     ZOOM_resultset_record(ZOOM_resultset r, size_t pos)
1765 {
1766     ZOOM_record rec = ZOOM_resultset_record_immediate(r, pos);
1767
1768     if (!rec)
1769     {
1770         /*
1771          * MIKE: I think force_sync should always be zero, but I don't
1772          * want to make this change until I get the go-ahead from
1773          * Adam, in case something depends on the old synchronous
1774          * behaviour.
1775          */
1776         int force_sync = 1;
1777         if (getenv("ZOOM_RECORD_NO_FORCE_SYNC")) force_sync = 0;
1778         ZOOM_resultset_retrieve(r, force_sync, pos, 1);
1779         rec = ZOOM_resultset_record_immediate(r, pos);
1780     }
1781     return rec;
1782 }
1783
1784 ZOOM_API(void)
1785     ZOOM_record_destroy(ZOOM_record rec)
1786 {
1787     if (!rec)
1788         return;
1789     if (rec->wrbuf_marc)
1790         wrbuf_destroy(rec->wrbuf_marc);
1791     if (rec->wrbuf_iconv)
1792         wrbuf_destroy(rec->wrbuf_iconv);
1793     if (rec->wrbuf_opac)
1794         wrbuf_destroy(rec->wrbuf_opac);
1795     odr_destroy(rec->odr);
1796     xfree(rec);
1797 }
1798
1799 static const char *marc_iconv_return(ZOOM_record rec, int marc_type,
1800                                      int *len,
1801                                      const char *buf, int sz,
1802                                      const char *record_charset)
1803 {
1804     char to[40];
1805     char from[40];
1806     yaz_iconv_t cd = 0;
1807     yaz_marc_t mt = yaz_marc_create();
1808
1809     *from = '\0';
1810     strcpy(to, "UTF-8");
1811     if (record_charset && *record_charset)
1812     {
1813         /* Use "from,to" or just "from" */
1814         const char *cp = strchr(record_charset, ',');
1815         int clen = strlen(record_charset);
1816         if (cp && cp[1])
1817         {
1818             strncpy( to, cp+1, sizeof(to)-1);
1819             to[sizeof(to)-1] = '\0';
1820             clen = cp - record_charset;
1821         }
1822         if (clen > sizeof(from)-1)
1823             clen = sizeof(from)-1;
1824
1825         if (clen)
1826             strncpy(from, record_charset, clen);
1827         from[clen] = '\0';
1828     }
1829
1830     if (*from && *to)
1831     {
1832         cd = yaz_iconv_open(to, from);
1833         yaz_marc_iconv(mt, cd);
1834     }
1835
1836     yaz_marc_xml(mt, marc_type);
1837     if (!rec->wrbuf_marc)
1838         rec->wrbuf_marc = wrbuf_alloc();
1839     wrbuf_rewind(rec->wrbuf_marc);
1840     if (yaz_marc_decode_wrbuf(mt, buf, sz, rec->wrbuf_marc) > 0)
1841     {
1842         yaz_marc_destroy(mt);
1843         if (cd)
1844             yaz_iconv_close(cd);
1845         if (len)
1846             *len = wrbuf_len(rec->wrbuf_marc);
1847         return wrbuf_cstr(rec->wrbuf_marc);
1848     }
1849     yaz_marc_destroy(mt);
1850     if (cd)
1851         yaz_iconv_close(cd);
1852     return 0;
1853 }
1854
1855 static const char *record_iconv_return(ZOOM_record rec, int *len,
1856                                        const char *buf, int sz,
1857                                        const char *record_charset)
1858 {
1859     char to[40];
1860     char from[40];
1861     yaz_iconv_t cd = 0;
1862
1863     *from = '\0';
1864     strcpy(to, "UTF-8");
1865
1866     if (record_charset && *record_charset)
1867     {
1868         /* Use "from,to" or just "from" */
1869         const char *cp = strchr(record_charset, ',');
1870         int clen = strlen(record_charset);
1871         if (cp && cp[1])
1872         {
1873             strncpy( to, cp+1, sizeof(to)-1);
1874             to[sizeof(to)-1] = '\0';
1875             clen = cp - record_charset;
1876         }
1877         if (clen > sizeof(from)-1)
1878             clen = sizeof(from)-1;
1879
1880         if (clen)
1881             strncpy(from, record_charset, clen);
1882         from[clen] = '\0';
1883     }
1884
1885     if (*from && *to && (cd = yaz_iconv_open(to, from)))
1886     {
1887         if (!rec->wrbuf_iconv)
1888             rec->wrbuf_iconv = wrbuf_alloc();
1889
1890         wrbuf_rewind(rec->wrbuf_iconv);
1891
1892         wrbuf_iconv_write(rec->wrbuf_iconv, cd, buf, sz);
1893         wrbuf_iconv_reset(rec->wrbuf_iconv, cd);
1894
1895         buf = wrbuf_cstr(rec->wrbuf_iconv);
1896         sz = wrbuf_len(rec->wrbuf_iconv);
1897         yaz_iconv_close(cd);
1898     }
1899     if (len)
1900         *len = sz;
1901     return buf;
1902 }
1903
1904 ZOOM_API(int)
1905     ZOOM_record_error(ZOOM_record rec, const char **cp,
1906                       const char **addinfo, const char **diagset)
1907 {
1908     Z_NamePlusRecord *npr;
1909
1910     if (!rec)
1911         return 0;
1912     npr = rec->npr;
1913     if (npr && npr->which == Z_NamePlusRecord_surrogateDiagnostic)
1914     {
1915         Z_DiagRec *diag_rec = npr->u.surrogateDiagnostic;
1916         int error = YAZ_BIB1_UNSPECIFIED_ERROR;
1917         const char *add = 0;
1918
1919         if (diag_rec->which == Z_DiagRec_defaultFormat)
1920         {
1921             Z_DefaultDiagFormat *ddf = diag_rec->u.defaultFormat;
1922             oid_class oclass;
1923
1924             error = *ddf->condition;
1925             switch (ddf->which)
1926             {
1927             case Z_DefaultDiagFormat_v2Addinfo:
1928                 add = ddf->u.v2Addinfo;
1929                 break;
1930             case Z_DefaultDiagFormat_v3Addinfo:
1931                 add = ddf->u.v3Addinfo;
1932                 break;
1933             }
1934             if (diagset)
1935                 *diagset =
1936                     yaz_oid_to_string(yaz_oid_std(),
1937                                       ddf->diagnosticSetId, &oclass);
1938         }
1939         else
1940         {
1941             if (diagset)
1942                 *diagset = "Bib-1";
1943         }
1944         if (addinfo)
1945             *addinfo = add ? add : "";
1946         if (cp)
1947             *cp = diagbib1_str(error);
1948         return error;
1949     }
1950     return 0;
1951 }
1952
1953 ZOOM_API(const char *)
1954     ZOOM_record_get(ZOOM_record rec, const char *type_spec, int *len)
1955 {
1956     char type[40];
1957     char charset[40];
1958     char xpath[512];
1959     const char *cp;
1960     int i;
1961     Z_NamePlusRecord *npr;
1962
1963     if (len)
1964         *len = 0; /* default return */
1965
1966     if (!rec)
1967         return 0;
1968     npr = rec->npr;
1969     if (!npr)
1970         return 0;
1971
1972     cp = type_spec;
1973     for (i = 0; cp[i] && i < sizeof(type)-1; i++)
1974     {
1975         if (cp[i] == ';' || cp[i] == ' ')
1976             break;
1977         type[i] = cp[i];
1978     }
1979     type[i] = '\0';
1980     charset[0] = '\0';
1981     while (type_spec[i] == ';')
1982     {
1983         i++;
1984         while (type_spec[i] == ' ')
1985             i++;
1986         if (!strncmp(type_spec+i, "charset=", 8))
1987         {
1988             int j = 0;
1989             i = i + 8; /* skip charset= */
1990             for (j = 0; type_spec[i]  && j < sizeof(charset)-1; i++, j++)
1991             {
1992                 if (type_spec[i] == ';' || type_spec[i] == ' ')
1993                     break;
1994                 charset[j] = cp[i];
1995             }
1996             charset[j] = '\0';
1997         }
1998         else if (!strncmp(type_spec+i, "xpath=", 6))
1999         {
2000             int j = 0;
2001             i = i + 6;
2002             for (j = 0; type_spec[i] && j < sizeof(xpath)-1; i++, j++)
2003                 xpath[j] = cp[i];
2004             xpath[j] = '\0';
2005         }
2006         while (type_spec[i] == ' ')
2007             i++;
2008     }
2009     if (!strcmp(type, "database"))
2010     {
2011         if (len)
2012             *len = (npr->databaseName ? strlen(npr->databaseName) : 0);
2013         return npr->databaseName;
2014     }
2015     else if (!strcmp(type, "syntax"))
2016     {
2017         const char *desc = 0;
2018         if (npr->which == Z_NamePlusRecord_databaseRecord)
2019         {
2020             Z_External *r = (Z_External *) npr->u.databaseRecord;
2021             desc = yaz_oid_to_string(yaz_oid_std(), r->direct_reference, 0);
2022         }
2023         if (!desc)
2024             desc = "none";
2025         if (len)
2026             *len = strlen(desc);
2027         return desc;
2028     }
2029     if (npr->which != Z_NamePlusRecord_databaseRecord)
2030         return 0;
2031
2032     /* from now on - we have a database record .. */
2033     if (!strcmp(type, "render"))
2034     {
2035         Z_External *r = (Z_External *) npr->u.databaseRecord;
2036         const Odr_oid *oid = r->direct_reference;
2037
2038         /* render bibliographic record .. */
2039         if (r->which == Z_External_OPAC)
2040         {
2041             r = r->u.opac->bibliographicRecord;
2042             if (!r)
2043                 return 0;
2044             oid = r->direct_reference;
2045         }
2046         if (r->which == Z_External_sutrs)
2047             return record_iconv_return(rec, len,
2048                                        (char*) r->u.sutrs->buf,
2049                                        r->u.sutrs->len,
2050                                        charset);
2051         else if (r->which == Z_External_octet)
2052         {
2053             if (yaz_oid_is_iso2709(oid))
2054             {
2055                 const char *ret_buf = marc_iconv_return(
2056                     rec, YAZ_MARC_LINE, len,
2057                     (const char *) r->u.octet_aligned->buf,
2058                     r->u.octet_aligned->len,
2059                     charset);
2060                 if (ret_buf)
2061                     return ret_buf;
2062             }
2063             return record_iconv_return(rec, len,
2064                                        (const char *) r->u.octet_aligned->buf,
2065                                        r->u.octet_aligned->len,
2066                                        charset);
2067         }
2068         else if (r->which == Z_External_grs1)
2069         {
2070             if (!rec->wrbuf_marc)
2071                 rec->wrbuf_marc = wrbuf_alloc();
2072             wrbuf_rewind(rec->wrbuf_marc);
2073             yaz_display_grs1(rec->wrbuf_marc, r->u.grs1, 0);
2074             return record_iconv_return(rec, len,
2075                                        wrbuf_buf(rec->wrbuf_marc),
2076                                        wrbuf_len(rec->wrbuf_marc),
2077                                        charset);
2078         }
2079         return 0;
2080     }
2081     else if (!strcmp(type, "xml"))
2082     {
2083         Z_External *r = (Z_External *) npr->u.databaseRecord;
2084         const Odr_oid *oid = r->direct_reference;
2085
2086         /* render bibliographic record .. */
2087         if (r->which == Z_External_OPAC)
2088         {
2089             r = r->u.opac->bibliographicRecord;
2090             if (!r)
2091                 return 0;
2092             oid = r->direct_reference;
2093         }
2094
2095         if (r->which == Z_External_sutrs)
2096             return record_iconv_return(rec, len,
2097                                        (const char *) r->u.sutrs->buf,
2098                                        r->u.sutrs->len,
2099                                        charset);
2100         else if (r->which == Z_External_octet)
2101         {
2102             int marc_decode_type = YAZ_MARC_MARCXML;
2103             if (yaz_oid_is_iso2709(oid))
2104             {
2105                 const char *ret_buf = marc_iconv_return(
2106                     rec, marc_decode_type, len,
2107                     (const char *) r->u.octet_aligned->buf,
2108                     r->u.octet_aligned->len,
2109                     charset);
2110                 if (ret_buf)
2111                     return ret_buf;
2112             }
2113             return record_iconv_return(rec, len,
2114                                        (const char *) r->u.octet_aligned->buf,
2115                                        r->u.octet_aligned->len,
2116                                        charset);
2117         }
2118         else if (r->which == Z_External_grs1)
2119         {
2120             if (len) *len = 5;
2121             return "GRS-1";
2122         }
2123         return 0;
2124     }
2125     else if (!strcmp(type, "raw"))
2126     {
2127         Z_External *r = (Z_External *) npr->u.databaseRecord;
2128
2129         if (r->which == Z_External_sutrs)
2130         {
2131             if (len) *len = r->u.sutrs->len;
2132             return (const char *) r->u.sutrs->buf;
2133         }
2134         else if (r->which == Z_External_octet)
2135         {
2136             if (len) *len = r->u.octet_aligned->len;
2137             return (const char *) r->u.octet_aligned->buf;
2138         }
2139         else /* grs-1, explain, OPAC, ... */
2140         {
2141             if (len) *len = -1;
2142             return (const char *) npr->u.databaseRecord;
2143         }
2144         return 0;
2145     }
2146     else if (!strcmp (type, "ext"))
2147     {
2148         if (len) *len = -1;
2149         return (const char *) npr->u.databaseRecord;
2150     }
2151     else if (!strcmp (type, "opac"))
2152
2153     {
2154         Z_External *r = (Z_External *) npr->u.databaseRecord;
2155         if (r->which == Z_External_OPAC)
2156         {
2157             if (!rec->wrbuf_opac)
2158                 rec->wrbuf_opac = wrbuf_alloc();
2159             wrbuf_rewind(rec->wrbuf_opac);
2160             yaz_display_OPAC(rec->wrbuf_opac, r->u.opac, 0);
2161             return record_iconv_return(rec, len,
2162                                        wrbuf_buf(rec->wrbuf_opac),
2163                                        wrbuf_len(rec->wrbuf_opac),
2164                                        charset);
2165         }
2166     }
2167     return 0;
2168 }
2169
2170 static int strcmp_null(const char *v1, const char *v2)
2171 {
2172     if (!v1 && !v2)
2173         return 0;
2174     if (!v1 || !v2)
2175         return -1;
2176     return strcmp(v1, v2);
2177 }
2178
2179 static size_t record_hash(int pos)
2180 {
2181     if (pos < 0)
2182         pos = 0;
2183     return pos % RECORD_HASH_SIZE;
2184 }
2185
2186 static void record_cache_add(ZOOM_resultset r, Z_NamePlusRecord *npr,
2187                              int pos,
2188                              const char *syntax, const char *elementSetName)
2189 {
2190     ZOOM_record_cache rc;
2191
2192     ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_RECV_RECORD);
2193     ZOOM_connection_put_event(r->connection, event);
2194
2195     for (rc = r->record_hash[record_hash(pos)]; rc; rc = rc->next)
2196     {
2197         if (pos == rc->pos)
2198         {
2199             if (strcmp_null(r->schema, rc->schema))
2200                 continue;
2201             if (strcmp_null(elementSetName,rc->elementSetName))
2202                 continue;
2203             if (strcmp_null(syntax, rc->syntax))
2204                 continue;
2205             /* not destroying rc->npr (it's handled by nmem )*/
2206             rc->rec.npr = npr;
2207             /* keeping wrbuf_marc too */
2208             return;
2209         }
2210     }
2211     rc = (ZOOM_record_cache) odr_malloc(r->odr, sizeof(*rc));
2212     rc->rec.npr = npr;
2213     rc->rec.odr = 0;
2214     rc->rec.wrbuf_marc = 0;
2215     rc->rec.wrbuf_iconv = 0;
2216     rc->rec.wrbuf_opac = 0;
2217     if (elementSetName)
2218         rc->elementSetName = odr_strdup(r->odr, elementSetName);
2219     else
2220         rc->elementSetName = 0;
2221
2222     if (syntax)
2223         rc->syntax = odr_strdup(r->odr, syntax);
2224     else
2225         rc->syntax = 0;
2226
2227     if (r->schema)
2228         rc->schema = odr_strdup(r->odr, r->schema);
2229     else
2230         rc->schema = 0;
2231
2232     rc->pos = pos;
2233     rc->next = r->record_hash[record_hash(pos)];
2234     r->record_hash[record_hash(pos)] = rc;
2235 }
2236
2237 static ZOOM_record record_cache_lookup(ZOOM_resultset r, int pos,
2238                                        const char *syntax,
2239                                        const char *elementSetName)
2240 {
2241     ZOOM_record_cache rc;
2242
2243     for (rc = r->record_hash[record_hash(pos)]; rc; rc = rc->next)
2244     {
2245         if (pos == rc->pos)
2246         {
2247             if (strcmp_null(r->schema, rc->schema))
2248                 continue;
2249             if (strcmp_null(elementSetName,rc->elementSetName))
2250                 continue;
2251             if (strcmp_null(syntax, rc->syntax))
2252                 continue;
2253             return &rc->rec;
2254         }
2255     }
2256     return 0;
2257 }
2258
2259 static void handle_records(ZOOM_connection c, Z_Records *sr,
2260                            int present_phase)
2261 {
2262     ZOOM_resultset resultset;
2263     int *start, *count;
2264     const char *syntax = 0, *elementSetName = 0;
2265
2266     if (!c->tasks)
2267         return ;
2268     switch (c->tasks->which)
2269     {
2270     case ZOOM_TASK_SEARCH:
2271         resultset = c->tasks->u.search.resultset;
2272         start = &c->tasks->u.search.start;
2273         count = &c->tasks->u.search.count;
2274         syntax = c->tasks->u.search.syntax;
2275         elementSetName = c->tasks->u.search.elementSetName;
2276         break;
2277     case ZOOM_TASK_RETRIEVE:
2278         resultset = c->tasks->u.retrieve.resultset;
2279         start = &c->tasks->u.retrieve.start;
2280         count = &c->tasks->u.retrieve.count;
2281         syntax = c->tasks->u.retrieve.syntax;
2282         elementSetName = c->tasks->u.retrieve.elementSetName;
2283         break;
2284     default:
2285         return;
2286     }
2287     if (sr && sr->which == Z_Records_NSD)
2288         response_default_diag(c, sr->u.nonSurrogateDiagnostic);
2289     else if (sr && sr->which == Z_Records_multipleNSD)
2290     {
2291         if (sr->u.multipleNonSurDiagnostics->num_diagRecs >= 1)
2292             response_diag(c, sr->u.multipleNonSurDiagnostics->diagRecs[0]);
2293         else
2294             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
2295     }
2296     else
2297     {
2298         if (*count + *start > resultset->size)
2299             *count = resultset->size - *start;
2300         if (*count < 0)
2301             *count = 0;
2302         if (sr && sr->which == Z_Records_DBOSD)
2303         {
2304             int i;
2305             NMEM nmem = odr_extract_mem(c->odr_in);
2306             Z_NamePlusRecordList *p =
2307                 sr->u.databaseOrSurDiagnostics;
2308             for (i = 0; i<p->num_records; i++)
2309             {
2310                 record_cache_add(resultset, p->records[i], i + *start,
2311                                  syntax, elementSetName);
2312             }
2313             *count -= i;
2314             if (*count < 0)
2315                 *count = 0;
2316             *start += i;
2317             yaz_log(log_details,
2318                     "handle_records resultset=%p start=%d count=%d",
2319                     resultset, *start, *count);
2320
2321             /* transfer our response to search_nmem .. we need it later */
2322             nmem_transfer(odr_getmem(resultset->odr), nmem);
2323             nmem_destroy(nmem);
2324             if (present_phase && p->num_records == 0)
2325             {
2326                 /* present response and we didn't get any records! */
2327                 Z_NamePlusRecord *myrec =
2328                     zget_surrogateDiagRec(resultset->odr, 0, 14, 0);
2329                 record_cache_add(resultset, myrec, *start,
2330                                  syntax, elementSetName);
2331             }
2332         }
2333         else if (present_phase)
2334         {
2335             /* present response and we didn't get any records! */
2336             Z_NamePlusRecord *myrec =
2337                 zget_surrogateDiagRec(resultset->odr, 0, 14, 0);
2338             record_cache_add(resultset, myrec, *start, syntax, elementSetName);
2339         }
2340     }
2341 }
2342
2343 static void handle_present_response(ZOOM_connection c, Z_PresentResponse *pr)
2344 {
2345     handle_records(c, pr->records, 1);
2346 }
2347
2348 static void handle_queryExpressionTerm(ZOOM_options opt, const char *name,
2349                                        Z_Term *term)
2350 {
2351     switch (term->which)
2352     {
2353     case Z_Term_general:
2354         ZOOM_options_setl(opt, name,
2355                           (const char *)(term->u.general->buf),
2356                           term->u.general->len);
2357         break;
2358     case Z_Term_characterString:
2359         ZOOM_options_set(opt, name, term->u.characterString);
2360         break;
2361     case Z_Term_numeric:
2362         ZOOM_options_set_int(opt, name, *term->u.numeric);
2363         break;
2364     }
2365 }
2366
2367 static void handle_queryExpression(ZOOM_options opt, const char *name,
2368                                    Z_QueryExpression *exp)
2369 {
2370     char opt_name[80];
2371
2372     switch (exp->which)
2373     {
2374     case Z_QueryExpression_term:
2375         if (exp->u.term && exp->u.term->queryTerm)
2376         {
2377             sprintf(opt_name, "%s.term", name);
2378             handle_queryExpressionTerm(opt, opt_name, exp->u.term->queryTerm);
2379         }
2380         break;
2381     case Z_QueryExpression_query:
2382         break;
2383     }
2384 }
2385
2386 static void handle_searchResult(ZOOM_connection c, ZOOM_resultset resultset,
2387                                 Z_OtherInformation *o)
2388 {
2389     int i;
2390     for (i = 0; o && i < o->num_elements; i++)
2391     {
2392         if (o->list[i]->which == Z_OtherInfo_externallyDefinedInfo)
2393         {
2394             Z_External *ext = o->list[i]->information.externallyDefinedInfo;
2395
2396             if (ext->which == Z_External_searchResult1)
2397             {
2398                 int j;
2399                 Z_SearchInfoReport *sr = ext->u.searchResult1;
2400
2401                 if (sr->num)
2402                     ZOOM_options_set_int(
2403                         resultset->options, "searchresult.size", sr->num);
2404
2405                 for (j = 0; j < sr->num; j++)
2406                 {
2407                     Z_SearchInfoReport_s *ent =
2408                         ext->u.searchResult1->elements[j];
2409                     char pref[80];
2410
2411                     sprintf(pref, "searchresult.%d", j);
2412
2413                     if (ent->subqueryId)
2414                     {
2415                         char opt_name[80];
2416                         sprintf(opt_name, "%s.id", pref);
2417                         ZOOM_options_set(resultset->options, opt_name,
2418                                          ent->subqueryId);
2419                     }
2420                     if (ent->subqueryExpression)
2421                     {
2422                         char opt_name[80];
2423                         sprintf(opt_name, "%s.subquery", pref);
2424                         handle_queryExpression(resultset->options, opt_name,
2425                                                ent->subqueryExpression);
2426                     }
2427                     if (ent->subqueryInterpretation)
2428                     {
2429                         char opt_name[80];
2430                         sprintf(opt_name, "%s.interpretation", pref);
2431                         handle_queryExpression(resultset->options, opt_name,
2432                                                ent->subqueryInterpretation);
2433                     }
2434                     if (ent->subqueryRecommendation)
2435                     {
2436                         char opt_name[80];
2437                         sprintf(opt_name, "%s.recommendation", pref);
2438                         handle_queryExpression(resultset->options, opt_name,
2439                                                ent->subqueryRecommendation);
2440                     }
2441                     if (ent->subqueryCount)
2442                     {
2443                         char opt_name[80];
2444                         sprintf(opt_name, "%s.count", pref);
2445                         ZOOM_options_set_int(resultset->options, opt_name,
2446                                              *ent->subqueryCount);
2447                     }
2448                 }
2449             }
2450         }
2451     }
2452 }
2453
2454 static void handle_search_response(ZOOM_connection c, Z_SearchResponse *sr)
2455 {
2456     ZOOM_resultset resultset;
2457     ZOOM_Event event;
2458
2459     if (!c->tasks || c->tasks->which != ZOOM_TASK_SEARCH)
2460         return ;
2461
2462     event = ZOOM_Event_create(ZOOM_EVENT_RECV_SEARCH);
2463     ZOOM_connection_put_event(c, event);
2464
2465     resultset = c->tasks->u.search.resultset;
2466
2467     if (sr->resultSetStatus)
2468     {
2469         ZOOM_options_set_int(resultset->options, "resultSetStatus",
2470                              *sr->resultSetStatus);
2471     }
2472     if (sr->presentStatus)
2473     {
2474         ZOOM_options_set_int(resultset->options, "presentStatus",
2475                              *sr->presentStatus);
2476     }
2477     handle_searchResult(c, resultset, sr->additionalSearchInfo);
2478
2479     resultset->size = *sr->resultCount;
2480     handle_records(c, sr->records, 0);
2481 }
2482
2483 static void sort_response(ZOOM_connection c, Z_SortResponse *res)
2484 {
2485     if (res->diagnostics && res->num_diagnostics > 0)
2486         response_diag(c, res->diagnostics[0]);
2487 }
2488
2489 static int scan_response(ZOOM_connection c, Z_ScanResponse *res)
2490 {
2491     NMEM nmem = odr_extract_mem(c->odr_in);
2492     ZOOM_scanset scan;
2493
2494     if (!c->tasks || c->tasks->which != ZOOM_TASK_SCAN)
2495         return 0;
2496     scan = c->tasks->u.scan.scan;
2497
2498     if (res->entries && res->entries->nonsurrogateDiagnostics)
2499         response_diag(c, res->entries->nonsurrogateDiagnostics[0]);
2500     scan->scan_response = res;
2501     scan->srw_scan_response = 0;
2502     nmem_transfer(odr_getmem(scan->odr), nmem);
2503     if (res->stepSize)
2504         ZOOM_options_set_int(scan->options, "stepSize", *res->stepSize);
2505     if (res->positionOfTerm)
2506         ZOOM_options_set_int(scan->options, "position", *res->positionOfTerm);
2507     if (res->scanStatus)
2508         ZOOM_options_set_int(scan->options, "scanStatus", *res->scanStatus);
2509     if (res->numberOfEntriesReturned)
2510         ZOOM_options_set_int(scan->options, "number",
2511                              *res->numberOfEntriesReturned);
2512     nmem_destroy(nmem);
2513     return 1;
2514 }
2515
2516 static zoom_ret send_sort(ZOOM_connection c,
2517                           ZOOM_resultset resultset)
2518 {
2519     if (c->error)
2520         resultset->r_sort_spec = 0;
2521     if (resultset->r_sort_spec)
2522     {
2523         Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_sortRequest);
2524         Z_SortRequest *req = apdu->u.sortRequest;
2525
2526         req->num_inputResultSetNames = 1;
2527         req->inputResultSetNames = (Z_InternationalString **)
2528             odr_malloc(c->odr_out, sizeof(*req->inputResultSetNames));
2529         req->inputResultSetNames[0] =
2530             odr_strdup(c->odr_out, resultset->setname);
2531         req->sortedResultSetName = odr_strdup(c->odr_out, resultset->setname);
2532         req->sortSequence = resultset->r_sort_spec;
2533         resultset->r_sort_spec = 0;
2534         return send_APDU(c, apdu);
2535     }
2536     return zoom_complete;
2537 }
2538
2539 static zoom_ret send_present(ZOOM_connection c)
2540 {
2541     Z_APDU *apdu = 0;
2542     Z_PresentRequest *req = 0;
2543     int i = 0;
2544     const char *syntax = 0;
2545     const char *elementSetName = 0;
2546     ZOOM_resultset  resultset;
2547     int *start, *count;
2548
2549     if (!c->tasks)
2550     {
2551         yaz_log(log_details, "%p send_present no tasks", c);
2552         return zoom_complete;
2553     }
2554
2555     switch (c->tasks->which)
2556     {
2557     case ZOOM_TASK_SEARCH:
2558         resultset = c->tasks->u.search.resultset;
2559         start = &c->tasks->u.search.start;
2560         count = &c->tasks->u.search.count;
2561         syntax = c->tasks->u.search.syntax;
2562         elementSetName = c->tasks->u.search.elementSetName;
2563         break;
2564     case ZOOM_TASK_RETRIEVE:
2565         resultset = c->tasks->u.retrieve.resultset;
2566         start = &c->tasks->u.retrieve.start;
2567         count = &c->tasks->u.retrieve.count;
2568         syntax = c->tasks->u.retrieve.syntax;
2569         elementSetName = c->tasks->u.retrieve.elementSetName;
2570         break;
2571     default:
2572         return zoom_complete;
2573     }
2574     yaz_log(log_details, "%p send_present start=%d count=%d",
2575             c, *start, *count);
2576
2577     if (*start < 0 || *count < 0 || *start + *count > resultset->size)
2578     {
2579         set_dset_error(c, YAZ_BIB1_PRESENT_REQUEST_OUT_OF_RANGE, "Bib-1",
2580                        "", 0);
2581     }
2582     if (c->error)                  /* don't continue on error */
2583         return zoom_complete;
2584     yaz_log(log_details, "send_present resultset=%p start=%d count=%d",
2585             resultset, *start, *count);
2586
2587     for (i = 0; i < *count; i++)
2588     {
2589         ZOOM_record rec =
2590             record_cache_lookup(resultset, i + *start, syntax, elementSetName);
2591         if (!rec)
2592             break;
2593         else
2594         {
2595             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_RECV_RECORD);
2596             ZOOM_connection_put_event(c, event);
2597         }
2598     }
2599     *start += i;
2600     *count -= i;
2601
2602     if (*count == 0)
2603     {
2604         yaz_log(log_details, "%p send_present skip=%d no more to fetch", c, i);
2605         return zoom_complete;
2606     }
2607
2608     apdu = zget_APDU(c->odr_out, Z_APDU_presentRequest);
2609     req = apdu->u.presentRequest;
2610
2611     if (i)
2612         yaz_log(log_details, "%p send_present skip=%d", c, i);
2613
2614     *req->resultSetStartPoint = *start + 1;
2615     *req->numberOfRecordsRequested = resultset->step>0 ?
2616         resultset->step : *count;
2617     if (*req->numberOfRecordsRequested + *start > resultset->size)
2618         *req->numberOfRecordsRequested = resultset->size - *start;
2619     assert(*req->numberOfRecordsRequested > 0);
2620
2621     if (syntax && *syntax)
2622         req->preferredRecordSyntax =
2623             zoom_yaz_str_to_z3950oid(c, CLASS_RECSYN, syntax);
2624
2625     if (resultset->schema && *resultset->schema)
2626     {
2627         Z_RecordComposition *compo = (Z_RecordComposition *)
2628             odr_malloc(c->odr_out, sizeof(*compo));
2629
2630         req->recordComposition = compo;
2631         compo->which = Z_RecordComp_complex;
2632         compo->u.complex = (Z_CompSpec *)
2633             odr_malloc(c->odr_out, sizeof(*compo->u.complex));
2634         compo->u.complex->selectAlternativeSyntax = (bool_t *)
2635             odr_malloc(c->odr_out, sizeof(bool_t));
2636         *compo->u.complex->selectAlternativeSyntax = 0;
2637
2638         compo->u.complex->generic = (Z_Specification *)
2639             odr_malloc(c->odr_out, sizeof(*compo->u.complex->generic));
2640
2641         compo->u.complex->generic->which = Z_Schema_oid;
2642         compo->u.complex->generic->schema.oid = (Odr_oid *)
2643             zoom_yaz_str_to_z3950oid (c, CLASS_SCHEMA, resultset->schema);
2644
2645         if (!compo->u.complex->generic->schema.oid)
2646         {
2647             /* OID wasn't a schema! Try record syntax instead. */
2648
2649             compo->u.complex->generic->schema.oid = (Odr_oid *)
2650                 zoom_yaz_str_to_z3950oid (c, CLASS_RECSYN, resultset->schema);
2651         }
2652         if (elementSetName && *elementSetName)
2653         {
2654             compo->u.complex->generic->elementSpec = (Z_ElementSpec *)
2655                 odr_malloc(c->odr_out, sizeof(Z_ElementSpec));
2656             compo->u.complex->generic->elementSpec->which =
2657                 Z_ElementSpec_elementSetName;
2658             compo->u.complex->generic->elementSpec->u.elementSetName =
2659                 odr_strdup(c->odr_out, elementSetName);
2660         }
2661         else
2662             compo->u.complex->generic->elementSpec = 0;
2663         compo->u.complex->num_dbSpecific = 0;
2664         compo->u.complex->dbSpecific = 0;
2665         compo->u.complex->num_recordSyntax = 0;
2666         compo->u.complex->recordSyntax = 0;
2667     }
2668     else if (elementSetName && *elementSetName)
2669     {
2670         Z_ElementSetNames *esn = (Z_ElementSetNames *)
2671             odr_malloc(c->odr_out, sizeof(*esn));
2672         Z_RecordComposition *compo = (Z_RecordComposition *)
2673             odr_malloc(c->odr_out, sizeof(*compo));
2674
2675         esn->which = Z_ElementSetNames_generic;
2676         esn->u.generic = odr_strdup(c->odr_out, elementSetName);
2677         compo->which = Z_RecordComp_simple;
2678         compo->u.simple = esn;
2679         req->recordComposition = compo;
2680     }
2681     req->resultSetId = odr_strdup(c->odr_out, resultset->setname);
2682     return send_APDU(c, apdu);
2683 }
2684
2685 ZOOM_API(ZOOM_scanset)
2686     ZOOM_connection_scan(ZOOM_connection c, const char *start)
2687 {
2688     ZOOM_scanset s;
2689     ZOOM_query q = ZOOM_query_create();
2690
2691     ZOOM_query_prefix(q, start);
2692
2693     s = ZOOM_connection_scan1(c, q);
2694     ZOOM_query_destroy(q);
2695     return s;
2696
2697 }
2698
2699 ZOOM_API(ZOOM_scanset)
2700     ZOOM_connection_scan1(ZOOM_connection c, ZOOM_query q)
2701 {
2702     ZOOM_scanset scan = 0;
2703
2704     if (!q->z_query)
2705         return 0;
2706     scan = (ZOOM_scanset) xmalloc(sizeof(*scan));
2707     scan->connection = c;
2708     scan->odr = odr_createmem(ODR_DECODE);
2709     scan->options = ZOOM_options_create_with_parent(c->options);
2710     scan->refcount = 1;
2711     scan->scan_response = 0;
2712     scan->srw_scan_response = 0;
2713
2714     scan->query = q;
2715     (q->refcount)++;
2716     scan->databaseNames = set_DatabaseNames(c, c->options,
2717                                             &scan->num_databaseNames,
2718                                             scan->odr);
2719
2720     if (1)
2721     {
2722         ZOOM_task task = ZOOM_connection_add_task(c, ZOOM_TASK_SCAN);
2723         task->u.scan.scan = scan;
2724
2725         (scan->refcount)++;
2726         if (!c->async)
2727         {
2728             while (ZOOM_event(1, &c))
2729                 ;
2730         }
2731     }
2732     return scan;
2733 }
2734
2735 ZOOM_API(void)
2736     ZOOM_scanset_destroy(ZOOM_scanset scan)
2737 {
2738     if (!scan)
2739         return;
2740     (scan->refcount)--;
2741     if (scan->refcount == 0)
2742     {
2743         ZOOM_query_destroy(scan->query);
2744
2745         odr_destroy(scan->odr);
2746
2747         ZOOM_options_destroy(scan->options);
2748         xfree(scan);
2749     }
2750 }
2751
2752 static zoom_ret send_package(ZOOM_connection c)
2753 {
2754     ZOOM_Event event;
2755
2756     yaz_log(log_details, "%p send_package", c);
2757     if (!c->tasks)
2758         return zoom_complete;
2759     assert (c->tasks->which == ZOOM_TASK_PACKAGE);
2760
2761     event = ZOOM_Event_create(ZOOM_EVENT_SEND_APDU);
2762     ZOOM_connection_put_event(c, event);
2763
2764     c->buf_out = c->tasks->u.package->buf_out;
2765     c->len_out = c->tasks->u.package->len_out;
2766
2767     return do_write(c);
2768 }
2769
2770 static zoom_ret ZOOM_connection_send_scan(ZOOM_connection c)
2771 {
2772     ZOOM_scanset scan;
2773     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_scanRequest);
2774     Z_ScanRequest *req = apdu->u.scanRequest;
2775
2776     yaz_log(log_details, "%p send_scan", c);
2777     if (!c->tasks)
2778         return zoom_complete;
2779     assert (c->tasks->which == ZOOM_TASK_SCAN);
2780     scan = c->tasks->u.scan.scan;
2781
2782     /* Z39.50 scan can only carry RPN */
2783     if (scan->query->z_query->which == Z_Query_type_1 ||
2784         scan->query->z_query->which == Z_Query_type_101)
2785     {
2786         Z_RPNQuery *rpn = scan->query->z_query->u.type_1;
2787         const char *cp = ZOOM_options_get(scan->options, "rpnCharset");
2788         if (cp)
2789         {
2790             yaz_iconv_t cd = yaz_iconv_open(cp, "UTF-8");
2791             if (cd)
2792             {
2793                 rpn = yaz_copy_z_RPNQuery(rpn, c->odr_out);
2794
2795                 yaz_query_charset_convert_rpnquery(
2796                     rpn, c->odr_out, cd);
2797                 yaz_iconv_close(cd);
2798             }
2799         }
2800         req->attributeSet = rpn->attributeSetId;
2801         if (!req->attributeSet)
2802             req->attributeSet = odr_oiddup(c->odr_out, yaz_oid_attset_bib_1);
2803         if (rpn->RPNStructure->which == Z_RPNStructure_simple &&
2804             rpn->RPNStructure->u.simple->which == Z_Operand_APT)
2805         {
2806             req->termListAndStartPoint =
2807                 rpn->RPNStructure->u.simple->u.attributesPlusTerm;
2808         }
2809         else
2810         {
2811             set_ZOOM_error(c, ZOOM_ERROR_INVALID_QUERY, 0);
2812             return zoom_complete;
2813         }
2814     }
2815     else
2816     {
2817         set_ZOOM_error(c, ZOOM_ERROR_UNSUPPORTED_QUERY, 0);
2818         return zoom_complete;
2819     }
2820
2821     *req->numberOfTermsRequested =
2822         ZOOM_options_get_int(scan->options, "number", 10);
2823
2824     req->preferredPositionInResponse =
2825         odr_intdup(c->odr_out,
2826                    ZOOM_options_get_int(scan->options, "position", 1));
2827
2828     req->stepSize =
2829         odr_intdup(c->odr_out,
2830                    ZOOM_options_get_int(scan->options, "stepSize", 0));
2831
2832     req->databaseNames = scan->databaseNames;
2833     req->num_databaseNames = scan->num_databaseNames;
2834
2835     return send_APDU(c, apdu);
2836 }
2837
2838 #if YAZ_HAVE_XML2
2839 static zoom_ret ZOOM_connection_srw_send_scan(ZOOM_connection c)
2840 {
2841     ZOOM_scanset scan;
2842     Z_SRW_PDU *sr = 0;
2843     const char *option_val = 0;
2844
2845     if (!c->tasks)
2846         return zoom_complete;
2847     assert (c->tasks->which == ZOOM_TASK_SCAN);
2848     scan = c->tasks->u.scan.scan;
2849
2850     sr = ZOOM_srw_get_pdu(c, Z_SRW_scan_request);
2851
2852     /* SRU scan can only carry CQL and PQF */
2853     if (scan->query->z_query->which == Z_Query_type_104)
2854     {
2855         sr->u.scan_request->query_type = Z_SRW_query_type_cql;
2856         sr->u.scan_request->scanClause.cql = scan->query->query_string;
2857     }
2858     else if (scan->query->z_query->which == Z_Query_type_1
2859              || scan->query->z_query->which == Z_Query_type_101)
2860     {
2861         sr->u.scan_request->query_type = Z_SRW_query_type_pqf;
2862         sr->u.scan_request->scanClause.pqf = scan->query->query_string;
2863     }
2864     else
2865     {
2866         set_ZOOM_error(c, ZOOM_ERROR_UNSUPPORTED_QUERY, 0);
2867         return zoom_complete;
2868     }
2869
2870     sr->u.scan_request->maximumTerms = odr_intdup(
2871         c->odr_out, ZOOM_options_get_int(scan->options, "number", 10));
2872
2873     sr->u.scan_request->responsePosition = odr_intdup(
2874         c->odr_out, ZOOM_options_get_int(scan->options, "position", 1));
2875
2876     option_val = ZOOM_options_get(scan->options, "extraArgs");
2877     if (option_val)
2878         sr->extra_args = odr_strdup(c->odr_out, option_val);
2879     return send_srw(c, sr);
2880 }
2881 #else
2882 static zoom_ret ZOOM_connection_srw_send_scan(ZOOM_connection c)
2883 {
2884     return zoom_complete;
2885 }
2886 #endif
2887
2888
2889 ZOOM_API(size_t)
2890     ZOOM_scanset_size(ZOOM_scanset scan)
2891 {
2892     if (!scan)
2893         return 0;
2894
2895     if (scan->scan_response && scan->scan_response->entries)
2896         return scan->scan_response->entries->num_entries;
2897     else if (scan->srw_scan_response)
2898         return scan->srw_scan_response->num_terms;
2899     return 0;
2900 }
2901
2902 static void ZOOM_scanset_term_x(ZOOM_scanset scan, size_t pos,
2903                                 int *occ,
2904                                 const char **value_term, size_t *value_len,
2905                                 const char **disp_term, size_t *disp_len)
2906 {
2907     size_t noent = ZOOM_scanset_size(scan);
2908
2909     *value_term = 0;
2910     *value_len = 0;
2911
2912     *disp_term = 0;
2913     *disp_len = 0;
2914
2915     *occ = 0;
2916     if (pos >= noent || pos < 0)
2917         return;
2918     if (scan->scan_response)
2919     {
2920         Z_ScanResponse *res = scan->scan_response;
2921         if (res->entries->entries[pos]->which == Z_Entry_termInfo)
2922         {
2923             Z_TermInfo *t = res->entries->entries[pos]->u.termInfo;
2924
2925             *value_term = (const char *) t->term->u.general->buf;
2926             *value_len = t->term->u.general->len;
2927             if (t->displayTerm)
2928             {
2929                 *disp_term = t->displayTerm;
2930                 *disp_len = strlen(*disp_term);
2931             }
2932             else if (t->term->which == Z_Term_general)
2933             {
2934                 *disp_term = (const char *) t->term->u.general->buf;
2935                 *disp_len = t->term->u.general->len;
2936             }
2937             *occ = t->globalOccurrences ? *t->globalOccurrences : 0;
2938         }
2939     }
2940     if (scan->srw_scan_response)
2941     {
2942         Z_SRW_scanResponse *res = scan->srw_scan_response;
2943         Z_SRW_scanTerm *t = res->terms + pos;
2944         if (t)
2945         {
2946             *value_term = t->value;
2947             *value_len = strlen(*value_term);
2948
2949             if (t->displayTerm)
2950                 *disp_term = t->displayTerm;
2951             else
2952                 *disp_term = t->value;
2953             *disp_len = strlen(*disp_term);
2954             *occ = t->numberOfRecords ? *t->numberOfRecords : 0;
2955         }
2956     }
2957 }
2958
2959 ZOOM_API(const char *)
2960     ZOOM_scanset_term(ZOOM_scanset scan, size_t pos,
2961                       int *occ, int *len)
2962 {
2963     const char *value_term = 0;
2964     size_t value_len = 0;
2965     const char *disp_term = 0;
2966     size_t disp_len = 0;
2967
2968     ZOOM_scanset_term_x(scan, pos, occ, &value_term, &value_len,
2969                         &disp_term, &disp_len);
2970
2971     *len = value_len;
2972     return value_term;
2973 }
2974
2975 ZOOM_API(const char *)
2976     ZOOM_scanset_display_term(ZOOM_scanset scan, size_t pos,
2977                               int *occ, int *len)
2978 {
2979     const char *value_term = 0;
2980     size_t value_len = 0;
2981     const char *disp_term = 0;
2982     size_t disp_len = 0;
2983
2984     ZOOM_scanset_term_x(scan, pos, occ, &value_term, &value_len,
2985                         &disp_term, &disp_len);
2986
2987     *len = disp_len;
2988     return disp_term;
2989 }
2990
2991 ZOOM_API(const char *)
2992     ZOOM_scanset_option_get(ZOOM_scanset scan, const char *key)
2993 {
2994     return ZOOM_options_get(scan->options, key);
2995 }
2996
2997 ZOOM_API(void)
2998     ZOOM_scanset_option_set(ZOOM_scanset scan, const char *key,
2999                             const char *val)
3000 {
3001     ZOOM_options_set(scan->options, key, val);
3002 }
3003
3004 static Z_APDU *create_es_package(ZOOM_package p, const Odr_oid *oid)
3005 {
3006     const char *str;
3007     Z_APDU *apdu = zget_APDU(p->odr_out, Z_APDU_extendedServicesRequest);
3008     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
3009
3010     str = ZOOM_options_get(p->options, "package-name");
3011     if (str && *str)
3012         req->packageName = odr_strdup(p->odr_out, str);
3013
3014     str = ZOOM_options_get(p->options, "user-id");
3015     if (str)
3016         req->userId = odr_strdup(p->odr_out, str);
3017
3018     req->packageType = odr_oiddup(p->odr_out, oid);
3019
3020     str = ZOOM_options_get(p->options, "function");
3021     if (str)
3022     {
3023         if (!strcmp (str, "create"))
3024             *req->function = Z_ExtendedServicesRequest_create;
3025         if (!strcmp (str, "delete"))
3026             *req->function = Z_ExtendedServicesRequest_delete;
3027         if (!strcmp (str, "modify"))
3028             *req->function = Z_ExtendedServicesRequest_modify;
3029     }
3030
3031     str = ZOOM_options_get(p->options, "waitAction");
3032     if (str)
3033     {
3034         if (!strcmp (str, "wait"))
3035             *req->waitAction = Z_ExtendedServicesRequest_wait;
3036         if (!strcmp (str, "waitIfPossible"))
3037             *req->waitAction = Z_ExtendedServicesRequest_waitIfPossible;
3038         if (!strcmp (str, "dontWait"))
3039             *req->waitAction = Z_ExtendedServicesRequest_dontWait;
3040         if (!strcmp (str, "dontReturnPackage"))
3041             *req->waitAction = Z_ExtendedServicesRequest_dontReturnPackage;
3042     }
3043     return apdu;
3044 }
3045
3046 static const char *ill_array_lookup(void *clientData, const char *idx)
3047 {
3048     ZOOM_package p = (ZOOM_package) clientData;
3049     return ZOOM_options_get(p->options, idx+4);
3050 }
3051
3052 static Z_External *encode_ill_request(ZOOM_package p)
3053 {
3054     ODR out = p->odr_out;
3055     ILL_Request *req;
3056     Z_External *r = 0;
3057     struct ill_get_ctl ctl;
3058
3059     ctl.odr = p->odr_out;
3060     ctl.clientData = p;
3061     ctl.f = ill_array_lookup;
3062
3063     req = ill_get_ILLRequest(&ctl, "ill", 0);
3064
3065     if (!ill_Request(out, &req, 0, 0))
3066     {
3067         int ill_request_size;
3068         char *ill_request_buf = odr_getbuf(out, &ill_request_size, 0);
3069         if (ill_request_buf)
3070             odr_setbuf(out, ill_request_buf, ill_request_size, 1);
3071         return 0;
3072     }
3073     else
3074     {
3075         int illRequest_size = 0;
3076         char *illRequest_buf = odr_getbuf(out, &illRequest_size, 0);
3077
3078         r = (Z_External *) odr_malloc(out, sizeof(*r));
3079         r->direct_reference = odr_oiddup(out, yaz_oid_general_isoill_1);
3080         r->indirect_reference = 0;
3081         r->descriptor = 0;
3082         r->which = Z_External_single;
3083
3084         r->u.single_ASN1_type =
3085             odr_create_Odr_oct(out,
3086                                (unsigned char *)illRequest_buf,
3087                                illRequest_size);
3088     }
3089     return r;
3090 }
3091
3092 static Z_ItemOrder *encode_item_order(ZOOM_package p)
3093 {
3094     Z_ItemOrder *req = (Z_ItemOrder *) odr_malloc(p->odr_out, sizeof(*req));
3095     const char *str;
3096
3097     req->which = Z_IOItemOrder_esRequest;
3098     req->u.esRequest = (Z_IORequest *)
3099         odr_malloc(p->odr_out,sizeof(Z_IORequest));
3100
3101     /* to keep part ... */
3102     req->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
3103         odr_malloc(p->odr_out,sizeof(Z_IOOriginPartToKeep));
3104     req->u.esRequest->toKeep->supplDescription = 0;
3105     req->u.esRequest->toKeep->contact = (Z_IOContact *)
3106         odr_malloc(p->odr_out, sizeof(*req->u.esRequest->toKeep->contact));
3107
3108     str = ZOOM_options_get(p->options, "contact-name");
3109     req->u.esRequest->toKeep->contact->name = str ?
3110         odr_strdup(p->odr_out, str) : 0;
3111
3112     str = ZOOM_options_get(p->options, "contact-phone");
3113     req->u.esRequest->toKeep->contact->phone = str ?
3114         odr_strdup(p->odr_out, str) : 0;
3115
3116     str = ZOOM_options_get(p->options, "contact-email");
3117     req->u.esRequest->toKeep->contact->email = str ?
3118         odr_strdup(p->odr_out, str) : 0;
3119
3120     req->u.esRequest->toKeep->addlBilling = 0;
3121
3122     /* not to keep part ... */
3123     req->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
3124         odr_malloc(p->odr_out,sizeof(Z_IOOriginPartNotToKeep));
3125
3126     str = ZOOM_options_get(p->options, "itemorder-setname");
3127     if (!str)
3128         str = "default";
3129
3130     if (!*str)
3131         req->u.esRequest->notToKeep->resultSetItem = 0;
3132     else
3133     {
3134         req->u.esRequest->notToKeep->resultSetItem = (Z_IOResultSetItem *)
3135             odr_malloc(p->odr_out, sizeof(Z_IOResultSetItem));
3136
3137         req->u.esRequest->notToKeep->resultSetItem->resultSetId =
3138             odr_strdup(p->odr_out, str);
3139         req->u.esRequest->notToKeep->resultSetItem->item =
3140             (int *) odr_malloc(p->odr_out, sizeof(int));
3141
3142         str = ZOOM_options_get(p->options, "itemorder-item");
3143         *req->u.esRequest->notToKeep->resultSetItem->item =
3144             (str ? atoi(str) : 1);
3145     }
3146
3147     str = ZOOM_options_get(p->options, "doc");
3148     if (str)
3149     {
3150         req->u.esRequest->notToKeep->itemRequest =
3151             z_ext_record_xml(p->odr_out, str, strlen(str));
3152     }
3153     else
3154         req->u.esRequest->notToKeep->itemRequest = encode_ill_request(p);
3155
3156     return req;
3157 }
3158
3159 Z_APDU *create_admin_package(ZOOM_package p, int type,
3160                              Z_ESAdminOriginPartToKeep **toKeepP,
3161                              Z_ESAdminOriginPartNotToKeep **notToKeepP)
3162 {
3163     Z_APDU *apdu = create_es_package(p, yaz_oid_extserv_admin);
3164     if (apdu)
3165     {
3166         Z_ESAdminOriginPartToKeep  *toKeep;
3167         Z_ESAdminOriginPartNotToKeep  *notToKeep;
3168         Z_External *r = (Z_External *) odr_malloc(p->odr_out, sizeof(*r));
3169         const char *first_db = "Default";
3170         int num_db;
3171         char **db = set_DatabaseNames(p->connection, p->options, &num_db,
3172                                       p->odr_out);
3173         if (num_db > 0)
3174             first_db = db[0];
3175
3176         r->direct_reference = odr_oiddup(p->odr_out, yaz_oid_extserv_admin);
3177         r->descriptor = 0;
3178         r->indirect_reference = 0;
3179         r->which = Z_External_ESAdmin;
3180
3181         r->u.adminService = (Z_Admin *)
3182             odr_malloc(p->odr_out, sizeof(*r->u.adminService));
3183         r->u.adminService->which = Z_Admin_esRequest;
3184         r->u.adminService->u.esRequest = (Z_AdminEsRequest *)
3185             odr_malloc(p->odr_out, sizeof(*r->u.adminService->u.esRequest));
3186
3187         toKeep = r->u.adminService->u.esRequest->toKeep =
3188             (Z_ESAdminOriginPartToKeep *)
3189             odr_malloc(p->odr_out, sizeof(*r->u.adminService->u.esRequest->toKeep));
3190         toKeep->which = type;
3191         toKeep->databaseName = odr_strdup(p->odr_out, first_db);
3192         toKeep->u.create = odr_nullval();
3193         apdu->u.extendedServicesRequest->taskSpecificParameters = r;
3194
3195         r->u.adminService->u.esRequest->notToKeep = notToKeep =
3196             (Z_ESAdminOriginPartNotToKeep *)
3197             odr_malloc(p->odr_out,
3198                        sizeof(*r->u.adminService->u.esRequest->notToKeep));
3199         notToKeep->which = Z_ESAdminOriginPartNotToKeep_recordsWillFollow;
3200         notToKeep->u.recordsWillFollow = odr_nullval();
3201         if (toKeepP)
3202             *toKeepP = toKeep;
3203         if (notToKeepP)
3204             *notToKeepP = notToKeep;
3205     }
3206     return apdu;
3207 }
3208
3209 static Z_APDU *create_xmlupdate_package(ZOOM_package p)
3210 {
3211     Z_APDU *apdu = create_es_package(p, yaz_oid_extserv_xml_es);
3212     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
3213     Z_External *ext = (Z_External *) odr_malloc(p->odr_out, sizeof(*ext));
3214     const char *doc = ZOOM_options_get(p->options, "doc");
3215
3216     if (!doc)
3217         doc = "";
3218
3219     req->taskSpecificParameters = ext;
3220     ext->direct_reference = req->packageType;
3221     ext->descriptor = 0;
3222     ext->indirect_reference = 0;
3223
3224     ext->which = Z_External_octet;
3225     ext->u.single_ASN1_type =
3226         odr_create_Odr_oct(p->odr_out, (const unsigned char *) doc,
3227                            strlen(doc));
3228     return apdu;
3229 }
3230
3231 static Z_APDU *create_update_package(ZOOM_package p)
3232 {
3233     Z_APDU *apdu = 0;
3234     const char *first_db = "Default";
3235     int num_db;
3236     char **db = set_DatabaseNames(p->connection, p->options, &num_db, p->odr_out);
3237     const char *action = ZOOM_options_get(p->options, "action");
3238     const char *recordIdOpaque = ZOOM_options_get(p->options, "recordIdOpaque");
3239     const char *recordIdNumber = ZOOM_options_get(p->options, "recordIdNumber");
3240     const char *record_buf = ZOOM_options_get(p->options, "record");
3241     const char *syntax_str = ZOOM_options_get(p->options, "syntax");
3242     const char *version = ZOOM_options_get(p->options, "updateVersion");
3243
3244     const char *correlationInfo_note =
3245         ZOOM_options_get(p->options, "correlationInfo.note");
3246     const char *correlationInfo_id =
3247         ZOOM_options_get(p->options, "correlationInfo.id");
3248     int action_no = -1;
3249     Odr_oid *syntax_oid = 0;
3250     const Odr_oid *package_oid = yaz_oid_extserv_database_update;
3251
3252     if (!version)
3253         version = "3";
3254     if (!syntax_str)
3255         syntax_str = "xml";
3256     if (!record_buf)
3257     {
3258         record_buf = "void";
3259         syntax_str = "SUTRS";
3260     }
3261
3262     if (syntax_str)
3263     {
3264         syntax_oid = yaz_string_to_oid_odr(yaz_oid_std(),
3265                                            CLASS_RECSYN, syntax_str,
3266                                            p->odr_out);
3267     }
3268     if (!syntax_oid)
3269         return 0;
3270
3271     if (num_db > 0)
3272         first_db = db[0];
3273
3274     switch(*version)
3275     {
3276     case '1':
3277         package_oid = yaz_oid_extserv_database_update_first_version;
3278         /* old update does not support specialUpdate */
3279         if (!action)
3280             action = "recordInsert";
3281         break;
3282     case '2':
3283         if (!action)
3284             action = "specialUpdate";
3285         package_oid = yaz_oid_extserv_database_update_second_version;
3286         break;
3287     case '3':
3288         if (!action)
3289             action = "specialUpdate";
3290         package_oid = yaz_oid_extserv_database_update;
3291         break;
3292     default:
3293         return 0;
3294     }
3295
3296     if (!strcmp(action, "recordInsert"))
3297         action_no = Z_IUOriginPartToKeep_recordInsert;
3298     else if (!strcmp(action, "recordReplace"))
3299         action_no = Z_IUOriginPartToKeep_recordReplace;
3300     else if (!strcmp(action, "recordDelete"))
3301         action_no = Z_IUOriginPartToKeep_recordDelete;
3302     else if (!strcmp(action, "elementUpdate"))
3303         action_no = Z_IUOriginPartToKeep_elementUpdate;
3304     else if (!strcmp(action, "specialUpdate"))
3305         action_no = Z_IUOriginPartToKeep_specialUpdate;
3306     else
3307         return 0;
3308
3309     apdu = create_es_package(p, package_oid);
3310     if (apdu)
3311     {
3312         Z_IUOriginPartToKeep *toKeep;
3313         Z_IUSuppliedRecords *notToKeep;
3314         Z_External *r = (Z_External *)
3315             odr_malloc(p->odr_out, sizeof(*r));
3316         const char *elementSetName =
3317             ZOOM_options_get(p->options, "elementSetName");
3318
3319         apdu->u.extendedServicesRequest->taskSpecificParameters = r;
3320
3321         r->direct_reference = odr_oiddup(p->odr_out, package_oid);
3322         r->descriptor = 0;
3323         r->which = Z_External_update;
3324         r->indirect_reference = 0;
3325         r->u.update = (Z_IUUpdate *)
3326             odr_malloc(p->odr_out, sizeof(*r->u.update));
3327
3328         r->u.update->which = Z_IUUpdate_esRequest;
3329         r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
3330             odr_malloc(p->odr_out, sizeof(*r->u.update->u.esRequest));
3331         toKeep = r->u.update->u.esRequest->toKeep =
3332             (Z_IUOriginPartToKeep *)
3333             odr_malloc(p->odr_out, sizeof(*toKeep));
3334
3335         toKeep->databaseName = odr_strdup(p->odr_out, first_db);
3336         toKeep->schema = 0;
3337
3338         toKeep->elementSetName = 0;
3339         if (elementSetName)
3340             toKeep->elementSetName = odr_strdup(p->odr_out, elementSetName);
3341
3342         toKeep->actionQualifier = 0;
3343         toKeep->action = odr_intdup(p->odr_out, action_no);
3344
3345         notToKeep = r->u.update->u.esRequest->notToKeep =
3346             (Z_IUSuppliedRecords *)
3347             odr_malloc(p->odr_out, sizeof(*notToKeep));
3348         notToKeep->num = 1;
3349         notToKeep->elements = (Z_IUSuppliedRecords_elem **)
3350             odr_malloc(p->odr_out, sizeof(*notToKeep->elements));
3351         notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
3352             odr_malloc(p->odr_out, sizeof(**notToKeep->elements));
3353         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
3354         if (recordIdOpaque)
3355         {
3356             notToKeep->elements[0]->u.opaque =
3357                 odr_create_Odr_oct(p->odr_out,
3358                                    (const unsigned char *) recordIdOpaque,
3359                                    strlen(recordIdOpaque));
3360         }
3361         else if (recordIdNumber)
3362         {
3363             notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_number;
3364
3365             notToKeep->elements[0]->u.number =
3366                 odr_intdup(p->odr_out, atoi(recordIdNumber));
3367         }
3368         else
3369             notToKeep->elements[0]->u.opaque = 0;
3370         notToKeep->elements[0]->supplementalId = 0;
3371         if (correlationInfo_note || correlationInfo_id)
3372         {
3373             Z_IUCorrelationInfo *ci;
3374             ci = notToKeep->elements[0]->correlationInfo =
3375                 odr_malloc(p->odr_out, sizeof(*ci));
3376             ci->note = correlationInfo_note ?
3377                 odr_strdup(p->odr_out, correlationInfo_note) : 0;
3378             ci->id = correlationInfo_id ?
3379                 odr_intdup(p->odr_out, atoi(correlationInfo_id)) : 0;
3380         }
3381         else
3382             notToKeep->elements[0]->correlationInfo = 0;
3383         notToKeep->elements[0]->record =
3384             z_ext_record_oid(p->odr_out, syntax_oid,
3385                              record_buf, strlen(record_buf));
3386     }
3387     if (0 && apdu)
3388     {
3389         ODR print = odr_createmem(ODR_PRINT);
3390
3391         z_APDU(print, &apdu, 0, 0);
3392         odr_destroy(print);
3393     }
3394     return apdu;
3395 }
3396
3397 ZOOM_API(void)
3398     ZOOM_package_send(ZOOM_package p, const char *type)
3399 {
3400     Z_APDU *apdu = 0;
3401     ZOOM_connection c;
3402     if (!p)
3403         return;
3404     c = p->connection;
3405     odr_reset(p->odr_out);
3406     xfree(p->buf_out);
3407     p->buf_out = 0;
3408     if (!strcmp(type, "itemorder"))
3409     {
3410         apdu = create_es_package(p, yaz_oid_extserv_item_order);
3411         if (apdu)
3412         {
3413             Z_External *r = (Z_External *) odr_malloc(p->odr_out, sizeof(*r));
3414
3415             r->direct_reference =
3416                 odr_oiddup(p->odr_out, yaz_oid_extserv_item_order);
3417             r->descriptor = 0;
3418             r->which = Z_External_itemOrder;
3419             r->indirect_reference = 0;
3420             r->u.itemOrder = encode_item_order(p);
3421
3422             apdu->u.extendedServicesRequest->taskSpecificParameters = r;
3423         }
3424     }
3425     else if (!strcmp(type, "create"))  /* create database */
3426     {
3427         apdu = create_admin_package(p, Z_ESAdminOriginPartToKeep_create,
3428                                     0, 0);
3429     }
3430     else if (!strcmp(type, "drop"))  /* drop database */
3431     {
3432         apdu = create_admin_package(p, Z_ESAdminOriginPartToKeep_drop,
3433                                     0, 0);
3434     }
3435     else if (!strcmp(type, "commit"))  /* commit changes */
3436     {
3437         apdu = create_admin_package(p, Z_ESAdminOriginPartToKeep_commit,
3438                                     0, 0);
3439     }
3440     else if (!strcmp(type, "update")) /* update record(s) */
3441     {
3442         apdu = create_update_package(p);
3443     }
3444     else if (!strcmp(type, "xmlupdate"))
3445     {
3446         apdu = create_xmlupdate_package(p);
3447     }
3448     if (apdu)
3449     {
3450         if (encode_APDU(p->connection, apdu, p->odr_out) == 0)
3451         {
3452             char *buf;
3453
3454             ZOOM_task task = ZOOM_connection_add_task(c, ZOOM_TASK_PACKAGE);
3455             task->u.package = p;
3456             buf = odr_getbuf(p->odr_out, &p->len_out, 0);
3457             p->buf_out = (char *) xmalloc(p->len_out);
3458             memcpy(p->buf_out, buf, p->len_out);
3459
3460             (p->refcount)++;
3461             if (!c->async)
3462             {
3463                 while (ZOOM_event(1, &c))
3464                     ;
3465             }
3466         }
3467     }
3468 }
3469
3470 ZOOM_API(ZOOM_package)
3471     ZOOM_connection_package(ZOOM_connection c, ZOOM_options options)
3472 {
3473     ZOOM_package p = (ZOOM_package) xmalloc(sizeof(*p));
3474
3475     p->connection = c;
3476     p->odr_out = odr_createmem(ODR_ENCODE);
3477     p->options = ZOOM_options_create_with_parent2(options, c->options);
3478     p->refcount = 1;
3479     p->buf_out = 0;
3480     p->len_out = 0;
3481     return p;
3482 }
3483
3484 ZOOM_API(void)
3485     ZOOM_package_destroy(ZOOM_package p)
3486 {
3487     if (!p)
3488         return;
3489     (p->refcount)--;
3490     if (p->refcount == 0)
3491     {
3492         odr_destroy(p->odr_out);
3493         xfree(p->buf_out);
3494
3495         ZOOM_options_destroy(p->options);
3496         xfree(p);
3497     }
3498 }
3499
3500 ZOOM_API(const char *)
3501     ZOOM_package_option_get(ZOOM_package p, const char *key)
3502 {
3503     return ZOOM_options_get(p->options, key);
3504 }
3505
3506
3507 ZOOM_API(void)
3508     ZOOM_package_option_set(ZOOM_package p, const char *key,
3509                             const char *val)
3510 {
3511     ZOOM_options_set(p->options, key, val);
3512 }
3513
3514 static int ZOOM_connection_exec_task(ZOOM_connection c)
3515 {
3516     ZOOM_task task = c->tasks;
3517     zoom_ret ret = zoom_complete;
3518
3519     if (!task)
3520         return 0;
3521     yaz_log(log_details, "%p ZOOM_connection_exec_task type=%d run=%d",
3522             c, task->which, task->running);
3523     if (c->error != ZOOM_ERROR_NONE)
3524     {
3525         yaz_log(log_details, "%p ZOOM_connection_exec_task "
3526                 "removing tasks because of error = %d", c, c->error);
3527         ZOOM_connection_remove_tasks(c);
3528         return 0;
3529     }
3530     if (task->running)
3531     {
3532         yaz_log(log_details, "%p ZOOM_connection_exec_task "
3533                 "task already running", c);
3534         return 0;
3535     }
3536     task->running = 1;
3537     ret = zoom_complete;
3538     if (c->cs || task->which == ZOOM_TASK_CONNECT)
3539     {
3540         switch (task->which)
3541         {
3542         case ZOOM_TASK_SEARCH:
3543             if (c->proto == PROTO_HTTP)
3544                 ret = ZOOM_connection_srw_send_search(c);
3545             else
3546                 ret = ZOOM_connection_send_search(c);
3547             break;
3548         case ZOOM_TASK_RETRIEVE:
3549             if (c->proto == PROTO_HTTP)
3550                 ret = ZOOM_connection_srw_send_search(c);
3551             else
3552                 ret = send_present(c);
3553             break;
3554         case ZOOM_TASK_CONNECT:
3555             ret = do_connect(c);
3556             break;
3557         case ZOOM_TASK_SCAN:
3558             if (c->proto == PROTO_HTTP)
3559                 ret = ZOOM_connection_srw_send_scan(c);
3560             else
3561                 ret = ZOOM_connection_send_scan(c);
3562             break;
3563         case ZOOM_TASK_PACKAGE:
3564             ret = send_package(c);
3565             break;
3566         case ZOOM_TASK_SORT:
3567             c->tasks->u.sort.resultset->r_sort_spec =
3568                 c->tasks->u.sort.q->sort_spec;
3569             ret = send_sort(c, c->tasks->u.sort.resultset);
3570             break;
3571         }
3572     }
3573     else
3574     {
3575         yaz_log(log_details, "%p ZOOM_connection_exec_task "
3576                 "remove tasks because no connection exist", c);
3577         ZOOM_connection_remove_tasks(c);
3578     }
3579     if (ret == zoom_complete)
3580     {
3581         yaz_log(log_details, "%p ZOOM_connection_exec_task "
3582                 "task removed (complete)", c);
3583         ZOOM_connection_remove_task(c);
3584         return 0;
3585     }
3586     yaz_log(log_details, "%p ZOOM_connection_exec_task "
3587             "task pending", c);
3588     return 1;
3589 }
3590
3591 static zoom_ret send_sort_present(ZOOM_connection c)
3592 {
3593     zoom_ret r = zoom_complete;
3594
3595     if (c->tasks && c->tasks->which == ZOOM_TASK_SEARCH)
3596         r = send_sort(c, c->tasks->u.search.resultset);
3597     if (r == zoom_complete)
3598         r = send_present(c);
3599     return r;
3600 }
3601
3602 static int es_response(ZOOM_connection c,
3603                        Z_ExtendedServicesResponse *res)
3604 {
3605     if (!c->tasks || c->tasks->which != ZOOM_TASK_PACKAGE)
3606         return 0;
3607     if (res->diagnostics && res->num_diagnostics > 0)
3608         response_diag(c, res->diagnostics[0]);
3609     if (res->taskPackage &&
3610         res->taskPackage->which == Z_External_extendedService)
3611     {
3612         Z_TaskPackage *taskPackage = res->taskPackage->u.extendedService;
3613         Odr_oct *id = taskPackage->targetReference;
3614
3615         if (id)
3616             ZOOM_options_setl(c->tasks->u.package->options,
3617                               "targetReference", (char*) id->buf, id->len);
3618     }
3619     if (res->taskPackage &&
3620         res->taskPackage->which == Z_External_octet)
3621     {
3622         Odr_oct *doc = res->taskPackage->u.octet_aligned;
3623         ZOOM_options_setl(c->tasks->u.package->options,
3624                           "xmlUpdateDoc", (char*) doc->buf, doc->len);
3625     }
3626     return 1;
3627 }
3628
3629 static void interpret_init_diag(ZOOM_connection c,
3630                                 Z_DiagnosticFormat *diag)
3631 {
3632     if (diag->num > 0)
3633     {
3634         Z_DiagnosticFormat_s *ds = diag->elements[0];
3635         if (ds->which == Z_DiagnosticFormat_s_defaultDiagRec)
3636             response_default_diag(c, ds->u.defaultDiagRec);
3637     }
3638 }
3639
3640
3641 static void interpret_otherinformation_field(ZOOM_connection c,
3642                                              Z_OtherInformation *ui)
3643 {
3644     int i;
3645     for (i = 0; i < ui->num_elements; i++)
3646     {
3647         Z_OtherInformationUnit *unit = ui->list[i];
3648         if (unit->which == Z_OtherInfo_externallyDefinedInfo &&
3649             unit->information.externallyDefinedInfo &&
3650             unit->information.externallyDefinedInfo->which ==
3651             Z_External_diag1)
3652         {
3653             interpret_init_diag(c, unit->information.externallyDefinedInfo->u.diag1);
3654         }
3655     }
3656 }
3657
3658
3659 static void set_init_option(const char *name, void *clientData) {
3660     ZOOM_connection c = (ZOOM_connection) clientData;
3661     char buf[80];
3662
3663     sprintf(buf, "init_opt_%.70s", name);
3664     ZOOM_connection_option_set(c, buf, "1");
3665 }
3666
3667
3668 static void recv_apdu(ZOOM_connection c, Z_APDU *apdu)
3669 {
3670     Z_InitResponse *initrs;
3671
3672     ZOOM_connection_set_mask(c, 0);
3673     yaz_log(log_details, "%p recv_apdu apdu->which=%d", c, apdu->which);
3674     switch(apdu->which)
3675     {
3676     case Z_APDU_initResponse:
3677         yaz_log(log_api, "%p recv_apdu: Received Init response", c);
3678         initrs = apdu->u.initResponse;
3679         ZOOM_connection_option_set(c, "serverImplementationId",
3680                                    initrs->implementationId ?
3681                                    initrs->implementationId : "");
3682         ZOOM_connection_option_set(c, "serverImplementationName",
3683                                    initrs->implementationName ?
3684                                    initrs->implementationName : "");
3685         ZOOM_connection_option_set(c, "serverImplementationVersion",
3686                                    initrs->implementationVersion ?
3687                                    initrs->implementationVersion : "");
3688         /* Set the three old options too, for old applications */
3689         ZOOM_connection_option_set(c, "targetImplementationId",
3690                                    initrs->implementationId ?
3691                                    initrs->implementationId : "");
3692         ZOOM_connection_option_set(c, "targetImplementationName",
3693                                    initrs->implementationName ?
3694                                    initrs->implementationName : "");
3695         ZOOM_connection_option_set(c, "targetImplementationVersion",
3696                                    initrs->implementationVersion ?
3697                                    initrs->implementationVersion : "");
3698
3699         /* Make initrs->options available as ZOOM-level options */
3700         yaz_init_opt_decode(initrs->options, set_init_option, (void*) c);
3701
3702         if (!*initrs->result)
3703         {
3704             Z_External *uif = initrs->userInformationField;
3705
3706             set_ZOOM_error(c, ZOOM_ERROR_INIT, 0); /* default error */
3707
3708             if (uif && uif->which == Z_External_userInfo1)
3709                 interpret_otherinformation_field(c, uif->u.userInfo1);
3710         }
3711         else
3712         {
3713             char *cookie =
3714                 yaz_oi_get_string_oid(&apdu->u.initResponse->otherInfo,
3715                                       yaz_oid_userinfo_cookie, 1, 0);
3716             xfree(c->cookie_in);
3717             c->cookie_in = 0;
3718             if (cookie)
3719                 c->cookie_in = xstrdup(cookie);
3720             if (ODR_MASK_GET(initrs->options, Z_Options_namedResultSets) &&
3721                 ODR_MASK_GET(initrs->protocolVersion, Z_ProtocolVersion_3))
3722                 c->support_named_resultsets = 1;
3723             if (c->tasks)
3724             {
3725                 assert(c->tasks->which == ZOOM_TASK_CONNECT);
3726                 ZOOM_connection_remove_task(c);
3727             }
3728             ZOOM_connection_exec_task(c);
3729         }
3730         if (ODR_MASK_GET(initrs->options, Z_Options_negotiationModel))
3731         {
3732             NMEM tmpmem = nmem_create();
3733             Z_CharSetandLanguageNegotiation *p =
3734                 yaz_get_charneg_record(initrs->otherInfo);
3735
3736             if (p)
3737             {
3738                 char *charset = NULL, *lang = NULL;
3739                 int sel;
3740
3741                 yaz_get_response_charneg(tmpmem, p, &charset, &lang, &sel);
3742                 yaz_log(log_details, "%p recv_apdu target accepted: "
3743                         "charset %s, language %s, select %d",
3744                         c,
3745                         charset ? charset : "none", lang ? lang : "none", sel);
3746                 if (charset)
3747                     ZOOM_connection_option_set(c, "negotiation-charset",
3748                                                charset);
3749                 if (lang)
3750                     ZOOM_connection_option_set(c, "negotiation-lang",
3751                                                lang);
3752
3753                 ZOOM_connection_option_set(
3754                     c,  "negotiation-charset-in-effect-for-records",
3755                     (sel != 0) ? "1" : "0");
3756                 nmem_destroy(tmpmem);
3757             }
3758         }
3759         break;
3760     case Z_APDU_searchResponse:
3761         yaz_log(log_api, "%p recv_apdu Search response", c);
3762         handle_search_response(c, apdu->u.searchResponse);
3763         if (send_sort_present(c) == zoom_complete)
3764             ZOOM_connection_remove_task(c);
3765         break;
3766     case Z_APDU_presentResponse:
3767         yaz_log(log_api, "%p recv_apdu Present response", c);
3768         handle_present_response(c, apdu->u.presentResponse);
3769         if (send_present(c) == zoom_complete)
3770             ZOOM_connection_remove_task(c);
3771         break;
3772     case Z_APDU_sortResponse:
3773         yaz_log(log_api, "%p recv_apdu Sort response", c);
3774         sort_response(c, apdu->u.sortResponse);
3775         if (send_present(c) == zoom_complete)
3776             ZOOM_connection_remove_task(c);
3777         break;
3778     case Z_APDU_scanResponse:
3779         yaz_log(log_api, "%p recv_apdu Scan response", c);
3780         scan_response(c, apdu->u.scanResponse);
3781         ZOOM_connection_remove_task(c);
3782         break;
3783     case Z_APDU_extendedServicesResponse:
3784         yaz_log(log_api, "%p recv_apdu Extended Services response", c);
3785         es_response(c, apdu->u.extendedServicesResponse);
3786         ZOOM_connection_remove_task(c);
3787         break;
3788     case Z_APDU_close:
3789         yaz_log(log_api, "%p recv_apdu Close PDU", c);
3790         if (!ZOOM_test_reconnect(c))
3791         {
3792             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, c->host_port);
3793             do_close(c);
3794         }
3795         break;
3796     default:
3797         yaz_log(log_api, "%p Received unknown PDU", c);
3798         set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
3799         do_close(c);
3800     }
3801 }
3802
3803 #if YAZ_HAVE_XML2
3804 static void handle_srw_response(ZOOM_connection c,
3805                                 Z_SRW_searchRetrieveResponse *res)
3806 {
3807     ZOOM_resultset resultset = 0;
3808     int i;
3809     NMEM nmem;
3810     ZOOM_Event event;
3811     int *start;
3812     const char *syntax, *elementSetName;
3813
3814     if (!c->tasks)
3815         return;
3816
3817     switch(c->tasks->which)
3818     {
3819     case ZOOM_TASK_SEARCH:
3820         resultset = c->tasks->u.search.resultset;
3821         start = &c->tasks->u.search.start;
3822         syntax = c->tasks->u.search.syntax;
3823         elementSetName = c->tasks->u.search.elementSetName;
3824         break;
3825     case ZOOM_TASK_RETRIEVE:
3826         resultset = c->tasks->u.retrieve.resultset;
3827         start = &c->tasks->u.retrieve.start;
3828         syntax = c->tasks->u.retrieve.syntax;
3829         elementSetName = c->tasks->u.retrieve.elementSetName;
3830         break;
3831     default:
3832         return;
3833     }
3834     event = ZOOM_Event_create(ZOOM_EVENT_RECV_SEARCH);
3835     ZOOM_connection_put_event(c, event);
3836
3837     resultset->size = 0;
3838
3839     if (res->resultSetId)
3840         ZOOM_resultset_option_set(resultset, "resultSetId", res->resultSetId);
3841
3842     yaz_log(log_details, "%p handle_srw_response got SRW response OK", c);
3843
3844     if (res->numberOfRecords)
3845         resultset->size = *res->numberOfRecords;
3846
3847     for (i = 0; i<res->num_records; i++)
3848     {
3849         int pos;
3850
3851         Z_NamePlusRecord *npr = (Z_NamePlusRecord *)
3852             odr_malloc(c->odr_in, sizeof(Z_NamePlusRecord));
3853
3854         if (res->records[i].recordPosition &&
3855             *res->records[i].recordPosition > 0)
3856             pos = *res->records[i].recordPosition - 1;
3857         else
3858             pos = *start + i;
3859
3860         npr->databaseName = 0;
3861         npr->which = Z_NamePlusRecord_databaseRecord;
3862         npr->u.databaseRecord = (Z_External *)
3863             odr_malloc(c->odr_in, sizeof(Z_External));
3864         npr->u.databaseRecord->descriptor = 0;
3865         npr->u.databaseRecord->direct_reference =
3866             odr_oiddup(c->odr_in, yaz_oid_recsyn_xml);
3867         npr->u.databaseRecord->which = Z_External_octet;
3868
3869         npr->u.databaseRecord->u.octet_aligned = (Odr_oct *)
3870             odr_malloc(c->odr_in, sizeof(Odr_oct));
3871         npr->u.databaseRecord->u.octet_aligned->buf = (unsigned char*)
3872             res->records[i].recordData_buf;
3873         npr->u.databaseRecord->u.octet_aligned->len =
3874             npr->u.databaseRecord->u.octet_aligned->size =
3875             res->records[i].recordData_len;
3876         record_cache_add(resultset, npr, pos, syntax, elementSetName);
3877     }
3878     if (res->num_diagnostics > 0)
3879         set_SRU_error(c, &res->diagnostics[0]);
3880     nmem = odr_extract_mem(c->odr_in);
3881     nmem_transfer(odr_getmem(resultset->odr), nmem);
3882     nmem_destroy(nmem);
3883 }
3884 #endif
3885
3886 #if YAZ_HAVE_XML2
3887 static void handle_srw_scan_response(ZOOM_connection c,
3888                                      Z_SRW_scanResponse *res)
3889 {
3890     NMEM nmem = odr_extract_mem(c->odr_in);
3891     ZOOM_scanset scan;
3892
3893     if (!c->tasks || c->tasks->which != ZOOM_TASK_SCAN)
3894         return;
3895     scan = c->tasks->u.scan.scan;
3896
3897     if (res->num_diagnostics > 0)
3898         set_SRU_error(c, &res->diagnostics[0]);
3899
3900     scan->scan_response = 0;
3901     scan->srw_scan_response = res;
3902     nmem_transfer(odr_getmem(scan->odr), nmem);
3903
3904     ZOOM_options_set_int(scan->options, "number", res->num_terms);
3905     nmem_destroy(nmem);
3906 }
3907 #endif
3908
3909 #if YAZ_HAVE_XML2
3910 static void handle_http(ZOOM_connection c, Z_HTTP_Response *hres)
3911 {
3912     int ret = -1;
3913     const char *addinfo = 0;
3914     const char *connection_head = z_HTTP_header_lookup(hres->headers,
3915                                                        "Connection");
3916     ZOOM_connection_set_mask(c, 0);
3917     yaz_log(log_details, "%p handle_http", c);
3918
3919     if (!yaz_srw_check_content_type(hres))
3920         addinfo = "content-type";
3921     else
3922     {
3923         Z_SOAP *soap_package = 0;
3924         ODR o = c->odr_in;
3925         Z_SOAP_Handler soap_handlers[2] = {
3926             {YAZ_XMLNS_SRU_v1_1, 0, (Z_SOAP_fun) yaz_srw_codec},
3927             {0, 0, 0}
3928         };
3929         ret = z_soap_codec(o, &soap_package,
3930                            &hres->content_buf, &hres->content_len,
3931                            soap_handlers);
3932         if (!ret && soap_package->which == Z_SOAP_generic &&
3933             soap_package->u.generic->no == 0)
3934         {
3935             Z_SRW_PDU *sr = (Z_SRW_PDU*) soap_package->u.generic->p;
3936
3937             ZOOM_options_set(c->options, "sru_version", sr->srw_version);
3938             if (sr->which == Z_SRW_searchRetrieve_response)
3939                 handle_srw_response(c, sr->u.response);
3940             else if (sr->which == Z_SRW_scan_response)
3941                 handle_srw_scan_response(c, sr->u.scan_response);
3942             else
3943                 ret = -1;
3944         }
3945         else if (!ret && (soap_package->which == Z_SOAP_fault
3946                           || soap_package->which == Z_SOAP_error))
3947         {
3948             set_HTTP_error(c, hres->code,
3949                            soap_package->u.fault->fault_code,
3950                            soap_package->u.fault->fault_string);
3951         }
3952         else
3953             ret = -1;
3954     }
3955     if (ret)
3956     {
3957         if (hres->code != 200)
3958             set_HTTP_error(c, hres->code, 0, 0);
3959         else
3960             set_ZOOM_error(c, ZOOM_ERROR_DECODE, addinfo);
3961         do_close(c);
3962     }
3963     ZOOM_connection_remove_task(c);
3964     if (!strcmp(hres->version, "1.0"))
3965     {
3966         /* HTTP 1.0: only if Keep-Alive we stay alive.. */
3967         if (!connection_head || strcmp(connection_head, "Keep-Alive"))
3968             do_close(c);
3969     }
3970     else
3971     {
3972         /* HTTP 1.1: only if no close we stay alive .. */
3973         if (connection_head && !strcmp(connection_head, "close"))
3974             do_close(c);
3975     }
3976 }
3977 #endif
3978
3979 static int do_read(ZOOM_connection c)
3980 {
3981     int r, more;
3982     ZOOM_Event event;
3983
3984     event = ZOOM_Event_create(ZOOM_EVENT_RECV_DATA);
3985     ZOOM_connection_put_event(c, event);
3986
3987     r = cs_get(c->cs, &c->buf_in, &c->len_in);
3988     more = cs_more(c->cs);
3989     yaz_log(log_details, "%p do_read len=%d more=%d", c, r, more);
3990     if (r == 1)
3991         return 0;
3992     if (r <= 0)
3993     {
3994         if (!ZOOM_test_reconnect(c))
3995         {
3996             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, c->host_port);
3997             do_close(c);
3998         }
3999     }
4000     else
4001     {
4002         Z_GDU *gdu;
4003         ZOOM_Event event;
4004
4005         odr_reset(c->odr_in);
4006         odr_setbuf(c->odr_in, c->buf_in, r, 0);
4007         event = ZOOM_Event_create(ZOOM_EVENT_RECV_APDU);
4008         ZOOM_connection_put_event(c, event);
4009
4010         if (!z_GDU(c->odr_in, &gdu, 0, 0))
4011         {
4012             int x;
4013             int err = odr_geterrorx(c->odr_in, &x);
4014             char msg[100];
4015             const char *element = odr_getelement(c->odr_in);
4016             yaz_snprintf(msg, sizeof(msg),
4017                     "ODR code %d:%d element=%s offset=%d",
4018                     err, x, element ? element : "<unknown>",
4019                     odr_offset(c->odr_in));
4020             set_ZOOM_error(c, ZOOM_ERROR_DECODE, msg);
4021             if (log_api)
4022             {
4023                 FILE *ber_file = yaz_log_file();
4024                 if (ber_file)
4025                     odr_dumpBER(ber_file, c->buf_in, r);
4026             }
4027             do_close(c);
4028         }
4029         else if (gdu->which == Z_GDU_Z3950)
4030             recv_apdu(c, gdu->u.z3950);
4031         else if (gdu->which == Z_GDU_HTTP_Response)
4032         {
4033 #if YAZ_HAVE_XML2
4034             handle_http(c, gdu->u.HTTP_Response);
4035 #else
4036             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
4037             do_close(c);
4038 #endif
4039         }
4040         c->reconnect_ok = 0;
4041     }
4042     return 1;
4043 }
4044
4045 static zoom_ret do_write_ex(ZOOM_connection c, char *buf_out, int len_out)
4046 {
4047     int r;
4048     ZOOM_Event event;
4049
4050     event = ZOOM_Event_create(ZOOM_EVENT_SEND_DATA);
4051     ZOOM_connection_put_event(c, event);
4052
4053     yaz_log(log_details, "%p do_write_ex len=%d", c, len_out);
4054     if ((r = cs_put(c->cs, buf_out, len_out)) < 0)
4055     {
4056         yaz_log(log_details, "%p do_write_ex write failed", c);
4057         if (ZOOM_test_reconnect(c))
4058         {
4059             return zoom_pending;
4060         }
4061         if (c->state == STATE_CONNECTING)
4062             set_ZOOM_error(c, ZOOM_ERROR_CONNECT, c->host_port);
4063         else
4064             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, c->host_port);
4065         do_close(c);
4066         return zoom_complete;
4067     }
4068     else if (r == 1)
4069     {
4070         int mask = ZOOM_SELECT_EXCEPT;
4071         if (c->cs->io_pending & CS_WANT_WRITE)
4072             mask += ZOOM_SELECT_WRITE;
4073         if (c->cs->io_pending & CS_WANT_READ)
4074             mask += ZOOM_SELECT_READ;
4075         ZOOM_connection_set_mask(c, mask);
4076         yaz_log(log_details, "%p do_write_ex write incomplete mask=%d",
4077                 c, c->mask);
4078     }
4079     else
4080     {
4081         ZOOM_connection_set_mask(c, ZOOM_SELECT_READ|ZOOM_SELECT_EXCEPT);
4082         yaz_log(log_details, "%p do_write_ex write complete mask=%d",
4083                 c, c->mask);
4084     }
4085     return zoom_pending;
4086 }
4087
4088 static zoom_ret do_write(ZOOM_connection c)
4089 {
4090     return do_write_ex(c, c->buf_out, c->len_out);
4091 }
4092
4093
4094 ZOOM_API(const char *)
4095     ZOOM_connection_option_get(ZOOM_connection c, const char *key)
4096 {
4097     return ZOOM_options_get(c->options, key);
4098 }
4099
4100 ZOOM_API(const char *)
4101     ZOOM_connection_option_getl(ZOOM_connection c, const char *key, int *lenp)
4102 {
4103     return ZOOM_options_getl(c->options, key, lenp);
4104 }
4105
4106 ZOOM_API(void)
4107     ZOOM_connection_option_set(ZOOM_connection c, const char *key,
4108                                const char *val)
4109 {
4110     ZOOM_options_set(c->options, key, val);
4111 }
4112
4113 ZOOM_API(void)
4114     ZOOM_connection_option_setl(ZOOM_connection c, const char *key,
4115                                 const char *val, int len)
4116 {
4117     ZOOM_options_setl(c->options, key, val, len);
4118 }
4119
4120 ZOOM_API(const char *)
4121     ZOOM_resultset_option_get(ZOOM_resultset r, const char *key)
4122 {
4123     return ZOOM_options_get(r->options, key);
4124 }
4125
4126 ZOOM_API(void)
4127     ZOOM_resultset_option_set(ZOOM_resultset r, const char *key,
4128                               const char *val)
4129 {
4130     ZOOM_options_set(r->options, key, val);
4131 }
4132
4133
4134 ZOOM_API(int)
4135     ZOOM_connection_errcode(ZOOM_connection c)
4136 {
4137     return ZOOM_connection_error(c, 0, 0);
4138 }
4139
4140 ZOOM_API(const char *)
4141     ZOOM_connection_errmsg(ZOOM_connection c)
4142 {
4143     const char *msg;
4144     ZOOM_connection_error(c, &msg, 0);
4145     return msg;
4146 }
4147
4148 ZOOM_API(const char *)
4149     ZOOM_connection_addinfo(ZOOM_connection c)
4150 {
4151     const char *addinfo;
4152     ZOOM_connection_error(c, 0, &addinfo);
4153     return addinfo;
4154 }
4155
4156 ZOOM_API(const char *)
4157     ZOOM_connection_diagset(ZOOM_connection c)
4158 {
4159     const char *diagset;
4160     ZOOM_connection_error_x(c, 0, 0, &diagset);
4161     return diagset;
4162 }
4163
4164 ZOOM_API(const char *)
4165     ZOOM_diag_str(int error)
4166 {
4167     switch (error)
4168     {
4169     case ZOOM_ERROR_NONE:
4170         return "No error";
4171     case ZOOM_ERROR_CONNECT:
4172         return "Connect failed";
4173     case ZOOM_ERROR_MEMORY:
4174         return "Out of memory";
4175     case ZOOM_ERROR_ENCODE:
4176         return "Encoding failed";
4177     case ZOOM_ERROR_DECODE:
4178         return "Decoding failed";
4179     case ZOOM_ERROR_CONNECTION_LOST:
4180         return "Connection lost";
4181     case ZOOM_ERROR_INIT:
4182         return "Init rejected";
4183     case ZOOM_ERROR_INTERNAL:
4184         return "Internal failure";
4185     case ZOOM_ERROR_TIMEOUT:
4186         return "Timeout";
4187     case ZOOM_ERROR_UNSUPPORTED_PROTOCOL:
4188         return "Unsupported protocol";
4189     case ZOOM_ERROR_UNSUPPORTED_QUERY:
4190         return "Unsupported query type";
4191     case ZOOM_ERROR_INVALID_QUERY:
4192         return "Invalid query";
4193     case ZOOM_ERROR_CQL_PARSE:
4194         return "CQL parsing error";
4195     case ZOOM_ERROR_CQL_TRANSFORM:
4196         return "CQL transformation error";
4197     case ZOOM_ERROR_CCL_CONFIG:
4198         return "CCL configuration error";
4199     case ZOOM_ERROR_CCL_PARSE:
4200         return "CCL parsing error";
4201     default:
4202         return diagbib1_str(error);
4203     }
4204 }
4205
4206 ZOOM_API(int)
4207     ZOOM_connection_error_x(ZOOM_connection c, const char **cp,
4208                             const char **addinfo, const char **diagset)
4209 {
4210     int error = c->error;
4211     if (cp)
4212     {
4213         if (!c->diagset || !strcmp(c->diagset, "ZOOM"))
4214             *cp = ZOOM_diag_str(error);
4215         else if (!strcmp(c->diagset, "HTTP"))
4216             *cp = z_HTTP_errmsg(c->error);
4217         else if (!strcmp(c->diagset, "Bib-1"))
4218             *cp = ZOOM_diag_str(error);
4219         else if (!strcmp(c->diagset, "info:srw/diagnostic/1"))
4220             *cp = yaz_diag_srw_str(c->error);
4221         else
4222             *cp = "Unknown error and diagnostic set";
4223     }
4224     if (addinfo)
4225         *addinfo = c->addinfo ? c->addinfo : "";
4226     if (diagset)
4227         *diagset = c->diagset ? c->diagset : "";
4228     return c->error;
4229 }
4230
4231 ZOOM_API(int)
4232     ZOOM_connection_error(ZOOM_connection c, const char **cp,
4233                           const char **addinfo)
4234 {
4235     return ZOOM_connection_error_x(c, cp, addinfo, 0);
4236 }
4237
4238 static void ZOOM_connection_do_io(ZOOM_connection c, int mask)
4239 {
4240     ZOOM_Event event = 0;
4241     int r = cs_look(c->cs);
4242     yaz_log(log_details, "%p ZOOM_connection_do_io mask=%d cs_look=%d",
4243             c, mask, r);
4244
4245     if (r == CS_NONE)
4246     {
4247         event = ZOOM_Event_create(ZOOM_EVENT_CONNECT);
4248         set_ZOOM_error(c, ZOOM_ERROR_CONNECT, c->host_port);
4249         do_close(c);
4250         ZOOM_connection_put_event(c, event);
4251     }
4252     else if (r == CS_CONNECT)
4253     {
4254         int ret = ret = cs_rcvconnect(c->cs);
4255         yaz_log(log_details, "%p ZOOM_connection_do_io "
4256                 "cs_rcvconnect returned %d", c, ret);
4257         if (ret == 1)
4258         {
4259             int mask = ZOOM_SELECT_EXCEPT;
4260             if (c->cs->io_pending & CS_WANT_WRITE)
4261                 mask += ZOOM_SELECT_WRITE;
4262             if (c->cs->io_pending & CS_WANT_READ)
4263                 mask += ZOOM_SELECT_READ;
4264             ZOOM_connection_set_mask(c, mask);
4265             event = ZOOM_Event_create(ZOOM_EVENT_NONE);
4266             ZOOM_connection_put_event(c, event);
4267         }
4268         else if (ret == 0)
4269         {
4270             event = ZOOM_Event_create(ZOOM_EVENT_CONNECT);
4271             ZOOM_connection_put_event(c, event);
4272             get_cert(c);
4273             if (c->proto == PROTO_Z3950)
4274                 ZOOM_connection_send_init(c);
4275             else
4276             {
4277                 /* no init request for SRW .. */
4278                 assert(c->tasks->which == ZOOM_TASK_CONNECT);
4279                 ZOOM_connection_remove_task(c);
4280                 ZOOM_connection_set_mask(c, 0);
4281                 ZOOM_connection_exec_task(c);
4282             }
4283             c->state = STATE_ESTABLISHED;
4284         }
4285         else
4286         {
4287             set_ZOOM_error(c, ZOOM_ERROR_CONNECT, c->host_port);
4288             do_close(c);
4289         }
4290     }
4291     else
4292     {
4293         if (mask & ZOOM_SELECT_EXCEPT)
4294         {
4295             if (!ZOOM_test_reconnect(c))
4296             {
4297                 set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, c->host_port);
4298                 do_close(c);
4299             }
4300             return;
4301         }
4302         if (mask & ZOOM_SELECT_READ)
4303             do_read(c);
4304         if (c->cs && (mask & ZOOM_SELECT_WRITE))
4305             do_write(c);
4306     }
4307 }
4308
4309 ZOOM_API(int)
4310     ZOOM_connection_last_event(ZOOM_connection cs)
4311 {
4312     if (!cs)
4313         return ZOOM_EVENT_NONE;
4314     return cs->last_event;
4315 }
4316
4317
4318 static void cql2pqf_wrbuf_puts(const char *buf, void *client_data)
4319 {
4320     WRBUF wrbuf = (WRBUF) client_data;
4321     wrbuf_puts(wrbuf, buf);
4322 }
4323
4324 /*
4325  * Returns an xmalloc()d string containing RPN that corresponds to the
4326  * CQL passed in.  On error, sets the Connection object's error state
4327  * and returns a null pointer.
4328  * ### We could cache CQL parser and/or transformer in Connection.
4329  */
4330 static char *cql2pqf(ZOOM_connection c, const char *cql)
4331 {
4332     CQL_parser parser;
4333     int error;
4334     const char *cqlfile;
4335     cql_transform_t trans;
4336     char *result = 0;
4337
4338     parser = cql_parser_create();
4339     if ((error = cql_parser_string(parser, cql)) != 0) {
4340         cql_parser_destroy(parser);
4341         set_ZOOM_error(c, ZOOM_ERROR_CQL_PARSE, cql);
4342         return 0;
4343     }
4344
4345     cqlfile = ZOOM_connection_option_get(c, "cqlfile");
4346     if (cqlfile == 0)
4347     {
4348         set_ZOOM_error(c, ZOOM_ERROR_CQL_TRANSFORM, "no CQL transform file");
4349     }
4350     else if ((trans = cql_transform_open_fname(cqlfile)) == 0)
4351     {
4352         char buf[512];
4353         sprintf(buf, "can't open CQL transform file '%.200s': %.200s",
4354                 cqlfile, strerror(errno));
4355         set_ZOOM_error(c, ZOOM_ERROR_CQL_TRANSFORM, buf);
4356     }
4357     else
4358     {
4359         WRBUF wrbuf_result = wrbuf_alloc();
4360         error = cql_transform(trans, cql_parser_result(parser),
4361                               cql2pqf_wrbuf_puts, wrbuf_result);
4362         if (error != 0) {
4363             char buf[512];
4364             const char *addinfo;
4365             error = cql_transform_error(trans, &addinfo);
4366             sprintf(buf, "%.200s (addinfo=%.200s)",
4367                     cql_strerror(error), addinfo);
4368             set_ZOOM_error(c, ZOOM_ERROR_CQL_TRANSFORM, buf);
4369         }
4370         else
4371         {
4372             result = xstrdup(wrbuf_cstr(wrbuf_result));
4373         }
4374         cql_transform_close(trans);
4375         wrbuf_destroy(wrbuf_result);
4376     }
4377     cql_parser_destroy(parser);
4378     return result;
4379 }
4380
4381 ZOOM_API(int) ZOOM_connection_fire_event_timeout(ZOOM_connection c)
4382 {
4383     if (c->mask)
4384     {
4385         ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_TIMEOUT);
4386         /* timeout and this connection was waiting */
4387         set_ZOOM_error(c, ZOOM_ERROR_TIMEOUT, 0);
4388         do_close(c);
4389         ZOOM_connection_put_event(c, event);
4390     }
4391     return 0;
4392 }
4393
4394 ZOOM_API(int)
4395     ZOOM_connection_process(ZOOM_connection c)
4396 {
4397     ZOOM_Event event;
4398     if (!c)
4399         return 0;
4400
4401     event = ZOOM_connection_get_event(c);
4402     if (event)
4403     {
4404         ZOOM_Event_destroy(event);
4405         return 1;
4406     }
4407     ZOOM_connection_exec_task(c);
4408     event = ZOOM_connection_get_event(c);
4409     if (event)
4410     {
4411         ZOOM_Event_destroy(event);
4412         return 1;
4413     }
4414     return 0;
4415 }
4416
4417 ZOOM_API(int)
4418     ZOOM_event_nonblock(int no, ZOOM_connection *cs)
4419 {
4420     int i;
4421
4422     yaz_log(log_details, "ZOOM_process_event(no=%d,cs=%p)", no, cs);
4423
4424     for (i = 0; i<no; i++)
4425     {
4426         ZOOM_connection c = cs[i];
4427
4428         if (c && ZOOM_connection_process(c))
4429             return i+1;
4430     }
4431     return 0;
4432 }
4433
4434 ZOOM_API(int) ZOOM_connection_fire_event_socket(ZOOM_connection c, int mask)
4435 {
4436     if (c->mask && mask)
4437         ZOOM_connection_do_io(c, mask);
4438     return 0;
4439 }
4440
4441 ZOOM_API(int) ZOOM_connection_get_socket(ZOOM_connection c)
4442 {
4443     if (c->cs)
4444         return cs_fileno(c->cs);
4445     return -1;
4446 }
4447
4448 ZOOM_API(int) ZOOM_connection_set_mask(ZOOM_connection c, int mask)
4449 {
4450     c->mask = mask;
4451     if (!c->cs)
4452         return -1;
4453     return 0;
4454 }
4455
4456 ZOOM_API(int) ZOOM_connection_get_mask(ZOOM_connection c)
4457 {
4458     if (c->cs)
4459         return c->mask;
4460     return 0;
4461 }
4462
4463 ZOOM_API(int) ZOOM_connection_get_timeout(ZOOM_connection c)
4464 {
4465     return ZOOM_options_get_int(c->options, "timeout", 30);
4466 }
4467
4468 /*
4469  * Local variables:
4470  * c-basic-offset: 4
4471  * indent-tabs-mode: nil
4472  * End:
4473  * vim: shiftwidth=4 tabstop=8 expandtab
4474  */
4475