tests: Fix sensitivity to record ordering in test-netflow output.
[sliver-openvswitch.git] / datapath / genl_exec.c
1 /*
2  * Copyright (c) 2007-2012 Nicira, Inc.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of version 2 of the GNU General Public
6  * License as published by the Free Software Foundation.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License
14  * along with this program; if not, write to the Free Software
15  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
16  * 02110-1301, USA
17  */
18
19 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
20
21 #include <linux/version.h>
22 #include <linux/completion.h>
23 #include <net/genetlink.h>
24 #include "genl_exec.h"
25
26 #if LINUX_VERSION_CODE < KERNEL_VERSION(2,6,35)
27
28 static DEFINE_MUTEX(genl_exec_lock);
29
30 static genl_exec_func_t  genl_exec_function;
31 static int               genl_exec_function_ret;
32 static void             *genl_exec_data;
33 static struct completion done;
34
35 static struct sk_buff *genlmsg_skb;
36
37 static int genl_exec_cmd(struct sk_buff *dummy, struct genl_info *dummy2)
38 {
39         genl_exec_function_ret = genl_exec_function(genl_exec_data);
40         complete(&done);
41         return 0;
42 }
43
44 enum exec_cmd {
45         GENL_EXEC_UNSPEC,
46         GENL_EXEC_RUN,
47 };
48
49 static struct genl_family genl_exec_family = {
50         .id = GENL_ID_GENERATE,
51         .name = "ovs_genl_exec",
52         .version = 1,
53 };
54
55 static struct genl_ops genl_exec_ops[] = {
56         {
57          .cmd = GENL_EXEC_RUN,
58          .doit = genl_exec_cmd,
59          .flags = CAP_NET_ADMIN,
60         },
61 };
62
63 int genl_exec_init(void)
64 {
65         int err;
66
67         err = genl_register_family_with_ops(&genl_exec_family,
68                         genl_exec_ops, ARRAY_SIZE(genl_exec_ops));
69
70         if (err)
71                 return err;
72
73         genlmsg_skb = genlmsg_new(0, GFP_KERNEL);
74         if (!genlmsg_skb) {
75                 genl_unregister_family(&genl_exec_family);
76                 return -ENOMEM;
77         }
78         return 0;
79 }
80
81 void genl_exec_exit(void)
82 {
83         kfree_skb(genlmsg_skb);
84         genl_unregister_family(&genl_exec_family);
85 }
86
87 /* genl_lock() is not exported from older kernel.
88  * Following function allows any function to be executed with
89  * genl_mutex held. */
90
91 int genl_exec(genl_exec_func_t func, void *data)
92 {
93         int ret;
94
95         mutex_lock(&genl_exec_lock);
96
97         init_completion(&done);
98         skb_get(genlmsg_skb);
99         genlmsg_put(genlmsg_skb, 0, 0, &genl_exec_family,
100                     NLM_F_REQUEST, GENL_EXEC_RUN);
101
102         genl_exec_function = func;
103         genl_exec_data = data;
104
105         /* There is no need to send msg to current namespace. */
106         ret = genlmsg_unicast(&init_net, genlmsg_skb, 0);
107
108         if (!ret) {
109                 wait_for_completion(&done);
110                 ret = genl_exec_function_ret;
111         } else {
112                 pr_err("genl_exec send error %d\n", ret);
113         }
114
115         /* Wait for genetlink to kfree skb. */
116         while (skb_shared(genlmsg_skb))
117                 cpu_relax();
118
119         genlmsg_skb->data = genlmsg_skb->head;
120         skb_reset_tail_pointer(genlmsg_skb);
121
122         mutex_unlock(&genl_exec_lock);
123
124         return ret;
125 }
126
127 #else
128
129 int genl_exec(genl_exec_func_t func, void *data)
130 {
131         int ret;
132
133         genl_lock();
134         ret = func(data);
135         genl_unlock();
136         return ret;
137 }
138
139 int genl_exec_init(void)
140 {
141         return 0;
142 }
143
144 void genl_exec_exit(void)
145 {
146 }
147 #endif