Add connection timeout

examples/http-client/main.c

@@ -13,11 +13,20 @@
 
 // The very first web page in history. You can replace it from command line
 static const char *s_url = "http://info.cern.ch/";
-static const char *s_post = NULL;  // POST data
+static const char *s_post_data = NULL;     // POST data
+static const int64_t s_timeout_ms = 1500;  // Connect timeout in milliseconds
 
 // Print HTTP response and signal that we're done
 static void fn(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {
-  if (ev == MG_EV_CONNECT) {
+  if (ev == MG_EV_OPEN) {
+    // Connection created. Store connect expiration time in c->label
+    *(int64_t *) c->label = mg_millis() + s_timeout_ms;
+  } else if (ev == MG_EV_POLL) {
+    if (mg_millis() > *(int64_t *) c->label &&
+        (c->is_connecting || c->is_resolving)) {
+      mg_error(c, "Connect timeout");
+    }
+  } else if (ev == MG_EV_CONNECT) {
     // Connected to server. Extract host name from URL
     struct mg_str host = mg_url_host(s_url);
 
@@ -28,16 +37,16 @@ static void fn(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {
     }
 
     // Send request
-    int content_length = s_post ? strlen(s_post) : 0;
+    int content_length = s_post_data ? strlen(s_post_data) : 0;
     mg_printf(c,
               "%s %s HTTP/1.0\r\n"
               "Host: %.*s\r\n"
               "Content-Type: octet-stream\r\n"
               "Content-Length: %d\r\n"
               "\r\n",
-              s_post ? "POST" : "GET", mg_url_uri(s_url), (int) host.len,
+              s_post_data ? "POST" : "GET", mg_url_uri(s_url), (int) host.len,
               host.ptr, content_length);
-    mg_send(c, s_post, content_length);
+    mg_send(c, s_post_data, content_length);
   } else if (ev == MG_EV_HTTP_MSG) {
     // Response is received. Print it
     struct mg_http_message *hm = (struct mg_http_message *) ev_data;
@@ -50,24 +59,13 @@ static void fn(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {
 }
 
 int main(int argc, char *argv[]) {
-  struct mg_mgr mgr;  // Event manager
-  bool done = false;  // Event handler flips it to true
-  int i;
-
-  // Parse command-line flags
-  for (i = 1; i < argc; i++) {
-    if (strcmp(argv[i], "-d") == 0) {
-      s_post = argv[++i];
-    } else {
-      break;
-    }
-  }
-  if (i < argc) s_url = argv[i];
-
-  mg_log_set("3");                          // Set to 0 to disable debug
-  mg_mgr_init(&mgr);                        // Initialise event manager
+  struct mg_mgr mgr;              // Event manager
+  bool done = false;              // Event handler flips it to true
+  if (argc > 1) s_url = argv[1];  // Use URL provided in the command line
+  mg_log_set("3");                // Set to 0 to disable debug
+  mg_mgr_init(&mgr);              // Initialise event manager
   mg_http_connect(&mgr, s_url, fn, &done);  // Create client connection
-  while (!done) mg_mgr_poll(&mgr, 1000);    // Infinite event loop
+  while (!done) mg_mgr_poll(&mgr, 50);      // Infinite event loop
   mg_mgr_free(&mgr);                        // Free resources
   return 0;
 }