lib/libifconfig/libifconfig_carp.c

   1 /*
   2  * Copyright (c) 1983, 1993
   3  *  The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. Neither the name of the University nor the names of its contributors
  14  *    may be used to endorse or promote products derived from this software
  15  *    without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  *
  29  * $FreeBSD$
  30  */
  31 #include <sys/param.h>
  32 #include <sys/ioctl.h>
  33
  34 #include <net/if.h>
  35 #include <netinet/ip_carp.h>
  36 #include <netinet/ip_carp_nl.h>
  37
  38 #include <netlink/netlink.h>
  39 #include <netlink/netlink_generic.h>
  40 #include <netlink/netlink_snl.h>
  41 #include <netlink/netlink_snl_generic.h>
  42 #include <netlink/netlink_snl_route.h>
  43
  44 #include <string.h>
  45 #include <strings.h>
  46
  47 #include "libifconfig.h"
  48 #include "libifconfig_internal.h"
  49
  50 #include <stdio.h>
  51
  52 #define _OUT(_field)    offsetof(struct ifconfig_carp, _field)
  53 static struct snl_attr_parser ap_carp_get[] = {
  54         { .type = CARP_NL_VHID, .off = _OUT(carpr_vhid), .cb = snl_attr_get_uint32 },
  55         { .type = CARP_NL_STATE, .off = _OUT(carpr_state), .cb = snl_attr_get_uint32 },
  56         { .type = CARP_NL_ADVBASE, .off = _OUT(carpr_advbase), .cb = snl_attr_get_int32 },
  57         { .type = CARP_NL_ADVSKEW, .off = _OUT(carpr_advskew), .cb = snl_attr_get_int32 },
  58         { .type = CARP_NL_KEY, .off = _OUT(carpr_key), .cb = snl_attr_copy_string, .arg_u32 = CARP_KEY_LEN },
  59         { .type = CARP_NL_ADDR, .off = _OUT(carpr_addr), .cb = snl_attr_get_in_addr },
  60         { .type = CARP_NL_ADDR6, .off = _OUT(carpr_addr6), .cb = snl_attr_get_in6_addr },
  61 };
  62 #undef _OUT
  63
  64 SNL_DECLARE_GENL_PARSER(carp_get_parser, ap_carp_get);
  65
  66 static int
  67 _ifconfig_carp_get(ifconfig_handle_t *h, const char *name,
  68     struct ifconfig_carp *carp, size_t ncarp, uint32_t vhid)
  69 {
  70         struct snl_state ss = {};
  71         struct snl_errmsg_data e = {};
  72         struct snl_writer nw;
  73         struct nlmsghdr *hdr;
  74         size_t i = 0;
  75         uint32_t seq_id;
  76         int family_id;
  77
  78         ifconfig_error_clear(h);
  79
  80         bzero(carp, sizeof(*carp) * ncarp);
  81
  82         if (! snl_init(&ss, NETLINK_GENERIC)) {
  83                 ifconfig_error(h, NETLINK, ENOTSUP);
  84                 return (-1);
  85         }
  86
  87         snl_init_writer(&ss, &nw);
  88
  89         family_id = snl_get_genl_family(&ss, CARP_NL_FAMILY_NAME);
  90         if (family_id == 0) {
  91                 ifconfig_error(h, NETLINK, EPROTONOSUPPORT);
  92                 goto out;
  93         }
  94
  95         hdr = snl_create_genl_msg_request(&nw, family_id, CARP_NL_CMD_GET);
  96         hdr->nlmsg_flags |= NLM_F_DUMP;
  97
  98         snl_add_msg_attr_string(&nw, CARP_NL_IFNAME, name);
  99
 100         if (vhid != 0)
 101                 snl_add_msg_attr_u32(&nw, CARP_NL_VHID, vhid);
 102
 103         hdr = snl_finalize_msg(&nw);
 104         if (hdr == NULL) {
 105                 ifconfig_error(h, NETLINK, ENOMEM);
 106                 goto out;
 107         }
 108         seq_id = hdr->nlmsg_seq;
 109         if (! snl_send_message(&ss, hdr)) {
 110                 ifconfig_error(h, NETLINK, EIO);
 111                 goto out;
 112         }
 113
 114         while ((hdr = snl_read_reply_multi(&ss, seq_id, &e)) != NULL) {
 115                 if (e.error != 0) {
 116                         ifconfig_error(h, NETLINK, e.error);
 117                         break;
 118                 }
 119
 120                 if (i >= ncarp) {
 121                         ifconfig_error(h, NETLINK, E2BIG);
 122                         break;
 123                 }
 124
 125                 memset(&carp[i], 0, sizeof(carp[0]));
 126                 if (! snl_parse_nlmsg(&ss, hdr, &carp_get_parser, &carp[i]))
 127                         continue;
 128
 129                 i++;
 130                 carp[0].carpr_count = i;
 131
 132                 if (i > ncarp) {
 133                         ifconfig_error(h, NETLINK, E2BIG);
 134                         break;
 135                 }
 136         }
 137
 138 out:
 139         snl_free(&ss);
 140
 141         return (h->error.errcode ? -1 : 0);
 142 }
 143
 144 int
 145 ifconfig_carp_set_info(ifconfig_handle_t *h, const char *name,
 146     const struct ifconfig_carp *carpr)
 147 {
 148         struct snl_state ss = {};
 149         struct snl_writer nw;
 150         struct nlmsghdr *hdr;
 151         int family_id;
 152         uint32_t seq_id;
 153
 154         ifconfig_error_clear(h);
 155
 156         if (! snl_init(&ss, NETLINK_GENERIC)) {
 157                 ifconfig_error(h, NETLINK, ENOTSUP);
 158                 return (-1);
 159         }
 160
 161         snl_init_writer(&ss, &nw);
 162
 163         family_id = snl_get_genl_family(&ss, CARP_NL_FAMILY_NAME);
 164         if (family_id == 0) {
 165                 ifconfig_error(h, NETLINK, EPROTONOSUPPORT);
 166                 return (-1);
 167         }
 168         hdr = snl_create_genl_msg_request(&nw, family_id, CARP_NL_CMD_SET);
 169
 170         snl_add_msg_attr_u32(&nw, CARP_NL_VHID, carpr->carpr_vhid);
 171         snl_add_msg_attr_u32(&nw, CARP_NL_STATE, carpr->carpr_state);
 172         snl_add_msg_attr_s32(&nw, CARP_NL_ADVBASE, carpr->carpr_advbase);
 173         snl_add_msg_attr_s32(&nw, CARP_NL_ADVSKEW, carpr->carpr_advskew);
 174         snl_add_msg_attr_string(&nw, CARP_NL_IFNAME, name);
 175         snl_add_msg_attr(&nw, CARP_NL_ADDR, sizeof(carpr->carpr_addr),
 176             &carpr->carpr_addr);
 177         snl_add_msg_attr(&nw, CARP_NL_ADDR6, sizeof(carpr->carpr_addr6),
 178             &carpr->carpr_addr6);
 179         snl_add_msg_attr_string(&nw, CARP_NL_KEY, carpr->carpr_key);
 180
 181         hdr = snl_finalize_msg(&nw);
 182         if (hdr == NULL) {
 183                 ifconfig_error(h, NETLINK, ENOMEM);
 184                 goto out;
 185         }
 186
 187         seq_id = hdr->nlmsg_seq;
 188         if (! snl_send_message(&ss, hdr)) {
 189                 ifconfig_error(h, NETLINK, EIO);
 190                 goto out;
 191         }
 192
 193         struct snl_errmsg_data e = { };
 194         if (! snl_read_reply_code(&ss, seq_id, &e))
 195                 ifconfig_error(h, NETLINK, e.error);
 196
 197 out:
 198         snl_free(&ss);
 199
 200         return (h->error.errcode ? -1 : 0);
 201 }
 202
 203 int
 204 ifconfig_carp_get_vhid(ifconfig_handle_t *h, const char *name,
 205     struct ifconfig_carp *carp, uint32_t vhid)
 206 {
 207         return (_ifconfig_carp_get(h, name, carp, 1, vhid));
 208 }
 209
 210 int
 211 ifconfig_carp_get_info(ifconfig_handle_t *h, const char *name,
 212     struct ifconfig_carp *carp, size_t ncarp)
 213 {
 214         return (_ifconfig_carp_get(h, name, carp, ncarp, 0));
 215 }