Added code to receive close
[yaz-moved-to-github.git] / server / seshigh.c
index a9dfd99..a062b3d 100644 (file)
@@ -4,7 +4,13 @@
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: seshigh.c,v $
- * Revision 1.44  1995-08-17 12:45:25  quinn
+ * Revision 1.46  1995-08-29 11:17:58  quinn
+ * Added code to receive close
+ *
+ * Revision 1.45  1995/08/21  09:11:00  quinn
+ * Smallish fixes to suppport new formats.
+ *
+ * Revision 1.44  1995/08/17  12:45:25  quinn
  * Fixed minor problems with GRS-1. Added support in c&s.
  *
  * Revision 1.43  1995/08/15  12:00:31  quinn
 #include <stdio.h>
 #include <unistd.h>
 #include <assert.h>
+#include <sys/time.h>
 
 #include <dmalloc.h>
 #include <comstack.h>
@@ -263,7 +270,7 @@ association *create_association(IOCHAN channel, COMSTACK link)
     new->input_buffer = 0;
     new->input_buffer_len = 0;
     new->backend = 0;
-    new->rejected = 0;
+    new->state = ASSOC_NEW;
     request_initq(&new->incoming);
     request_initq(&new->outgoing);
     new->proto = cs_getproto(link);
@@ -288,6 +295,33 @@ void destroy_association(association *h)
     free(h);
 }
 
+static void do_close(association *a, int reason, char *message)
+{
+    Z_APDU apdu;
+    Z_Close *cls = zget_Close(a->encode);
+    request *req = request_get();
+
+    /* Purge request queue */
+    while (request_deq(&a->incoming));
+    while (request_deq(&a->outgoing));
+    if (a->version >= 3)
+    {
+       logf(LOG_DEBUG, "Generating Close PDU");
+       apdu.which = Z_APDU_close;
+       apdu.u.close = cls;
+       *cls->closeReason = reason;
+       cls->diagnosticInformation = message;
+       process_response(a, req, &apdu);
+       iochan_settimeout(a->client_chan, 60);
+    }
+    else
+    {
+       logf(LOG_DEBUG, "v2 client. No Close PDU");
+       iochan_setevent(a->client_chan, EVENT_TIMEOUT); /* force imm close */
+    }
+    a->state = ASSOC_DEAD;
+}
+
 /*
  * This is where PDUs from the client are read and the further
  * processing is initiated. Flow of control moves down through the
@@ -307,10 +341,18 @@ void ir_session(IOCHAN h, int event)
     assert(h && conn && assoc);
     if (event == EVENT_TIMEOUT)
     {
-       logf(LOG_LOG, "Timeout - closing connection.");
-       cs_close(conn);
-       destroy_association(assoc);
-       iochan_destroy(h);
+       if (assoc->state != ASSOC_UP)
+       {
+           logf(LOG_LOG, "Final timeout - closing connection.");
+           cs_close(conn);
+           destroy_association(assoc);
+           iochan_destroy(h);
+       }
+       else
+       {
+           logf(LOG_LOG, "Session idle too long. Sending close.");
+           do_close(assoc, Z_Close_lackOfActivity, 0);
+       }
        return;
     }
     if (event & EVENT_INPUT || event & EVENT_WORK) /* input */
@@ -320,7 +362,7 @@ void ir_session(IOCHAN h, int event)
            logf(LOG_DEBUG, "ir_session (input)");
            assert(assoc && conn);
            /* We aren't speaking to this fellow */
-           if (assoc->rejected)
+           if (assoc->state == ASSOC_DEAD)
            {
                logf(LOG_LOG, "Closed connection after reject");
                cs_close(conn);
@@ -353,9 +395,7 @@ void ir_session(IOCHAN h, int event)
                    odr_errlist[odr_geterror(assoc->decode)]);
                logf(LOG_LOG, "PDU dump:");
                odr_dumpBER(log_file(), assoc->input_buffer, res);
-               cs_close(conn);
-               destroy_association(assoc);
-               iochan_destroy(h);
+               do_close(assoc, Z_Close_protocolError, "Malformed package");
                return;
            }
            req->request_mem = odr_extract_mem(assoc->decode);
@@ -372,11 +412,7 @@ void ir_session(IOCHAN h, int event)
        req = request_head(&assoc->incoming);
        if (req->state == REQUEST_IDLE)
            if (process_request(assoc) < 0)
-           {
-               cs_close(conn);
-               destroy_association(assoc);
-               iochan_destroy(h);
-           }
+               do_close(assoc, Z_Close_systemProblem, "Unknown error");
     }
     if (event & EVENT_OUTPUT)
     {
@@ -492,9 +528,7 @@ void backend_response(IOCHAN i, int event)
     if ((res && process_response(assoc, req, res) < 0) || fd < 0)
     {
        logf(LOG_LOG, "Fatal error when talking to backend");
-       cs_close(assoc->client_link);
-       destroy_association(assoc);
-       iochan_destroy(assoc->client_chan);
+       do_close(assoc, Z_Close_systemProblem, 0);
        iochan_destroy(i);
        return;
     }
@@ -528,9 +562,11 @@ static int process_response(association *assoc, request *req, Z_APDU *res)
        odr_reset(assoc->print);
     }
     /* change this when we make the backend reentrant */
-    assert(req == request_head(&assoc->incoming));
-    req->state = REQUEST_IDLE;
-    request_deq(&assoc->incoming);
+    if (req == request_head(&assoc->incoming))
+    {
+       req->state = REQUEST_IDLE;
+       request_deq(&assoc->incoming);
+    }
     request_enq(&assoc->outgoing, req);
     /* turn the work over to the ir_session handler */
     iochan_setflag(assoc->client_chan, EVENT_OUTPUT);
