]>
Commit | Line | Data |
---|---|---|
2c06d9a9 DDP |
1 | /* |
2 | * Copyright (c) 2015 Nicira, Inc. | |
3 | * | |
4 | * Licensed under the Apache License, Version 2.0 (the "License"); | |
5 | * you may not use this file except in compliance with the License. | |
6 | * You may obtain a copy of the License at: | |
7 | * | |
8 | * http://www.apache.org/licenses/LICENSE-2.0 | |
9 | * | |
10 | * Unless required by applicable law or agreed to in writing, software | |
11 | * distributed under the License is distributed on an "AS IS" BASIS, | |
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
13 | * See the License for the specific language governing permissions and | |
14 | * limitations under the License. | |
15 | */ | |
16 | ||
17 | #include <config.h> | |
18 | ||
19 | #include <linux/netfilter/nfnetlink.h> | |
20 | ||
21 | #include "ct-dpif.h" | |
22 | #include "netlink-conntrack.h" | |
23 | #include "netlink-notifier.h" | |
24 | #include "ovstest.h" | |
25 | #include "poll-loop.h" | |
26 | ||
27 | /* Monitor command */ | |
28 | struct test_change { | |
29 | enum nl_ct_event_type type; | |
30 | struct ct_dpif_entry entry; | |
31 | }; | |
32 | ||
33 | static bool | |
34 | event_parse(struct ofpbuf *buf, void *change_) | |
35 | { | |
36 | struct test_change *change = change_; | |
37 | ||
38 | return nl_ct_parse_entry(buf, &change->entry, &change->type); | |
39 | } | |
40 | ||
41 | static void | |
42 | event_print(const void *change_, void *aux OVS_UNUSED) | |
43 | { | |
44 | const struct test_change *change = change_; | |
45 | ||
46 | if (change) { | |
47 | struct ds ds = DS_EMPTY_INITIALIZER; | |
48 | ||
49 | nl_ct_format_event_entry(&change->entry, change->type, &ds, true, | |
50 | true); | |
51 | printf("%s\n", ds_cstr(&ds)); | |
52 | ds_destroy(&ds); | |
53 | } | |
54 | } | |
55 | ||
56 | static void | |
57 | test_nl_ct_monitor(struct ovs_cmdl_context *ctx OVS_UNUSED) | |
58 | { | |
59 | int groups [] = { | |
60 | NFNLGRP_CONNTRACK_DESTROY, | |
61 | NFNLGRP_CONNTRACK_NEW, | |
62 | NFNLGRP_CONNTRACK_UPDATE, | |
63 | }; | |
64 | ||
65 | struct nln *nlns[ARRAY_SIZE(groups)]; | |
66 | struct nln_notifier *notifiers[ARRAY_SIZE(groups)]; | |
67 | ||
68 | struct test_change change; | |
69 | ||
70 | unsigned i; | |
71 | ||
72 | for (i = 0; i < ARRAY_SIZE(groups); i++) { | |
73 | nlns[i] = nln_create(NETLINK_NETFILTER, groups[i], event_parse, | |
74 | &change); | |
75 | ||
76 | notifiers[i] = nln_notifier_create(nlns[i], event_print, NULL); | |
77 | } | |
78 | ||
79 | for (;;) { | |
80 | for (i = 0; i < ARRAY_SIZE(groups); i++) { | |
81 | nln_run(nlns[i]); | |
82 | nln_wait(nlns[i]); | |
83 | } | |
84 | poll_block(); | |
85 | } | |
86 | ||
87 | for (i = 0; i < ARRAY_SIZE(groups); i++) { | |
88 | nln_notifier_destroy(notifiers[i]); | |
89 | nln_destroy(nlns[i]); | |
90 | } | |
91 | } | |
92 | \f | |
93 | /* Dump command */ | |
94 | static void | |
95 | test_nl_ct_dump(struct ovs_cmdl_context *ctx) | |
96 | { | |
97 | struct nl_ct_dump_state *dump; | |
98 | uint16_t zone, *pzone = NULL; | |
99 | struct ct_dpif_entry entry; | |
100 | int err; | |
101 | ||
102 | if (ctx->argc >= 2) { | |
103 | if (!ovs_scan(ctx->argv[1], "zone=%"SCNu16, &zone)) { | |
104 | ovs_fatal(0, "Error parsing zone= specifier"); | |
105 | } | |
106 | pzone = &zone; | |
107 | } | |
108 | err = nl_ct_dump_start(&dump, pzone); | |
109 | if (err) { | |
110 | ovs_fatal(err, "Error creating conntrack netlink dump"); | |
111 | } | |
112 | ||
113 | do { | |
114 | err = nl_ct_dump_next(dump, &entry); | |
115 | if (!err) { | |
116 | struct ds ds = DS_EMPTY_INITIALIZER; | |
117 | ||
118 | ct_dpif_format_entry(&entry, &ds, true, true); | |
119 | printf("%s\n", ds_cstr(&ds)); | |
120 | ds_destroy(&ds); | |
121 | } | |
122 | } while (!err); | |
123 | ||
124 | if (err != EOF) { | |
125 | ovs_fatal(err, "Error dumping conntrack netlink entry"); | |
126 | } | |
127 | nl_ct_dump_done(dump); | |
128 | } | |
129 | \f | |
130 | /* Flush command */ | |
131 | static void | |
132 | test_nl_ct_flush(struct ovs_cmdl_context *ctx OVS_UNUSED) | |
133 | { | |
134 | int err; | |
135 | ||
136 | if (ctx->argc >= 2) { | |
137 | uint16_t zone; | |
138 | ||
139 | if (ovs_scan(ctx->argv[1], "zone=%"SCNu16, &zone)) { | |
140 | err = nl_ct_flush_zone(zone); | |
141 | } else { | |
142 | ovs_fatal(0, "Error parsing zone= specifier"); | |
143 | } | |
144 | } else { | |
145 | err = nl_ct_flush(); | |
146 | } | |
147 | if (err) { | |
148 | ovs_fatal(err, "Error flushing conntrack netlink"); | |
149 | } | |
150 | } | |
151 | \f | |
152 | static const struct ovs_cmdl_command commands[] = { | |
153 | /* Linux netlink connection tracker interface test. */ | |
154 | ||
155 | /* Prints all the entries in the connection table and exits. */ | |
156 | {"dump", "[zone=zone]", 0, 1, test_nl_ct_dump}, | |
157 | /* Listens to all the connection tracking events and prints them to | |
158 | * standard output until killed. */ | |
159 | {"monitor", "", 0, 0, test_nl_ct_monitor}, | |
160 | /* Flushes all the entries from all the tables.. */ | |
161 | {"flush", "[zone=zone]", 0, 1, test_nl_ct_flush}, | |
162 | ||
163 | {NULL, NULL, 0, 0, NULL}, | |
164 | }; | |
165 | ||
166 | static void | |
167 | test_netlink_conntrack(int argc, char *argv[]) | |
168 | { | |
169 | struct ovs_cmdl_context ctx = { | |
170 | .argc = argc - 1, | |
171 | .argv = argv + 1, | |
172 | }; | |
173 | set_program_name(argv[0]); | |
174 | ovs_cmdl_run_command(&ctx, commands); | |
175 | } | |
176 | ||
177 | OVSTEST_REGISTER("test-netlink-conntrack", test_netlink_conntrack); |