src/zoom-c.c

   1 /*
   2  * Copyright (c) 2000-2003, Index Data
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: zoom-c.c,v 1.10 2003-12-04 12:57:30 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, "pass");
 899     char *version;
 900
 901     ODR_MASK_SET(ireq->options, Z_Options_search);
 902     ODR_MASK_SET(ireq->options, Z_Options_present);
 903     ODR_MASK_SET(ireq->options, Z_Options_scan);
 904     ODR_MASK_SET(ireq->options, Z_Options_sort);
 905     ODR_MASK_SET(ireq->options, Z_Options_extendedServices);
 906     ODR_MASK_SET(ireq->options, Z_Options_namedResultSets);
 907
 908     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_1);
 909     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_2);
 910     ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_3);
 911
 912     /* Index Data's Z39.50 Implementor Id is 81 */
 913     ireq->implementationId = odr_prepend(c->odr_out,
 914         ZOOM_options_get(c->options, "implementationId"),
 915         odr_prepend(c->odr_out, "81", ireq->implementationId));
 916
 917     ireq->implementationName = odr_prepend(c->odr_out,
 918         ZOOM_options_get(c->options, "implementationName"),
 919         odr_prepend(c->odr_out, "ZOOM-C", ireq->implementationName));
 920
 921     version = odr_strdup(c->odr_out, "$Revision: 1.10 $");
 922     if (strlen(version) > 10)   /* check for unexpanded CVS strings */
 923         version[strlen(version)-2] = '\0';
 924     ireq->implementationVersion = odr_prepend(c->odr_out,
 925         ZOOM_options_get(c->options, "implementationVersion"),
 926         odr_prepend(c->odr_out, &version[11], ireq->implementationVersion));
 927
 928     *ireq->maximumRecordSize =
 929         ZOOM_options_get_int (c->options, "maximumRecordSize", 1024*1024);
 930     *ireq->preferredMessageSize =
 931         ZOOM_options_get_int (c->options, "preferredMessageSize", 1024*1024);
 932
 933     if (auth_groupId || auth_password)
 934     {
 935         Z_IdPass *pass = (Z_IdPass *) odr_malloc(c->odr_out, sizeof(*pass));
 936         int i = 0;
 937         pass->groupId = 0;
 938         if (auth_groupId && *auth_groupId)
 939         {
 940             pass->groupId = (char *)
 941                 odr_malloc(c->odr_out, strlen(auth_groupId)+1);
 942             strcpy(pass->groupId, auth_groupId);
 943             i++;
 944         }
 945         pass->userId = 0;
 946         if (auth_userId && *auth_userId)
 947         {
 948             pass->userId = (char *)
 949                 odr_malloc(c->odr_out, strlen(auth_userId)+1);
 950             strcpy(pass->userId, auth_userId);
 951             i++;
 952         }
 953         pass->password = 0;
 954         if (auth_password && *auth_password)
 955         {
 956             pass->password = (char *)
 957                 odr_malloc(c->odr_out, strlen(auth_password)+1);
 958             strcpy(pass->password, auth_password);
 959             i++;
 960         }
 961         if (i)
 962         {
 963             auth->which = Z_IdAuthentication_idPass;
 964             auth->u.idPass = pass;
 965             ireq->idAuthentication = auth;
 966         }
 967     }
 968     else if (auth_userId)
 969     {
 970         auth->which = Z_IdAuthentication_open;
 971         auth->u.open = (char *)
 972             odr_malloc(c->odr_out, strlen(auth_userId)+1);
 973         strcpy(auth->u.open, auth_userId);
 974         ireq->idAuthentication = auth;
 975     }
 976     if (c->proxy)
 977         yaz_oi_set_string_oidval(&ireq->otherInfo, c->odr_out,
 978                                  VAL_PROXY, 1, c->host_port);
 979     if (c->charset||c->lang)
 980     {
 981         Z_OtherInformation **oi;
 982         Z_OtherInformationUnit *oi_unit;
 983
 984         yaz_oi_APDU(apdu, &oi);
 985
 986         if ((oi_unit = yaz_oi_update(oi, c->odr_out, NULL, 0, 0)))
 987         {
 988             ODR_MASK_SET(ireq->options, Z_Options_negotiationModel);
 989
 990             oi_unit->which = Z_OtherInfo_externallyDefinedInfo;
 991             oi_unit->information.externallyDefinedInfo =
 992                 yaz_set_proposal_charneg
 993                 (c->odr_out,
 994                  (const char **)&c->charset, (c->charset) ? 1:0,
 995                  (const char **)&c->lang, (c->lang) ? 1:0, 1);
 996         }
 997     }
 998     assert (apdu);
 999     return send_APDU (c, apdu);
