tcp: add IP_PORT_SET() macro

tvheadend · Mar 11, 2015 · 53984a3 · 53984a3
1 parent af698f0
commit 53984a3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/tcp.c b/src/tcp.c
@@ -771,10 +771,7 @@ tcp_default_ip_addr ( struct sockaddr_storage *deflt )
                 IP_IN_ADDR(ss)) <= 0)
     return -1;
 
-  if (ss.ss_family == AF_INET)
-    IP_AS_V4(ss, port) = htons(53);
-  else
-    IP_AS_V6(ss, port) = htons(53);
+  IP_PORT_SET(ss, htons(53));
 
   sock = tvh_socket(ss.ss_family, SOCK_STREAM, 0);
   if (sock < 0)

diff --git a/src/tcp.h b/src/tcp.h
@@ -36,6 +36,10 @@
   ((storage).ss_family == AF_INET6 ? \
       ((struct sockaddr_in6 *)&(storage))->sin6_port : \
       ((struct sockaddr_in  *)&(storage))->sin_port)
+#define IP_PORT_SET(storage, port) \
+  if ((storage).ss_family == AF_INET6) \
+      ((struct sockaddr_in6 *)&(storage))->sin6_port = (port); else \
+      ((struct sockaddr_in  *)&(storage))->sin_port  = (port);
 
 typedef struct tcp_server_ops
 {