407 lines
7.9 KiB
C
407 lines
7.9 KiB
C
/*
|
|
* lib/socket.c Netlink Socket
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation version 2.1
|
|
* of the License.
|
|
*
|
|
* Copyright (c) 2003-2008 Thomas Graf <tgraf@suug.ch>
|
|
*/
|
|
|
|
/**
|
|
* @ingroup core
|
|
* @defgroup socket Socket
|
|
* @{
|
|
*/
|
|
|
|
#include <netlink-local.h>
|
|
#include <netlink/netlink.h>
|
|
#include <netlink/utils.h>
|
|
#include <netlink/handlers.h>
|
|
#include <netlink/msg.h>
|
|
#include <netlink/attr.h>
|
|
|
|
static uint32_t used_ports_map[32];
|
|
|
|
static uint32_t generate_local_port(void)
|
|
{
|
|
int i, n;
|
|
uint32_t pid = getpid() & 0x3FFFFF;
|
|
|
|
for (i = 0; i < 32; i++) {
|
|
if (used_ports_map[i] == 0xFFFFFFFF)
|
|
continue;
|
|
|
|
for (n = 0; n < 32; n++) {
|
|
if (1UL & (used_ports_map[i] >> n))
|
|
continue;
|
|
|
|
used_ports_map[i] |= (1UL << n);
|
|
n += (i * 32);
|
|
|
|
/* PID_MAX_LIMIT is currently at 2^22, leaving 10 bit
|
|
* to, i.e. 1024 unique ports per application. */
|
|
return pid + (n << 22);
|
|
|
|
}
|
|
}
|
|
|
|
/* Out of sockets in our own PID namespace, what to do? FIXME */
|
|
return UINT_MAX;
|
|
}
|
|
|
|
static void release_local_port(uint32_t port)
|
|
{
|
|
int nr;
|
|
|
|
if (port == UINT_MAX)
|
|
return;
|
|
|
|
nr = port >> 22;
|
|
used_ports_map[nr / 32] &= ~((nr % 32) + 1);
|
|
}
|
|
|
|
/**
|
|
* @name Allocation
|
|
* @{
|
|
*/
|
|
|
|
static struct nl_sock *__alloc_socket(struct nl_cb *cb)
|
|
{
|
|
struct nl_sock *sk;
|
|
|
|
sk = calloc(1, sizeof(*sk));
|
|
if (!sk)
|
|
return NULL;
|
|
|
|
sk->s_fd = -1;
|
|
sk->s_cb = cb;
|
|
sk->s_local.nl_family = AF_NETLINK;
|
|
sk->s_peer.nl_family = AF_NETLINK;
|
|
sk->s_seq_expect = sk->s_seq_next = time(0);
|
|
sk->s_local.nl_pid = generate_local_port();
|
|
if (sk->s_local.nl_pid == UINT_MAX) {
|
|
nl_socket_free(sk);
|
|
return NULL;
|
|
}
|
|
|
|
return sk;
|
|
}
|
|
|
|
/**
|
|
* Allocate new netlink socket
|
|
*
|
|
* @return Newly allocated netlink socket or NULL.
|
|
*/
|
|
struct nl_sock *nl_socket_alloc(void)
|
|
{
|
|
struct nl_cb *cb;
|
|
|
|
cb = nl_cb_alloc(NL_CB_DEFAULT);
|
|
if (!cb)
|
|
return NULL;
|
|
|
|
return __alloc_socket(cb);
|
|
}
|
|
|
|
/**
|
|
* Allocate new socket with custom callbacks
|
|
* @arg cb Callback handler
|
|
*
|
|
* The reference to the callback handler is taken into account
|
|
* automatically, it is released again upon calling nl_socket_free().
|
|
*
|
|
*@return Newly allocted socket handle or NULL.
|
|
*/
|
|
struct nl_sock *nl_socket_alloc_cb(struct nl_cb *cb)
|
|
{
|
|
if (cb == NULL)
|
|
BUG();
|
|
|
|
return __alloc_socket(nl_cb_get(cb));
|
|
}
|
|
|
|
/**
|
|
* Free a netlink socket.
|
|
* @arg sk Netlink socket.
|
|
*/
|
|
void nl_socket_free(struct nl_sock *sk)
|
|
{
|
|
if (!sk)
|
|
return;
|
|
|
|
if (sk->s_fd >= 0)
|
|
close(sk->s_fd);
|
|
|
|
if (!(sk->s_flags & NL_OWN_PORT))
|
|
release_local_port(sk->s_local.nl_pid);
|
|
|
|
nl_cb_put(sk->s_cb);
|
|
free(sk);
|
|
}
|
|
|
|
/** @} */
|
|
|
|
/**
|
|
* @name Sequence Numbers
|
|
* @{
|
|
*/
|
|
|
|
static int noop_seq_check(struct nl_msg *msg, void *arg)
|
|
{
|
|
return NL_OK;
|
|
}
|
|
|
|
|
|
/**
|
|
* Disable sequence number checking.
|
|
* @arg sk Netlink socket.
|
|
*
|
|
* Disables checking of sequence numbers on the netlink socket This is
|
|
* required to allow messages to be processed which were not requested by
|
|
* a preceding request message, e.g. netlink events.
|
|
*
|
|
* @note This function modifies the NL_CB_SEQ_CHECK configuration in
|
|
* the callback handle associated with the socket.
|
|
*/
|
|
void nl_socket_disable_seq_check(struct nl_sock *sk)
|
|
{
|
|
nl_cb_set(sk->s_cb, NL_CB_SEQ_CHECK,
|
|
NL_CB_CUSTOM, noop_seq_check, NULL);
|
|
}
|
|
|
|
/** @} */
|
|
|
|
/**
|
|
* Set local port of socket
|
|
* @arg sk Netlink socket.
|
|
* @arg port Local port identifier
|
|
*
|
|
* Assigns a local port identifier to the socket. If port is 0
|
|
* a unique port identifier will be generated automatically.
|
|
*/
|
|
void nl_socket_set_local_port(struct nl_sock *sk, uint32_t port)
|
|
{
|
|
if (port == 0) {
|
|
port = generate_local_port();
|
|
sk->s_flags &= ~NL_OWN_PORT;
|
|
} else {
|
|
if (!(sk->s_flags & NL_OWN_PORT))
|
|
release_local_port(sk->s_local.nl_pid);
|
|
sk->s_flags |= NL_OWN_PORT;
|
|
}
|
|
|
|
sk->s_local.nl_pid = port;
|
|
}
|
|
|
|
/** @} */
|
|
|
|
/**
|
|
* @name Group Subscriptions
|
|
* @{
|
|
*/
|
|
|
|
/**
|
|
* Join groups
|
|
* @arg sk Netlink socket
|
|
* @arg group Group identifier
|
|
*
|
|
* Joins the specified groups using the modern socket option which
|
|
* is available since kernel version 2.6.14. It allows joining an
|
|
* almost arbitary number of groups without limitation. The list
|
|
* of groups has to be terminated by 0 (%NFNLGRP_NONE).
|
|
*
|
|
* Make sure to use the correct group definitions as the older
|
|
* bitmask definitions for nl_join_groups() are likely to still
|
|
* be present for backward compatibility reasons.
|
|
*
|
|
* @return 0 on sucess or a negative error code.
|
|
*/
|
|
int nl_socket_add_memberships(struct nl_sock *sk, int group, ...)
|
|
{
|
|
int err;
|
|
va_list ap;
|
|
|
|
if (sk->s_fd == -1)
|
|
return -NLE_BAD_SOCK;
|
|
|
|
va_start(ap, group);
|
|
|
|
while (group != 0) {
|
|
if (group < 0)
|
|
return -NLE_INVAL;
|
|
|
|
err = setsockopt(sk->s_fd, SOL_NETLINK, NETLINK_ADD_MEMBERSHIP,
|
|
&group, sizeof(group));
|
|
if (err < 0)
|
|
return -nl_syserr2nlerr(errno);
|
|
|
|
group = va_arg(ap, int);
|
|
}
|
|
|
|
va_end(ap);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Leave groups
|
|
* @arg sk Netlink socket
|
|
* @arg group Group identifier
|
|
*
|
|
* Leaves the specified groups using the modern socket option
|
|
* which is available since kernel version 2.6.14. The list of groups
|
|
* has to terminated by 0 (%NFNLGRP_NONE).
|
|
*
|
|
* @see nl_socket_add_membership
|
|
* @return 0 on success or a negative error code.
|
|
*/
|
|
int nl_socket_drop_memberships(struct nl_sock *sk, int group, ...)
|
|
{
|
|
int err;
|
|
va_list ap;
|
|
|
|
if (sk->s_fd == -1)
|
|
return -NLE_BAD_SOCK;
|
|
|
|
va_start(ap, group);
|
|
|
|
while (group != 0) {
|
|
if (group < 0)
|
|
return -NLE_INVAL;
|
|
|
|
err = setsockopt(sk->s_fd, SOL_NETLINK, NETLINK_DROP_MEMBERSHIP,
|
|
&group, sizeof(group));
|
|
if (err < 0)
|
|
return -nl_syserr2nlerr(errno);
|
|
|
|
group = va_arg(ap, int);
|
|
}
|
|
|
|
va_end(ap);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
/** @} */
|
|
|
|
/**
|
|
* Set file descriptor of socket to non-blocking state
|
|
* @arg sk Netlink socket.
|
|
*
|
|
* @return 0 on success or a negative error code.
|
|
*/
|
|
int nl_socket_set_nonblocking(struct nl_sock *sk)
|
|
{
|
|
if (sk->s_fd == -1)
|
|
return -NLE_BAD_SOCK;
|
|
|
|
if (fcntl(sk->s_fd, F_SETFL, O_NONBLOCK) < 0)
|
|
return -nl_syserr2nlerr(errno);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/** @} */
|
|
|
|
/**
|
|
* @name Utilities
|
|
* @{
|
|
*/
|
|
|
|
/**
|
|
* Set socket buffer size of netlink socket.
|
|
* @arg sk Netlink socket.
|
|
* @arg rxbuf New receive socket buffer size in bytes.
|
|
* @arg txbuf New transmit socket buffer size in bytes.
|
|
*
|
|
* Sets the socket buffer size of a netlink socket to the specified
|
|
* values \c rxbuf and \c txbuf. Providing a value of \c 0 assumes a
|
|
* good default value.
|
|
*
|
|
* @note It is not required to call this function prior to nl_connect().
|
|
* @return 0 on sucess or a negative error code.
|
|
*/
|
|
int nl_socket_set_buffer_size(struct nl_sock *sk, int rxbuf, int txbuf)
|
|
{
|
|
int err;
|
|
|
|
if (rxbuf <= 0)
|
|
rxbuf = 32768;
|
|
|
|
if (txbuf <= 0)
|
|
txbuf = 32768;
|
|
|
|
if (sk->s_fd == -1)
|
|
return -NLE_BAD_SOCK;
|
|
|
|
err = setsockopt(sk->s_fd, SOL_SOCKET, SO_SNDBUF,
|
|
&txbuf, sizeof(txbuf));
|
|
if (err < 0)
|
|
return -nl_syserr2nlerr(errno);
|
|
|
|
err = setsockopt(sk->s_fd, SOL_SOCKET, SO_RCVBUF,
|
|
&rxbuf, sizeof(rxbuf));
|
|
if (err < 0)
|
|
return -nl_syserr2nlerr(errno);
|
|
|
|
sk->s_flags |= NL_SOCK_BUFSIZE_SET;
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Enable/disable credential passing on netlink socket.
|
|
* @arg sk Netlink socket.
|
|
* @arg state New state (0 - disabled, 1 - enabled)
|
|
*
|
|
* @return 0 on success or a negative error code
|
|
*/
|
|
int nl_socket_set_passcred(struct nl_sock *sk, int state)
|
|
{
|
|
int err;
|
|
|
|
if (sk->s_fd == -1)
|
|
return -NLE_BAD_SOCK;
|
|
|
|
err = setsockopt(sk->s_fd, SOL_SOCKET, SO_PASSCRED,
|
|
&state, sizeof(state));
|
|
if (err < 0)
|
|
return -nl_syserr2nlerr(errno);
|
|
|
|
if (state)
|
|
sk->s_flags |= NL_SOCK_PASSCRED;
|
|
else
|
|
sk->s_flags &= ~NL_SOCK_PASSCRED;
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Enable/disable receival of additional packet information
|
|
* @arg sk Netlink socket.
|
|
* @arg state New state (0 - disabled, 1 - enabled)
|
|
*
|
|
* @return 0 on success or a negative error code
|
|
*/
|
|
int nl_socket_recv_pktinfo(struct nl_sock *sk, int state)
|
|
{
|
|
int err;
|
|
|
|
if (sk->s_fd == -1)
|
|
return -NLE_BAD_SOCK;
|
|
|
|
err = setsockopt(sk->s_fd, SOL_NETLINK, NETLINK_PKTINFO,
|
|
&state, sizeof(state));
|
|
if (err < 0)
|
|
return -nl_syserr2nlerr(errno);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/** @} */
|
|
|
|
/** @} */
|