107 lines
2.7 KiB
C
107 lines
2.7 KiB
C
/* $OpenBSD: util.c,v 1.3 2019/07/03 03:24:03 deraadt Exp $ */
|
|
|
|
/*
|
|
* Copyright (c) 2013 Internet Initiative Japan Inc.
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
|
|
#include <netdb.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
|
|
#include "util.h"
|
|
|
|
/*
|
|
* Convert argument like "192.168.160.1:1723/tcp" or "[::1]:1723/tcp" to
|
|
* match getaddrinfo(3)'s specification and pass them to getaddrinfo(3).
|
|
*/
|
|
int
|
|
addrport_parse(const char *addrport, int proto, struct addrinfo **p_ai)
|
|
{
|
|
char *servp, *nodep, *slash, buf[256];
|
|
struct addrinfo hints;
|
|
|
|
strlcpy(buf, addrport, sizeof(buf));
|
|
if (buf[0] == '[' && (servp = strchr(buf, ']')) != NULL) {
|
|
nodep = buf + 1;
|
|
*servp++ = '\0';
|
|
if (*servp != ':')
|
|
servp = NULL;
|
|
} else {
|
|
nodep = buf;
|
|
servp = strrchr(nodep, ':');
|
|
}
|
|
if (servp != NULL) {
|
|
*servp = '\0';
|
|
servp++;
|
|
slash = strrchr(servp, '/');
|
|
if (slash != NULL) {
|
|
/*
|
|
* Ignore like "/tcp"
|
|
*/
|
|
*slash = '\0';
|
|
slash++;
|
|
}
|
|
} else
|
|
servp = NULL;
|
|
memset(&hints, 0, sizeof(hints));
|
|
hints.ai_flags = AI_NUMERICHOST;
|
|
hints.ai_family = AF_UNSPEC;
|
|
switch (proto) {
|
|
case IPPROTO_TCP:
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
break;
|
|
case IPPROTO_UDP:
|
|
hints.ai_socktype = SOCK_DGRAM;
|
|
break;
|
|
}
|
|
hints.ai_protocol = proto;
|
|
|
|
return (getaddrinfo(nodep, servp, &hints, p_ai));
|
|
}
|
|
|
|
/*
|
|
* Make a string like "192.168.160.1:1723" or "[::1]:1723" from a struct
|
|
* sockaddr
|
|
*/
|
|
const char *
|
|
addrport_tostring(struct sockaddr *sa, socklen_t salen, char *buf, size_t lbuf)
|
|
{
|
|
char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV];
|
|
int ret;
|
|
|
|
if (getnameinfo(sa, salen, hbuf, sizeof(hbuf), sbuf, sizeof(sbuf),
|
|
NI_NUMERICHOST | NI_NUMERICSERV) != 0)
|
|
return (NULL);
|
|
|
|
switch (sa->sa_family) {
|
|
case AF_INET6:
|
|
ret = snprintf(buf, lbuf, "[%s]:%s", hbuf, sbuf);
|
|
break;
|
|
|
|
case AF_INET:
|
|
ret = snprintf(buf, lbuf, "%s:%s", hbuf, sbuf);
|
|
break;
|
|
|
|
default:
|
|
return "error";
|
|
}
|
|
|
|
if (ret < 0 || ret >= (int)lbuf)
|
|
return "(error)";
|
|
return (buf);
|
|
}
|