MARC records embedded in C code for test server
[yaz-moved-to-github.git] / ztest / ztest.c
index 45ae5b3..a021b64 100644 (file)
@@ -2,7 +2,7 @@
  * Copyright (c) 1995-2002, Index Data.
  * See the file LICENSE for details.
  *
- * $Id: ztest.c,v 1.47 2002-01-17 21:04:24 adam Exp $
+ * $Id: ztest.c,v 1.52 2002-03-18 21:33:48 adam Exp $
  */
 
 /*
@@ -20,7 +20,9 @@
 #include <yaz/ill.h>
 #endif
 
-Z_GenericRecord *read_grs1(FILE *f, ODR o);
+Z_GenericRecord *dummy_grs_record (int num, ODR o);
+char *dummy_marc_record (int num, ODR odr);
+char *dummy_xml_record (int num, ODR odr);
 
 int ztest_search (void *handle, bend_search_rr *rr);
 int ztest_sort (void *handle, bend_sort_rr *rr);
@@ -41,15 +43,12 @@ int ztest_search (void *handle, bend_search_rr *rr)
         rr->errstring = rr->basenames[0];
         return 0;
     }
-    rr->hits = rand() % 22;
+    rr->hits = rand() % 24;
     return 0;
 }
 
-int ztest_present (void *handle, bend_present_rr *rr)
-{
-    return 0;
-}
 
+/* this huge function handles extended services */
 int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
 {
     /* user-defined handle - created in bend_init */
@@ -74,9 +73,7 @@ int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
     {
        Z_ItemOrder *it = rr->esr->taskSpecificParameters->u.itemOrder;
        yaz_log (LOG_LOG, "Received ItemOrder");
-       switch (it->which)
-       {
-       case Z_IOItemOrder_esRequest:
+        if (it->which == Z_IOItemOrder_esRequest)
        {
            Z_IORequest *ir = it->u.esRequest;
            Z_IOOriginPartToKeep *k = ir->toKeep;
@@ -135,12 +132,6 @@ int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
                                     "Couldn't decode ItemRequest %s near %d",
                                        odr_errmsg(odr_geterror(rr->decode)),
                                        odr_offset(rr->decode));
-#if 0
-                                yaz_log(LOG_LOG, "PDU dump:");
-                                odr_dumpBER(yaz_log_file(),
-                                     r->u.single_ASN1_type->buf,
-                                     r->u.single_ASN1_type->len);
-#endif
                             }
                            else
                                yaz_log(LOG_LOG, "Decode ItemRequest OK");
@@ -187,8 +178,54 @@ int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
                }
            }
 #endif
-       }
-       break;
+            if (k)
+            {
+
+               Z_External *ext = (Z_External *)
+                    odr_malloc (rr->stream, sizeof(*ext));
+               Z_IUOriginPartToKeep *keep = (Z_IUOriginPartToKeep *)
+                    odr_malloc (rr->stream, sizeof(*keep));
+               Z_IOTargetPart *targetPart = (Z_IOTargetPart *)
+                   odr_malloc (rr->stream, sizeof(*targetPart));
+
+               rr->taskPackage = (Z_TaskPackage *)
+                    odr_malloc (rr->stream, sizeof(*rr->taskPackage));
+               rr->taskPackage->packageType =
+                   odr_oiddup (rr->stream, rr->esr->packageType);
+               rr->taskPackage->packageName = 0;
+               rr->taskPackage->userId = 0;
+               rr->taskPackage->retentionTime = 0;
+               rr->taskPackage->permissions = 0;
+               rr->taskPackage->description = 0;
+               rr->taskPackage->targetReference = (Odr_oct *)
+                   odr_malloc (rr->stream, sizeof(Odr_oct));
+               rr->taskPackage->targetReference->buf =
+                   (unsigned char *) odr_strdup (rr->stream, "911");
+               rr->taskPackage->targetReference->len =
+                   rr->taskPackage->targetReference->size =
+                   strlen((char *) (rr->taskPackage->targetReference->buf));
+               rr->taskPackage->creationDateTime = 0;
+               rr->taskPackage->taskStatus = odr_intdup(rr->stream, 0);
+               rr->taskPackage->packageDiagnostics = 0;
+               rr->taskPackage->taskSpecificParameters = ext;
+
+               ext->direct_reference =
+                   odr_oiddup (rr->stream, rr->esr->packageType);
+               ext->indirect_reference = 0;
+               ext->descriptor = 0;
+               ext->which = Z_External_itemOrder;
+               ext->u.itemOrder = (Z_ItemOrder *)
+                   odr_malloc (rr->stream, sizeof(*ext->u.update));
+               ext->u.itemOrder->which = Z_IOItemOrder_taskPackage;
+               ext->u.itemOrder->u.taskPackage =  (Z_IOTaskPackage *)
+                   odr_malloc (rr->stream, sizeof(Z_IOTaskPackage));
+               ext->u.itemOrder->u.taskPackage->originPart = k;
+               ext->u.itemOrder->u.taskPackage->targetPart = targetPart;
+
+                targetPart->itemRequest = 0;
+                targetPart->statusOrErrorReport = 0;
+                targetPart->auxiliaryStatus = 0;
+            }
        }
     }
     else if (rr->esr->taskSpecificParameters->which == Z_External_update)
