net.c

   1 HEADER #include <netdb.h>
   2
   3 #include <sys/types.h>
   4 #include <sys/socket.h>
   5 #include <netinet/in.h>
   6 #include <netinet/tcp.h>
   7 #include <unistd.h>
   8 #include <errno.h>
   9
  10 HEADER extern const struct addrinfo libjh_tcp_hints;
  11 HEADER #define JH_TCP_HINTS (&libjh_tcp_hints)
  12 const struct addrinfo libjh_tcp_hints = {
  13   .ai_flags = AI_ADDRCONFIG,
  14   .ai_family = AF_UNSPEC,
  15   .ai_socktype = SOCK_STREAM,
  16   .ai_protocol = 0
  17 };
  18
  19 // negative return value: interpret as a getaddrinfo() error (non-gai errors become EAI_SYSTEM)
  20 // >=0 return value: interpret as resulting fd
  21 PUBLIC_FN int netopen(const char *node, const char *service, const struct addrinfo *hints) {
  22   struct addrinfo *addrs;
  23   int gai_res = getaddrinfo(node, service, hints, &addrs);
  24   if (gai_res) return gai_res;
  25
  26   int s = socket(addrs[0].ai_family, addrs[0].ai_socktype, addrs[0].ai_protocol);
  27   if (s == -1) goto err_socket;
  28
  29   if (connect(s, addrs[0].ai_addr, addrs[0].ai_addrlen)) goto err_connect;
  30
  31   freeaddrinfo(addrs);
  32   return s;
  33
  34 err_connect:;
  35   int errno_ = errno;
  36   close(s);
  37   errno = errno_;
  38 err_socket:
  39   freeaddrinfo(addrs);
  40   return EAI_SYSTEM;
  41 }
  42
  43 PUBLIC_FN int fnetopen(FILE **in, FILE **out, const char *node, const char *service, const struct addrinfo *hints) {
  44   int fd = netopen(node, service, hints);
  45   if (fd < 0) return EAI_SYSTEM;
  46   int ret = fopen_bistream(in, out, fd, 0);
  47   if (ret) close(fd);
  48   return ret;
  49 }