Client accepts multiple database names.
[yaz-moved-to-github.git] / client / client.c
index 9646c2e..5edf457 100644 (file)
@@ -1,10 +1,59 @@
 /*
- * Copyright (c) 1995, Index Data.
+ * Copyright (c) 1995-1996, Index Data.
  * See the file LICENSE for details.
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: client.c,v $
- * Revision 1.27  1995-12-12 16:37:02  quinn
+ * Revision 1.43  1996-11-08 11:03:26  adam
+ * Client accepts multiple database names.
+ *
+ * Revision 1.42  1996/10/08 10:44:57  quinn
+ * Resolved conflicts.
+ *
+ * Revision 1.41  1996/10/07  15:29:03  quinn
+ * Work
+ *
+ * Revision 1.40  1996/08/29  14:19:34  quinn
+ * Fixed conflict (CVS)
+ *
+ * Revision 1.39  1996/08/27  10:43:22  quinn
+ * Made select() optional
+ *
+ * Revision 1.38  1996/08/12  14:09:11  adam
+ * Default prefix query attribute set defined by using p_query_attset.
+ *
+ * Revision 1.37  1996/07/06  19:58:29  quinn
+ * System headerfiles gathered in yconfig
+ *
+ * Revision 1.36  1996/06/10  08:53:47  quinn
+ * Added Summary
+ *
+ * Revision 1.35  1996/06/03  09:45:50  quinn
+ * Added display of OIDs in the GRS routine.
+ *
+ * Revision 1.34  1996/05/09  07:26:49  quinn
+ * *** empty log message ***
+ *
+ * Revision 1.33  1996/05/09  07:25:22  quinn
+ * Small
+ *
+ * Revision 1.32  1996/03/15  11:05:33  adam
+ * The user can set the preferred query type (prefix, ccl, ..) with the
+ * querytype command.
+ *
+ * Revision 1.31  1996/02/20  12:51:54  quinn
+ * Fixed problems with EXTERNAL.
+ *
+ * Revision 1.30  1996/02/12  18:18:09  quinn
+ * Fidgeting.
+ *
+ * Revision 1.29  1996/01/02  08:57:25  quinn
+ * Changed enums in the ASN.1 .h files to #defines. Changed oident.class to oclass
+ *
+ * Revision 1.28  1995/12/14  11:09:31  quinn
+ * Added Explain record syntax to the format command.
+ *
+ * Revision 1.27  1995/12/12  16:37:02  quinn
  * Added destroy element to data1_node.
  *
  * Revision 1.26  1995/12/12  14:11:00  quinn
  * to illustrate the use of the YAZ service-level API.
  */
 
+#include <yconfig.h>
 #include <stdio.h>
 #include <stdlib.h>
-#ifdef WINDOWS
 #include <time.h>
-#else
-#include <sys/time.h>
-#endif
 #include <assert.h>
-#ifdef _AIX
-#include <sys/select.h>
-#endif
 
 #include <comstack.h>
 #include <tcpip.h>
 #include <marcdisp.h>
 #include <diagbib1.h>
 
-#ifdef RPN_QUERY
-#ifdef PREFIX_QUERY
 #include <pquery.h>
-#else
+
+#if CCL2RPN
 #include <yaz-ccl.h>
 #endif
-#endif
 
 #define C_PROMPT "Z> "
 
 static ODR out, in, print;              /* encoding and decoding streams */
 static COMSTACK conn = 0;               /* our z-association */
 static Z_IdAuthentication *auth = 0;    /* our current auth definition */
-static char database[512] = "Default";  /* Database name */
+static char *databaseNames[128];
+static int num_databaseNames = 0;
 static int setnumber = 0;               /* current result set number */
 static int smallSetUpperBound = 0;
 static int largeSetLowerBound = 1;
@@ -138,15 +180,24 @@ static int setno = 1;                   /* current set offset */
 static int protocol = PROTO_Z3950;      /* current app protocol */
 static int recordsyntax = VAL_USMARC;
 static int sent_close = 0;
-static ODR_MEM session_mem;                /* memory handle for init-response */
-static Z_InitResponse *session = 0;        /* session parameters */
+static ODR_MEM session_mem;             /* memory handle for init-response */
+static Z_InitResponse *session = 0;     /* session parameters */
 static char last_scan[512] = "0";
 static char last_cmd[100] = "?";