@@ -244,6 +281,7 @@ int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
                     odr_malloc (rr->stream, sizeof(*keep));
                Z_IUTargetPart *targetPart = (Z_IUTargetPart *)
                    odr_malloc (rr->stream, sizeof(*targetPart));
+
                rr->taskPackage = (Z_TaskPackage *)
                     odr_malloc (rr->stream, sizeof(*rr->taskPackage));
                rr->taskPackage->packageType =
@@ -289,10 +327,29 @@ int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
                targetPart->updateStatus = odr_intdup (rr->stream, 1);
                targetPart->num_globalDiagnostics = 0;
                targetPart->globalDiagnostics = (Z_DiagRec **) odr_nullval();
-               targetPart->num_taskPackageRecords = 0;
-               targetPart->taskPackageRecords =
-                    (Z_IUTaskPackageRecordStructure **) odr_nullval();
-           }
+               targetPart->num_taskPackageRecords = 1;
+               targetPart->taskPackageRecords = 
+                    (Z_IUTaskPackageRecordStructure **)
+                    odr_malloc (rr->stream,
+                                sizeof(Z_IUTaskPackageRecordStructure *));
+               targetPart->taskPackageRecords[0] =
+                    (Z_IUTaskPackageRecordStructure *)
+                    odr_malloc (rr->stream,
+                                sizeof(Z_IUTaskPackageRecordStructure));
+                
+               targetPart->taskPackageRecords[0]->which =
+                    Z_IUTaskPackageRecordStructure_record;
+               targetPart->taskPackageRecords[0]->u.record = 
+                    z_ext_record (rr->stream, VAL_SUTRS, "test", 4);
+               targetPart->taskPackageRecords[0]->correlationInfo = 0; 
+               targetPart->taskPackageRecords[0]->recordStatus =
+                    odr_intdup (rr->stream,
+                                Z_IUTaskPackageRecordStructure_success);  
+               targetPart->taskPackageRecords[0]->num_supplementalDiagnostics
+                    = 0;
+
+               targetPart->taskPackageRecords[0]->supplementalDiagnostics = 0;
+            }
            if (notToKeep)
            {
                int i;
@@ -343,6 +400,7 @@ int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
     return 0;
 }
 
+/* result set delete */
 int ztest_delete (void *handle, bend_delete_rr *rr)
 {
     if (rr->num_setnames == 1 && !strcmp (rr->setnames[0], "1"))
@@ -360,111 +418,31 @@ int ztest_sort (void *handle, bend_sort_rr *rr)
     return 0;
 }
 