@@ -554,6 +590,7 @@ static Z_APDU *process_initRequest(association *assoc, request *reqb)
     Z_InitResponse *resp = apdu->u.initResponse;
     bend_initrequest binitreq;
     bend_initresult *binitres;
+    char options[100];
 
     logf(LOG_LOG, "Got initRequest");
     if (req->implementationId)
@@ -573,26 +610,57 @@ static Z_APDU *process_initRequest(association *assoc, request *reqb)
 
     assoc->backend = binitres->handle;
     resp->referenceId = req->referenceId;
+    *options = '\0';
     /* let's tell the client what we can do */
     if (ODR_MASK_GET(req->options, Z_Options_search))
+    {
        ODR_MASK_SET(resp->options, Z_Options_search);
+       strcat(options, "srch");
+    }
     if (ODR_MASK_GET(req->options, Z_Options_present))
+    {
        ODR_MASK_SET(resp->options, Z_Options_present);
+       strcat(options, " prst");
+    }
 #if 0
     if (ODR_MASK_GET(req->options, Z_Options_delSet))
+    {
        ODR_MASK_SET(&options, Z_Options_delSet);
+       strcat(options, " del");
+    }
 #endif
     if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
+    {
        ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
+       strcat(options, " namedresults");
+    }
     if (ODR_MASK_GET(req->options, Z_Options_scan))
+    {
        ODR_MASK_SET(resp->options, Z_Options_scan);
+       strcat(options, " scan");
+    }
     if (ODR_MASK_GET(req->options, Z_Options_concurrentOperations))
+    {
        ODR_MASK_SET(resp->options, Z_Options_concurrentOperations);
+       strcat(options, " concurop");
+    }
 
     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_1))
+    {
        ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_1);
+       assoc->version = 2; /* 1 & 2 are equivalent */
+    }
     if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_2))
+    {
        ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_2);
+       assoc->version = 2;
+    }
+    if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_3))
+    {
+       ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_3);
+       assoc->version = 3;
+    }
+    logf(LOG_LOG, "Negotiated to v%d: %s", assoc->version, options);
     assoc->maximumRecordSize = *req->maximumRecordSize;
     if (assoc->maximumRecordSize > control_block->maxrecordsize)
        assoc->maximumRecordSize = control_block->maxrecordsize;
@@ -606,8 +674,10 @@ static Z_APDU *process_initRequest(association *assoc, request *reqb)
     {
        logf(LOG_LOG, "Connection rejected by backend.");
        *resp->result = 0;
-       assoc->rejected = 1;
+       assoc->state = ASSOC_DEAD;
     }
+    else
+       assoc->state = ASSOC_UP;
     return apdu;
 }
 
@@ -756,7 +826,14 @@ static Z_Records *pack_records(association *a, char *setname, int start,
        bend_fetchresult *fres;
        Z_NamePlusRecord *thisrec;
        Z_DatabaseRecord *thisext;
-
+       int this_length;
+
+       /*
+        * we get the number of bytes allocated on the stream before any
+        * allocation done by the backend - this should give us a reasonable
+        * idea of the total size of the data so far.
+        */
+       total_length = odr_total(a->encode);
        if (reclist.num_records == MAX_RECORDS - 1)
        {
            *pres = Z_PRES_PARTIAL_2;
@@ -778,19 +855,23 @@ static Z_Records *pack_records(association *a, char *setname, int start,
            *pres = Z_PRES_FAILURE;
            return diagrec(a->proto, fres->errcode, fres->errstring);
        }
+       if (fres->len >= 0)
+           this_length = fres->len;
+       else
+           this_length = odr_total(a->encode) - total_length;
        logf(LOG_DEBUG, "  fetched record, len=%d, total=%d",
-           fres->len, total_length);
-       if (fres->len + total_length > a->preferredMessageSize)
+           this_length, total_length);
+       if (this_length + total_length > a->preferredMessageSize)
        {
            /* record is small enough, really */
-           if (fres->len <= a->preferredMessageSize)
+           if (this_length <= a->preferredMessageSize)
            {
                logf(LOG_DEBUG, "  Dropped last normal-sized record");
                *pres = Z_PRES_PARTIAL_2;
                break;
            }
            /* record can only be fetched by itself */
-           if (fres->len < a->maximumRecordSize)
+           if (this_length < a->maximumRecordSize)
            {
                logf(LOG_DEBUG, "  Record > prefmsgsz");
                if (toget > 1)
@@ -799,7 +880,6 @@ static Z_Records *pack_records(association *a, char *setname, int start,
                    reclist.records[reclist.num_records] =
                         surrogatediagrec(a->proto, fres->basename, 16, 0);
                    reclist.num_records++;
-                   total_length += 10; /* totally arbitrary */
                    continue;
                }
            }
@@ -809,7 +889,6 @@ static Z_Records *pack_records(association *a, char *setname, int start,
                reclist.records[reclist.num_records] =
                    surrogatediagrec(a->proto, fres->basename, 17, 0);
                reclist.num_records++;
-               total_length += 10; /* totally arbitrary */
                continue;
            }
        }
@@ -877,7 +956,6 @@ static Z_Records *pack_records(association *a, char *setname, int start,
        }
        reclist.records[reclist.num_records] = thisrec;
        reclist.num_records++;
-       total_length += fres->len;
        *next = fres->last_in_set ? 0 : recno + 1;
     }
     *num = reclist.num_records;