Changeset 6629 for pymolproxy

Timestamp:

Nov 14, 2016 6:28:38 PM (8 years ago)

Author:

ldelgass

Message:

Add idle timeout option to pymolproxy

File:

• pymolproxy/trunk/pymolproxy.c (modified) (8 diffs)

pymolproxy/trunk/pymolproxy.c

@@ -104 +104 @@
 static int pymolIsAlive = TRUE;
 static int statsFile = -1;
+static long idleTimeout = -1L;
 
 #define READ_DEBUG  1
@@ -2266 +2267 @@
     PymolProxy *p = clientData;
     Tcl_DString command;
+    fd_set readFds;
+    int ret = 1;
     struct timeval tv, *tvPtr;
+    int polling = 0;
 
 #if READ_DEBUG
@@ -2273 +2277 @@
     Tcl_DStringInit(&command);
     while (pymolIsAlive) {
-        tvPtr = NULL;
+        FD_ZERO(&readFds);
+        FD_SET(p->client.fd, &readFds);
+
+        if (idleTimeout >= 0L) {
+            tv.tv_sec = idleTimeout;
+            tv.tv_usec = 0L;
+            polling = (tv.tv_sec == 0 && tv.tv_usec == 0) ? 1 : 0;
+            tvPtr = &tv;
+        } else {
+            tvPtr = NULL;
+        }
 #if READ_DEBUG
         DEBUG("Start I/O set");
 #endif
-        while ((pymolIsAlive) && (WaitForNextLine(&p->client, tvPtr))) {
+        while (pymolIsAlive &&
+               (IsLineAvailable(&p->client) ||
+                (ret = select(p->client.fd+1, &readFds, NULL, NULL, tvPtr)) > 0)) {
             size_t numBytes;
             const char *line;
@@ -2302 +2318 @@
                     DEBUG("TCL_BREAK found");
 #endif
+                    ret = 2;
                     break;              /* This was caused by a "imgflush"
                                          * command. Break out of the read
                                          * loop and allow a new image to be
                                          * rendered. */
+                } else if (result != TCL_OK) {
+                    ret = 0;
+                } else {
+                    ret = 3;
                 }
                 if (p->flags & FORCE_UPDATE) {
@@ -2314 +2335 @@
                 }
             }
+            polling = 1;
             tv.tv_sec = 0L;
             tv.tv_usec = 0L;            /* On successive reads, we break
                                          * out if no data is available. */
+            FD_SET(p->client.fd, &readFds);
             tvPtr = &tv;
         }
@@ -2322 +2345 @@
         DEBUG("Finish I/O set");
 #endif
+        if (ret < 0) {
+            DEBUG("Error in select(): %s", strerror(errno));
+            goto done;
+        }
+        if (!polling && ret == 0 && idleTimeout > 0L) {
+            // If idle timeout expired, disconnect
+            DEBUG("Exiting server after timeout waiting for client command");
+            goto done;
+        }
         /* Handle all the pending setting changes now. */
         UpdateSettings(p);
@@ -2434 +2466 @@
     size_t numBytes;
 
+    while (1) {
+        int c = getopt(argc, argv, "t:");
+        if (c == -1) {
+            break;
+        }
+        switch (c) {
+        case 't':
+            idleTimeout = atol(optarg);
+            break;
+        case '?':
+            break;
+        default:
+            return 1;
+        }
+    }
+
     frecord = NULL;
     if (recording) {
@@ -2452 +2500 @@
     DEBUG("Starting pymolproxy");
 
-    InitProxy(&proxy, argv + 1);
+    InitProxy(&proxy, argv + optind);
     if (pthread_create(&thread1, NULL, &ClientToServer, &proxy) < 0) {
         ERROR("Can't create reader thread: %s", strerror(errno));