-#ifdef RPN_QUERY
-#ifndef PREFIX_QUERY
+static FILE *marcdump = 0;
+static char marcdump_file[512] = "marc.out";
+
+typedef enum {
+    QueryType_Prefix,
+    QueryType_CCL,
+    QueryType_CCL2RPN
+} QueryType;
+
+static QueryType queryType = QueryType_Prefix;
+
+#if CCL2RPN
 static CCL_bibset bibset;               /* CCL bibset handle */
 #endif
-#endif
 
 static void send_apdu(Z_APDU *a)
 {
@@ -159,7 +210,7 @@ static void send_apdu(Z_APDU *a)
         exit(1);
     }
     buf = odr_getbuf(out, &len, 0);
-    odr_reset(out); /* release the APDU */
+    odr_reset(out); /* release the APDU structure  */
     if (cs_put(conn, buf, len) < 0)
     {
         fprintf(stderr, "cs_put: %s", cs_errmsg(cs_errno(conn)));
@@ -354,6 +405,21 @@ static void display_grs1(Z_GenericRecord *r, int level)
             printf("%s\n", t->content->u.string);
         else if (t->content->which == Z_ElementData_numeric)
            printf("%d\n", *t->content->u.numeric);
+       else if (t->content->which == Z_ElementData_oid)
+       {
+           int *ip = t->content->u.oid;
+           oident *oent;
+
+           if ((oent = oid_getentbyoid(t->content->u.oid)))
+               printf("OID: %s\n", oent->desc);
+           else
+           {
+               printf("{");
+               while (ip && *ip >= 0)
+                   printf(" %d", *(ip++));
+               printf(" }\n");
+           }
+       }
        else if (t->content->which == Z_ElementData_noDataRequested)
            printf("[No data requested]\n");
        else if (t->content->which == Z_ElementData_elementEmpty)
@@ -383,6 +449,9 @@ static void display_record(Z_DatabaseRecord *p)
     Z_External *r = (Z_External*) p;
     oident *ent = oid_getentbyoid(r->direct_reference);
 
+    /*
+     * Tell the user what we got.
+     */
     if (r->direct_reference)
     {
         printf("Record type: ");
@@ -394,7 +463,41 @@ static void display_record(Z_DatabaseRecord *p)
             odr_reset(print);
         }
     }
-    if (r->which == Z_External_octet && p->u.octet_aligned->len)
+
+    /* Check if this is a known, ASN.1 type tucked away in an octet string */
+    if (ent && r->which == Z_External_octet)
+    {
+       Z_ext_typeent *type = z_ext_getentbyref(ent->value);
+       void *rr;
+
+       if (type)
+       {
+           /*
+            * Call the given decoder to process the record.
+            */
+           odr_setbuf(in, (char*)p->u.octet_aligned->buf,
+               p->u.octet_aligned->len, 0);
+           if (!(*type->fun)(in, &rr, 0))
+           {
+               odr_perror(in, "Decoding constructed record.");
+               fprintf(stderr, "[Near %d]\n", odr_offset(in));
+               fprintf(stderr, "Packet dump:\n---------\n");
+               odr_dumpBER(stderr, (char*)p->u.octet_aligned->buf,
+                   p->u.octet_aligned->len);
+               fprintf(stderr, "---------\n");
+               exit(1);
+           }
+           /*
+            * Note: we throw away the original, BER-encoded record here.
+            * Do something else with it if you want to keep it.
+            */
+           r->u.sutrs = rr;    /* we don't actually check the type here. */
+           r->which = type->what;
+       }
+    }
+    if (ent->value == VAL_SOIF)
+        printf("%.*s", r->u.octet_aligned->len, r->u.octet_aligned->buf);
+    else if (r->which == Z_External_octet && p->u.octet_aligned->len)
         marc_display ((char*)p->u.octet_aligned->buf, stdout);
     else if (ent->value == VAL_SUTRS)
     {
@@ -445,11 +548,11 @@ static void display_diagrec(Z_DiagRec *p)
         r = p->u.defaultFormat;
 #endif
     if (!(ent = oid_getentbyoid(r->diagnosticSetId)) ||
-        ent->class != CLASS_DIAGSET || ent->value != VAL_BIB1)
+        ent->oclass != CLASS_DIAGSET || ent->value != VAL_BIB1)
         printf("Missing or unknown diagset\n");
     printf("    [%d] %s", *r->condition, diagbib1_str(*r->condition));
     if (r->addinfo && *r->addinfo)
-        printf(" -- %s\n", r->addinfo);
+        printf(" -- '%s'\n", r->addinfo);
     else
         printf("\n");
 }
