Add support to terminate monitor thread

Create a local pipe to track when socket connection times out
and send a terminate on the monitor thread

Bug: 5337272
Change-Id: I80ed3f8450b4d54d1b9e2bbc3cc84f10d0ae7a04

1 files changed, 46 insertions, 2 deletions

diff --git a/wifi/wifi.c b/wifi/wifi.c
index 332133c..903065c 100644
--- a/wifi/wifi.c
+++ b/wifi/wifi.c
@@ -36,6 +36,8 @@
 
 static struct wpa_ctrl *ctrl_conn;
 static struct wpa_ctrl *monitor_conn;
+/* pipe used to exit from a blocking read */
+static int exit_pipe[2] = {-1, -1};
 
 extern int do_dhcp();
 extern int ifc_init();
@@ -610,6 +612,14 @@ int wifi_connect_to_supplicant()
         ctrl_conn = monitor_conn = NULL;
         return -1;
     }
+
+    if (pipe(exit_pipe) == -1) {
+        wpa_ctrl_close(monitor_conn);
+        wpa_ctrl_close(ctrl_conn);
+        ctrl_conn = monitor_conn = NULL;
+        return -1;
+    }
+
     return 0;
 }
 
@@ -624,6 +634,8 @@ int wifi_send_command(struct wpa_ctrl *ctrl, const char *cmd, char *reply, size_
     ret = wpa_ctrl_request(ctrl, cmd, strlen(cmd), reply, reply_len, NULL);
     if (ret == -2) {
         LOGD("'%s' command timed out.\n", cmd);
+        /* unblocks the monitor receive socket for termination */
+        write(exit_pipe[1], "T", 1);
         return -2;
     } else if (ret < 0 || strncmp(reply, "FAIL", 4) == 0) {
         return -1;
@@ -634,6 +646,28 @@ int wifi_send_command(struct wpa_ctrl *ctrl, const char *cmd, char *reply, size_
     return 0;
 }
 
+int wifi_ctrl_recv(struct wpa_ctrl *ctrl, char *reply, size_t *reply_len)
+{
+    int res;
+    int ctrlfd = wpa_ctrl_get_fd(ctrl);
+    int high = (ctrlfd > exit_pipe[0]) ? ctrlfd : exit_pipe[0];
+    fd_set rfds;
+
+    FD_ZERO(&rfds);
+    FD_SET(ctrlfd, &rfds);
+    FD_SET(exit_pipe[0], &rfds);
+    res = select(high + 1, &rfds, NULL, NULL, NULL);
+    if (res < 0)
+        return res;
+    if (FD_ISSET(ctrlfd, &rfds)) {
+        return wpa_ctrl_recv(ctrl, reply, reply_len);
+    } else {
+        LOGD("Received on exit pipe, terminate");
+        return -1;
+    }
+    return 0;
+}
+
 int wifi_wait_for_event(char *buf, size_t buflen)
 {
     size_t nread = buflen - 1;
@@ -650,9 +684,9 @@ int wifi_wait_for_event(char *buf, size_t buflen)
         return strlen(buf);
     }
 
-    result = wpa_ctrl_recv(monitor_conn, buf, &nread);
+    result = wifi_ctrl_recv(monitor_conn, buf, &nread);
     if (result < 0) {
-        LOGD("wpa_ctrl_recv failed: %s\n", strerror(errno));
+        LOGD("wifi_ctrl_recv failed: %s\n", strerror(errno));
         strncpy(buf, WPA_EVENT_TERMINATING " - recv error", buflen-1);
         buf[buflen-1] = '\0';
         return strlen(buf);
@@ -700,6 +734,16 @@ void wifi_close_supplicant_connection()
         monitor_conn = NULL;
     }
 
+    if (exit_pipe[0] >= 0) {
+        close(exit_pipe[0]);
+        exit_pipe[0] = -1;
+    }
+
+    if (exit_pipe[1] >= 0) {
+        close(exit_pipe[1]);
+        exit_pipe[1] = -1;
+    }
+
     while (count-- > 0) {
         if (property_get(SUPP_PROP_NAME, supp_status, NULL)) {
             if (strcmp(supp_status, "stopped") == 0)