You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
4.2 KiB
121 lines
4.2 KiB
/*
|
|
* Copyright (c) 2017 JingPiao Chen <chenjingpiao@gmail.com>
|
|
* Copyright (c) 2017 The strace developers.
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* 3. The name of the author may not be used to endorse or promote products
|
|
* derived from this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
|
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
|
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
|
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "tests.h"
|
|
|
|
#ifdef HAVE_LINUX_NETFILTER_NFNETLINK_ACCT_H
|
|
|
|
# include <stdio.h>
|
|
# include <sys/socket.h>
|
|
# include "netlink.h"
|
|
# include <linux/netfilter/nfnetlink.h>
|
|
# include <linux/netfilter/nfnetlink_acct.h>
|
|
|
|
static void
|
|
test_nlmsg_type(const int fd)
|
|
{
|
|
long rc;
|
|
struct nlmsghdr nlh = {
|
|
.nlmsg_len = sizeof(nlh),
|
|
.nlmsg_flags = NLM_F_REQUEST,
|
|
};
|
|
|
|
nlh.nlmsg_type = NFNL_SUBSYS_ACCT << 8 | NFNL_MSG_ACCT_NEW;
|
|
rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
|
|
printf("sendto(%d, {len=%u"
|
|
", type=NFNL_SUBSYS_ACCT<<8|NFNL_MSG_ACCT_NEW"
|
|
", flags=NLM_F_REQUEST, seq=0, pid=0}"
|
|
", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
|
|
fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
|
|
|
|
nlh.nlmsg_type = NFNL_SUBSYS_ACCT << 8 | 0xff;
|
|
rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
|
|
printf("sendto(%d, {len=%u"
|
|
", type=NFNL_SUBSYS_ACCT<<8|0xff /* NFNL_MSG_ACCT_??? */"
|
|
", flags=NLM_F_REQUEST, seq=0, pid=0}"
|
|
", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
|
|
fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
|
|
}
|
|
|
|
static void
|
|
test_nlmsg_flags(const int fd)
|
|
{
|
|
long rc;
|
|
struct nlmsghdr nlh = {
|
|
.nlmsg_len = sizeof(nlh),
|
|
};
|
|
|
|
nlh.nlmsg_type = NFNL_SUBSYS_ACCT << 8 | NFNL_MSG_ACCT_NEW;
|
|
nlh.nlmsg_flags = NLM_F_REQUEST | NLM_F_CREATE;
|
|
rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
|
|
printf("sendto(%d, {len=%u"
|
|
", type=NFNL_SUBSYS_ACCT<<8|NFNL_MSG_ACCT_NEW"
|
|
", flags=NLM_F_REQUEST|NLM_F_CREATE, seq=0, pid=0}"
|
|
", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
|
|
fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
|
|
|
|
nlh.nlmsg_type = NFNL_SUBSYS_ACCT << 8 | NFNL_MSG_ACCT_GET;
|
|
nlh.nlmsg_flags = NLM_F_REQUEST | NLM_F_MATCH;
|
|
rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
|
|
printf("sendto(%d, {len=%u"
|
|
", type=NFNL_SUBSYS_ACCT<<8|NFNL_MSG_ACCT_GET"
|
|
", flags=NLM_F_REQUEST|NLM_F_MATCH, seq=0, pid=0}"
|
|
", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
|
|
fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
|
|
|
|
nlh.nlmsg_type = NFNL_SUBSYS_ACCT << 8 | NFNL_MSG_ACCT_DEL;
|
|
nlh.nlmsg_flags = NLM_F_REQUEST | NLM_F_NONREC;
|
|
rc = sendto(fd, &nlh, nlh.nlmsg_len, MSG_DONTWAIT, NULL, 0);
|
|
printf("sendto(%d, {len=%u"
|
|
", type=NFNL_SUBSYS_ACCT<<8|NFNL_MSG_ACCT_DEL"
|
|
", flags=NLM_F_REQUEST|NLM_F_NONREC, seq=0, pid=0}"
|
|
", %u, MSG_DONTWAIT, NULL, 0) = %s\n",
|
|
fd, nlh.nlmsg_len, nlh.nlmsg_len, sprintrc(rc));
|
|
}
|
|
|
|
int
|
|
main(void)
|
|
{
|
|
skip_if_unavailable("/proc/self/fd/");
|
|
|
|
int fd = create_nl_socket(NETLINK_NETFILTER);
|
|
|
|
test_nlmsg_type(fd);
|
|
test_nlmsg_flags(fd);
|
|
|
|
puts("+++ exited with 0 +++");
|
|
|
|
return 0;
|
|
}
|
|
|
|
#else
|
|
|
|
SKIP_MAIN_UNDEFINED("HAVE_LINUX_NETFILTER_NFNETLINK_ACCT_H")
|
|
|
|
#endif
|