@@ -482,32 +585,27 @@ static int send_searchRequest(char *arg)
 {
     Z_APDU *apdu = zget_APDU(out, Z_APDU_searchRequest);
     Z_SearchRequest *req = apdu->u.searchRequest;
-    char *databaseNames = database;
     Z_Query query;
-#ifdef RPN_QUERY
-#ifndef PREFIX_QUERY
+#if CCL2RPN
     struct ccl_rpn_node *rpn;
     int error, pos;
-#endif
+    oident bib1;
 #endif
     char setstring[100];
-#ifdef RPN_QUERY
     Z_RPNQuery *RPNquery;
-    oident bib1;
-#else
     Odr_oct ccl_query;
-#endif
 
-#ifdef RPN_QUERY
-#ifndef PREFIX_QUERY
-    rpn = ccl_find_str(bibset, arg, &error, &pos);
-    if (error)
+#if CCL2RPN
+    if (queryType == QueryType_CCL2RPN)
     {
-        printf("CCL ERROR: %s\n", ccl_err_msg(error));
-        return 0;
+        rpn = ccl_find_str(bibset, arg, &error, &pos);
+        if (error)
+        {
+            printf("CCL ERROR: %s\n", ccl_err_msg(error));
+            return 0;
+        }
     }
 #endif
-#endif
 
     if (!strcmp(arg, "@big")) /* strictly for troublemaking */
     {
@@ -535,42 +633,51 @@ static int send_searchRequest(char *arg)
         oident prefsyn;
 
         prefsyn.proto = protocol;
-        prefsyn.class = CLASS_RECSYN;
+        prefsyn.oclass = CLASS_RECSYN;
         prefsyn.value = recordsyntax;
-        req->preferredRecordSyntax = odr_oiddup(out, oid_getoidbyent(&prefsyn));
+        req->preferredRecordSyntax =
+            odr_oiddup(out, oid_getoidbyent(&prefsyn));
         req->smallSetElementSetNames =
             req->mediumSetElementSetNames = elementSetNames;
     }
-    req->num_databaseNames = 1;
-    req->databaseNames = &databaseNames;
+    req->num_databaseNames = num_databaseNames;
+    req->databaseNames = databaseNames;
 
     req->query = &query;
 
-#ifdef RPN_QUERY
-    query.which = Z_Query_type_1;
-
-#ifndef PREFIX_QUERY
-    assert((RPNquery = ccl_rpn_query(rpn)));
-#else
-    RPNquery = p_query_rpn (out, arg);
-    if (!RPNquery)
+    switch (queryType)
     {
-        printf("Prefix query error\n");
+    case QueryType_Prefix:
+        query.which = Z_Query_type_1;
+        RPNquery = p_query_rpn (out, protocol, arg);
+        if (!RPNquery)
+        {
+            printf("Prefix query error\n");
+            return 0;
+        }
+        query.u.type_1 = RPNquery;
+        break;
+    case QueryType_CCL:
+        query.which = Z_Query_type_2;
+        query.u.type_2 = &ccl_query;
+        ccl_query.buf = (unsigned char*) arg;
+        ccl_query.len = strlen(arg);
+        break;
+#if CCL2RPN
+    case QueryType_CCL2RPN:
+        query.which = Z_Query_type_1;
+        assert((RPNquery = ccl_rpn_query(rpn)));
+        bib1.proto = protocol;
+        bib1.oclass = CLASS_ATTSET;
+        bib1.value = VAL_BIB1;
+        RPNquery->attributeSetId = oid_getoidbyent(&bib1);
+        query.u.type_1 = RPNquery;
+        break;
+#endif
+    default:
+        printf ("Unsupported query type\n");
         return 0;
     }
-#endif
-    bib1.proto = protocol;
-    bib1.class = CLASS_ATTSET;
-    bib1.value = VAL_BIB1;
-    RPNquery->attributeSetId = oid_getoidbyent(&bib1);
-    query.u.type_1 = RPNquery;
-#else
-    query.which = Z_Query_type_2;
-    query.u.type_2 = &ccl_query;
-    ccl_query.buf = (unsigned char*) arg;
-    ccl_query.len = strlen(arg);
-#endif
-
     send_apdu(apdu);
     setno = 1;
     printf("Sent searchRequest.\n");
@@ -641,12 +748,30 @@ static int cmd_status(char *arg)
 
 static int cmd_base(char *arg)
 {
+    int i;
+    char *cp;
+
     if (!*arg)
     {
-        printf("Usage: base <database>\n");
+        printf("Usage: base <database> <database> ...\n");
         return 0;
     }
-    strcpy(database, arg);
+    for (i = 0; i<num_databaseNames; i++)
+        xfree (databaseNames[i]);
+    num_databaseNames = 0;
+    while (1)
+    {
+        if (!(cp = strchr(arg, ' ')))
+            cp = arg + strlen(arg);
+        if (cp - arg < 1)
+            break;
+        databaseNames[num_databaseNames] = xmalloc (1 + cp - arg);
+        memcpy (databaseNames[num_databaseNames], arg, cp - arg);
+        databaseNames[num_databaseNames++][cp - arg] = '\0';
+        if (!*cp)
+            break;
+        arg = cp+1;
+    }
     return 1;
 }
 
@@ -690,10 +815,28 @@ static int send_presentRequest(char *arg)
         sprintf(setstring, "%d", setnumber);
         req->resultSetId = setstring;
     }
