1
0
mirror of https://github.com/haiwen/ccnet-server.git synced 2025-04-28 10:33:20 +00:00
ccnet-server/lib/net.c
2016-08-19 13:54:34 +08:00

469 lines
10 KiB
C

/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
#ifdef WIN32
#define WINVER 0x0501
#endif
#include "include.h"
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <string.h>
#ifdef WIN32
#include <inttypes.h>
#include <winsock2.h>
#include <ctype.h>
#include <ws2tcpip.h>
#define UNUSED
#else
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <sys/un.h>
#include <net/if.h>
#include <netinet/tcp.h>
#endif
#include <fcntl.h>
#include "net.h"
#ifdef WIN32
#ifndef inet_aton
int inet_aton(const char *string, struct in_addr *addr)
{
addr->s_addr = inet_addr(string);
if (addr->s_addr != -1 || strcmp("255.255.255.255", string) == 0)
return 1;
return 0;
}
#endif
#endif //WIN32
int
ccnet_netSetTOS (evutil_socket_t s, int tos)
{
#ifdef IP_TOS
return setsockopt( s, IPPROTO_IP, IP_TOS, (char*)&tos, sizeof( tos ) );
#else
return 0;
#endif
}
static evutil_socket_t
makeSocketNonBlocking (evutil_socket_t fd)
{
if (fd >= 0)
{
if (evutil_make_socket_nonblocking(fd))
{
ccnet_warning ("Couldn't make socket nonblock: %s",
evutil_socket_error_to_string(EVUTIL_SOCKET_ERROR()));
evutil_closesocket(fd);
fd = -1;
}
}
return fd;
}
static evutil_socket_t
createSocket (int family, int nonblock)
{
evutil_socket_t fd;
int ret;
fd = socket (family, SOCK_STREAM, 0);
if (fd < 0) {
ccnet_warning("create Socket failed %d\n", fd);
} else if (nonblock) {
int nodelay = 1;
fd = makeSocketNonBlocking( fd );
ret = setsockopt (fd, IPPROTO_TCP, TCP_NODELAY,
(char *)&nodelay, sizeof(nodelay));
if (ret < 0) {
ccnet_warning("setsockopt failed\n");
evutil_closesocket(fd);
return -1;
}
}
return fd;
}
evutil_socket_t
ccnet_net_open_tcp (const struct sockaddr *sa, int nonblock)
{
evutil_socket_t s;
int sa_len;
if( (s = createSocket(sa->sa_family, nonblock)) < 0 )
return -1;
#ifndef WIN32
if (sa->sa_family == AF_INET)
sa_len = sizeof (struct sockaddr_in);
else
sa_len = sizeof (struct sockaddr_in6);
#else
if (sa->sa_family == AF_INET)
sa_len = sizeof (struct sockaddr_in);
else
return -1;
#endif
if( (connect(s, sa, sa_len) < 0)
#ifdef WIN32
&& (sockerrno != WSAEWOULDBLOCK)
#endif
&& (sockerrno != EINPROGRESS) )
{
evutil_closesocket(s);
s = -1;
}
return s;
}
evutil_socket_t
ccnet_net_bind_tcp (int port, int nonblock)
{
#ifndef WIN32
int sockfd, n;
struct addrinfo hints, *res, *ressave;
char buf[10];
memset (&hints, 0,sizeof (struct addrinfo));
hints.ai_flags = AI_PASSIVE;
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
snprintf (buf, sizeof(buf), "%d", port);
if ( (n = getaddrinfo(NULL, buf, &hints, &res) ) != 0) {
ccnet_warning ("getaddrinfo fails: %s\n", gai_strerror(n));
return -1;
}
ressave = res;
do {
int on = 1;
sockfd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
if (sockfd < 0)
continue; /* error - try next one */
if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0) {
ccnet_warning ("setsockopt of SO_REUSEADDR error\n");
continue;
}
if (nonblock)
sockfd = makeSocketNonBlocking (sockfd);
if (sockfd < 0)
continue; /* error - try next one */
if (bind(sockfd, res->ai_addr, res->ai_addrlen) == 0)
break; /* success */
close(sockfd); /* bind error - close and try next one */
} while ( (res = res->ai_next) != NULL);
freeaddrinfo (ressave);
if (res == NULL) {
ccnet_warning ("bind fails: %s\n", strerror(errno));
return -1;
}
return sockfd;
#else
evutil_socket_t s;
struct sockaddr_in sock;
const int type = AF_INET;
#if defined( SO_REUSEADDR ) || defined( SO_REUSEPORT )
int optval;
#endif
if ((s = createSocket(type, nonblock)) < 0)
return -1;
optval = 1;
setsockopt (s, SOL_SOCKET, SO_REUSEADDR, (char*)&optval, sizeof(optval));
memset(&sock, 0, sizeof(sock));
sock.sin_family = AF_INET;
sock.sin_addr.s_addr = INADDR_ANY;
sock.sin_port = htons(port);
if ( bind(s, (struct sockaddr *)&sock, sizeof(struct sockaddr_in)) < 0)
{
ccnet_warning ("bind fails: %s\n", strerror(errno));
evutil_closesocket (s);
return -1;
}
if (nonblock)
s = makeSocketNonBlocking (s);
return s;
#endif
}
evutil_socket_t
ccnet_net_accept (evutil_socket_t b, struct sockaddr_storage *cliaddr,
socklen_t *len, int nonblock)
{
evutil_socket_t s;
/* int nodelay = 1; */
s = accept (b, (struct sockaddr *)cliaddr, len);
/* setsockopt (s, IPPROTO_TCP, TCP_NODELAY, &nodelay, sizeof(nodelay)); */
if (nonblock)
makeSocketNonBlocking(s);
return s;
}
evutil_socket_t
ccnet_net_bind_v4 (const char *ipaddr, int *port)
{
evutil_socket_t sockfd;
struct sockaddr_in addr;
int on = 1;
sockfd = socket (AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
ccnet_warning("create socket failed: %s\n", strerror(errno));
exit(-1);
}
memset (&addr, 0, sizeof (struct sockaddr_in));
addr.sin_family = AF_INET;
if (inet_aton(ipaddr, &addr.sin_addr) == 0) {
ccnet_warning ("Bad ip address %s\n", ipaddr);
return -1;
}
addr.sin_port = htons (*port);
if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on)) < 0)
{
ccnet_warning ("setsockopt of SO_REUSEADDR error: %s\n",
strerror(errno));
return -1;
}
if ( bind(sockfd, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
ccnet_warning ("Bind error: %s\n", strerror (errno));
return -1;
}
if (*port == 0) {
struct sockaddr_storage ss;
socklen_t len;
len = sizeof(ss);
if (getsockname(sockfd, (struct sockaddr *)&ss, &len) < 0) {
ccnet_warning ("getsockname error: %s\n", strerror(errno));
return -1;
}
*port = sock_port ((struct sockaddr *)&ss);
}
return sockfd;
}
char *
sock_ntop(const struct sockaddr *sa, socklen_t salen)
{
static char str[128]; /* Unix domain is largest */
switch (sa->sa_family) {
case AF_INET: {
struct sockaddr_in *sin = (struct sockaddr_in *) sa;
if (evutil_inet_ntop(AF_INET, &sin->sin_addr, str, sizeof(str)) == NULL)
return(NULL);
return(str);
}
#ifdef IPv6
case AF_INET6: {
struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *) sa;
if (evutil_inet_ntop(AF_INET6, &sin6->sin6_addr, str, sizeof(str) - 1) == NULL)
return(NULL);
return (str);
}
#endif
#ifndef WIN32
#ifdef AF_UNIX
case AF_UNIX: {
struct sockaddr_un *unp = (struct sockaddr_un *) sa;
/* OK to have no pathname bound to the socket: happens on
every connect() unless client calls bind() first. */
if (unp->sun_path[0] == 0)
strcpy(str, "(no pathname bound)");
else
snprintf(str, sizeof(str), "%s", unp->sun_path);
return(str);
}
#endif
#endif
default:
snprintf(str, sizeof(str), "sock_ntop: unknown AF_xxx: %d, len %d",
sa->sa_family, salen);
return(str);
}
return (NULL);
}
int
sock_pton (const char *addr_str, uint16_t port, struct sockaddr_storage *sa)
{
struct sockaddr_in *saddr = (struct sockaddr_in *) sa;
#ifndef WIN32
struct sockaddr_in6 *saddr6 = (struct sockaddr_in6 *) sa;
#endif
if (evutil_inet_pton (AF_INET, addr_str, &saddr->sin_addr) == 1 ) {
saddr->sin_family = AF_INET;
saddr->sin_port = htons (port);
return 0;
}
#ifndef WIN32
else if (evutil_inet_pton (AF_INET6, addr_str, &saddr6->sin6_addr) == 1)
{
saddr6->sin6_family = AF_INET6;
saddr6->sin6_port = htons (port);
return 0;
}
#endif
return -1;
}
/* return 1 if addr_str is a valid ipv4 or ipv6 address */
int
is_valid_ipaddr (const char *addr_str)
{
struct sockaddr_storage addr;
if (!addr_str)
return 0;
if (sock_pton(addr_str, 0, &addr) < 0)
return 0;
return 1;
}
uint16_t
sock_port (const struct sockaddr *sa)
{
switch (sa->sa_family) {
case AF_INET: {
struct sockaddr_in *sin = (struct sockaddr_in *) sa;
return ntohs(sin->sin_port);
}
#ifdef IPv6
case AF_INET6: {
struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *) sa;
return ntohs(sin6->sin6_port);
}
#endif
default:
return 0;
}
return 0;
}
evutil_socket_t
udp_client (const char *host, const char *serv,
struct sockaddr **saptr, socklen_t *lenp)
{
evutil_socket_t sockfd;
int n;
struct addrinfo hints, *res, *ressave;
memset (&hints, 0, sizeof(struct addrinfo));
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_DGRAM;
if ((n = getaddrinfo(host, serv, &hints, &res)) != 0) {
ccnet_warning ("udp_client error for %s, %s: %s",
host, serv, gai_strerror(n));
return -1;
}
ressave = res;
do {
sockfd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
if (sockfd >= 0)
break; /* success */
} while ( (res = res->ai_next) != NULL);
if (res == NULL) { /* errno set from final socket() */
ccnet_warning ("udp_client error for %s, %s", host, serv);
freeaddrinfo (ressave);
return -1;
}
*saptr = malloc(res->ai_addrlen);
memcpy(*saptr, res->ai_addr, res->ai_addrlen);
*lenp = res->ai_addrlen;
freeaddrinfo(ressave);
return (sockfd);
}
int
family_to_level(int family)
{
switch (family) {
case AF_INET:
return IPPROTO_IP;
#ifdef IPV6
case AF_INET6:
return IPPROTO_IPV6;
#endif
default:
return -1;
}
}
int
sockfd_to_family(evutil_socket_t sockfd)
{
struct sockaddr_storage ss;
socklen_t len;
len = sizeof(ss);
if (getsockname(sockfd, (struct sockaddr *) &ss, &len) < 0)
return(-1);
return(ss.ss_family);
}