xref: /OK3568_Linux_fs/kernel/arch/um/drivers/vector_user.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun  * Copyright (C) 2001 - 2007 Jeff Dike (jdike@{addtoit,linux.intel}.com)
4*4882a593Smuzhiyun  */
5*4882a593Smuzhiyun 
6*4882a593Smuzhiyun #include <stdio.h>
7*4882a593Smuzhiyun #include <unistd.h>
8*4882a593Smuzhiyun #include <stdarg.h>
9*4882a593Smuzhiyun #include <errno.h>
10*4882a593Smuzhiyun #include <stddef.h>
11*4882a593Smuzhiyun #include <string.h>
12*4882a593Smuzhiyun #include <sys/ioctl.h>
13*4882a593Smuzhiyun #include <net/if.h>
14*4882a593Smuzhiyun #include <linux/if_tun.h>
15*4882a593Smuzhiyun #include <arpa/inet.h>
16*4882a593Smuzhiyun #include <sys/types.h>
17*4882a593Smuzhiyun #include <sys/stat.h>
18*4882a593Smuzhiyun #include <fcntl.h>
19*4882a593Smuzhiyun #include <sys/socket.h>
20*4882a593Smuzhiyun #include <sys/un.h>
21*4882a593Smuzhiyun #include <netinet/ip.h>
22*4882a593Smuzhiyun #include <linux/if_ether.h>
23*4882a593Smuzhiyun #include <linux/if_packet.h>
24*4882a593Smuzhiyun #include <sys/wait.h>
25*4882a593Smuzhiyun #include <sys/uio.h>
26*4882a593Smuzhiyun #include <linux/virtio_net.h>
27*4882a593Smuzhiyun #include <netdb.h>
28*4882a593Smuzhiyun #include <stdlib.h>
29*4882a593Smuzhiyun #include <os.h>
30*4882a593Smuzhiyun #include <limits.h>
31*4882a593Smuzhiyun #include <um_malloc.h>
32*4882a593Smuzhiyun #include "vector_user.h"
33*4882a593Smuzhiyun 
34*4882a593Smuzhiyun #define ID_GRE 0
35*4882a593Smuzhiyun #define ID_L2TPV3 1
36*4882a593Smuzhiyun #define ID_BESS 2
37*4882a593Smuzhiyun #define ID_MAX 2
38*4882a593Smuzhiyun 
39*4882a593Smuzhiyun #define TOKEN_IFNAME "ifname"
40*4882a593Smuzhiyun #define TOKEN_SCRIPT "ifup"
41*4882a593Smuzhiyun 
42*4882a593Smuzhiyun #define TRANS_RAW "raw"
43*4882a593Smuzhiyun #define TRANS_RAW_LEN strlen(TRANS_RAW)
44*4882a593Smuzhiyun 
45*4882a593Smuzhiyun #define TRANS_FD "fd"
46*4882a593Smuzhiyun #define TRANS_FD_LEN strlen(TRANS_FD)
47*4882a593Smuzhiyun 
48*4882a593Smuzhiyun #define VNET_HDR_FAIL "could not enable vnet headers on fd %d"
49*4882a593Smuzhiyun #define TUN_GET_F_FAIL "tapraw: TUNGETFEATURES failed: %s"
50*4882a593Smuzhiyun #define L2TPV3_BIND_FAIL "l2tpv3_open : could not bind socket err=%i"
51*4882a593Smuzhiyun #define UNIX_BIND_FAIL "unix_open : could not bind socket err=%i"
52*4882a593Smuzhiyun #define BPF_ATTACH_FAIL "Failed to attach filter size %d prog %px to %d, err %d\n"
53*4882a593Smuzhiyun #define BPF_DETACH_FAIL "Failed to detach filter size %d prog %px to %d, err %d\n"
54*4882a593Smuzhiyun 
55*4882a593Smuzhiyun #define MAX_UN_LEN 107
56*4882a593Smuzhiyun 
57*4882a593Smuzhiyun static const char padchar[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
58*4882a593Smuzhiyun static const char *template = "tapXXXXXX";
59*4882a593Smuzhiyun 
60*4882a593Smuzhiyun /* This is very ugly and brute force lookup, but it is done
61*4882a593Smuzhiyun  * only once at initialization so not worth doing hashes or
62*4882a593Smuzhiyun  * anything more intelligent
63*4882a593Smuzhiyun  */
64*4882a593Smuzhiyun 
uml_vector_fetch_arg(struct arglist * ifspec,char * token)65*4882a593Smuzhiyun char *uml_vector_fetch_arg(struct arglist *ifspec, char *token)
66*4882a593Smuzhiyun {
67*4882a593Smuzhiyun 	int i;
68*4882a593Smuzhiyun 
69*4882a593Smuzhiyun 	for (i = 0; i < ifspec->numargs; i++) {
70*4882a593Smuzhiyun 		if (strcmp(ifspec->tokens[i], token) == 0)
71*4882a593Smuzhiyun 			return ifspec->values[i];
72*4882a593Smuzhiyun 	}
73*4882a593Smuzhiyun 	return NULL;
74*4882a593Smuzhiyun 
75*4882a593Smuzhiyun }
76*4882a593Smuzhiyun 
uml_parse_vector_ifspec(char * arg)77*4882a593Smuzhiyun struct arglist *uml_parse_vector_ifspec(char *arg)
78*4882a593Smuzhiyun {
79*4882a593Smuzhiyun 	struct arglist *result;
80*4882a593Smuzhiyun 	int pos, len;
81*4882a593Smuzhiyun 	bool parsing_token = true, next_starts = true;
82*4882a593Smuzhiyun 
83*4882a593Smuzhiyun 	if (arg == NULL)
84*4882a593Smuzhiyun 		return NULL;
85*4882a593Smuzhiyun 	result = uml_kmalloc(sizeof(struct arglist), UM_GFP_KERNEL);
86*4882a593Smuzhiyun 	if (result == NULL)
87*4882a593Smuzhiyun 		return NULL;
88*4882a593Smuzhiyun 	result->numargs = 0;
89*4882a593Smuzhiyun 	len = strlen(arg);
90*4882a593Smuzhiyun 	for (pos = 0; pos < len; pos++) {
91*4882a593Smuzhiyun 		if (next_starts) {
92*4882a593Smuzhiyun 			if (parsing_token) {
93*4882a593Smuzhiyun 				result->tokens[result->numargs] = arg + pos;
94*4882a593Smuzhiyun 			} else {
95*4882a593Smuzhiyun 				result->values[result->numargs] = arg + pos;
96*4882a593Smuzhiyun 				result->numargs++;
97*4882a593Smuzhiyun 			}
98*4882a593Smuzhiyun 			next_starts = false;
99*4882a593Smuzhiyun 		}
100*4882a593Smuzhiyun 		if (*(arg + pos) == '=') {
101*4882a593Smuzhiyun 			if (parsing_token)
102*4882a593Smuzhiyun 				parsing_token = false;
103*4882a593Smuzhiyun 			else
104*4882a593Smuzhiyun 				goto cleanup;
105*4882a593Smuzhiyun 			next_starts = true;
106*4882a593Smuzhiyun 			(*(arg + pos)) = '\0';
107*4882a593Smuzhiyun 		}
108*4882a593Smuzhiyun 		if (*(arg + pos) == ',') {
109*4882a593Smuzhiyun 			parsing_token = true;
110*4882a593Smuzhiyun 			next_starts = true;
111*4882a593Smuzhiyun 			(*(arg + pos)) = '\0';
112*4882a593Smuzhiyun 		}
113*4882a593Smuzhiyun 	}
114*4882a593Smuzhiyun 	return result;
115*4882a593Smuzhiyun cleanup:
116*4882a593Smuzhiyun 	printk(UM_KERN_ERR "vector_setup - Couldn't parse '%s'\n", arg);
117*4882a593Smuzhiyun 	kfree(result);
118*4882a593Smuzhiyun 	return NULL;
119*4882a593Smuzhiyun }
120*4882a593Smuzhiyun 
121*4882a593Smuzhiyun /*
122*4882a593Smuzhiyun  * Socket/FD configuration functions. These return an structure
123*4882a593Smuzhiyun  * of rx and tx descriptors to cover cases where these are not
124*4882a593Smuzhiyun  * the same (f.e. read via raw socket and write via tap).
125*4882a593Smuzhiyun  */
126*4882a593Smuzhiyun 
127*4882a593Smuzhiyun #define PATH_NET_TUN "/dev/net/tun"
128*4882a593Smuzhiyun 
129*4882a593Smuzhiyun 
create_tap_fd(char * iface)130*4882a593Smuzhiyun static int create_tap_fd(char *iface)
131*4882a593Smuzhiyun {
132*4882a593Smuzhiyun 	struct ifreq ifr;
133*4882a593Smuzhiyun 	int fd = -1;
134*4882a593Smuzhiyun 	int err = -ENOMEM, offload;
135*4882a593Smuzhiyun 
136*4882a593Smuzhiyun 	fd = open(PATH_NET_TUN, O_RDWR);
137*4882a593Smuzhiyun 	if (fd < 0) {
138*4882a593Smuzhiyun 		printk(UM_KERN_ERR "uml_tap: failed to open tun device\n");
139*4882a593Smuzhiyun 		goto tap_fd_cleanup;
140*4882a593Smuzhiyun 	}
141*4882a593Smuzhiyun 	memset(&ifr, 0, sizeof(ifr));
142*4882a593Smuzhiyun 	ifr.ifr_flags = IFF_TAP | IFF_NO_PI | IFF_VNET_HDR;
143*4882a593Smuzhiyun 	strncpy((char *)&ifr.ifr_name, iface, sizeof(ifr.ifr_name) - 1);
144*4882a593Smuzhiyun 
145*4882a593Smuzhiyun 	err = ioctl(fd, TUNSETIFF, (void *) &ifr);
146*4882a593Smuzhiyun 	if (err != 0) {
147*4882a593Smuzhiyun 		printk(UM_KERN_ERR "uml_tap: failed to select tap interface\n");
148*4882a593Smuzhiyun 		goto tap_fd_cleanup;
149*4882a593Smuzhiyun 	}
150*4882a593Smuzhiyun 
151*4882a593Smuzhiyun 	offload = TUN_F_CSUM | TUN_F_TSO4 | TUN_F_TSO6;
152*4882a593Smuzhiyun 	ioctl(fd, TUNSETOFFLOAD, offload);
153*4882a593Smuzhiyun 	return fd;
154*4882a593Smuzhiyun tap_fd_cleanup:
155*4882a593Smuzhiyun 	if (fd >= 0)
156*4882a593Smuzhiyun 		os_close_file(fd);
157*4882a593Smuzhiyun 	return err;
158*4882a593Smuzhiyun }
159*4882a593Smuzhiyun 
create_raw_fd(char * iface,int flags,int proto)160*4882a593Smuzhiyun static int create_raw_fd(char *iface, int flags, int proto)
161*4882a593Smuzhiyun {
162*4882a593Smuzhiyun 	struct ifreq ifr;
163*4882a593Smuzhiyun 	int fd = -1;
164*4882a593Smuzhiyun 	struct sockaddr_ll sock;
165*4882a593Smuzhiyun 	int err = -ENOMEM;
166*4882a593Smuzhiyun 
167*4882a593Smuzhiyun 	fd = socket(AF_PACKET, SOCK_RAW, flags);
168*4882a593Smuzhiyun 	if (fd == -1) {
169*4882a593Smuzhiyun 		err = -errno;
170*4882a593Smuzhiyun 		goto raw_fd_cleanup;
171*4882a593Smuzhiyun 	}
172*4882a593Smuzhiyun 	memset(&ifr, 0, sizeof(ifr));
173*4882a593Smuzhiyun 	strncpy((char *)&ifr.ifr_name, iface, sizeof(ifr.ifr_name) - 1);
174*4882a593Smuzhiyun 	if (ioctl(fd, SIOCGIFINDEX, (void *) &ifr) < 0) {
175*4882a593Smuzhiyun 		err = -errno;
176*4882a593Smuzhiyun 		goto raw_fd_cleanup;
177*4882a593Smuzhiyun 	}
178*4882a593Smuzhiyun 
179*4882a593Smuzhiyun 	sock.sll_family = AF_PACKET;
180*4882a593Smuzhiyun 	sock.sll_protocol = htons(proto);
181*4882a593Smuzhiyun 	sock.sll_ifindex = ifr.ifr_ifindex;
182*4882a593Smuzhiyun 
183*4882a593Smuzhiyun 	if (bind(fd,
184*4882a593Smuzhiyun 		(struct sockaddr *) &sock, sizeof(struct sockaddr_ll)) < 0) {
185*4882a593Smuzhiyun 		err = -errno;
186*4882a593Smuzhiyun 		goto raw_fd_cleanup;
187*4882a593Smuzhiyun 	}
188*4882a593Smuzhiyun 	return fd;
189*4882a593Smuzhiyun raw_fd_cleanup:
190*4882a593Smuzhiyun 	printk(UM_KERN_ERR "user_init_raw: init failed, error %d", err);
191*4882a593Smuzhiyun 	if (fd >= 0)
192*4882a593Smuzhiyun 		os_close_file(fd);
193*4882a593Smuzhiyun 	return err;
194*4882a593Smuzhiyun }
195*4882a593Smuzhiyun 
196*4882a593Smuzhiyun 
user_init_tap_fds(struct arglist * ifspec)197*4882a593Smuzhiyun static struct vector_fds *user_init_tap_fds(struct arglist *ifspec)
198*4882a593Smuzhiyun {
199*4882a593Smuzhiyun 	int fd = -1, i;
200*4882a593Smuzhiyun 	char *iface;
201*4882a593Smuzhiyun 	struct vector_fds *result = NULL;
202*4882a593Smuzhiyun 	bool dynamic = false;
203*4882a593Smuzhiyun 	char dynamic_ifname[IFNAMSIZ];
204*4882a593Smuzhiyun 	char *argv[] = {NULL, NULL, NULL, NULL};
205*4882a593Smuzhiyun 
206*4882a593Smuzhiyun 	iface = uml_vector_fetch_arg(ifspec, TOKEN_IFNAME);
207*4882a593Smuzhiyun 	if (iface == NULL) {
208*4882a593Smuzhiyun 		dynamic = true;
209*4882a593Smuzhiyun 		iface = dynamic_ifname;
210*4882a593Smuzhiyun 		srand(getpid());
211*4882a593Smuzhiyun 	}
212*4882a593Smuzhiyun 
213*4882a593Smuzhiyun 	result = uml_kmalloc(sizeof(struct vector_fds), UM_GFP_KERNEL);
214*4882a593Smuzhiyun 	if (result == NULL) {
215*4882a593Smuzhiyun 		printk(UM_KERN_ERR "uml_tap: failed to allocate file descriptors\n");
216*4882a593Smuzhiyun 		goto tap_cleanup;
217*4882a593Smuzhiyun 	}
218*4882a593Smuzhiyun 	result->rx_fd = -1;
219*4882a593Smuzhiyun 	result->tx_fd = -1;
220*4882a593Smuzhiyun 	result->remote_addr = NULL;
221*4882a593Smuzhiyun 	result->remote_addr_size = 0;
222*4882a593Smuzhiyun 
223*4882a593Smuzhiyun 	/* TAP */
224*4882a593Smuzhiyun 	do {
225*4882a593Smuzhiyun 		if (dynamic) {
226*4882a593Smuzhiyun 			strcpy(iface, template);
227*4882a593Smuzhiyun 			for (i = 0; i < strlen(iface); i++) {
228*4882a593Smuzhiyun 				if (iface[i] == 'X') {
229*4882a593Smuzhiyun 					iface[i] = padchar[rand() % strlen(padchar)];
230*4882a593Smuzhiyun 				}
231*4882a593Smuzhiyun 			}
232*4882a593Smuzhiyun 		}
233*4882a593Smuzhiyun 		fd = create_tap_fd(iface);
234*4882a593Smuzhiyun 		if ((fd < 0) && (!dynamic)) {
235*4882a593Smuzhiyun 			printk(UM_KERN_ERR "uml_tap: failed to create tun interface\n");
236*4882a593Smuzhiyun 			goto tap_cleanup;
237*4882a593Smuzhiyun 		}
238*4882a593Smuzhiyun 		result->tx_fd = fd;
239*4882a593Smuzhiyun 		result->rx_fd = fd;
240*4882a593Smuzhiyun 	} while (fd < 0);
241*4882a593Smuzhiyun 
242*4882a593Smuzhiyun 	argv[0] = uml_vector_fetch_arg(ifspec, TOKEN_SCRIPT);
243*4882a593Smuzhiyun 	if (argv[0]) {
244*4882a593Smuzhiyun 		argv[1] = iface;
245*4882a593Smuzhiyun 		run_helper(NULL, NULL, argv);
246*4882a593Smuzhiyun 	}
247*4882a593Smuzhiyun 
248*4882a593Smuzhiyun 	return result;
249*4882a593Smuzhiyun tap_cleanup:
250*4882a593Smuzhiyun 	printk(UM_KERN_ERR "user_init_tap: init failed, error %d", fd);
251*4882a593Smuzhiyun 	kfree(result);
252*4882a593Smuzhiyun 	return NULL;
253*4882a593Smuzhiyun }
254*4882a593Smuzhiyun 
user_init_hybrid_fds(struct arglist * ifspec)255*4882a593Smuzhiyun static struct vector_fds *user_init_hybrid_fds(struct arglist *ifspec)
256*4882a593Smuzhiyun {
257*4882a593Smuzhiyun 	char *iface;
258*4882a593Smuzhiyun 	struct vector_fds *result = NULL;
259*4882a593Smuzhiyun 	char *argv[] = {NULL, NULL, NULL, NULL};
260*4882a593Smuzhiyun 
261*4882a593Smuzhiyun 	iface = uml_vector_fetch_arg(ifspec, TOKEN_IFNAME);
262*4882a593Smuzhiyun 	if (iface == NULL) {
263*4882a593Smuzhiyun 		printk(UM_KERN_ERR "uml_tap: failed to parse interface spec\n");
264*4882a593Smuzhiyun 		goto hybrid_cleanup;
265*4882a593Smuzhiyun 	}
266*4882a593Smuzhiyun 
267*4882a593Smuzhiyun 	result = uml_kmalloc(sizeof(struct vector_fds), UM_GFP_KERNEL);
268*4882a593Smuzhiyun 	if (result == NULL) {
269*4882a593Smuzhiyun 		printk(UM_KERN_ERR "uml_tap: failed to allocate file descriptors\n");
270*4882a593Smuzhiyun 		goto hybrid_cleanup;
271*4882a593Smuzhiyun 	}
272*4882a593Smuzhiyun 	result->rx_fd = -1;
273*4882a593Smuzhiyun 	result->tx_fd = -1;
274*4882a593Smuzhiyun 	result->remote_addr = NULL;
275*4882a593Smuzhiyun 	result->remote_addr_size = 0;
276*4882a593Smuzhiyun 
277*4882a593Smuzhiyun 	/* TAP */
278*4882a593Smuzhiyun 
279*4882a593Smuzhiyun 	result->tx_fd = create_tap_fd(iface);
280*4882a593Smuzhiyun 	if (result->tx_fd < 0) {
281*4882a593Smuzhiyun 		printk(UM_KERN_ERR "uml_tap: failed to create tun interface: %i\n", result->tx_fd);
282*4882a593Smuzhiyun 		goto hybrid_cleanup;
283*4882a593Smuzhiyun 	}
284*4882a593Smuzhiyun 
285*4882a593Smuzhiyun 	/* RAW */
286*4882a593Smuzhiyun 
287*4882a593Smuzhiyun 	result->rx_fd = create_raw_fd(iface, ETH_P_ALL, ETH_P_ALL);
288*4882a593Smuzhiyun 	if (result->rx_fd == -1) {
289*4882a593Smuzhiyun 		printk(UM_KERN_ERR
290*4882a593Smuzhiyun 			"uml_tap: failed to create paired raw socket: %i\n", result->rx_fd);
291*4882a593Smuzhiyun 		goto hybrid_cleanup;
292*4882a593Smuzhiyun 	}
293*4882a593Smuzhiyun 
294*4882a593Smuzhiyun 	argv[0] = uml_vector_fetch_arg(ifspec, TOKEN_SCRIPT);
295*4882a593Smuzhiyun 	if (argv[0]) {
296*4882a593Smuzhiyun 		argv[1] = iface;
297*4882a593Smuzhiyun 		run_helper(NULL, NULL, argv);
298*4882a593Smuzhiyun 	}
299*4882a593Smuzhiyun 	return result;
300*4882a593Smuzhiyun hybrid_cleanup:
301*4882a593Smuzhiyun 	printk(UM_KERN_ERR "user_init_hybrid: init failed");
302*4882a593Smuzhiyun 	kfree(result);
303*4882a593Smuzhiyun 	return NULL;
304*4882a593Smuzhiyun }
305*4882a593Smuzhiyun 
user_init_unix_fds(struct arglist * ifspec,int id)306*4882a593Smuzhiyun static struct vector_fds *user_init_unix_fds(struct arglist *ifspec, int id)
307*4882a593Smuzhiyun {
308*4882a593Smuzhiyun 	int fd = -1;
309*4882a593Smuzhiyun 	int socktype;
310*4882a593Smuzhiyun 	char *src, *dst;
311*4882a593Smuzhiyun 	struct vector_fds *result = NULL;
312*4882a593Smuzhiyun 	struct sockaddr_un *local_addr = NULL, *remote_addr = NULL;
313*4882a593Smuzhiyun 
314*4882a593Smuzhiyun 	src = uml_vector_fetch_arg(ifspec, "src");
315*4882a593Smuzhiyun 	dst = uml_vector_fetch_arg(ifspec, "dst");
316*4882a593Smuzhiyun 	result = uml_kmalloc(sizeof(struct vector_fds), UM_GFP_KERNEL);
317*4882a593Smuzhiyun 	if (result == NULL) {
318*4882a593Smuzhiyun 		printk(UM_KERN_ERR "unix open:cannot allocate remote addr");
319*4882a593Smuzhiyun 		goto unix_cleanup;
320*4882a593Smuzhiyun 	}
321*4882a593Smuzhiyun 	remote_addr = uml_kmalloc(sizeof(struct sockaddr_un), UM_GFP_KERNEL);
322*4882a593Smuzhiyun 	if (remote_addr == NULL) {
323*4882a593Smuzhiyun 		printk(UM_KERN_ERR "unix open:cannot allocate remote addr");
324*4882a593Smuzhiyun 		goto unix_cleanup;
325*4882a593Smuzhiyun 	}
326*4882a593Smuzhiyun 
327*4882a593Smuzhiyun 	switch (id) {
328*4882a593Smuzhiyun 	case ID_BESS:
329*4882a593Smuzhiyun 		socktype = SOCK_SEQPACKET;
330*4882a593Smuzhiyun 		if ((src != NULL) && (strlen(src) <= MAX_UN_LEN)) {
331*4882a593Smuzhiyun 			local_addr = uml_kmalloc(sizeof(struct sockaddr_un), UM_GFP_KERNEL);
332*4882a593Smuzhiyun 			if (local_addr == NULL) {
333*4882a593Smuzhiyun 				printk(UM_KERN_ERR "bess open:cannot allocate local addr");
334*4882a593Smuzhiyun 				goto unix_cleanup;
335*4882a593Smuzhiyun 			}
336*4882a593Smuzhiyun 			local_addr->sun_family = AF_UNIX;
337*4882a593Smuzhiyun 			memcpy(local_addr->sun_path, src, strlen(src) + 1);
338*4882a593Smuzhiyun 		}
339*4882a593Smuzhiyun 		if ((dst == NULL) || (strlen(dst) > MAX_UN_LEN))
340*4882a593Smuzhiyun 			goto unix_cleanup;
341*4882a593Smuzhiyun 		remote_addr->sun_family = AF_UNIX;
342*4882a593Smuzhiyun 		memcpy(remote_addr->sun_path, dst, strlen(dst) + 1);
343*4882a593Smuzhiyun 		break;
344*4882a593Smuzhiyun 	default:
345*4882a593Smuzhiyun 		printk(KERN_ERR "Unsupported unix socket type\n");
346*4882a593Smuzhiyun 		return NULL;
347*4882a593Smuzhiyun 	}
348*4882a593Smuzhiyun 
349*4882a593Smuzhiyun 	fd = socket(AF_UNIX, socktype, 0);
350*4882a593Smuzhiyun 	if (fd == -1) {
351*4882a593Smuzhiyun 		printk(UM_KERN_ERR
352*4882a593Smuzhiyun 			"unix open: could not open socket, error = %d",
353*4882a593Smuzhiyun 			-errno
354*4882a593Smuzhiyun 		);
355*4882a593Smuzhiyun 		goto unix_cleanup;
356*4882a593Smuzhiyun 	}
357*4882a593Smuzhiyun 	if (local_addr != NULL) {
358*4882a593Smuzhiyun 		if (bind(fd, (struct sockaddr *) local_addr, sizeof(struct sockaddr_un))) {
359*4882a593Smuzhiyun 			printk(UM_KERN_ERR UNIX_BIND_FAIL, errno);
360*4882a593Smuzhiyun 			goto unix_cleanup;
361*4882a593Smuzhiyun 		}
362*4882a593Smuzhiyun 	}
363*4882a593Smuzhiyun 	switch (id) {
364*4882a593Smuzhiyun 	case ID_BESS:
365*4882a593Smuzhiyun 		if (connect(fd, (const struct sockaddr *) remote_addr, sizeof(struct sockaddr_un)) < 0) {
366*4882a593Smuzhiyun 			printk(UM_KERN_ERR "bess open:cannot connect to %s %i", remote_addr->sun_path, -errno);
367*4882a593Smuzhiyun 			goto unix_cleanup;
368*4882a593Smuzhiyun 		}
369*4882a593Smuzhiyun 		break;
370*4882a593Smuzhiyun 	}
371*4882a593Smuzhiyun 	result->rx_fd = fd;
372*4882a593Smuzhiyun 	result->tx_fd = fd;
373*4882a593Smuzhiyun 	result->remote_addr_size = sizeof(struct sockaddr_un);
374*4882a593Smuzhiyun 	result->remote_addr = remote_addr;
375*4882a593Smuzhiyun 	return result;
376*4882a593Smuzhiyun unix_cleanup:
377*4882a593Smuzhiyun 	if (fd >= 0)
378*4882a593Smuzhiyun 		os_close_file(fd);
379*4882a593Smuzhiyun 	kfree(remote_addr);
380*4882a593Smuzhiyun 	kfree(result);
381*4882a593Smuzhiyun 	return NULL;
382*4882a593Smuzhiyun }
383*4882a593Smuzhiyun 
strtofd(const char * nptr)384*4882a593Smuzhiyun static int strtofd(const char *nptr)
385*4882a593Smuzhiyun {
386*4882a593Smuzhiyun 	long fd;
387*4882a593Smuzhiyun 	char *endptr;
388*4882a593Smuzhiyun 
389*4882a593Smuzhiyun 	if (nptr == NULL)
390*4882a593Smuzhiyun 		return -1;
391*4882a593Smuzhiyun 
392*4882a593Smuzhiyun 	errno = 0;
393*4882a593Smuzhiyun 	fd = strtol(nptr, &endptr, 10);
394*4882a593Smuzhiyun 	if (nptr == endptr ||
395*4882a593Smuzhiyun 		errno != 0 ||
396*4882a593Smuzhiyun 		*endptr != '\0' ||
397*4882a593Smuzhiyun 		fd < 0 ||
398*4882a593Smuzhiyun 		fd > INT_MAX) {
399*4882a593Smuzhiyun 		return -1;
400*4882a593Smuzhiyun 	}
401*4882a593Smuzhiyun 	return fd;
402*4882a593Smuzhiyun }
403*4882a593Smuzhiyun 
user_init_fd_fds(struct arglist * ifspec)404*4882a593Smuzhiyun static struct vector_fds *user_init_fd_fds(struct arglist *ifspec)
405*4882a593Smuzhiyun {
406*4882a593Smuzhiyun 	int fd = -1;
407*4882a593Smuzhiyun 	char *fdarg = NULL;
408*4882a593Smuzhiyun 	struct vector_fds *result = NULL;
409*4882a593Smuzhiyun 
410*4882a593Smuzhiyun 	fdarg = uml_vector_fetch_arg(ifspec, "fd");
411*4882a593Smuzhiyun 	fd = strtofd(fdarg);
412*4882a593Smuzhiyun 	if (fd == -1) {
413*4882a593Smuzhiyun 		printk(UM_KERN_ERR "fd open: bad or missing fd argument");
414*4882a593Smuzhiyun 		goto fd_cleanup;
415*4882a593Smuzhiyun 	}
416*4882a593Smuzhiyun 
417*4882a593Smuzhiyun 	result = uml_kmalloc(sizeof(struct vector_fds), UM_GFP_KERNEL);
418*4882a593Smuzhiyun 	if (result == NULL) {
419*4882a593Smuzhiyun 		printk(UM_KERN_ERR "fd open: allocation failed");
420*4882a593Smuzhiyun 		goto fd_cleanup;
421*4882a593Smuzhiyun 	}
422*4882a593Smuzhiyun 
423*4882a593Smuzhiyun 	result->rx_fd = fd;
424*4882a593Smuzhiyun 	result->tx_fd = fd;
425*4882a593Smuzhiyun 	result->remote_addr_size = 0;
426*4882a593Smuzhiyun 	result->remote_addr = NULL;
427*4882a593Smuzhiyun 	return result;
428*4882a593Smuzhiyun 
429*4882a593Smuzhiyun fd_cleanup:
430*4882a593Smuzhiyun 	if (fd >= 0)
431*4882a593Smuzhiyun 		os_close_file(fd);
432*4882a593Smuzhiyun 	kfree(result);
433*4882a593Smuzhiyun 	return NULL;
434*4882a593Smuzhiyun }
435*4882a593Smuzhiyun 
user_init_raw_fds(struct arglist * ifspec)436*4882a593Smuzhiyun static struct vector_fds *user_init_raw_fds(struct arglist *ifspec)
437*4882a593Smuzhiyun {
438*4882a593Smuzhiyun 	int rxfd = -1, txfd = -1;
439*4882a593Smuzhiyun 	int err = -ENOMEM;
440*4882a593Smuzhiyun 	char *iface;
441*4882a593Smuzhiyun 	struct vector_fds *result = NULL;
442*4882a593Smuzhiyun 	char *argv[] = {NULL, NULL, NULL, NULL};
443*4882a593Smuzhiyun 
444*4882a593Smuzhiyun 	iface = uml_vector_fetch_arg(ifspec, TOKEN_IFNAME);
445*4882a593Smuzhiyun 	if (iface == NULL)
446*4882a593Smuzhiyun 		goto raw_cleanup;
447*4882a593Smuzhiyun 
448*4882a593Smuzhiyun 	rxfd = create_raw_fd(iface, ETH_P_ALL, ETH_P_ALL);
449*4882a593Smuzhiyun 	if (rxfd == -1) {
450*4882a593Smuzhiyun 		err = -errno;
451*4882a593Smuzhiyun 		goto raw_cleanup;
452*4882a593Smuzhiyun 	}
453*4882a593Smuzhiyun 	txfd = create_raw_fd(iface, 0, ETH_P_IP); /* Turn off RX on this fd */
454*4882a593Smuzhiyun 	if (txfd == -1) {
455*4882a593Smuzhiyun 		err = -errno;
456*4882a593Smuzhiyun 		goto raw_cleanup;
457*4882a593Smuzhiyun 	}
458*4882a593Smuzhiyun 	result = uml_kmalloc(sizeof(struct vector_fds), UM_GFP_KERNEL);
459*4882a593Smuzhiyun 	if (result != NULL) {
460*4882a593Smuzhiyun 		result->rx_fd = rxfd;
461*4882a593Smuzhiyun 		result->tx_fd = txfd;
462*4882a593Smuzhiyun 		result->remote_addr = NULL;
463*4882a593Smuzhiyun 		result->remote_addr_size = 0;
464*4882a593Smuzhiyun 	}
465*4882a593Smuzhiyun 	argv[0] = uml_vector_fetch_arg(ifspec, TOKEN_SCRIPT);
466*4882a593Smuzhiyun 	if (argv[0]) {
467*4882a593Smuzhiyun 		argv[1] = iface;
468*4882a593Smuzhiyun 		run_helper(NULL, NULL, argv);
469*4882a593Smuzhiyun 	}
470*4882a593Smuzhiyun 	return result;
471*4882a593Smuzhiyun raw_cleanup:
472*4882a593Smuzhiyun 	printk(UM_KERN_ERR "user_init_raw: init failed, error %d", err);
473*4882a593Smuzhiyun 	kfree(result);
474*4882a593Smuzhiyun 	return NULL;
475*4882a593Smuzhiyun }
476*4882a593Smuzhiyun 
477*4882a593Smuzhiyun 
uml_raw_enable_qdisc_bypass(int fd)478*4882a593Smuzhiyun bool uml_raw_enable_qdisc_bypass(int fd)
479*4882a593Smuzhiyun {
480*4882a593Smuzhiyun 	int optval = 1;
481*4882a593Smuzhiyun 
482*4882a593Smuzhiyun 	if (setsockopt(fd,
483*4882a593Smuzhiyun 		SOL_PACKET, PACKET_QDISC_BYPASS,
484*4882a593Smuzhiyun 		&optval, sizeof(optval)) != 0) {
485*4882a593Smuzhiyun 		return false;
486*4882a593Smuzhiyun 	}
487*4882a593Smuzhiyun 	return true;
488*4882a593Smuzhiyun }
489*4882a593Smuzhiyun 
uml_raw_enable_vnet_headers(int fd)490*4882a593Smuzhiyun bool uml_raw_enable_vnet_headers(int fd)
491*4882a593Smuzhiyun {
492*4882a593Smuzhiyun 	int optval = 1;
493*4882a593Smuzhiyun 
494*4882a593Smuzhiyun 	if (setsockopt(fd,
495*4882a593Smuzhiyun 		SOL_PACKET, PACKET_VNET_HDR,
496*4882a593Smuzhiyun 		&optval, sizeof(optval)) != 0) {
497*4882a593Smuzhiyun 		printk(UM_KERN_INFO VNET_HDR_FAIL, fd);
498*4882a593Smuzhiyun 		return false;
499*4882a593Smuzhiyun 	}
500*4882a593Smuzhiyun 	return true;
501*4882a593Smuzhiyun }
uml_tap_enable_vnet_headers(int fd)502*4882a593Smuzhiyun bool uml_tap_enable_vnet_headers(int fd)
503*4882a593Smuzhiyun {
504*4882a593Smuzhiyun 	unsigned int features;
505*4882a593Smuzhiyun 	int len = sizeof(struct virtio_net_hdr);
506*4882a593Smuzhiyun 
507*4882a593Smuzhiyun 	if (ioctl(fd, TUNGETFEATURES, &features) == -1) {
508*4882a593Smuzhiyun 		printk(UM_KERN_INFO TUN_GET_F_FAIL, strerror(errno));
509*4882a593Smuzhiyun 		return false;
510*4882a593Smuzhiyun 	}
511*4882a593Smuzhiyun 	if ((features & IFF_VNET_HDR) == 0) {
512*4882a593Smuzhiyun 		printk(UM_KERN_INFO "tapraw: No VNET HEADER support");
513*4882a593Smuzhiyun 		return false;
514*4882a593Smuzhiyun 	}
515*4882a593Smuzhiyun 	ioctl(fd, TUNSETVNETHDRSZ, &len);
516*4882a593Smuzhiyun 	return true;
517*4882a593Smuzhiyun }
518*4882a593Smuzhiyun 
user_init_socket_fds(struct arglist * ifspec,int id)519*4882a593Smuzhiyun static struct vector_fds *user_init_socket_fds(struct arglist *ifspec, int id)
520*4882a593Smuzhiyun {
521*4882a593Smuzhiyun 	int err = -ENOMEM;
522*4882a593Smuzhiyun 	int fd = -1, gairet;
523*4882a593Smuzhiyun 	struct addrinfo srchints;
524*4882a593Smuzhiyun 	struct addrinfo dsthints;
525*4882a593Smuzhiyun 	bool v6, udp;
526*4882a593Smuzhiyun 	char *value;
527*4882a593Smuzhiyun 	char *src, *dst, *srcport, *dstport;
528*4882a593Smuzhiyun 	struct addrinfo *gairesult = NULL;
529*4882a593Smuzhiyun 	struct vector_fds *result = NULL;
530*4882a593Smuzhiyun 
531*4882a593Smuzhiyun 
532*4882a593Smuzhiyun 	value = uml_vector_fetch_arg(ifspec, "v6");
533*4882a593Smuzhiyun 	v6 = false;
534*4882a593Smuzhiyun 	udp = false;
535*4882a593Smuzhiyun 	if (value != NULL) {
536*4882a593Smuzhiyun 		if (strtol((const char *) value, NULL, 10) > 0)
537*4882a593Smuzhiyun 			v6 = true;
538*4882a593Smuzhiyun 	}
539*4882a593Smuzhiyun 
540*4882a593Smuzhiyun 	value = uml_vector_fetch_arg(ifspec, "udp");
541*4882a593Smuzhiyun 	if (value != NULL) {
542*4882a593Smuzhiyun 		if (strtol((const char *) value, NULL, 10) > 0)
543*4882a593Smuzhiyun 			udp = true;
544*4882a593Smuzhiyun 	}
545*4882a593Smuzhiyun 	src = uml_vector_fetch_arg(ifspec, "src");
546*4882a593Smuzhiyun 	dst = uml_vector_fetch_arg(ifspec, "dst");
547*4882a593Smuzhiyun 	srcport = uml_vector_fetch_arg(ifspec, "srcport");
548*4882a593Smuzhiyun 	dstport = uml_vector_fetch_arg(ifspec, "dstport");
549*4882a593Smuzhiyun 
550*4882a593Smuzhiyun 	memset(&dsthints, 0, sizeof(dsthints));
551*4882a593Smuzhiyun 
552*4882a593Smuzhiyun 	if (v6)
553*4882a593Smuzhiyun 		dsthints.ai_family = AF_INET6;
554*4882a593Smuzhiyun 	else
555*4882a593Smuzhiyun 		dsthints.ai_family = AF_INET;
556*4882a593Smuzhiyun 
557*4882a593Smuzhiyun 	switch (id) {
558*4882a593Smuzhiyun 	case ID_GRE:
559*4882a593Smuzhiyun 		dsthints.ai_socktype = SOCK_RAW;
560*4882a593Smuzhiyun 		dsthints.ai_protocol = IPPROTO_GRE;
561*4882a593Smuzhiyun 		break;
562*4882a593Smuzhiyun 	case ID_L2TPV3:
563*4882a593Smuzhiyun 		if (udp) {
564*4882a593Smuzhiyun 			dsthints.ai_socktype = SOCK_DGRAM;
565*4882a593Smuzhiyun 			dsthints.ai_protocol = 0;
566*4882a593Smuzhiyun 		} else {
567*4882a593Smuzhiyun 			dsthints.ai_socktype = SOCK_RAW;
568*4882a593Smuzhiyun 			dsthints.ai_protocol = IPPROTO_L2TP;
569*4882a593Smuzhiyun 		}
570*4882a593Smuzhiyun 		break;
571*4882a593Smuzhiyun 	default:
572*4882a593Smuzhiyun 		printk(KERN_ERR "Unsupported socket type\n");
573*4882a593Smuzhiyun 		return NULL;
574*4882a593Smuzhiyun 	}
575*4882a593Smuzhiyun 	memcpy(&srchints, &dsthints, sizeof(struct addrinfo));
576*4882a593Smuzhiyun 
577*4882a593Smuzhiyun 	gairet = getaddrinfo(src, srcport, &dsthints, &gairesult);
578*4882a593Smuzhiyun 	if ((gairet != 0) || (gairesult == NULL)) {
579*4882a593Smuzhiyun 		printk(UM_KERN_ERR
580*4882a593Smuzhiyun 			"socket_open : could not resolve src, error = %s",
581*4882a593Smuzhiyun 			gai_strerror(gairet)
582*4882a593Smuzhiyun 		);
583*4882a593Smuzhiyun 		return NULL;
584*4882a593Smuzhiyun 	}
585*4882a593Smuzhiyun 	fd = socket(gairesult->ai_family,
586*4882a593Smuzhiyun 		gairesult->ai_socktype, gairesult->ai_protocol);
587*4882a593Smuzhiyun 	if (fd == -1) {
588*4882a593Smuzhiyun 		printk(UM_KERN_ERR
589*4882a593Smuzhiyun 			"socket_open : could not open socket, error = %d",
590*4882a593Smuzhiyun 			-errno
591*4882a593Smuzhiyun 		);
592*4882a593Smuzhiyun 		goto cleanup;
593*4882a593Smuzhiyun 	}
594*4882a593Smuzhiyun 	if (bind(fd,
595*4882a593Smuzhiyun 		(struct sockaddr *) gairesult->ai_addr,
596*4882a593Smuzhiyun 		gairesult->ai_addrlen)) {
597*4882a593Smuzhiyun 		printk(UM_KERN_ERR L2TPV3_BIND_FAIL, errno);
598*4882a593Smuzhiyun 		goto cleanup;
599*4882a593Smuzhiyun 	}
600*4882a593Smuzhiyun 
601*4882a593Smuzhiyun 	if (gairesult != NULL)
602*4882a593Smuzhiyun 		freeaddrinfo(gairesult);
603*4882a593Smuzhiyun 
604*4882a593Smuzhiyun 	gairesult = NULL;
605*4882a593Smuzhiyun 
606*4882a593Smuzhiyun 	gairet = getaddrinfo(dst, dstport, &dsthints, &gairesult);
607*4882a593Smuzhiyun 	if ((gairet != 0) || (gairesult == NULL)) {
608*4882a593Smuzhiyun 		printk(UM_KERN_ERR
609*4882a593Smuzhiyun 			"socket_open : could not resolve dst, error = %s",
610*4882a593Smuzhiyun 			gai_strerror(gairet)
611*4882a593Smuzhiyun 		);
612*4882a593Smuzhiyun 		return NULL;
613*4882a593Smuzhiyun 	}
614*4882a593Smuzhiyun 
615*4882a593Smuzhiyun 	result = uml_kmalloc(sizeof(struct vector_fds), UM_GFP_KERNEL);
616*4882a593Smuzhiyun 	if (result != NULL) {
617*4882a593Smuzhiyun 		result->rx_fd = fd;
618*4882a593Smuzhiyun 		result->tx_fd = fd;
619*4882a593Smuzhiyun 		result->remote_addr = uml_kmalloc(
620*4882a593Smuzhiyun 			gairesult->ai_addrlen, UM_GFP_KERNEL);
621*4882a593Smuzhiyun 		if (result->remote_addr == NULL)
622*4882a593Smuzhiyun 			goto cleanup;
623*4882a593Smuzhiyun 		result->remote_addr_size = gairesult->ai_addrlen;
624*4882a593Smuzhiyun 		memcpy(
625*4882a593Smuzhiyun 			result->remote_addr,
626*4882a593Smuzhiyun 			gairesult->ai_addr,
627*4882a593Smuzhiyun 			gairesult->ai_addrlen
628*4882a593Smuzhiyun 		);
629*4882a593Smuzhiyun 	}
630*4882a593Smuzhiyun 	freeaddrinfo(gairesult);
631*4882a593Smuzhiyun 	return result;
632*4882a593Smuzhiyun cleanup:
633*4882a593Smuzhiyun 	if (gairesult != NULL)
634*4882a593Smuzhiyun 		freeaddrinfo(gairesult);
635*4882a593Smuzhiyun 	printk(UM_KERN_ERR "user_init_socket: init failed, error %d", err);
636*4882a593Smuzhiyun 	if (fd >= 0)
637*4882a593Smuzhiyun 		os_close_file(fd);
638*4882a593Smuzhiyun 	if (result != NULL) {
639*4882a593Smuzhiyun 		kfree(result->remote_addr);
640*4882a593Smuzhiyun 		kfree(result);
641*4882a593Smuzhiyun 	}
642*4882a593Smuzhiyun 	return NULL;
643*4882a593Smuzhiyun }
644*4882a593Smuzhiyun 
uml_vector_user_open(int unit,struct arglist * parsed)645*4882a593Smuzhiyun struct vector_fds *uml_vector_user_open(
646*4882a593Smuzhiyun 	int unit,
647*4882a593Smuzhiyun 	struct arglist *parsed
648*4882a593Smuzhiyun )
649*4882a593Smuzhiyun {
650*4882a593Smuzhiyun 	char *transport;
651*4882a593Smuzhiyun 
652*4882a593Smuzhiyun 	if (parsed == NULL) {
653*4882a593Smuzhiyun 		printk(UM_KERN_ERR "no parsed config for unit %d\n", unit);
654*4882a593Smuzhiyun 		return NULL;
655*4882a593Smuzhiyun 	}
656*4882a593Smuzhiyun 	transport = uml_vector_fetch_arg(parsed, "transport");
657*4882a593Smuzhiyun 	if (transport == NULL) {
658*4882a593Smuzhiyun 		printk(UM_KERN_ERR "missing transport for unit %d\n", unit);
659*4882a593Smuzhiyun 		return NULL;
660*4882a593Smuzhiyun 	}
661*4882a593Smuzhiyun 	if (strncmp(transport, TRANS_RAW, TRANS_RAW_LEN) == 0)
662*4882a593Smuzhiyun 		return user_init_raw_fds(parsed);
663*4882a593Smuzhiyun 	if (strncmp(transport, TRANS_HYBRID, TRANS_HYBRID_LEN) == 0)
664*4882a593Smuzhiyun 		return user_init_hybrid_fds(parsed);
665*4882a593Smuzhiyun 	if (strncmp(transport, TRANS_TAP, TRANS_TAP_LEN) == 0)
666*4882a593Smuzhiyun 		return user_init_tap_fds(parsed);
667*4882a593Smuzhiyun 	if (strncmp(transport, TRANS_GRE, TRANS_GRE_LEN) == 0)
668*4882a593Smuzhiyun 		return user_init_socket_fds(parsed, ID_GRE);
669*4882a593Smuzhiyun 	if (strncmp(transport, TRANS_L2TPV3, TRANS_L2TPV3_LEN) == 0)
670*4882a593Smuzhiyun 		return user_init_socket_fds(parsed, ID_L2TPV3);
671*4882a593Smuzhiyun 	if (strncmp(transport, TRANS_BESS, TRANS_BESS_LEN) == 0)
672*4882a593Smuzhiyun 		return user_init_unix_fds(parsed, ID_BESS);
673*4882a593Smuzhiyun 	if (strncmp(transport, TRANS_FD, TRANS_FD_LEN) == 0)
674*4882a593Smuzhiyun 		return user_init_fd_fds(parsed);
675*4882a593Smuzhiyun 	return NULL;
676*4882a593Smuzhiyun }
677*4882a593Smuzhiyun 
678*4882a593Smuzhiyun 
uml_vector_sendmsg(int fd,void * hdr,int flags)679*4882a593Smuzhiyun int uml_vector_sendmsg(int fd, void *hdr, int flags)
680*4882a593Smuzhiyun {
681*4882a593Smuzhiyun 	int n;
682*4882a593Smuzhiyun 
683*4882a593Smuzhiyun 	CATCH_EINTR(n = sendmsg(fd, (struct msghdr *) hdr,  flags));
684*4882a593Smuzhiyun 	if ((n < 0) && (errno == EAGAIN))
685*4882a593Smuzhiyun 		return 0;
686*4882a593Smuzhiyun 	if (n >= 0)
687*4882a593Smuzhiyun 		return n;
688*4882a593Smuzhiyun 	else
689*4882a593Smuzhiyun 		return -errno;
690*4882a593Smuzhiyun }
691*4882a593Smuzhiyun 
uml_vector_recvmsg(int fd,void * hdr,int flags)692*4882a593Smuzhiyun int uml_vector_recvmsg(int fd, void *hdr, int flags)
693*4882a593Smuzhiyun {
694*4882a593Smuzhiyun 	int n;
695*4882a593Smuzhiyun 	struct msghdr *msg = (struct msghdr *) hdr;
696*4882a593Smuzhiyun 
697*4882a593Smuzhiyun 	CATCH_EINTR(n = readv(fd, msg->msg_iov, msg->msg_iovlen));
698*4882a593Smuzhiyun 	if ((n < 0) && (errno == EAGAIN))
699*4882a593Smuzhiyun 		return 0;
700*4882a593Smuzhiyun 	if (n >= 0)
701*4882a593Smuzhiyun 		return n;
702*4882a593Smuzhiyun 	else
703*4882a593Smuzhiyun 		return -errno;
704*4882a593Smuzhiyun }
705*4882a593Smuzhiyun 
uml_vector_writev(int fd,void * hdr,int iovcount)706*4882a593Smuzhiyun int uml_vector_writev(int fd, void *hdr, int iovcount)
707*4882a593Smuzhiyun {
708*4882a593Smuzhiyun 	int n;
709*4882a593Smuzhiyun 
710*4882a593Smuzhiyun 	CATCH_EINTR(n = writev(fd, (struct iovec *) hdr,  iovcount));
711*4882a593Smuzhiyun 	if ((n < 0) && ((errno == EAGAIN) || (errno == ENOBUFS)))
712*4882a593Smuzhiyun 		return 0;
713*4882a593Smuzhiyun 	if (n >= 0)
714*4882a593Smuzhiyun 		return n;
715*4882a593Smuzhiyun 	else
716*4882a593Smuzhiyun 		return -errno;
717*4882a593Smuzhiyun }
718*4882a593Smuzhiyun 
uml_vector_sendmmsg(int fd,void * msgvec,unsigned int vlen,unsigned int flags)719*4882a593Smuzhiyun int uml_vector_sendmmsg(
720*4882a593Smuzhiyun 	int fd,
721*4882a593Smuzhiyun 	void *msgvec,
722*4882a593Smuzhiyun 	unsigned int vlen,
723*4882a593Smuzhiyun 	unsigned int flags)
724*4882a593Smuzhiyun {
725*4882a593Smuzhiyun 	int n;
726*4882a593Smuzhiyun 
727*4882a593Smuzhiyun 	CATCH_EINTR(n = sendmmsg(fd, (struct mmsghdr *) msgvec, vlen, flags));
728*4882a593Smuzhiyun 	if ((n < 0) && ((errno == EAGAIN) || (errno == ENOBUFS)))
729*4882a593Smuzhiyun 		return 0;
730*4882a593Smuzhiyun 	if (n >= 0)
731*4882a593Smuzhiyun 		return n;
732*4882a593Smuzhiyun 	else
733*4882a593Smuzhiyun 		return -errno;
734*4882a593Smuzhiyun }
735*4882a593Smuzhiyun 
uml_vector_recvmmsg(int fd,void * msgvec,unsigned int vlen,unsigned int flags)736*4882a593Smuzhiyun int uml_vector_recvmmsg(
737*4882a593Smuzhiyun 	int fd,
738*4882a593Smuzhiyun 	void *msgvec,
739*4882a593Smuzhiyun 	unsigned int vlen,
740*4882a593Smuzhiyun 	unsigned int flags)
741*4882a593Smuzhiyun {
742*4882a593Smuzhiyun 	int n;
743*4882a593Smuzhiyun 
744*4882a593Smuzhiyun 	CATCH_EINTR(
745*4882a593Smuzhiyun 		n = recvmmsg(fd, (struct mmsghdr *) msgvec, vlen, flags, 0));
746*4882a593Smuzhiyun 	if ((n < 0) && (errno == EAGAIN))
747*4882a593Smuzhiyun 		return 0;
748*4882a593Smuzhiyun 	if (n >= 0)
749*4882a593Smuzhiyun 		return n;
750*4882a593Smuzhiyun 	else
751*4882a593Smuzhiyun 		return -errno;
752*4882a593Smuzhiyun }
uml_vector_attach_bpf(int fd,void * bpf)753*4882a593Smuzhiyun int uml_vector_attach_bpf(int fd, void *bpf)
754*4882a593Smuzhiyun {
755*4882a593Smuzhiyun 	struct sock_fprog *prog = bpf;
756*4882a593Smuzhiyun 
757*4882a593Smuzhiyun 	int err = setsockopt(fd, SOL_SOCKET, SO_ATTACH_FILTER, bpf, sizeof(struct sock_fprog));
758*4882a593Smuzhiyun 
759*4882a593Smuzhiyun 	if (err < 0)
760*4882a593Smuzhiyun 		printk(KERN_ERR BPF_ATTACH_FAIL, prog->len, prog->filter, fd, -errno);
761*4882a593Smuzhiyun 	return err;
762*4882a593Smuzhiyun }
763*4882a593Smuzhiyun 
uml_vector_detach_bpf(int fd,void * bpf)764*4882a593Smuzhiyun int uml_vector_detach_bpf(int fd, void *bpf)
765*4882a593Smuzhiyun {
766*4882a593Smuzhiyun 	struct sock_fprog *prog = bpf;
767*4882a593Smuzhiyun 
768*4882a593Smuzhiyun 	int err = setsockopt(fd, SOL_SOCKET, SO_DETACH_FILTER, bpf, sizeof(struct sock_fprog));
769*4882a593Smuzhiyun 	if (err < 0)
770*4882a593Smuzhiyun 		printk(KERN_ERR BPF_DETACH_FAIL, prog->len, prog->filter, fd, -errno);
771*4882a593Smuzhiyun 	return err;
772*4882a593Smuzhiyun }
uml_vector_default_bpf(void * mac)773*4882a593Smuzhiyun void *uml_vector_default_bpf(void *mac)
774*4882a593Smuzhiyun {
775*4882a593Smuzhiyun 	struct sock_filter *bpf;
776*4882a593Smuzhiyun 	uint32_t *mac1 = (uint32_t *)(mac + 2);
777*4882a593Smuzhiyun 	uint16_t *mac2 = (uint16_t *) mac;
778*4882a593Smuzhiyun 	struct sock_fprog *bpf_prog;
779*4882a593Smuzhiyun 
780*4882a593Smuzhiyun 	bpf_prog = uml_kmalloc(sizeof(struct sock_fprog), UM_GFP_KERNEL);
781*4882a593Smuzhiyun 	if (bpf_prog) {
782*4882a593Smuzhiyun 		bpf_prog->len = DEFAULT_BPF_LEN;
783*4882a593Smuzhiyun 		bpf_prog->filter = NULL;
784*4882a593Smuzhiyun 	} else {
785*4882a593Smuzhiyun 		return NULL;
786*4882a593Smuzhiyun 	}
787*4882a593Smuzhiyun 	bpf = uml_kmalloc(
788*4882a593Smuzhiyun 		sizeof(struct sock_filter) * DEFAULT_BPF_LEN, UM_GFP_KERNEL);
789*4882a593Smuzhiyun 	if (bpf) {
790*4882a593Smuzhiyun 		bpf_prog->filter = bpf;
791*4882a593Smuzhiyun 		/* ld	[8] */
792*4882a593Smuzhiyun 		bpf[0] = (struct sock_filter){ 0x20, 0, 0, 0x00000008 };
793*4882a593Smuzhiyun 		/* jeq	#0xMAC[2-6] jt 2 jf 5*/
794*4882a593Smuzhiyun 		bpf[1] = (struct sock_filter){ 0x15, 0, 3, ntohl(*mac1)};
795*4882a593Smuzhiyun 		/* ldh	[6] */
796*4882a593Smuzhiyun 		bpf[2] = (struct sock_filter){ 0x28, 0, 0, 0x00000006 };
797*4882a593Smuzhiyun 		/* jeq	#0xMAC[0-1] jt 4 jf 5 */
798*4882a593Smuzhiyun 		bpf[3] = (struct sock_filter){ 0x15, 0, 1, ntohs(*mac2)};
799*4882a593Smuzhiyun 		/* ret	#0 */
800*4882a593Smuzhiyun 		bpf[4] = (struct sock_filter){ 0x6, 0, 0, 0x00000000 };
801*4882a593Smuzhiyun 		/* ret	#0x40000 */
802*4882a593Smuzhiyun 		bpf[5] = (struct sock_filter){ 0x6, 0, 0, 0x00040000 };
803*4882a593Smuzhiyun 	} else {
804*4882a593Smuzhiyun 		kfree(bpf_prog);
805*4882a593Smuzhiyun 		bpf_prog = NULL;
806*4882a593Smuzhiyun 	}
807*4882a593Smuzhiyun 	return bpf_prog;
808*4882a593Smuzhiyun }
809*4882a593Smuzhiyun 
810*4882a593Smuzhiyun /* Note - this function requires a valid mac being passed as an arg */
811*4882a593Smuzhiyun 
uml_vector_user_bpf(char * filename)812*4882a593Smuzhiyun void *uml_vector_user_bpf(char *filename)
813*4882a593Smuzhiyun {
814*4882a593Smuzhiyun 	struct sock_filter *bpf;
815*4882a593Smuzhiyun 	struct sock_fprog *bpf_prog;
816*4882a593Smuzhiyun 	struct stat statbuf;
817*4882a593Smuzhiyun 	int res, ffd = -1;
818*4882a593Smuzhiyun 
819*4882a593Smuzhiyun 	if (filename == NULL)
820*4882a593Smuzhiyun 		return NULL;
821*4882a593Smuzhiyun 
822*4882a593Smuzhiyun 	if (stat(filename, &statbuf) < 0) {
823*4882a593Smuzhiyun 		printk(KERN_ERR "Error %d reading bpf file", -errno);
824*4882a593Smuzhiyun 		return false;
825*4882a593Smuzhiyun 	}
826*4882a593Smuzhiyun 	bpf_prog = uml_kmalloc(sizeof(struct sock_fprog), UM_GFP_KERNEL);
827*4882a593Smuzhiyun 	if (bpf_prog == NULL) {
828*4882a593Smuzhiyun 		printk(KERN_ERR "Failed to allocate bpf prog buffer");
829*4882a593Smuzhiyun 		return NULL;
830*4882a593Smuzhiyun 	}
831*4882a593Smuzhiyun 	bpf_prog->len = statbuf.st_size / sizeof(struct sock_filter);
832*4882a593Smuzhiyun 	bpf_prog->filter = NULL;
833*4882a593Smuzhiyun 	ffd = os_open_file(filename, of_read(OPENFLAGS()), 0);
834*4882a593Smuzhiyun 	if (ffd < 0) {
835*4882a593Smuzhiyun 		printk(KERN_ERR "Error %d opening bpf file", -errno);
836*4882a593Smuzhiyun 		goto bpf_failed;
837*4882a593Smuzhiyun 	}
838*4882a593Smuzhiyun 	bpf = uml_kmalloc(statbuf.st_size, UM_GFP_KERNEL);
839*4882a593Smuzhiyun 	if (bpf == NULL) {
840*4882a593Smuzhiyun 		printk(KERN_ERR "Failed to allocate bpf buffer");
841*4882a593Smuzhiyun 		goto bpf_failed;
842*4882a593Smuzhiyun 	}
843*4882a593Smuzhiyun 	bpf_prog->filter = bpf;
844*4882a593Smuzhiyun 	res = os_read_file(ffd, bpf, statbuf.st_size);
845*4882a593Smuzhiyun 	if (res < statbuf.st_size) {
846*4882a593Smuzhiyun 		printk(KERN_ERR "Failed to read bpf program %s, error %d", filename, res);
847*4882a593Smuzhiyun 		kfree(bpf);
848*4882a593Smuzhiyun 		goto bpf_failed;
849*4882a593Smuzhiyun 	}
850*4882a593Smuzhiyun 	os_close_file(ffd);
851*4882a593Smuzhiyun 	return bpf_prog;
852*4882a593Smuzhiyun bpf_failed:
853*4882a593Smuzhiyun 	if (ffd > 0)
854*4882a593Smuzhiyun 		os_close_file(ffd);
855*4882a593Smuzhiyun 	kfree(bpf_prog);
856*4882a593Smuzhiyun 	return NULL;
857*4882a593Smuzhiyun }
858