carp.c revision 228571 
1/*	$FreeBSD: head/sbin/ifconfig/ifcarp.c 228571 2011-12-16 12:16:56Z 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 <net/if_var.h>
40#include <netinet/in.h>
41#include <netinet/in_var.h>
42#include <netinet/ip_carp.h>
43
44#include <ctype.h>
45#include <stdio.h>
46#include <string.h>
47#include <stdlib.h>
48#include <unistd.h>
49#include <err.h>
50#include <errno.h>
51
52#include "ifconfig.h"
53
54static const char *carp_states[] = { CARP_STATES };
55
56static void carp_status(int s);
57static void setcarp_vhid(const char *, int, int, const struct afswtch *rafp);
58static void setcarp_callback(int, void *);
59static void setcarp_advbase(const char *,int, int, const struct afswtch *rafp);
60static void setcarp_advskew(const char *, int, int, const struct afswtch *rafp);
61static void setcarp_passwd(const char *, int, int, const struct afswtch *rafp);
62
63static int carpr_vhid = -1;
64static int carpr_advskew = -1;
65static int carpr_advbase = -1;
66static int carpr_state = -1;
67static unsigned char const *carpr_key;
68
69static void
70carp_status(int s)
71{
72	struct carpreq carpr[CARP_MAXVHID];
73	int i;
74
75	bzero(carpr, sizeof(struct carpreq) * CARP_MAXVHID);
76	carpr[0].carpr_count = CARP_MAXVHID;
77	ifr.ifr_data = (caddr_t)&carpr;
78
79	if (ioctl(s, SIOCGVH, (caddr_t)&ifr) == -1)
80		return;
81
82	for (i = 0; i < carpr[0].carpr_count; i++) {
83		printf("\tcarp: %s vhid %d advbase %d advskew %d",
84		    carp_states[carpr[i].carpr_state], carpr[i].carpr_vhid,
85		    carpr[i].carpr_advbase, carpr[i].carpr_advskew);
86		if (printkeys && carpr[i].carpr_key[0] != '\0')
87			printf(" key \"%s\"\n", carpr[i].carpr_key);
88		else
89			printf("\n");
90	}
91}
92
93static void
94setcarp_vhid(const char *val, int d, int s, const struct afswtch *afp)
95{
96
97	carpr_vhid = atoi(val);
98
99	if (carpr_vhid <= 0 || carpr_vhid > CARP_MAXVHID)
100		errx(1, "vhid must be greater than 0 and less than %u",
101		    CARP_MAXVHID);
102
103	switch (afp->af_af) {
104#ifdef INET
105	case AF_INET:
106	    {
107		struct in_aliasreq *ifra;
108
109		ifra = (struct in_aliasreq *)afp->af_addreq;
110		ifra->ifra_vhid = carpr_vhid;
111		break;
112	    }
113#endif
114#ifdef INET6
115	case AF_INET6:
116	    {
117		struct in6_aliasreq *ifra;
118
119		ifra = (struct in6_aliasreq *)afp->af_addreq;
120		ifra->ifra_vhid = carpr_vhid;
121		break;
122	    }
123#endif
124	default:
125		errx(1, "%s doesn't support carp(4)", afp->af_name);
126	}
127
128	callback_register(setcarp_callback, NULL);
129}
130
131static void
132setcarp_callback(int s, void *arg __unused)
133{
134	struct carpreq carpr;
135
136	bzero(&carpr, sizeof(struct carpreq));
137	carpr.carpr_vhid = carpr_vhid;
138	carpr.carpr_count = 1;
139	ifr.ifr_data = (caddr_t)&carpr;
140
141	if (ioctl(s, SIOCGVH, (caddr_t)&ifr) == -1 && errno != ENOENT)
142		err(1, "SIOCGVH");
143
144	if (carpr_key != NULL)
145		/* XXX Should hash the password into the key here? */
146		strlcpy(carpr.carpr_key, carpr_key, CARP_KEY_LEN);
147	if (carpr_advskew > -1)
148		carpr.carpr_advskew = carpr_advskew;
149	if (carpr_advbase > -1)
150		carpr.carpr_advbase = carpr_advbase;
151	if (carpr_state > -1)
152		carpr.carpr_state = carpr_state;
153
154	if (ioctl(s, SIOCSVH, (caddr_t)&ifr) == -1)
155		err(1, "SIOCSVH");
156}
157
158static void
159setcarp_passwd(const char *val, int d, int s, const struct afswtch *afp)
160{
161
162	if (carpr_vhid == -1)
163		errx(1, "passwd requires vhid");
164
165	carpr_key = val;
166}
167
168static void
169setcarp_advskew(const char *val, int d, int s, const struct afswtch *afp)
170{
171
172	if (carpr_vhid == -1)
173		errx(1, "advskew requires vhid");
174
175	carpr_advskew = atoi(val);
176}
177
178static void
179setcarp_advbase(const char *val, int d, int s, const struct afswtch *afp)
180{
181
182	if (carpr_vhid == -1)
183		errx(1, "advbase requires vhid");
184
185	carpr_advbase = atoi(val);
186}
187
188static void
189setcarp_state(const char *val, int d, int s, const struct afswtch *afp)
190{
191	int i;
192
193	if (carpr_vhid == -1)
194		errx(1, "state requires vhid");
195
196	for (i = 0; i <= CARP_MAXSTATE; i++)
197		if (strcasecmp(carp_states[i], val) == 0) {
198			carpr_state = i;
199			return;
200		}
201
202	errx(1, "unknown state");
203}
204
205static struct cmd carp_cmds[] = {
206	DEF_CMD_ARG("advbase",	setcarp_advbase),
207	DEF_CMD_ARG("advskew",	setcarp_advskew),
208	DEF_CMD_ARG("pass",	setcarp_passwd),
209	DEF_CMD_ARG("vhid",	setcarp_vhid),
210	DEF_CMD_ARG("state",	setcarp_state),
211};
212static struct afswtch af_carp = {
213	.af_name	= "af_carp",
214	.af_af		= AF_UNSPEC,
215	.af_other_status = carp_status,
216};
217
218static __constructor void
219carp_ctor(void)
220{
221#define	N(a)	(sizeof(a) / sizeof(a[0]))
222	int i;
223
224	for (i = 0; i < N(carp_cmds);  i++)
225		cmd_register(&carp_cmds[i]);
226	af_register(&af_carp);
227#undef N
228}
229