2 * $Id: zoom-c.c,v 1.15 2001-12-30 22:21:11 adam Exp $
4 * ZOOM layer for C, connections, result sets, queries.
7 #include <yaz/xmalloc.h>
8 #include <yaz/otherinfo.h>
10 #include <yaz/pquery.h>
11 #include <yaz/diagbib1.h>
19 static ZOOM_Event ZOOM_Event_create (int kind)
21 ZOOM_Event event = xmalloc (sizeof(*event));
28 static void ZOOM_Event_destroy (ZOOM_Event event)
33 static void ZOOM_connection_put_event (ZOOM_connection c, ZOOM_Event event)
35 // put in back of queue
38 c->m_queue_back->prev = event;
39 assert (c->m_queue_front);
43 assert (!c->m_queue_front);
44 c->m_queue_front = event;
46 event->next = c->m_queue_back;
48 c->m_queue_back = event;
51 static ZOOM_Event ZOOM_connection_get_event(ZOOM_connection c)
53 // get from front of queue
54 ZOOM_Event event = c->m_queue_front;
57 assert (c->m_queue_back);
58 c->m_queue_front = event->prev;
61 assert (c->m_queue_back);
62 c->m_queue_front->next = 0;
66 c->last_event = event->kind;
70 static void clear_error (ZOOM_connection c)
72 c->error = ZOOM_ERROR_NONE;
77 ZOOM_task ZOOM_connection_add_task (ZOOM_connection c, int which)
79 ZOOM_task *taskp = &c->tasks;
81 taskp = &(*taskp)->next;
82 *taskp = xmalloc (sizeof(**taskp));
83 (*taskp)->running = 0;
84 (*taskp)->which = which;
90 void ZOOM_connection_remove_task (ZOOM_connection c)
92 ZOOM_task task = c->tasks;
96 c->tasks = task->next;
99 case ZOOM_TASK_SEARCH:
100 ZOOM_resultset_destroy (task->u.search.resultset);
102 case ZOOM_TASK_RETRIEVE:
103 ZOOM_resultset_destroy (task->u.retrieve.resultset);
105 case ZOOM_TASK_CONNECT:
108 ZOOM_scanset_destroy (task->u.scan.scan);
117 void ZOOM_connection_remove_tasks (ZOOM_connection c)
120 ZOOM_connection_remove_task(c);
123 static ZOOM_record record_cache_lookup (ZOOM_resultset r,
125 const char *elementSetName);
127 ZOOM_connection ZOOM_connection_create (ZOOM_options options)
129 ZOOM_connection c = xmalloc (sizeof(*c));
133 c->state = STATE_IDLE;
134 c->error = ZOOM_ERROR_NONE;
142 c->options = ZOOM_options_create_with_parent(options);
151 c->odr_in = odr_createmem (ODR_DECODE);
152 c->odr_out = odr_createmem (ODR_ENCODE);
155 c->support_named_resultsets = 0;
156 c->last_event = ZOOM_EVENT_NONE;
158 c->m_queue_front = 0;
163 /* set database names. Take local databases (if set); otherwise
164 take databases given in ZURL (if set); otherwise use Default */
165 static char **set_DatabaseNames (ZOOM_connection con, ZOOM_options options,
168 char **databaseNames;
171 const char *cp = ZOOM_options_get (options, "databaseName");
175 cp = strchr (con->host_port, '/');
182 while ((c = strchr(c, '+')))
190 databaseNames = odr_malloc (con->odr_out, no * sizeof(*databaseNames));
194 c = strchr (cp, '+');
202 /* cp ptr to first char of db name, c is char
204 databaseNames[no] = odr_malloc (con->odr_out, 1+c-cp);
205 memcpy (databaseNames[no], cp, c-cp);
206 databaseNames[no++][c-cp] = '\0';
211 databaseNames[no] = NULL;
213 return databaseNames;
216 ZOOM_connection ZOOM_connection_new (const char *host, int portnum)
218 ZOOM_connection c = ZOOM_connection_create (0);
220 ZOOM_connection_connect (c, host, portnum);
224 void ZOOM_connection_connect(ZOOM_connection c,
225 const char *host, int portnum)
230 val = ZOOM_options_get (c->options, "proxy");
232 c->proxy = xstrdup (val);
239 sprintf (hostn, "%.80s:%d", host, portnum);
240 c->host_port = xstrdup(hostn);
243 c->host_port = xstrdup(host);
245 ZOOM_options_set(c->options, "host", c->host_port);
247 c->async = ZOOM_options_get_bool (c->options, "async", 0);
249 task = ZOOM_connection_add_task (c, ZOOM_TASK_CONNECT);
253 while (ZOOM_event (1, &c))
258 ZOOM_query ZOOM_query_create(void)
260 ZOOM_query s = xmalloc (sizeof(*s));
265 s->odr = odr_createmem (ODR_ENCODE);
270 void ZOOM_query_destroy(ZOOM_query s)
276 yaz_log (LOG_DEBUG, "ZOOM_query_destroy count=%d", s->refcount);
277 if (s->refcount == 0)
279 odr_destroy (s->odr);
284 int ZOOM_query_prefix(ZOOM_query s, const char *str)
286 s->query = odr_malloc (s->odr, sizeof(*s->query));
287 s->query->which = Z_Query_type_1;
288 s->query->u.type_1 = p_query_rpn(s->odr, PROTO_Z3950, str);
289 if (!s->query->u.type_1)
294 int ZOOM_query_sortby(ZOOM_query s, const char *criteria)
296 s->sort_spec = yaz_sort_spec (s->odr, criteria);
302 static int do_write(ZOOM_connection c);
304 void ZOOM_connection_destroy(ZOOM_connection c)
311 for (r = c->resultsets; r; r = r->next)
316 odr_destroy (c->odr_in);
317 odr_destroy (c->odr_out);
318 ZOOM_options_destroy (c->options);
319 ZOOM_connection_remove_tasks (c);
320 xfree (c->host_port);
324 void ZOOM_resultset_addref (ZOOM_resultset r)
329 ZOOM_resultset ZOOM_resultset_create ()
331 ZOOM_resultset r = xmalloc (sizeof(*r));
335 r->odr = odr_createmem (ODR_ENCODE);
349 ZOOM_resultset ZOOM_connection_search_pqf(ZOOM_connection c, const char *q)
352 ZOOM_query s = ZOOM_query_create();
354 ZOOM_query_prefix (s, q);
356 r = ZOOM_connection_search (c, s);
357 ZOOM_query_destroy (s);
361 ZOOM_resultset ZOOM_connection_search(ZOOM_connection c, ZOOM_query q)
363 ZOOM_resultset r = ZOOM_resultset_create ();
367 r->r_sort_spec = q->sort_spec;
368 r->r_query = q->query;
371 r->options = ZOOM_options_create_with_parent(c->options);
373 r->start = ZOOM_options_get_int(r->options, "start", 0);
374 r->count = ZOOM_options_get_int(r->options, "count", 0);
375 r->piggyback = ZOOM_options_get_bool (r->options, "piggyback", 1);
376 cp = ZOOM_options_get (r->options, "setname");
378 r->setname = xstrdup (cp);
382 r->next = c->resultsets;
385 task = ZOOM_connection_add_task (c, ZOOM_TASK_SEARCH);
386 task->u.search.resultset = r;
387 ZOOM_resultset_addref (r);
393 while (ZOOM_event (1, &c))
399 void ZOOM_resultset_destroy(ZOOM_resultset r)
404 yaz_log (LOG_DEBUG, "destroy r = %p count=%d", r, r->refcount);
405 if (r->refcount == 0)
409 /* remove ourselves from the resultsets in connection */
410 ZOOM_resultset *rp = &r->connection->resultsets;
413 assert (*rp); /* we must be in this list!! */
415 { /* OK, we're here - take us out of it */
422 ZOOM_query_destroy (r->search);
423 ZOOM_options_destroy (r->options);
424 odr_destroy (r->odr);
430 size_t ZOOM_resultset_size (ZOOM_resultset r)
435 static void do_close (ZOOM_connection c)
441 c->state = STATE_IDLE;
444 static void ZOOM_resultset_retrieve (ZOOM_resultset r,
445 int force_sync, int start, int count)
456 task = ZOOM_connection_add_task (c, ZOOM_TASK_RETRIEVE);
457 task->u.retrieve.resultset = r;
458 task->u.retrieve.start = start;
459 task->u.retrieve.count = count;
461 ZOOM_resultset_addref (r);
463 if (!r->connection->async || force_sync)
464 while (r->connection && ZOOM_event (1, &r->connection))
468 void ZOOM_resultset_records (ZOOM_resultset r, ZOOM_record *recs,
469 size_t start, size_t count)
471 int force_present = 0;
477 ZOOM_resultset_retrieve (r, force_present, start, count);
481 for (i = 0; i< count; i++)
482 recs[i] = ZOOM_resultset_record_immediate (r, i+start);
486 static int do_connect (ZOOM_connection c)
489 const char *effective_host;
492 effective_host = c->proxy;
494 effective_host = c->host_port;
496 yaz_log (LOG_DEBUG, "do_connect host=%s", effective_host);
499 c->cs = cs_create_host (effective_host, 0, &add);
503 int ret = cs_connect (c->cs, add);
504 yaz_log (LOG_DEBUG, "cs_connect returned %d", ret);
507 c->state = STATE_CONNECTING;
508 c->mask = ZOOM_SELECT_READ | ZOOM_SELECT_WRITE |
513 c->state = STATE_IDLE;
514 c->error = ZOOM_ERROR_CONNECT;
518 int z3950_connection_socket(ZOOM_connection c)
521 return cs_fileno(c->cs);
525 int z3950_connection_mask(ZOOM_connection c)
532 static int encode_APDU(ZOOM_connection c, Z_APDU *a, ODR out)
537 sprintf (str, "send_APDU t=%p type=%d", c, a->which);
540 Z_OtherInformation **oi;
542 yaz_oi_set_string_oidval(oi, out, VAL_COOKIE, 1, c->cookie_out);
544 if (!z_APDU(out, &a, 0, 0))
546 FILE *outf = fopen("/tmp/apdu.txt", "w");
549 ODR odr_pr = odr_createmem(ODR_PRINT);
550 fprintf (outf, "a=%p\n", a);
551 odr_setprint(odr_pr, outf);
552 z_APDU(odr_pr, &a, 0, 0);
556 c->error = ZOOM_ERROR_ENCODE;
564 static int send_APDU (ZOOM_connection c, Z_APDU *a)
568 if (encode_APDU(c, a, c->odr_out))
570 c->buf_out = odr_getbuf(c->odr_out, &c->len_out, 0);
571 event = ZOOM_Event_create (ZOOM_EVENT_SEND_APDU);
572 ZOOM_connection_put_event (c, event);
573 odr_reset(c->odr_out);
578 static int ZOOM_connection_send_init (ZOOM_connection c)
581 Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_initRequest);
582 Z_InitRequest *ireq = apdu->u.initRequest;
583 Z_IdAuthentication *auth = odr_malloc(c->odr_out, sizeof(*auth));
584 const char *auth_groupId = ZOOM_options_get (c->options, "group");
585 const char *auth_userId = ZOOM_options_get (c->options, "user");
586 const char *auth_password = ZOOM_options_get (c->options, "pass");
588 ODR_MASK_SET(ireq->options, Z_Options_search);
589 ODR_MASK_SET(ireq->options, Z_Options_present);
590 ODR_MASK_SET(ireq->options, Z_Options_scan);
591 ODR_MASK_SET(ireq->options, Z_Options_sort);
592 ODR_MASK_SET(ireq->options, Z_Options_extendedServices);
593 ODR_MASK_SET(ireq->options, Z_Options_namedResultSets);
595 ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_1);
596 ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_2);
597 ODR_MASK_SET(ireq->protocolVersion, Z_ProtocolVersion_3);
599 impname = ZOOM_options_get (c->options, "implementationName");
600 ireq->implementationName =
601 odr_malloc (c->odr_out, 15 + (impname ? strlen(impname) : 0));
602 strcpy (ireq->implementationName, "");
605 strcat (ireq->implementationName, impname);
606 strcat (ireq->implementationName, "/");
608 strcat (ireq->implementationName, "ZOOM-C/YAZ");
610 *ireq->maximumRecordSize =
611 ZOOM_options_get_int (c->options, "maximumRecordSize", 1024*1024);
612 *ireq->preferredMessageSize =
613 ZOOM_options_get_int (c->options, "preferredMessageSize", 1024*1024);
615 if (auth_groupId || auth_password)
617 Z_IdPass *pass = odr_malloc(c->odr_out, sizeof(*pass));
620 if (auth_groupId && *auth_groupId)
622 pass->groupId = odr_malloc(c->odr_out, strlen(auth_groupId)+1);
623 strcpy(pass->groupId, auth_groupId);
627 if (auth_userId && *auth_userId)
629 pass->userId = odr_malloc(c->odr_out, strlen(auth_userId)+1);
630 strcpy(pass->userId, auth_userId);
634 if (auth_password && *auth_password)
636 pass->password = odr_malloc(c->odr_out, strlen(auth_password)+1);
637 strcpy(pass->password, auth_password);
642 auth->which = Z_IdAuthentication_idPass;
643 auth->u.idPass = pass;
644 ireq->idAuthentication = auth;
647 else if (auth_userId)
649 auth->which = Z_IdAuthentication_open;
650 auth->u.open = odr_malloc(c->odr_out, strlen(auth_userId)+1);
651 strcpy(auth->u.open, auth_userId);
652 ireq->idAuthentication = auth;
655 yaz_oi_set_string_oidval(&ireq->otherInfo, c->odr_out,
656 VAL_PROXY, 1, c->host_port);
663 static int ZOOM_connection_send_search (ZOOM_connection c)
666 int lslb, ssub, mspn;
668 Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_searchRequest);
669 Z_SearchRequest *search_req = apdu->u.searchRequest;
670 const char *elementSetName;
671 const char *smallSetElementSetName;
672 const char *mediumSetElementSetName;
676 assert (c->tasks->which == ZOOM_TASK_SEARCH);
678 r = c->tasks->u.search.resultset;
681 ZOOM_options_get (r->options, "elementSetName");
682 smallSetElementSetName =
683 ZOOM_options_get (r->options, "smallSetElementSetName");
684 mediumSetElementSetName =
685 ZOOM_options_get (r->options, "mediumSetElementSetName");
687 ZOOM_options_get (r->options, "schema");
689 if (!smallSetElementSetName)
690 smallSetElementSetName = elementSetName;
692 if (!mediumSetElementSetName)
693 mediumSetElementSetName = elementSetName;
698 /* prepare query for the search request */
699 search_req->query = r->r_query;
701 search_req->databaseNames =
702 set_DatabaseNames (c, r->options, &search_req->num_databaseNames);
704 /* get syntax (no need to provide unless piggyback is in effect) */
705 syntax = ZOOM_options_get (r->options, "preferredRecordSyntax");
707 lslb = ZOOM_options_get_int (r->options, "largeSetLowerBound", -1);
708 ssub = ZOOM_options_get_int (r->options, "smallSetUpperBound", -1);
709 mspn = ZOOM_options_get_int (r->options, "mediumSetPresentNumber", -1);
710 if (lslb != -1 && ssub != -1 && mspn != -1)
712 /* So're a Z39.50 expert? Let's hope you don't do sort */
713 *search_req->largeSetLowerBound = lslb;
714 *search_req->smallSetUpperBound = ssub;
715 *search_req->mediumSetPresentNumber = mspn;
717 else if (r->start == 0 && r->count > 0
718 && r->piggyback && !r->r_sort_spec && !schema)
720 /* Regular piggyback - do it unless we're going to do sort */
721 *search_req->largeSetLowerBound = 2000000000;
722 *search_req->smallSetUpperBound = r->count;
723 *search_req->mediumSetPresentNumber = r->count;
724 smallSetElementSetName = 0; /* no need to provide this */
728 /* non-piggyback. Need not provide elementsets or syntaxes .. */
729 smallSetElementSetName = 0;
730 mediumSetElementSetName = 0;
733 if (smallSetElementSetName && *smallSetElementSetName)
735 Z_ElementSetNames *esn = odr_malloc (c->odr_out, sizeof(*esn));
737 esn->which = Z_ElementSetNames_generic;
738 esn->u.generic = odr_strdup (c->odr_out, smallSetElementSetName);
739 search_req->smallSetElementSetNames = esn;
741 if (mediumSetElementSetName && *mediumSetElementSetName)
743 Z_ElementSetNames *esn = odr_malloc (c->odr_out, sizeof(*esn));
745 esn->which = Z_ElementSetNames_generic;
746 esn->u.generic = odr_strdup (c->odr_out, mediumSetElementSetName);
747 search_req->mediumSetElementSetNames = esn;
750 search_req->preferredRecordSyntax =
751 yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, syntax);
755 if (c->support_named_resultsets)
759 /* find the lowest unused ordinal so that we re-use
760 result sets on the server. */
761 for (ord = 1; ; ord++)
764 sprintf (setname, "%d", ord);
765 for (rp = c->resultsets; rp; rp = rp->next)
766 if (rp->setname && !strcmp (rp->setname, setname))
771 r->setname = xstrdup (setname);
772 yaz_log (LOG_DEBUG, "allocating %s", r->setname);
775 r->setname = xstrdup ("default");
776 ZOOM_options_set (r->options, "setname", r->setname);
778 search_req->resultSetName = odr_strdup(c->odr_out, r->setname);
779 /* send search request */
785 static void response_diag (ZOOM_connection c, Z_DiagRec *p)
787 Z_DefaultDiagFormat *r;
792 if (p->which != Z_DiagRec_defaultFormat)
794 c->error = ZOOM_ERROR_DECODE;
797 r = p->u.defaultFormat;
800 case Z_DefaultDiagFormat_v2Addinfo:
801 addinfo = r->u.v2Addinfo;
803 case Z_DefaultDiagFormat_v3Addinfo:
804 addinfo = r->u.v3Addinfo;
808 c->addinfo = xstrdup (addinfo);
809 c->error = *r->condition;
812 ZOOM_record ZOOM_record_clone (ZOOM_record srec)
819 odr_enc = odr_createmem(ODR_ENCODE);
820 if (!z_NamePlusRecord (odr_enc, &srec->npr, 0, 0))
822 buf = odr_getbuf (odr_enc, &size, 0);
824 nrec = xmalloc (sizeof(*nrec));
825 nrec->odr = odr_createmem(ODR_DECODE);
826 nrec->wrbuf_marc = 0;
827 odr_setbuf (nrec->odr, buf, size, 0);
828 z_NamePlusRecord (nrec->odr, &nrec->npr, 0, 0);
830 odr_destroy (odr_enc);
834 ZOOM_record ZOOM_resultset_record_immediate (ZOOM_resultset s,size_t pos)
836 return record_cache_lookup (s, pos, 0);
839 ZOOM_record ZOOM_resultset_record (ZOOM_resultset r, size_t pos)
841 ZOOM_resultset_retrieve (r, 1, pos, 1);
842 return ZOOM_resultset_record_immediate (r, pos);
845 void ZOOM_record_destroy (ZOOM_record rec)
850 wrbuf_free (rec->wrbuf_marc, 1);
851 odr_destroy (rec->odr);
855 void *ZOOM_record_get (ZOOM_record rec, const char *type, size_t *len)
857 Z_NamePlusRecord *npr;
863 if (!strcmp (type, "database"))
865 return npr->databaseName;
867 else if (!strcmp (type, "syntax"))
869 if (npr->which == Z_NamePlusRecord_databaseRecord)
871 Z_External *r = (Z_External *) npr->u.databaseRecord;
872 oident *ent = oid_getentbyoid(r->direct_reference);
878 else if (!strcmp (type, "render"))
880 if (npr->which == Z_NamePlusRecord_databaseRecord)
882 Z_External *r = (Z_External *) npr->u.databaseRecord;
883 oident *ent = oid_getentbyoid(r->direct_reference);
885 if (r->which == Z_External_sutrs)
887 *len = r->u.sutrs->len;
888 return r->u.sutrs->buf;
890 else if (r->which == Z_External_octet)
899 case VAL_APPLICATION_XML:
902 if (!rec->wrbuf_marc)
903 rec->wrbuf_marc = wrbuf_alloc();
904 if (marc_display_wrbuf (r->u.octet_aligned->buf,
906 r->u.octet_aligned->len) > 0)
908 *len = wrbuf_len(rec->wrbuf_marc);
909 return wrbuf_buf(rec->wrbuf_marc);
912 *len = r->u.octet_aligned->len;
913 return r->u.octet_aligned->buf;
915 else if (r->which == Z_External_grs1)
923 else if (!strcmp (type, "raw"))
925 if (npr->which == Z_NamePlusRecord_databaseRecord)
928 return (Z_External *) npr->u.databaseRecord;
935 static void record_cache_add (ZOOM_resultset r,
936 Z_NamePlusRecord *npr,
938 const char *elementSetName)
940 ZOOM_record_cache rc;
942 for (rc = r->record_cache; rc; rc = rc->next)
946 if ((!elementSetName && !rc->elementSetName)
947 || (elementSetName && rc->elementSetName &&
948 !strcmp (elementSetName, rc->elementSetName)))
950 /* not destroying rc->npr (it's handled by nmem )*/
952 /* keeping wrbuf_marc too */
958 rc = odr_malloc (r->odr, sizeof(*rc));
961 rc->rec.wrbuf_marc = 0;
963 rc->elementSetName = odr_strdup (r->odr, elementSetName);
965 rc->elementSetName = 0;
967 rc->next = r->record_cache;
968 r->record_cache = rc;
971 static ZOOM_record record_cache_lookup (ZOOM_resultset r,
973 const char *elementSetName)
975 ZOOM_record_cache rc;
977 for (rc = r->record_cache; rc; rc = rc->next)
981 if ((!elementSetName && !rc->elementSetName)
982 || (elementSetName && rc->elementSetName &&
983 !strcmp (elementSetName, rc->elementSetName)))
990 static void handle_records (ZOOM_connection c, Z_Records *sr,
993 ZOOM_resultset resultset;
997 switch (c->tasks->which)
999 case ZOOM_TASK_SEARCH:
1000 resultset = c->tasks->u.search.resultset;
1002 case ZOOM_TASK_RETRIEVE:
1003 resultset = c->tasks->u.retrieve.resultset;
1008 if (sr && sr->which == Z_Records_NSD)
1010 Z_DiagRec dr, *dr_p = &dr;
1011 dr.which = Z_DiagRec_defaultFormat;
1012 dr.u.defaultFormat = sr->u.nonSurrogateDiagnostic;
1014 response_diag (c, dr_p);
1016 else if (sr && sr->which == Z_Records_multipleNSD)
1018 if (sr->u.multipleNonSurDiagnostics->num_diagRecs >= 1)
1019 response_diag(c, sr->u.multipleNonSurDiagnostics->diagRecs[0]);
1021 c->error = ZOOM_ERROR_DECODE;
1025 if (resultset->count + resultset->start > resultset->size)
1026 resultset->count = resultset->size - resultset->start;
1027 if (resultset->count < 0)
1028 resultset->count = 0;
1029 if (sr && sr->which == Z_Records_DBOSD)
1032 NMEM nmem = odr_extract_mem (c->odr_in);
1033 Z_NamePlusRecordList *p =
1034 sr->u.databaseOrSurDiagnostics;
1035 for (i = 0; i<p->num_records; i++)
1037 record_cache_add (resultset, p->records[i],
1038 i+ resultset->start, 0);
1040 /* transfer our response to search_nmem .. we need it later */
1041 nmem_transfer (resultset->odr->mem, nmem);
1042 nmem_destroy (nmem);
1043 if (present_phase && p->num_records == 0)
1045 /* present response and we didn't get any records! */
1046 c->error = ZOOM_ERROR_DECODE;
1049 else if (present_phase)
1051 /* present response and we didn't get any records! */
1052 c->error = ZOOM_ERROR_DECODE;
1057 static void handle_present_response (ZOOM_connection c, Z_PresentResponse *pr)
1059 handle_records (c, pr->records, 1);
1062 static void handle_search_response (ZOOM_connection c, Z_SearchResponse *sr)
1064 ZOOM_resultset resultset;
1066 yaz_log (LOG_DEBUG, "got search response");
1068 if (!c->tasks || c->tasks->which != ZOOM_TASK_SEARCH)
1071 resultset = c->tasks->u.search.resultset;
1073 resultset->size = *sr->resultCount;
1074 handle_records (c, sr->records, 0);
1077 static void sort_response (ZOOM_connection c, Z_SortResponse *res)
1079 if (res->diagnostics && res->num_diagnostics > 0)
1080 response_diag (c, res->diagnostics[0]);
1083 static int scan_response (ZOOM_connection c, Z_ScanResponse *res)
1085 NMEM nmem = odr_extract_mem (c->odr_in);
1088 if (!c->tasks || c->tasks->which != ZOOM_TASK_SCAN)
1090 scan = c->tasks->u.scan.scan;
1092 if (res->entries && res->entries->nonsurrogateDiagnostics)
1093 response_diag(c, res->entries->nonsurrogateDiagnostics[0]);
1094 scan->scan_response = res;
1095 nmem_transfer (scan->odr->mem, nmem);
1096 nmem_destroy (nmem);
1099 static int send_sort (ZOOM_connection c)
1101 ZOOM_resultset resultset;
1103 if (!c->tasks || c->tasks->which != ZOOM_TASK_SEARCH)
1106 resultset = c->tasks->u.search.resultset;
1110 resultset->r_sort_spec = 0;
1113 if (resultset->r_sort_spec)
1115 Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_sortRequest);
1116 Z_SortRequest *req = apdu->u.sortRequest;
1118 req->num_inputResultSetNames = 1;
1119 req->inputResultSetNames = (Z_InternationalString **)
1120 odr_malloc (c->odr_out, sizeof(*req->inputResultSetNames));
1121 req->inputResultSetNames[0] =
1122 odr_strdup (c->odr_out, resultset->setname);
1123 req->sortedResultSetName = odr_strdup (c->odr_out, resultset->setname);
1124 req->sortSequence = resultset->r_sort_spec;
1125 resultset->r_sort_spec = 0;
1126 send_APDU (c, apdu);
1132 static int send_present (ZOOM_connection c)
1134 Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_presentRequest);
1135 Z_PresentRequest *req = apdu->u.presentRequest;
1137 const char *syntax =
1138 ZOOM_options_get (c->options, "preferredRecordSyntax");
1139 const char *element =
1140 ZOOM_options_get (c->options, "elementSetName");
1141 const char *schema =
1142 ZOOM_options_get (c->options, "schema");
1143 ZOOM_resultset resultset;
1148 switch (c->tasks->which)
1150 case ZOOM_TASK_SEARCH:
1151 resultset = c->tasks->u.search.resultset;
1153 case ZOOM_TASK_RETRIEVE:
1154 resultset = c->tasks->u.retrieve.resultset;
1155 resultset->start = c->tasks->u.retrieve.start;
1156 resultset->count = c->tasks->u.retrieve.count;
1158 if (resultset->start >= resultset->size)
1160 if (resultset->start + resultset->count > resultset->size)
1161 resultset->count = resultset->size - resultset->start;
1167 if (c->error) /* don't continue on error */
1169 if (resultset->start < 0)
1171 for (i = 0; i<resultset->count; i++)
1174 record_cache_lookup (resultset, i + resultset->start, 0);
1178 if (i == resultset->count)
1181 resultset->start += i;
1182 resultset->count -= i;
1183 *req->resultSetStartPoint = resultset->start + 1;
1184 *req->numberOfRecordsRequested = resultset->count;
1185 assert (*req->numberOfRecordsRequested > 0);
1187 if (syntax && *syntax)
1188 req->preferredRecordSyntax =
1189 yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, syntax);
1191 if (schema && *schema)
1193 Z_RecordComposition *compo = odr_malloc (c->odr_out, sizeof(*compo));
1195 req->recordComposition = compo;
1196 compo->which = Z_RecordComp_complex;
1197 compo->u.complex = (Z_CompSpec *)
1198 odr_malloc(c->odr_out, sizeof(*compo->u.complex));
1199 compo->u.complex->selectAlternativeSyntax = (bool_t *)
1200 odr_malloc(c->odr_out, sizeof(bool_t));
1201 *compo->u.complex->selectAlternativeSyntax = 0;
1203 compo->u.complex->generic = (Z_Specification *)
1204 odr_malloc(c->odr_out, sizeof(*compo->u.complex->generic));
1206 compo->u.complex->generic->schema = (Odr_oid *)
1207 yaz_str_to_z3950oid (c->odr_out, CLASS_SCHEMA, schema);
1209 if (!compo->u.complex->generic->schema)
1211 /* OID wasn't a schema! Try record syntax instead. */
1213 compo->u.complex->generic->schema = (Odr_oid *)
1214 yaz_str_to_z3950oid (c->odr_out, CLASS_RECSYN, schema);
1216 if (element && *element)
1218 compo->u.complex->generic->elementSpec = (Z_ElementSpec *)
1219 odr_malloc(c->odr_out, sizeof(Z_ElementSpec));
1220 compo->u.complex->generic->elementSpec->which =
1221 Z_ElementSpec_elementSetName;
1222 compo->u.complex->generic->elementSpec->u.elementSetName =
1223 odr_strdup (c->odr_out, element);
1226 compo->u.complex->generic->elementSpec = 0;
1227 compo->u.complex->num_dbSpecific = 0;
1228 compo->u.complex->dbSpecific = 0;
1229 compo->u.complex->num_recordSyntax = 0;
1230 compo->u.complex->recordSyntax = 0;
1232 else if (element && *element)
1234 Z_ElementSetNames *esn = odr_malloc (c->odr_out, sizeof(*esn));
1235 Z_RecordComposition *compo = odr_malloc (c->odr_out, sizeof(*compo));
1237 esn->which = Z_ElementSetNames_generic;
1238 esn->u.generic = odr_strdup (c->odr_out, element);
1239 compo->which = Z_RecordComp_simple;
1240 compo->u.simple = esn;
1241 req->recordComposition = compo;
1243 req->resultSetId = odr_strdup(c->odr_out, resultset->setname);
1244 send_APDU (c, apdu);
1248 ZOOM_scanset ZOOM_connection_scan (ZOOM_connection c, const char *start)
1250 ZOOM_scanset scan = xmalloc (sizeof(*scan));
1252 scan->connection = c;
1253 scan->odr = odr_createmem (ODR_DECODE);
1254 scan->options = ZOOM_options_create_with_parent (c->options);
1256 scan->scan_response = 0;
1258 if ((scan->termListAndStartPoint =
1259 p_query_scan(scan->odr, PROTO_Z3950, &scan->attributeSet,
1262 ZOOM_task task = ZOOM_connection_add_task (c, ZOOM_TASK_SCAN);
1263 task->u.scan.scan = scan;
1268 while (ZOOM_event (1, &c))
1275 void ZOOM_scanset_destroy (ZOOM_scanset scan)
1280 if (scan->refcount == 0)
1282 odr_destroy (scan->odr);
1284 ZOOM_options_destroy (scan->options);
1289 int send_scan (ZOOM_connection c)
1292 Z_APDU *apdu = zget_APDU(c->odr_out, Z_APDU_scanRequest);
1293 Z_ScanRequest *req = apdu->u.scanRequest;
1296 assert (c->tasks->which == ZOOM_TASK_SCAN);
1297 scan = c->tasks->u.scan.scan;
1299 req->termListAndStartPoint = scan->termListAndStartPoint;
1300 req->attributeSet = scan->attributeSet;
1302 *req->numberOfTermsRequested =
1303 ZOOM_options_get_int(scan->options, "number", 10);
1305 req->preferredPositionInResponse =
1306 odr_intdup (c->odr_out,
1307 ZOOM_options_get_int(scan->options, "position", 1));
1310 odr_intdup (c->odr_out,
1311 ZOOM_options_get_int(scan->options, "stepSize", 0));
1313 req->databaseNames = set_DatabaseNames (c, scan->options,
1314 &req->num_databaseNames);
1316 send_APDU (c, apdu);
1321 size_t ZOOM_scanset_size (ZOOM_scanset scan)
1323 if (!scan || !scan->scan_response || !scan->scan_response->entries)
1325 return scan->scan_response->entries->num_entries;
1328 const char *ZOOM_scanset_term (ZOOM_scanset scan, size_t i, int *occ, size_t *len)
1330 const char *term = 0;
1331 size_t noent = ZOOM_scanset_size (scan);
1332 Z_ScanResponse *res = scan->scan_response;
1338 if (res->entries->entries[i]->which == Z_Entry_termInfo)
1340 Z_TermInfo *t = res->entries->entries[i]->u.termInfo;
1342 if (t->term->which == Z_Term_general)
1344 term = t->term->u.general->buf;
1345 *len = t->term->u.general->len;
1347 *occ = t->globalOccurrences ? *t->globalOccurrences : 0;
1352 static int ZOOM_connection_exec_task (ZOOM_connection c)
1354 ZOOM_task task = c->tasks;
1356 yaz_log (LOG_DEBUG, "ZOOM_connection_exec_task");
1359 if (c->error != ZOOM_ERROR_NONE ||
1360 (!c->cs && task->which != ZOOM_TASK_CONNECT))
1362 ZOOM_connection_remove_tasks (c);
1365 yaz_log (LOG_DEBUG, "ZOOM_connection_exec_task type=%d", task->which);
1369 switch (task->which)
1371 case ZOOM_TASK_SEARCH:
1372 /* see if search hasn't been sent yet. */
1373 if (ZOOM_connection_send_search (c))
1376 case ZOOM_TASK_RETRIEVE:
1377 if (send_present (c))
1380 case ZOOM_TASK_CONNECT:
1384 case ZOOM_TASK_SCAN:
1388 ZOOM_connection_remove_task (c);
1392 static int send_sort_present (ZOOM_connection c)
1394 int r = send_sort (c);
1396 r = send_present (c);
1400 static void handle_apdu (ZOOM_connection c, Z_APDU *apdu)
1402 Z_InitResponse *initrs;
1404 yaz_log (LOG_DEBUG, "hande_apdu type=%d", apdu->which);
1408 case Z_APDU_initResponse:
1409 initrs = apdu->u.initResponse;
1410 if (!*initrs->result)
1412 c->error = ZOOM_ERROR_INIT;
1417 yaz_oi_get_string_oidval (&apdu->u.initResponse->otherInfo,
1419 xfree (c->cookie_in);
1422 c->cookie_in = xstrdup(cookie);
1423 if (ODR_MASK_GET(initrs->options, Z_Options_namedResultSets) &&
1424 ODR_MASK_GET(initrs->protocolVersion, Z_ProtocolVersion_3))
1425 c->support_named_resultsets = 1;
1428 assert (c->tasks->which == ZOOM_TASK_CONNECT);
1429 ZOOM_connection_remove_task (c);
1431 ZOOM_connection_exec_task (c);
1434 case Z_APDU_searchResponse:
1435 handle_search_response (c, apdu->u.searchResponse);
1436 if (!send_sort_present (c))
1437 ZOOM_connection_remove_task (c);
1439 case Z_APDU_presentResponse:
1440 handle_present_response (c, apdu->u.presentResponse);
1441 if (!send_present (c))
1442 ZOOM_connection_remove_task (c);
1444 case Z_APDU_sortResponse:
1445 sort_response (c, apdu->u.sortResponse);
1446 if (!send_present (c))
1447 ZOOM_connection_remove_task (c);
1449 case Z_APDU_scanResponse:
1450 scan_response (c, apdu->u.scanResponse);
1451 ZOOM_connection_remove_task (c);
1455 static int do_read (ZOOM_connection c)
1461 event = ZOOM_Event_create (ZOOM_EVENT_RECV_DATA);
1462 ZOOM_connection_put_event (c, event);
1464 r = cs_get (c->cs, &c->buf_in, &c->len_in);
1469 c->error= ZOOM_ERROR_CONNECTION_LOST;
1475 odr_reset (c->odr_in);
1476 odr_setbuf (c->odr_in, c->buf_in, r, 0);
1477 event = ZOOM_Event_create (ZOOM_EVENT_RECV_APDU);
1478 ZOOM_connection_put_event (c, event);
1479 if (!z_APDU (c->odr_in, &apdu, 0, 0))
1481 c->error = ZOOM_ERROR_DECODE;
1486 handle_apdu (c, apdu);
1492 static int do_write_ex (ZOOM_connection c, char *buf_out, int len_out)
1497 event = ZOOM_Event_create(ZOOM_EVENT_SEND_DATA);
1498 ZOOM_connection_put_event (c, event);
1500 if ((r=cs_put (c->cs, buf_out, len_out)) < 0)
1502 if (c->state == STATE_CONNECTING)
1503 c->error = ZOOM_ERROR_CONNECT;
1505 c->error = ZOOM_ERROR_CONNECTION_LOST;
1511 c->state = STATE_ESTABLISHED;
1512 c->mask = ZOOM_SELECT_READ|ZOOM_SELECT_WRITE|ZOOM_SELECT_EXCEPT;
1516 c->state = STATE_ESTABLISHED;
1517 c->mask = ZOOM_SELECT_READ|ZOOM_SELECT_EXCEPT;
1522 static int do_write(ZOOM_connection c)
1524 return do_write_ex (c, c->buf_out, c->len_out);
1528 const char *ZOOM_connection_option_get (ZOOM_connection c, const char *key)
1530 return ZOOM_options_get (c->options, key);
1533 void ZOOM_connection_option_set (ZOOM_connection c, const char *key,
1536 ZOOM_options_set (c->options, key, val);
1539 const char *ZOOM_resultset_option_get (ZOOM_resultset r, const char *key)
1541 return ZOOM_options_get (r->options, key);
1544 void ZOOM_resultset_option_set (ZOOM_resultset r, const char *key,
1547 ZOOM_options_set (r->options, key, val);
1551 int ZOOM_connection_errcode (ZOOM_connection c)
1553 return ZOOM_connection_error (c, 0, 0);
1556 const char *ZOOM_connection_errmsg (ZOOM_connection c)
1559 ZOOM_connection_error (c, &msg, 0);
1563 const char *ZOOM_connection_addinfo (ZOOM_connection c)
1565 const char *addinfo;
1566 ZOOM_connection_error (c, 0, &addinfo);
1570 int ZOOM_connection_error (ZOOM_connection c, const char **cp,
1571 const char **addinfo)
1573 int error = c->error;
1578 case ZOOM_ERROR_NONE:
1579 *cp = "No error"; break;
1580 case ZOOM_ERROR_CONNECT:
1581 *cp = "Connect failed"; break;
1582 case ZOOM_ERROR_MEMORY:
1583 *cp = "Out of memory"; break;
1584 case ZOOM_ERROR_ENCODE:
1585 *cp = "Encoding failed"; break;
1586 case ZOOM_ERROR_DECODE:
1587 *cp = "Decoding failed"; break;
1588 case ZOOM_ERROR_CONNECTION_LOST:
1589 *cp = "Connection lost"; break;
1590 case ZOOM_ERROR_INIT:
1591 *cp = "Init rejected"; break;
1592 case ZOOM_ERROR_INTERNAL:
1593 *cp = "Internal failure"; break;
1594 case ZOOM_ERROR_TIMEOUT:
1595 *cp = "Timeout"; break;
1597 *cp = diagbib1_str (error);
1603 *addinfo = c->addinfo;
1610 int ZOOM_connection_do_io(ZOOM_connection c, int mask)
1612 ZOOM_Event event = 0;
1614 int r = cs_look(c->cs);
1615 yaz_log (LOG_LOG, "ZOOM_connection_do_io c=%p mask=%d cs_look=%d",
1620 event = ZOOM_Event_create (ZOOM_EVENT_IO_CONNECT);
1621 c->error = ZOOM_ERROR_CONNECT;
1623 ZOOM_connection_put_event (c, event);
1625 else if (r == CS_CONNECT)
1627 event = ZOOM_Event_create (ZOOM_EVENT_IO_CONNECT);
1628 yaz_log (LOG_LOG, "calling rcvconnect");
1629 if (cs_rcvconnect (c->cs) < 0)
1631 c->error = ZOOM_ERROR_CONNECT;
1633 ZOOM_connection_put_event (c, event);
1637 ZOOM_connection_put_event (c, event);
1638 ZOOM_connection_send_init (c);
1643 if (mask & ZOOM_SELECT_READ)
1645 if (c->cs && (mask & ZOOM_SELECT_WRITE))
1649 yaz_log (LOG_DEBUG, "ZOOM_connection_do_io c=%p mask=%d", c, mask);
1650 if (c->state == STATE_CONNECTING)
1652 event = ZOOM_Event_create (ZOOM_EVENT_CONNECT);
1653 if (mask & ZOOM_SELECT_WRITE)
1655 ZOOM_connection_put_event (c, event);
1656 ZOOM_connection_send_init (c);
1660 c->error = ZOOM_ERROR_CONNECT;
1662 ZOOM_connection_put_event (c, event);
1665 else if (c->state == STATE_ESTABLISHED)
1667 if (mask & ZOOM_SELECT_READ)
1669 if (c->cs && (mask & ZOOM_SELECT_WRITE))
1674 event = ZOOM_Event_create (ZOOM_EVENT_UNKNOWN);
1675 ZOOM_connection_put_event (c, event);
1676 c->error = ZOOM_ERROR_INTERNAL;
1683 int ZOOM_connection_last_event(ZOOM_connection cs)
1686 return ZOOM_EVENT_NONE;
1687 return cs->last_event;
1690 int ZOOM_event (int no, ZOOM_connection *cs)
1693 struct pollfd pollfds[1024];
1694 ZOOM_connection poll_cs[1024];
1697 fd_set input, output, except;
1702 for (i = 0; i<no; i++)
1704 ZOOM_connection c = cs[i];
1706 if (c && (event = ZOOM_connection_get_event(c)))
1708 ZOOM_Event_destroy (event);
1712 for (i = 0; i<no; i++)
1714 ZOOM_connection c = cs[i];
1716 if (c && ZOOM_connection_exec_task (c))
1718 if ((event = ZOOM_connection_get_event(c)))
1720 ZOOM_Event_destroy (event);
1736 for (i = 0; i<no; i++)
1738 ZOOM_connection c = cs[i];
1743 fd = z3950_connection_socket(c);
1744 mask = z3950_connection_mask(c);
1754 short poll_events = 0;
1756 if (mask & ZOOM_SELECT_READ)
1757 poll_events += POLLIN;
1758 if (mask & ZOOM_SELECT_WRITE)
1759 poll_events += POLLOUT;
1760 if (mask & ZOOM_SELECT_EXCEPT)
1761 poll_events += POLLERR;
1762 pollfds[nfds].fd = fd;
1763 pollfds[nfds].events = poll_events;
1764 pollfds[nfds].revents = 0;
1769 if (mask & ZOOM_SELECT_READ)
1771 FD_SET (fd, &input);
1774 if (mask & ZOOM_SELECT_WRITE)
1776 FD_SET (fd, &output);
1779 if (mask & ZOOM_SELECT_EXCEPT)
1781 FD_SET (fd, &except);
1789 yaz_log (LOG_DEBUG, "poll start");
1790 r = poll (pollfds, nfds, 15000);
1791 yaz_log (LOG_DEBUG, "poll stop, returned r=%d", r);
1792 for (i = 0; i<nfds; i++)
1794 ZOOM_connection c = poll_cs[i];
1798 if (pollfds[i].revents & POLLIN)
1799 mask += ZOOM_SELECT_READ;
1800 if (pollfds[i].revents & POLLOUT)
1801 mask += ZOOM_SELECT_WRITE;
1802 if (pollfds[i].revents & POLLERR)
1803 mask += ZOOM_SELECT_EXCEPT;
1805 ZOOM_connection_do_io(c, mask);
1807 else if (r == 0 && c->mask)
1809 ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_TIMEOUT);
1810 /* timeout and this connection was waiting */
1811 c->error = ZOOM_ERROR_TIMEOUT;
1813 ZOOM_connection_put_event(c, event);
1817 yaz_log (LOG_DEBUG, "select start");
1818 r = select (max_fd+1, &input, &output, &except, &tv);
1819 yaz_log (LOG_DEBUG, "select stop, returned r=%d", r);
1820 for (i = 0; i<no; i++)
1822 ZOOM_connection c = cs[i];
1827 fd = z3950_connection_socket(c);
1831 /* no timeout and real socket */
1832 if (FD_ISSET(fd, &input))
1833 mask += ZOOM_SELECT_READ;
1834 if (FD_ISSET(fd, &output))
1835 mask += ZOOM_SELECT_WRITE;
1836 if (FD_ISSET(fd, &except))
1837 mask += ZOOM_SELECT_EXCEPT;
1839 ZOOM_connection_do_io(c, mask);
1841 if (r == 0 && c->mask)
1843 ZOOM_Event event = ZOOM_Event_create(ZOOM_EVENT_IO_TIMEOUT);
1844 /* timeout and this connection was waiting */
1845 c->error = ZOOM_ERROR_TIMEOUT;
1847 yaz_log (LOG_DEBUG, "timeout");
1848 ZOOM_connection_put_event(c, event);
1852 for (i = 0; i<no; i++)
1854 ZOOM_connection c = cs[i];
1856 if (c && (event = ZOOM_connection_get_event(c)))
1858 ZOOM_Event_destroy (event);