xref: /freebsd/sbin/ifconfig/ifgroup.c (revision f5147e312f43a9050468de539aeafa072caa1a60)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2006 Max Laier. All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25  * SUCH DAMAGE.
26  */
27 
28 #ifndef lint
29 static const char rcsid[] =
30   "$FreeBSD$";
31 #endif /* not lint */
32 
33 #include <sys/param.h>
34 #include <sys/ioctl.h>
35 #include <sys/socket.h>
36 #include <net/if.h>
37 
38 #include <ctype.h>
39 #include <err.h>
40 #include <errno.h>
41 #include <stdio.h>
42 #include <stdlib.h>
43 #include <string.h>
44 #include <unistd.h>
45 
46 #include "ifconfig.h"
47 
48 /* ARGSUSED */
49 static void
50 setifgroup(const char *group_name, int d, int s, const struct afswtch *rafp)
51 {
52 	struct ifgroupreq ifgr;
53 
54 	memset(&ifgr, 0, sizeof(ifgr));
55 	strlcpy(ifgr.ifgr_name, name, IFNAMSIZ);
56 
57 	if (group_name[0] && isdigit(group_name[strlen(group_name) - 1]))
58 		errx(1, "setifgroup: group names may not end in a digit");
59 
60 	if (strlcpy(ifgr.ifgr_group, group_name, IFNAMSIZ) >= IFNAMSIZ)
61 		errx(1, "setifgroup: group name too long");
62 	if (ioctl(s, SIOCAIFGROUP, (caddr_t)&ifgr) == -1 && errno != EEXIST)
63 		err(1," SIOCAIFGROUP");
64 }
65 
66 /* ARGSUSED */
67 static void
68 unsetifgroup(const char *group_name, int d, int s, const struct afswtch *rafp)
69 {
70 	struct ifgroupreq ifgr;
71 
72 	memset(&ifgr, 0, sizeof(ifgr));
73 	strlcpy(ifgr.ifgr_name, name, IFNAMSIZ);
74 
75 	if (group_name[0] && isdigit(group_name[strlen(group_name) - 1]))
76 		errx(1, "unsetifgroup: group names may not end in a digit");
77 
78 	if (strlcpy(ifgr.ifgr_group, group_name, IFNAMSIZ) >= IFNAMSIZ)
79 		errx(1, "unsetifgroup: group name too long");
80 	if (ioctl(s, SIOCDIFGROUP, (caddr_t)&ifgr) == -1 && errno != ENOENT)
81 		err(1, "SIOCDIFGROUP");
82 }
83 
84 static void
85 getifgroups(int s)
86 {
87 	int			 len, cnt;
88 	struct ifgroupreq	 ifgr;
89 	struct ifg_req		*ifg;
90 
91 	memset(&ifgr, 0, sizeof(ifgr));
92 	strlcpy(ifgr.ifgr_name, name, IFNAMSIZ);
93 
94 	if (ioctl(s, SIOCGIFGROUP, (caddr_t)&ifgr) == -1) {
95 		if (errno == EINVAL || errno == ENOTTY)
96 			return;
97 		else
98 			err(1, "SIOCGIFGROUP");
99 	}
100 
101 	len = ifgr.ifgr_len;
102 	ifgr.ifgr_groups =
103 	    (struct ifg_req *)calloc(len / sizeof(struct ifg_req),
104 	    sizeof(struct ifg_req));
105 	if (ifgr.ifgr_groups == NULL)
106 		err(1, "getifgroups");
107 	if (ioctl(s, SIOCGIFGROUP, (caddr_t)&ifgr) == -1)
108 		err(1, "SIOCGIFGROUP");
109 
110 	cnt = 0;
111 	ifg = ifgr.ifgr_groups;
112 	for (; ifg && len >= sizeof(struct ifg_req); ifg++) {
113 		len -= sizeof(struct ifg_req);
114 		if (strcmp(ifg->ifgrq_group, "all")) {
115 			if (cnt == 0)
116 				printf("\tgroups: ");
117 			cnt++;
118 			printf("%s ", ifg->ifgrq_group);
119 		}
120 	}
121 	if (cnt)
122 		printf("\n");
123 
124 	free(ifgr.ifgr_groups);
125 }
126 
127 static void
128 printgroup(const char *groupname)
129 {
130 	struct ifgroupreq	 ifgr;
131 	struct ifg_req		*ifg;
132 	int			 len, cnt = 0;
133 	int			 s;
134 
135 	s = socket(AF_LOCAL, SOCK_DGRAM, 0);
136 	if (s == -1)
137 		err(1, "socket(AF_LOCAL,SOCK_DGRAM)");
138 	bzero(&ifgr, sizeof(ifgr));
139 	strlcpy(ifgr.ifgr_name, groupname, sizeof(ifgr.ifgr_name));
140 	if (ioctl(s, SIOCGIFGMEMB, (caddr_t)&ifgr) == -1) {
141 		if (errno == EINVAL || errno == ENOTTY ||
142 		    errno == ENOENT)
143 			exit(0);
144 		else
145 			err(1, "SIOCGIFGMEMB");
146 	}
147 
148 	len = ifgr.ifgr_len;
149 	if ((ifgr.ifgr_groups = calloc(1, len)) == NULL)
150 		err(1, "printgroup");
151 	if (ioctl(s, SIOCGIFGMEMB, (caddr_t)&ifgr) == -1)
152 		err(1, "SIOCGIFGMEMB");
153 
154 	for (ifg = ifgr.ifgr_groups; ifg && len >= sizeof(struct ifg_req);
155 	    ifg++) {
156 		len -= sizeof(struct ifg_req);
157 		printf("%s\n", ifg->ifgrq_member);
158 		cnt++;
159 	}
160 	free(ifgr.ifgr_groups);
161 
162 	exit(0);
163 }
164 
165 static struct cmd group_cmds[] = {
166 	DEF_CMD_ARG("group",	setifgroup),
167 	DEF_CMD_ARG("-group",	unsetifgroup),
168 };
169 static struct afswtch af_group = {
170 	.af_name	= "af_group",
171 	.af_af		= AF_UNSPEC,
172 	.af_other_status = getifgroups,
173 };
174 static struct option group_gopt = { "g:", "[-g groupname]", printgroup };
175 
176 static __constructor void
177 group_ctor(void)
178 {
179 	int i;
180 
181 	for (i = 0; i < nitems(group_cmds);  i++)
182 		cmd_register(&group_cmds[i]);
183 	af_register(&af_group);
184 	opt_register(&group_gopt);
185 }
186