1000 }
1001
1002 #if HAVE_XML2
1003 static zoom_ret send_srw (ZOOM_connection c, Z_SRW_PDU *sr)
1004 {
1005     char ctype[50];
1006     Z_SOAP_Handler h[2] = {
1007         {"http://www.loc.gov/zing/srw/v1.0/", 0, (Z_SOAP_fun) yaz_srw_codec},
1008         {0, 0, 0}
1009     };
1010     ODR o = odr_createmem(ODR_ENCODE);
1011     int ret;
1012     Z_SOAP *p = odr_malloc(o, sizeof(*p));
1013     Z_GDU *gdu;
1014     ZOOM_Event event;
1015
1016     gdu = z_get_HTTP_Request(c->odr_out);
1017     gdu->u.HTTP_Request->path = c->path;
1018
1019     if (c->host_port)
1020     {
1021         const char *cp0 = strstr(c->host_port, "://");
1022         const char *cp1 = 0;
1023         if (cp0)
1024             cp0 = cp0+3;
1025         else
1026             cp0 = c->host_port;
1027
1028         cp1 = strchr(cp0, '/');
1029         if (!cp1)
1030             cp1 = cp0+strlen(cp0);
1031
1032         if (cp0 && cp1)
1033         {
1034             char *h = odr_malloc(c->odr_out, cp1 - cp0 + 1);
1035             memcpy (h, cp0, cp1 - cp0);
1036             h[cp1-cp0] = '\0';
1037             z_HTTP_header_add(c->odr_out, &gdu->u.HTTP_Request->headers,
1038                               "Host", h);
1039         }
1040     }
1041
1042     strcpy(ctype, "text/xml");
1043     if (c->charset && strlen(c->charset) < 20)
1044     {
1045         strcat(ctype, "; charset=");
1046         strcat(ctype, c->charset);
1047     }
1048     z_HTTP_header_add(c->odr_out, &gdu->u.HTTP_Request->headers,
1049                       "Content-Type", ctype);
1050     z_HTTP_header_add(c->odr_out, &gdu->u.HTTP_Request->headers,
1051                       "SOAPAction", "\"\"");
1052     p->which = Z_SOAP_generic;
1053     p->u.generic = odr_malloc(o, sizeof(*p->u.generic));
1054     p->u.generic->no = 0;
1055     p->u.generic->ns = 0;
1056     p->u.generic->p = sr;
1057     p->ns = "http://schemas.xmlsoap.org/soap/envelope/";
1058
1059     ret = z_soap_codec_enc(o, &p,
1060                            &gdu->u.HTTP_Request->content_buf,
1061                            &gdu->u.HTTP_Request->content_len, h,
1062                            c->charset);
1063
1064     if (!z_GDU(c->odr_out, &gdu, 0, 0))
1065         return zoom_complete;
1066     c->buf_out = odr_getbuf(c->odr_out, &c->len_out, 0);
1067
1068     odr_destroy(o);
1069
1070     event = ZOOM_Event_create (ZOOM_EVENT_SEND_APDU);
1071     ZOOM_connection_put_event (c, event);
1072     odr_reset(c->odr_out);
1073     return do_write (c);
1074 }
1075 #endif
1076
1077 #if HAVE_XML2
1078 static zoom_ret ZOOM_connection_srw_send_search(ZOOM_connection c)
1079 {
1080     int i;
1081     ZOOM_resultset resultset = 0;
1082     Z_SRW_PDU *sr = 0;
1083     const char *recordPacking = 0;
1084
1085     if (c->error)                  /* don't continue on error */
1086         return zoom_complete;
1087     assert (c->tasks);
1088     if (c->tasks->which == ZOOM_TASK_SEARCH)
1089     {
1090         resultset = c->tasks->u.search.resultset;
1091         resultset->setname = xstrdup ("default");
1092         ZOOM_options_set (resultset->options, "setname", resultset->setname);
1093     }
1094     else if(c->tasks->which == ZOOM_TASK_RETRIEVE)
1095     {
1096         resultset = c->tasks->u.retrieve.resultset;
1097
1098         resultset->start = c->tasks->u.retrieve.start;
1099         resultset->count = c->tasks->u.retrieve.count;
1100
1101         if (resultset->start >= resultset->size)
1102             return zoom_complete;
1103         if (resultset->start + resultset->count > resultset->size)
1104             resultset->count = resultset->size - resultset->start;
1105
1106         for (i = 0; i<resultset->count; i++)
1107         {
1108             ZOOM_record rec =
1109                 record_cache_lookup (resultset, i + resultset->start);
1110             if (!rec)
1111                 break;
1112         }
1113         if (i == resultset->count)
1114             return zoom_complete;
1115     }
1116     assert(resultset->query);
1117
1118     sr = yaz_srw_get(c->odr_out, Z_SRW_searchRetrieve_request);
1119
1120     if (resultset->query->z_query->which == Z_Query_type_104
1121         && resultset->query->z_query->u.type_104->which == Z_External_CQL)
1122     {
1123
1124         sr->u.request->query_type = Z_SRW_query_type_cql;
1125         sr->u.request->query.cql =resultset->query->z_query->u.type_104->u.cql;
1126     }
1127     else if (resultset->query->z_query->which == Z_Query_type_1 &&
1128              resultset->query->z_query->u.type_1)
1129     {
1130         sr->u.request->query_type = Z_SRW_query_type_pqf;
1131         sr->u.request->query.pqf = resultset->query->query_string;
1132     }
1133     else
1134     {
1135         set_ZOOM_error(c, ZOOM_ERROR_UNSUPPORTED_QUERY, 0);
1136         return zoom_complete;
1137     }
1138     sr->u.request->startRecord = odr_intdup (c->odr_out, resultset->start + 1);
1139     sr->u.request->maximumRecords = odr_intdup (
1140         c->odr_out, resultset->step>0 ? resultset->step : resultset->count);
1141     sr->u.request->recordSchema = resultset->schema;
1142
1143     recordPacking = ZOOM_resultset_option_get (resultset, "recordPacking");
1144
1145     if (recordPacking)
1146         sr->u.request->recordPacking = odr_strdup(c->odr_out, recordPacking);
1147
1148     return send_srw(c, sr);
1149 }
1150 #else
1151 static zoom_ret ZOOM_connection_srw_send_search(ZOOM_connection c)
1152 {
1153     return zoom_complete;
1154 }
1155 #endif
1156
1157 static zoom_ret ZOOM_connection_send_search (ZOOM_connection c)
1158 {
1159     ZOOM_resultset r;
1160     int lslb, ssub, mspn;
1161     const char *syntax;
1162     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_searchRequest);
1163     Z_SearchRequest *search_req = apdu->u.searchRequest;
1164     const char *elementSetName;
1165     const char *smallSetElementSetName;
1166     const char *mediumSetElementSetName;
1167
1168     assert (c->tasks);
1169     assert (c->tasks->which == ZOOM_TASK_SEARCH);
1170
1171     r = c->tasks->u.search.resultset;
1172
1173     elementSetName =
1174         ZOOM_options_get (r->options, "elementSetName");
1175     smallSetElementSetName  =
1176         ZOOM_options_get (r->options, "smallSetElementSetName");
1177     mediumSetElementSetName =
1178         ZOOM_options_get (r->options, "mediumSetElementSetName");
1179
1180     if (!smallSetElementSetName)
1181         smallSetElementSetName = elementSetName;
1182
1183     if (!mediumSetElementSetName)
1184         mediumSetElementSetName = elementSetName;
1185
1186     assert (r);
1187     assert (r->query);
1188
1189     /* prepare query for the search request */
1190     search_req->query = r->query->z_query;
1191
1192     search_req->databaseNames =
1193         set_DatabaseNames (c, r->options, &search_req->num_databaseNames);
1194
1195     /* get syntax (no need to provide unless piggyback is in effect) */
1196     syntax = ZOOM_options_get (r->options, "preferredRecordSyntax");
1197
1198     lslb = ZOOM_options_get_int (r->options, "largeSetLowerBound", -1);
1199     ssub = ZOOM_options_get_int (r->options, "smallSetUpperBound", -1);
1200     mspn = ZOOM_options_get_int (r->options, "mediumSetPresentNumber", -1);
1201     if (lslb != -1 && ssub != -1 && mspn != -1)
1202     {
1203         /* So're a Z39.50 expert? Let's hope you don't do sort */
1204         *search_req->largeSetLowerBound = lslb;
1205         *search_req->smallSetUpperBound = ssub;
1206         *search_req->mediumSetPresentNumber = mspn;
1207     }
1208     else if (r->start == 0 && r->count > 0
1209              && r->piggyback && !r->r_sort_spec && !r->schema)
1210     {
1211         /* Regular piggyback - do it unless we're going to do sort */
1212         *search_req->largeSetLowerBound = 2000000000;
1213         *search_req->smallSetUpperBound = 1;
1214         *search_req->mediumSetPresentNumber = r->step>0 ? r->step : r->count;
1215     }
1216     else
1217     {
1218         /* non-piggyback. Need not provide elementsets or syntaxes .. */
1219         smallSetElementSetName = 0;
1220         mediumSetElementSetName = 0;
1221         syntax = 0;
1222     }
1223     if (smallSetElementSetName && *smallSetElementSetName)
1224     {
1225         Z_ElementSetNames *esn = (Z_ElementSetNames *)
1226             odr_malloc (c->odr_out, sizeof(*esn));
1227
1228         esn->which = Z_ElementSetNames_generic;
1229         esn->u.generic = odr_strdup (c->odr_out, smallSetElementSetName);
1230         search_req->smallSetElementSetNames = esn;
1231     }
1232     if (mediumSetElementSetName && *mediumSetElementSetName)
1233     {
1234         Z_ElementSetNames *esn = (Z_ElementSetNames *)
1235             odr_malloc (c->odr_out, sizeof(*esn));
1236
1237         esn->which = Z_ElementSetNames_generic;
1238         esn->u.generic = odr_strdup (c->odr_out, mediumSetElementSetName);
1239         search_req->mediumSetElementSetNames = esn;
1240     }
1241     if (syntax)
1242         search_req->preferredRecordSyntax =
1243             yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, syntax);
1244
1245     if (!r->setname)
1246     {
1247         if (c->support_named_resultsets)
1248         {
1249             char setname[14];
1250             int ord;
1251             /* find the lowest unused ordinal so that we re-use
1252                result sets on the server. */
1253             for (ord = 1; ; ord++)
1254             {
1255                 ZOOM_resultset rp;
1256                 sprintf (setname, "%d", ord);
1257                 for (rp = c->resultsets; rp; rp = rp->next)
1258                     if (rp->setname && !strcmp (rp->setname, setname))
1259                         break;
1260                 if (!rp)
1261                     break;
1262             }
1263             r->setname = xstrdup (setname);
1264             yaz_log (LOG_DEBUG, "allocating set %s", r->setname);
1265         }
1266         else
1267             r->setname = xstrdup ("default");
1268         ZOOM_options_set (r->options, "setname", r->setname);
1269     }
1270     search_req->resultSetName = odr_strdup(c->odr_out, r->setname);
1271     /* send search request */
1272     return send_APDU (c, apdu);
1273 }
1274
1275 static void response_diag (ZOOM_connection c, Z_DiagRec *p)
1276 {
1277     int oclass;
1278     Z_DefaultDiagFormat *r;
1279     char *addinfo = 0;
1280
1281     xfree (c->addinfo);
1282     c->addinfo = 0;
1283     if (p->which != Z_DiagRec_defaultFormat)
1284     {
1285         set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1286         return;
1287     }
1288     r = p->u.defaultFormat;
1289     switch (r->which)
1290     {
1291     case Z_DefaultDiagFormat_v2Addinfo:
1292         addinfo = r->u.v2Addinfo;
1293         break;
1294     case Z_DefaultDiagFormat_v3Addinfo:
1295         addinfo = r->u.v3Addinfo;
1296         break;
1297     }
1298     set_dset_error(c, *r->condition,
1299                    yaz_z3950oid_to_str(r->diagnosticSetId, &oclass),
1300                    addinfo, 0);
1301 }
1302
1303 ZOOM_API(ZOOM_record)
1304 ZOOM_record_clone (ZOOM_record srec)
1305 {
1306     char *buf;
1307     int size;
1308     ODR odr_enc;
1309     ZOOM_record nrec;
1310
1311     odr_enc = odr_createmem(ODR_ENCODE);
1312     if (!z_NamePlusRecord (odr_enc, &srec->npr, 0, 0))
1313         return 0;
1314     buf = odr_getbuf (odr_enc, &size, 0);
1315
1316     nrec = (ZOOM_record) xmalloc (sizeof(*nrec));
1317     nrec->odr = odr_createmem(ODR_DECODE);
1318     nrec->wrbuf_marc = 0;
1319     nrec->wrbuf_iconv = 0;
1320     nrec->wrbuf_opac = 0;
1321     odr_setbuf (nrec->odr, buf, size, 0);
1322     z_NamePlusRecord (nrec->odr, &nrec->npr, 0, 0);
1323
1324     odr_destroy (odr_enc);
1325     return nrec;
1326 }
1327
1328 ZOOM_API(ZOOM_record)
1329 ZOOM_resultset_record_immediate (ZOOM_resultset s,size_t pos)
1330 {
1331     return record_cache_lookup (s, pos);
1332 }
1333
1334 ZOOM_API(ZOOM_record)
1335 ZOOM_resultset_record (ZOOM_resultset r, size_t pos)
1336 {
1337     ZOOM_record rec = ZOOM_resultset_record_immediate(r, pos);
1338
1339     if (!rec)
1340     {
1341         ZOOM_resultset_retrieve (r, 1, pos, 1);
1342         rec = ZOOM_resultset_record_immediate (r, pos);
1343     }
1344     return rec;
1345 }
1346
1347 ZOOM_API(void)
1348 ZOOM_record_destroy (ZOOM_record rec)
1349 {
1350     if (!rec)
1351         return;
1352     if (rec->wrbuf_marc)
1353         wrbuf_free (rec->wrbuf_marc, 1);
1354     if (rec->wrbuf_iconv)
1355         wrbuf_free (rec->wrbuf_iconv, 1);
1356     if (rec->wrbuf_opac)
1357         wrbuf_free (rec->wrbuf_opac, 1);
1358     odr_destroy (rec->odr);
1359     xfree (rec);
1360 }
1361
1362 static const char *record_iconv_return(ZOOM_record rec, int *len,
1363                                        const char *buf, int sz,
1364                                        const char *record_charset)
1365 {
1366     char to[40];
1367     char from[40];
1368     yaz_iconv_t cd = 0;
1369
1370     *from = '\0';
1371     strcpy(to, "UTF-8");
1372     if (record_charset && *record_charset)
1373     {
1374         /* Use "from,to" or just "from" */
1375         const char *cp =strchr(record_charset, ',');
1376         int clen = strlen(record_charset);
1377         if (cp && cp[1])
1378         {
1379             strncpy( to, cp+1, sizeof(to)-1);
1380             to[sizeof(to)-1] = '\0';
1381             clen = cp - record_charset;
1382         }
1383         if (clen > sizeof(from)-1)
1384             clen = sizeof(from)-1;
1385
1386         if (clen)
1387             strncpy(from, record_charset, clen);
1388         from[clen] = '\0';
1389     }
1390
1391     if (*from && *to && (cd = yaz_iconv_open(to, from)))
1392     {
1393         char outbuf[12];
1394         size_t inbytesleft = sz;
1395         const char *inp = buf;
1396
1397         if (!rec->wrbuf_iconv)
1398             rec->wrbuf_iconv = wrbuf_alloc();
1399
1400         wrbuf_rewind(rec->wrbuf_iconv);
1401
1402         while (inbytesleft)
1403         {
1404             size_t outbytesleft = sizeof(outbuf);
1405             char *outp = outbuf;
1406             size_t r = yaz_iconv (cd, (char**) &inp,
1407                                   &inbytesleft,
1408                                   &outp, &outbytesleft);
1409             if (r == (size_t) (-1))
1410             {
1411                 int e = yaz_iconv_error(cd);
1412                 if (e != YAZ_ICONV_E2BIG)
1413                     break;
1414             }
1415             wrbuf_write(rec->wrbuf_iconv, outbuf, outp - outbuf);
1416         }
1417         wrbuf_puts(rec->wrbuf_iconv, "");
1418         buf = wrbuf_buf(rec->wrbuf_iconv);
1419         sz = wrbuf_len(rec->wrbuf_iconv);
1420         yaz_iconv_close(cd);
1421     }
1422     if (len)
1423         *len = sz;
1424     return buf;
1425 }
1426
1427 ZOOM_API(const char *)
1428 ZOOM_record_get (ZOOM_record rec, const char *type_spec, int *len)
1429 {
1430     char type[40];
1431     char charset[40];
1432     const char *cp;
1433     int i;
1434     Z_NamePlusRecord *npr;
1435
1436     if (len)
1437         *len = 0; /* default return */
1438
1439     if (!rec)
1440         return 0;
1441     npr = rec->npr;
1442     if (!npr)
1443         return 0;
1444
1445     cp = type_spec;
1446     for (i = 0; cp[i] && i < sizeof(type)-1; i++)
1447     {
1448         if (cp[i] == ';' || cp[i] == ' ')
1449             break;
1450         type[i] = cp[i];
1451     }
1452     type[i] = '\0';
1453     charset[0] = '\0';
1454     if (type_spec[i] == ';')
1455     {
1456         i++;
1457         while (type_spec[i] == ' ')
1458             i++;
1459         if (!strncmp(type_spec+i, "charset=", 8))
1460         {
1461             cp = type_spec+i+8;
1462             for (i = 0; cp[i] && i < sizeof(charset)-1; i++)
1463             {
1464                 if (cp[i] == ';' || cp[i] == ' ')
1465                     break;
1466                 charset[i] = cp[i];
1467             }
1468             charset[i] = '\0';
1469         }
1470     }
1471
1472     if (!strcmp (type, "database"))
1473     {
1474         if (len)
1475             *len = (npr->databaseName ? strlen(npr->databaseName) : 0);
1476         return npr->databaseName;
1477     }
1478     else if (!strcmp (type, "syntax"))
1479     {
1480         const char *desc = 0;
1481         if (npr->which == Z_NamePlusRecord_databaseRecord)
1482         {
1483             Z_External *r = (Z_External *) npr->u.databaseRecord;
1484             oident *ent = oid_getentbyoid(r->direct_reference);
1485             if (ent)
1486                 desc = ent->desc;
1487         }
1488         if (!desc)
1489             desc = "none";
1490         if (len)
1491             *len = strlen(desc);
1492         return desc;
1493     }
1494     if (npr->which != Z_NamePlusRecord_databaseRecord)
1495         return 0;
1496
1497     /* from now on - we have a database record .. */
1498     if (!strcmp (type, "render"))
1499     {
1500         Z_External *r = (Z_External *) npr->u.databaseRecord;
1501         oident *ent = oid_getentbyoid(r->direct_reference);
1502
1503         /* render bibliographic record .. */
1504         if (r->which == Z_External_OPAC)
1505         {
1506             r = r->u.opac->bibliographicRecord;
1507             if (!r)
1508                 return 0;
1509             ent = oid_getentbyoid(r->direct_reference);
1510         }
1511         if (r->which == Z_External_sutrs)
1512             return record_iconv_return(rec, len,
1513                                        r->u.sutrs->buf, r->u.sutrs->len,
1514                                        charset);
1515         else if (r->which == Z_External_octet)
1516         {
1517             yaz_marc_t mt;
1518             switch (ent->value)
1519             {
1520             case VAL_SOIF:
1521             case VAL_HTML:
1522             case VAL_SUTRS:
1523                 break;
1524             case VAL_TEXT_XML:
1525             case VAL_APPLICATION_XML:
1526                 break;
1527             default:
1528                 if (!rec->wrbuf_marc)
1529                     rec->wrbuf_marc = wrbuf_alloc();
1530
1531                 mt = yaz_marc_create();
1532                 wrbuf_rewind (rec->wrbuf_marc);
1533                 if (yaz_marc_decode_wrbuf (
1534                         mt, (const char *) r->u.octet_aligned->buf,
1535                         r->u.octet_aligned->len,
1536                         rec->wrbuf_marc) > 0)
1537                 {
1538                     yaz_marc_destroy(mt);
1539                     return record_iconv_return(rec, len,
1540                                                wrbuf_buf(rec->wrbuf_marc),
1541                                                wrbuf_len(rec->wrbuf_marc),
1542                                                charset);
1543                 }
1544                 yaz_marc_destroy(mt);
1545             }
1546             return record_iconv_return(rec, len,
1547                                        (const char *) r->u.octet_aligned->buf,
1548                                        r->u.octet_aligned->len,
1549                                        charset);
1550         }
1551         else if (r->which == Z_External_grs1)
1552         {
1553             if (!rec->wrbuf_marc)
1554                 rec->wrbuf_marc = wrbuf_alloc();
1555             wrbuf_rewind (rec->wrbuf_marc);
1556             yaz_display_grs1(rec->wrbuf_marc, r->u.grs1, 0);
1557             return record_iconv_return(rec, len,
1558                                        wrbuf_buf(rec->wrbuf_marc),
1559                                        wrbuf_len(rec->wrbuf_marc),
1560                                        charset);
1561         }
1562         return 0;
1563     }
1564     else if (!strcmp (type, "xml") || !strcmp(type, "oai"))
1565     {
1566         Z_External *r = (Z_External *) npr->u.databaseRecord;
1567         oident *ent = oid_getentbyoid(r->direct_reference);
1568
1569         /* render bibliographic record .. */
1570         if (r->which == Z_External_OPAC)
1571         {
1572             r = r->u.opac->bibliographicRecord;
1573             if (!r)
1574                 return 0;
1575             ent = oid_getentbyoid(r->direct_reference);
1576         }
1577
1578         if (r->which == Z_External_sutrs)
1579             return record_iconv_return(rec, len,
1580                                        (const char *) r->u.sutrs->buf,
1581                                        r->u.sutrs->len,
1582                                        charset);
1583         else if (r->which == Z_External_octet)
1584         {
1585             yaz_marc_t mt;
1586             int marc_decode_type = YAZ_MARC_MARCXML;
1587
1588             if (!strcmp(type, "oai"))
1589                 marc_decode_type = YAZ_MARC_OAIMARC;
1590             switch (ent->value)
1591             {
1592             case VAL_SOIF:
1593             case VAL_HTML:
1594             case VAL_SUTRS:
1595                 break;
1596             case VAL_TEXT_XML:
1597             case VAL_APPLICATION_XML:
1598                 break;
1599             default:
1600                 if (!rec->wrbuf_marc)
1601                     rec->wrbuf_marc = wrbuf_alloc();
1602                 wrbuf_rewind (rec->wrbuf_marc);
1603                 mt = yaz_marc_create();
1604
1605                 yaz_marc_xml(mt, YAZ_MARC_MARCXML);
1606                 if (yaz_marc_decode_wrbuf (
1607                         mt, (const char *) r->u.octet_aligned->buf,
1608                         r->u.octet_aligned->len,
1609                         rec->wrbuf_marc) > 0)
1610                 {
1611                     yaz_marc_destroy(mt);
1612                     return record_iconv_return(rec, len,
1613                                                wrbuf_buf(rec->wrbuf_marc),
1614                                                wrbuf_len(rec->wrbuf_marc),
1615                                                charset);
1616                 }
1617                 yaz_marc_destroy(mt);
1618             }
1619             return record_iconv_return(rec, len,
1620                                        (const char *) r->u.octet_aligned->buf,
1621                                        r->u.octet_aligned->len,
1622                                        charset);
1623         }
1624         else if (r->which == Z_External_grs1)
1625         {
1626             if (len) *len = 5;
1627             return "GRS-1";
1628         }
1629         return 0;
1630     }
1631     else if (!strcmp (type, "raw"))
1632     {
1633         Z_External *r = (Z_External *) npr->u.databaseRecord;
1634
1635         if (r->which == Z_External_sutrs)
1636         {
1637             if (len) *len = r->u.sutrs->len;
1638             return (const char *) r->u.sutrs->buf;
1639         }
1640         else if (r->which == Z_External_octet)
1641         {
1642             if (len) *len = r->u.octet_aligned->len;
1643             return (const char *) r->u.octet_aligned->buf;
1644         }
1645         else /* grs-1, explain, OPAC, ... */
1646         {
1647             if (len) *len = -1;
1648             return (const char *) npr->u.databaseRecord;
1649         }
1650         return 0;
1651     }
1652     else if (!strcmp (type, "ext"))
1653     {
1654         if (len) *len = -1;
1655         return (const char *) npr->u.databaseRecord;
1656     }
1657     else if (!strcmp (type, "opac"))
1658
1659     {
1660         Z_External *r = (Z_External *) npr->u.databaseRecord;
1661         if (r->which == Z_External_OPAC)
1662         {
1663             if (!rec->wrbuf_opac)
1664                 rec->wrbuf_opac = wrbuf_alloc();
1665             wrbuf_rewind (rec->wrbuf_opac);
1666             yaz_display_OPAC(rec->wrbuf_opac, r->u.opac, 0);
1667             return record_iconv_return(rec, len,
1668                                        wrbuf_buf(rec->wrbuf_opac),
1669                                        wrbuf_len(rec->wrbuf_opac),
1670                                        charset);
1671         }
1672     }
1673     return 0;
1674 }
1675
1676 static int strcmp_null(const char *v1, const char *v2)
1677 {
1678     if (!v1 && !v2)
1679         return 0;
1680     if (!v1 || !v2)
1681         return -1;
1682     return strcmp(v1, v2);
1683 }
1684
1685 static void record_cache_add (ZOOM_resultset r, Z_NamePlusRecord *npr,
1686                               int pos)
1687 {
1688     ZOOM_record_cache rc;
1689     const char *elementSetName =
1690         ZOOM_resultset_option_get (r, "elementSetName");
1691     const char *syntax =
1692         ZOOM_resultset_option_get (r, "preferredRecordSyntax");
1693
1694     ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_RECV_RECORD);
1695     ZOOM_connection_put_event(r->connection, event);
1696
1697     for (rc = r->record_cache; rc; rc = rc->next)
1698     {
1699         if (pos == rc->pos)
1700         {
1701             if (strcmp_null(r->schema, rc->schema))
1702                 continue;
1703             if (strcmp_null(elementSetName,rc->elementSetName))
1704                 continue;
1705             if (strcmp_null(syntax, rc->syntax))
1706                 continue;
1707             /* not destroying rc->npr (it's handled by nmem )*/
1708             rc->rec.npr = npr;
1709             /* keeping wrbuf_marc too */
1710             return;
1711         }
1712     }
1713     rc = (ZOOM_record_cache) odr_malloc (r->odr, sizeof(*rc));
1714     rc->rec.npr = npr;
1715     rc->rec.odr = 0;
1716     rc->rec.wrbuf_marc = 0;
1717     rc->rec.wrbuf_iconv = 0;
1718     rc->rec.wrbuf_opac = 0;
1719     if (elementSetName)
1720         rc->elementSetName = odr_strdup (r->odr, elementSetName);
1721     else
1722         rc->elementSetName = 0;
1723
1724     if (syntax)
1725         rc->syntax = odr_strdup (r->odr, syntax);
1726     else
1727         rc->syntax = 0;
1728
1729     if (r->schema)
1730         rc->schema = odr_strdup (r->odr, r->schema);
1731     else
1732         rc->schema = 0;
1733
1734     rc->pos = pos;
1735     rc->next = r->record_cache;
1736     r->record_cache = rc;
1737 }
1738
1739 static ZOOM_record record_cache_lookup (ZOOM_resultset r, int pos)
1740 {
1741     ZOOM_record_cache rc;
1742     const char *elementSetName =
1743         ZOOM_resultset_option_get (r, "elementSetName");
1744     const char *syntax =
1745         ZOOM_resultset_option_get (r, "preferredRecordSyntax");
1746
1747     for (rc = r->record_cache; rc; rc = rc->next)
1748     {
1749         if (pos == rc->pos)
1750         {
1751             if (strcmp_null(r->schema, rc->schema))
1752                 continue;
1753             if (strcmp_null(elementSetName,rc->elementSetName))
1754                 continue;
1755             if (strcmp_null(syntax, rc->syntax))
1756                 continue;
1757             return &rc->rec;
1758         }
1759     }
1760     return 0;
1761 }
1762
1763 static void handle_records (ZOOM_connection c, Z_Records *sr,
1764                             int present_phase)
1765 {
1766     ZOOM_resultset resultset;
1767
1768     if (!c->tasks)
1769         return ;
1770     switch (c->tasks->which)
1771     {
1772     case ZOOM_TASK_SEARCH:
1773         resultset = c->tasks->u.search.resultset;
1774         break;
1775     case ZOOM_TASK_RETRIEVE:
1776         resultset = c->tasks->u.retrieve.resultset;
1777         break;
1778     default:
1779         return;
1780     }
1781     if (sr && sr->which == Z_Records_NSD)
1782     {
1783         Z_DiagRec dr, *dr_p = &dr;
1784         dr.which = Z_DiagRec_defaultFormat;
1785         dr.u.defaultFormat = sr->u.nonSurrogateDiagnostic;
1786
1787         response_diag (c, dr_p);
1788     }
1789     else if (sr && sr->which == Z_Records_multipleNSD)
1790     {
1791         if (sr->u.multipleNonSurDiagnostics->num_diagRecs >= 1)
1792             response_diag(c, sr->u.multipleNonSurDiagnostics->diagRecs[0]);
1793         else
1794             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1795     }
1796     else
1797     {
1798         if (resultset->count + resultset->start > resultset->size)
1799             resultset->count = resultset->size - resultset->start;
1800         if (resultset->count < 0)
1801             resultset->count = 0;
1802         if (sr && sr->which == Z_Records_DBOSD)
1803         {
1804             int i;
1805             NMEM nmem = odr_extract_mem (c->odr_in);
1806             Z_NamePlusRecordList *p =
1807                 sr->u.databaseOrSurDiagnostics;
1808             for (i = 0; i<p->num_records; i++)
1809             {
1810                 record_cache_add (resultset, p->records[i],
1811                                   i+ resultset->start);
1812             }
1813             /* transfer our response to search_nmem .. we need it later */
1814             nmem_transfer (resultset->odr->mem, nmem);
1815             nmem_destroy (nmem);
1816             if (present_phase && p->num_records == 0)
1817             {
1818                 /* present response and we didn't get any records! */
1819                 set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1820             }
1821         }
1822         else if (present_phase)
1823         {
1824             /* present response and we didn't get any records! */
1825             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
1826         }
1827     }
1828 }
1829
1830 static void handle_present_response (ZOOM_connection c, Z_PresentResponse *pr)
1831 {
1832     handle_records (c, pr->records, 1);
1833 }
1834
1835 static void handle_search_response (ZOOM_connection c, Z_SearchResponse *sr)
1836 {
1837     ZOOM_resultset resultset;
1838     ZOOM_Event event;
1839
1840     yaz_log (LOG_DEBUG, "got search response");
1841
1842     if (!c->tasks || c->tasks->which != ZOOM_TASK_SEARCH)
1843         return ;
1844
1845     event = ZOOM_Event_create(ZOOM_EVENT_RECV_SEARCH);
1846     ZOOM_connection_put_event(c, event);
1847
1848     resultset = c->tasks->u.search.resultset;
1849
1850     resultset->size = *sr->resultCount;
1851     handle_records (c, sr->records, 0);
1852 }
1853
1854 static void sort_response (ZOOM_connection c, Z_SortResponse *res)
1855 {
1856     if (res->diagnostics && res->num_diagnostics > 0)
1857         response_diag (c, res->diagnostics[0]);
1858 }
1859
1860 static int scan_response (ZOOM_connection c, Z_ScanResponse *res)
1861 {
1862     NMEM nmem = odr_extract_mem (c->odr_in);
1863     ZOOM_scanset scan;
1864
1865     if (!c->tasks || c->tasks->which != ZOOM_TASK_SCAN)
1866         return 0;
1867     scan = c->tasks->u.scan.scan;
1868
1869     if (res->entries && res->entries->nonsurrogateDiagnostics)
1870         response_diag(c, res->entries->nonsurrogateDiagnostics[0]);
1871     scan->scan_response = res;
1872     nmem_transfer (scan->odr->mem, nmem);
1873     if (res->stepSize)
1874         ZOOM_options_set_int (scan->options, "stepSize", *res->stepSize);
1875     if (res->positionOfTerm)
1876         ZOOM_options_set_int (scan->options, "position", *res->positionOfTerm);
1877     if (res->scanStatus)
1878         ZOOM_options_set_int (scan->options, "scanStatus", *res->scanStatus);
1879     if (res->numberOfEntriesReturned)
1880         ZOOM_options_set_int (scan->options, "number",
1881                               *res->numberOfEntriesReturned);
1882     nmem_destroy (nmem);
1883     return 1;
1884 }
1885
1886 static zoom_ret send_sort (ZOOM_connection c)
1887 {
1888     ZOOM_resultset  resultset;
1889
1890     if (!c->tasks || c->tasks->which != ZOOM_TASK_SEARCH)
1891         return zoom_complete;
1892
1893     resultset = c->tasks->u.search.resultset;
1894
1895     if (c->error)
1896     {
1897         resultset->r_sort_spec = 0;
1898         return zoom_complete;
1899     }
1900     if (resultset->r_sort_spec)
1901     {
1902         Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_sortRequest);
1903         Z_SortRequest *req = apdu->u.sortRequest;
1904
1905         req->num_inputResultSetNames = 1;
1906         req->inputResultSetNames = (Z_InternationalString **)
1907             odr_malloc (c->odr_out, sizeof(*req->inputResultSetNames));
1908         req->inputResultSetNames[0] =
1909             odr_strdup (c->odr_out, resultset->setname);
1910         req->sortedResultSetName = odr_strdup (c->odr_out, resultset->setname);
1911         req->sortSequence = resultset->r_sort_spec;
1912         resultset->r_sort_spec = 0;
1913         return send_APDU (c, apdu);
1914     }
1915     return zoom_complete;
1916 }
1917
1918 static zoom_ret send_present (ZOOM_connection c)
1919 {
1920     Z_APDU *apdu = 0;
1921     Z_PresentRequest *req = 0;
1922     int i = 0;
1923     const char *syntax = 0;
1924     const char *elementSetName = 0;
1925     ZOOM_resultset  resultset;
1926
1927     if (!c->tasks)
1928         return zoom_complete;
1929
1930     switch (c->tasks->which)
1931     {
1932     case ZOOM_TASK_SEARCH:
1933         resultset = c->tasks->u.search.resultset;
1934         break;
1935     case ZOOM_TASK_RETRIEVE:
1936         resultset = c->tasks->u.retrieve.resultset;
1937         resultset->start = c->tasks->u.retrieve.start;
1938         resultset->count = c->tasks->u.retrieve.count;
1939
1940         if (resultset->start >= resultset->size)
1941             return zoom_complete;
1942         if (resultset->start + resultset->count > resultset->size)
1943             resultset->count = resultset->size - resultset->start;
1944         break;
1945     default:
1946         return zoom_complete;
1947     }
1948
1949     syntax = ZOOM_resultset_option_get (resultset, "preferredRecordSyntax");
1950     elementSetName = ZOOM_resultset_option_get (resultset, "elementSetName");
1951
1952     if (c->error)                  /* don't continue on error */
1953         return zoom_complete;
1954     if (resultset->start < 0)
1955         return zoom_complete;
1956     for (i = 0; i<resultset->count; i++)
1957     {
1958         ZOOM_record rec =
1959             record_cache_lookup (resultset, i + resultset->start);
1960         if (!rec)
1961             break;
1962     }
1963     if (i == resultset->count)
1964         return zoom_complete;
1965
1966     apdu = zget_APDU(c->odr_out, Z_APDU_presentRequest);
1967     req = apdu->u.presentRequest;
1968
1969     resultset->start += i;
1970     resultset->count -= i;
1971     *req->resultSetStartPoint = resultset->start + 1;
1972     *req->numberOfRecordsRequested = resultset->step>0 ?
1973         resultset->step : resultset->count;
1974     assert (*req->numberOfRecordsRequested > 0);
1975
1976     if (syntax && *syntax)
1977         req->preferredRecordSyntax =
1978             yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, syntax);
1979
1980     if (resultset->schema && *resultset->schema)
1981     {
1982         Z_RecordComposition *compo = (Z_RecordComposition *)
1983             odr_malloc (c->odr_out, sizeof(*compo));
1984
1985         req->recordComposition = compo;
1986         compo->which = Z_RecordComp_complex;
1987         compo->u.complex = (Z_CompSpec *)
1988             odr_malloc(c->odr_out, sizeof(*compo->u.complex));
1989         compo->u.complex->selectAlternativeSyntax = (bool_t *)
1990             odr_malloc(c->odr_out, sizeof(bool_t));
1991         *compo->u.complex->selectAlternativeSyntax = 0;
1992
1993         compo->u.complex->generic = (Z_Specification *)
1994             odr_malloc(c->odr_out, sizeof(*compo->u.complex->generic));
1995
1996         compo->u.complex->generic->which = Z_Schema_oid;
1997         compo->u.complex->generic->schema.oid = (Odr_oid *)
1998             yaz_str_to_z3950oid (c->odr_out, CLASS_SCHEMA, resultset->schema);
1999
2000         if (!compo->u.complex->generic->schema.oid)
2001         {
2002             /* OID wasn't a schema! Try record syntax instead. */
2003
2004             compo->u.complex->generic->schema.oid = (Odr_oid *)
2005                 yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, resultset->schema);
2006         }
2007         if (elementSetName && *elementSetName)
2008         {
2009             compo->u.complex->generic->elementSpec = (Z_ElementSpec *)
2010                 odr_malloc(c->odr_out, sizeof(Z_ElementSpec));
2011             compo->u.complex->generic->elementSpec->which =
2012                 Z_ElementSpec_elementSetName;
2013             compo->u.complex->generic->elementSpec->u.elementSetName =
2014                 odr_strdup (c->odr_out, elementSetName);
2015         }
2016         else
2017             compo->u.complex->generic->elementSpec = 0;
2018         compo->u.complex->num_dbSpecific = 0;
2019         compo->u.complex->dbSpecific = 0;
2020         compo->u.complex->num_recordSyntax = 0;
2021         compo->u.complex->recordSyntax = 0;
2022     }
2023     else if (elementSetName && *elementSetName)
2024     {
2025         Z_ElementSetNames *esn = (Z_ElementSetNames *)
2026             odr_malloc (c->odr_out, sizeof(*esn));
2027         Z_RecordComposition *compo = (Z_RecordComposition *)
2028             odr_malloc (c->odr_out, sizeof(*compo));
2029
2030         esn->which = Z_ElementSetNames_generic;
2031         esn->u.generic = odr_strdup (c->odr_out, elementSetName);
2032         compo->which = Z_RecordComp_simple;
2033         compo->u.simple = esn;
2034         req->recordComposition = compo;
2035     }
2036     req->resultSetId = odr_strdup(c->odr_out, resultset->setname);
2037     return send_APDU (c, apdu);
2038 }
2039
2040 ZOOM_API(ZOOM_scanset)
2041 ZOOM_connection_scan (ZOOM_connection c, const char *start)
2042 {
2043     ZOOM_scanset scan = (ZOOM_scanset) xmalloc (sizeof(*scan));
2044
2045     scan->connection = c;
2046     scan->odr = odr_createmem (ODR_DECODE);
2047     scan->options = ZOOM_options_create_with_parent (c->options);
2048     scan->refcount = 1;
2049     scan->scan_response = 0;
2050
2051     if ((scan->termListAndStartPoint =
2052          p_query_scan(scan->odr, PROTO_Z3950, &scan->attributeSet,
2053                       start)))
2054     {
2055         ZOOM_task task = ZOOM_connection_add_task (c, ZOOM_TASK_SCAN);
2056         task->u.scan.scan = scan;
2057
2058         (scan->refcount)++;
2059         if (!c->async)
2060         {
2061             while (ZOOM_event (1, &c))
2062                 ;
2063         }
2064     }
2065     return scan;
2066 }
2067
2068 ZOOM_API(void)
2069 ZOOM_scanset_destroy (ZOOM_scanset scan)
2070 {
2071     if (!scan)
2072         return;
2073     (scan->refcount)--;
2074     if (scan->refcount == 0)
2075     {
2076         odr_destroy (scan->odr);
2077
2078         ZOOM_options_destroy (scan->options);
2079         xfree (scan);
2080     }
2081 }
2082
2083 static zoom_ret send_package (ZOOM_connection c)
2084 {
2085     ZOOM_Event event;
2086     if (!c->tasks)
2087         return zoom_complete;
2088     assert (c->tasks->which == ZOOM_TASK_PACKAGE);
2089
2090     event = ZOOM_Event_create (ZOOM_EVENT_SEND_APDU);
2091     ZOOM_connection_put_event (c, event);
2092
2093     return do_write_ex (c, c->tasks->u.package->buf_out,
2094                         c->tasks->u.package->len_out);
2095 }
2096
2097 static zoom_ret send_scan (ZOOM_connection c)
2098 {
2099     ZOOM_scanset scan;
2100     Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_scanRequest);
2101     Z_ScanRequest *req = apdu->u.scanRequest;
2102     if (!c->tasks)
2103         return zoom_complete;
2104     assert (c->tasks->which == ZOOM_TASK_SCAN);
2105     scan = c->tasks->u.scan.scan;
2106
2107     req->termListAndStartPoint = scan->termListAndStartPoint;
2108     req->attributeSet = scan->attributeSet;
2109
2110     *req->numberOfTermsRequested =
2111         ZOOM_options_get_int(scan->options, "number", 10);
2112
2113     req->preferredPositionInResponse =
2114         odr_intdup (c->odr_out,
2115                     ZOOM_options_get_int(scan->options, "position", 1));
2116
2117     req->stepSize =
2118         odr_intdup (c->odr_out,
2119                     ZOOM_options_get_int(scan->options, "stepSize", 0));
2120
2121     req->databaseNames = set_DatabaseNames (c, scan->options,
2122                                             &req->num_databaseNames);
2123
2124     return send_APDU (c, apdu);
2125 }
2126
2127 ZOOM_API(size_t)
2128 ZOOM_scanset_size (ZOOM_scanset scan)
2129 {
2130     if (!scan || !scan->scan_response || !scan->scan_response->entries)
2131         return 0;
2132     return scan->scan_response->entries->num_entries;
2133 }
2134
2135 ZOOM_API(const char *)
2136 ZOOM_scanset_term (ZOOM_scanset scan, size_t pos,
2137                    int *occ, int *len)
2138 {
2139     const char *term = 0;
2140     size_t noent = ZOOM_scanset_size (scan);
2141     Z_ScanResponse *res = scan->scan_response;
2142
2143     *len = 0;
2144     *occ = 0;
2145     if (pos >= noent)
2146         return 0;
2147     if (res->entries->entries[pos]->which == Z_Entry_termInfo)
2148     {
2149         Z_TermInfo *t = res->entries->entries[pos]->u.termInfo;
2150
2151         if (t->term->which == Z_Term_general)
2152         {
2153             term = (const char *) t->term->u.general->buf;
2154             *len = t->term->u.general->len;
2155         }
2156         *occ = t->globalOccurrences ? *t->globalOccurrences : 0;
2157     }
2158     return term;
2159 }
2160
2161 ZOOM_API(const char *)
2162 ZOOM_scanset_display_term (ZOOM_scanset scan, size_t pos,
2163                            int *occ, int *len)
2164 {
2165     const char *term = 0;
2166     size_t noent = ZOOM_scanset_size (scan);
2167     Z_ScanResponse *res = scan->scan_response;
2168
2169     *len = 0;
2170     *occ = 0;
2171     if (pos >= noent)
2172         return 0;
2173     if (res->entries->entries[pos]->which == Z_Entry_termInfo)
2174     {
2175         Z_TermInfo *t = res->entries->entries[pos]->u.termInfo;
2176
2177         if (t->displayTerm)
2178         {
2179             term = (const char *) t->term->u.general->buf;
2180             *len = strlen(term);
2181         }
2182         else if (t->term->which == Z_Term_general)
2183         {
2184             term = (const char *) t->term->u.general->buf;
2185             *len = t->term->u.general->len;
2186         }
2187         *occ = t->globalOccurrences ? *t->globalOccurrences : 0;
2188     }
2189     return term;
2190 }
2191
2192 ZOOM_API(const char *)
2193 ZOOM_scanset_option_get (ZOOM_scanset scan, const char *key)
2194 {
2195     return ZOOM_options_get (scan->options, key);
2196 }
2197
2198 ZOOM_API(void)
2199 ZOOM_scanset_option_set (ZOOM_scanset scan, const char *key,
2200                               const char *val)
2201 {
2202     ZOOM_options_set (scan->options, key, val);
2203 }
2204
2205 static Z_APDU *create_es_package (ZOOM_package p, int type)
2206 {
2207     const char *str;
2208     Z_APDU *apdu = zget_APDU(p->odr_out, Z_APDU_extendedServicesRequest);
2209     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2210
2211     *req->function = Z_ExtendedServicesRequest_create;
2212
2213     str = ZOOM_options_get(p->options, "package-name");
2214     if (str && *str)
2215         req->packageName = nmem_strdup (p->odr_out->mem, str);
2216
2217     str = ZOOM_options_get(p->options, "user-id");
2218     if (str)
2219         req->userId = nmem_strdup (p->odr_out->mem, str);
2220
2221     req->packageType = yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2222                                               type);
2223
2224     str = ZOOM_options_get(p->options, "function");
2225     if (str)
2226     {
2227         if (!strcmp (str, "create"))
2228             *req->function = 1;
2229         if (!strcmp (str, "delete"))
2230             *req->function = 2;
2231         if (!strcmp (str, "modify"))
2232             *req->function = 3;
2233     }
2234     return apdu;
2235 }
2236
2237 static const char *ill_array_lookup (void *clientData, const char *idx)
2238 {
2239     ZOOM_package p = (ZOOM_package) clientData;
2240     return ZOOM_options_get (p->options, idx+4);
2241 }
2242
2243 static Z_External *encode_ill_request (ZOOM_package p)
2244 {
2245     ODR out = p->odr_out;
2246     ILL_Request *req;
2247     Z_External *r = 0;
2248     struct ill_get_ctl ctl;
2249
2250     ctl.odr = p->odr_out;
2251     ctl.clientData = p;
2252     ctl.f = ill_array_lookup;
2253
2254     req = ill_get_ILLRequest(&ctl, "ill", 0);
2255
2256     if (!ill_Request (out, &req, 0, 0))
2257     {
2258         int ill_request_size;
2259         char *ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
2260         if (ill_request_buf)
2261             odr_setbuf (out, ill_request_buf, ill_request_size, 1);
2262         return 0;
2263     }
2264     else
2265     {
2266         oident oid;
2267         int illRequest_size = 0;
2268         char *illRequest_buf = odr_getbuf (out, &illRequest_size, 0);
2269
2270         oid.proto = PROTO_GENERAL;
2271         oid.oclass = CLASS_GENERAL;
2272         oid.value = VAL_ISO_ILL_1;
2273
2274         r = (Z_External *) odr_malloc (out, sizeof(*r));
2275         r->direct_reference = odr_oiddup(out,oid_getoidbyent(&oid));
2276         r->indirect_reference = 0;
2277         r->descriptor = 0;
2278         r->which = Z_External_single;
2279
2280         r->u.single_ASN1_type = (Odr_oct *)
2281             odr_malloc (out, sizeof(*r->u.single_ASN1_type));
2282         r->u.single_ASN1_type->buf = (unsigned char*)
2283             odr_malloc (out, illRequest_size);
2284         r->u.single_ASN1_type->len = illRequest_size;
2285         r->u.single_ASN1_type->size = illRequest_size;
2286         memcpy (r->u.single_ASN1_type->buf, illRequest_buf, illRequest_size);
2287     }
2288     return r;
2289 }
2290
2291 static Z_ItemOrder *encode_item_order(ZOOM_package p)
2292 {
2293     Z_ItemOrder *req = (Z_ItemOrder *) odr_malloc (p->odr_out, sizeof(*req));
2294     const char *str;
2295
2296     req->which=Z_IOItemOrder_esRequest;
2297     req->u.esRequest = (Z_IORequest *)
2298         odr_malloc(p->odr_out,sizeof(Z_IORequest));
2299
2300     /* to keep part ... */
2301     req->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
2302         odr_malloc(p->odr_out,sizeof(Z_IOOriginPartToKeep));
2303     req->u.esRequest->toKeep->supplDescription = 0;
2304     req->u.esRequest->toKeep->contact = (Z_IOContact *)
2305         odr_malloc (p->odr_out, sizeof(*req->u.esRequest->toKeep->contact));
2306
2307     str = ZOOM_options_get(p->options, "contact-name");
2308     req->u.esRequest->toKeep->contact->name = str ?
2309         nmem_strdup (p->odr_out->mem, str) : 0;
2310
2311     str = ZOOM_options_get(p->options, "contact-phone");
2312     req->u.esRequest->toKeep->contact->phone = str ?
2313         nmem_strdup (p->odr_out->mem, str) : 0;
2314
2315     str = ZOOM_options_get(p->options, "contact-email");
2316     req->u.esRequest->toKeep->contact->email = str ?
2317         nmem_strdup (p->odr_out->mem, str) : 0;
2318
2319     req->u.esRequest->toKeep->addlBilling = 0;
2320
2321     /* not to keep part ... */
2322     req->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
2323         odr_malloc(p->odr_out,sizeof(Z_IOOriginPartNotToKeep));
2324
2325     str = ZOOM_options_get(p->options, "itemorder-setname");
2326     if (!str)
2327         str = "default";
2328
2329     if (!*str)
2330         req->u.esRequest->notToKeep->resultSetItem = 0;
2331     else
2332     {
2333         req->u.esRequest->notToKeep->resultSetItem = (Z_IOResultSetItem *)
2334            odr_malloc(p->odr_out, sizeof(Z_IOResultSetItem));
2335
2336         req->u.esRequest->notToKeep->resultSetItem->resultSetId =
2337            nmem_strdup (p->odr_out->mem, str);
2338         req->u.esRequest->notToKeep->resultSetItem->item =
2339             (int *) odr_malloc(p->odr_out, sizeof(int));
2340
2341         str = ZOOM_options_get(p->options, "itemorder-item");
2342         *req->u.esRequest->notToKeep->resultSetItem->item =
2343             (str ? atoi(str) : 1);
2344     }
2345     req->u.esRequest->notToKeep->itemRequest = encode_ill_request(p);
2346
2347     return req;
2348 }
2349
2350 Z_APDU *create_admin_package(ZOOM_package p, int type,
2351                              Z_ESAdminOriginPartToKeep **toKeepP,
2352                              Z_ESAdminOriginPartNotToKeep **notToKeepP)
2353 {
2354     Z_APDU *apdu = create_es_package (p, VAL_ADMINSERVICE);
2355     if (apdu)
2356     {
2357         Z_ESAdminOriginPartToKeep  *toKeep;
2358         Z_ESAdminOriginPartNotToKeep  *notToKeep;
2359         Z_External *r = (Z_External *) odr_malloc (p->odr_out, sizeof(*r));
2360         const char *first_db = "Default";
2361         int num_db;
2362         char **db = set_DatabaseNames(p->connection, p->options, &num_db);
2363         if (num_db > 0)
2364             first_db = db[0];
2365
2366         r->direct_reference =
2367             yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2368                                    VAL_ADMINSERVICE);
2369         r->descriptor = 0;
2370         r->indirect_reference = 0;
2371         r->which = Z_External_ESAdmin;
2372
2373         r->u.adminService = (Z_Admin *)
2374             odr_malloc(p->odr_out, sizeof(*r->u.adminService));
2375         r->u.adminService->which = Z_Admin_esRequest;
2376         r->u.adminService->u.esRequest = (Z_AdminEsRequest *)
2377             odr_malloc(p->odr_out, sizeof(*r->u.adminService->u.esRequest));
2378
2379         toKeep = r->u.adminService->u.esRequest->toKeep =
2380             (Z_ESAdminOriginPartToKeep *)
2381             odr_malloc(p->odr_out, sizeof(*r->u.adminService->u.esRequest->toKeep));
2382         toKeep->which=type;
2383         toKeep->databaseName = odr_strdup(p->odr_out, first_db);
2384         toKeep->u.create=odr_nullval();
2385         apdu->u.extendedServicesRequest->taskSpecificParameters = r;
2386
2387         r->u.adminService->u.esRequest->notToKeep = notToKeep =
2388             (Z_ESAdminOriginPartNotToKeep *)
2389             odr_malloc(p->odr_out,
2390                        sizeof(*r->u.adminService->u.esRequest->notToKeep));
2391         notToKeep->which=Z_ESAdminOriginPartNotToKeep_recordsWillFollow;
2392         notToKeep->u.recordsWillFollow=odr_nullval();
2393         if (toKeepP)
2394             *toKeepP = toKeep;
2395         if (notToKeepP)
2396             *notToKeepP = notToKeep;
2397     }
2398     return apdu;
2399 }
2400
2401 static Z_APDU *create_update_package(ZOOM_package p)
2402 {
2403     Z_APDU *apdu = 0;
2404     const char *first_db = "Default";
2405     int num_db;
2406     char **db = set_DatabaseNames(p->connection, p->options, &num_db);
2407     const char *action = ZOOM_options_get(p->options, "action");
2408     const char *recordId = ZOOM_options_get(p->options, "recordId");
2409     const char *record_buf = ZOOM_options_get(p->options, "record");
2410     const char *syntax_str = ZOOM_options_get(p->options, "syntax");
2411     int syntax_oid = VAL_NONE;
2412     int action_no = -1;
2413
2414     if (syntax_str)
2415         syntax_oid = oid_getvalbyname(syntax_str);
2416     if (!record_buf)
2417     {
2418         record_buf = "void";
2419         syntax_oid = VAL_SUTRS;
2420     }
2421     if (syntax_oid != VAL_NONE)
2422         syntax_oid = VAL_TEXT_XML;
2423
2424     if (num_db > 0)
2425         first_db = db[0];
2426
2427     if (!action)
2428         action = "specialUpdate";
2429
2430     if (!strcmp(action, "recordInsert"))
2431         action_no = Z_IUOriginPartToKeep_recordInsert;
2432     else if (!strcmp(action, "recordReplace"))
2433         action_no = Z_IUOriginPartToKeep_recordReplace;
2434     else if (!strcmp(action, "recordDelete"))
2435         action_no = Z_IUOriginPartToKeep_recordDelete;
2436     else if (!strcmp(action, "elementUpdate"))
2437         action_no = Z_IUOriginPartToKeep_elementUpdate;
2438     else if (!strcmp(action, "specialUpdate"))
2439         action_no = Z_IUOriginPartToKeep_specialUpdate;
2440     else
2441         return 0;
2442
2443     apdu = create_es_package (p, VAL_DBUPDATE);
2444     if (apdu)
2445     {
2446         Z_IUOriginPartToKeep *toKeep;
2447         Z_IUSuppliedRecords *notToKeep;
2448         Z_External *r = (Z_External *)
2449             odr_malloc (p->odr_out, sizeof(*r));
2450
2451         apdu->u.extendedServicesRequest->taskSpecificParameters = r;
2452
2453         r->direct_reference =
2454             yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2455                                    VAL_DBUPDATE);
2456         r->descriptor = 0;
2457         r->which = Z_External_update;
2458         r->indirect_reference = 0;
2459         r->u.update = (Z_IUUpdate *)
2460             odr_malloc(p->odr_out, sizeof(*r->u.update));
2461
2462         r->u.update->which = Z_IUUpdate_esRequest;
2463         r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
2464             odr_malloc(p->odr_out, sizeof(*r->u.update->u.esRequest));
2465         toKeep = r->u.update->u.esRequest->toKeep =
2466             (Z_IUOriginPartToKeep *)
2467             odr_malloc(p->odr_out, sizeof(*toKeep));
2468
2469         toKeep->databaseName = odr_strdup(p->odr_out, first_db);
2470         toKeep->schema = 0;
2471         toKeep->elementSetName = 0;
2472         toKeep->actionQualifier = 0;
2473         toKeep->action = odr_intdup(p->odr_out, action_no);
2474
2475         notToKeep = r->u.update->u.esRequest->notToKeep =
2476             (Z_IUSuppliedRecords *)
2477             odr_malloc(p->odr_out, sizeof(*notToKeep));
2478         notToKeep->num = 1;
2479         notToKeep->elements = (Z_IUSuppliedRecords_elem **)
2480             odr_malloc(p->odr_out, sizeof(*notToKeep->elements));
2481         notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
2482             odr_malloc(p->odr_out, sizeof(**notToKeep->elements));
2483         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2484         if (recordId)
2485         {
2486             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2487                 odr_malloc (p->odr_out, sizeof(Odr_oct));
2488             notToKeep->elements[0]->u.opaque->size = strlen(recordId);
2489             notToKeep->elements[0]->u.opaque->len = strlen(recordId);
2490             notToKeep->elements[0]->u.opaque->buf =
2491                 odr_strdup(p->odr_out, recordId);
2492         }
2493         else
2494             notToKeep->elements[0]->u.opaque = 0;
2495         notToKeep->elements[0]->supplementalId = 0;
2496         notToKeep->elements[0]->correlationInfo = 0;
2497         notToKeep->elements[0]->record =
2498             z_ext_record(p->odr_out, syntax_oid,
2499                          record_buf, strlen(record_buf));
2500     }
2501     if (0 && apdu)
2502     {
2503        ODR print = odr_createmem(ODR_PRINT);
2504
2505        z_APDU(print, &apdu, 0, 0);
2506        odr_destroy(print);
2507     }
2508     return apdu;
2509 }
2510
2511 ZOOM_API(void)
2512     ZOOM_package_send (ZOOM_package p, const char *type)
2513 {
2514     Z_APDU *apdu = 0;
2515     ZOOM_connection c;
2516     if (!p)
2517         return;
2518     c = p->connection;
2519     odr_reset (p->odr_out);
2520     xfree (p->buf_out);
2521     p->buf_out = 0;
2522     if (!strcmp(type, "itemorder"))
2523     {
2524         apdu = create_es_package (p, VAL_ITEMORDER);
2525         if (apdu)
2526         {
2527             Z_External *r = (Z_External *) odr_malloc (p->odr_out, sizeof(*r));
2528
2529             r->direct_reference =
2530                 yaz_oidval_to_z3950oid(p->odr_out, CLASS_EXTSERV,
2531                                        VAL_ITEMORDER);
2532             r->descriptor = 0;
2533             r->which = Z_External_itemOrder;
2534             r->indirect_reference = 0;
2535             r->u.itemOrder = encode_item_order (p);
2536
2537             apdu->u.extendedServicesRequest->taskSpecificParameters = r;
2538         }
2539     }
2540     if (!strcmp(type, "create"))  /* create database */
2541     {
2542         apdu = create_admin_package(p, Z_ESAdminOriginPartToKeep_create,
2543                                     0, 0);
2544     }
2545     if (!strcmp(type, "drop"))  /* drop database */
2546     {
2547         apdu = create_admin_package(p, Z_ESAdminOriginPartToKeep_drop,
2548                                     0, 0);
2549     }
2550     if (!strcmp(type, "update")) /* update record(s) */
2551     {
2552         apdu = create_update_package(p);
2553     }
2554     if (apdu)
2555     {
2556         if (encode_APDU(p->connection, apdu, p->odr_out) == 0)
2557         {
2558             char *buf;
2559
2560             ZOOM_task task = ZOOM_connection_add_task (c, ZOOM_TASK_PACKAGE);
2561             task->u.package = p;
2562             buf = odr_getbuf(p->odr_out, &p->len_out, 0);
2563             p->buf_out = (char *) xmalloc (p->len_out);
2564             memcpy (p->buf_out, buf, p->len_out);
2565
2566             (p->refcount)++;
2567             if (!c->async)
2568             {
2569                 while (ZOOM_event (1, &c))
2570                     ;
2571             }
2572         }
2573     }
2574 }
2575
2576 ZOOM_API(ZOOM_package)
2577     ZOOM_connection_package (ZOOM_connection c, ZOOM_options options)
2578 {
2579     ZOOM_package p = (ZOOM_package) xmalloc (sizeof(*p));
2580
2581     p->connection = c;
2582     p->odr_out = odr_createmem (ODR_ENCODE);
2583     p->options = ZOOM_options_create_with_parent2 (options, c->options);
2584     p->refcount = 1;
2585     p->buf_out = 0;
2586     p->len_out = 0;
2587     return p;
2588 }
2589
2590 ZOOM_API(void)
2591     ZOOM_package_destroy(ZOOM_package p)
2592 {
2593     if (!p)
2594         return;
2595     (p->refcount)--;
2596     if (p->refcount == 0)
2597     {
2598         odr_destroy (p->odr_out);
2599         xfree (p->buf_out);
2600
2601         ZOOM_options_destroy (p->options);
2602         xfree (p);
2603     }
2604 }
2605
2606 ZOOM_API(const char *)
2607 ZOOM_package_option_get (ZOOM_package p, const char *key)
2608 {
2609     return ZOOM_options_get (p->options, key);
2610 }
2611
2612
2613 ZOOM_API(void)
2614 ZOOM_package_option_set (ZOOM_package p, const char *key,
2615                               const char *val)
2616 {
2617     ZOOM_options_set (p->options, key, val);
2618 }
2619
2620 static int ZOOM_connection_exec_task (ZOOM_connection c)
2621 {
2622     ZOOM_task task = c->tasks;
2623     zoom_ret ret = zoom_complete;
2624
2625     if (!task)
2626     {
2627         yaz_log (LOG_DEBUG, "ZOOM_connection_exec_task task=<null>");
2628         return 0;
2629     }
2630     yaz_log (LOG_DEBUG, "ZOOM_connection_exec_task type=%d run=%d",
2631              task->which, task->running);
2632     if (c->error != ZOOM_ERROR_NONE)
2633     {
2634         yaz_log (LOG_DEBUG, "remove tasks because of error = %d", c->error);
2635         ZOOM_connection_remove_tasks (c);
2636         return 0;
2637     }
2638     if (task->running)
2639     {
2640         yaz_log (LOG_DEBUG, "task already running");
2641         return 0;
2642     }
2643     task->running = 1;
2644     ret = zoom_complete;
2645     if (c->cs || task->which == ZOOM_TASK_CONNECT)
2646     {
2647         switch (task->which)
2648         {
2649         case ZOOM_TASK_SEARCH:
2650             if (c->proto == PROTO_HTTP)
2651                 ret = ZOOM_connection_srw_send_search(c);
2652             else
2653                 ret = ZOOM_connection_send_search(c);
2654             break;
2655         case ZOOM_TASK_RETRIEVE:
2656             if (c->proto == PROTO_HTTP)
2657                 ret = ZOOM_connection_srw_send_search(c);
2658             else
2659                 ret = send_present (c);
2660             break;
2661         case ZOOM_TASK_CONNECT:
2662             ret = do_connect(c);
2663             break;
2664         case ZOOM_TASK_SCAN:
2665             ret = send_scan(c);
2666             break;
2667         case ZOOM_TASK_PACKAGE:
2668             ret = send_package(c);
2669             break;
2670         }
2671     }
2672     else
2673     {
2674         yaz_log (LOG_DEBUG, "remove tasks because no connection exist");
2675         ZOOM_connection_remove_tasks (c);
2676     }
2677     if (ret == zoom_complete)
2678     {
2679         yaz_log (LOG_DEBUG, "task removed (complete)");
2680         ZOOM_connection_remove_task (c);
2681         return 0;
2682     }
2683     yaz_log (LOG_DEBUG, "task pending");
2684     return 1;
2685 }
2686
2687 static zoom_ret send_sort_present (ZOOM_connection c)
2688 {
2689     zoom_ret r = send_sort (c);
2690     if (r == zoom_complete)
2691         r = send_present (c);
2692     return r;
2693 }
2694
2695 static int es_response (ZOOM_connection c,
2696                         Z_ExtendedServicesResponse *res)
2697 {
2698     if (!c->tasks || c->tasks->which != ZOOM_TASK_PACKAGE)
2699         return 0;
2700     if (res->diagnostics && res->num_diagnostics > 0)
2701         response_diag(c, res->diagnostics[0]);
2702     if (res->taskPackage &&
2703         res->taskPackage->which == Z_External_extendedService)
2704     {
2705         Z_TaskPackage *taskPackage = res->taskPackage->u.extendedService;
2706         Odr_oct *id = taskPackage->targetReference;
2707
2708         if (id)
2709             ZOOM_options_setl (c->tasks->u.package->options,
2710                                "targetReference", (char*) id->buf, id->len);
2711     }
2712     return 1;
2713 }
2714
2715
2716 static void handle_apdu (ZOOM_connection c, Z_APDU *apdu)
2717 {
2718     Z_InitResponse *initrs;
2719
2720     c->mask = 0;
2721     yaz_log (LOG_DEBUG, "recv APDU type=%d", apdu->which);
2722     switch(apdu->which)
2723     {
2724     case Z_APDU_initResponse:
2725         initrs = apdu->u.initResponse;
2726         ZOOM_connection_option_set(c, "serverImplementationId",
2727                                    initrs->implementationId ?
2728                                    initrs->implementationId : "");
2729         ZOOM_connection_option_set(c, "serverImplementationName",
2730                                    initrs->implementationName ?
2731                                    initrs->implementationName : "");
2732         ZOOM_connection_option_set(c, "serverImplementationVersion",
2733                                    initrs->implementationVersion ?
2734                                    initrs->implementationVersion : "");
2735         /* Set the three old options too, for old applications */
2736         ZOOM_connection_option_set(c, "targetImplementationId",
2737                                    initrs->implementationId ?
2738                                    initrs->implementationId : "");
2739         ZOOM_connection_option_set(c, "targetImplementationName",
2740                                    initrs->implementationName ?
2741                                    initrs->implementationName : "");
2742         ZOOM_connection_option_set(c, "targetImplementationVersion",
2743                                    initrs->implementationVersion ?
2744                                    initrs->implementationVersion : "");
2745         if (!*initrs->result)
2746         {
2747             set_ZOOM_error(c, ZOOM_ERROR_INIT, 0);
2748         }
2749         else
2750         {
2751             char *cookie =
2752                 yaz_oi_get_string_oidval (&apdu->u.initResponse->otherInfo,
2753                                           VAL_COOKIE, 1, 0);
2754             xfree (c->cookie_in);
2755             c->cookie_in = 0;
2756             if (cookie)
2757                 c->cookie_in = xstrdup(cookie);
2758             if (ODR_MASK_GET(initrs->options, Z_Options_namedResultSets) &&
2759                 ODR_MASK_GET(initrs->protocolVersion, Z_ProtocolVersion_3))
2760                 c->support_named_resultsets = 1;
2761             if (c->tasks)
2762             {
2763                 assert (c->tasks->which == ZOOM_TASK_CONNECT);
2764                 ZOOM_connection_remove_task (c);
2765             }
2766             ZOOM_connection_exec_task (c);
2767         }
2768         if (ODR_MASK_GET(initrs->options, Z_Options_negotiationModel))
2769         {
2770             NMEM tmpmem = nmem_create();
2771             Z_CharSetandLanguageNegotiation *p =
2772                 yaz_get_charneg_record(initrs->otherInfo);
2773
2774             if (p)
2775             {
2776                 char *charset=NULL, *lang=NULL;
2777                 int sel;
2778
2779                 yaz_get_response_charneg(tmpmem, p, &charset, &lang, &sel);
2780                 yaz_log(LOG_DEBUG, "Target accepted: charset %s, "
2781                         "language %s, select %d",
2782                         charset ? charset : "none", lang ? lang : "none", sel);
2783                 if (charset)
2784                     ZOOM_connection_option_set (c, "negotiation-charset",
2785                                                 charset);
2786                 if (lang)
2787                     ZOOM_connection_option_set (c, "negotiation-lang",
2788                                                 lang);
2789                 nmem_destroy(tmpmem);
2790             }
2791         }
2792         break;
2793     case Z_APDU_searchResponse:
2794         handle_search_response (c, apdu->u.searchResponse);
2795         if (send_sort_present (c) == zoom_complete)
2796             ZOOM_connection_remove_task (c);
2797         break;
2798     case Z_APDU_presentResponse:
2799         handle_present_response (c, apdu->u.presentResponse);
2800         if (send_present (c) == zoom_complete)
2801             ZOOM_connection_remove_task (c);
2802         break;
2803     case Z_APDU_sortResponse:
2804         sort_response (c, apdu->u.sortResponse);
2805         if (send_present (c) == zoom_complete)
2806             ZOOM_connection_remove_task (c);
2807         break;
2808     case Z_APDU_scanResponse:
2809         scan_response (c, apdu->u.scanResponse);
2810         ZOOM_connection_remove_task (c);
2811         break;
2812     case Z_APDU_extendedServicesResponse:
2813         es_response (c, apdu->u.extendedServicesResponse);
2814         ZOOM_connection_remove_task (c);
2815         break;
2816     case Z_APDU_close:
2817         if (c->reconnect_ok)
2818         {
2819             do_close(c);
2820             c->tasks->running = 0;
2821             ZOOM_connection_insert_task (c, ZOOM_TASK_CONNECT);
2822         }
2823         else
2824         {
2825             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, 0);
2826             do_close(c);
2827         }
2828         break;
2829     default:
2830         set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
2831         do_close(c);
2832     }
2833 }
2834
2835 #if HAVE_XML2
2836 static void handle_srw_response(ZOOM_connection c,
2837                                 Z_SRW_searchRetrieveResponse *res)
2838 {
2839     ZOOM_resultset resultset = 0;
2840     int i;
2841     NMEM nmem;
2842     ZOOM_Event event;
2843
2844     if (!c->tasks)
2845         return;
2846
2847     if (c->tasks->which == ZOOM_TASK_SEARCH)
2848         resultset = c->tasks->u.search.resultset;
2849     else if (c->tasks->which == ZOOM_TASK_RETRIEVE)
2850         resultset = c->tasks->u.retrieve.resultset;
2851     else
2852         return ;
2853
2854     event = ZOOM_Event_create(ZOOM_EVENT_RECV_SEARCH);
2855     ZOOM_connection_put_event(c, event);
2856
2857     resultset->size = 0;
2858
2859     yaz_log(LOG_DEBUG, "got SRW response OK");
2860
2861     if (res->numberOfRecords)
2862         resultset->size = *res->numberOfRecords;
2863
2864     for (i = 0; i<res->num_records; i++)
2865     {
2866         int pos;
2867
2868         Z_NamePlusRecord *npr = (Z_NamePlusRecord *)
2869             odr_malloc(c->odr_in, sizeof(Z_NamePlusRecord));
2870
2871         if (res->records[i].recordPosition &&
2872             *res->records[i].recordPosition > 0)
2873             pos = *res->records[i].recordPosition - 1;
2874         else
2875             pos = resultset->start + i;
2876
2877         npr->databaseName = 0;
2878         npr->which = Z_NamePlusRecord_databaseRecord;
2879         npr->u.databaseRecord = (Z_External *)
2880             odr_malloc(c->odr_in, sizeof(Z_External));
2881         npr->u.databaseRecord->descriptor = 0;
2882         npr->u.databaseRecord->direct_reference =
2883             yaz_oidval_to_z3950oid(c->odr_in, CLASS_RECSYN, VAL_TEXT_XML);
2884         npr->u.databaseRecord->which = Z_External_octet;
2885         npr->u.databaseRecord->u.octet_aligned = (Odr_oct *)
2886             odr_malloc(c->odr_in, sizeof(Odr_oct));
2887         npr->u.databaseRecord->u.octet_aligned->buf =
2888             res->records[i].recordData_buf;
2889         npr->u.databaseRecord->u.octet_aligned->len =
2890             npr->u.databaseRecord->u.octet_aligned->size =
2891             res->records[i].recordData_len;
2892         record_cache_add (resultset, npr, pos);
2893     }
2894     if (res->num_diagnostics > 0)
2895     {
2896         set_dset_error(c, *res->diagnostics[0].code, "SRW",
2897                        res->diagnostics[0].details, 0);
2898     }
2899     nmem = odr_extract_mem(c->odr_in);
2900     nmem_transfer(resultset->odr->mem, nmem);
2901     nmem_destroy(nmem);
2902 }
2903 #endif
2904
2905 #if HAVE_XML2
2906 static void handle_http(ZOOM_connection c, Z_HTTP_Response *hres)
2907 {
2908     int ret = -1;
2909     const char *content_type = z_HTTP_header_lookup(hres->headers,
2910                                                     "Content-Type");
2911     const char *connection_head = z_HTTP_header_lookup(hres->headers,
2912                                                        "Connection");
2913     c->mask = 0;
2914     yaz_log (LOG_DEBUG, "handle_http");
2915
2916     if (content_type && !yaz_strcmp_del("text/xml", content_type, "; "))
2917     {
2918         Z_SOAP *soap_package = 0;
2919         ODR o = odr_createmem(ODR_DECODE);
2920         Z_SOAP_Handler soap_handlers[2] = {
2921             {"http://www.loc.gov/zing/srw/v1.0/", 0,
2922              (Z_SOAP_fun) yaz_srw_codec},
2923             {0, 0, 0}
2924         };
2925         ret = z_soap_codec(o, &soap_package,
2926                            &hres->content_buf, &hres->content_len,
2927                            soap_handlers);
2928         if (!ret && soap_package->which == Z_SOAP_generic &&
2929             soap_package->u.generic->no == 0)
2930         {
2931             Z_SRW_PDU *sr = soap_package->u.generic->p;
2932             if (sr->which == Z_SRW_searchRetrieve_response)
2933                 handle_srw_response(c, sr->u.response);
2934             else
2935                 ret = -1;
2936         }
2937         else if (!ret && (soap_package->which == Z_SOAP_fault
2938                           || soap_package->which == Z_SOAP_error))
2939         {
2940             set_HTTP_error(c, hres->code,
2941                            soap_package->u.fault->fault_code,
2942                            soap_package->u.fault->fault_string);
2943         }
2944         else
2945             ret = -1;
2946         odr_destroy(o);
2947     }
2948     if (ret)
2949     {
2950         if (hres->code != 200)
2951             set_HTTP_error(c, hres->code, 0, 0);
2952         else
2953             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
2954         do_close (c);
2955     }
2956     ZOOM_connection_remove_task(c);
2957     if (!strcmp(hres->version, "1.0"))
2958     {
2959         /* HTTP 1.0: only if Keep-Alive we stay alive.. */
2960         if (!connection_head || strcmp(connection_head, "Keep-Alive"))
2961             do_close(c);
2962     }
2963     else
2964     {
2965         /* HTTP 1.1: only if no close we stay alive .. */
2966         if (connection_head && !strcmp(connection_head, "close"))
2967             do_close(c);
2968     }
2969 }
2970 #endif
2971
2972 static int do_read (ZOOM_connection c)
2973 {
2974     int r, more;
2975     ZOOM_Event event;
2976
2977     event = ZOOM_Event_create (ZOOM_EVENT_RECV_DATA);
2978     ZOOM_connection_put_event (c, event);
2979
2980     r = cs_get (c->cs, &c->buf_in, &c->len_in);
2981     more = cs_more(c->cs);
2982     yaz_log (LOG_DEBUG, "do_read len=%d more=%d", r, more);
2983     if (r == 1)
2984         return 0;
2985     if (r <= 0)
2986     {
2987         if (c->reconnect_ok)
2988         {
2989             do_close (c);
2990             c->reconnect_ok = 0;
2991             yaz_log (LOG_DEBUG, "reconnect read");
2992             c->tasks->running = 0;
2993             ZOOM_connection_insert_task (c, ZOOM_TASK_CONNECT);
2994         }
2995         else
2996         {
2997             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, 0);
2998             do_close (c);
2999         }
3000     }
3001     else
3002     {
3003         Z_GDU *gdu;
3004         ZOOM_Event event;
3005
3006         odr_reset (c->odr_in);
3007         odr_setbuf (c->odr_in, c->buf_in, r, 0);
3008         event = ZOOM_Event_create (ZOOM_EVENT_RECV_APDU);
3009         ZOOM_connection_put_event (c, event);
3010
3011         if (!z_GDU (c->odr_in, &gdu, 0, 0))
3012         {
3013             int x;
3014             int err = odr_geterrorx(c->odr_in, &x);
3015             char msg[60];
3016             const char *element = odr_getelement(c->odr_in);
3017             sprintf (msg, "ODR code %d:%d element=%-20s",
3018                      err, x, element ? element : "<unknown>");
3019             set_ZOOM_error(c, ZOOM_ERROR_DECODE, msg);
3020             do_close (c);
3021         }
3022         else if (gdu->which == Z_GDU_Z3950)
3023             handle_apdu (c, gdu->u.z3950);
3024         else if (gdu->which == Z_GDU_HTTP_Response)
3025         {
3026 #if HAVE_XML2
3027             handle_http (c, gdu->u.HTTP_Response);
3028 #else
3029             set_ZOOM_error(c, ZOOM_ERROR_DECODE, 0);
3030             do_close (c);
3031 #endif
3032         }
3033         c->reconnect_ok = 0;
3034     }
3035     return 1;
3036 }
3037
3038 static zoom_ret do_write_ex (ZOOM_connection c, char *buf_out, int len_out)
3039 {
3040     int r;
3041     ZOOM_Event event;
3042
3043     event = ZOOM_Event_create(ZOOM_EVENT_SEND_DATA);
3044     ZOOM_connection_put_event (c, event);
3045
3046     yaz_log (LOG_DEBUG, "do_write_ex len=%d", len_out);
3047     if ((r=cs_put (c->cs, buf_out, len_out)) < 0)
3048     {
3049         if (c->reconnect_ok)
3050         {
3051             do_close (c);
3052             c->reconnect_ok = 0;
3053             yaz_log (LOG_DEBUG, "reconnect write");
3054             c->tasks->running = 0;
3055             ZOOM_connection_insert_task (c, ZOOM_TASK_CONNECT);
3056             return zoom_pending;
3057         }
3058         if (c->state == STATE_CONNECTING)
3059             set_ZOOM_error(c, ZOOM_ERROR_CONNECT, 0);
3060         else
3061             set_ZOOM_error(c, ZOOM_ERROR_CONNECTION_LOST, 0);
3062         do_close (c);
3063         return zoom_complete;
3064     }
3065     else if (r == 1)
3066     {
3067         c->mask = ZOOM_SELECT_EXCEPT;
3068         if (c->cs->io_pending & CS_WANT_WRITE)
3069             c->mask += ZOOM_SELECT_WRITE;
3070         if (c->cs->io_pending & CS_WANT_READ)
3071             c->mask += ZOOM_SELECT_READ;
3072         yaz_log (LOG_DEBUG, "do_write_ex 1 mask=%d", c->mask);
3073     }
3074     else
3075     {
3076         c->mask = ZOOM_SELECT_READ|ZOOM_SELECT_EXCEPT;
3077         yaz_log (LOG_DEBUG, "do_write_ex 2 mask=%d", c->mask);
3078     }
3079     return zoom_pending;
3080 }
3081
3082 static zoom_ret do_write(ZOOM_connection c)
3083 {
3084     return do_write_ex (c, c->buf_out, c->len_out);
3085 }
3086
3087
3088 ZOOM_API(const char *)
3089 ZOOM_connection_option_get (ZOOM_connection c, const char *key)
3090 {
3091     return ZOOM_options_get (c->options, key);
3092 }
3093
3094 ZOOM_API(void)
3095 ZOOM_connection_option_set (ZOOM_connection c, const char *key,
3096                                   const char *val)
3097 {
3098     ZOOM_options_set (c->options, key, val);
3099 }
3100
3101 ZOOM_API(const char *)
3102 ZOOM_resultset_option_get (ZOOM_resultset r, const char *key)
3103 {
3104     return ZOOM_options_get (r->options, key);
3105 }
3106
3107 ZOOM_API(void)
3108 ZOOM_resultset_option_set (ZOOM_resultset r, const char *key,
3109                                   const char *val)
3110 {
3111     ZOOM_options_set (r->options, key, val);
3112 }
3113
3114
3115 ZOOM_API(int)
3116 ZOOM_connection_errcode (ZOOM_connection c)
3117 {
3118     return ZOOM_connection_error (c, 0, 0);
3119 }
3120
3121 ZOOM_API(const char *)
3122 ZOOM_connection_errmsg (ZOOM_connection c)
3123 {
3124     const char *msg;
3125     ZOOM_connection_error (c, &msg, 0);
3126     return msg;
3127 }
3128
3129 ZOOM_API(const char *)
3130 ZOOM_connection_addinfo (ZOOM_connection c)
3131 {
3132     const char *addinfo;
3133     ZOOM_connection_error (c, 0, &addinfo);
3134     return addinfo;
3135 }
3136
3137 ZOOM_API(const char *)
3138 ZOOM_diag_str (int error)
3139 {
3140     switch (error)
3141     {
3142     case ZOOM_ERROR_NONE:
3143         return "No error";
3144     case ZOOM_ERROR_CONNECT:
3145         return "Connect failed";
3146     case ZOOM_ERROR_MEMORY:
3147         return "Out of memory";
3148     case ZOOM_ERROR_ENCODE:
3149         return "Encoding failed";
3150     case ZOOM_ERROR_DECODE:
3151         return "Decoding failed";
3152     case ZOOM_ERROR_CONNECTION_LOST:
3153         return "Connection lost";
3154     case ZOOM_ERROR_INIT:
3155         return "Init rejected";
3156     case ZOOM_ERROR_INTERNAL:
3157         return "Internal failure";
3158     case ZOOM_ERROR_TIMEOUT:
3159         return "Timeout";
3160     case ZOOM_ERROR_UNSUPPORTED_PROTOCOL:
3161         return "Unsupported protocol";
3162     case ZOOM_ERROR_UNSUPPORTED_QUERY:
3163         return "Unsupported query type";
3164     default:
3165         return diagbib1_str (error);
3166     }
3167 }
3168
3169 ZOOM_API(int)
3170 ZOOM_connection_error_x (ZOOM_connection c, const char **cp,
3171                          const char **addinfo, const char **diagset)
3172 {
3173     int error = c->error;
3174     if (cp)
3175     {
3176         if (!c->diagset || !strcmp(c->diagset, "ZOOM"))
3177             *cp = ZOOM_diag_str(error);
3178         else if (!strcmp(c->diagset, "HTTP"))
3179             *cp = z_HTTP_errmsg(c->error);
3180         else if (!strcmp(c->diagset, "Bib-1"))
3181             *cp = ZOOM_diag_str(error);
3182         else if (!strcmp(c->diagset, "SRW"))
3183             *cp = yaz_diag_srw_str(c->error);
3184         else
3185             *cp = "Unknown error and diagnostic set";
3186     }
3187     if (addinfo)
3188         *addinfo = c->addinfo ? c->addinfo : "";
3189     if (diagset)
3190         *diagset = c->diagset ? c->diagset : "";
3191     return c->error;
3192 }
3193
3194 ZOOM_API(int)
3195 ZOOM_connection_error (ZOOM_connection c, const char **cp,
3196                        const char **addinfo)
3197 {
3198     return ZOOM_connection_error_x(c, cp, addinfo, 0);
3199 }
3200
3201 static int ZOOM_connection_do_io(ZOOM_connection c, int mask)
3202 {
3203     ZOOM_Event event = 0;
3204     int r = cs_look(c->cs);
3205     yaz_log (LOG_DEBUG, "ZOOM_connection_do_io c=%p mask=%d cs_look=%d",
3206              c, mask, r);
3207
3208     if (r == CS_NONE)
3209     {
3210         event = ZOOM_Event_create (ZOOM_EVENT_CONNECT);
3211         set_ZOOM_error(c, ZOOM_ERROR_CONNECT, 0);
3212         do_close (c);
3213         ZOOM_connection_put_event (c, event);
3214     }
3215     else if (r == CS_CONNECT)
3216     {
3217         int ret;
3218         event = ZOOM_Event_create (ZOOM_EVENT_CONNECT);
3219
3220         ret = cs_rcvconnect (c->cs);
3221         yaz_log (LOG_DEBUG, "cs_rcvconnect returned %d", ret);
3222         if (ret == 1)
3223         {
3224             c->mask = ZOOM_SELECT_EXCEPT;
3225             if (c->cs->io_pending & CS_WANT_WRITE)
3226                 c->mask += ZOOM_SELECT_WRITE;
3227             if (c->cs->io_pending & CS_WANT_READ)
3228                 c->mask += ZOOM_SELECT_READ;
3229             ZOOM_connection_put_event (c, event);
3230         }
3231         else if (ret == 0)
3232         {
3233             ZOOM_connection_put_event (c, event);
3234             if (c->proto == PROTO_Z3950)
3235                 ZOOM_connection_send_init(c);
3236             else
3237             {
3238                 /* no init request for SRW .. */
3239                 assert (c->tasks->which == ZOOM_TASK_CONNECT);
3240                 ZOOM_connection_remove_task (c);
3241                 c->mask = 0;
3242                 ZOOM_connection_exec_task (c);
3243             }
3244             c->state = STATE_ESTABLISHED;
3245         }
3246         else
3247         {
3248             set_ZOOM_error(c, ZOOM_ERROR_CONNECT, 0);
3249             do_close (c);
3250             ZOOM_connection_put_event (c, event);
3251         }
3252     }
3253     else
3254     {
3255         if (mask & ZOOM_SELECT_READ)
3256             do_read (c);
3257         if (c->cs && (mask & ZOOM_SELECT_WRITE))
3258             do_write (c);
3259     }
3260     return 1;
3261 }
3262
3263 ZOOM_API(int)
3264 ZOOM_connection_last_event(ZOOM_connection cs)
3265 {
3266     if (!cs)
3267         return ZOOM_EVENT_NONE;
3268     return cs->last_event;
3269 }
3270
3271 ZOOM_API(int)
3272 ZOOM_event (int no, ZOOM_connection *cs)
3273 {
3274     int timeout = 5000;
3275 #if HAVE_SYS_POLL_H
3276     struct pollfd pollfds[1024];
3277     ZOOM_connection poll_cs[1024];
3278 #else
3279     struct timeval tv;
3280     fd_set input, output, except;
3281 #endif
3282     int i, r, nfds;
3283     int max_fd = 0;
3284
3285     for (i = 0; i<no; i++)
3286     {
3287         ZOOM_connection c = cs[i];
3288         ZOOM_Event event;
3289         if (c && (event = ZOOM_connection_get_event(c)))
3290         {
3291             ZOOM_Event_destroy (event);
3292             return i+1;
3293         }
3294     }
3295     for (i = 0; i<no; i++)
3296     {
3297         ZOOM_connection c = cs[i];
3298         ZOOM_Event event;
3299         if (c && ZOOM_connection_exec_task (c))
3300         {
3301             if ((event = ZOOM_connection_get_event(c)))
3302             {
3303                 ZOOM_Event_destroy (event);
3304                 return i+1;
3305             }
3306         }
3307     }
3308 #if HAVE_SYS_POLL_H
3309
3310 #else
3311     FD_ZERO (&input);
3312     FD_ZERO (&output);
3313     FD_ZERO (&except);
3314 #endif
3315     nfds = 0;
3316     for (i = 0; i<no; i++)
3317     {
3318         ZOOM_connection c = cs[i];
3319         int fd, mask;
3320         int this_timeout;
3321
3322         if (!c)
3323             continue;
3324         fd = z3950_connection_socket(c);
3325         mask = z3950_connection_mask(c);
3326
3327         if (fd == -1)
3328             continue;
3329         if (max_fd < fd)
3330             max_fd = fd;
3331
3332         this_timeout = ZOOM_options_get_int (c->options, "timeout", -1);
3333         if (this_timeout != -1 && this_timeout < timeout)
3334             timeout = this_timeout;
3335 #if HAVE_SYS_POLL_H
3336         if (mask)
3337         {
3338             short poll_events = 0;
3339
3340             if (mask & ZOOM_SELECT_READ)
3341                 poll_events += POLLIN;
3342             if (mask & ZOOM_SELECT_WRITE)
3343                 poll_events += POLLOUT;
3344             if (mask & ZOOM_SELECT_EXCEPT)
3345                 poll_events += POLLERR;
3346             pollfds[nfds].fd = fd;
3347             pollfds[nfds].events = poll_events;
3348             pollfds[nfds].revents = 0;
3349             poll_cs[nfds] = c;
3350             nfds++;
3351         }
3352 #else
3353         if (mask & ZOOM_SELECT_READ)
3354         {
3355             FD_SET (fd, &input);
3356             nfds++;
3357         }
3358         if (mask & ZOOM_SELECT_WRITE)
3359         {
3360             FD_SET (fd, &output);
3361             nfds++;
3362         }
3363         if (mask & ZOOM_SELECT_EXCEPT)
3364         {
3365             FD_SET (fd, &except);
3366             nfds++;
3367         }
3368 #endif
3369     }
3370     if (timeout >= 5000)
3371         timeout = 30;
3372
3373     if (!nfds)
3374         return 0;
3375
3376 #if HAVE_SYS_POLL_H
3377     r = poll (pollfds, nfds, timeout * 1000);
3378     for (i = 0; i<nfds; i++)
3379     {
3380         ZOOM_connection c = poll_cs[i];
3381         if (r && c->mask)
3382         {
3383             int mask = 0;
3384             if (pollfds[i].revents & POLLIN)
3385                 mask += ZOOM_SELECT_READ;
3386             if (pollfds[i].revents & POLLOUT)
3387                 mask += ZOOM_SELECT_WRITE;
3388             if (pollfds[i].revents & POLLERR)
3389                 mask += ZOOM_SELECT_EXCEPT;
3390             if (mask)
3391                 ZOOM_connection_do_io(c, mask);
3392         }
3393         else if (r == 0 && c->mask)
3394         {
3395             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_TIMEOUT);
3396             /* timeout and this connection was waiting */
3397             set_ZOOM_error(c, ZOOM_ERROR_TIMEOUT, 0);
3398             do_close (c);
3399             ZOOM_connection_put_event(c, event);
3400         }
3401     }
3402 #else
3403     tv.tv_sec = timeout;
3404     tv.tv_usec = 0;
3405     yaz_log (LOG_DEBUG, "select start");
3406     r = select (max_fd+1, &input, &output, &except, &tv);
3407     yaz_log (LOG_DEBUG, "select stop, returned r=%d", r);
3408     for (i = 0; i<no; i++)
3409     {
3410         ZOOM_connection c = cs[i];
3411         int fd, mask;
3412
3413         if (!c)
3414             continue;
3415         fd = z3950_connection_socket(c);
3416         mask = 0;
3417         if (r && c->mask)
3418         {
3419             /* no timeout and real socket */
3420             if (FD_ISSET(fd, &input))
3421                 mask += ZOOM_SELECT_READ;
3422             if (FD_ISSET(fd, &output))
3423                 mask += ZOOM_SELECT_WRITE;
3424             if (FD_ISSET(fd, &except))
3425                 mask += ZOOM_SELECT_EXCEPT;
3426             if (mask)
3427                 ZOOM_connection_do_io(c, mask);
3428         }
3429         if (r == 0 && c->mask)
3430         {
3431             ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_TIMEOUT);
3432             /* timeout and this connection was waiting */
3433             set_ZOOM_error(c, ZOOM_ERROR_TIMEOUT, 0);
3434             do_close (c);
3435             yaz_log (LOG_DEBUG, "timeout");
3436             ZOOM_connection_put_event(c, event);
3437         }
3438     }
3439 #endif
3440     for (i = 0; i<no; i++)
3441     {
3442         ZOOM_connection c = cs[i];
3443         ZOOM_Event event;
3444         if (c && (event = ZOOM_connection_get_event(c)))
3445         {
3446             ZOOM_Event_destroy (event);
3447             return i+1;
3448         }
3449     }
3450     return 0;
3451 }