1 // SPDX-License-Identifier: GPL-2.0-only
2 // Copyright (c) 2020 Facebook Inc.
3 
4 #include <linux/debugfs.h>
5 #include <linux/netdevice.h>
6 #include <linux/slab.h>
7 #include <net/udp_tunnel.h>
8 
9 #include "netdevsim.h"
10 
11 static int
12 nsim_udp_tunnel_set_port(struct net_device *dev, unsigned int table,
13 			 unsigned int entry, struct udp_tunnel_info *ti)
14 {
15 	struct netdevsim *ns = netdev_priv(dev);
16 	int ret;
17 
18 	ret = -ns->udp_ports.inject_error;
19 	ns->udp_ports.inject_error = 0;
20 
21 	if (ns->udp_ports.sleep)
22 		msleep(ns->udp_ports.sleep);
23 
24 	if (!ret) {
25 		if (ns->udp_ports.ports[table][entry]) {
26 			WARN(1, "entry already in use\n");
27 			ret = -EBUSY;
28 		} else {
29 			ns->udp_ports.ports[table][entry] =
30 				be16_to_cpu(ti->port) << 16 | ti->type;
31 		}
32 	}
33 
34 	netdev_info(dev, "set [%d, %d] type %d family %d port %d - %d\n",
35 		    table, entry, ti->type, ti->sa_family, ntohs(ti->port),
36 		    ret);
37 	return ret;
38 }
39 
40 static int
41 nsim_udp_tunnel_unset_port(struct net_device *dev, unsigned int table,
42 			   unsigned int entry, struct udp_tunnel_info *ti)
43 {
44 	struct netdevsim *ns = netdev_priv(dev);
45 	int ret;
46 
47 	ret = -ns->udp_ports.inject_error;
48 	ns->udp_ports.inject_error = 0;
49 
50 	if (ns->udp_ports.sleep)
51 		msleep(ns->udp_ports.sleep);
52 	if (!ret) {
53 		u32 val = be16_to_cpu(ti->port) << 16 | ti->type;
54 
55 		if (val == ns->udp_ports.ports[table][entry]) {
56 			ns->udp_ports.ports[table][entry] = 0;
57 		} else {
58 			WARN(1, "entry not installed %x vs %x\n",
59 			     val, ns->udp_ports.ports[table][entry]);
60 			ret = -ENOENT;
61 		}
62 	}
63 
64 	netdev_info(dev, "unset [%d, %d] type %d family %d port %d - %d\n",
65 		    table, entry, ti->type, ti->sa_family, ntohs(ti->port),
66 		    ret);
67 	return ret;
68 }
69 
70 static int
71 nsim_udp_tunnel_sync_table(struct net_device *dev, unsigned int table)
72 {
73 	struct netdevsim *ns = netdev_priv(dev);
74 	struct udp_tunnel_info ti;
75 	unsigned int i;
76 	int ret;
77 
78 	ret = -ns->udp_ports.inject_error;
79 	ns->udp_ports.inject_error = 0;
80 
81 	for (i = 0; i < NSIM_UDP_TUNNEL_N_PORTS; i++) {
82 		udp_tunnel_nic_get_port(dev, table, i, &ti);
83 		ns->udp_ports.ports[table][i] =
84 			be16_to_cpu(ti.port) << 16 | ti.type;
85 	}
86 
87 	return ret;
88 }
89 
90 static const struct udp_tunnel_nic_info nsim_udp_tunnel_info = {
91 	.set_port	= nsim_udp_tunnel_set_port,
92 	.unset_port	= nsim_udp_tunnel_unset_port,
93 	.sync_table	= nsim_udp_tunnel_sync_table,
94 
95 	.tables = {
96 		{
97 			.n_entries	= NSIM_UDP_TUNNEL_N_PORTS,
98 			.tunnel_types	= UDP_TUNNEL_TYPE_VXLAN,
99 		},
100 		{
101 			.n_entries	= NSIM_UDP_TUNNEL_N_PORTS,
102 			.tunnel_types	= UDP_TUNNEL_TYPE_GENEVE |
103 					  UDP_TUNNEL_TYPE_VXLAN_GPE,
104 		},
105 	},
106 };
107 
108 static ssize_t
109 nsim_udp_tunnels_info_reset_write(struct file *file, const char __user *data,
110 				  size_t count, loff_t *ppos)
111 {
112 	struct net_device *dev = file->private_data;
113 	struct netdevsim *ns = netdev_priv(dev);
114 
115 	memset(ns->udp_ports.ports, 0, sizeof(ns->udp_ports.__ports));
116 	rtnl_lock();
117 	udp_tunnel_nic_reset_ntf(dev);
118 	rtnl_unlock();
119 
120 	return count;
121 }
122 
123 static const struct file_operations nsim_udp_tunnels_info_reset_fops = {
124 	.open = simple_open,
125 	.write = nsim_udp_tunnels_info_reset_write,
126 	.llseek = generic_file_llseek,
127 };
128 
129 int nsim_udp_tunnels_info_create(struct nsim_dev *nsim_dev,
130 				 struct net_device *dev)
131 {
132 	struct netdevsim *ns = netdev_priv(dev);
133 	struct udp_tunnel_nic_info *info;
134 
135 	if (nsim_dev->udp_ports.shared && nsim_dev->udp_ports.open_only) {
136 		dev_err(&nsim_dev->nsim_bus_dev->dev,
137 			"shared can't be used in conjunction with open_only\n");
138 		return -EINVAL;
139 	}
140 
141 	if (!nsim_dev->udp_ports.shared)
142 		ns->udp_ports.ports = ns->udp_ports.__ports;
143 	else
144 		ns->udp_ports.ports = nsim_dev->udp_ports.__ports;
145 
146 	debugfs_create_u32("udp_ports_inject_error", 0600,
147 			   ns->nsim_dev_port->ddir,
148 			   &ns->udp_ports.inject_error);
149 
150 	ns->udp_ports.dfs_ports[0].array = ns->udp_ports.ports[0];
151 	ns->udp_ports.dfs_ports[0].n_elements = NSIM_UDP_TUNNEL_N_PORTS;
152 	debugfs_create_u32_array("udp_ports_table0", 0400,
153 				 ns->nsim_dev_port->ddir,
154 				 &ns->udp_ports.dfs_ports[0]);
155 
156 	ns->udp_ports.dfs_ports[1].array = ns->udp_ports.ports[1];
157 	ns->udp_ports.dfs_ports[1].n_elements = NSIM_UDP_TUNNEL_N_PORTS;
158 	debugfs_create_u32_array("udp_ports_table1", 0400,
159 				 ns->nsim_dev_port->ddir,
160 				 &ns->udp_ports.dfs_ports[1]);
161 
162 	debugfs_create_file("udp_ports_reset", 0200, ns->nsim_dev_port->ddir,
163 			    dev, &nsim_udp_tunnels_info_reset_fops);
164 
165 	/* Note: it's not normal to allocate the info struct like this!
166 	 * Drivers are expected to use a static const one, here we're testing.
167 	 */
168 	info = kmemdup(&nsim_udp_tunnel_info, sizeof(nsim_udp_tunnel_info),
169 		       GFP_KERNEL);
170 	if (!info)
171 		return -ENOMEM;
172 	ns->udp_ports.sleep = nsim_dev->udp_ports.sleep;
173 
174 	if (nsim_dev->udp_ports.sync_all) {
175 		info->set_port = NULL;
176 		info->unset_port = NULL;
177 	} else {
178 		info->sync_table = NULL;
179 	}
180 
181 	if (ns->udp_ports.sleep)
182 		info->flags |= UDP_TUNNEL_NIC_INFO_MAY_SLEEP;
183 	if (nsim_dev->udp_ports.open_only)
184 		info->flags |= UDP_TUNNEL_NIC_INFO_OPEN_ONLY;
185 	if (nsim_dev->udp_ports.ipv4_only)
186 		info->flags |= UDP_TUNNEL_NIC_INFO_IPV4_ONLY;
187 	if (nsim_dev->udp_ports.shared)
188 		info->shared = &nsim_dev->udp_ports.utn_shared;
189 	if (nsim_dev->udp_ports.static_iana_vxlan)
190 		info->flags |= UDP_TUNNEL_NIC_INFO_STATIC_IANA_VXLAN;
191 
192 	dev->udp_tunnel_nic_info = info;
193 	return 0;
194 }
195 
196 void nsim_udp_tunnels_info_destroy(struct net_device *dev)
197 {
198 	kfree(dev->udp_tunnel_nic_info);
199 	dev->udp_tunnel_nic_info = NULL;
200 }
201 
202 void nsim_udp_tunnels_debugfs_create(struct nsim_dev *nsim_dev)
203 {
204 	debugfs_create_bool("udp_ports_sync_all", 0600, nsim_dev->ddir,
205 			    &nsim_dev->udp_ports.sync_all);
206 	debugfs_create_bool("udp_ports_open_only", 0600, nsim_dev->ddir,
207 			    &nsim_dev->udp_ports.open_only);
208 	debugfs_create_bool("udp_ports_ipv4_only", 0600, nsim_dev->ddir,
209 			    &nsim_dev->udp_ports.ipv4_only);
210 	debugfs_create_bool("udp_ports_shared", 0600, nsim_dev->ddir,
211 			    &nsim_dev->udp_ports.shared);
212 	debugfs_create_bool("udp_ports_static_iana_vxlan", 0600, nsim_dev->ddir,
213 			    &nsim_dev->udp_ports.static_iana_vxlan);
214 	debugfs_create_u32("udp_ports_sleep", 0600, nsim_dev->ddir,
215 			   &nsim_dev->udp_ports.sleep);
216 }
217