carp.c revision 281143 
1/*	$FreeBSD: head/sbin/ifconfig/carp.c 281143 2015-04-06 09:42:23Z glebius $ */
2/*	from $OpenBSD: ifconfig.c,v 1.82 2003/10/19 05:43:35 mcbride Exp $ */
3
4/*
5 * Copyright (c) 2002 Michael Shalayeff. All rights reserved.
6 * Copyright (c) 2003 Ryan McBride. All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR OR HIS RELATIVES BE LIABLE FOR ANY DIRECT,
21 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
23 * SERVICES; LOSS OF MIND, USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
25 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
26 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
27 * THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30#include <sys/param.h>
31#include <sys/ioctl.h>
32#include <sys/socket.h>
33#include <sys/sockio.h>
34
35#include <stdlib.h>
36#include <unistd.h>
37
38#include <net/if.h>
39#include <netinet/in.h>
40#include <netinet/in_var.h>
41#include <netinet/ip_carp.h>
42
43#include <ctype.h>
44#include <stdio.h>
45#include <string.h>
46#include <stdlib.h>
47#include <unistd.h>
48#include <err.h>
49#include <errno.h>
50
51#include "ifconfig.h"
52
53static const char *carp_states[] = { CARP_STATES };
54
55static void carp_status(int s);
56static void setcarp_vhid(const char *, int, int, const struct afswtch *rafp);
57static void setcarp_callback(int, void *);
58static void setcarp_advbase(const char *,int, int, const struct afswtch *rafp);
59static void setcarp_advskew(const char *, int, int, const struct afswtch *rafp);
60static void setcarp_passwd(const char *, int, int, const struct afswtch *rafp);
61
62static int carpr_vhid = -1;
63static int carpr_advskew = -1;
64static int carpr_advbase = -1;
65static int carpr_state = -1;
66static unsigned char const *carpr_key;
67
68static void
69carp_status(int s)
70{
71	struct carpreq carpr[CARP_MAXVHID];
72	int i;
73
74	bzero(carpr, sizeof(struct carpreq) * CARP_MAXVHID);
75	carpr[0].carpr_count = CARP_MAXVHID;
76	ifr.ifr_data = (caddr_t)&carpr;
77
78	if (ioctl(s, SIOCGVH, (caddr_t)&ifr) == -1)
79		return;
80
81	for (i = 0; i < carpr[0].carpr_count; i++) {
82		printf("\tcarp: %s vhid %d advbase %d advskew %d",
83		    carp_states[carpr[i].carpr_state], carpr[i].carpr_vhid,
84		    carpr[i].carpr_advbase, carpr[i].carpr_advskew);
85		if (printkeys && carpr[i].carpr_key[0] != '\0')
86			printf(" key \"%s\"\n", carpr[i].carpr_key);
87		else
88			printf("\n");
89	}
90}
91
92static void
93setcarp_vhid(const char *val, int d, int s, const struct afswtch *afp)
94{
95
96	carpr_vhid = atoi(val);
97
98	if (carpr_vhid <= 0 || carpr_vhid > CARP_MAXVHID)
99		errx(1, "vhid must be greater than 0 and less than %u",
100		    CARP_MAXVHID);
101
102	switch (afp->af_af) {
103#ifdef INET
104	case AF_INET:
105	    {
106		struct in_aliasreq *ifra;
107
108		ifra = (struct in_aliasreq *)afp->af_addreq;
109		ifra->ifra_vhid = carpr_vhid;
110		break;
111	    }
112#endif
113#ifdef INET6
114	case AF_INET6:
115	    {
116		struct in6_aliasreq *ifra;
117
118		ifra = (struct in6_aliasreq *)afp->af_addreq;
119		ifra->ifra_vhid = carpr_vhid;
120		break;
121	    }
122#endif
123	default:
124		errx(1, "%s doesn't support carp(4)", afp->af_name);
125	}
126
127	callback_register(setcarp_callback, NULL);
128}
129
130static void
131setcarp_callback(int s, void *arg __unused)
132{
133	struct carpreq carpr;
134
135	bzero(&carpr, sizeof(struct carpreq));
136	carpr.carpr_vhid = carpr_vhid;
137	carpr.carpr_count = 1;
138	ifr.ifr_data = (caddr_t)&carpr;
139
140	if (ioctl(s, SIOCGVH, (caddr_t)&ifr) == -1 && errno != ENOENT)
141		err(1, "SIOCGVH");
142
143	if (carpr_key != NULL)
144		/* XXX Should hash the password into the key here? */
145		strlcpy(carpr.carpr_key, carpr_key, CARP_KEY_LEN);
146	if (carpr_advskew > -1)
147		carpr.carpr_advskew = carpr_advskew;
148	if (carpr_advbase > -1)
149		carpr.carpr_advbase = carpr_advbase;
150	if (carpr_state > -1)
151		carpr.carpr_state = carpr_state;
152
153	if (ioctl(s, SIOCSVH, (caddr_t)&ifr) == -1)
154		err(1, "SIOCSVH");
155}
156
157static void
158setcarp_passwd(const char *val, int d, int s, const struct afswtch *afp)
159{
160
161	if (carpr_vhid == -1)
162		errx(1, "passwd requires vhid");
163
164	carpr_key = val;
165}
166
167static void
168setcarp_advskew(const char *val, int d, int s, const struct afswtch *afp)
169{
170
171	if (carpr_vhid == -1)
172		errx(1, "advskew requires vhid");
173
174	carpr_advskew = atoi(val);
175}
176
177static void
178setcarp_advbase(const char *val, int d, int s, const struct afswtch *afp)
179{
180
181	if (carpr_vhid == -1)
182		errx(1, "advbase requires vhid");
183
184	carpr_advbase = atoi(val);
185}
186
187static void
188setcarp_state(const char *val, int d, int s, const struct afswtch *afp)
189{
190	int i;
191
192	if (carpr_vhid == -1)
193		errx(1, "state requires vhid");
194
195	for (i = 0; i <= CARP_MAXSTATE; i++)
196		if (strcasecmp(carp_states[i], val) == 0) {
197			carpr_state = i;
198			return;
199		}
200
201	errx(1, "unknown state");
202}
203
204static struct cmd carp_cmds[] = {
205	DEF_CMD_ARG("advbase",	setcarp_advbase),
206	DEF_CMD_ARG("advskew",	setcarp_advskew),
207	DEF_CMD_ARG("pass",	setcarp_passwd),
208	DEF_CMD_ARG("vhid",	setcarp_vhid),
209	DEF_CMD_ARG("state",	setcarp_state),
210};
211static struct afswtch af_carp = {
212	.af_name	= "af_carp",
213	.af_af		= AF_UNSPEC,
214	.af_other_status = carp_status,
215};
216
217static __constructor void
218carp_ctor(void)
219{
220#define	N(a)	(sizeof(a) / sizeof(a[0]))
221	int i;
222
223	for (i = 0; i < N(carp_cmds);  i++)
224		cmd_register(&carp_cmds[i]);
225	af_register(&af_carp);
226#undef N
227}
228