int ipvs_init(void)
{
socklen_t len;
ipvs_func = ipvs_init;
#ifdef LIBIPVS_USE_NL
try_nl = 1;
if (ipvs_nl_send_message(NULL, NULL, NULL) == 0) {
try_nl = 1;
return ipvs_getinfo();
}
try_nl = 0;
#endif
len = sizeof(ipvs_info);
if ((sockfd = socket(AF_INET, SOCK_RAW | SOCK_CLOEXEC, IPPROTO_RAW)) == -1)
return -1;
#ifndef _HAVE_SOCK_CLOEXEC_
if (set_sock_flags(sockfd, F_SETFD, FD_CLOEXEC))
return -1;
#endif
if (getsockopt(sockfd, IPPROTO_IP, IP_VS_SO_GET_INFO,
(char *)&ipvs_info, &len))
return -1;
return 0;
}
int ipvs_init(void)
{
socklen_t len;
len = sizeof(ipvs_info);
if ((sockfd = socket(AF_INET, SOCK_RAW | SOCK_CLOEXEC, IPPROTO_RAW)) == -1)
return -1;
#ifndef _HAVE_SOCK_CLOEXEC_
if (set_sock_flags(sockfd, F_SETFD, FD_CLOEXEC))
log_message(LOG_INFO, "Unable to set CLOEXEC on ipvs socket - %s (%d)", strerror(errno), errno);
#endif
ipvs_cmd = GET_CMD(IP_VS_SO_GET_INFO);
if (getsockopt(sockfd, IPPROTO_IP, IP_VS_SO_GET_INFO,
(char *)&ipvs_info, &len))
return -1;
return 0;
}
static int
if_ethtool_probe(const char *ifname)
{
int fd = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0);
int status = 0;
if (fd < 0)
return -1;
#ifndef _HAVE_SOCK_CLOEXEC_
if (set_sock_flags(fd, F_SETFD, FD_CLOEXEC))
log_message(LOG_INFO, "Unable to set CLOEXEC on ethtool_probe socket - %s (%d)", strerror(errno), errno);
#endif
memset(&ifr, 0, sizeof (struct ifreq));
strncpy(ifr.ifr_name, ifname, sizeof (ifr.ifr_name));
status = if_ethtool_status(fd);
close(fd);
return status;
}
/* connect remote SMTP server */
static void
smtp_connect(smtp_t * smtp)
{
enum connect_result status;
if ((smtp->fd = socket(global_data->smtp_server.ss_family, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP)) == -1) {
DBG("SMTP connect fail to create socket.");
free_smtp_all(smtp);
return;
}
#if !HAVE_DECL_SOCK_CLOEXEC
if (set_sock_flags(smtp->fd, F_SETFD, FD_CLOEXEC))
log_message(LOG_INFO, "Unable to set CLOEXEC on smtp_connect socket - %s (%d)", strerror(errno), errno);
#endif
status = tcp_connect(smtp->fd, &global_data->smtp_server);
/* Handle connection status code */
thread_add_event(master, SMTP_FSM[status].send, smtp, smtp->fd);
}
static void
if_ioctl_flags(interface_t * ifp)
{
int fd = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0);
if (fd < 0)
return;
#ifndef _HAVE_SOCK_CLOEXEC_
if (set_sock_flags(fd, F_SETFD, FD_CLOEXEC))
log_message(LOG_INFO, "Unable to set CLOEXEC on ioctl_flags socket - %s (%d)", strerror(errno), errno);
#endif
memset(&ifr, 0, sizeof (struct ifreq));
strncpy(ifr.ifr_name, ifp->ifname, sizeof (ifr.ifr_name));
if (ioctl(fd, SIOCGIFFLAGS, &ifr) < 0) {
close(fd);
return;
}
ifp->flags = ifr.ifr_flags;
close(fd);
}
static int
if_mii_probe(const char *ifname)
{
uint16_t *data = (uint16_t *) (&ifr.ifr_data);
int phy_id;
int fd = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0);
int status = 0;
if (fd < 0)
return -1;
#ifndef _HAVE_SOCK_CLOEXEC_
if (set_sock_flags(fd, F_SETFD, FD_CLOEXEC))
log_message(LOG_INFO, "Unable to set CLOEXEC on mii_probe socket - %s (%d)", strerror(errno), errno);
#endif
memset(&ifr, 0, sizeof (struct ifreq));
strncpy(ifr.ifr_name, ifname, sizeof (ifr.ifr_name));
if (ioctl(fd, SIOCGMIIPHY, &ifr) < 0) {
close(fd);
return -1;
}
/* check if the driver reports BMSR using the MII interface, as we
* will need this and we already know that some don't support it.
*/
phy_id = data[0]; /* save it in case it is overwritten */
data[1] = 1;
if (ioctl(fd, SIOCGMIIREG, &ifr) < 0) {
close(fd);
return -1;
}
data[0] = phy_id;
/* Dump the MII transceiver */
status = if_mii_status(fd);
close(fd);
return status;
}
/* Create a socket to netlink interface_t */
static int
netlink_socket(nl_handle_t *nl, int flags, int group, ...)
{
int ret;
va_list gp;
memset(nl, 0, sizeof (*nl));
#ifdef _HAVE_LIBNL3_
/* We need to keep libnl3 in step with our netlink socket creation. */
nl->sk = nl_socket_alloc();
if ( nl->sk == NULL ) {
log_message(LOG_INFO, "Netlink: Cannot allocate netlink socket" );
return -1;
}
ret = nl_connect(nl->sk, NETLINK_ROUTE);
if (ret != 0) {
log_message(LOG_INFO, "Netlink: Cannot open netlink socket : (%d)", ret);
return -1;
}
/* Unfortunately we can't call nl_socket_add_memberships() with variadic arguments
* from a variadic argument list passed to us
*/
va_start(gp, group);
while (group != 0) {
if (group < 0) {
va_end(gp);
return -1;
}
if ((ret = nl_socket_add_membership(nl->sk, group))) {
log_message(LOG_INFO, "Netlink: Cannot add socket membership 0x%x : (%d)", group, ret);
return -1;
}
group = va_arg(gp,int);
}
va_end(gp);
if (flags & SOCK_NONBLOCK) {
if ((ret = nl_socket_set_nonblocking(nl->sk))) {
log_message(LOG_INFO, "Netlink: Cannot set netlink socket non-blocking : (%d)", ret);
return -1;
}
}
if ((ret = nl_socket_set_buffer_size(nl->sk, IF_DEFAULT_BUFSIZE, 0))) {
log_message(LOG_INFO, "Netlink: Cannot set netlink buffer size : (%d)", ret);
return -1;
}
nl->nl_pid = nl_socket_get_local_port(nl->sk);
nl->fd = nl_socket_get_fd(nl->sk);
/* Set CLOEXEC */
fcntl(nl->fd, F_SETFD, fcntl(nl->fd, F_GETFD) | FD_CLOEXEC);
#else
socklen_t addr_len;
struct sockaddr_nl snl;
#if !HAVE_DECL_SOCK_NONBLOCK
int sock_flags = flags;
flags &= ~SOCK_NONBLOCK;
#endif
nl->fd = socket(AF_NETLINK, SOCK_RAW | SOCK_CLOEXEC | flags, NETLINK_ROUTE);
if (nl->fd < 0) {
log_message(LOG_INFO, "Netlink: Cannot open netlink socket : (%s)",
strerror(errno));
return -1;
}
#if !HAVE_DECL_SOCK_NONBLOCK
if ((sock_flags & SOCK_NONBLOCK) &&
set_sock_flags(nl->fd, F_SETFL, O_NONBLOCK))
return -1;
#endif
memset(&snl, 0, sizeof (snl));
snl.nl_family = AF_NETLINK;
ret = bind(nl->fd, (struct sockaddr *) &snl, sizeof (snl));
if (ret < 0) {
log_message(LOG_INFO, "Netlink: Cannot bind netlink socket : (%s)",
strerror(errno));
close(nl->fd);
return -1;
}
/* Join the requested groups */
va_start(gp, group);
while (group != 0) {
if (group < 0) {
va_end(gp);
return -1;
}
ret = setsockopt(nl->fd, SOL_NETLINK, NETLINK_ADD_MEMBERSHIP, &group, sizeof(group));
if (ret < 0) {
log_message(LOG_INFO, "Netlink: Cannot add membership on netlink socket : (%s)",
strerror(errno));
va_end(gp);
return -1;
}
group = va_arg(gp,int);
}
va_end(gp);
addr_len = sizeof (snl);
ret = getsockname(nl->fd, (struct sockaddr *) &snl, &addr_len);
if (ret < 0 || addr_len != sizeof (snl)) {
log_message(LOG_INFO, "Netlink: Cannot getsockname : (%s)",
strerror(errno));
close(nl->fd);
return -1;
}
if (snl.nl_family != AF_NETLINK) {
log_message(LOG_INFO, "Netlink: Wrong address family %d",
snl.nl_family);
close(nl->fd);
return -1;
}
/* Save the port id for checking message source later */
nl->nl_pid = snl.nl_pid;
/* Set default rcvbuf size */
if_setsockopt_rcvbuf(&nl->fd, IF_DEFAULT_BUFSIZE);
#endif
nl->seq = (uint32_t)time(NULL);
if (nl->fd < 0)
return -1;
return ret;
}
