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