socket-util: Make TCP open function support no default port.
authorBen Pfaff <blp@nicira.com>
Thu, 19 Nov 2009 21:44:49 +0000 (13:44 -0800)
committerBen Pfaff <blp@nicira.com>
Mon, 23 Nov 2009 20:25:13 +0000 (12:25 -0800)
Until now, tcp_open_active() and tcp_open_passive() have only been used
in situations where there is a reasonable default port, e.g. OFP_TCP_PORT.
But for NetFlow there is no universal default, so enhance these functions
so that they can require the user to specify a port explicitly.

Crossported from the 'db' branch, where this is useful for JSON-RPC, which
also has no widely known port.

lib/socket-util.c

index 3fcd5a1e8e79f73210c0d2ec70feeb33b370a1dd..e400bb543abc1cdbf34919aa84e9d111083c0849 100644 (file)
@@ -292,8 +292,9 @@ guess_netmask(uint32_t ip)
 }
 
 /* Opens a non-blocking TCP socket and connects to 'target', which should be a
- * string in the format "<host>[:<port>]", where <host> is required and <port>
- * is optional, with 'default_port' assumed if <port> is omitted.
+ * string in the format "<host>[:<port>]".  <host> is required.  If
+ * 'default_port' is nonzero then <port> is optional and defaults to
+ * 'default_port'.
  *
  * On success, returns 0 (indicating connection complete) or EAGAIN (indicating
  * connection in progress), in which case the new file descriptor is stored
@@ -335,6 +336,10 @@ tcp_open_active(const char *target_, uint16_t default_port,
     }
     if (port_string && atoi(port_string)) {
         sin.sin_port = htons(atoi(port_string));
+    } else if (!default_port) {
+        VLOG_ERR("%s: port number must be specified", target_);
+        error = EAFNOSUPPORT;
+        goto exit;
     }
 
     /* Create non-blocking socket. */
@@ -376,10 +381,10 @@ exit:
 }
 
 /* Opens a non-blocking TCP socket, binds to 'target', and listens for incoming
- * connections.  'target' should be a string in the format "[<port>][:<ip>]",
- * where both <port> and <ip> are optional.  If <port> is omitted, it defaults
- * to 'default_port'; if <ip> is omitted it defaults to the wildcard IP
- * address.
+ * connections.  'target' should be a string in the format "[<port>][:<ip>]".
+ * <port> may be omitted if 'default_port' is nonzero, in which case it
+ * defaults to 'default_port'.  If <ip> is omitted it defaults to the wildcard
+ * IP address.
  *
  * The socket will have SO_REUSEADDR turned on.
  *
@@ -406,6 +411,10 @@ tcp_open_passive(const char *target_, uint16_t default_port)
     port_string = strsep(&string_ptr, ":");
     if (port_string && atoi(port_string)) {
         sin.sin_port = htons(atoi(port_string));
+    } else if (!default_port) {
+        VLOG_ERR("%s: port number must be specified", target_);
+        error = EAFNOSUPPORT;
+        goto exit;
     }
 
     /* Parse optional bind IP. */