-static int atoin (const char *buf, int n)
-{
-    int val = 0;
-    while (--n >= 0)
-    {
-        if (isdigit(*buf))
-            val = val*10 + (*buf - '0');
-        buf++;
-    }
-    return val;
-}
-
-char *marc_read(FILE *inf, ODR odr)
-{
-    char length_str[5];
-    size_t size;
-    char *buf;
-
-    if (fread (length_str, 1, 5, inf) != 5)
-        return NULL;
-    size = atoin (length_str, 5);
-    if (size <= 6)
-        return NULL;
-    if (!(buf = (char*) odr_malloc (odr, size+1)))
-        return NULL;
-    if (fread (buf+5, 1, size-5, inf) != (size-5))
-    {
-        xfree (buf);
-        return NULL;
-    }
-    memcpy (buf, length_str, 5);
-    buf[size] = '\0';
-    return buf;
-}
-
-static char *dummy_database_record (int num, ODR odr)
-{
-    FILE *inf = fopen ("dummy-records", "r");
-    char *buf = 0;
-
-    if (!inf)
-       return NULL;
-    if (num == 98) 
-    {   /* this will generate a very bad MARC record (testing only) */
-        buf = (char*) odr_malloc(odr, 2101);
-        memset(buf, '7', 2100);
-        buf[2100] = '\0';
-    }
-    else
-    {
-        /* OK, try to get proper MARC records from the file */
-        while (--num >= 0)
-        {
-            buf = marc_read (inf, odr);
-            if (!buf)
-                break;
-        }
-    }
-    fclose(inf);
-    return buf;
-}
 
-static Z_GenericRecord *dummy_grs_record (int num, ODR o)
+/* present request handler */
+int ztest_present (void *handle, bend_present_rr *rr)
 {
-    FILE *f = fopen("dummy-grs", "r");
-    char line[512];
-    Z_GenericRecord *r = 0;
-    int n;
-
-    if (!f)
-       return 0;
-    while (fgets(line, 512, f))
-       if (*line == '#' && sscanf(line, "#%d", &n) == 1 && n == num)
-       {
-           r = read_grs1(f, o);
-           break;
-       }
-    fclose(f);
-    return r;
+    return 0;
 }
 
+/* retrieval of a single record (present, and piggy back search) */
 int ztest_fetch(void *handle, bend_fetch_rr *r)
 {
     char *cp;
     r->errstring = 0;
     r->last_in_set = 0;
-    r->basename = "DUMMY";
+    r->basename = "Default";
     r->output_format = r->request_format;  
     if (r->request_format == VAL_SUTRS)
     {
-#if 0
-/* this section returns a huge record (for testing non-blocking write, etc) */
-       r->len = 980000;
-       r->record = odr_malloc (r->stream, r->len);
-       memset (r->record, 'x', r->len);
-#else
-/* this section returns a small record */
+        /* this section returns a small record */
        char buf[100];
-
+        
        sprintf(buf, "This is dummy SUTRS record number %d\n", r->number);
 
        r->len = strlen(buf);
        r->record = (char *) odr_malloc (r->stream, r->len+1);
        strcpy(r->record, buf);
-#endif
     }
     else if (r->request_format == VAL_GRS1)
     {
@@ -476,7 +454,41 @@ int ztest_fetch(void *handle, bend_fetch_rr *r)
            return 0;
        }
     }
-    else if ((cp = dummy_database_record(r->number, r->stream)))
+    else if (r->request_format == VAL_POSTSCRIPT)
+    {
+        char fname[20];
+        FILE *f;
+        long size;
+
+        sprintf (fname, "part.%d.ps", r->number);
+        f = fopen(fname, "rb");
+        if (!f)
+       {
+            r->errcode = 13;
+            return 0;
+        }
+        fseek (f, 0L, SEEK_END);
+        size = ftell (f);
+        if (size <= 0 || size >= 5000000)
+        {
+            r->errcode = 14;
+            return 0;
+        }
+        fseek (f, 0L, SEEK_SET);
+        r->record = (char*) odr_malloc (r->stream, size);
+        r->len = size;
+        r->output_format = VAL_POSTSCRIPT;
+        fread (r->record, size, 1, f);
+        fclose (f);
+    }
+    else if (r->request_format == VAL_TEXT_XML &&
+             (cp = dummy_xml_record (r->number, r->stream)))
+    {
+        r->len = strlen(cp);
+        r->record = cp;
+        r->output_format = VAL_TEXT_XML;
+    }
+    else if ((cp = dummy_marc_record(r->number, r->stream)))
     {
        r->len = strlen(cp);
        r->record = cp;