+
+
+#if 0
+    if (1)
+    {
+       static Z_Range range;
+       static Z_Range *rangep = &range;
+    req->num_ranges = 1;
+#endif
+
+
+
+
+
+
+
+
+
     req->resultSetStartPoint = &setno;
     req->numberOfRecordsRequested = &nos;
     prefsyn.proto = protocol;
-    prefsyn.class = CLASS_RECSYN;
+    prefsyn.oclass = CLASS_RECSYN;
     prefsyn.value = recordsyntax;
     req->preferredRecordSyntax = oid_getoidbyent(&prefsyn);
     if (elementSetNames)
@@ -781,16 +924,11 @@ int send_scanrequest(char *string, int pp, int num)
 {
     Z_APDU *apdu = zget_APDU(out, Z_APDU_scanRequest);
     Z_ScanRequest *req = apdu->u.scanRequest;
-    char *db = database;
-    oident attset;
-
-    req->num_databaseNames = 1;
-    req->databaseNames = &db;
-    attset.proto = protocol;
-    attset.class = CLASS_ATTSET;
-    attset.value = VAL_BIB1;
-    req->attributeSet = oid_getoidbyent(&attset);
-    req->termListAndStartPoint = p_query_scan(out, string);
+
+    req->num_databaseNames = num_databaseNames;
+    req->databaseNames = databaseNames;
+    req->termListAndStartPoint = p_query_scan(out, protocol,
+                                              &req->attributeSet, string);
     req->numberOfTermsRequested = &num;
     req->preferredPositionInResponse = &pp;
     send_apdu(apdu);
@@ -885,15 +1023,39 @@ int cmd_format(char *arg)
         recordsyntax = VAL_DANMARC;
         return 1;
     }
+    else if (!strcmp(arg, "ukmarc"))
+    {
+        printf("Preferred format is UKMARC\n");
+        recordsyntax = VAL_UKMARC;
+        return 1;
+    }
     else if (!strcmp(arg, "grs1"))
     {
         printf("Preferred format is GRS1\n");
         recordsyntax = VAL_GRS1;
         return 1;
     }
+    else if (!strcmp(arg, "soif"))
+    {
+        printf("Preferred format is SOIF\n");
+        recordsyntax = VAL_SOIF;
+        return 1;
+    }
+    else if (!strcmp(arg, "summary"))
+    {
+        printf("Preferred format is Summary\n");
+        recordsyntax = VAL_SUMMARY;
+        return 1;
+    }
+    else if (!strcmp(arg, "explain"))
+    {
+        printf("Preferred format is Explain\n");
+        recordsyntax = VAL_EXPLAIN;
+        return 1;
+    }
     else
     {
-        printf("Specify one of {sutrs,usmarc,danmarc,grs1}.\n");
+        printf("Specify one of {sutrs,usmarc,danmarc,ukmarc,grs1,summary,explain}.\n");
         return 0;
     }
 }
@@ -915,6 +1077,47 @@ int cmd_elements(char *arg)
     return 1;
 }
 
