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