ctdb/server/ipalloc_common.c

   1 /*
   2    ctdb ip takeover code
   3
   4    Copyright (C) Ronnie Sahlberg  2007
   5    Copyright (C) Andrew Tridgell  2007
   6    Copyright (C) Martin Schwenke  2011
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "replace.h"
  23 #include "system/network.h"
  24
  25 #include "ctdb_private.h"
  26
  27 #include "lib/util/time.h"
  28
  29 #include "lib/util/debug.h"
  30 #include "common/logging.h"
  31
  32 #include "common/common.h"
  33 #include "common/rb_tree.h"
  34
  35 #include "protocol/protocol_util.h"
  36
  37 #include "server/ipalloc_private.h"
  38
  39 #define TAKEOVER_TIMEOUT() timeval_current_ofs(ctdb->tunable.takeover_timeout,0)
  40
  41 /* Given a physical node, return the number of
  42    public addresses that is currently assigned to this node.
  43 */
  44 int node_ip_coverage(int32_t pnn, struct public_ip_list *ips)
  45 {
  46         int num=0;
  47
  48         for (;ips;ips=ips->next) {
  49                 if (ips->pnn == pnn) {
  50                         num++;
  51                 }
  52         }
  53         return num;
  54 }
  55
  56
  57 /* Can the given node host the given IP: is the public IP known to the
  58  * node and is NOIPHOST unset?
  59 */
  60 static bool can_node_host_ip(struct ipalloc_state *ipalloc_state,
  61                              int32_t pnn,
  62                              struct public_ip_list *ip)
  63 {
  64         return bitmap_query(ip->available_on, pnn);
  65 }
  66
  67 bool can_node_takeover_ip(struct ipalloc_state *ipalloc_state,
  68                           int32_t pnn,
  69                           struct public_ip_list *ip)
  70 {
  71         if (ipalloc_state->no_ip_takeover) {
  72                 return false;
  73         }
  74
  75         return can_node_host_ip(ipalloc_state, pnn, ip);
  76 }
  77
  78 /* search the node lists list for a node to takeover this ip.
  79    pick the node that currently are serving the least number of ips
  80    so that the ips get spread out evenly.
  81 */
  82 int find_takeover_node(struct ipalloc_state *ipalloc_state,
  83                        struct public_ip_list *ip)
  84 {
  85         int pnn, min=0, num;
  86         int i, numnodes;
  87
  88         numnodes = ipalloc_state->num;
  89         pnn    = -1;
  90         for (i=0; i<numnodes; i++) {
  91                 /* verify that this node can serve this ip */
  92                 if (!can_node_takeover_ip(ipalloc_state, i, ip)) {
  93                         /* no it couldnt   so skip to the next node */
  94                         continue;
  95                 }
  96
  97                 num = node_ip_coverage(i, ipalloc_state->all_ips);
  98                 /* was this the first node we checked ? */
  99                 if (pnn == -1) {
 100                         pnn = i;
 101                         min  = num;
 102                 } else {
 103                         if (num < min) {
 104                                 pnn = i;
 105                                 min  = num;
 106                         }
 107                 }
 108         }
 109         if (pnn == -1) {
 110                 DEBUG(DEBUG_WARNING,(__location__ " Could not find node to take over public address '%s'\n",
 111                                      ctdb_sock_addr_to_string(ipalloc_state,
 112                                                               &ip->addr,
 113                                                               false)));
 114
 115                 return -1;
 116         }
 117
 118         ip->pnn = pnn;
 119         return 0;
 120 }
 121
 122 uint32_t *ip_key(ctdb_sock_addr *ip)
 123 {
 124         static uint32_t key[IP_KEYLEN];
 125
 126         bzero(key, sizeof(key));
 127
 128         switch (ip->sa.sa_family) {
 129         case AF_INET:
 130                 key[3]  = htonl(ip->ip.sin_addr.s_addr);
 131                 break;
 132         case AF_INET6: {
 133                 uint32_t *s6_a32 = (uint32_t *)&(ip->ip6.sin6_addr.s6_addr);
 134                 key[0]  = htonl(s6_a32[0]);
 135                 key[1]  = htonl(s6_a32[1]);
 136                 key[2]  = htonl(s6_a32[2]);
 137                 key[3]  = htonl(s6_a32[3]);
 138                 break;
 139         }
 140         default:
 141                 DEBUG(DEBUG_ERR, (__location__ " ERROR, unknown family passed :%u\n", ip->sa.sa_family));
 142                 return key;
 143         }
 144
 145         return key;
 146 }
 147
 148 /* Allocate any unassigned IPs just by looping through the IPs and
 149  * finding the best node for each.
 150  */
 151 void basic_allocate_unassigned(struct ipalloc_state *ipalloc_state)
 152 {
 153         struct public_ip_list *t;
 154
 155         /* loop over all ip's and find a physical node to cover for
 156            each unassigned ip.
 157         */
 158         for (t = ipalloc_state->all_ips; t != NULL; t = t->next) {
 159                 if (t->pnn == -1) {
 160                         if (find_takeover_node(ipalloc_state, t)) {
 161                                 DEBUG(DEBUG_WARNING,
 162                                       ("Failed to find node to cover ip %s\n",
 163                                        ctdb_sock_addr_to_string(ipalloc_state,
 164                                                                 &t->addr,
 165                                                                 false)));
 166                         }
 167                 }
 168         }
 169 }
 170
 171 void unassign_unsuitable_ips(struct ipalloc_state *ipalloc_state)
 172 {
 173         struct public_ip_list *t;
 174
 175         /* verify that the assigned nodes can serve that public ip
 176            and set it to -1 if not
 177         */
 178         for (t = ipalloc_state->all_ips; t != NULL; t = t->next) {
 179                 if (t->pnn == -1) {
 180                         continue;
 181                 }
 182                 if (!can_node_host_ip(ipalloc_state, t->pnn, t) != 0) {
 183                         /* this node can not serve this ip. */
 184                         DEBUG(DEBUG_DEBUG,("Unassign IP: %s from %d\n",
 185                                            ctdb_sock_addr_to_string(
 186                                                    ipalloc_state,
 187                                                    &t->addr, false),
 188                                            t->pnn));
 189                         t->pnn = -1;
 190                 }
 191         }
 192 }