+int cmd_attributeset(char *arg)
+{
+    char what[100];
+
+    if (!arg || !*arg)
+    {
+       printf("Usage: attributeset <setname>\n");
+       return 0;
+    }
+    sscanf(arg, "%s", what);
+    if (p_query_attset (what))
+    {
+       printf("Unknown attribute set name\n");
+       return 0;
+    }
+    return 1;
+}
+
+int cmd_querytype (char *arg)
+{
+    if (!strcmp (arg, "ccl"))
+        queryType = QueryType_CCL;
+    else if (!strcmp (arg, "prefix"))
+        queryType = QueryType_Prefix;
+#if CCL2RPN
+    else if (!strcmp (arg, "ccl2rpn") || !strcmp (arg, "cclrpn"))
+        queryType = QueryType_CCL2RPN;
+#endif
+    else
+    {
+        printf ("Querytype must be one of:\n");
+        printf (" prefix         - Prefix query\n");
+        printf (" ccl            - CCL query\n");
+#if CCL2RPN
+        printf (" ccl2rpn        - CCL query converted to RPN\n");
+#endif
+        return 0;
+    }
+    return 1;
+}
+
 int cmd_close(char *arg)
 {
     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
@@ -929,11 +1132,9 @@ int cmd_close(char *arg)
 
 static void initialize(void)
 {
-#ifdef RPN_QUERY
-#ifndef PREFIX_QUERY
+#if CCL2RPN
     FILE *inf;
 #endif
-#endif
 
     if (!(out = odr_createmem(ODR_ENCODE)) ||
         !(in = odr_createmem(ODR_DECODE)) ||
@@ -944,8 +1145,7 @@ static void initialize(void)
     }
     setvbuf(stdout, 0, _IONBF, 0);
 
-#ifdef RPN_QUERY
-#ifndef PREFIX_QUERY
+#if CCL2RPN
     bibset = ccl_qual_mk (); 
     inf = fopen ("default.bib", "r");
     if (inf)
@@ -954,23 +1154,22 @@ static void initialize(void)
         fclose (inf);
     }
 #endif
-#endif
 }
 
-static int client(void)
+static int client(int wait)
 {
     static struct {
         char *cmd;
         int (*fun)(char *arg);
         char *ad;
     } cmd[] = {
-        {"open", cmd_open, "('tcp'|'osi')':'[<TSEL>'/']<HOST>[':'<PORT>]"},
+        {"open", cmd_open, "('tcp'|'osi')':'[<tsel>'/']<host>[':'<port>]"},
         {"quit", cmd_quit, ""},
-        {"find", cmd_find, "<CCL-QUERY>"},
-        {"base", cmd_base, "<BASE-NAME>"},
-        {"show", cmd_show, "<REC#>['+'<#RECS>]"},
-        {"scan", cmd_scan, "<TERM>"},
-        {"authentication", cmd_authentication, "<ACCTSTRING>"},
+        {"find", cmd_find, "<query>"},
+        {"base", cmd_base, "<base-name>"},
+        {"show", cmd_show, "<rec#>['+'<#recs>]"},
+        {"scan", cmd_scan, "<term>"},
+        {"authentication", cmd_authentication, "<acctstring>"},
         {"lslb", cmd_lslb, "<largeSetLowerBound>"},
         {"ssub", cmd_ssub, "<smallSetUpperBound>"},
         {"mspn", cmd_mspn, "<mediumSetPresentNumber>"},
@@ -980,6 +1179,8 @@ static int client(void)
         {"format", cmd_format, "<recordsyntax>"},
         {"elements", cmd_elements, "<elementSetName>"},
         {"close", cmd_close, ""},
+       {"attributeset", cmd_attributeset, "<attrset>"},
+        {"querytype", cmd_querytype, "<type>"},
         {0,0}
     };
     char *netbuffer= 0;
@@ -990,9 +1191,12 @@ static int client(void)
     while (1)
     {
         int res;
+#ifdef USE_SELECT
         fd_set input;
+#endif
         char line[1024], word[1024], arg[1024];
 
+#ifdef USE_SELECT
         FD_ZERO(&input);
         FD_SET(0, &input);
         if (conn)
@@ -1004,7 +1208,10 @@ static int client(void)
         }
         if (!res)
             continue;
-        if (FD_ISSET(0, &input))
+        if (!wait && FD_ISSET(0, &input))
+#else
+       if (!wait)
+#endif
         {
             /* quick & dirty way to get a command line. */
             if (!gets(line))
@@ -1032,9 +1239,14 @@ static int client(void)
                 res = 1;
             }
             if (res < 2)
+           {
                 printf(C_PROMPT);
+               continue;
+           }
         }
+#ifdef USE_SELECT
         if (conn && FD_ISSET(cs_fileno(conn), &input))
+#endif
         {
             do
             {
@@ -1101,6 +1313,7 @@ static int client(void)
             }
             while (cs_more(conn));
         }
+       wait = 0;
     }
     return 0;
 }
@@ -1108,9 +1321,15 @@ static int client(void)
 int main(int argc, char **argv)
 {
     initialize();
+    cmd_base("Default");
     if (argc > 1)
         cmd_open(argv[1]);
     else
         printf(C_PROMPT);
-    return client();
+    if (*marcdump_file && !(marcdump = fopen(marcdump_file, "a")))
+    {
+       perror(marcdump_file);
+       exit(1);
+    }
+    return client((argc > 1));
 }