src/zoom-c.c

   1 /*
   2  * Copyright (c) 2000-2004, Index Data
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: zoom-c.c,v 1.19 2004-01-16 10:04:54 adam Exp $
   6  *
   7  * ZOOM layer for C, connections, result sets, queries.
   8  */
   9 #include <assert.h>
  10 #include <string.h>
  11 #include "zoom-p.h"
  12
  13 #include <yaz/yaz-util.h>
  14 #include <yaz/xmalloc.h>
  15 #include <yaz/otherinfo.h>
  16 #include <yaz/log.h>
  17 #include <yaz/pquery.h>
  18 #include <yaz/marcdisp.h>
  19 #include <yaz/diagbib1.h>
  20 #include <yaz/charneg.h>
  21 #include <yaz/ill.h>
  22 #include <yaz/srw.h>
  23
  24 #if HAVE_SYS_POLL_H
  25 #include <sys/poll.h>
  26 #endif
  27
  28 typedef enum {
  29     zoom_pending,
  30     zoom_complete
  31 } zoom_ret;
  32
  33 static zoom_ret ZOOM_connection_send_init (ZOOM_connection c);
  34 static zoom_ret do_write_ex (ZOOM_connection c, char *buf_out, int len_out);
  35
  36 static ZOOM_Event ZOOM_Event_create (int kind)
  37 {
  38     ZOOM_Event event = (ZOOM_Event) xmalloc (sizeof(*event));
  39     event->kind = kind;
  40     event->next = 0;
  41     event->prev = 0;
  42     return event;
  43 }
  44
  45 static void ZOOM_Event_destroy (ZOOM_Event event)
  46 {
  47     xfree (event);
  48 }
  49
  50 static void ZOOM_connection_put_event (ZOOM_connection c, ZOOM_Event event)
  51 {
  52     if (c->m_queue_back)
  53     {
  54         c->m_queue_back->prev = event;
  55         assert (c->m_queue_front);
  56     }
  57     else
  58     {
  59         assert (!c->m_queue_front);
  60         c->m_queue_front = event;
  61     }
  62     event->next = c->m_queue_back;
  63     event->prev = 0;
  64     c->m_queue_back = event;
  65 }
  66
  67 static ZOOM_Event ZOOM_connection_get_event(ZOOM_connection c)
  68 {
  69     ZOOM_Event event = c->m_queue_front;
  70     if (!event)
  71     {
  72         c->last_event = ZOOM_EVENT_NONE;
  73         return 0;
  74     }
  75     assert (c->m_queue_back);
  76     c->m_queue_front = event->prev;
  77     if (c->m_queue_front)
  78     {
  79         assert (c->m_queue_back);
  80         c->m_queue_front->next = 0;
  81     }
  82     else
  83         c->m_queue_back = 0;
  84     c->last_event = event->kind;
  85     return event;
  86 }
  87
  88
  89 static void set_dset_error (ZOOM_connection c, int error,
  90                             const char *dset,
  91                             const char *addinfo, const char *addinfo2)
  92 {
  93     xfree (c->addinfo);
  94     c->addinfo = 0;
  95     c->error = error;
  96     c->diagset = dset;
  97     if (addinfo && addinfo2)
  98     {
  99         c->addinfo = xmalloc(strlen(addinfo) + strlen(addinfo2) + 2);
 100         strcpy(c->addinfo, addinfo);
 101         strcat(c->addinfo, addinfo2);
 102     }
 103     else if (addinfo)
 104         c->addinfo = xstrdup(addinfo);
 105     if (error)
 106         yaz_log(LOG_DEBUG, "Error %s %s:%d %s %s",
 107                 c->host_port ? c->host_port : "<>", dset, error,
 108                 addinfo ? addinfo : "",
 109                 addinfo2 ? addinfo2 : "");
 110 }
 111
 112 #if HAVE_XML2
 113 static void set_HTTP_error (ZOOM_connection c, int error,
 114                             const char *addinfo, const char *addinfo2)
 115 {
 116     set_dset_error(c, error, "HTTP", addinfo, addinfo2);
 117 }
 118 #endif
 119
 120 static void set_ZOOM_error (ZOOM_connection c, int error,
 121                             const char *addinfo)
 122 {
 123     set_dset_error(c, error, "ZOOM", addinfo, 0);
 124 }
 125
 126 static void clear_error (ZOOM_connection c)
 127 {
 128
 129     switch (c->error)
 130     {
 131     case ZOOM_ERROR_CONNECT:
 132     case ZOOM_ERROR_MEMORY:
 133     case ZOOM_ERROR_DECODE:
 134     case ZOOM_ERROR_CONNECTION_LOST:
 135     case ZOOM_ERROR_INIT:
 136     case ZOOM_ERROR_INTERNAL:
 137     case ZOOM_ERROR_UNSUPPORTED_PROTOCOL:
 138         break;
 139     default:
 140         set_ZOOM_error(c, ZOOM_ERROR_NONE, 0);
 141     }
 142 }
 143
 144 ZOOM_task ZOOM_connection_add_task (ZOOM_connection c, int which)
 145 {
 146     ZOOM_task *taskp = &c->tasks;
 147     while (*taskp)
 148         taskp = &(*taskp)->next;
 149     *taskp = (ZOOM_task) xmalloc (sizeof(**taskp));
 150     (*taskp)->running = 0;
 151     (*taskp)->which = which;
 152     (*taskp)->next = 0;
 153     clear_error (c);
 154     return *taskp;
 155 }
 156
 157 ZOOM_task ZOOM_connection_insert_task (ZOOM_connection c, int which)
 158 {
 159     ZOOM_task task = (ZOOM_task) xmalloc (sizeof(*task));
 160
 161     task->next = c->tasks;
 162     c->tasks = task;
 163
 164     task->running = 0;
 165     task->which = which;
 166     clear_error (c);
 167     return task;
 168 }
 169
 170 void ZOOM_connection_remove_task (ZOOM_connection c)
 171 {
 172     ZOOM_task task = c->tasks;
 173
 174     if (task)
 175     {
 176         c->tasks = task->next;
 177         switch (task->which)
 178         {
 179         case ZOOM_TASK_SEARCH:
 180
 181             ZOOM_resultset_destroy (task->u.search.resultset);
 182             break;
 183         case ZOOM_TASK_RETRIEVE:
 184             ZOOM_resultset_destroy (task->u.retrieve.resultset);
 185             break;
 186         case ZOOM_TASK_CONNECT:
 187             break;
 188         case ZOOM_TASK_SCAN:
 189             ZOOM_scanset_destroy (task->u.scan.scan);
 190             break;
 191         case ZOOM_TASK_PACKAGE:
 192             ZOOM_package_destroy (task->u.package);
 193             break;
 194         default:
 195             assert (0);
 196         }
 197         xfree (task);
 198     }
 199 }
 200
 201 static int ZOOM_connection_exec_task (ZOOM_connection c);
 202
 203 void ZOOM_connection_remove_tasks (ZOOM_connection c)
 204 {
 205     while (c->tasks)
 206         ZOOM_connection_remove_task(c);
 207 }
 208
 209 static ZOOM_record record_cache_lookup (ZOOM_resultset r, int pos);
 210
 211 ZOOM_API(ZOOM_connection)
 212 ZOOM_connection_create (ZOOM_options options)
 213 {
 214     ZOOM_connection c = (ZOOM_connection) xmalloc (sizeof(*c));
 215
 216     c->proto = PROTO_Z3950;
 217     c->cs = 0;
 218     c->mask = 0;
 219     c->reconnect_ok = 0;
 220     c->state = STATE_IDLE;
 221     c->addinfo = 0;
 222     set_ZOOM_error(c, ZOOM_ERROR_NONE, 0);
 223     c->buf_in = 0;
 224     c->len_in = 0;
 225     c->buf_out = 0;
 226     c->len_out = 0;
 227     c->resultsets = 0;
 228
 229     c->options = ZOOM_options_create_with_parent(options);
 230
 231     c->host_port = 0;
 232     c->path = 0;
 233     c->proxy = 0;
 234
 235     c->charset = c->lang = 0;
 236
 237     c->cookie_out = 0;
 238     c->cookie_in = 0;
 239     c->client_IP = 0;
 240     c->tasks = 0;
 241
 242     c->odr_in = odr_createmem (ODR_DECODE);
 243     c->odr_out = odr_createmem (ODR_ENCODE);
 244
 245     c->async = 0;
 246     c->support_named_resultsets = 0;
 247     c->last_event = ZOOM_EVENT_NONE;
 248
 249     c->m_queue_front = 0;
 250     c->m_queue_back = 0;
 251     return c;
 252 }
 253
 254 /* set database names. Take local databases (if set); otherwise
 255    take databases given in ZURL (if set); otherwise use Default */
 256 static char **set_DatabaseNames (ZOOM_connection con, ZOOM_options options,
 257                                  int *num)
 258 {
 259     char **databaseNames;
 260     const char *c;
 261     int no = 2;
 262     const char *cp = ZOOM_options_get (options, "databaseName");
 263
 264     if (!cp || !*cp)
 265     {
 266         if (strncmp (con->host_port, "unix:", 5) == 0)
 267             cp = strchr (con->host_port+5, ':');
 268         else
 269             cp = strchr (con->host_port, '/');
 270         if (cp)
 271             cp++;
 272     }
 273     if (cp)
 274     {
 275         c = cp;
 276         while ((c = strchr(c, '+')))
 277         {
 278             c++;
 279             no++;
 280         }
 281     }
 282     else
 283         cp = "Default";
 284     databaseNames = (char**)
 285         odr_malloc (con->odr_out, no * sizeof(*databaseNames));
 286     no = 0;
 287     while (*cp)
 288     {
 289         c = strchr (cp, '+');
 290         if (!c)
 291             c = cp + strlen(cp);
 292         else if (c == cp)
 293         {
 294             cp++;
 295             continue;
 296         }
 297         /* cp ptr to first char of db name, c is char
 298            following db name */
 299         databaseNames[no] = (char*) odr_malloc (con->odr_out, 1+c-cp);
 300         memcpy (databaseNames[no], cp, c-cp);
 301         databaseNames[no++][c-cp] = '\0';
 302         cp = c;
 303         if (*cp)
 304             cp++;
 305     }
 306     databaseNames[no] = NULL;
 307     *num = no;
 308     return databaseNames;
 309 }
 310
 311 ZOOM_API(ZOOM_connection)
 312 ZOOM_connection_new (const char *host, int portnum)
 313 {
 314     ZOOM_connection c = ZOOM_connection_create (0);
 315
 316     ZOOM_connection_connect (c, host, portnum);
 317     return c;
 318 }
 319
 320 ZOOM_API(void)
 321 ZOOM_connection_connect(ZOOM_connection c,
 322                         const char *host, int portnum)
 323 {
 324     const char *val;
 325     ZOOM_task task;
 326
 327     if (c->cs)
 328     {
 329         yaz_log (LOG_DEBUG, "reconnect");
 330         c->reconnect_ok = 1;
 331         return;
 332     }
 333     yaz_log(LOG_DEBUG, "connect");
 334     xfree (c->proxy);
 335     val = ZOOM_options_get (c->options, "proxy");
 336     if (val && *val)
 337         c->proxy = xstrdup (val);
 338     else
 339         c->proxy = 0;
 340
 341     xfree (c->charset);
 342     val = ZOOM_options_get (c->options, "charset");
 343     if (val && *val)
 344         c->charset = xstrdup (val);
 345     else
 346         c->charset = 0;
 347
 348     xfree (c->lang);
 349     val = ZOOM_options_get (c->options, "lang");
 350     if (val && *val)
 351         c->lang = xstrdup (val);
 352     else
 353         c->lang = 0;
 354
 355     xfree (c->host_port);
 356     if (portnum)
 357     {
 358         char hostn[128];
 359         sprintf (hostn, "%.80s:%d", host, portnum);
 360         c->host_port = xstrdup(hostn);
 361     }
 362     else
 363         c->host_port = xstrdup(host);
 364
 365     ZOOM_options_set(c->options, "host", c->host_port);
 366
 367     val = ZOOM_options_get (c->options, "cookie");
 368     if (val && *val)
 369         c->cookie_out = xstrdup (val);
 370
 371     val = ZOOM_options_get (c->options, "clientIP");
 372     if (val && *val)
 373         c->client_IP = xstrdup (val);
 374
 375     c->async = ZOOM_options_get_bool (c->options, "async", 0);
 376
 377     set_ZOOM_error(c, ZOOM_ERROR_NONE, 0);
 378
 379     task = ZOOM_connection_add_task (c, ZOOM_TASK_CONNECT);
 380
 381     if (!c->async)
 382     {
 383         while (ZOOM_event (1, &c))
 384             ;
 385     }
 386 }
 387
 388 ZOOM_API(ZOOM_query)
 389 ZOOM_query_create(void)
 390 {
 391     ZOOM_query s = (ZOOM_query) xmalloc (sizeof(*s));
 392
 393     s->refcount = 1;
 394     s->z_query = 0;
 395     s->sort_spec = 0;
 396     s->odr = odr_createmem (ODR_ENCODE);
 397     s->query_string = 0;
 398
 399     return s;
 400 }
 401
 402 ZOOM_API(void)
 403 ZOOM_query_destroy(ZOOM_query s)
 404 {
 405     if (!s)
 406         return;
 407
 408     (s->refcount)--;
 409     yaz_log (LOG_DEBUG, "ZOOM_query_destroy count=%d", s->refcount);
 410     if (s->refcount == 0)
 411     {
 412         odr_destroy (s->odr);
 413         xfree (s);
 414     }
 415 }
 416
 417 ZOOM_API(int)
 418 ZOOM_query_prefix(ZOOM_query s, const char *str)
 419 {
 420     s->query_string = odr_strdup(s->odr, str);
 421     s->z_query = (Z_Query *) odr_malloc (s->odr, sizeof(*s->z_query));
 422     s->z_query->which = Z_Query_type_1;
 423     s->z_query->u.type_1 =  p_query_rpn(s->odr, PROTO_Z3950, str);
 424     if (!s->z_query->u.type_1)
 425         return -1;
 426     return 0;
 427 }
 428
 429 ZOOM_API(int)
 430 ZOOM_query_cql(ZOOM_query s, const char *str)
 431 {
 432     Z_External *ext;
 433
 434     s->query_string = odr_strdup(s->odr, str);
 435
 436     ext = (Z_External *) odr_malloc(s->odr, sizeof(*ext));
 437     ext->direct_reference = odr_getoidbystr(s->odr, "1.2.840.10003.16.2");
 438     ext->indirect_reference = 0;
 439     ext->descriptor = 0;
 440     ext->which = Z_External_CQL;
 441     ext->u.cql = s->query_string;
 442
 443     s->z_query = (Z_Query *) odr_malloc (s->odr, sizeof(*s->z_query));
 444     s->z_query->which = Z_Query_type_104;
 445     s->z_query->u.type_104 =  ext;
 446
 447     return 0;
 448 }
 449
 450 ZOOM_API(int)
 451 ZOOM_query_sortby(ZOOM_query s, const char *criteria)
 452 {
 453     s->sort_spec = yaz_sort_spec (s->odr, criteria);
 454     if (!s->sort_spec)
 455         return -1;
 456     return 0;
 457 }
 458
 459 static zoom_ret do_write(ZOOM_connection c);
 460
 461 ZOOM_API(void)
 462 ZOOM_connection_destroy(ZOOM_connection c)
 463 {
 464     ZOOM_resultset r;
 465     if (!c)
 466         return;
 467     if (c->cs)
 468         cs_close (c->cs);
 469     for (r = c->resultsets; r; r = r->next)
 470         r->connection = 0;
 471
 472     xfree (c->buf_in);
 473     xfree (c->addinfo);
 474     odr_destroy (c->odr_in);
 475     odr_destroy (c->odr_out);
 476     ZOOM_options_destroy (c->options);
 477     ZOOM_connection_remove_tasks (c);
 478     xfree (c->host_port);
 479     xfree (c->path);
 480     xfree (c->proxy);
 481     xfree (c->charset);
 482     xfree (c->lang);
 483     xfree (c->cookie_out);
 484     xfree (c->cookie_in);
 485     xfree (c->client_IP);
 486     xfree (c);
 487 }
 488
 489 void ZOOM_resultset_addref (ZOOM_resultset r)
 490 {
 491     if (r)
 492     {
 493         (r->refcount)++;
 494         yaz_log (LOG_DEBUG, "ZOOM_resultset_addref r=%p count=%d",
 495                  r, r->refcount);
 496     }
 497 }
 498 ZOOM_resultset ZOOM_resultset_create ()
 499 {
 500     ZOOM_resultset r = (ZOOM_resultset) xmalloc (sizeof(*r));
 501
 502     yaz_log (LOG_DEBUG, "ZOOM_resultset_create r = %p", r);
 503     r->refcount = 1;
 504     r->size = 0;
 505     r->odr = odr_createmem (ODR_ENCODE);
 506     r->start = 0;
 507     r->piggyback = 1;
 508     r->setname = 0;
 509     r->schema = 0;
 510     r->count = 0;
 511     r->step = 0;
 512     r->record_cache = 0;
 513     r->r_sort_spec = 0;
 514     r->query = 0;
 515     r->connection = 0;
 516     r->next = 0;
 517     return r;
 518 }
 519
 520 ZOOM_API(ZOOM_resultset)
 521 ZOOM_connection_search_pqf(ZOOM_connection c, const char *q)
 522 {
 523     ZOOM_resultset r;
 524     ZOOM_query s = ZOOM_query_create();
 525
 526     ZOOM_query_prefix (s, q);
 527
 528     r = ZOOM_connection_search (c, s);
 529     ZOOM_query_destroy (s);
 530     return r;
 531 }
 532
 533 ZOOM_API(ZOOM_resultset)
 534 ZOOM_connection_search(ZOOM_connection c, ZOOM_query q)
 535 {
 536     ZOOM_resultset r = ZOOM_resultset_create ();
 537     ZOOM_task task;
 538     const char *cp;
 539
 540     r->r_sort_spec = q->sort_spec;
 541     r->query = q;
 542
 543     r->options = ZOOM_options_create_with_parent(c->options);
 544
 545     r->start = ZOOM_options_get_int(r->options, "start", 0);
 546     r->count = ZOOM_options_get_int(r->options, "count", 0);
 547     {
 548         /* If "presentChunk" is defined use that; otherwise "step" */
 549         const char *cp = ZOOM_options_get (r->options, "presentChunk");
 550         r->step = ZOOM_options_get_int(r->options,
 551                                        (cp != 0 ? "presentChunk": "step"), 0);
 552     }
 553     r->piggyback = ZOOM_options_get_bool (r->options, "piggyback", 1);
 554     cp = ZOOM_options_get (r->options, "setname");
 555     if (cp)
 556         r->setname = xstrdup(cp);
 557     cp = ZOOM_options_get (r->options, "schema");
 558     if (cp)
 559         r->schema = xstrdup(cp);
 560
 561     r->connection = c;
 562
 563     r->next = c->resultsets;
 564     c->resultsets = r;
 565
 566     if (c->host_port && c->proto == PROTO_HTTP)
 567     {
 568         if (!c->cs)
 569         {
 570             yaz_log(LOG_DEBUG, "NO COMSTACK");
 571             ZOOM_connection_add_task(c, ZOOM_TASK_CONNECT);
 572         }
 573         else
 574         {
 575             yaz_log(LOG_DEBUG, "PREPARE FOR RECONNECT");
 576             c->reconnect_ok = 1;
 577         }
 578     }
 579
 580     task = ZOOM_connection_add_task (c, ZOOM_TASK_SEARCH);
 581     task->u.search.resultset = r;
 582     ZOOM_resultset_addref (r);
 583
 584     (q->refcount)++;
 585
 586     if (!c->async)
 587     {
 588         while (ZOOM_event (1, &c))
 589             ;
 590     }
 591     return r;
 592 }
 593
 594 ZOOM_API(void)
 595 ZOOM_resultset_destroy(ZOOM_resultset r)
 596 {
 597     if (!r)
 598         return;
 599     (r->refcount)--;
 600     yaz_log (LOG_DEBUG, "ZOOM_resultset_destroy r = %p count=%d",
 601              r, r->refcount);
 602     if (r->refcount == 0)
 603     {
 604         ZOOM_record_cache rc;
 605
 606         for (rc = r->record_cache; rc; rc = rc->next)
 607         {
 608             if (rc->rec.wrbuf_marc)
 609                 wrbuf_free (rc->rec.wrbuf_marc, 1);
 610             if (rc->rec.wrbuf_iconv)
 611                 wrbuf_free (rc->rec.wrbuf_iconv, 1);
 612             if (rc->rec.wrbuf_opac)
 613                 wrbuf_free (rc->rec.wrbuf_opac, 1);
 614         }
 615         if (r->connection)
 616         {
 617             /* remove ourselves from the resultsets in connection */
 618             ZOOM_resultset *rp = &r->connection->resultsets;
 619             while (1)
 620             {
 621                 assert (*rp);   /* we must be in this list!! */
 622                 if (*rp == r)
 623                 {   /* OK, we're here - take us out of it */
 624                     *rp = (*rp)->next;
 625                     break;
 626                 }
 627                 rp = &(*rp)->next;
 628             }
 629         }
 630         ZOOM_query_destroy (r->query);
 631         ZOOM_options_destroy (r->options);
 632         odr_destroy (r->odr);
 633         xfree (r->setname);
 634         xfree (r->schema);
 635         xfree (r);
 636     }
 637 }
 638
 639 ZOOM_API(size_t)
 640 ZOOM_resultset_size (ZOOM_resultset r)
 641 {
 642     return r->size;
 643 }
 644
 645 static void do_close (ZOOM_connection c)
 646 {
 647     if (c->cs)
 648         cs_close(c->cs);
 649     c->cs = 0;
 650     c->mask = 0;
 651     c->state = STATE_IDLE;
 652 }
 653
 654 static void ZOOM_resultset_retrieve (ZOOM_resultset r,
 655                                      int force_sync, int start, int count)
 656 {
 657     ZOOM_task task;
 658     ZOOM_connection c;
 659     const char *cp;
 660
 661     if (!r)
 662         return;
 663     c = r->connection;
 664     if (!c)
 665         return;
 666
 667     if (c->host_port && c->proto == PROTO_HTTP)
 668     {
 669         if (!c->cs)
 670         {
 671             yaz_log(LOG_DEBUG, "NO COMSTACK");
 672             ZOOM_connection_add_task(c, ZOOM_TASK_CONNECT);
 673         }
 674         else
 675         {
 676             yaz_log(LOG_DEBUG, "PREPARE FOR RECONNECT");
 677             c->reconnect_ok = 1;
 678         }
 679     }
 680     task = ZOOM_connection_add_task (c, ZOOM_TASK_RETRIEVE);
 681     task->u.retrieve.resultset = r;
 682     task->u.retrieve.start = start;
 683     task->u.retrieve.count = count;
 684
 685     cp = ZOOM_options_get (r->options, "schema");
 686     if (cp)
 687     {
 688         if (!r->schema || strcmp(r->schema, cp))
 689         {
 690             xfree(r->schema);
 691             r->schema = xstrdup(cp);
 692         }
 693     }
 694
 695     ZOOM_resultset_addref (r);
 696
 697     if (!r->connection->async || force_sync)
 698         while (r->connection && ZOOM_event (1, &r->connection))
 699             ;
 700 }
 701
 702 ZOOM_API(void)
 703 ZOOM_resultset_records (ZOOM_resultset r, ZOOM_record *recs,
 704                         size_t start, size_t count)
 705 {
 706     int force_present = 0;
 707
 708     if (!r)
 709         return ;
 710     if (count && recs)
 711         force_present = 1;
 712     ZOOM_resultset_retrieve (r, force_present, start, count);
 713     if (force_present)
 714     {
 715         size_t i;
 716         for (i = 0; i< count; i++)
 717             recs[i] = ZOOM_resultset_record_immediate (r, i+start);
 718     }
 719 }
 720
 721 static zoom_ret do_connect (ZOOM_connection c)
 722 {
 723     void *add;
 724     const char *effective_host;
 725
 726     if (c->proxy)
 727         effective_host = c->proxy;
 728     else
 729         effective_host = c->host_port;
 730
 731     yaz_log (LOG_DEBUG, "do_connect host=%s", effective_host);
 732
 733     assert (!c->cs);
 734     c->cs = cs_create_host (effective_host, 0, &add);
 735
 736     if (c->cs && c->cs->protocol == PROTO_HTTP)
 737     {
 738 #if HAVE_XML2
 739         const char *path = 0;
 740
 741         c->proto = PROTO_HTTP;
 742         cs_get_host_args(c->host_port, &path);
 743         xfree(c->path);
 744         c->path = xmalloc(strlen(path)+2);
 745         c->path[0] = '/';
 746         strcpy (c->path+1, path);
 747 #else
 748         set_ZOOM_error(c, ZOOM_ERROR_UNSUPPORTED_PROTOCOL, "SRW");
 749         do_close(c);
 750         return zoom_complete;
 751 #endif
 752     }
 753     if (c->cs)
 754     {
 755         int ret = cs_connect (c->cs, add);
 756         if (ret == 0)
 757         {
 758             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_CONNECT);
 759             ZOOM_connection_put_event(c, event);
 760             if (c->proto == PROTO_Z3950)
 761                 ZOOM_connection_send_init(c);
 762             else
 763             {
 764                 /* no init request for SRW .. */
 765                 assert (c->tasks->which == ZOOM_TASK_CONNECT);
 766                 ZOOM_connection_remove_task (c);
 767                 c->mask = 0;
 768                 ZOOM_connection_exec_task (c);
 769             }
 770             c->state = STATE_ESTABLISHED;
 771             return zoom_pending;
 772         }
 773         else if (ret > 0)
 774         {
 775             c->state = STATE_CONNECTING;
 776             c->mask = ZOOM_SELECT_EXCEPT;
 777             if (c->cs->io_pending & CS_WANT_WRITE)
 778                 c->mask += ZOOM_SELECT_WRITE;
 779             if (c->cs->io_pending & CS_WANT_READ)
 780                 c->mask += ZOOM_SELECT_READ;
 781             return zoom_pending;
 782         }
 783     }
 784     c->state = STATE_IDLE;
 785     set_ZOOM_error(c, ZOOM_ERROR_CONNECT, effective_host);
 786     return zoom_complete;
 787 }
 788
 789 int z3950_connection_socket(ZOOM_connection c)
 790 {
 791     if (c->cs)
 792         return cs_fileno(c->cs);
 793     return -1;
 794 }
 795
 796 int z3950_connection_mask(ZOOM_connection c)
 797 {
 798     if (c->cs)
 799         return c->mask;
 800     return 0;
 801 }
 802
 803 static void otherInfo_attach (ZOOM_connection c, Z_APDU *a, ODR out)
 804 {
 805     int i;
 806     for (i = 0; i<200; i++)
 807     {
 808         size_t len;
 809         Z_OtherInformation **oi;
 810         char buf[80];
 811         const char *val;
 812         const char *cp;
 813         int oidval;
 814
 815         sprintf (buf, "otherInfo%d", i);
 816         val = ZOOM_options_get (c->options, buf);
 817         if (!val)
 818             break;
 819         cp = strchr (val, ':');
 820         if (!cp)
 821             continue;
 822         len = cp - val;
 823         if (len >= sizeof(buf))
 824             len = sizeof(buf)-1;
 825         memcpy (buf, val, len);
 826         buf[len] = '\0';
 827         oidval = oid_getvalbyname (buf);
 828         if (oidval == VAL_NONE)
 829             continue;
 830
 831         yaz_oi_APDU(a, &oi);
 832         yaz_oi_set_string_oidval(oi, out, oidval, 1, cp+1);
 833     }
 834 }
 835
 836 static int encode_APDU(ZOOM_connection c, Z_APDU *a, ODR out)
 837 {
 838     assert (a);
 839     if (c->cookie_out)
 840     {
 841         Z_OtherInformation **oi;
 842         yaz_oi_APDU(a, &oi);
 843         yaz_oi_set_string_oidval(oi, out, VAL_COOKIE, 1, c->cookie_out);
 844     }
 845     if (c->client_IP)
 846     {
 847         Z_OtherInformation **oi;
 848         yaz_oi_APDU(a, &oi);
 849         yaz_oi_set_string_oidval(oi, out, VAL_CLIENT_IP, 1, c->client_IP);
 850     }
 851     otherInfo_attach (c, a, out);
 852     if (!z_APDU(out, &a, 0, 0))
 853     {
 854         FILE *outf = fopen("/tmp/apdu.txt", "a");
 855         if (a && outf)
 856         {
 857             ODR odr_pr = odr_createmem(ODR_PRINT);
 858             fprintf (outf, "a=%p\n", a);
 859             odr_setprint(odr_pr, outf);
 860             z_APDU(odr_pr, &a, 0, 0);
 861             odr_destroy(odr_pr);
 862         }
 863         yaz_log (LOG_DEBUG, "encoding failed");
 864         set_ZOOM_error(c, ZOOM_ERROR_ENCODE, 0);
 865         odr_reset(out);
 866         return -1;
 867     }
 868
 869     return 0;
 870 }
 871
 872 static zoom_ret send_APDU (ZOOM_connection c, Z_APDU *a)
 873 {
 874     ZOOM_Event event;
 875     assert (a);
 876     if (encode_APDU(c, a, c->odr_out))
 877         return zoom_complete;
 878     yaz_log(LOG_DEBUG, "send APDU type=%d", a->which);
 879     c->buf_out = odr_getbuf(c->odr_out, &c->len_out, 0);
 880     event = ZOOM_Event_create (ZOOM_EVENT_SEND_APDU);
 881     ZOOM_connection_put_event (c, event);
 882     odr_reset(c->odr_out);
 883     return do_write (c);
 884 }
 885
 886 /* returns 1 if PDU was sent OK (still pending )
 887            0 if PDU was not sent OK (nothing to wait for)
 888 */
 889
 890 static zoom_ret ZOOM_connection_send_init (ZOOM_connection c)
 891 {
 892     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_initRequest);
 893     Z_InitRequest *ireq = apdu->u.initRequest;
 894     Z_IdAuthentication *auth = (Z_IdAuthentication *)
 895         odr_malloc(c->odr_out, sizeof(*auth));
 896     const char *auth_groupId = ZOOM_options_get (c->options, "group");
 897     const char *auth_userId = ZOOM_options_get (c->options, "user");
 898     const char *auth_password = ZOOM_options_get (c->options, "password");
 899     char *version;
 900
 901     /* support the pass for backwards compatibility */
 902     if (!auth_password)
 903         auth_password = ZOOM_options_get (c->options, "pass");
 904
 905     ODR_MASK_SET(ireq->options, Z_Options_search);
 906     ODR_MASK_SET(ireq->options, Z_Options_present);
 907     ODR_MASK_SET(ireq->options, Z_Options_scan);
 908     ODR_MASK_SET(ireq->options, Z_Options_sort);
 909     ODR_MASK_SET(ireq->options, Z_Options_extendedServices);
 910     ODR_MASK_SET(ireq->options, Z_Options_namedResultSets);
 911
 912     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_1);
 913     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_2);
 914     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_3);
 915
 916     /* Index Data's Z39.50 Implementor Id is 81 */
 917     ireq->implementationId = odr_prepend(c->odr_out,
 918         ZOOM_options_get(c->options, "implementationId"),
 919         odr_prepend(c->odr_out, "81", ireq->implementationId));
 920
 921     ireq->implementationName = odr_prepend(c->odr_out,
 922         ZOOM_options_get(c->options, "implementationName"),
 923         odr_prepend(c->odr_out, "ZOOM-C", ireq->implementationName));
 924
 925     version = odr_strdup(c->odr_out, "$Revision: 1.19 $");
 926     if (strlen(version) > 10)   /* check for unexpanded CVS strings */
 927         version[strlen(version)-2] = '\0';
 928     ireq->implementationVersion = odr_prepend(c->odr_out,
 929         ZOOM_options_get(c->options, "implementationVersion"),
 930         odr_prepend(c->odr_out, &version[11], ireq->implementationVersion));
 931
 932     *ireq->maximumRecordSize =
 933         ZOOM_options_get_int (c->options, "maximumRecordSize", 1024*1024);
 934     *ireq->preferredMessageSize =
 935         ZOOM_options_get_int (c->options, "preferredMessageSize", 1024*1024);
 936
 937     if (auth_groupId || auth_password)
 938     {
 939         Z_IdPass *pass = (Z_IdPass *) odr_malloc(c->odr_out, sizeof(*pass));
 940         int i = 0;
 941         pass->groupId = 0;
 942         if (auth_groupId && *auth_groupId)
 943         {
 944             pass->groupId = (char *)
 945                 odr_malloc(c->odr_out, strlen(auth_groupId)+1);
 946             strcpy(pass->groupId, auth_groupId);
 947             i++;
 948         }
 949         pass->userId = 0;
 950         if (auth_userId && *auth_userId)
 951         {
 952             pass->userId = (char *)
 953                 odr_malloc(c->odr_out, strlen(auth_userId)+1);
 954             strcpy(pass->userId, auth_userId);
 955             i++;
 956         }
 957         pass->password = 0;
 958         if (auth_password && *auth_password)
 959         {
 960             pass->password = (char *)
 961                 odr_malloc(c->odr_out, strlen(auth_password)+1);
 962             strcpy(pass->password, auth_password);
 963             i++;
 964         }
 965         if (i)
 966         {
 967             auth->which = Z_IdAuthentication_idPass;
 968             auth->u.idPass = pass;
 969             ireq->idAuthentication = auth;
 970         }
 971     }
 972     else if (auth_userId)
 973     {
 974         auth->which = Z_IdAuthentication_open;
 975         auth->u.open = (char *)
 976             odr_malloc(c->odr_out, strlen(auth_userId)+1);
 977         strcpy(auth->u.open, auth_userId);
 978         ireq->idAuthentication = auth;
 979     }
 980     if (c->proxy)
 981         yaz_oi_set_string_oidval(&ireq->otherInfo, c->odr_out,
 982                                  VAL_PROXY, 1, c->host_port);
 983     if (c->charset||c->lang)
 984     {
 985         Z_OtherInformation **oi;
 986         Z_OtherInformationUnit *oi_unit;
 987
 988         yaz_oi_APDU(apdu, &oi);
 989
 990         if ((oi_unit = yaz_oi_update(oi, c->odr_out, NULL, 0, 0)))
 991         {
 992             ODR_MASK_SET(ireq->options, Z_Options_negotiationModel);
 993
 994             oi_unit->which = Z_OtherInfo_externallyDefinedInfo;
 995             oi_unit->information.externallyDefinedInfo =
 996                 yaz_set_proposal_charneg
 997                 (c->odr_out,
 998                  (const char **)&c->charset, (c->charset) ? 1:0,
 999                  (const char **)&c->lang, (c->lang) ? 1:0, 1);
1000         }
1001     }
1002     assert (apdu);
1003     return send_APDU (c, apdu);
1004 }
1005
1006 #if HAVE_XML2
1007 static zoom_ret send_srw (ZOOM_connection c, Z_SRW_PDU *sr)
1008 {
1009     char ctype[50];
1010     Z_SOAP_Handler h[2] = {
1011         {"http://www.loc.gov/zing/srw/", 0, (Z_SOAP_fun) yaz_srw_codec},
1012         {0, 0, 0}
1013     };
1014     ODR o = odr_createmem(ODR_ENCODE);
1015     int ret;
1016     Z_SOAP *p = odr_malloc(o, sizeof(*p));
1017     Z_GDU *gdu;
1018     ZOOM_Event event;
1019
1020     gdu = z_get_HTTP_Request(c->odr_out);
1021     gdu->u.HTTP_Request->path = c->path;
1022
1023     if (c->host_port)
1024     {
1025         const char *cp0 = strstr(c->host_port, "://");
1026         const char *cp1 = 0;
1027         if (cp0)
1028             cp0 = cp0+3;
1029         else
1030             cp0 = c->host_port;
1031
1032         cp1 = strchr(cp0, '/');
1033         if (!cp1)
1034             cp1 = cp0+strlen(cp0);
1035
1036         if (cp0 && cp1)
1037         {
1038             char *h = odr_malloc(c->odr_out, cp1 - cp0 + 1);
1039             memcpy (h, cp0, cp1 - cp0);
1040             h[cp1-cp0] = '\0';
1041             z_HTTP_header_add(c->odr_out, &gdu->u.HTTP_Request->headers,
1042                               "Host", h);
1043         }
1044     }
1045
1046     strcpy(ctype, "text/xml");
1047     if (c->charset && strlen(c->charset) < 20)
1048     {
1049         strcat(ctype, "; charset=");
1050         strcat(ctype, c->charset);
1051     }
1052     z_HTTP_header_add(c->odr_out, &gdu->u.HTTP_Request->headers,
1053                       "Content-Type", ctype);
1054     z_HTTP_header_add(c->odr_out, &gdu->u.HTTP_Request->headers,
1055                       "SOAPAction", "\"\"");
1056     p->which = Z_SOAP_generic;
1057     p->u.generic = odr_malloc(o, sizeof(*p->u.generic));
1058     p->u.generic->no = 0;
1059     p->u.generic->ns = 0;
1060     p->u.generic->p = sr;
1061     p->ns = "http://schemas.xmlsoap.org/soap/envelope/";
1062
1063     ret = z_soap_codec_enc(o, &p,
1064                            &gdu->u.HTTP_Request->content_buf,
1065                            &gdu->u.HTTP_Request->content_len, h,
1066                            c->charset);
1067
1068     if (!z_GDU(c->odr_out, &gdu, 0, 0))
1069         return zoom_complete;
1070     c->buf_out = odr_getbuf(c->odr_out, &c->len_out, 0);
1071
1072     odr_destroy(o);
1073
1074     event = ZOOM_Event_create (ZOOM_EVENT_SEND_APDU);
1075     ZOOM_connection_put_event (c, event);
1076     odr_reset(c->odr_out);
1077     return do_write (c);
1078 }
1079 #endif
1080
1081 #if HAVE_XML2
1082 static zoom_ret ZOOM_connection_srw_send_search(ZOOM_connection c)
1083 {
1084     int i;
1085     ZOOM_resultset resultset = 0;
1086     Z_SRW_PDU *sr = 0;
1087     const char *recordPacking = 0;
1088
1089     if (c->error)                  /* don't continue on error */
1090         return zoom_complete;
1091     assert (c->tasks);
1092     if (c->tasks->which == ZOOM_TASK_SEARCH)
1093     {
1094         resultset = c->tasks->u.search.resultset;
1095         resultset->setname = xstrdup ("default");
1096         ZOOM_options_set (resultset->options, "setname", resultset->setname);
1097     }
1098     else if(c->tasks->which == ZOOM_TASK_RETRIEVE)
1099     {
1100         resultset = c->tasks->u.retrieve.resultset;
1101
1102         resultset->start = c->tasks->u.retrieve.start;
1103         resultset->count = c->tasks->u.retrieve.count;
1104
1105         if (resultset->start >= resultset->size)
1106             return zoom_complete;
1107         if (resultset->start + resultset->count > resultset->size)
1108             resultset->count = resultset->size - resultset->start;
1109
1110         for (i = 0; i<resultset->count; i++)
1111         {
1112             ZOOM_record rec =
1113                 record_cache_lookup (resultset, i + resultset->start);
1114             if (!rec)
1115                 break;
1116         }
1117         if (i == resultset->count)
1118             return zoom_complete;
1119     }
1120     assert(resultset->query);
1121
1122     sr = yaz_srw_get(c->odr_out, Z_SRW_searchRetrieve_request);
1123
1124     if (resultset->query->z_query->which == Z_Query_type_104
1125         && resultset->query->z_query->u.type_104->which == Z_External_CQL)
1126     {
1127
1128         sr->u.request->query_type = Z_SRW_query_type_cql;
1129         sr->u.request->query.cql =resultset->query->z_query->u.type_104->u.cql;
1130     }
1131     else if (resultset->query->z_query->which == Z_Query_type_1 &&
1132              resultset->query->z_query->u.type_1)
1133     {
1134         sr->u.request->query_type = Z_SRW_query_type_pqf;
1135         sr->u.request->query.pqf = resultset->query->query_string;
1136     }
1137     else
1138     {
1139         set_ZOOM_error(c, ZOOM_ERROR_UNSUPPORTED_QUERY, 0);
1140         return zoom_complete;
1141     }
1142     sr->u.request->startRecord = odr_intdup (c->odr_out, resultset->start + 1);
1143     sr->u.request->maximumRecords = odr_intdup (
1144         c->odr_out, resultset->step>0 ? resultset->step : resultset->count);
1145     sr->u.request->recordSchema = resultset->schema;
1146
1147     recordPacking = ZOOM_resultset_option_get (resultset, "recordPacking");
1148
1149     if (recordPacking)
1150         sr->u.request->recordPacking = odr_strdup(c->odr_out, recordPacking);
1151
1152     return send_srw(c, sr);
1153 }
1154 #else
1155 static zoom_ret ZOOM_connection_srw_send_search(ZOOM_connection c)
1156 {
1157     return zoom_complete;
1158 }
1159 #endif
1160
1161 static zoom_ret ZOOM_connection_send_search (ZOOM_connection c)
1162 {
1163     ZOOM_resultset r;
1164     int lslb, ssub, mspn;
1165     const char *syntax;
1166     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_searchRequest);
1167     Z_SearchRequest *search_req = apdu->u.searchRequest;
1168     const char *elementSetName;
1169     const char *smallSetElementSetName;
1170     const char *mediumSetElementSetName;
1171
1172     assert (c->tasks);
1173     assert (c->tasks->which == ZOOM_TASK_SEARCH);
1174
1175     r = c->tasks->u.search.resultset;
1176
1177     elementSetName =
1178         ZOOM_options_get (r->options, "elementSetName");
1179     smallSetElementSetName  =
1180         ZOOM_options_get (r->options, "smallSetElementSetName");
1181     mediumSetElementSetName =
1182         ZOOM_options_get (r->options, "mediumSetElementSetName");
1183
1184     if (!smallSetElementSetName)
1185         smallSetElementSetName = elementSetName;
1186
1187     if (!mediumSetElementSetName)
1188         mediumSetElementSetName = elementSetName;
1189
1190     assert (r);
1191     assert (r->query);
1192
1193     /* prepare query for the search request */
1194     search_req->query = r->query->z_query;
1195
1196     search_req->databaseNames =
1197         set_DatabaseNames (c, r->options, &search_req->num_databaseNames);
1198
1199     /* get syntax (no need to provide unless piggyback is in effect) */
1200     syntax = ZOOM_options_get (r->options, "preferredRecordSyntax");
1201
1202     lslb = ZOOM_options_get_int (r->options, "largeSetLowerBound", -1);
1203     ssub = ZOOM_options_get_int (r->options, "smallSetUpperBound", -1);
1204     mspn = ZOOM_options_get_int (r->options, "mediumSetPresentNumber", -1);
1205     if (lslb != -1 && ssub != -1 && mspn != -1)
1206     {
1207         /* So're a Z39.50 expert? Let's hope you don't do sort */
1208         *search_req->largeSetLowerBound = lslb;
1209         *search_req->smallSetUpperBound = ssub;
1210         *search_req->mediumSetPresentNumber = mspn;
1211     }
1212     else if (r->start == 0 && r->count > 0
1213              && r->piggyback && !r->r_sort_spec && !r->schema)
1214     {
1215         /* Regular piggyback - do it unless we're going to do sort */
1216         *search_req->largeSetLowerBound = 2000000000;
1217         *search_req->smallSetUpperBound = 1;
1218         *search_req->mediumSetPresentNumber = r->step>0 ? r->step : r->count;
1219     }
1220     else
1221     {
1222         /* non-piggyback. Need not provide elementsets or syntaxes .. */
1223         smallSetElementSetName = 0;
1224         mediumSetElementSetName = 0;
1225         syntax = 0;
1226     }
1227     if (smallSetElementSetName && *smallSetElementSetName)
1228     {
1229         Z_ElementSetNames *esn = (Z_ElementSetNames *)
1230             odr_malloc (c->odr_out, sizeof(*esn));
1231
1232         esn->which = Z_ElementSetNames_generic;
1233         esn->u.generic = odr_strdup (c->odr_out, smallSetElementSetName);
1234         search_req->smallSetElementSetNames = esn;
1235     }
1236     if (mediumSetElementSetName && *mediumSetElementSetName)
1237     {
1238         Z_ElementSetNames *esn = (Z_ElementSetNames *)
1239             odr_malloc (c->odr_out, sizeof(*esn));
1240
1241         esn->which = Z_ElementSetNames_generic;
1242         esn->u.generic = odr_strdup (c->odr_out, mediumSetElementSetName);
1243         search_req->mediumSetElementSetNames = esn;
1244     }
1245     if (syntax)
1246         search_req->preferredRecordSyntax =
1247             yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, syntax);
1248
1249     if (!r->setname)
1250     {
1251         if (c->support_named_resultsets)
1252         {
1253             char setname[14];
1254             int ord;
1255             /* find the lowest unused ordinal so that we re-use
1256                result sets on the server. */
1257             for (ord = 1; ; ord++)
1258             {
1259                 ZOOM_resultset rp;
1260                 sprintf (setname, "%d", ord);
1261                 for (rp = c->resultsets; rp; rp = rp->next)
1262                     if (rp->setname && !strcmp (rp->setname, setname))
1263                         break;
1264                 if (!rp)
1265                     break;
1266             }
1267             r->setname = xstrdup (setname);
1268             yaz_log (LOG_DEBUG, "allocating set %s", r->setname);
1269         }
1270         else
1271             r->setname = xstrdup ("default");
1272         ZOOM_options_set (r->options, "setname", r->setname);
1273     }
1274     search_req->resultSetName = odr_strdup(c->odr_out, r->setname);
1275     /* send search request */
1276     return send_APDU (c, apdu);
1277 }
1278
1279 static void response_diag (ZOOM_connection c, Z_DiagRec *p)
1280 {
1281     int oclass;
1282     Z_DefaultDiagFormat *r;
1283     char *addinfo = 0;
1284
1285     xfree (c->addinfo);
1286     c->addinfo = 0;
1287     if (p->which != Z_DiagRec_defaultFormat)
1288     {
1289         set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1290         return;
1291     }
1292     r = p->u.defaultFormat;
1293     switch (r->which)
1294     {
1295     case Z_DefaultDiagFormat_v2Addinfo:
1296         addinfo = r->u.v2Addinfo;
1297         break;
1298     case Z_DefaultDiagFormat_v3Addinfo:
1299         addinfo = r->u.v3Addinfo;
1300         break;
1301     }
1302     set_dset_error(c, *r->condition,
1303                    yaz_z3950oid_to_str(r->diagnosticSetId, &oclass),
1304                    addinfo, 0);
1305 }
1306
1307 ZOOM_API(ZOOM_record)
1308 ZOOM_record_clone (ZOOM_record srec)
1309 {
1310     char *buf;
1311     int size;
1312     ODR odr_enc;
1313     ZOOM_record nrec;
1314
1315     odr_enc = odr_createmem(ODR_ENCODE);
1316     if (!z_NamePlusRecord (odr_enc, &srec->npr, 0, 0))
1317         return 0;
1318     buf = odr_getbuf (odr_enc, &size, 0);
1319
1320     nrec = (ZOOM_record) xmalloc (sizeof(*nrec));
1321     nrec->odr = odr_createmem(ODR_DECODE);
1322     nrec->wrbuf_marc = 0;
1323     nrec->wrbuf_iconv = 0;
1324     nrec->wrbuf_opac = 0;
1325     odr_setbuf (nrec->odr, buf, size, 0);
1326     z_NamePlusRecord (nrec->odr, &nrec->npr, 0, 0);
1327
1328     odr_destroy (odr_enc);
1329     return nrec;
1330 }
1331
1332 ZOOM_API(ZOOM_record)
1333 ZOOM_resultset_record_immediate (ZOOM_resultset s,size_t pos)
1334 {
1335     return record_cache_lookup (s, pos);
1336 }
1337
1338 ZOOM_API(ZOOM_record)
1339 ZOOM_resultset_record (ZOOM_resultset r, size_t pos)
1340 {
1341     ZOOM_record rec = ZOOM_resultset_record_immediate(r, pos);
1342
1343     if (!rec)
1344     {
1345         ZOOM_resultset_retrieve (r, 1, pos, 1);
1346         rec = ZOOM_resultset_record_immediate (r, pos);
1347     }
1348     return rec;
1349 }
1350
1351 ZOOM_API(void)
1352 ZOOM_record_destroy (ZOOM_record rec)
1353 {
1354     if (!rec)
1355         return;
1356     if (rec->wrbuf_marc)
1357         wrbuf_free (rec->wrbuf_marc, 1);
1358     if (rec->wrbuf_iconv)
1359         wrbuf_free (rec->wrbuf_iconv, 1);
1360     if (rec->wrbuf_opac)
1361         wrbuf_free (rec->wrbuf_opac, 1);
1362     odr_destroy (rec->odr);
1363     xfree (rec);
1364 }
1365
1366 static const char *marc_iconv_return(ZOOM_record rec, int marc_type,
1367                                      int *len,
1368                                      const char *buf, int sz,
1369                                      const char *record_charset)
1370 {
1371     char to[40];
1372     char from[40];
1373     yaz_iconv_t cd = 0;
1374     yaz_marc_t mt = yaz_marc_create();
1375
1376     *from = '\0';
1377     strcpy(to, "UTF-8");
1378     if (record_charset && *record_charset)
1379     {
1380         /* Use "from,to" or just "from" */
1381         const char *cp =strchr(record_charset, ',');
1382         int clen = strlen(record_charset);
1383         if (cp && cp[1])
1384         {
1385             strncpy( to, cp+1, sizeof(to)-1);
1386             to[sizeof(to)-1] = '\0';
1387             clen = cp - record_charset;
1388         }
1389         if (clen > sizeof(from)-1)
1390             clen = sizeof(from)-1;
1391
1392         if (clen)
1393             strncpy(from, record_charset, clen);
1394         from[clen] = '\0';
1395     }
1396
1397     if (*from && *to)
1398     {
1399         cd = yaz_iconv_open(to, from);
1400         yaz_marc_iconv(mt, cd);
1401     }
1402
1403     yaz_marc_xml(mt, marc_type);
1404     if (!rec->wrbuf_marc)
1405         rec->wrbuf_marc = wrbuf_alloc();
1406     wrbuf_rewind (rec->wrbuf_marc);
1407     if (yaz_marc_decode_wrbuf (mt, buf, sz, rec->wrbuf_marc) > 0)
1408     {
1409         yaz_marc_destroy(mt);
1410         if (cd)
1411             yaz_iconv_close(cd);
1412         if (len)
1413             *len = wrbuf_len(rec->wrbuf_marc);
1414         return wrbuf_buf(rec->wrbuf_marc);
1415     }
1416     yaz_marc_destroy(mt);
1417     if (cd)
1418         yaz_iconv_close(cd);
1419     return 0;
1420 }
1421
1422 static const char *record_iconv_return(ZOOM_record rec, int *len,
1423                                        const char *buf, int sz,
1424                                        const char *record_charset)
1425 {
1426     char to[40];
1427     char from[40];
1428     yaz_iconv_t cd = 0;
1429
1430     *from = '\0';
1431     strcpy(to, "UTF-8");
1432     if (record_charset && *record_charset)
1433     {
1434         /* Use "from,to" or just "from" */
1435         const char *cp =strchr(record_charset, ',');
1436         int clen = strlen(record_charset);
1437         if (cp && cp[1])
1438         {
1439             strncpy( to, cp+1, sizeof(to)-1);
1440             to[sizeof(to)-1] = '\0';
1441             clen = cp - record_charset;
1442         }
1443         if (clen > sizeof(from)-1)
1444             clen = sizeof(from)-1;
1445
1446         if (clen)
1447             strncpy(from, record_charset, clen);
1448         from[clen] = '\0';
1449     }
1450
1451     if (*from && *to && (cd = yaz_iconv_open(to, from)))
1452     {
1453         char outbuf[12];
1454         size_t inbytesleft = sz;
1455         const char *inp = buf;
1456
1457         if (!rec->wrbuf_iconv)
1458             rec->wrbuf_iconv = wrbuf_alloc();
1459
1460         wrbuf_rewind(rec->wrbuf_iconv);
1461
1462         while (inbytesleft)
1463         {
1464             size_t outbytesleft = sizeof(outbuf);
1465             char *outp = outbuf;
1466             size_t r = yaz_iconv (cd, (char**) &inp,
1467                                   &inbytesleft,
1468                                   &outp, &outbytesleft);
1469             if (r == (size_t) (-1))
1470             {
1471                 int e = yaz_iconv_error(cd);
1472                 if (e != YAZ_ICONV_E2BIG)
1473                     break;
1474             }
1475             wrbuf_write(rec->wrbuf_iconv, outbuf, outp - outbuf);
1476         }
1477         wrbuf_puts(rec->wrbuf_iconv, "");
1478         buf = wrbuf_buf(rec->wrbuf_iconv);
1479         sz = wrbuf_len(rec->wrbuf_iconv);
1480         yaz_iconv_close(cd);
1481     }
1482     if (len)
1483         *len = sz;
1484     return buf;
1485 }
1486
1487 ZOOM_API(const char *)
1488 ZOOM_record_get (ZOOM_record rec, const char *type_spec, int *len)
1489 {
1490     char type[40];
1491     char charset[40];
1492     char xpath[512];
1493     const char *cp;
1494     int i;
1495     Z_NamePlusRecord *npr;
1496
1497     if (len)
1498         *len = 0; /* default return */
1499
1500     if (!rec)
1501         return 0;
1502     npr = rec->npr;
1503     if (!npr)
1504         return 0;
1505
1506     cp = type_spec;
1507     for (i = 0; cp[i] && i < sizeof(type)-1; i++)
1508     {
1509         if (cp[i] == ';' || cp[i] == ' ')
1510             break;
1511         type[i] = cp[i];
1512     }
1513     type[i] = '\0';
1514     charset[0] = '\0';
1515     while (type_spec[i] == ';')
1516     {
1517         i++;
1518         while (type_spec[i] == ' ')
1519             i++;
1520         if (!strncmp(type_spec+i, "charset=", 8))
1521         {
1522             cp = type_spec+i+8;
1523             for (i = 0; cp[i] && i < sizeof(charset)-1; i++)
1524             {
1525                 if (cp[i] == ';' || cp[i] == ' ')
1526                     break;
1527                 charset[i] = cp[i];
1528             }
1529             charset[i] = '\0';
1530         }
1531         else if (!strncmp(type_spec+i, "xpath=", 6))
1532         {
1533             cp = type_spec+i+6;
1534             for (i = 0; cp[i] && i < sizeof(xpath)-1; i++)
1535                 xpath[i] = cp[i];
1536             xpath[i] = '\0';
1537         }
1538         while (type_spec[i] == ' ')
1539             i++;
1540     }
1541     if (!strcmp (type, "database"))
1542     {
1543         if (len)
1544             *len = (npr->databaseName ? strlen(npr->databaseName) : 0);
1545         return npr->databaseName;
1546     }
1547     else if (!strcmp (type, "syntax"))
1548     {
1549         const char *desc = 0;
1550         if (npr->which == Z_NamePlusRecord_databaseRecord)
1551         {
1552             Z_External *r = (Z_External *) npr->u.databaseRecord;
1553             oident *ent = oid_getentbyoid(r->direct_reference);
1554             if (ent)
1555                 desc = ent->desc;
1556         }
1557         if (!desc)
1558             desc = "none";
1559         if (len)
1560             *len = strlen(desc);
1561         return desc;
1562     }
1563     if (npr->which != Z_NamePlusRecord_databaseRecord)
1564         return 0;
1565
1566     /* from now on - we have a database record .. */
1567     if (!strcmp (type, "render"))
1568     {
1569         Z_External *r = (Z_External *) npr->u.databaseRecord;
1570         oident *ent = oid_getentbyoid(r->direct_reference);
1571
1572         /* render bibliographic record .. */
1573         if (r->which == Z_External_OPAC)
1574         {
1575             r = r->u.opac->bibliographicRecord;
1576             if (!r)
1577                 return 0;
1578             ent = oid_getentbyoid(r->direct_reference);
1579         }
1580         if (r->which == Z_External_sutrs)
1581             return record_iconv_return(rec, len,
1582                                        r->u.sutrs->buf, r->u.sutrs->len,
1583                                        charset);
1584         else if (r->which == Z_External_octet)
1585         {
1586             const char *ret_buf;
1587             switch (ent->value)
1588             {
1589             case VAL_SOIF:
1590             case VAL_HTML:
1591             case VAL_SUTRS:
1592                 break;
1593             case VAL_TEXT_XML:
1594             case VAL_APPLICATION_XML:
1595                 break;
1596             default:
1597                 ret_buf = marc_iconv_return(
1598                     rec, YAZ_MARC_LINE, len,
1599                     (const char *) r->u.octet_aligned->buf,
1600                     r->u.octet_aligned->len,
1601                     charset);
1602                 if (ret_buf)
1603                     return ret_buf;
1604             }
1605             return record_iconv_return(rec, len,
1606                                        (const char *) r->u.octet_aligned->buf,
1607                                        r->u.octet_aligned->len,
1608                                        charset);
1609         }
1610         else if (r->which == Z_External_grs1)
1611         {
1612             if (!rec->wrbuf_marc)
1613                 rec->wrbuf_marc = wrbuf_alloc();
1614             wrbuf_rewind (rec->wrbuf_marc);
1615             yaz_display_grs1(rec->wrbuf_marc, r->u.grs1, 0);
1616             return record_iconv_return(rec, len,
1617                                        wrbuf_buf(rec->wrbuf_marc),
1618                                        wrbuf_len(rec->wrbuf_marc),
1619                                        charset);
1620         }
1621         return 0;
1622     }
1623     else if (!strcmp (type, "xml") || !strcmp(type, "oai"))
1624     {
1625         Z_External *r = (Z_External *) npr->u.databaseRecord;
1626         oident *ent = oid_getentbyoid(r->direct_reference);
1627
1628         /* render bibliographic record .. */
1629         if (r->which == Z_External_OPAC)
1630         {
1631             r = r->u.opac->bibliographicRecord;
1632             if (!r)
1633                 return 0;
1634             ent = oid_getentbyoid(r->direct_reference);
1635         }
1636
1637         if (r->which == Z_External_sutrs)
1638             return record_iconv_return(rec, len,
1639                                        (const char *) r->u.sutrs->buf,
1640                                        r->u.sutrs->len,
1641                                        charset);
1642         else if (r->which == Z_External_octet)
1643         {
1644             const char *ret_buf;
1645             int marc_decode_type = YAZ_MARC_MARCXML;
1646
1647             if (!strcmp(type, "oai"))
1648                 marc_decode_type = YAZ_MARC_OAIMARC;
1649             switch (ent->value)
1650             {
1651             case VAL_SOIF:
1652             case VAL_HTML:
1653             case VAL_SUTRS:
1654                 break;
1655             case VAL_TEXT_XML:
1656             case VAL_APPLICATION_XML:
1657                 break;
1658             default:
1659                 ret_buf = marc_iconv_return(
1660                     rec, marc_decode_type, len,
1661                     (const char *) r->u.octet_aligned->buf,
1662                     r->u.octet_aligned->len,
1663                     charset);
1664                 if (ret_buf)
1665                     return ret_buf;
1666             }
1667             return record_iconv_return(rec, len,
1668                                        (const char *) r->u.octet_aligned->buf,
1669                                        r->u.octet_aligned->len,
1670                                        charset);
1671         }
1672         else if (r->which == Z_External_grs1)
1673         {
1674             if (len) *len = 5;
1675             return "GRS-1";
1676         }
1677         return 0;
1678     }
1679     else if (!strcmp (type, "raw"))
1680     {
1681         Z_External *r = (Z_External *) npr->u.databaseRecord;
1682
1683         if (r->which == Z_External_sutrs)
1684         {
1685             if (len) *len = r->u.sutrs->len;
1686             return (const char *) r->u.sutrs->buf;
1687         }
1688         else if (r->which == Z_External_octet)
1689         {
1690             if (len) *len = r->u.octet_aligned->len;
1691             return (const char *) r->u.octet_aligned->buf;
1692         }
1693         else /* grs-1, explain, OPAC, ... */
1694         {
1695             if (len) *len = -1;
1696             return (const char *) npr->u.databaseRecord;
1697         }
1698         return 0;
1699     }
1700     else if (!strcmp (type, "ext"))
1701     {
1702         if (len) *len = -1;
1703         return (const char *) npr->u.databaseRecord;
1704     }
1705     else if (!strcmp (type, "opac"))
1706
1707     {
1708         Z_External *r = (Z_External *) npr->u.databaseRecord;
1709         if (r->which == Z_External_OPAC)
1710         {
1711             if (!rec->wrbuf_opac)
1712                 rec->wrbuf_opac = wrbuf_alloc();
1713             wrbuf_rewind (rec->wrbuf_opac);
1714             yaz_display_OPAC(rec->wrbuf_opac, r->u.opac, 0);
1715             return record_iconv_return(rec, len,
1716                                        wrbuf_buf(rec->wrbuf_opac),
1717                                        wrbuf_len(rec->wrbuf_opac),
1718                                        charset);
1719         }
1720     }
1721     return 0;
1722 }
1723
1724 static int strcmp_null(const char *v1, const char *v2)
1725 {
1726     if (!v1 && !v2)
1727         return 0;
1728     if (!v1 || !v2)
1729         return -1;
1730     return strcmp(v1, v2);
1731 }
1732
1733 static void record_cache_add (ZOOM_resultset r, Z_NamePlusRecord *npr,
1734                               int pos)
1735 {
1736     ZOOM_record_cache rc;
1737     const char *elementSetName =
1738         ZOOM_resultset_option_get (r, "elementSetName");
1739     const char *syntax =
1740         ZOOM_resultset_option_get (r, "preferredRecordSyntax");
1741
1742     ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_RECV_RECORD);
1743     ZOOM_connection_put_event(r->connection, event);
1744
1745     for (rc = r->record_cache; rc; rc = rc->next)
1746     {
1747         if (pos == rc->pos)
1748         {
1749             if (strcmp_null(r->schema, rc->schema))
1750                 continue;
1751             if (strcmp_null(elementSetName,rc->elementSetName))
1752                 continue;
1753             if (strcmp_null(syntax, rc->syntax))
1754                 continue;
1755             /* not destroying rc->npr (it's handled by nmem )*/
1756             rc->rec.npr = npr;
1757             /* keeping wrbuf_marc too */
1758             return;
1759         }
1760     }
1761     rc = (ZOOM_record_cache) odr_malloc (r->odr, sizeof(*rc));
1762     rc->rec.npr = npr;
1763     rc->rec.odr = 0;
1764     rc->rec.wrbuf_marc = 0;
1765     rc->rec.wrbuf_iconv = 0;
1766     rc->rec.wrbuf_opac = 0;
1767     if (elementSetName)
1768         rc->elementSetName = odr_strdup (r->odr, elementSetName);
1769     else
1770         rc->elementSetName = 0;
1771
1772     if (syntax)
1773         rc->syntax = odr_strdup (r->odr, syntax);
1774     else
1775         rc->syntax = 0;
1776
1777     if (r->schema)
1778         rc->schema = odr_strdup (r->odr, r->schema);
1779     else
1780         rc->schema = 0;
1781
1782     rc->pos = pos;
1783     rc->next = r->record_cache;
1784     r->record_cache = rc;
1785 }
1786
1787 static ZOOM_record record_cache_lookup (ZOOM_resultset r, int pos)
1788 {
1789     ZOOM_record_cache rc;
1790     const char *elementSetName =
1791         ZOOM_resultset_option_get (r, "elementSetName");
1792     const char *syntax =
1793         ZOOM_resultset_option_get (r, "preferredRecordSyntax");
1794
1795     for (rc = r->record_cache; rc; rc = rc->next)
1796     {
1797         if (pos == rc->pos)
1798         {
1799             if (strcmp_null(r->schema, rc->schema))
1800                 continue;
1801             if (strcmp_null(elementSetName,rc->elementSetName))
1802                 continue;
1803             if (strcmp_null(syntax, rc->syntax))
1804                 continue;
1805             return &rc->rec;
1806         }
1807     }
1808     return 0;
1809 }
1810
1811 static void handle_records (ZOOM_connection c, Z_Records *sr,
1812                             int present_phase)
1813 {
1814     ZOOM_resultset resultset;
1815
1816     if (!c->tasks)
1817         return ;
1818     switch (c->tasks->which)
1819     {
1820     case ZOOM_TASK_SEARCH:
1821         resultset = c->tasks->u.search.resultset;
1822         break;
1823     case ZOOM_TASK_RETRIEVE:
1824         resultset = c->tasks->u.retrieve.resultset;
1825         break;
1826     default:
1827         return;
1828     }
1829     if (sr && sr->which == Z_Records_NSD)
1830     {
1831         Z_DiagRec dr, *dr_p = &dr;
1832         dr.which = Z_DiagRec_defaultFormat;
1833         dr.u.defaultFormat = sr->u.nonSurrogateDiagnostic;
1834
1835         response_diag (c, dr_p);
1836     }
1837     else if (sr && sr->which == Z_Records_multipleNSD)
1838     {
1839         if (sr->u.multipleNonSurDiagnostics->num_diagRecs >= 1)
1840             response_diag(c, sr->u.multipleNonSurDiagnostics->diagRecs[0]);
1841         else
1842             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1843     }
1844     else
1845     {
1846         if (resultset->count + resultset->start > resultset->size)
1847             resultset->count = resultset->size - resultset->start;
1848         if (resultset->count < 0)
1849             resultset->count = 0;
1850         if (sr && sr->which == Z_Records_DBOSD)
1851         {
1852             int i;
1853             NMEM nmem = odr_extract_mem (c->odr_in);
1854             Z_NamePlusRecordList *p =
1855                 sr->u.databaseOrSurDiagnostics;
1856             for (i = 0; i<p->num_records; i++)
1857             {
1858                 record_cache_add (resultset, p->records[i],
1859                                   i+ resultset->start);
1860             }
1861             /* transfer our response to search_nmem .. we need it later */
1862             nmem_transfer (resultset->odr->mem, nmem);
1863             nmem_destroy (nmem);
1864             if (present_phase && p->num_records == 0)
1865             {
1866                 /* present response and we didn't get any records! */
1867                 set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1868             }
1869         }
1870         else if (present_phase)
1871         {
1872             /* present response and we didn't get any records! */
1873             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1874         }
1875     }
1876 }
1877
1878 static void handle_present_response (ZOOM_connection c, Z_PresentResponse *pr)
1879 {
1880     handle_records (c, pr->records, 1);
1881 }
1882
1883 static void handle_search_response (ZOOM_connection c, Z_SearchResponse *sr)
1884 {
1885     ZOOM_resultset resultset;
1886     ZOOM_Event event;
1887
1888     yaz_log (LOG_DEBUG, "got search response");
1889
1890     if (!c->tasks || c->tasks->which != ZOOM_TASK_SEARCH)
1891         return ;
1892
1893     event = ZOOM_Event_create(ZOOM_EVENT_RECV_SEARCH);
1894     ZOOM_connection_put_event(c, event);
1895
1896     resultset = c->tasks->u.search.resultset;
1897
1898     resultset->size = *sr->resultCount;
1899     handle_records (c, sr->records, 0);
1900 }
1901
1902 static void sort_response (ZOOM_connection c, Z_SortResponse *res)
1903 {
1904     if (res->diagnostics && res->num_diagnostics > 0)
1905         response_diag (c, res->diagnostics[0]);
1906 }
1907
1908 static int scan_response (ZOOM_connection c, Z_ScanResponse *res)
1909 {
1910     NMEM nmem = odr_extract_mem (c->odr_in);
1911     ZOOM_scanset scan;
1912
1913     if (!c->tasks || c->tasks->which != ZOOM_TASK_SCAN)
1914         return 0;
1915     scan = c->tasks->u.scan.scan;
1916
1917     if (res->entries && res->entries->nonsurrogateDiagnostics)
1918         response_diag(c, res->entries->nonsurrogateDiagnostics[0]);
1919     scan->scan_response = res;
1920     nmem_transfer (scan->odr->mem, nmem);
1921     if (res->stepSize)
1922         ZOOM_options_set_int (scan->options, "stepSize", *res->stepSize);
1923     if (res->positionOfTerm)
1924         ZOOM_options_set_int (scan->options, "position", *res->positionOfTerm);
1925     if (res->scanStatus)
1926         ZOOM_options_set_int (scan->options, "scanStatus", *res->scanStatus);
1927     if (res->numberOfEntriesReturned)
1928         ZOOM_options_set_int (scan->options, "number",
1929                               *res->numberOfEntriesReturned);
1930     nmem_destroy (nmem);
1931     return 1;
1932 }
1933
1934 static zoom_ret send_sort (ZOOM_connection c)
1935 {
1936     ZOOM_resultset  resultset;
1937
1938     if (!c->tasks || c->tasks->which != ZOOM_TASK_SEARCH)
1939         return zoom_complete;
1940
1941     resultset = c->tasks->u.search.resultset;
1942
1943     if (c->error)
1944     {
1945         resultset->r_sort_spec = 0;
1946         return zoom_complete;
1947     }
1948     if (resultset->r_sort_spec)
1949     {
1950         Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_sortRequest);
1951         Z_SortRequest *req = apdu->u.sortRequest;
1952
1953         req->num_inputResultSetNames = 1;
1954         req->inputResultSetNames = (Z_InternationalString **)
1955             odr_malloc (c->odr_out, sizeof(*req->inputResultSetNames));
1956         req->inputResultSetNames[0] =
1957             odr_strdup (c->odr_out, resultset->setname);
1958         req->sortedResultSetName = odr_strdup (c->odr_out, resultset->setname);
1959         req->sortSequence = resultset->r_sort_spec;
1960         resultset->r_sort_spec = 0;
1961         return send_APDU (c, apdu);
1962     }
1963     return zoom_complete;
1964 }
1965
1966 static zoom_ret send_present (ZOOM_connection c)
1967 {
1968     Z_APDU *apdu = 0;
1969     Z_PresentRequest *req = 0;
1970     int i = 0;
1971     const char *syntax = 0;
1972     const char *elementSetName = 0;
1973     ZOOM_resultset  resultset;
1974
1975     if (!c->tasks)
1976         return zoom_complete;
1977
1978     switch (c->tasks->which)
1979     {
1980     case ZOOM_TASK_SEARCH:
1981         resultset = c->tasks->u.search.resultset;
1982         break;
1983     case ZOOM_TASK_RETRIEVE:
1984         resultset = c->tasks->u.retrieve.resultset;
1985         resultset->start = c->tasks->u.retrieve.start;
1986         resultset->count = c->tasks->u.retrieve.count;
1987
1988         if (resultset->start >= resultset->size)
1989             return zoom_complete;
1990         if (resultset->start + resultset->count > resultset->size)
1991             resultset->count = resultset->size - resultset->start;
1992         break;
1993     default:
1994         return zoom_complete;
1995     }
1996
1997     syntax = ZOOM_resultset_option_get (resultset, "preferredRecordSyntax");
1998     elementSetName = ZOOM_resultset_option_get (resultset, "elementSetName");
1999
2000     if (c->error)                  /* don't continue on error */
2001         return zoom_complete;
2002     if (resultset->start < 0)
2003         return zoom_complete;
2004     for (i = 0; i<resultset->count; i++)
2005     {
2006         ZOOM_record rec =
2007             record_cache_lookup (resultset, i + resultset->start);
2008         if (!rec)
2009             break;
2010     }
2011     if (i == resultset->count)
2012         return zoom_complete;
2013
2014     apdu = zget_APDU(c->odr_out, Z_APDU_presentRequest);
2015     req = apdu->u.presentRequest;
2016
2017     resultset->start += i;
2018     resultset->count -= i;
2019     *req->resultSetStartPoint = resultset->start + 1;
2020     *req->numberOfRecordsRequested = resultset->step>0 ?
2021         resultset->step : resultset->count;
2022     assert (*req->numberOfRecordsRequested > 0);
2023
2024     if (syntax && *syntax)
2025         req->preferredRecordSyntax =
2026             yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, syntax);
2027
2028     if (resultset->schema && *resultset->schema)
2029     {
2030         Z_RecordComposition *compo = (Z_RecordComposition *)
2031             odr_malloc (c->odr_out, sizeof(*compo));
2032
2033         req->recordComposition = compo;
2034         compo->which = Z_RecordComp_complex;
2035         compo->u.complex = (Z_CompSpec *)
2036             odr_malloc(c->odr_out, sizeof(*compo->u.complex));
2037         compo->u.complex->selectAlternativeSyntax = (bool_t *)
2038             odr_malloc(c->odr_out, sizeof(bool_t));
2039         *compo->u.complex->selectAlternativeSyntax = 0;
2040
2041         compo->u.complex->generic = (Z_Specification *)
2042             odr_malloc(c->odr_out, sizeof(*compo->u.complex->generic));
2043
2044         compo->u.complex->generic->which = Z_Schema_oid;
2045         compo->u.complex->generic->schema.oid = (Odr_oid *)
2046             yaz_str_to_z3950oid (c->odr_out, CLASS_SCHEMA, resultset->schema);
2047
2048         if (!compo->u.complex->generic->schema.oid)
2049         {
2050             /* OID wasn't a schema! Try record syntax instead. */
2051
2052             compo->u.complex->generic->schema.oid = (Odr_oid *)
2053                 yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, resultset->schema);
2054         }
2055         if (elementSetName && *elementSetName)
2056         {
2057             compo->u.complex->generic->elementSpec = (Z_ElementSpec *)
2058                 odr_malloc(c->odr_out, sizeof(Z_ElementSpec));
2059             compo->u.complex->generic->elementSpec->which =
2060                 Z_ElementSpec_elementSetName;
2061             compo->u.complex->generic->elementSpec->u.elementSetName =
2062                 odr_strdup (c->odr_out, elementSetName);
2063         }
2064         else
2065             compo->u.complex->generic->elementSpec = 0;
2066         compo->u.complex->num_dbSpecific = 0;
2067         compo->u.complex->dbSpecific = 0;
2068         compo->u.complex->num_recordSyntax = 0;
2069         compo->u.complex->recordSyntax = 0;
2070     }
2071     else if (elementSetName && *elementSetName)
2072     {
2073         Z_ElementSetNames *esn = (Z_ElementSetNames *)
2074             odr_malloc (c->odr_out, sizeof(*esn));
2075         Z_RecordComposition *compo = (Z_RecordComposition *)
2076             odr_malloc (c->odr_out, sizeof(*compo));
2077
2078         esn->which = Z_ElementSetNames_generic;
2079         esn->u.generic = odr_strdup (c->odr_out, elementSetName);
2080         compo->which = Z_RecordComp_simple;
2081         compo->u.simple = esn;
2082         req->recordComposition = compo;
2083     }
2084     req->resultSetId = odr_strdup(c->odr_out, resultset->setname);
2085     return send_APDU (c, apdu);
2086 }
2087
2088 ZOOM_API(ZOOM_scanset)
2089 ZOOM_connection_scan (ZOOM_connection c, const char *start)
2090 {
2091     ZOOM_scanset scan = (ZOOM_scanset) xmalloc (sizeof(*scan));
2092
2093     scan->connection = c;
2094     scan->odr = odr_createmem (ODR_DECODE);
2095     scan->options = ZOOM_options_create_with_parent (c->options);
2096     scan->refcount = 1;
2097     scan->scan_response = 0;
2098
2099     if ((scan->termListAndStartPoint =
2100          p_query_scan(scan->odr, PROTO_Z3950, &scan->attributeSet,
2101                       start)))
2102     {
2103         ZOOM_task task = ZOOM_connection_add_task (c, ZOOM_TASK_SCAN);
2104         task->u.scan.scan = scan;
2105
2106         (scan->refcount)++;
2107         if (!c->async)
2108         {
2109             while (ZOOM_event (1, &c))
2110                 ;
2111         }
2112     }
2113     return scan;
2114 }
2115
2116 ZOOM_API(void)
2117 ZOOM_scanset_destroy (ZOOM_scanset scan)
2118 {
2119     if (!scan)
2120         return;
2121     (scan->refcount)--;
2122     if (scan->refcount == 0)
2123     {
2124         odr_destroy (scan->odr);
2125
2126         ZOOM_options_destroy (scan->options);
2127         xfree (scan);
2128     }
2129 }
2130
2131 static zoom_ret send_package (ZOOM_connection c)
2132 {
2133     ZOOM_Event event;
2134     if (!c->tasks)
2135         return zoom_complete;
2136     assert (c->tasks->which == ZOOM_TASK_PACKAGE);
2137
2138     event = ZOOM_Event_create (ZOOM_EVENT_SEND_APDU);
2139     ZOOM_connection_put_event (c, event);
2140
2141     return do_write_ex (c, c->tasks->u.package->buf_out,
2142                         c->tasks->u.package->len_out);
2143 }
2144
2145 static zoom_ret send_scan (ZOOM_connection c)
2146 {
2147     ZOOM_scanset scan;
2148     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_scanRequest);
2149     Z_ScanRequest *req = apdu->u.scanRequest;
2150     if (!c->tasks)
2151         return zoom_complete;
2152     assert (c->tasks->which == ZOOM_TASK_SCAN);
2153     scan = c->tasks->u.scan.scan;
2154
2155     req->termListAndStartPoint = scan->termListAndStartPoint;
2156     req->attributeSet = scan->attributeSet;
2157
2158     *req->numberOfTermsRequested =
2159         ZOOM_options_get_int(scan->options, "number", 10);
2160
2161     req->preferredPositionInResponse =
2162         odr_intdup (c->odr_out,
2163                     ZOOM_options_get_int(scan->options, "position", 1));
2164
2165     req->stepSize =
2166         odr_intdup (c->odr_out,
2167                     ZOOM_options_get_int(scan->options, "stepSize", 0));
2168
2169     req->databaseNames = set_DatabaseNames (c, scan->options,
2170                                             &req->num_databaseNames);
2171
2172     return send_APDU (c, apdu);
2173 }
2174
2175 ZOOM_API(size_t)
2176 ZOOM_scanset_size (ZOOM_scanset scan)
2177 {
2178     if (!scan || !scan->scan_response || !scan->scan_response->entries)
2179         return 0;
2180     return scan->scan_response->entries->num_entries;
2181 }
2182
2183 ZOOM_API(const char *)
2184 ZOOM_scanset_term (ZOOM_scanset scan, size_t pos,
2185                    int *occ, int *len)
2186 {
2187     const char *term = 0;
2188     size_t noent = ZOOM_scanset_size (scan);
2189     Z_ScanResponse *res = scan->scan_response;
2190
2191     *len = 0;
2192     *occ = 0;
2193     if (pos >= noent)
2194         return 0;
2195     if (res->entries->entries[pos]->which == Z_Entry_termInfo)
2196     {
2197         Z_TermInfo *t = res->entries->entries[pos]->u.termInfo;
2198
2199         if (t->term->which == Z_Term_general)
2200         {
2201             term = (const char *) t->term->u.general->buf;
2202             *len = t->term->u.general->len;
2203         }
2204         *occ = t->globalOccurrences ? *t->globalOccurrences : 0;
2205     }
2206     return term;
2207 }
2208
2209 ZOOM_API(const char *)
2210 ZOOM_scanset_display_term (ZOOM_scanset scan, size_t pos,
2211                            int *occ, int *len)
2212 {
2213     const char *term = 0;
2214     size_t noent = ZOOM_scanset_size (scan);
2215     Z_ScanResponse *res = scan->scan_response;
2216
2217     *len = 0;
2218     *occ = 0;
2219     if (pos >= noent)
2220         return 0;
2221     if (res->entries->entries[pos]->which == Z_Entry_termInfo)
2222     {
2223         Z_TermInfo *t = res->entries->entries[pos]->u.termInfo;
2224
2225         if (t->displayTerm)
2226         {
2227             term = t->displayTerm;
2228             *len = strlen(term);
2229         }
2230         else if (t->term->which == Z_Term_general)
2231         {
2232             term = (const char *) t->term->u.general->buf;
2233             *len = t->term->u.general->len;
2234         }
2235         *occ = t->globalOccurrences ? *t->globalOccurrences : 0;
2236     }
2237     return term;
2238 }
2239
2240 ZOOM_API(const char *)
2241 ZOOM_scanset_option_get (ZOOM_scanset scan, const char *key)
2242 {
2243     return ZOOM_options_get (scan->options, key);
2244 }
2245
2246 ZOOM_API(void)
2247 ZOOM_scanset_option_set (ZOOM_scanset scan, const char *key,
2248                               const char *val)
2249 {
2250     ZOOM_options_set (scan->options, key, val);
2251 }
2252
2253 static Z_APDU *create_es_package (ZOOM_package p, int type)
2254 {
2255     const char *str;
2256     Z_APDU *apdu = zget_APDU(p->odr_out, Z_APDU_extendedServicesRequest);
2257     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2258
2259     *req->function = Z_ExtendedServicesRequest_create;
2260
2261     str = ZOOM_options_get(p->options, "package-name");
2262     if (str && *str)
2263         req->packageName = nmem_strdup (p->odr_out->mem, str);
2264
2265     str = ZOOM_options_get(p->options, "user-id");
2266     if (str)
2267         req->userId = nmem_strdup (p->odr_out->mem, str);
2268
2269     req->packageType = yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2270                                               type);
2271
2272     str = ZOOM_options_get(p->options, "function");
2273     if (str)
2274     {
2275         if (!strcmp (str, "create"))
2276             *req->function = 1;
2277         if (!strcmp (str, "delete"))
2278             *req->function = 2;
2279         if (!strcmp (str, "modify"))
2280             *req->function = 3;
2281     }
2282     return apdu;
2283 }
2284
2285 static const char *ill_array_lookup (void *clientData, const char *idx)
2286 {
2287     ZOOM_package p = (ZOOM_package) clientData;
2288     return ZOOM_options_get (p->options, idx+4);
2289 }
2290
2291 static Z_External *encode_ill_request (ZOOM_package p)
2292 {
2293     ODR out = p->odr_out;
2294     ILL_Request *req;
2295     Z_External *r = 0;
2296     struct ill_get_ctl ctl;
2297
2298     ctl.odr = p->odr_out;
2299     ctl.clientData = p;
2300     ctl.f = ill_array_lookup;
2301
2302     req = ill_get_ILLRequest(&ctl, "ill", 0);
2303
2304     if (!ill_Request (out, &req, 0, 0))
2305     {
2306         int ill_request_size;
2307         char *ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
2308         if (ill_request_buf)
2309             odr_setbuf (out, ill_request_buf, ill_request_size, 1);
2310         return 0;
2311     }
2312     else
2313     {
2314         oident oid;
2315         int illRequest_size = 0;
2316         char *illRequest_buf = odr_getbuf (out, &illRequest_size, 0);
2317
2318         oid.proto = PROTO_GENERAL;
2319         oid.oclass = CLASS_GENERAL;
2320         oid.value = VAL_ISO_ILL_1;
2321
2322         r = (Z_External *) odr_malloc (out, sizeof(*r));
2323         r->direct_reference = odr_oiddup(out,oid_getoidbyent(&oid));
2324         r->indirect_reference = 0;
2325         r->descriptor = 0;
2326         r->which = Z_External_single;
2327
2328         r->u.single_ASN1_type = (Odr_oct *)
2329             odr_malloc (out, sizeof(*r->u.single_ASN1_type));
2330         r->u.single_ASN1_type->buf = (unsigned char*)
2331             odr_malloc (out, illRequest_size);
2332         r->u.single_ASN1_type->len = illRequest_size;
2333         r->u.single_ASN1_type->size = illRequest_size;
2334         memcpy (r->u.single_ASN1_type->buf, illRequest_buf, illRequest_size);
2335     }
2336     return r;
2337 }
2338
2339 static Z_ItemOrder *encode_item_order(ZOOM_package p)
2340 {
2341     Z_ItemOrder *req = (Z_ItemOrder *) odr_malloc (p->odr_out, sizeof(*req));
2342     const char *str;
2343
2344     req->which=Z_IOItemOrder_esRequest;
2345     req->u.esRequest = (Z_IORequest *)
2346         odr_malloc(p->odr_out,sizeof(Z_IORequest));
2347
2348     /* to keep part ... */
2349     req->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
2350         odr_malloc(p->odr_out,sizeof(Z_IOOriginPartToKeep));
2351     req->u.esRequest->toKeep->supplDescription = 0;
2352     req->u.esRequest->toKeep->contact = (Z_IOContact *)
2353         odr_malloc (p->odr_out, sizeof(*req->u.esRequest->toKeep->contact));
2354
2355     str = ZOOM_options_get(p->options, "contact-name");
2356     req->u.esRequest->toKeep->contact->name = str ?
2357         nmem_strdup (p->odr_out->mem, str) : 0;
2358
2359     str = ZOOM_options_get(p->options, "contact-phone");
2360     req->u.esRequest->toKeep->contact->phone = str ?
2361         nmem_strdup (p->odr_out->mem, str) : 0;
2362
2363     str = ZOOM_options_get(p->options, "contact-email");
2364     req->u.esRequest->toKeep->contact->email = str ?
2365         nmem_strdup (p->odr_out->mem, str) : 0;
2366
2367     req->u.esRequest->toKeep->addlBilling = 0;
2368
2369     /* not to keep part ... */
2370     req->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
2371         odr_malloc(p->odr_out,sizeof(Z_IOOriginPartNotToKeep));
2372
2373     str = ZOOM_options_get(p->options, "itemorder-setname");
2374     if (!str)
2375         str = "default";
2376
2377     if (!*str)
2378         req->u.esRequest->notToKeep->resultSetItem = 0;
2379     else
2380     {
2381         req->u.esRequest->notToKeep->resultSetItem = (Z_IOResultSetItem *)
2382            odr_malloc(p->odr_out, sizeof(Z_IOResultSetItem));
2383
2384         req->u.esRequest->notToKeep->resultSetItem->resultSetId =
2385            nmem_strdup (p->odr_out->mem, str);
2386         req->u.esRequest->notToKeep->resultSetItem->item =
2387             (int *) odr_malloc(p->odr_out, sizeof(int));
2388
2389         str = ZOOM_options_get(p->options, "itemorder-item");
2390         *req->u.esRequest->notToKeep->resultSetItem->item =
2391             (str ? atoi(str) : 1);
2392     }
2393     req->u.esRequest->notToKeep->itemRequest = encode_ill_request(p);
2394
2395     return req;
2396 }
2397
2398 Z_APDU *create_admin_package(ZOOM_package p, int type,
2399                              Z_ESAdminOriginPartToKeep **toKeepP,
2400                              Z_ESAdminOriginPartNotToKeep **notToKeepP)
2401 {
2402     Z_APDU *apdu = create_es_package (p, VAL_ADMINSERVICE);
2403     if (apdu)
2404     {
2405         Z_ESAdminOriginPartToKeep  *toKeep;
2406         Z_ESAdminOriginPartNotToKeep  *notToKeep;
2407         Z_External *r = (Z_External *) odr_malloc (p->odr_out, sizeof(*r));
2408         const char *first_db = "Default";
2409         int num_db;
2410         char **db = set_DatabaseNames(p->connection, p->options, &num_db);
2411         if (num_db > 0)
2412             first_db = db[0];
2413
2414         r->direct_reference =
2415             yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2416                                    VAL_ADMINSERVICE);
2417         r->descriptor = 0;
2418         r->indirect_reference = 0;
2419         r->which = Z_External_ESAdmin;
2420
2421         r->u.adminService = (Z_Admin *)
2422             odr_malloc(p->odr_out, sizeof(*r->u.adminService));
2423         r->u.adminService->which = Z_Admin_esRequest;
2424         r->u.adminService->u.esRequest = (Z_AdminEsRequest *)
2425             odr_malloc(p->odr_out, sizeof(*r->u.adminService->u.esRequest));
2426
2427         toKeep = r->u.adminService->u.esRequest->toKeep =
2428             (Z_ESAdminOriginPartToKeep *)
2429             odr_malloc(p->odr_out, sizeof(*r->u.adminService->u.esRequest->toKeep));
2430         toKeep->which=type;
2431         toKeep->databaseName = odr_strdup(p->odr_out, first_db);
2432         toKeep->u.create=odr_nullval();
2433         apdu->u.extendedServicesRequest->taskSpecificParameters = r;
2434
2435         r->u.adminService->u.esRequest->notToKeep = notToKeep =
2436             (Z_ESAdminOriginPartNotToKeep *)
2437             odr_malloc(p->odr_out,
2438                        sizeof(*r->u.adminService->u.esRequest->notToKeep));
2439         notToKeep->which=Z_ESAdminOriginPartNotToKeep_recordsWillFollow;
2440         notToKeep->u.recordsWillFollow=odr_nullval();
2441         if (toKeepP)
2442             *toKeepP = toKeep;
2443         if (notToKeepP)
2444             *notToKeepP = notToKeep;
2445     }
2446     return apdu;
2447 }
2448
2449 static Z_APDU *create_update_package(ZOOM_package p)
2450 {
2451     Z_APDU *apdu = 0;
2452     const char *first_db = "Default";
2453     int num_db;
2454     char **db = set_DatabaseNames(p->connection, p->options, &num_db);
2455     const char *action = ZOOM_options_get(p->options, "action");
2456     const char *recordId = ZOOM_options_get(p->options, "recordId");
2457     const char *record_buf = ZOOM_options_get(p->options, "record");
2458     const char *syntax_str = ZOOM_options_get(p->options, "syntax");
2459     int syntax_oid = VAL_NONE;
2460     int action_no = -1;
2461
2462     if (syntax_str)
2463         syntax_oid = oid_getvalbyname(syntax_str);
2464     if (!record_buf)
2465     {
2466         record_buf = "void";
2467         syntax_oid = VAL_SUTRS;
2468     }
2469     if (syntax_oid != VAL_NONE)
2470         syntax_oid = VAL_TEXT_XML;
2471
2472     if (num_db > 0)
2473         first_db = db[0];
2474
2475     if (!action)
2476         action = "specialUpdate";
2477
2478     if (!strcmp(action, "recordInsert"))
2479         action_no = Z_IUOriginPartToKeep_recordInsert;
2480     else if (!strcmp(action, "recordReplace"))
2481         action_no = Z_IUOriginPartToKeep_recordReplace;
2482     else if (!strcmp(action, "recordDelete"))
2483         action_no = Z_IUOriginPartToKeep_recordDelete;
2484     else if (!strcmp(action, "elementUpdate"))
2485         action_no = Z_IUOriginPartToKeep_elementUpdate;
2486     else if (!strcmp(action, "specialUpdate"))
2487         action_no = Z_IUOriginPartToKeep_specialUpdate;
2488     else
2489         return 0;
2490
2491     apdu = create_es_package (p, VAL_DBUPDATE);
2492     if (apdu)
2493     {
2494         Z_IUOriginPartToKeep *toKeep;
2495         Z_IUSuppliedRecords *notToKeep;
2496         Z_External *r = (Z_External *)
2497             odr_malloc (p->odr_out, sizeof(*r));
2498
2499         apdu->u.extendedServicesRequest->taskSpecificParameters = r;
2500
2501         r->direct_reference =
2502             yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2503                                    VAL_DBUPDATE);
2504         r->descriptor = 0;
2505         r->which = Z_External_update;
2506         r->indirect_reference = 0;
2507         r->u.update = (Z_IUUpdate *)
2508             odr_malloc(p->odr_out, sizeof(*r->u.update));
2509
2510         r->u.update->which = Z_IUUpdate_esRequest;
2511         r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
2512             odr_malloc(p->odr_out, sizeof(*r->u.update->u.esRequest));
2513         toKeep = r->u.update->u.esRequest->toKeep =
2514             (Z_IUOriginPartToKeep *)
2515             odr_malloc(p->odr_out, sizeof(*toKeep));
2516
2517         toKeep->databaseName = odr_strdup(p->odr_out, first_db);
2518         toKeep->schema = 0;
2519         toKeep->elementSetName = 0;
2520         toKeep->actionQualifier = 0;
2521         toKeep->action = odr_intdup(p->odr_out, action_no);
2522
2523         notToKeep = r->u.update->u.esRequest->notToKeep =
2524             (Z_IUSuppliedRecords *)
2525             odr_malloc(p->odr_out, sizeof(*notToKeep));
2526         notToKeep->num = 1;
2527         notToKeep->elements = (Z_IUSuppliedRecords_elem **)
2528             odr_malloc(p->odr_out, sizeof(*notToKeep->elements));
2529         notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
2530             odr_malloc(p->odr_out, sizeof(**notToKeep->elements));
2531         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2532         if (recordId)
2533         {
2534             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2535                 odr_malloc (p->odr_out, sizeof(Odr_oct));
2536             notToKeep->elements[0]->u.opaque->size = strlen(recordId);
2537             notToKeep->elements[0]->u.opaque->len = strlen(recordId);
2538             notToKeep->elements[0]->u.opaque->buf =
2539                 odr_strdup(p->odr_out, recordId);
2540         }
2541         else
2542             notToKeep->elements[0]->u.opaque = 0;
2543         notToKeep->elements[0]->supplementalId = 0;
2544         notToKeep->elements[0]->correlationInfo = 0;
2545         notToKeep->elements[0]->record =
2546             z_ext_record(p->odr_out, syntax_oid,
2547                          record_buf, strlen(record_buf));
2548     }
2549     if (0 && apdu)
2550     {
2551        ODR print = odr_createmem(ODR_PRINT);
2552
2553        z_APDU(print, &apdu, 0, 0);
2554        odr_destroy(print);
2555     }
2556     return apdu;
2557 }
2558
2559 ZOOM_API(void)
2560     ZOOM_package_send (ZOOM_package p, const char *type)
2561 {
2562     Z_APDU *apdu = 0;
2563     ZOOM_connection c;
2564     if (!p)
2565         return;
2566     c = p->connection;
2567     odr_reset (p->odr_out);
2568     xfree (p->buf_out);
2569     p->buf_out = 0;
2570     if (!strcmp(type, "itemorder"))
2571     {
2572         apdu = create_es_package (p, VAL_ITEMORDER);
2573         if (apdu)
2574         {
2575             Z_External *r = (Z_External *) odr_malloc (p->odr_out, sizeof(*r));
2576
2577             r->direct_reference =
2578                 yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2579                                        VAL_ITEMORDER);
2580             r->descriptor = 0;
2581             r->which = Z_External_itemOrder;
2582             r->indirect_reference = 0;
2583             r->u.itemOrder = encode_item_order (p);
2584
2585             apdu->u.extendedServicesRequest->taskSpecificParameters = r;
2586         }
2587     }
2588     if (!strcmp(type, "create"))  /* create database */
2589     {
2590         apdu = create_admin_package(p, Z_ESAdminOriginPartToKeep_create,
2591                                     0, 0);
2592     }
2593     if (!strcmp(type, "drop"))  /* drop database */
2594     {
2595         apdu = create_admin_package(p, Z_ESAdminOriginPartToKeep_drop,
2596                                     0, 0);
2597     }
2598     if (!strcmp(type, "update")) /* update record(s) */
2599     {
2600         apdu = create_update_package(p);
2601     }
2602     if (apdu)
2603     {
2604         if (encode_APDU(p->connection, apdu, p->odr_out) == 0)
2605         {
2606             char *buf;
2607
2608             ZOOM_task task = ZOOM_connection_add_task (c, ZOOM_TASK_PACKAGE);
2609             task->u.package = p;
2610             buf = odr_getbuf(p->odr_out, &p->len_out, 0);
2611             p->buf_out = (char *) xmalloc (p->len_out);
2612             memcpy (p->buf_out, buf, p->len_out);
2613
2614             (p->refcount)++;
2615             if (!c->async)
2616             {
2617                 while (ZOOM_event (1, &c))
2618                     ;
2619             }
2620         }
2621     }
2622 }
2623
2624 ZOOM_API(ZOOM_package)
2625     ZOOM_connection_package (ZOOM_connection c, ZOOM_options options)
2626 {
2627     ZOOM_package p = (ZOOM_package) xmalloc (sizeof(*p));
2628
2629     p->connection = c;
2630     p->odr_out = odr_createmem (ODR_ENCODE);
2631     p->options = ZOOM_options_create_with_parent2 (options, c->options);
2632     p->refcount = 1;
2633     p->buf_out = 0;
2634     p->len_out = 0;
2635     return p;
2636 }
2637
2638 ZOOM_API(void)
2639     ZOOM_package_destroy(ZOOM_package p)
2640 {
2641     if (!p)
2642         return;
2643     (p->refcount)--;
2644     if (p->refcount == 0)
2645     {
2646         odr_destroy (p->odr_out);
2647         xfree (p->buf_out);
2648
2649         ZOOM_options_destroy (p->options);
2650         xfree (p);
2651     }
2652 }
2653
2654 ZOOM_API(const char *)
2655 ZOOM_package_option_get (ZOOM_package p, const char *key)
2656 {
2657     return ZOOM_options_get (p->options, key);
2658 }
2659
2660
2661 ZOOM_API(void)
2662 ZOOM_package_option_set (ZOOM_package p, const char *key,
2663                               const char *val)
2664 {
2665     ZOOM_options_set (p->options, key, val);
2666 }
2667
2668 static int ZOOM_connection_exec_task (ZOOM_connection c)
2669 {
2670     ZOOM_task task = c->tasks;
2671     zoom_ret ret = zoom_complete;
2672
2673     if (!task)
2674     {
2675         yaz_log (LOG_DEBUG, "ZOOM_connection_exec_task task=<null>");
2676         return 0;
2677     }
2678     yaz_log (LOG_DEBUG, "ZOOM_connection_exec_task type=%d run=%d",
2679              task->which, task->running);
2680     if (c->error != ZOOM_ERROR_NONE)
2681     {
2682         yaz_log (LOG_DEBUG, "remove tasks because of error = %d", c->error);
2683         ZOOM_connection_remove_tasks (c);
2684         return 0;
2685     }
2686     if (task->running)
2687     {
2688         yaz_log (LOG_DEBUG, "task already running");
2689         return 0;
2690     }
2691     task->running = 1;
2692     ret = zoom_complete;
2693     if (c->cs || task->which == ZOOM_TASK_CONNECT)
2694     {
2695         switch (task->which)
2696         {
2697         case ZOOM_TASK_SEARCH:
2698             if (c->proto == PROTO_HTTP)
2699                 ret = ZOOM_connection_srw_send_search(c);
2700             else
2701                 ret = ZOOM_connection_send_search(c);
2702             break;
2703         case ZOOM_TASK_RETRIEVE:
2704             if (c->proto == PROTO_HTTP)
2705                 ret = ZOOM_connection_srw_send_search(c);
2706             else
2707                 ret = send_present (c);
2708             break;
2709         case ZOOM_TASK_CONNECT:
2710             ret = do_connect(c);
2711             break;
2712         case ZOOM_TASK_SCAN:
2713             ret = send_scan(c);
2714             break;
2715         case ZOOM_TASK_PACKAGE:
2716             ret = send_package(c);
2717             break;
2718         }
2719     }
2720     else
2721     {
2722         yaz_log (LOG_DEBUG, "remove tasks because no connection exist");
2723         ZOOM_connection_remove_tasks (c);
2724     }
2725     if (ret == zoom_complete)
2726     {
2727         yaz_log (LOG_DEBUG, "task removed (complete)");
2728         ZOOM_connection_remove_task (c);
2729         return 0;
2730     }
2731     yaz_log (LOG_DEBUG, "task pending");
2732     return 1;
2733 }
2734
2735 static zoom_ret send_sort_present (ZOOM_connection c)
2736 {
2737     zoom_ret r = send_sort (c);
2738     if (r == zoom_complete)
2739         r = send_present (c);
2740     return r;
2741 }
2742
2743 static int es_response (ZOOM_connection c,
2744                         Z_ExtendedServicesResponse *res)
2745 {
2746     if (!c->tasks || c->tasks->which != ZOOM_TASK_PACKAGE)
2747         return 0;
2748     if (res->diagnostics && res->num_diagnostics > 0)
2749         response_diag(c, res->diagnostics[0]);
2750     if (res->taskPackage &&
2751         res->taskPackage->which == Z_External_extendedService)
2752     {
2753         Z_TaskPackage *taskPackage = res->taskPackage->u.extendedService;
2754         Odr_oct *id = taskPackage->targetReference;
2755
2756         if (id)
2757             ZOOM_options_setl (c->tasks->u.package->options,
2758                                "targetReference", (char*) id->buf, id->len);
2759     }
2760     return 1;
2761 }
2762
2763
2764 static void handle_apdu (ZOOM_connection c, Z_APDU *apdu)
2765 {
2766     Z_InitResponse *initrs;
2767
2768     c->mask = 0;
2769     yaz_log (LOG_DEBUG, "recv APDU type=%d", apdu->which);
2770     switch(apdu->which)
2771     {
2772     case Z_APDU_initResponse:
2773         initrs = apdu->u.initResponse;
2774         ZOOM_connection_option_set(c, "serverImplementationId",
2775                                    initrs->implementationId ?
2776                                    initrs->implementationId : "");
2777         ZOOM_connection_option_set(c, "serverImplementationName",
2778                                    initrs->implementationName ?
2779                                    initrs->implementationName : "");
2780         ZOOM_connection_option_set(c, "serverImplementationVersion",
2781                                    initrs->implementationVersion ?
2782                                    initrs->implementationVersion : "");
2783         /* Set the three old options too, for old applications */
2784         ZOOM_connection_option_set(c, "targetImplementationId",
2785                                    initrs->implementationId ?
2786                                    initrs->implementationId : "");
2787         ZOOM_connection_option_set(c, "targetImplementationName",
2788                                    initrs->implementationName ?
2789                                    initrs->implementationName : "");
2790         ZOOM_connection_option_set(c, "targetImplementationVersion",
2791                                    initrs->implementationVersion ?
2792                                    initrs->implementationVersion : "");
2793         if (!*initrs->result)
2794         {
2795             set_ZOOM_error(c, ZOOM_ERROR_INIT, 0);
2796         }
2797         else
2798         {
2799             char *cookie =
2800                 yaz_oi_get_string_oidval (&apdu->u.initResponse->otherInfo,
2801                                           VAL_COOKIE, 1, 0);
2802             xfree (c->cookie_in);
2803             c->cookie_in = 0;
2804             if (cookie)
2805                 c->cookie_in = xstrdup(cookie);
2806             if (ODR_MASK_GET(initrs->options, Z_Options_namedResultSets) &&
2807                 ODR_MASK_GET(initrs->protocolVersion, Z_ProtocolVersion_3))
2808                 c->support_named_resultsets = 1;
2809             if (c->tasks)
2810             {
2811                 assert (c->tasks->which == ZOOM_TASK_CONNECT);
2812                 ZOOM_connection_remove_task (c);
2813             }
2814             ZOOM_connection_exec_task (c);
2815         }
2816         if (ODR_MASK_GET(initrs->options, Z_Options_negotiationModel))
2817         {
2818             NMEM tmpmem = nmem_create();
2819             Z_CharSetandLanguageNegotiation *p =
2820                 yaz_get_charneg_record(initrs->otherInfo);
2821
2822             if (p)
2823             {
2824                 char *charset=NULL, *lang=NULL;
2825                 int sel;
2826
2827                 yaz_get_response_charneg(tmpmem, p, &charset, &lang, &sel);
2828                 yaz_log(LOG_DEBUG, "Target accepted: charset %s, "
2829                         "language %s, select %d",
2830                         charset ? charset : "none", lang ? lang : "none", sel);
2831                 if (charset)
2832                     ZOOM_connection_option_set (c, "negotiation-charset",
2833                                                 charset);
2834                 if (lang)
2835                     ZOOM_connection_option_set (c, "negotiation-lang",
2836                                                 lang);
2837                 nmem_destroy(tmpmem);
2838             }
2839         }
2840         break;
2841     case Z_APDU_searchResponse:
2842         handle_search_response (c, apdu->u.searchResponse);
2843         if (send_sort_present (c) == zoom_complete)
2844             ZOOM_connection_remove_task (c);
2845         break;
2846     case Z_APDU_presentResponse:
2847         handle_present_response (c, apdu->u.presentResponse);
2848         if (send_present (c) == zoom_complete)
2849             ZOOM_connection_remove_task (c);
2850         break;
2851     case Z_APDU_sortResponse:
2852         sort_response (c, apdu->u.sortResponse);
2853         if (send_present (c) == zoom_complete)
2854             ZOOM_connection_remove_task (c);
2855         break;
2856     case Z_APDU_scanResponse:
2857         scan_response (c, apdu->u.scanResponse);
2858         ZOOM_connection_remove_task (c);
2859         break;
2860     case Z_APDU_extendedServicesResponse:
2861         es_response (c, apdu->u.extendedServicesResponse);
2862         ZOOM_connection_remove_task (c);
2863         break;
2864     case Z_APDU_close:
2865         if (c->reconnect_ok)
2866         {
2867             do_close(c);
2868             c->tasks->running = 0;
2869             ZOOM_connection_insert_task (c, ZOOM_TASK_CONNECT);
2870         }
2871         else
2872         {
2873             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, 0);
2874             do_close(c);
2875         }
2876         break;
2877     default:
2878         set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
2879         do_close(c);
2880     }
2881 }
2882
2883 #if HAVE_XML2
2884 static void handle_srw_response(ZOOM_connection c,
2885                                 Z_SRW_searchRetrieveResponse *res)
2886 {
2887     ZOOM_resultset resultset = 0;
2888     int i;
2889     NMEM nmem;
2890     ZOOM_Event event;
2891
2892     if (!c->tasks)
2893         return;
2894
2895     if (c->tasks->which == ZOOM_TASK_SEARCH)
2896         resultset = c->tasks->u.search.resultset;
2897     else if (c->tasks->which == ZOOM_TASK_RETRIEVE)
2898         resultset = c->tasks->u.retrieve.resultset;
2899     else
2900         return ;
2901
2902     event = ZOOM_Event_create(ZOOM_EVENT_RECV_SEARCH);
2903     ZOOM_connection_put_event(c, event);
2904
2905     resultset->size = 0;
2906
2907     yaz_log(LOG_DEBUG, "got SRW response OK");
2908
2909     if (res->numberOfRecords)
2910         resultset->size = *res->numberOfRecords;
2911
2912     for (i = 0; i<res->num_records; i++)
2913     {
2914         int pos;
2915
2916         Z_NamePlusRecord *npr = (Z_NamePlusRecord *)
2917             odr_malloc(c->odr_in, sizeof(Z_NamePlusRecord));
2918
2919         if (res->records[i].recordPosition &&
2920             *res->records[i].recordPosition > 0)
2921             pos = *res->records[i].recordPosition - 1;
2922         else
2923             pos = resultset->start + i;
2924
2925         npr->databaseName = 0;
2926         npr->which = Z_NamePlusRecord_databaseRecord;
2927         npr->u.databaseRecord = (Z_External *)
2928             odr_malloc(c->odr_in, sizeof(Z_External));
2929         npr->u.databaseRecord->descriptor = 0;
2930         npr->u.databaseRecord->direct_reference =
2931             yaz_oidval_to_z3950oid(c->odr_in, CLASS_RECSYN, VAL_TEXT_XML);
2932         npr->u.databaseRecord->which = Z_External_octet;
2933         npr->u.databaseRecord->u.octet_aligned = (Odr_oct *)
2934             odr_malloc(c->odr_in, sizeof(Odr_oct));
2935         npr->u.databaseRecord->u.octet_aligned->buf =
2936             res->records[i].recordData_buf;
2937         npr->u.databaseRecord->u.octet_aligned->len =
2938             npr->u.databaseRecord->u.octet_aligned->size =
2939             res->records[i].recordData_len;
2940         record_cache_add (resultset, npr, pos);
2941     }
2942     if (res->num_diagnostics > 0)
2943     {
2944         set_dset_error(c, *res->diagnostics[0].code, "SRW",
2945                        res->diagnostics[0].details, 0);
2946     }
2947     nmem = odr_extract_mem(c->odr_in);
2948     nmem_transfer(resultset->odr->mem, nmem);
2949     nmem_destroy(nmem);
2950 }
2951 #endif
2952
2953 #if HAVE_XML2
2954 static void handle_http(ZOOM_connection c, Z_HTTP_Response *hres)
2955 {
2956     int ret = -1;
2957     const char *content_type = z_HTTP_header_lookup(hres->headers,
2958                                                     "Content-Type");
2959     const char *connection_head = z_HTTP_header_lookup(hres->headers,
2960                                                        "Connection");
2961     c->mask = 0;
2962     yaz_log (LOG_DEBUG, "handle_http");
2963
2964     if (content_type && !yaz_strcmp_del("text/xml", content_type, "; "))
2965     {
2966         Z_SOAP *soap_package = 0;
2967         ODR o = c->odr_in;
2968         Z_SOAP_Handler soap_handlers[2] = {
2969             {"http://www.loc.gov/zing/srw/", 0,
2970              (Z_SOAP_fun) yaz_srw_codec},
2971             {0, 0, 0}
2972         };
2973         ret = z_soap_codec(o, &soap_package,
2974                            &hres->content_buf, &hres->content_len,
2975                            soap_handlers);
2976         if (!ret && soap_package->which == Z_SOAP_generic &&
2977             soap_package->u.generic->no == 0)
2978         {
2979             Z_SRW_PDU *sr = soap_package->u.generic->p;
2980             if (sr->which == Z_SRW_searchRetrieve_response)
2981                 handle_srw_response(c, sr->u.response);
2982             else
2983                 ret = -1;
2984         }
2985         else if (!ret && (soap_package->which == Z_SOAP_fault
2986                           || soap_package->which == Z_SOAP_error))
2987         {
2988             set_HTTP_error(c, hres->code,
2989                            soap_package->u.fault->fault_code,
2990                            soap_package->u.fault->fault_string);
2991         }
2992         else
2993             ret = -1;
2994     }
2995     if (ret)
2996     {
2997         if (hres->code != 200)
2998             set_HTTP_error(c, hres->code, 0, 0);
2999         else
3000             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
3001         do_close (c);
3002     }
3003     ZOOM_connection_remove_task(c);
3004     if (!strcmp(hres->version, "1.0"))
3005     {
3006         /* HTTP 1.0: only if Keep-Alive we stay alive.. */
3007         if (!connection_head || strcmp(connection_head, "Keep-Alive"))
3008             do_close(c);
3009     }
3010     else
3011     {
3012         /* HTTP 1.1: only if no close we stay alive .. */
3013         if (connection_head && !strcmp(connection_head, "close"))
3014             do_close(c);
3015     }
3016 }
3017 #endif
3018
3019 static int do_read (ZOOM_connection c)
3020 {
3021     int r, more;
3022     ZOOM_Event event;
3023
3024     event = ZOOM_Event_create (ZOOM_EVENT_RECV_DATA);
3025     ZOOM_connection_put_event (c, event);
3026
3027     r = cs_get (c->cs, &c->buf_in, &c->len_in);
3028     more = cs_more(c->cs);
3029     yaz_log (LOG_DEBUG, "do_read len=%d more=%d", r, more);
3030     if (r == 1)
3031         return 0;
3032     if (r <= 0)
3033     {
3034         if (c->reconnect_ok)
3035         {
3036             do_close (c);
3037             c->reconnect_ok = 0;
3038             yaz_log (LOG_DEBUG, "reconnect read");
3039             c->tasks->running = 0;
3040             ZOOM_connection_insert_task (c, ZOOM_TASK_CONNECT);
3041         }
3042         else
3043         {
3044             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, 0);
3045             do_close (c);
3046         }
3047     }
3048     else
3049     {
3050         Z_GDU *gdu;
3051         ZOOM_Event event;
3052
3053         odr_reset (c->odr_in);
3054         odr_setbuf (c->odr_in, c->buf_in, r, 0);
3055         event = ZOOM_Event_create (ZOOM_EVENT_RECV_APDU);
3056         ZOOM_connection_put_event (c, event);
3057
3058         if (!z_GDU (c->odr_in, &gdu, 0, 0))
3059         {
3060             int x;
3061             int err = odr_geterrorx(c->odr_in, &x);
3062             char msg[60];
3063             const char *element = odr_getelement(c->odr_in);
3064             sprintf (msg, "ODR code %d:%d element=%-20s",
3065                      err, x, element ? element : "<unknown>");
3066             set_ZOOM_error(c, ZOOM_ERROR_DECODE, msg);
3067             do_close (c);
3068         }
3069         else if (gdu->which == Z_GDU_Z3950)
3070             handle_apdu (c, gdu->u.z3950);
3071         else if (gdu->which == Z_GDU_HTTP_Response)
3072         {
3073 #if HAVE_XML2
3074             handle_http (c, gdu->u.HTTP_Response);
3075 #else
3076             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
3077             do_close (c);
3078 #endif
3079         }
3080         c->reconnect_ok = 0;
3081     }
3082     return 1;
3083 }
3084
3085 static zoom_ret do_write_ex (ZOOM_connection c, char *buf_out, int len_out)
3086 {
3087     int r;
3088     ZOOM_Event event;
3089
3090     event = ZOOM_Event_create(ZOOM_EVENT_SEND_DATA);
3091     ZOOM_connection_put_event (c, event);
3092
3093     yaz_log (LOG_DEBUG, "do_write_ex len=%d", len_out);
3094     if ((r=cs_put (c->cs, buf_out, len_out)) < 0)
3095     {
3096         if (c->reconnect_ok)
3097         {
3098             do_close (c);
3099             c->reconnect_ok = 0;
3100             yaz_log (LOG_DEBUG, "reconnect write");
3101             c->tasks->running = 0;
3102             ZOOM_connection_insert_task (c, ZOOM_TASK_CONNECT);
3103             return zoom_pending;
3104         }
3105         if (c->state == STATE_CONNECTING)
3106             set_ZOOM_error(c, ZOOM_ERROR_CONNECT, 0);
3107         else
3108             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, 0);
3109         do_close (c);
3110         return zoom_complete;
3111     }
3112     else if (r == 1)
3113     {
3114         c->mask = ZOOM_SELECT_EXCEPT;
3115         if (c->cs->io_pending & CS_WANT_WRITE)
3116             c->mask += ZOOM_SELECT_WRITE;
3117         if (c->cs->io_pending & CS_WANT_READ)
3118             c->mask += ZOOM_SELECT_READ;
3119         yaz_log (LOG_DEBUG, "do_write_ex 1 mask=%d", c->mask);
3120     }
3121     else
3122     {
3123         c->mask = ZOOM_SELECT_READ|ZOOM_SELECT_EXCEPT;
3124         yaz_log (LOG_DEBUG, "do_write_ex 2 mask=%d", c->mask);
3125     }
3126     return zoom_pending;
3127 }
3128
3129 static zoom_ret do_write(ZOOM_connection c)
3130 {
3131     return do_write_ex (c, c->buf_out, c->len_out);
3132 }
3133
3134
3135 ZOOM_API(const char *)
3136 ZOOM_connection_option_get (ZOOM_connection c, const char *key)
3137 {
3138     return ZOOM_options_get (c->options, key);
3139 }
3140
3141 ZOOM_API(void)
3142 ZOOM_connection_option_set (ZOOM_connection c, const char *key,
3143                                   const char *val)
3144 {
3145     ZOOM_options_set (c->options, key, val);
3146 }
3147
3148 ZOOM_API(const char *)
3149 ZOOM_resultset_option_get (ZOOM_resultset r, const char *key)
3150 {
3151     return ZOOM_options_get (r->options, key);
3152 }
3153
3154 ZOOM_API(void)
3155 ZOOM_resultset_option_set (ZOOM_resultset r, const char *key,
3156                                   const char *val)
3157 {
3158     ZOOM_options_set (r->options, key, val);
3159 }
3160
3161
3162 ZOOM_API(int)
3163 ZOOM_connection_errcode (ZOOM_connection c)
3164 {
3165     return ZOOM_connection_error (c, 0, 0);
3166 }
3167
3168 ZOOM_API(const char *)
3169 ZOOM_connection_errmsg (ZOOM_connection c)
3170 {
3171     const char *msg;
3172     ZOOM_connection_error (c, &msg, 0);
3173     return msg;
3174 }
3175
3176 ZOOM_API(const char *)
3177 ZOOM_connection_addinfo (ZOOM_connection c)
3178 {
3179     const char *addinfo;
3180     ZOOM_connection_error (c, 0, &addinfo);
3181     return addinfo;
3182 }
3183
3184 ZOOM_API(const char *)
3185 ZOOM_diag_str (int error)
3186 {
3187     switch (error)
3188     {
3189     case ZOOM_ERROR_NONE:
3190         return "No error";
3191     case ZOOM_ERROR_CONNECT:
3192         return "Connect failed";
3193     case ZOOM_ERROR_MEMORY:
3194         return "Out of memory";
3195     case ZOOM_ERROR_ENCODE:
3196         return "Encoding failed";
3197     case ZOOM_ERROR_DECODE:
3198         return "Decoding failed";
3199     case ZOOM_ERROR_CONNECTION_LOST:
3200         return "Connection lost";
3201     case ZOOM_ERROR_INIT:
3202         return "Init rejected";
3203     case ZOOM_ERROR_INTERNAL:
3204         return "Internal failure";
3205     case ZOOM_ERROR_TIMEOUT:
3206         return "Timeout";
3207     case ZOOM_ERROR_UNSUPPORTED_PROTOCOL:
3208         return "Unsupported protocol";
3209     case ZOOM_ERROR_UNSUPPORTED_QUERY:
3210         return "Unsupported query type";
3211     default:
3212         return diagbib1_str (error);
3213     }
3214 }
3215
3216 ZOOM_API(int)
3217 ZOOM_connection_error_x (ZOOM_connection c, const char **cp,
3218                          const char **addinfo, const char **diagset)
3219 {
3220     int error = c->error;
3221     if (cp)
3222     {
3223         if (!c->diagset || !strcmp(c->diagset, "ZOOM"))
3224             *cp = ZOOM_diag_str(error);
3225         else if (!strcmp(c->diagset, "HTTP"))
3226             *cp = z_HTTP_errmsg(c->error);
3227         else if (!strcmp(c->diagset, "Bib-1"))
3228             *cp = ZOOM_diag_str(error);
3229         else if (!strcmp(c->diagset, "SRW"))
3230             *cp = yaz_diag_srw_str(c->error);
3231         else
3232             *cp = "Unknown error and diagnostic set";
3233     }
3234     if (addinfo)
3235         *addinfo = c->addinfo ? c->addinfo : "";
3236     if (diagset)
3237         *diagset = c->diagset ? c->diagset : "";
3238     return c->error;
3239 }
3240
3241 ZOOM_API(int)
3242 ZOOM_connection_error (ZOOM_connection c, const char **cp,
3243                        const char **addinfo)
3244 {
3245     return ZOOM_connection_error_x(c, cp, addinfo, 0);
3246 }
3247
3248 static int ZOOM_connection_do_io(ZOOM_connection c, int mask)
3249 {
3250     ZOOM_Event event = 0;
3251     int r = cs_look(c->cs);
3252     yaz_log (LOG_DEBUG, "ZOOM_connection_do_io c=%p mask=%d cs_look=%d",
3253              c, mask, r);
3254
3255     if (r == CS_NONE)
3256     {
3257         event = ZOOM_Event_create (ZOOM_EVENT_CONNECT);
3258         set_ZOOM_error(c, ZOOM_ERROR_CONNECT, 0);
3259         do_close (c);
3260         ZOOM_connection_put_event (c, event);
3261     }
3262     else if (r == CS_CONNECT)
3263     {
3264         int ret;
3265         event = ZOOM_Event_create (ZOOM_EVENT_CONNECT);
3266
3267         ret = cs_rcvconnect (c->cs);
3268         yaz_log (LOG_DEBUG, "cs_rcvconnect returned %d", ret);
3269         if (ret == 1)
3270         {
3271             c->mask = ZOOM_SELECT_EXCEPT;
3272             if (c->cs->io_pending & CS_WANT_WRITE)
3273                 c->mask += ZOOM_SELECT_WRITE;
3274             if (c->cs->io_pending & CS_WANT_READ)
3275                 c->mask += ZOOM_SELECT_READ;
3276             ZOOM_connection_put_event (c, event);
3277         }
3278         else if (ret == 0)
3279         {
3280             ZOOM_connection_put_event (c, event);
3281             if (c->proto == PROTO_Z3950)
3282                 ZOOM_connection_send_init(c);
3283             else
3284             {
3285                 /* no init request for SRW .. */
3286                 assert (c->tasks->which == ZOOM_TASK_CONNECT);
3287                 ZOOM_connection_remove_task (c);
3288                 c->mask = 0;
3289                 ZOOM_connection_exec_task (c);
3290             }
3291             c->state = STATE_ESTABLISHED;
3292         }
3293         else
3294         {
3295             set_ZOOM_error(c, ZOOM_ERROR_CONNECT, 0);
3296             do_close (c);
3297             ZOOM_connection_put_event (c, event);
3298         }
3299     }
3300     else
3301     {
3302         if (mask & ZOOM_SELECT_READ)
3303             do_read (c);
3304         if (c->cs && (mask & ZOOM_SELECT_WRITE))
3305             do_write (c);
3306     }
3307     return 1;
3308 }
3309
3310 ZOOM_API(int)
3311 ZOOM_connection_last_event(ZOOM_connection cs)
3312 {
3313     if (!cs)
3314         return ZOOM_EVENT_NONE;
3315     return cs->last_event;
3316 }
3317
3318 ZOOM_API(int)
3319 ZOOM_event (int no, ZOOM_connection *cs)
3320 {
3321     int timeout = 5000;
3322 #if HAVE_SYS_POLL_H
3323     struct pollfd pollfds[1024];
3324     ZOOM_connection poll_cs[1024];
3325 #else
3326     struct timeval tv;
3327     fd_set input, output, except;
3328 #endif
3329     int i, r, nfds;
3330     int max_fd = 0;
3331
3332     for (i = 0; i<no; i++)
3333     {
3334         ZOOM_connection c = cs[i];
3335         ZOOM_Event event;
3336         if (c && (event = ZOOM_connection_get_event(c)))
3337         {
3338             ZOOM_Event_destroy (event);
3339             return i+1;
3340         }
3341     }
3342     for (i = 0; i<no; i++)
3343     {
3344         ZOOM_connection c = cs[i];
3345         ZOOM_Event event;
3346         if (c && ZOOM_connection_exec_task (c))
3347         {
3348             if ((event = ZOOM_connection_get_event(c)))
3349             {
3350                 ZOOM_Event_destroy (event);
3351                 return i+1;
3352             }
3353         }
3354     }
3355 #if HAVE_SYS_POLL_H
3356
3357 #else
3358     FD_ZERO (&input);
3359     FD_ZERO (&output);
3360     FD_ZERO (&except);
3361 #endif
3362     nfds = 0;
3363     for (i = 0; i<no; i++)
3364     {
3365         ZOOM_connection c = cs[i];
3366         int fd, mask;
3367         int this_timeout;
3368
3369         if (!c)
3370             continue;
3371         fd = z3950_connection_socket(c);
3372         mask = z3950_connection_mask(c);
3373
3374         if (fd == -1)
3375             continue;
3376         if (max_fd < fd)
3377             max_fd = fd;
3378
3379         this_timeout = ZOOM_options_get_int (c->options, "timeout", -1);
3380         if (this_timeout != -1 && this_timeout < timeout)
3381             timeout = this_timeout;
3382 #if HAVE_SYS_POLL_H
3383         if (mask)
3384         {
3385             short poll_events = 0;
3386
3387             if (mask & ZOOM_SELECT_READ)
3388                 poll_events += POLLIN;
3389             if (mask & ZOOM_SELECT_WRITE)
3390                 poll_events += POLLOUT;
3391             if (mask & ZOOM_SELECT_EXCEPT)
3392                 poll_events += POLLERR;
3393             pollfds[nfds].fd = fd;
3394             pollfds[nfds].events = poll_events;
3395             pollfds[nfds].revents = 0;
3396             poll_cs[nfds] = c;
3397             nfds++;
3398         }
3399 #else
3400         if (mask & ZOOM_SELECT_READ)
3401         {
3402             FD_SET (fd, &input);
3403             nfds++;
3404         }
3405         if (mask & ZOOM_SELECT_WRITE)
3406         {
3407             FD_SET (fd, &output);
3408             nfds++;
3409         }
3410         if (mask & ZOOM_SELECT_EXCEPT)
3411         {
3412             FD_SET (fd, &except);
3413             nfds++;
3414         }
3415 #endif
3416     }
3417     if (timeout >= 5000)
3418         timeout = 30;
3419
3420     if (!nfds)
3421         return 0;
3422
3423 #if HAVE_SYS_POLL_H
3424     r = poll (pollfds, nfds, timeout * 1000);
3425     for (i = 0; i<nfds; i++)
3426     {
3427         ZOOM_connection c = poll_cs[i];
3428         if (r && c->mask)
3429         {
3430             int mask = 0;
3431             if (pollfds[i].revents & POLLIN)
3432                 mask += ZOOM_SELECT_READ;
3433             if (pollfds[i].revents & POLLOUT)
3434                 mask += ZOOM_SELECT_WRITE;
3435             if (pollfds[i].revents & POLLERR)
3436                 mask += ZOOM_SELECT_EXCEPT;
3437             if (mask)
3438                 ZOOM_connection_do_io(c, mask);
3439         }
3440         else if (r == 0 && c->mask)
3441         {
3442             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_TIMEOUT);
3443             /* timeout and this connection was waiting */
3444             set_ZOOM_error(c, ZOOM_ERROR_TIMEOUT, 0);
3445             do_close (c);
3446             ZOOM_connection_put_event(c, event);
3447         }
3448     }
3449 #else
3450     tv.tv_sec = timeout;
3451     tv.tv_usec = 0;
3452     yaz_log (LOG_DEBUG, "select start");
3453     r = select (max_fd+1, &input, &output, &except, &tv);
3454     yaz_log (LOG_DEBUG, "select stop, returned r=%d", r);
3455     for (i = 0; i<no; i++)
3456     {
3457         ZOOM_connection c = cs[i];
3458         int fd, mask;
3459
3460         if (!c)
3461             continue;
3462         fd = z3950_connection_socket(c);
3463         mask = 0;
3464         if (r && c->mask)
3465         {
3466             /* no timeout and real socket */
3467             if (FD_ISSET(fd, &input))
3468                 mask += ZOOM_SELECT_READ;
3469             if (FD_ISSET(fd, &output))
3470                 mask += ZOOM_SELECT_WRITE;
3471             if (FD_ISSET(fd, &except))
3472                 mask += ZOOM_SELECT_EXCEPT;
3473             if (mask)
3474                 ZOOM_connection_do_io(c, mask);
3475         }
3476         if (r == 0 && c->mask)
3477         {
3478             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_TIMEOUT);
3479             /* timeout and this connection was waiting */
3480             set_ZOOM_error(c, ZOOM_ERROR_TIMEOUT, 0);
3481             do_close (c);
3482             yaz_log (LOG_DEBUG, "timeout");
3483             ZOOM_connection_put_event(c, event);
3484         }
3485     }
3486 #endif
3487     for (i = 0; i<no; i++)
3488     {
3489         ZOOM_connection c = cs[i];
3490         ZOOM_Event event;
3491         if (c && (event = ZOOM_connection_get_event(c)))
3492         {
3493             ZOOM_Event_destroy (event);
3494             return i+1;
3495         }
3496     }
3497     return 0;
3498 }