mirror of
http://git.haproxy.org/git/haproxy.git/
synced 2024-12-17 08:54:41 +00:00
a4d0361969
Add plug_qdisc.c source file which may help in how to programatically use plug queueing disciplines with its README file. Such code may be useful to reproduce painful network application bugs.
87 lines
2.3 KiB
C
87 lines
2.3 KiB
C
#include <stdint.h>
|
|
#include <netlink/cache.h>
|
|
#include <netlink/cli/utils.h>
|
|
#include <netlink/cli/tc.h>
|
|
#include <netlink/cli/qdisc.h>
|
|
#include <netlink/cli/link.h>
|
|
#include <netlink/route/qdisc/plug.h>
|
|
|
|
/*
|
|
* XXX Please, first note that this code is not safe. XXX
|
|
* It was developed fast so that to reproduce a bug.
|
|
* You will certainly have to adapt it to your application.
|
|
* But at least it gives an idea about how to programatically use plug
|
|
* queueing disciplines.
|
|
*/
|
|
|
|
static struct nl_sock *nl_sock;
|
|
static struct nl_cache *link_cache;
|
|
static struct rtnl_qdisc *qdisc;
|
|
static struct rtnl_tc *tc;
|
|
|
|
static int qdisc_init(void)
|
|
{
|
|
nl_sock = nl_cli_alloc_socket();
|
|
nl_cli_connect(nl_sock, NETLINK_ROUTE);
|
|
link_cache = nl_cli_link_alloc_cache(nl_sock);
|
|
qdisc = nl_cli_qdisc_alloc();
|
|
tc = (struct rtnl_tc *)qdisc;
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* Stop buffering and release all buffered and incoming 'qdisc'
|
|
* queueing discipline traffic.
|
|
*/
|
|
int plug_qdisc_release_indefinite_buffer(void)
|
|
{
|
|
rtnl_qdisc_plug_release_indefinite(qdisc);
|
|
return rtnl_qdisc_add(nl_sock, qdisc, 0);
|
|
}
|
|
|
|
/* Start buffering incoming 'qdisc' queueing discipline traffic. */
|
|
int plug_qdisc_plug_buffer(void)
|
|
{
|
|
rtnl_qdisc_plug_buffer(qdisc);
|
|
return rtnl_qdisc_add(nl_sock, qdisc, 0);
|
|
}
|
|
|
|
/* Create a plug qdisc attached to 'device' network device with 'parent'
|
|
* as parent, with 'id' as ID and 'limit' as buffer size.
|
|
* This is equivalent to use nl-qdisc-add tool like that:
|
|
* $ nl-qdisc-add --dev=<device> --parent=<parent> --id=<id> plug --limit <limit>
|
|
* $ nl-qdisc-add --dev=<device> --parent=<parent> --id=<id> --update plug --release-indefinite
|
|
*/
|
|
int plug_qdisc_attach(char *device, char *parent, char *id, uint32_t limit)
|
|
{
|
|
int ret;
|
|
|
|
if (!tc && qdisc_init() == -1)
|
|
return -1;
|
|
|
|
nl_cli_tc_parse_dev(tc, link_cache, device);
|
|
nl_cli_tc_parse_parent(tc, parent);
|
|
if (!rtnl_tc_get_ifindex(tc))
|
|
return -1;
|
|
|
|
if (!rtnl_tc_get_parent(tc))
|
|
return -1;
|
|
if (id)
|
|
nl_cli_tc_parse_handle(tc, id, 1);
|
|
|
|
rtnl_tc_set_kind(tc, "plug");
|
|
if (limit)
|
|
rtnl_qdisc_plug_set_limit(qdisc, limit);
|
|
|
|
ret = rtnl_qdisc_add(nl_sock, qdisc, NLM_F_CREATE);
|
|
if (ret < 0) {
|
|
fprintf(stderr, "Could add attach qdisc: %s\n", nl_geterror(ret));
|
|
return -1;
|
|
}
|
|
/* Release buffer. */
|
|
plug_qdisc_release_indefinite_buffer();
|
|
|
|
return 0;
|
|
}
|
|
|