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("unable to connect to remote host (%s)",
  58                                 inet_ntoa(((struct sockaddr_in *)s_addr)->sin_addr));
  59                 bb_perror_msg_and_die("unable to connect to remote host");
  60         }
  61 }
  62
  63 int xconnect_tcp_v4(struct sockaddr_in *s_addr)
  64 {
  65         int s = xsocket(AF_INET, SOCK_STREAM, 0);
  66         xconnect(s, (struct sockaddr*) s_addr, sizeof(*s_addr));
  67         return s;
  68 }