sbin/ifconfig/carp.c

   1 /*      $FreeBSD$ */
   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
  54 static const char *carp_states[] = { CARP_STATES };
  55
  56 static void carp_status(int s);
  57 static void setcarp_vhid(const char *, int, int, const struct afswtch *rafp);
  58 static void setcarp_callback(int, void *);
  59 static void setcarp_advbase(const char *,int, int, const struct afswtch *rafp);
  60 static void setcarp_advskew(const char *, int, int, const struct afswtch *rafp);
  61 static void setcarp_passwd(const char *, int, int, const struct afswtch *rafp);
  62
  63 static int carpr_vhid = -1;
  64 static int carpr_advskew = -1;
  65 static int carpr_advbase = -1;
  66 static int carpr_state = -1;
  67 static unsigned char const *carpr_key;
  68
  69 static void
  70 carp_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
  93 static void
  94 setcarp_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
 131 static void
 132 setcarp_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
 158 static void
 159 setcarp_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
 168 static void
 169 setcarp_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
 178 static void
 179 setcarp_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
 188 static void
 189 setcarp_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
 205 static 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 };
 212 static struct afswtch af_carp = {
 213         .af_name        = "af_carp",
 214         .af_af          = AF_UNSPEC,
 215         .af_other_status = carp_status,
 216 };
 217
 218 static __constructor void
 219 carp_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 }