def test_s_proto():
"""Test socket protocol."""
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
msg = nlmsg_alloc_simple(0, 0)
assert -1 == msg.nm_protocol
nl_complete_msg(sk, msg)
assert NETLINK_ROUTE == msg.nm_protocol
nl_socket_free(sk)
def test_nl_recv():
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
// Send data to the kernel.
struct nl_sock *sk = nl_socket_alloc();
sk->s_seq_next = 0;
nl_connect(sk, NETLINK_ROUTE);
int ret = nl_send_simple(sk, 0, NLM_F_REQUEST, NULL, 0);
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
// nl_recvmsgs_default(sk);
// return 0;
unsigned char *buf = NULL;
struct sockaddr_nl nla = {0};
printf("%d == nla.nl_family\n", nla.nl_family);
int n = nl_recv(sk, &nla, &buf, NULL);
printf("Bytes Recv: %d\n", n);
int i = 0; for (i = 0; i<n; i++) printf("%02x", buf[i]); printf("\n");
printf("%d == nla.nl_family\n", nla.nl_family);
return 0;
}
// Expected output:
// Bytes Sent: 16
// 0 == nla.nl_family
// Bytes Recv: 36
// 240000000200000000000000844c000000000000100000000000050000000000844c0000
// 16 == nla.nl_family
// Output delta:
// 240000000200000000000000ac4e000000000000100000000000050000000000ac4e0000
"""
sk = nl_socket_alloc()
sk.s_seq_next = 0
nl_connect(sk, NETLINK_ROUTE)
assert 16 == nl_send_simple(sk, 0, NLM_F_REQUEST, None)
buf = bytearray()
nla = sockaddr_nl()
assert 0 == nla.nl_family
assert 36 == nl_recv(sk, nla, buf, None)
nl_socket_free(sk)
buf_hex = binascii.hexlify(buffer(buf)).decode('ascii')
assert re.match(
r'240000000200000000000000....000000000000100000000000050000000000....0000',
buf_hex)
assert 16 == nla.nl_family
def test_nl_recv():
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
// Send data to the kernel.
struct nl_sock *sk = nl_socket_alloc();
sk->s_seq_next = 0;
nl_connect(sk, NETLINK_ROUTE);
int ret = nl_send_simple(sk, 0, NLM_F_REQUEST, NULL, 0);
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
// nl_recvmsgs_default(sk);
// return 0;
unsigned char *buf = NULL;
struct sockaddr_nl nla = {0};
printf("%d == nla.nl_family\n", nla.nl_family);
int n = nl_recv(sk, &nla, &buf, NULL);
printf("Bytes Recv: %d\n", n);
int i = 0; for (i = 0; i<n; i++) printf("%02x", buf[i]); printf("\n");
printf("%d == nla.nl_family\n", nla.nl_family);
return 0;
}
// Expected output:
// Bytes Sent: 16
// 0 == nla.nl_family
// Bytes Recv: 36
// 240000000200000000000000844c000000000000100000000000050000000000844c0000
// 16 == nla.nl_family
// Output delta:
// 240000000200000000000000ac4e000000000000100000000000050000000000ac4e0000
"""
sk = nl_socket_alloc()
sk.s_seq_next = 0
nl_connect(sk, NETLINK_ROUTE)
assert 16 == nl_send_simple(sk, 0, NLM_F_REQUEST, None)
buf = bytearray()
nla = sockaddr_nl()
assert 0 == nla.nl_family
assert 36 == nl_recv(sk, nla, buf, None)
nl_socket_free(sk)
buf_hex = binascii.hexlify(buffer(buf)).decode('ascii')
assert re.match(r'240000000200000000000000....000000000000100000000000050000000000....0000', buf_hex)
assert 16 == nla.nl_family
def test_dissect(monkeypatch):
r"""Diff of C code (from test_bare()) to test against.
--- test_bare.c 2015-02-08 12:43:15.543135855 -0800
+++ test_dissect.c 2015-02-08 13:25:31.375715668 -0800
@@ -16,8 +16,22 @@
sk->s_local.nl_pid = 0;
sk->s_seq_next = 0;
- int ret = nl_send_simple(sk, RTM_GETLINK, 0, NULL, 0);
+ struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
+ struct nl_msg *msg = nlmsg_alloc_simple(RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP);
+ nlmsg_append(msg, &rt_hdr, sizeof(rt_hdr), NLMSG_ALIGNTO);
+ nl_complete_msg(sk, msg);
+ struct nlmsghdr *nlh = nlmsg_hdr(msg);
+ printf("%d == nlh->nlmsg_len\n", nlh->nlmsg_len);
+ printf("%d == nlh->nlmsg_type\n", nlh->nlmsg_type);
+ printf("%d == nlh->nlmsg_flags\n", nlh->nlmsg_flags);
+ printf("%d == nlh->nlmsg_seq\n", nlh->nlmsg_seq);
+ printf("%d == nlh->nlmsg_pid\n", nlh->nlmsg_pid);
+
+ struct iovec iov = { .iov_base = nlh, .iov_len = nlh->nlmsg_len };
+ struct msghdr hdr = { .msg_iov = &iov, .msg_iovlen = 1, };
+
+ int ret = nl_sendmsg(sk, msg, &hdr);
printf("Bytes: %d\n", ret);
return 0;
}
"""
monkeypatch.setattr(libnl.socket_, 'generate_local_port', lambda: 0)
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
msg = nlmsg_alloc_simple(RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP)
nlmsg_append(msg, rt_hdr, rt_hdr.SIZEOF, NLMSG_ALIGNTO)
nl_complete_msg(sk, msg)
nlh = nlmsg_hdr(msg)
assert 20 == nlh.nlmsg_len
assert 18 == nlh.nlmsg_type
assert 773 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
assert b'FAAAABIABQMAAAAAAAAAABEAAAA=' == base64.b64encode(
buffer(nlh.bytearray[:nlh.nlmsg_len]))
nl_socket_free(sk)
def test_dissect(monkeypatch):
r"""Diff of C code (from test_bare()) to test against.
--- test_bare.c 2015-02-08 12:43:15.543135855 -0800
+++ test_dissect.c 2015-02-08 13:25:31.375715668 -0800
@@ -16,8 +16,22 @@
sk->s_local.nl_pid = 0;
sk->s_seq_next = 0;
- int ret = nl_send_simple(sk, RTM_GETLINK, 0, NULL, 0);
+ struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
+ struct nl_msg *msg = nlmsg_alloc_simple(RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP);
+ nlmsg_append(msg, &rt_hdr, sizeof(rt_hdr), NLMSG_ALIGNTO);
+ nl_complete_msg(sk, msg);
+ struct nlmsghdr *nlh = nlmsg_hdr(msg);
+ printf("%d == nlh->nlmsg_len\n", nlh->nlmsg_len);
+ printf("%d == nlh->nlmsg_type\n", nlh->nlmsg_type);
+ printf("%d == nlh->nlmsg_flags\n", nlh->nlmsg_flags);
+ printf("%d == nlh->nlmsg_seq\n", nlh->nlmsg_seq);
+ printf("%d == nlh->nlmsg_pid\n", nlh->nlmsg_pid);
+
+ struct iovec iov = { .iov_base = nlh, .iov_len = nlh->nlmsg_len };
+ struct msghdr hdr = { .msg_iov = &iov, .msg_iovlen = 1, };
+
+ int ret = nl_sendmsg(sk, msg, &hdr);
printf("Bytes: %d\n", ret);
return 0;
}
"""
monkeypatch.setattr(libnl.socket_, 'generate_local_port', lambda: 0)
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
msg = nlmsg_alloc_simple(RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP)
nlmsg_append(msg, rt_hdr, rt_hdr.SIZEOF, NLMSG_ALIGNTO)
nl_complete_msg(sk, msg)
nlh = nlmsg_hdr(msg)
assert 20 == nlh.nlmsg_len
assert 18 == nlh.nlmsg_type
assert 773 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
assert b'FAAAABIABQMAAAAAAAAAABEAAAA=' == base64.b64encode(buffer(nlh.bytearray[:nlh.nlmsg_len]))
nl_socket_free(sk)
def test_error_nle_bad_sock():
"""Test for NLE_BAD_SOCK."""
sk = nl_socket_alloc()
msg = nlmsg_alloc_simple(0, 0)
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
nl_complete_msg(sk, msg)
message = 'Hello World!\n'
iov = bytes(message.encode('ascii'))
hdr = msghdr(msg_iov=iov)
assert -3 == nl_sendmsg(sk, msg, hdr)
def test_default(tcp_server):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0)
// (nc -l 2000 &); sleep 0.1; ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
struct sockaddr_in sin = { .sin_port = htons(2000), .sin_family = AF_INET, };
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
int fd = socket(AF_INET, SOCK_STREAM, 0);
connect(fd, (struct sockaddr *) &sin, sizeof(sin));
struct nl_sock *sk = nl_socket_alloc();
struct nl_msg *msg = nlmsg_alloc_simple(0, 0);
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
sk->s_local.nl_pid = 0;
nl_complete_msg(sk, msg);
char message[] = "Hello World!\n";
struct iovec iov = { .iov_base = message, .iov_len = sizeof(message), };
struct msghdr hdr = { .msg_iov = &iov, .msg_iovlen = 1, };
int ret = nl_sendmsg(sk, msg, &hdr);
printf("Bytes: %d\n", ret); // 14
return 0;
}
// Expected bash output:
// Hello World!
// Bytes: 14
"""
sk = nl_socket_alloc()
msg = nlmsg_alloc_simple(0, 0)
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 0
nl_complete_msg(sk, msg)
message = 'Hello World!\n\0'
iov = bytes(message.encode('ascii'))
hdr = msghdr(msg_iov=iov)
assert 14 == nl_sendmsg(sk, msg, hdr)
assert [iov] == tcp_server.data
nl_socket_free(sk)
def main():
"""Main function called upon script execution."""
# First open a socket to the kernel. Same one used for sending and receiving.
sk = nl_socket_alloc() # Creates an `nl_sock` instance.
ret = nl_connect(sk,
NETLINK_ROUTE) # Create file descriptor and bind socket.
if ret < 0:
reason = errmsg[abs(ret)]
return error('nl_connect() returned {0} ({1})'.format(ret, reason))
# Next we send the request to the kernel.
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, rt_hdr,
rt_hdr.SIZEOF)
if ret < 0:
reason = errmsg[abs(ret)]
return error('nl_send_simple() returned {0} ({1})'.format(ret, reason))
print('Sent {0} bytes to the kernel.'.format(ret))
# Finally we'll retrieve the kernel's answer, process it, and call any callbacks attached to the `nl_sock` instance.
nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback,
None) # Add callback to the `nl_sock` instance.
ret = nl_recvmsgs_default(
sk) # Get kernel's answer, and call attached callbacks.
if ret < 0:
reason = errmsg[abs(ret)]
return error('nl_recvmsgs_default() returned {0} ({1})'.format(
ret, reason))
def test_bare(tcp_server, monkeypatch):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && (nc -l 2000 |base64 &) && sleep 0.1 && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
struct sockaddr_in sin = { .sin_port = htons(2000), .sin_family = AF_INET, };
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
int fd = socket(AF_INET, SOCK_STREAM, 0);
connect(fd, (struct sockaddr *) &sin, sizeof(sin));
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
sk->s_local.nl_pid = 0;
sk->s_seq_next = 0;
int ret = nl_send_simple(sk, RTM_GETLINK, 0, NULL, 0);
printf("Bytes: %d\n", ret);
return 0;
}
// Expected bash output:
// Bytes: 16
// EAAAABIABQAAAAAAAAAAAA==
"""
monkeypatch.setattr(libnl.socket_, 'generate_local_port', lambda: 0)
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
assert 16 == nl_send_simple(sk, RTM_GETLINK, 0, None)
assert 1 == len(tcp_server.data)
assert b'EAAAABIABQAAAAAAAAAAAA==' == base64.b64encode(
buffer(tcp_server.data[0]))
nl_socket_free(sk)
def test_bare(tcp_server, monkeypatch):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && (nc -l 2000 |base64 &) && sleep 0.1 && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
struct sockaddr_in sin = { .sin_port = htons(2000), .sin_family = AF_INET, };
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
int fd = socket(AF_INET, SOCK_STREAM, 0);
connect(fd, (struct sockaddr *) &sin, sizeof(sin));
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
sk->s_local.nl_pid = 0;
sk->s_seq_next = 0;
int ret = nl_send_simple(sk, RTM_GETLINK, 0, NULL, 0);
printf("Bytes: %d\n", ret);
return 0;
}
// Expected bash output:
// Bytes: 16
// EAAAABIABQAAAAAAAAAAAA==
"""
monkeypatch.setattr(libnl.socket_, 'generate_local_port', lambda: 0)
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
assert 16 == nl_send_simple(sk, RTM_GETLINK, 0, None)
assert 1 == len(tcp_server.data)
assert b'EAAAABIABQAAAAAAAAAAAA==' == base64.b64encode(buffer(tcp_server.data[0]))
nl_socket_free(sk)
def test_seq():
"""Test sequences."""
msg = nlmsg_alloc()
nlh = nlmsg_hdr(msg)
assert 16 == nlh.nlmsg_len
assert 0 == nlh.nlmsg_type
assert 0 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.s_local.nl_pid = 0
nl_complete_msg(sk, msg)
assert 1423351063 <= nlh.nlmsg_seq
nlh.nlmsg_seq = 1423350947
assert 1423350947 == nlh.nlmsg_seq
nlh.nlmsg_pid = 0 # sk.s_local.nl_pid is read-only in Python.
assert b'EAAAAAAABQCjnNZUAAAAAA==' == base64.b64encode(buffer(nlh.bytearray[:nlh.nlmsg_len]))
def test_seq():
"""Test sequences."""
msg = nlmsg_alloc()
nlh = nlmsg_hdr(msg)
assert 16 == nlh.nlmsg_len
assert 0 == nlh.nlmsg_type
assert 0 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.s_local.nl_pid = 0
nl_complete_msg(sk, msg)
assert 1423351063 <= nlh.nlmsg_seq
nlh.nlmsg_seq = 1423350947
assert 1423350947 == nlh.nlmsg_seq
nlh.nlmsg_pid = 0 # sk.s_local.nl_pid is read-only in Python.
assert b'EAAAAAAABQCjnNZUAAAAAA==' == base64.b64encode(
buffer(nlh.bytearray[:nlh.nlmsg_len]))
def test_full(tcp_server, monkeypatch):
r"""Diff of C code (from test_bare()) to test against.
--- test_bare.c 2015-02-08 12:43:15.543135855 -0800
+++ test_full.c 2015-02-08 12:43:08.533183752 -0800
@@ -13,10 +13,11 @@
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
- sk->s_local.nl_pid = 0;
- sk->s_seq_next = 0;
+ sk->s_local.nl_pid = 1234;
+ sk->s_seq_next = 10;
- int ret = nl_send_simple(sk, RTM_GETLINK, 0, NULL, 0);
+ struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
+ int ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, &rt_hdr, sizeof(rt_hdr));
printf("Bytes: %d\n", ret);
return 0;
"""
monkeypatch.setattr(libnl.socket_, 'generate_local_port', lambda: 1234)
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 1234
sk.s_seq_next = 10
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP,
rt_hdr, rt_hdr.SIZEOF)
assert 1 == len(tcp_server.data)
assert b'FAAAABIABQMKAAAA0gQAABEAAAA=' == base64.b64encode(
buffer(tcp_server.data[0]))
nl_socket_free(sk)
def test_two_attrs():
"""Test with two attributes."""
msg = nlmsg_alloc()
assert 0 == nla_put_u32(msg, 4, 8)
nlh = nlmsg_hdr(msg)
assert 24 == nlh.nlmsg_len
assert 0 == nlh.nlmsg_type
assert 0 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
assert 0 == nla_put_u64(msg, 5, 17)
assert 36 == nlh.nlmsg_len
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
nl_complete_msg(sk, msg)
assert 0 == nlh.nlmsg_seq
nlh.nlmsg_pid = 0 # sk.s_local.nl_pid is read-only in Python.
expected = b'JAAAAAAABQAAAAAAAAAAAAgABAAIAAAADAAFABEAAAAAAAAA'
assert expected == base64.b64encode(buffer(nlh.bytearray[:nlh.nlmsg_len]))
def test_two_attrs():
"""Test with two attributes."""
msg = nlmsg_alloc()
assert 0 == nla_put_u32(msg, 4, 8)
nlh = nlmsg_hdr(msg)
assert 24 == nlh.nlmsg_len
assert 0 == nlh.nlmsg_type
assert 0 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
assert 0 == nla_put_u64(msg, 5, 17)
assert 36 == nlh.nlmsg_len
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
nl_complete_msg(sk, msg)
assert 0 == nlh.nlmsg_seq
nlh.nlmsg_pid = 0 # sk.s_local.nl_pid is read-only in Python.
expected = b'JAAAAAAABQAAAAAAAAAAAAgABAAIAAAADAAFABEAAAAAAAAA'
assert expected == base64.b64encode(buffer(nlh.bytearray[:nlh.nlmsg_len]))
def test_full(tcp_server, monkeypatch):
r"""Diff of C code (from test_bare()) to test against.
--- test_bare.c 2015-02-08 12:43:15.543135855 -0800
+++ test_full.c 2015-02-08 12:43:08.533183752 -0800
@@ -13,10 +13,11 @@
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
- sk->s_local.nl_pid = 0;
- sk->s_seq_next = 0;
+ sk->s_local.nl_pid = 1234;
+ sk->s_seq_next = 10;
- int ret = nl_send_simple(sk, RTM_GETLINK, 0, NULL, 0);
+ struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
+ int ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, &rt_hdr, sizeof(rt_hdr));
printf("Bytes: %d\n", ret);
return 0;
"""
monkeypatch.setattr(libnl.socket_, 'generate_local_port', lambda: 1234)
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 1234
sk.s_seq_next = 10
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, rt_hdr, rt_hdr.SIZEOF)
assert 1 == len(tcp_server.data)
assert b'FAAAABIABQMKAAAA0gQAABEAAAA=' == base64.b64encode(buffer(tcp_server.data[0]))
nl_socket_free(sk)
def genl_connect(sk):
"""Connect a Generic Netlink socket.
https://github.com/thom311/libnl/blob/libnl3_2_25/lib/genl/genl.c#L45
This function expects a nl_socket class instance previously allocated via nl_socket_alloc(). It calls nl_connect()
to create the local socket file descriptor and binds the socket to the NETLINK_GENERIC Netlink protocol.
Using this function is equivalent to:
nl_connect(sk, NETLINK_GENERIC)
Positional arguments:
sk -- unconnected Netlink socket (nl_sock class instance).
Returns:
0 on success or a negative error code.
"""
return int(nl_connect(sk, NETLINK_GENERIC))
def genl_connect(sk):
"""Connect a Generic Netlink socket.
https://github.com/thom311/libnl/blob/libnl3_2_25/lib/genl/genl.c#L45
This function expects a nl_socket class instance previously allocated via nl_socket_alloc(). It calls nl_connect()
to create the local socket file descriptor and binds the socket to the NETLINK_GENERIC Netlink protocol.
Using this function is equivalent to:
nl_connect(sk, NETLINK_GENERIC)
Positional arguments:
sk -- unconnected Netlink socket (nl_sock class instance).
Returns:
0 on success or a negative error code.
"""
return int(nl_connect(sk, NETLINK_GENERIC))
def main():
"""Main function called upon script execution."""
# First open a socket to the kernel. Same one used for sending and receiving.
sk = nl_socket_alloc() # Creates an `nl_sock` instance.
ret = nl_connect(sk, NETLINK_ROUTE) # Create file descriptor and bind socket.
if ret < 0:
reason = errmsg[abs(ret)]
return error('nl_connect() returned {0} ({1})'.format(ret, reason))
# Next we send the request to the kernel.
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, rt_hdr, rt_hdr.SIZEOF)
if ret < 0:
reason = errmsg[abs(ret)]
return error('nl_send_simple() returned {0} ({1})'.format(ret, reason))
print('Sent {0} bytes to the kernel.'.format(ret))
# Finally we'll retrieve the kernel's answer, process it, and call any callbacks attached to the `nl_sock` instance.
nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback, None) # Add callback to the `nl_sock` instance.
ret = nl_recvmsgs_default(sk) # Get kernel's answer, and call attached callbacks.
if ret < 0:
reason = errmsg[abs(ret)]
return error('nl_recvmsgs_default() returned {0} ({1})'.format(ret, reason))
def test_socket(tcp_server):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && (nc -l 2000 |base64 &) && sleep 0.1 && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
struct nl_msg *msg = nlmsg_alloc();
printf("%d == nla_put_u32()\n", nla_put_u32(msg, 4, 8));
struct nlmsghdr *nlh = nlmsg_hdr(msg);
struct nlattr *attr = nlmsg_find_attr(nlh, 0, 4);
printf("%d == nla_type(attr)\n", nla_type(attr));
printf("%d == nla_get_u32(attr)\n", nla_get_u32(attr));
printf("%d == attr->nla_len\n", attr->nla_len);
struct sockaddr_in sin = { .sin_port = htons(2000), .sin_family = AF_INET, };
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
int fd = socket(AF_INET, SOCK_STREAM, 0);
connect(fd, (struct sockaddr *) &sin, sizeof(sin));
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
sk->s_local.nl_pid = 0;
nl_complete_msg(sk, msg);
struct iovec iov = { .iov_base = attr, .iov_len = attr->nla_len };
struct msghdr hdr = { .msg_iov = &iov, .msg_iovlen = 1, };
int ret = nl_sendmsg(sk, msg, &hdr);
printf("Bytes: %d\n", ret); // 14
return 0;
}
// Expected bash output:
// 0 == nla_put_u32()
// 4 == nla_type(attr)
// 8 == nla_get_u32(attr)
// 8 == attr->nla_len
// Bytes: 8
// CAAEAAgAAAA=
"""
msg = nlmsg_alloc()
assert 0 == libnl.attr.nla_put_u32(msg, 4, 8)
nlh = nlmsg_hdr(msg)
attr = nlmsg_find_attr(nlh, 0, 4)
assert 4 == libnl.attr.nla_type(attr)
assert 8 == libnl.attr.nla_get_u32(attr)
assert 8 == attr.nla_len
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 0
nl_complete_msg(sk, msg)
iov = attr.bytearray[:attr.nla_len]
hdr = msghdr(msg_iov=iov)
assert 8 == nl_sendmsg(sk, msg, hdr)
assert 1 == len(tcp_server.data)
assert b'CAAEAAgAAAA=' == base64.b64encode(buffer(iov))
assert b'CAAEAAgAAAA=' == base64.b64encode(buffer(tcp_server.data[0]))
nl_socket_free(sk)
def test_nl_socket_modify_cb(log, ifaces):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 ./a.out
#include <netlink/msg.h>
static int callback(struct nl_msg *msg, void *arg) {
printf("Got something.\n");
nl_msg_dump(msg, stdout);
return NL_OK;
}
int main() {
// Send data to the kernel.
struct nl_sock *sk = nl_socket_alloc();
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
int ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, &rt_hdr, sizeof(rt_hdr));
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback, NULL);
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x1b840b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x1b840b8: Allocated new simple message
// nlmsg_reserve: msg 0x1b840b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0x1b840b8: Appended 1 bytes with padding 4
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0x1b840b8, 0 remaining
// nlmsg_free: msg 0x1b840b8: Freed
// Bytes Sent: 20
// recvmsgs: Attempting to read from 0x1b84080
// recvmsgs: recvmsgs(0x1b84080): Read 3364 bytes
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=1116
// Got something.
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1116
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424133909
// .port = 6192
// [PAYLOAD] 1100 octets
// 00 00 04 03 01 00 00 00 49 00 01 00 00 00 00 00 ........I.......
// 07 00 03 00 6c 6f 00 00 08 00 0d 00 00 00 00 00 ....lo..........
// <trimmed>
// 00 00 00 00 00 00 00 00 00 00 00 00 ............
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=1124
// Got something.
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424133909
// .port = 6192
// [PAYLOAD] 1108 octets
// 00 00 01 00 02 00 00 00 43 10 01 00 00 00 00 00 ........C.......
// 09 00 03 00 65 74 68 30 00 00 00 00 08 00 0d 00 ....eth0........
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=1124
// Got something.
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424133909
// .port = 6192
// [PAYLOAD] 1108 octets
// 00 00 01 00 04 00 00 00 03 10 00 00 00 00 00 00 ................
// 0a 00 03 00 77 6c 61 6e 30 00 00 00 08 00 0d 00 ....wlan0.......
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// recvmsgs: Attempting to read from 0x1b84080
// recvmsgs: recvmsgs(0x1b84080): Read 20 bytes
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=20
// recvmsgs: recvmsgs(0x1b84080): Increased expected sequence number to 1424133910
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
got_something = list()
def callback(msg, arg):
got_something.append(arg)
nl_msg_dump(msg)
return NL_OK
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, rt_hdr, rt_hdr.SIZEOF)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096', log, True)
assert match('nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message', log, True)
assert match('nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20', log, True)
assert match('nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log, True)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback, 95)
assert 0 == nl_recvmsgs_default(sk)
assert [95] * len(ifaces) == got_something
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log, True)
for _ in ifaces:
if 'Attempting to read' in log[0]:
# Lots of network interfaces on this host.
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=\d{3,}', log, True)
assert match('nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------', log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = \d{3,}', log, True)
assert match('print_hdr: .type = 16 <0x10>', log)
assert match('print_hdr: .flags = 2 <MULTI>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: \[PAYLOAD\] \d{3,} octets', log, True)
rem = log.index('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------')
assert 20 < rem # At least check that there were a lot of log statements skipped.
log = log[rem:]
assert match('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------', log)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 20 bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=20', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}', log, True)
nl_socket_free(sk)
assert not log
def test_nl_socket_modify_cb_error_verbose(log):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 NLCB=verbose ./a.out
#include <netlink/msg.h>
static int callback(struct sockaddr_nl *nla, struct nlmsgerr *err, void *arg) {
int *ret = arg;
*ret = err->error;
printf("Got something.\n");
return NL_STOP;
}
int main() {
// Send data to the kernel.
struct nl_sock *sk = nl_socket_alloc();
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
int ret = nl_send_simple(sk, RTM_GETLINK, 0, &rt_hdr, sizeof(rt_hdr));
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
int err = 0;
nl_socket_modify_err_cb(sk, NL_CB_CUSTOM, callback, &err);
printf("%d == err\n", err);
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
printf("%d == err\n", err);
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x124e0b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x124e0b8: Allocated new simple message
// nlmsg_reserve: msg 0x124e0b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0x124e0b8: Appended 1 bytes with padding 4
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0x124e0b8, 0 remaining
// nlmsg_free: msg 0x124e0b8: Freed
// Bytes Sent: 20
// 0 == err
// recvmsgs: Attempting to read from 0x124e080
// recvmsgs: recvmsgs(0x124e080): Read 40 bytes
// recvmsgs: recvmsgs(0x124e080): Processing valid message...
// __nlmsg_alloc: msg 0x12520c0: Allocated new message, maxlen=40
// recvmsgs: recvmsgs(0x124e080): Increased expected sequence number to 1424136270
// Got something.
// nlmsg_free: Returned message reference 0x12520c0, 0 remaining
// nlmsg_free: msg 0x12520c0: Freed
// -7 == nl_recvmsgs_default(sk)
// -22 == err
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
got_something = list()
def callback(_, err, arg):
arg.append(err.error)
return NL_STOP
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, 0, rt_hdr, rt_hdr.SIZEOF)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096', log, True)
assert match('nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message', log, True)
assert match('nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20', log, True)
assert match('nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log, True)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_socket_modify_err_cb(sk, NL_CB_CUSTOM, callback, got_something)
assert -7 == nl_recvmsgs_default(sk)
assert [-22] == got_something
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 40 bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=40', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}', log, True)
nl_socket_free(sk)
assert not log
def test_error(log):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 NLCB=debug ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
// Send data to the kernel.
printf("Begin main()\n");
struct nl_sock *sk = nl_socket_alloc();
printf("Allocated socket.\n");
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
int ret = nl_send_simple(sk, 0, NLM_F_REQUEST, NULL, 0);
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// Begin main()
// Allocated socket.
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x3df0b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x3df0b8: Allocated new simple message
// -- Debug: Sent Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 16
// .type = 0 <0x0>
// .flags = 5 <REQUEST,ACK>
// .seq = 1423967746
// .port = 29930
// --------------------------- END NETLINK MESSAGE ---------------------------
// nl_sendmsg: sent 16 bytes
// nlmsg_free: Returned message reference 0x3df0b8, 0 remaining
// nlmsg_free: msg 0x3df0b8: Freed
// Bytes Sent: 16
// recvmsgs: Attempting to read from 0x3df080
// recvmsgs: recvmsgs(0x3df080): Read 36 bytes
// recvmsgs: recvmsgs(0x3df080): Processing valid message...
// __nlmsg_alloc: msg 0x3e30c0: Allocated new message, maxlen=36
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 36
// .type = 2 <ERROR>
// .flags = 0 <>
// .seq = 1423967746
// .port = 29930
// [ERRORMSG] 20 octets
// .error = 0 "Success"
// [ORIGINAL MESSAGE] 16 octets
// __nlmsg_alloc: msg 0x3e3128: Allocated new message, maxlen=4096
// .nlmsg_len = 16
// .type = 0 <0x0>
// .flags = 5 <REQUEST,ACK>
// .seq = 1423967746
// .port = 29930
// nlmsg_free: Returned message reference 0x3e3128, 0 remaining
// nlmsg_free: msg 0x3e3128: Freed
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x3df080): Increased expected sequence number to 1423967746
// -- Debug: ACK: type=ERROR length=36 flags=<> sequence-nr=1423967746 pid=29930
// nlmsg_free: Returned message reference 0x3e30c0, 0 remaining
// nlmsg_free: msg 0x3e30c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
assert 16 == nl_send_simple(sk, 0, NLM_F_REQUEST, None)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096',
log, True)
assert match(
'nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message',
log, True)
assert match('nl_msg_out_handler_debug: -- Debug: Sent Message:', log)
assert match(
'nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------',
log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 16', log)
assert match('print_hdr: .type = 0 <0x0>', log)
assert match('print_hdr: .flags = 5 <REQUEST,ACK>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------',
log)
assert match('nl_sendmsg: sent 16 bytes', log)
assert not log
assert 0 == nl_recvmsgs_default(sk)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 36 bytes', log, True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log,
True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=36', log,
True)
assert match('nl_msg_in_handler_debug: -- Debug: Received Message:', log)
assert match(
'nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------',
log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 36', log)
assert match('print_hdr: .type = 2 <ERROR>', log)
assert match('print_hdr: .flags = 0 <>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('dump_error_msg: [ERRORMSG] 20 octets', log)
assert match('dump_error_msg: .error = 0 "Success"', log)
assert match('dump_error_msg: [ORIGINAL MESSAGE] 16 octets', log)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096',
log, True)
assert match('print_hdr: .nlmsg_len = 16', log)
assert match('print_hdr: .type = 0 <0x0>', log)
assert match('print_hdr: .flags = 5 <REQUEST,ACK>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------',
log)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}',
log, True)
assert match(
'nl_ack_handler_debug: -- Debug: ACK: type=ERROR length=36 flags=<> sequence-nr=\d{10,} pid=\d{3,}',
log, True)
nl_socket_free(sk)
assert not log
def test_multipart_verbose(log, ifaces):
"""Expected output (trimmed).
// nl_cache_mngt_register: Registered cache operations genl/family
// Begin main()
// Allocated socket.
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0xa180b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0xa180b8: Allocated new simple message
// nlmsg_reserve: msg 0xa180b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0xa180b8: Appended 1 bytes with padding 4
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0xa180b8, 0 remaining
// nlmsg_free: msg 0xa180b8: Freed
// Bytes Sent: 20
// recvmsgs: Attempting to read from 0xa18080
// recvmsgs: recvmsgs(0xa18080): Read 3364 bytes
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=1116
// -- Warning: unhandled valid message: type=0x10 length=1116 flags=<MULTI> sequence-nr=1424132449 pid=5810
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=1124
// -- Warning: unhandled valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424132449 pid=5810
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=1124
// -- Warning: unhandled valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424132449 pid=5810
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// recvmsgs: Attempting to read from 0xa18080
// recvmsgs: recvmsgs(0xa18080): Read 20 bytes
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=20
// recvmsgs: recvmsgs(0xa18080): Increased expected sequence number to 1424132450
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP,
rt_hdr, rt_hdr.SIZEOF)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096',
log, True)
assert match(
'nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message',
log, True)
assert match(
'nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20',
log, True)
assert match(
'nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log,
True)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_recvmsgs_default(sk)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log,
True)
for _ in ifaces:
if 'Attempting to read' in log[0]:
# Lots of network interfaces on this host.
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...',
log, True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=\d{3,}',
log, True)
assert match(
'nl_valid_handler_verbose: -- Warning: unhandled valid message: type=0x10 length=\d{3,} '
'flags=<MULTI> sequence-nr=\d{10,} pid=\d{3,}', log, True)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 20 bytes', log, True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log,
True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=20', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}',
log, True)
nl_socket_free(sk)
assert not log
def test_nl_connect():
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && ./a.out
#include <netlink/msg.h>
#include <dirent.h>
struct nl_cb {
nl_recvmsg_msg_cb_t cb_set[NL_CB_TYPE_MAX+1]; void * cb_args[NL_CB_TYPE_MAX+1]; nl_recvmsg_err_cb_t cb_err;
void * cb_err_arg; int (*cb_recvmsgs_ow)(struct nl_sock *, struct nl_cb *);
int (*cb_recv_ow)(struct nl_sock *, struct sockaddr_nl *, unsigned char **, struct ucred **);
int (*cb_send_ow)(struct nl_sock *, struct nl_msg *); int cb_refcnt; enum nl_cb_type cb_active;
};
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
void print_fd_count() {
int fd_count = 0; char buf[64]; struct dirent *dp;
snprintf(buf, 256, "/proc/self/fd/");
DIR *dir = opendir(buf); while ((dp = readdir(dir)) != NULL) fd_count++; closedir(dir);
printf("fd_count: %d\n", fd_count);
}
void print(struct nl_sock *sk) {
printf("sk.s_local.nl_family = %d\n", sk->s_local.nl_family);
printf("sk.s_local.nl_pid = %d # changes every process, remains same throughout proc.\n", sk->s_local.nl_pid);
printf("sk.s_local.nl_groups = %d\n", sk->s_local.nl_groups);
printf("sk.s_peer.nl_family = %d\n", sk->s_peer.nl_family);
printf("sk.s_peer.nl_pid = %d\n", sk->s_peer.nl_pid);
printf("sk.s_peer.nl_groups = %d\n", sk->s_peer.nl_groups);
printf("sk.s_fd = %d\n", sk->s_fd);
printf("sk.s_proto = %d\n", sk->s_proto);
printf("sk.s_flags = %d\n", sk->s_flags);
printf("sk.s_cb.cb_active = %d\n", sk->s_cb->cb_active);
printf("addr: sk.s_cb.cb_err = %p\n", sk->s_cb->cb_err);
}
int main() {
struct nl_sock *sk = nl_socket_alloc();
print_fd_count();
print(sk);
printf("\n");
printf("nl_connect(): %d\n", nl_connect(sk, NETLINK_ROUTE));
print_fd_count();
print(sk);
nl_socket_free(sk);
print_fd_count();
print(sk);
printf("\n");
sk = nl_socket_alloc();
printf("nl_connect(): %d\n", nl_connect(sk, NETLINK_GENERIC));
print_fd_count();
print(sk);
nl_socket_free(sk);
print_fd_count();
printf("\n");
return 0;
}
// Expected output:
// fd_count: 6
// sk.s_local.nl_family = 16
// sk.s_local.nl_pid = 3121 # changes every process, remains same throughout proc.
// sk.s_local.nl_groups = 0
// sk.s_peer.nl_family = 16
// sk.s_peer.nl_pid = 0
// sk.s_peer.nl_groups = 0
// sk.s_fd = -1
// sk.s_proto = 0
// sk.s_flags = 0
// sk.s_cb.cb_active = 11
// addr: sk.s_cb.cb_err = (nil)
//
// nl_connect(): 0
// fd_count: 7
// sk.s_local.nl_family = 16
// sk.s_local.nl_pid = 3121 # changes every process, remains same throughout proc.
// sk.s_local.nl_groups = 0
// sk.s_peer.nl_family = 16
// sk.s_peer.nl_pid = 0
// sk.s_peer.nl_groups = 0
// sk.s_fd = 3
// sk.s_proto = 0
// sk.s_flags = 1
// sk.s_cb.cb_active = 11
// addr: sk.s_cb.cb_err = (nil)
// fd_count: 6
// sk.s_local.nl_family = 16
// sk.s_local.nl_pid = 167775232 # changes every process, remains same throughout proc.
// sk.s_local.nl_groups = 0
// sk.s_peer.nl_family = 16
// sk.s_peer.nl_pid = 0
// sk.s_peer.nl_groups = 0
// sk.s_fd = 3
// sk.s_proto = 0
// sk.s_flags = 1
// sk.s_cb.cb_active = 18774
// addr: sk.s_cb.cb_err = 0x310a0010
//
// nl_connect(): 0
// fd_count: 7
// sk.s_local.nl_family = 16
// sk.s_local.nl_pid = 3121 # changes every process, remains same throughout proc.
// sk.s_local.nl_groups = 0
// sk.s_peer.nl_family = 16
// sk.s_peer.nl_pid = 0
// sk.s_peer.nl_groups = 0
// sk.s_fd = 3
// sk.s_proto = 16
// sk.s_flags = 1
// sk.s_cb.cb_active = 11
// addr: sk.s_cb.cb_err = (nil)
// fd_count: 6
"""
initial_fd_count = len(os.listdir('/proc/self/fd'))
assert 2 <= initial_fd_count
# Allocate but don't connect/bind.
sk = nl_socket_alloc()
assert initial_fd_count == len(os.listdir('/proc/self/fd'))
assert 16 == sk.s_local.nl_family
assert 0 < sk.s_local.nl_pid
assert 0 == sk.s_local.nl_groups
assert 16 == sk.s_peer.nl_family
assert 0 == sk.s_peer.nl_pid
assert 0 == sk.s_peer.nl_groups
assert -1 == sk.s_fd
assert 0 == sk.s_proto
assert 0 == sk.s_flags
assert 11 == sk.s_cb.cb_active
assert sk.s_cb.cb_err is None
persistent_pid = int(sk.s_local.nl_pid)
# Connect, then close the socket at the end.
assert 0 == nl_connect(sk, NETLINK_ROUTE)
assert initial_fd_count + 1 == len(os.listdir('/proc/self/fd'))
assert 16 == sk.s_local.nl_family
assert persistent_pid == sk.s_local.nl_pid
assert 0 == sk.s_local.nl_groups
assert 16 == sk.s_peer.nl_family
assert 0 == sk.s_peer.nl_pid
assert 0 == sk.s_peer.nl_groups
assert 0 < sk.s_fd
assert 0 == sk.s_proto
assert 1 == sk.s_flags
assert 11 == sk.s_cb.cb_active
assert sk.s_cb.cb_err is None
persistent_fd = int(sk.s_fd)
nl_socket_free(sk)
assert initial_fd_count == len(os.listdir('/proc/self/fd'))
assert 16 == sk.s_local.nl_family
# assert persistent_pid == sk.s_local.nl_pid # In C, pointer points to deallocated memory. In Python, leave alone.
assert 0 == sk.s_local.nl_groups
assert 16 == sk.s_peer.nl_family
assert 0 == sk.s_peer.nl_pid
assert 0 == sk.s_peer.nl_groups
# assert persistent_fd == sk.s_fd # In C, s_fd is a regular int. In Python, it's a class property.
assert 0 == sk.s_proto
assert 1 == sk.s_flags
# assert 11 == sk.s_cb.cb_active # In C, pointer points to deallocated memory. In Python, leave alone.
# assert sk.s_cb.cb_err is None # In C, pointer points to deallocated memory. In Python, leave alone.
# Re-allocate and connect again, pid should be the same as the previous session.
sk = nl_socket_alloc()
assert 0 == nl_connect(sk, NETLINK_GENERIC)
assert initial_fd_count + 1 == len(os.listdir('/proc/self/fd'))
assert 16 == sk.s_local.nl_family
assert persistent_pid == sk.s_local.nl_pid
assert 0 == sk.s_local.nl_groups
assert 16 == sk.s_peer.nl_family
assert 0 == sk.s_peer.nl_pid
assert 0 == sk.s_peer.nl_groups
assert persistent_fd == sk.s_fd
assert 16 == sk.s_proto
assert 1 == sk.s_flags
assert 11 == sk.s_cb.cb_active
assert sk.s_cb.cb_err is None
nl_socket_free(sk)
assert initial_fd_count == len(os.listdir('/proc/self/fd'))
def test_multipart(log, ifaces):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 NLCB=debug ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
// Send data to the kernel.
printf("Begin main()\n");
struct nl_sock *sk = nl_socket_alloc();
printf("Allocated socket.\n");
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
int ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, &rt_hdr, sizeof(rt_hdr));
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// Begin main()
// Allocated socket.
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x1bbe0b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x1bbe0b8: Allocated new simple message
// nlmsg_reserve: msg 0x1bbe0b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0x1bbe0b8: Appended 1 bytes with padding 4
// -- Debug: Sent Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 20
// .type = 18 <0x12>
// .flags = 773 <REQUEST,ACK,ROOT,MATCH>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 4 octets
// 11 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0x1bbe0b8, 0 remaining
// nlmsg_free: msg 0x1bbe0b8: Freed
// Bytes Sent: 20
// recvmsgs: Attempting to read from 0x1bbe080
// recvmsgs: recvmsgs(0x1bbe080): Read 3364 bytes
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=1116
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1116
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 1100 octets
// 00 00 04 03 01 00 00 00 49 00 01 00 00 00 00 00 ........I.......
// <trimmed>
// 00 00 00 00 00 00 00 00 00 00 00 00 ............
// --------------------------- END NETLINK MESSAGE ---------------------------
// -- Debug: Unhandled Valid message: type=0x10 length=1116 flags=<MULTI> sequence-nr=1424053819 pid=18409
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=1124
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 1108 octets
// 00 00 01 00 02 00 00 00 43 10 01 00 00 00 00 00 ........C.......
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// -- Debug: Unhandled Valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424053819 pid=18409
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=1124
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 1108 octets
// 00 00 01 00 03 00 00 00 03 10 00 00 00 00 00 00 ................
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// -- Debug: Unhandled Valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424053819 pid=18409
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// recvmsgs: Attempting to read from 0x1bbe080
// recvmsgs: recvmsgs(0x1bbe080): Read 20 bytes
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=20
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 20
// .type = 3 <DONE>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 4 octets
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x1bbe080): Increased expected sequence number to 1424053820
// -- Debug: End of multipart message block: type=DONE length=20 flags=<MULTI> sequence-nr=1424053819 pid=18409
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP,
rt_hdr, rt_hdr.SIZEOF)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096',
log, True)
assert match(
'nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message',
log, True)
assert match(
'nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20',
log, True)
assert match(
'nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log,
True)
assert match('nl_msg_out_handler_debug: -- Debug: Sent Message:', log)
assert match(
'nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------',
log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 20', log)
assert match('print_hdr: .type = 18 <0x12>', log)
assert match('print_hdr: .flags = 773 <REQUEST,ACK,ROOT,MATCH>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: [PAYLOAD] 4 octets', log)
assert match(
'dump_hex: 11 00 00 00 ....',
log)
assert match(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------',
log)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_recvmsgs_default(sk)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log,
True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=\d{3,}',
log, True)
assert match('nl_msg_in_handler_debug: -- Debug: Received Message:', log)
assert match(
'nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------',
log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = \d{3,}', log, True)
assert match('print_hdr: .type = 16 <0x10>', log)
assert match('print_hdr: .flags = 2 <MULTI>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: \[PAYLOAD\] \d{3,} octets', log, True)
assert match(
'dump_hex: 00 00 04 03 01 00 00 00 49 00 01 00 00 00 00 00 ........I.......',
log)
assert match(
'dump_hex: 07 00 03 00 6c 6f 00 00 08 00 0d 00 00 00 00 00 ....lo..........',
log)
assert match(
'dump_hex: 05 00 10 00 00 00 00 00 05 00 11 00 00 00 00 00 ................',
log)
for _ in ifaces:
# Done testing this payload. Differs too much between Travis and Raspbian, and probably others.
rem = log.index(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------'
)
assert 20 < rem # At least check that there were a lot of log statements skipped.
log = log[rem:]
assert match(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------',
log)
assert match(
'nl_valid_handler_debug: -- Debug: Unhandled Valid message: type=0x10 length=\d{3,} flags=<MULTI> '
'sequence-nr=\d{10,} pid=\d{3,}', log, True)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 20 bytes', log, True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log,
True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=20', log,
True)
assert match('nl_msg_in_handler_debug: -- Debug: Received Message:', log)
assert match(
'nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------',
log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 20', log)
assert match('print_hdr: .type = 3 <DONE>', log)
assert match('print_hdr: .flags = 2 <MULTI>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: [PAYLOAD] 4 octets', log)
assert match(
'dump_hex: 00 00 00 00 ....',
log)
assert match(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------',
log)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}',
log, True)
assert match(
'nl_finish_handler_debug: -- Debug: End of multipart message block: type=DONE length=20 flags=<MULTI> '
'sequence-nr=\d{10,} pid=\d{3,}', log, True)
nl_socket_free(sk)
assert not log
def test_socket(tcp_server):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && (nc -l 2000 |base64 &) && sleep 0.1 && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
struct nl_msg *msg = nlmsg_alloc();
printf("%d == nla_put_u32()\n", nla_put_u32(msg, 4, 8));
struct nlmsghdr *nlh = nlmsg_hdr(msg);
struct nlattr *attr = nlmsg_find_attr(nlh, 0, 4);
printf("%d == nla_type(attr)\n", nla_type(attr));
printf("%d == nla_get_u32(attr)\n", nla_get_u32(attr));
printf("%d == attr->nla_len\n", attr->nla_len);
struct sockaddr_in sin = { .sin_port = htons(2000), .sin_family = AF_INET, };
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
int fd = socket(AF_INET, SOCK_STREAM, 0);
connect(fd, (struct sockaddr *) &sin, sizeof(sin));
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
sk->s_local.nl_pid = 0;
nl_complete_msg(sk, msg);
struct iovec iov = { .iov_base = attr, .iov_len = attr->nla_len };
struct msghdr hdr = { .msg_iov = &iov, .msg_iovlen = 1, };
int ret = nl_sendmsg(sk, msg, &hdr);
printf("Bytes: %d\n", ret); // 14
return 0;
}
// Expected bash output:
// 0 == nla_put_u32()
// 4 == nla_type(attr)
// 8 == nla_get_u32(attr)
// 8 == attr->nla_len
// Bytes: 8
// CAAEAAgAAAA=
"""
msg = nlmsg_alloc()
assert 0 == libnl.attr.nla_put_u32(msg, 4, 8)
nlh = nlmsg_hdr(msg)
attr = nlmsg_find_attr(nlh, 0, 4)
assert 4 == libnl.attr.nla_type(attr)
assert 8 == libnl.attr.nla_get_u32(attr)
assert 8 == attr.nla_len
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 0
nl_complete_msg(sk, msg)
iov = attr.bytearray[: attr.nla_len]
hdr = msghdr(msg_iov=iov)
assert 8 == nl_sendmsg(sk, msg, hdr)
assert 1 == len(tcp_server.data)
assert b"CAAEAAgAAAA=" == base64.b64encode(buffer(iov))
assert b"CAAEAAgAAAA=" == base64.b64encode(buffer(tcp_server.data[0]))
nl_socket_free(sk)
def test_multipart_verbose(log, ifaces):
"""Expected output (trimmed).
// nl_cache_mngt_register: Registered cache operations genl/family
// Begin main()
// Allocated socket.
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0xa180b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0xa180b8: Allocated new simple message
// nlmsg_reserve: msg 0xa180b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0xa180b8: Appended 1 bytes with padding 4
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0xa180b8, 0 remaining
// nlmsg_free: msg 0xa180b8: Freed
// Bytes Sent: 20
// recvmsgs: Attempting to read from 0xa18080
// recvmsgs: recvmsgs(0xa18080): Read 3364 bytes
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=1116
// -- Warning: unhandled valid message: type=0x10 length=1116 flags=<MULTI> sequence-nr=1424132449 pid=5810
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=1124
// -- Warning: unhandled valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424132449 pid=5810
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=1124
// -- Warning: unhandled valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424132449 pid=5810
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// recvmsgs: Attempting to read from 0xa18080
// recvmsgs: recvmsgs(0xa18080): Read 20 bytes
// recvmsgs: recvmsgs(0xa18080): Processing valid message...
// __nlmsg_alloc: msg 0xa1c0c0: Allocated new message, maxlen=20
// recvmsgs: recvmsgs(0xa18080): Increased expected sequence number to 1424132450
// nlmsg_free: Returned message reference 0xa1c0c0, 0 remaining
// nlmsg_free: msg 0xa1c0c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, rt_hdr, rt_hdr.SIZEOF)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096', log, True)
assert match('nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message', log, True)
assert match('nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20', log, True)
assert match('nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log, True)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_recvmsgs_default(sk)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log, True)
for _ in ifaces:
if 'Attempting to read' in log[0]:
# Lots of network interfaces on this host.
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=\d{3,}', log, True)
assert match('nl_valid_handler_verbose: -- Warning: unhandled valid message: type=0x10 length=\d{3,} '
'flags=<MULTI> sequence-nr=\d{10,} pid=\d{3,}', log, True)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 20 bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=20', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}', log, True)
nl_socket_free(sk)
assert not log
def test_error(log):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 NLCB=debug ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
// Send data to the kernel.
printf("Begin main()\n");
struct nl_sock *sk = nl_socket_alloc();
printf("Allocated socket.\n");
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
int ret = nl_send_simple(sk, 0, NLM_F_REQUEST, NULL, 0);
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// Begin main()
// Allocated socket.
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x3df0b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x3df0b8: Allocated new simple message
// -- Debug: Sent Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 16
// .type = 0 <0x0>
// .flags = 5 <REQUEST,ACK>
// .seq = 1423967746
// .port = 29930
// --------------------------- END NETLINK MESSAGE ---------------------------
// nl_sendmsg: sent 16 bytes
// nlmsg_free: Returned message reference 0x3df0b8, 0 remaining
// nlmsg_free: msg 0x3df0b8: Freed
// Bytes Sent: 16
// recvmsgs: Attempting to read from 0x3df080
// recvmsgs: recvmsgs(0x3df080): Read 36 bytes
// recvmsgs: recvmsgs(0x3df080): Processing valid message...
// __nlmsg_alloc: msg 0x3e30c0: Allocated new message, maxlen=36
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 36
// .type = 2 <ERROR>
// .flags = 0 <>
// .seq = 1423967746
// .port = 29930
// [ERRORMSG] 20 octets
// .error = 0 "Success"
// [ORIGINAL MESSAGE] 16 octets
// __nlmsg_alloc: msg 0x3e3128: Allocated new message, maxlen=4096
// .nlmsg_len = 16
// .type = 0 <0x0>
// .flags = 5 <REQUEST,ACK>
// .seq = 1423967746
// .port = 29930
// nlmsg_free: Returned message reference 0x3e3128, 0 remaining
// nlmsg_free: msg 0x3e3128: Freed
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x3df080): Increased expected sequence number to 1423967746
// -- Debug: ACK: type=ERROR length=36 flags=<> sequence-nr=1423967746 pid=29930
// nlmsg_free: Returned message reference 0x3e30c0, 0 remaining
// nlmsg_free: msg 0x3e30c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
assert 16 == nl_send_simple(sk, 0, NLM_F_REQUEST, None)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096', log, True)
assert match('nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message', log, True)
assert match('nl_msg_out_handler_debug: -- Debug: Sent Message:', log)
assert match('nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------', log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 16', log)
assert match('print_hdr: .type = 0 <0x0>', log)
assert match('print_hdr: .flags = 5 <REQUEST,ACK>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------', log)
assert match('nl_sendmsg: sent 16 bytes', log)
assert not log
assert 0 == nl_recvmsgs_default(sk)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 36 bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=36', log, True)
assert match('nl_msg_in_handler_debug: -- Debug: Received Message:', log)
assert match('nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------', log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 36', log)
assert match('print_hdr: .type = 2 <ERROR>', log)
assert match('print_hdr: .flags = 0 <>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('dump_error_msg: [ERRORMSG] 20 octets', log)
assert match('dump_error_msg: .error = 0 "Success"', log)
assert match('dump_error_msg: [ORIGINAL MESSAGE] 16 octets', log)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096', log, True)
assert match('print_hdr: .nlmsg_len = 16', log)
assert match('print_hdr: .type = 0 <0x0>', log)
assert match('print_hdr: .flags = 5 <REQUEST,ACK>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------', log)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}', log, True)
assert match('nl_ack_handler_debug: -- Debug: ACK: type=ERROR length=36 flags=<> sequence-nr=\d{10,} pid=\d{3,}',
log, True)
nl_socket_free(sk)
assert not log
def test_socket(tcp_server):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && (nc -l 2000 |base64 &) && sleep 0.1 && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
struct nl_msg *msg = nlmsg_alloc();
struct nlmsghdr *nlh = nlmsg_hdr(msg);
printf("%d == nlh->nlmsg_len\n", nlh->nlmsg_len);
printf("%d == nlh->nlmsg_type\n", nlh->nlmsg_type);
printf("%d == nlh->nlmsg_flags\n", nlh->nlmsg_flags);
printf("%d == nlh->nlmsg_seq\n", nlh->nlmsg_seq);
printf("%d == nlh->nlmsg_pid\n", nlh->nlmsg_pid);
struct sockaddr_in sin = { .sin_port = htons(2000), .sin_family = AF_INET, };
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
int fd = socket(AF_INET, SOCK_STREAM, 0);
connect(fd, (struct sockaddr *) &sin, sizeof(sin));
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
sk->s_local.nl_pid = 0;
sk->s_seq_next = 0;
nl_complete_msg(sk, msg);
printf("%d == nlh->nlmsg_seq\n", nlh->nlmsg_seq);
struct iovec iov = { .iov_base = nlh, .iov_len = nlh->nlmsg_len };
struct msghdr hdr = { .msg_iov = &iov, .msg_iovlen = 1, };
int ret = nl_sendmsg(sk, msg, &hdr);
printf("Bytes: %d\n", ret);
return 0;
}
// Expected bash output:
// 16 == nlh->nlmsg_len
// 0 == nlh->nlmsg_type
// 0 == nlh->nlmsg_flags
// 0 == nlh->nlmsg_seq
// 0 == nlh->nlmsg_pid
// 0 == nlh->nlmsg_seq
// Bytes: 16
// EAAAAAAABQAAAAAAAAAAAA==
"""
msg = nlmsg_alloc()
nlh = nlmsg_hdr(msg)
assert 16 == nlh.nlmsg_len
assert 0 == nlh.nlmsg_type
assert 0 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
nl_complete_msg(sk, msg)
assert 0 == nlh.nlmsg_seq
nlh.nlmsg_pid = 0 # sk.s_local.nl_pid is read-only in Python.
iov = nlh.bytearray[:nlh.nlmsg_len]
hdr = msghdr(msg_iov=iov)
assert 16 == nl_sendmsg(sk, msg, hdr)
assert 1 == len(tcp_server.data)
assert b'EAAAAAAABQAAAAAAAAAAAA==' == base64.b64encode(buffer(iov))
assert b'EAAAAAAABQAAAAAAAAAAAA==' == base64.b64encode(
buffer(tcp_server.data[0]))
nl_socket_free(sk)
def test_multipart(log, ifaces):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 NLCB=debug ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
// Send data to the kernel.
printf("Begin main()\n");
struct nl_sock *sk = nl_socket_alloc();
printf("Allocated socket.\n");
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
int ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, &rt_hdr, sizeof(rt_hdr));
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// Begin main()
// Allocated socket.
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x1bbe0b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x1bbe0b8: Allocated new simple message
// nlmsg_reserve: msg 0x1bbe0b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0x1bbe0b8: Appended 1 bytes with padding 4
// -- Debug: Sent Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 20
// .type = 18 <0x12>
// .flags = 773 <REQUEST,ACK,ROOT,MATCH>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 4 octets
// 11 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0x1bbe0b8, 0 remaining
// nlmsg_free: msg 0x1bbe0b8: Freed
// Bytes Sent: 20
// recvmsgs: Attempting to read from 0x1bbe080
// recvmsgs: recvmsgs(0x1bbe080): Read 3364 bytes
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=1116
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1116
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 1100 octets
// 00 00 04 03 01 00 00 00 49 00 01 00 00 00 00 00 ........I.......
// <trimmed>
// 00 00 00 00 00 00 00 00 00 00 00 00 ............
// --------------------------- END NETLINK MESSAGE ---------------------------
// -- Debug: Unhandled Valid message: type=0x10 length=1116 flags=<MULTI> sequence-nr=1424053819 pid=18409
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=1124
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 1108 octets
// 00 00 01 00 02 00 00 00 43 10 01 00 00 00 00 00 ........C.......
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// -- Debug: Unhandled Valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424053819 pid=18409
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=1124
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 1108 octets
// 00 00 01 00 03 00 00 00 03 10 00 00 00 00 00 00 ................
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// -- Debug: Unhandled Valid message: type=0x10 length=1124 flags=<MULTI> sequence-nr=1424053819 pid=18409
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// recvmsgs: Attempting to read from 0x1bbe080
// recvmsgs: recvmsgs(0x1bbe080): Read 20 bytes
// recvmsgs: recvmsgs(0x1bbe080): Processing valid message...
// __nlmsg_alloc: msg 0x1bc20c0: Allocated new message, maxlen=20
// -- Debug: Received Message:
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 20
// .type = 3 <DONE>
// .flags = 2 <MULTI>
// .seq = 1424053819
// .port = 18409
// [PAYLOAD] 4 octets
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x1bbe080): Increased expected sequence number to 1424053820
// -- Debug: End of multipart message block: type=DONE length=20 flags=<MULTI> sequence-nr=1424053819 pid=18409
// nlmsg_free: Returned message reference 0x1bc20c0, 0 remaining
// nlmsg_free: msg 0x1bc20c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, rt_hdr, rt_hdr.SIZEOF)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096', log, True)
assert match('nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message', log, True)
assert match('nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20', log, True)
assert match('nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log, True)
assert match('nl_msg_out_handler_debug: -- Debug: Sent Message:', log)
assert match('nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------', log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 20', log)
assert match('print_hdr: .type = 18 <0x12>', log)
assert match('print_hdr: .flags = 773 <REQUEST,ACK,ROOT,MATCH>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: [PAYLOAD] 4 octets', log)
assert match('dump_hex: 11 00 00 00 ....', log)
assert match('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------', log)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_recvmsgs_default(sk)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=\d{3,}', log, True)
assert match('nl_msg_in_handler_debug: -- Debug: Received Message:', log)
assert match('nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------', log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = \d{3,}', log, True)
assert match('print_hdr: .type = 16 <0x10>', log)
assert match('print_hdr: .flags = 2 <MULTI>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: \[PAYLOAD\] \d{3,} octets', log, True)
assert match('dump_hex: 00 00 04 03 01 00 00 00 49 00 01 00 00 00 00 00 ........I.......', log)
assert match('dump_hex: 07 00 03 00 6c 6f 00 00 08 00 0d 00 00 00 00 00 ....lo..........', log)
assert match('dump_hex: 05 00 10 00 00 00 00 00 05 00 11 00 00 00 00 00 ................', log)
for _ in ifaces:
# Done testing this payload. Differs too much between Travis and Raspbian, and probably others.
rem = log.index('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------')
assert 20 < rem # At least check that there were a lot of log statements skipped.
log = log[rem:]
assert match('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------', log)
assert match('nl_valid_handler_debug: -- Debug: Unhandled Valid message: type=0x10 length=\d{3,} flags=<MULTI> '
'sequence-nr=\d{10,} pid=\d{3,}', log, True)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 20 bytes', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log, True)
assert match('nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=20', log, True)
assert match('nl_msg_in_handler_debug: -- Debug: Received Message:', log)
assert match('nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------', log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = 20', log)
assert match('print_hdr: .type = 3 <DONE>', log)
assert match('print_hdr: .flags = 2 <MULTI>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: [PAYLOAD] 4 octets', log)
assert match('dump_hex: 00 00 00 00 ....', log)
assert match('nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------', log)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}', log, True)
assert match('nl_finish_handler_debug: -- Debug: End of multipart message block: type=DONE length=20 flags=<MULTI> '
'sequence-nr=\d{10,} pid=\d{3,}', log, True)
nl_socket_free(sk)
assert not log
def test_socket(tcp_server):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && (nc -l 2000 |base64 &) && sleep 0.1 && ./a.out
#include <netlink/msg.h>
struct nl_sock {
struct sockaddr_nl s_local; struct sockaddr_nl s_peer; int s_fd; int s_proto; unsigned int s_seq_next;
unsigned int s_seq_expect; int s_flags; struct nl_cb *s_cb; size_t s_bufsize;
};
int main() {
struct nl_msg *msg = nlmsg_alloc();
struct nlmsghdr *nlh = nlmsg_hdr(msg);
printf("%d == nlh->nlmsg_len\n", nlh->nlmsg_len);
printf("%d == nlh->nlmsg_type\n", nlh->nlmsg_type);
printf("%d == nlh->nlmsg_flags\n", nlh->nlmsg_flags);
printf("%d == nlh->nlmsg_seq\n", nlh->nlmsg_seq);
printf("%d == nlh->nlmsg_pid\n", nlh->nlmsg_pid);
struct sockaddr_in sin = { .sin_port = htons(2000), .sin_family = AF_INET, };
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
int fd = socket(AF_INET, SOCK_STREAM, 0);
connect(fd, (struct sockaddr *) &sin, sizeof(sin));
struct nl_sock *sk = nl_socket_alloc();
nl_connect(sk, NETLINK_ROUTE);
sk->s_fd = fd;
sk->s_local.nl_pid = 0;
sk->s_seq_next = 0;
nl_complete_msg(sk, msg);
printf("%d == nlh->nlmsg_seq\n", nlh->nlmsg_seq);
struct iovec iov = { .iov_base = nlh, .iov_len = nlh->nlmsg_len };
struct msghdr hdr = { .msg_iov = &iov, .msg_iovlen = 1, };
int ret = nl_sendmsg(sk, msg, &hdr);
printf("Bytes: %d\n", ret);
return 0;
}
// Expected bash output:
// 16 == nlh->nlmsg_len
// 0 == nlh->nlmsg_type
// 0 == nlh->nlmsg_flags
// 0 == nlh->nlmsg_seq
// 0 == nlh->nlmsg_pid
// 0 == nlh->nlmsg_seq
// Bytes: 16
// EAAAAAAABQAAAAAAAAAAAA==
"""
msg = nlmsg_alloc()
nlh = nlmsg_hdr(msg)
assert 16 == nlh.nlmsg_len
assert 0 == nlh.nlmsg_type
assert 0 == nlh.nlmsg_flags
assert 0 == nlh.nlmsg_seq
assert 0 == nlh.nlmsg_pid
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
sk.socket_instance.close()
sk.socket_instance = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.socket_instance.connect(tcp_server.server.server_address)
sk.s_local.nl_pid = 0
sk.s_seq_next = 0
nl_complete_msg(sk, msg)
assert 0 == nlh.nlmsg_seq
nlh.nlmsg_pid = 0 # sk.s_local.nl_pid is read-only in Python.
iov = nlh.bytearray[:nlh.nlmsg_len]
hdr = msghdr(msg_iov=iov)
assert 16 == nl_sendmsg(sk, msg, hdr)
assert 1 == len(tcp_server.data)
assert b'EAAAAAAABQAAAAAAAAAAAA==' == base64.b64encode(buffer(iov))
assert b'EAAAAAAABQAAAAAAAAAAAA==' == base64.b64encode(buffer(tcp_server.data[0]))
nl_socket_free(sk)
def test_nl_socket_modify_cb(log, ifaces):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 ./a.out
#include <netlink/msg.h>
static int callback(struct nl_msg *msg, void *arg) {
printf("Got something.\n");
nl_msg_dump(msg, stdout);
return NL_OK;
}
int main() {
// Send data to the kernel.
struct nl_sock *sk = nl_socket_alloc();
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
int ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, &rt_hdr, sizeof(rt_hdr));
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback, NULL);
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x1b840b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x1b840b8: Allocated new simple message
// nlmsg_reserve: msg 0x1b840b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0x1b840b8: Appended 1 bytes with padding 4
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0x1b840b8, 0 remaining
// nlmsg_free: msg 0x1b840b8: Freed
// Bytes Sent: 20
// recvmsgs: Attempting to read from 0x1b84080
// recvmsgs: recvmsgs(0x1b84080): Read 3364 bytes
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=1116
// Got something.
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1116
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424133909
// .port = 6192
// [PAYLOAD] 1100 octets
// 00 00 04 03 01 00 00 00 49 00 01 00 00 00 00 00 ........I.......
// 07 00 03 00 6c 6f 00 00 08 00 0d 00 00 00 00 00 ....lo..........
// <trimmed>
// 00 00 00 00 00 00 00 00 00 00 00 00 ............
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=1124
// Got something.
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424133909
// .port = 6192
// [PAYLOAD] 1108 octets
// 00 00 01 00 02 00 00 00 43 10 01 00 00 00 00 00 ........C.......
// 09 00 03 00 65 74 68 30 00 00 00 00 08 00 0d 00 ....eth0........
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=1124
// Got something.
// -------------------------- BEGIN NETLINK MESSAGE ---------------------------
// [NETLINK HEADER] 16 octets
// .nlmsg_len = 1124
// .type = 16 <0x10>
// .flags = 2 <MULTI>
// .seq = 1424133909
// .port = 6192
// [PAYLOAD] 1108 octets
// 00 00 01 00 04 00 00 00 03 10 00 00 00 00 00 00 ................
// 0a 00 03 00 77 6c 61 6e 30 00 00 00 08 00 0d 00 ....wlan0.......
// <trimmed>
// 00 00 00 00 ....
// --------------------------- END NETLINK MESSAGE ---------------------------
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// recvmsgs: Attempting to read from 0x1b84080
// recvmsgs: recvmsgs(0x1b84080): Read 20 bytes
// recvmsgs: recvmsgs(0x1b84080): Processing valid message...
// __nlmsg_alloc: msg 0x1b880c0: Allocated new message, maxlen=20
// recvmsgs: recvmsgs(0x1b84080): Increased expected sequence number to 1424133910
// nlmsg_free: Returned message reference 0x1b880c0, 0 remaining
// nlmsg_free: msg 0x1b880c0: Freed
// 0 == nl_recvmsgs_default(sk)
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
got_something = list()
def callback(msg, arg):
got_something.append(arg)
nl_msg_dump(msg)
return NL_OK
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP,
rt_hdr, rt_hdr.SIZEOF)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096',
log, True)
assert match(
'nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message',
log, True)
assert match(
'nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20',
log, True)
assert match(
'nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log,
True)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback,
95)
assert 0 == nl_recvmsgs_default(sk)
assert [95] * len(ifaces) == got_something
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log,
True)
for _ in ifaces:
if 'Attempting to read' in log[0]:
# Lots of network interfaces on this host.
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read \d{4,} bytes', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...',
log, True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=\d{3,}',
log, True)
assert match(
'nl_msg_dump: -------------------------- BEGIN NETLINK MESSAGE ---------------------------',
log)
assert match('nl_msg_dump: [NETLINK HEADER] 16 octets', log)
assert match('print_hdr: .nlmsg_len = \d{3,}', log, True)
assert match('print_hdr: .type = 16 <0x10>', log)
assert match('print_hdr: .flags = 2 <MULTI>', log)
assert match('print_hdr: .seq = \d{10}', log, True)
assert match('print_hdr: .port = \d{3,}', log, True)
assert match('print_msg: \[PAYLOAD\] \d{3,} octets', log, True)
rem = log.index(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------'
)
assert 20 < rem # At least check that there were a lot of log statements skipped.
log = log[rem:]
assert match(
'nl_msg_dump: --------------------------- END NETLINK MESSAGE ---------------------------',
log)
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 20 bytes', log, True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log,
True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=20', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}',
log, True)
nl_socket_free(sk)
assert not log
def test_list_interfaces(ifacesi):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && ./a.out
#include <netlink/msg.h>
static int callback(struct nl_msg *msg, void *arg) {
struct nlmsghdr *nlh = nlmsg_hdr(msg);
struct ifinfomsg *iface = NLMSG_DATA(nlh);
struct rtattr *hdr = IFLA_RTA(iface);
int remaining = nlh->nlmsg_len - NLMSG_LENGTH(sizeof(*iface));
while (RTA_OK(hdr, remaining)) {
if (hdr->rta_type == IFLA_IFNAME) {
printf("Found network interface %d: %s\n", iface->ifi_index, (char *) RTA_DATA(hdr));
}
hdr = RTA_NEXT(hdr, remaining);
}
return NL_OK;
}
int main() {
// Send data to the kernel.
struct nl_sock *sk = nl_socket_alloc();
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
int ret = nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, &rt_hdr, sizeof(rt_hdr));
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback, NULL);
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
nl_socket_free(sk);
return 0;
}
// Expected output:
// 0 == nl_connect(sk, NETLINK_ROUTE)
// Bytes Sent: 20
// Found network interface 1: lo
// Found network interface 2: eth0
// Found network interface 4: wlan0
// 0 == nl_recvmsgs_default(sk)
"""
got_something = dict()
def callback(msg, arg):
nlh = nlmsg_hdr(msg)
iface = ifinfomsg(nlmsg_data(nlh))
hdr = IFLA_RTA(iface)
remaining = c_int(nlh.nlmsg_len - NLMSG_LENGTH(iface.SIZEOF))
while RTA_OK(hdr, remaining):
if hdr.rta_type == IFLA_IFNAME:
arg[int(iface.ifi_index)] = str(get_string(RTA_DATA(hdr)).decode('ascii'))
hdr = RTA_NEXT(hdr, remaining)
return NL_OK
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, NLM_F_REQUEST | NLM_F_DUMP, rt_hdr, rt_hdr.SIZEOF)
assert 0 == nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, callback, got_something)
assert 0 == nl_recvmsgs_default(sk)
assert dict(ifacesi) == got_something
nl_socket_free(sk)
def test_nl_socket_modify_cb_error_verbose(log):
r"""C code to test against.
// gcc a.c $(pkg-config --cflags --libs libnl-genl-3.0) && NLDBG=4 NLCB=verbose ./a.out
#include <netlink/msg.h>
static int callback(struct sockaddr_nl *nla, struct nlmsgerr *err, void *arg) {
int *ret = arg;
*ret = err->error;
printf("Got something.\n");
return NL_STOP;
}
int main() {
// Send data to the kernel.
struct nl_sock *sk = nl_socket_alloc();
printf("%d == nl_connect(sk, NETLINK_ROUTE)\n", nl_connect(sk, NETLINK_ROUTE));
struct rtgenmsg rt_hdr = { .rtgen_family = AF_PACKET, };
int ret = nl_send_simple(sk, RTM_GETLINK, 0, &rt_hdr, sizeof(rt_hdr));
printf("Bytes Sent: %d\n", ret);
// Retrieve kernel's response.
int err = 0;
nl_socket_modify_err_cb(sk, NL_CB_CUSTOM, callback, &err);
printf("%d == err\n", err);
printf("%d == nl_recvmsgs_default(sk)\n", nl_recvmsgs_default(sk));
printf("%d == err\n", err);
nl_socket_free(sk);
return 0;
}
// Expected output (trimmed):
// nl_cache_mngt_register: Registered cache operations genl/family
// 0 == nl_connect(sk, NETLINK_ROUTE)
// __nlmsg_alloc: msg 0x124e0b8: Allocated new message, maxlen=4096
// nlmsg_alloc_simple: msg 0x124e0b8: Allocated new simple message
// nlmsg_reserve: msg 0x124e0b8: Reserved 4 (1) bytes, pad=4, nlmsg_len=20
// nlmsg_append: msg 0x124e0b8: Appended 1 bytes with padding 4
// nl_sendmsg: sent 20 bytes
// nlmsg_free: Returned message reference 0x124e0b8, 0 remaining
// nlmsg_free: msg 0x124e0b8: Freed
// Bytes Sent: 20
// 0 == err
// recvmsgs: Attempting to read from 0x124e080
// recvmsgs: recvmsgs(0x124e080): Read 40 bytes
// recvmsgs: recvmsgs(0x124e080): Processing valid message...
// __nlmsg_alloc: msg 0x12520c0: Allocated new message, maxlen=40
// recvmsgs: recvmsgs(0x124e080): Increased expected sequence number to 1424136270
// Got something.
// nlmsg_free: Returned message reference 0x12520c0, 0 remaining
// nlmsg_free: msg 0x12520c0: Freed
// -7 == nl_recvmsgs_default(sk)
// -22 == err
// nl_cache_mngt_unregister: Unregistered cache operations genl/family
"""
got_something = list()
def callback(_, err, arg):
arg.append(err.error)
return NL_STOP
del log[:]
sk = nl_socket_alloc()
nl_connect(sk, NETLINK_ROUTE)
rt_hdr = rtgenmsg(rtgen_family=socket.AF_PACKET)
assert 20 == nl_send_simple(sk, RTM_GETLINK, 0, rt_hdr, rt_hdr.SIZEOF)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=4096',
log, True)
assert match(
'nlmsg_alloc_simple: msg 0x[a-f0-9]+: Allocated new simple message',
log, True)
assert match(
'nlmsg_reserve: msg 0x[a-f0-9]+: Reserved 4 \(1\) bytes, pad=4, nlmsg_len=20',
log, True)
assert match(
'nlmsg_append: msg 0x[a-f0-9]+: Appended 1 bytes with padding 4', log,
True)
assert match('nl_sendmsg: sent 20 bytes', log)
assert not log
assert 0 == nl_socket_modify_err_cb(sk, NL_CB_CUSTOM, callback,
got_something)
assert -7 == nl_recvmsgs_default(sk)
assert [-22] == got_something
assert match('recvmsgs: Attempting to read from 0x[a-f0-9]+', log, True)
assert match('recvmsgs: recvmsgs\(0x[a-f0-9]+\): Read 40 bytes', log, True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Processing valid message...', log,
True)
assert match(
'nlmsg_alloc: msg 0x[a-f0-9]+: Allocated new message, maxlen=40', log,
True)
assert match(
'recvmsgs: recvmsgs\(0x[a-f0-9]+\): Increased expected sequence number to \d{4,}',
log, True)
nl_socket_free(sk)
assert not log
