libbb/xconnect.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Connect to host at port using address resolution from getaddrinfo
   6  *
   7  */
   8
   9 #include "libbb.h"
  10
  11 /* Return network byte ordered port number for a service.
  12  * If "port" is a number use it as the port.
  13  * If "port" is a name it is looked up in /etc/services, if it isnt found return
  14  * default_port
  15  */
  16 unsigned short bb_lookup_port(const char *port, const char *protocol, unsigned short default_port)
  17 {
  18         unsigned short port_nr = htons(default_port);
  19         if (port) {
  20                 char *endptr;
  21                 int old_errno;
  22                 long port_long;
  23
  24                 /* Since this is a lib function, we're not allowed to reset errno to 0.
  25                  * Doing so could break an app that is deferring checking of errno. */
  26                 old_errno = errno;
  27                 errno = 0;
  28                 port_long = strtol(port, &endptr, 10);
  29                 if (errno != 0 || *endptr!='\0' || endptr==port || port_long < 0 || port_long > 65535) {
  30                         struct servent *tserv = getservbyname(port, protocol);
  31                         if (tserv) {
  32                                 port_nr = tserv->s_port;
  33                         }
  34                 } else {
  35                         port_nr = htons(port_long);
  36                 }
  37                 errno = old_errno;
  38         }
  39         return port_nr;
  40 }
  41
  42 void bb_lookup_host(struct sockaddr_in *s_in, const char *host)
  43 {
  44         struct hostent *he;
  45
  46         memset(s_in, 0, sizeof(struct sockaddr_in));
  47         s_in->sin_family = AF_INET;
  48         he = xgethostbyname(host);
  49         memcpy(&(s_in->sin_addr), he->h_addr_list[0], he->h_length);
  50 }
  51
  52 void xconnect(int s, const struct sockaddr *s_addr, socklen_t addrlen)
  53 {
  54         if (connect(s, s_addr, addrlen) < 0) {
  55                 if (ENABLE_FEATURE_CLEAN_UP) close(s);
  56                 if (s_addr->sa_family == AF_INET)
  57                         bb_perror_msg_and_die("%s (%s)",
  58                                 "cannot connect to remote host",
  59                                 inet_ntoa(((struct sockaddr_in *)s_addr)->sin_addr));
  60                 bb_perror_msg_and_die("cannot connect to remote host");
  61         }
  62 }
  63
  64 int xconnect_tcp_v4(struct sockaddr_in *s_addr)
  65 {
  66         int s = xsocket(AF_INET, SOCK_STREAM, 0);
  67         xconnect(s, (struct sockaddr*) s_addr, sizeof(*s_addr));
  68         return s;
  69 }
  70
  71 static const int one = 1;
  72 int setsockopt_reuseaddr(int fd)
  73 {
  74         return setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
  75 }
  76 int setsockopt_broadcast(int fd)
  77 {
  78         return setsockopt(fd, SOL_SOCKET, SO_BROADCAST, &one, sizeof(one));
  79 }
  80
  81 int dotted2sockaddr(const char *dotted, struct sockaddr* sp, int socklen)
  82 {
  83         union {
  84                 struct in_addr a4;
  85 #if ENABLE_FEATURE_IPV6
  86                 struct in6_addr a6;
  87 #endif
  88         } a;
  89
  90         /* TODO maybe: port spec? like n.n.n.n:nn */
  91
  92 #if ENABLE_FEATURE_IPV6
  93         if (socklen >= sizeof(struct sockaddr_in6)
  94          && inet_pton(AF_INET6, dotted, &a.a6) > 0
  95         ) {
  96                 ((struct sockaddr_in6*)sp)->sin6_family = AF_INET6;
  97                 ((struct sockaddr_in6*)sp)->sin6_addr = a.a6;
  98                 /* ((struct sockaddr_in6*)sp)->sin6_port = */
  99                 return 0; /* success */
 100         }
 101 #endif
 102         if (socklen >= sizeof(struct sockaddr_in)
 103          && inet_pton(AF_INET, dotted, &a.a4) > 0
 104         ) {
 105                 ((struct sockaddr_in*)sp)->sin_family = AF_INET;
 106                 ((struct sockaddr_in*)sp)->sin_addr = a.a4;
 107                 /* ((struct sockaddr_in*)sp)->sin_port = */
 108                 return 0; /* success */
 109         }
 110         return 1;
 111 }
 112
 113 int xsocket_stream_ip4or6(sa_family_t *fp)
 114 {
 115         int fd;
 116 #if ENABLE_FEATURE_IPV6
 117         fd = socket(AF_INET6, SOCK_STREAM, 0);
 118         if (fp) *fp = AF_INET6;
 119         if (fd < 0)
 120 #endif
 121         {
 122                 fd = xsocket(AF_INET, SOCK_STREAM, 0);
 123                 if (fp) *fp = AF_INET;
 124         }
 125         return fd;
 126 }
 127
 128 int create_and_bind_socket_ip4or6(const char *hostaddr, int port)
 129 {
 130         int fd;
 131         sockaddr_inet sa;
 132
 133         memset(&sa, 0, sizeof(sa));
 134         if (hostaddr) {
 135                 if (dotted2sockaddr(hostaddr, &sa.sa, sizeof(sa)))
 136                         bb_error_msg_and_die("bad address '%s'", hostaddr);
 137                 /* user specified bind addr dictates family */
 138                 fd = xsocket(sa.sa.sa_family, SOCK_STREAM, 0);
 139         } else
 140                 fd = xsocket_stream_ip4or6(&sa.sa.sa_family);
 141         setsockopt_reuseaddr(fd);
 142
 143         /* if (port >= 0) { */
 144 #if ENABLE_FEATURE_IPV6
 145                 if (sa.sa.sa_family == AF_INET6 /* && !sa.sin6.sin6_port */)
 146                         sa.sin6.sin6_port = htons(port);
 147 #endif
 148                 if (sa.sa.sa_family == AF_INET /* && !sa.sin.sin_port */)
 149                         sa.sin.sin_port = htons(port);
 150         /* } */
 151
 152         xbind(fd, &sa.sa, sizeof(sa));
 153         return fd;
 154 }