static int usock_inet(int type, const char *host, const char *service, int socktype, bool server)
{
struct addrinfo *result, *rp;
struct addrinfo hints = {
.ai_family = (type & USOCK_IPV6ONLY) ? AF_INET6 :
(type & USOCK_IPV4ONLY) ? AF_INET : AF_UNSPEC,
.ai_socktype = socktype,
.ai_flags = AI_ADDRCONFIG
| ((type & USOCK_SERVER) ? AI_PASSIVE : 0)
| ((type & USOCK_NUMERIC) ? AI_NUMERICHOST : 0),
};
int sock = -1;
if (getaddrinfo(host, service, &hints, &result))
return -1;
for (rp = result; rp != NULL; rp = rp->ai_next) {
sock = usock_connect(rp->ai_addr, rp->ai_addrlen, rp->ai_family, socktype, server);
if (sock >= 0)
break;
}
freeaddrinfo(result);
return sock;
}
const char *usock_port(int port)
{
static char buffer[sizeof("65535\0")];
if (port < 0 || port > 65535)
return NULL;
snprintf(buffer, sizeof(buffer), "%u", port);
return buffer;
}
int usock(int type, const char *host, const char *service) {
int socktype = ((type & 0xff) == USOCK_TCP) ? SOCK_STREAM : SOCK_DGRAM;
bool server = !!(type & USOCK_SERVER);
int sock;
if (type & USOCK_UNIX)
sock = usock_unix(host, socktype, server);
else
sock = usock_inet(type, host, service, socktype, server);
if (sock < 0)
return -1;
usock_set_flags(sock, type);
return sock;
}
static int usock_unix(const char *host, int socktype, bool server)
{
struct sockaddr_un sun = {.sun_family = AF_UNIX};
if (strlen(host) >= sizeof(sun.sun_path)) {
errno = EINVAL;
return -1;
}
strcpy(sun.sun_path, host);
return usock_connect((struct sockaddr*)&sun, sizeof(sun), AF_UNIX, socktype, server);
}
