Fix proxy loop

src/lib/proxy.c

@@ -352,6 +352,7 @@ static int proxy_connect(proxy_ctx_t *proxy, host_config_t *conf, http_res *res,
     proxy->initialized = 1;
     proxy->cnx_s = clock_micros();
     proxy->host = conf->name;
+    proxy->http_timeout = 0;
 
     info(BLUE_STR "Established new connection with " BLD_STR "[%s]:%i", addr_buf, conf->proxy.port);
 
@@ -361,26 +362,12 @@ static int proxy_connect(proxy_ctx_t *proxy, host_config_t *conf, http_res *res,
 int proxy_init(proxy_ctx_t **proxy_ptr, http_req *req, http_res *res, http_status_ctx *ctx, host_config_t *conf, sock *client, http_status *custom_status, char *err_msg) {
     char buffer[CHUNK_SIZE], err_buf[256];
     long ret;
-    int tries = 0, retry = 1;
+    int tries = 0, retry = 1, srv_error = 0;
 
     *proxy_ptr = proxy_get_by_conf(conf);
     proxy_ctx_t *proxy = *proxy_ptr;
     proxy->client = NULL;
 
-    while (retry) {
-        errno = 0;
-
-        if (!proxy->initialized || sock_has_pending(&proxy->proxy) != 0) {
-            if (proxy->initialized)
-                proxy_close(proxy);
-
-            retry = 0;
-            tries++;
-
-            if (proxy_connect(proxy, conf, res, ctx, err_msg) != 0)
-                continue;
-        }
-
     const char *connection = http_get_header_field(&req->hdr, "Connection");
     if (strcontains(connection, "upgrade") || strcontains(connection, "Upgrade")) {
         const char *upgrade = http_get_header_field(&req->hdr, "Upgrade");
@@ -404,6 +391,23 @@ int proxy_init(proxy_ctx_t **proxy_ptr, http_req *req, http_res *res, http_statu
         return -1;
     }
 
+    while (retry) {
+        errno = 0;
+
+        if (!proxy->initialized || sock_has_pending(&proxy->proxy) != 0 || srv_error ||
+           (proxy->http_timeout != 0 && (clock_micros() - proxy->proxy.ts_last) > proxy->http_timeout))
+        {
+            if (proxy->initialized)
+                proxy_close(proxy);
+
+            retry = 0;
+            srv_error = 0;
+            tries++;
+
+            if (proxy_connect(proxy, conf, res, ctx, err_msg) != 0)
+                continue;
+        }
+
         ret = http_send_request(&proxy->proxy, req);
         if (ret < 0) {
             res->status = http_get_status(502);
@@ -411,6 +415,7 @@ int proxy_init(proxy_ctx_t **proxy_ptr, http_req *req, http_res *res, http_statu
             error("Unable to send request to server (1)");
             sprintf(err_msg, "Unable to send request to server: %s.", error_str(errno, err_buf, sizeof(err_buf)));
             retry = tries < 4;
+            srv_error = 1;
             continue;
         }
 
@@ -431,8 +436,7 @@ int proxy_init(proxy_ctx_t **proxy_ptr, http_req *req, http_res *res, http_statu
                 ctx->origin = SERVER_REQ;
                 error("Unable to send request to server (2)");
                 sprintf(err_msg, "Unable to send request to server: %s.", error_str(errno, err_buf, sizeof(err_buf)));
-                retry = tries < 4;
+                return -1;
-                continue;
             } else if (ret == -1) {
                 res->status = http_get_status(400);
                 ctx->origin = CLIENT_REQ;
@@ -458,8 +462,7 @@ int proxy_init(proxy_ctx_t **proxy_ptr, http_req *req, http_res *res, http_statu
             }
             error("Unable to receive response from server");
             sprintf(err_msg, "Unable to receive response from server: %s.", error_str(errno, err_buf, sizeof(err_buf)));
-            retry = tries < 4;
+            return -1;
-            continue;
         }
         buffer[ret] = 0;
 
@@ -533,6 +536,8 @@ int proxy_init(proxy_ctx_t **proxy_ptr, http_req *req, http_res *res, http_statu
         }
         sock_recv_x(&proxy->proxy, buffer, header_len, 0);
 
+        // TODO read timeout from Keep-Alive
+
         ret = proxy_response_header(req, res, conf);
         if (ret != 0) {
             res->status = http_get_status(500);

src/lib/proxy.h

@@ -22,6 +22,7 @@ typedef struct {
     unsigned char initialized:1, in_use:1;
     sock proxy;
     long cnx_s, cnx_e;
+    long http_timeout;
     char *host;
     void *client;
 } proxy_ctx_t;

src/lib/sock.c

@@ -87,6 +87,9 @@ int sock_init(sock *s, int fd, int flags) {
     s->ts_start = clock_micros();
     s->ts_last = s->ts_start;
     s->timeout_us = -1;
+    s->ssl = NULL;
+    s->addr = NULL;
+    s->s_addr = NULL;
 
     return 0;
 }
@@ -200,7 +203,7 @@ int sock_set_timeout(sock *s, double sec) {
 }
 
 long sock_send(sock *s, void *buf, unsigned long len, int flags) {
-    if (s->socket == 0) {
+    if (s->socket < 0) {
         errno = ENOTCONN;
         return -1;
     }
@@ -242,7 +245,7 @@ long sock_send_x(sock *s, void *buf, unsigned long len, int flags) {
 }
 
 long sock_recv(sock *s, void *buf, unsigned long len, int flags) {
-    if (s->socket == 0) {
+    if (s->socket < 0) {
         errno = ENOTCONN;
         return -1;
     }
@@ -377,7 +380,7 @@ int sock_close(sock *s) {
         s->ssl = NULL;
     }
     close(s->socket);
-    s->socket = 0;
+    s->socket = -1;
     s->enc = 0, s->pipe = 0;
     errno = e;
     return 0;