1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
/* vi: set sw=4 ts=4:
*
* vconfig.c - Creates virtual ethernet devices.
*
* Copyright 2012 Sandeep Sharma <sandeep.jack2756@gmail.com>, Kyungwan Han <asura321@gmail.com>
*
* Not in SUSv4.
USE_VCONFIG(NEWTOY(vconfig, "<2>4", TOYFLAG_NEEDROOT|TOYFLAG_SBIN))
config VCONFIG
bool "vconfig"
default y
help
usage: vconfig COMMAND [OPTIONS]
add [interface-name] [vlan_id]
rem [vlan-name]
set_flag [interface-name] [flag-num] [0 | 1]
set_egress_map [vlan-name] [skb_priority] [vlan_qos]
set_ingress_map [vlan-name] [skb_priority] [vlan_qos]
set_name_type [name-type]
Create and remove virtual ethernet devices
*/
#include "toys.h"
#include "toynet.h"
#include <linux/if_vlan.h>
#include <linux/sockios.h>
/*
* convert str to long within given range
*/
static int strtol_range(char *str, int min, int max)
{
char *endptr = NULL;
errno = 0;
long ret_value = strtol(str, &endptr, 10);
if(errno) perror_exit("Invalid num %s", str);
else {
if(endptr && (*endptr != '\0' || endptr == str))
perror_exit("Not a valid num %s", str);
}
if(ret_value >= min && ret_value <= max) return ret_value;
else perror_exit("Number %s is not in valid [%d-%d] Range\n", str, min, max);
}
/*
* vconfig main function.
*/
void vconfig_main(void)
{
#define MAX_VLAN_ID 4094
struct vlan_ioctl_args request;
char *interface_name = NULL;
unsigned int name_type = VLAN_NAME_TYPE_PLUS_VID;
char *cmd;
int fd = 0;
int vlan_id = 0;
if((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) perror_exit("Can't open socket"); //Use socket instead of open
memset(&request, 0, sizeof(struct vlan_ioctl_args)); // Null set all the VLAN info's.
cmd = toys.optargs[0]; // Fetch cmd and proceed.
if(strcmp(cmd, "set_name_type") == 0) {
if(strcmp(toys.optargs[1], "VLAN_PLUS_VID") == 0) {
name_type = VLAN_NAME_TYPE_PLUS_VID;
}
else if(strcmp(toys.optargs[1], "VLAN_PLUS_VID_NO_PAD") == 0) {
name_type = VLAN_NAME_TYPE_PLUS_VID_NO_PAD;
}
else if(strcmp(toys.optargs[1], "DEV_PLUS_VID") == 0) {
name_type = VLAN_NAME_TYPE_RAW_PLUS_VID;
}
else if(strcmp(toys.optargs[1], "DEV_PLUS_VID_NO_PAD") == 0) {
name_type = VLAN_NAME_TYPE_RAW_PLUS_VID_NO_PAD;
}
else perror_exit("ERROR: Invalid name type");
request.u.name_type = name_type;
request.cmd = SET_VLAN_NAME_TYPE_CMD;
if(ioctl(fd, SIOCSIFVLAN, &request) == 0) {
xprintf("Successful set_name_type for VLAN subsystem\n");
exit(EXIT_SUCCESS);
}
else perror_exit("Failed to set set_name_type:");
}
else {
interface_name = toys.optargs[1]; // Store interface name.
if(strlen(interface_name) > 15) perror_exit("ERROR:if_name length can not be greater than 15");
strcpy(request.device1, interface_name); //we had exited if interface_name length greater than 15, so here it never overflows.
}
if(strcmp(cmd, "add") == 0) {
request.cmd = ADD_VLAN_CMD;
if(toys.optargs[2]) vlan_id = strtol_range(toys.optargs[2], 0, MAX_VLAN_ID);
else vlan_id = 0;
request.u.VID = vlan_id;
}
else if(strcmp(cmd, "rem") == 0) {
request.cmd = DEL_VLAN_CMD;
}
else if(strcmp(cmd, "set_flag") == 0) {
request.cmd = SET_VLAN_FLAG_CMD;
if(toys.optargs[2]) request.u.flag = strtol_range(toys.optargs[2], 0, 1);
else request.u.flag = 0;
if(toys.optargs[3]) request.vlan_qos = strtol_range(toys.optargs[3], 0, 7);
else request.vlan_qos = 0;
}
else if(strcmp(cmd, "set_egress_map") == 0) {
request.cmd = SET_VLAN_EGRESS_PRIORITY_CMD;
if(toys.optargs[2]) request.u.skb_priority = strtol_range(toys.optargs[2], 0, INT_MAX);
else request.u.skb_priority = 0;
if(toys.optargs[3]) request.vlan_qos = strtol_range(toys.optargs[3], 0, 7);
else request.vlan_qos = 0;
}
else if(strcmp(cmd, "set_ingress_map") == 0) {
request.cmd = SET_VLAN_INGRESS_PRIORITY_CMD;
if(toys.optargs[2]) request.u.skb_priority = strtol_range(toys.optargs[2], 0, INT_MAX);
else request.u.skb_priority = 0;
if(toys.optargs[3]) request.vlan_qos = strtol_range(toys.optargs[3], 0, 7); //To set flag we must have to set vlan_qos
else request.vlan_qos = 0;
}
else {
xclose(fd);
perror_exit("Unknown command %s", cmd);
}
if(ioctl(fd, SIOCSIFVLAN, &request) == 0) {
if(strcmp(cmd, "add") == 0 && vlan_id == 1)
xprintf("WARNING: VLAN 1 does not work with many switches,consider another number if you have problems.\n");
xprintf("Successful %s on device %s\n", cmd, interface_name);
}
else perror_exit("Failed to %s, on vlan subsystem %s.", cmd, interface_name);
}
|
