src/zoom-c.c

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