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
29static 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 <libifconfig.h>
47
48#include "ifconfig.h"
49
50/* ARGSUSED */
51static void
52setifgroup(const char *group_name, int d, int s, const struct afswtch *rafp)
53{
54	struct ifgroupreq ifgr;
55
56	memset(&ifgr, 0, sizeof(ifgr));
57	strlcpy(ifgr.ifgr_name, name, IFNAMSIZ);
58
59	if (group_name[0] && isdigit(group_name[strlen(group_name) - 1]))
60		errx(1, "setifgroup: group names may not end in a digit");
61
62	if (strlcpy(ifgr.ifgr_group, group_name, IFNAMSIZ) >= IFNAMSIZ)
63		errx(1, "setifgroup: group name too long");
64	if (ioctl(s, SIOCAIFGROUP, (caddr_t)&ifgr) == -1 && errno != EEXIST)
65		err(1," SIOCAIFGROUP");
66}
67
68/* ARGSUSED */
69static void
70unsetifgroup(const char *group_name, int d, int s, const struct afswtch *rafp)
71{
72	struct ifgroupreq ifgr;
73
74	memset(&ifgr, 0, sizeof(ifgr));
75	strlcpy(ifgr.ifgr_name, name, IFNAMSIZ);
76
77	if (group_name[0] && isdigit(group_name[strlen(group_name) - 1]))
78		errx(1, "unsetifgroup: group names may not end in a digit");
79
80	if (strlcpy(ifgr.ifgr_group, group_name, IFNAMSIZ) >= IFNAMSIZ)
81		errx(1, "unsetifgroup: group name too long");
82	if (ioctl(s, SIOCDIFGROUP, (caddr_t)&ifgr) == -1 && errno != ENOENT)
83		err(1, "SIOCDIFGROUP");
84}
85
86static void
87getifgroups(int s)
88{
89	struct ifgroupreq ifgr;
90	size_t cnt;
91
92	if (ifconfig_get_groups(lifh, name, &ifgr) == -1)
93		return;
94
95	cnt = 0;
96	for (size_t i = 0; i < ifgr.ifgr_len / sizeof(struct ifg_req); ++i) {
97		struct ifg_req *ifg = &ifgr.ifgr_groups[i];
98
99		if (strcmp(ifg->ifgrq_group, "all")) {
100			if (cnt == 0)
101				printf("\tgroups:");
102			cnt++;
103			printf(" %s", ifg->ifgrq_group);
104		}
105	}
106	if (cnt)
107		printf("\n");
108
109	free(ifgr.ifgr_groups);
110}
111
112static void
113printgroup(const char *groupname)
114{
115	struct ifgroupreq	 ifgr;
116	struct ifg_req		*ifg;
117	int			 len, cnt = 0;
118	int			 s;
119
120	s = socket(AF_LOCAL, SOCK_DGRAM, 0);
121	if (s == -1)
122		err(1, "socket(AF_LOCAL,SOCK_DGRAM)");
123	bzero(&ifgr, sizeof(ifgr));
124	strlcpy(ifgr.ifgr_name, groupname, sizeof(ifgr.ifgr_name));
125	if (ioctl(s, SIOCGIFGMEMB, (caddr_t)&ifgr) == -1) {
126		if (errno == EINVAL || errno == ENOTTY ||
127		    errno == ENOENT)
128			exit(exit_code);
129		else
130			err(1, "SIOCGIFGMEMB");
131	}
132
133	len = ifgr.ifgr_len;
134	if ((ifgr.ifgr_groups = calloc(1, len)) == NULL)
135		err(1, "printgroup");
136	if (ioctl(s, SIOCGIFGMEMB, (caddr_t)&ifgr) == -1)
137		err(1, "SIOCGIFGMEMB");
138
139	for (ifg = ifgr.ifgr_groups; ifg && len >= sizeof(struct ifg_req);
140	    ifg++) {
141		len -= sizeof(struct ifg_req);
142		printf("%s\n", ifg->ifgrq_member);
143		cnt++;
144	}
145	free(ifgr.ifgr_groups);
146
147	exit(exit_code);
148}
149
150static struct cmd group_cmds[] = {
151	DEF_CMD_ARG("group",	setifgroup),
152	DEF_CMD_ARG("-group",	unsetifgroup),
153};
154static struct afswtch af_group = {
155	.af_name	= "af_group",
156	.af_af		= AF_UNSPEC,
157	.af_other_status = getifgroups,
158};
159static struct option group_gopt = { "g:", "[-g groupname]", printgroup };
160
161static __constructor void
162group_ctor(void)
163{
164	int i;
165
166	for (i = 0; i < nitems(group_cmds);  i++)
167		cmd_register(&group_cmds[i]);
168	af_register(&af_group);
169	opt_register(&group_gopt);
170}
171