ctdb-ipalloc: Move ipalloc state initialisation to ipalloc.c

[sharpe/samba-autobuild/.git] / ctdb / server / ipalloc.c

diff --git a/ctdb/server/ipalloc.c b/ctdb/server/ipalloc.c
index 295671a924f53bddc03e242ea7f5eff2fe1cb33f..b1a74bcfd1b2ff23d55d652cd95840d34c61f1f9 100644 (file)
--- a/ctdb/server/ipalloc.c
+++ b/ctdb/server/ipalloc.c
@@ -31,6 +31,50 @@
 
 #include "server/ipalloc_private.h"
 
+/* Initialise main ipalloc state and sub-structures */
+struct ipalloc_state *
+ipalloc_state_init(TALLOC_CTX *mem_ctx,
+                  uint32_t num_nodes,
+                  enum ipalloc_algorithm algorithm,
+                  bool no_ip_failback,
+                  uint32_t *force_rebalance_nodes)
+{
+       struct ipalloc_state *ipalloc_state =
+               talloc_zero(mem_ctx, struct ipalloc_state);
+       if (ipalloc_state == NULL) {
+               DEBUG(DEBUG_ERR, (__location__ " Out of memory\n"));
+               return NULL;
+       }
+
+       ipalloc_state->num = num_nodes;
+
+       ipalloc_state->noiptakeover =
+               talloc_zero_array(ipalloc_state,
+                                 bool,
+                                 ipalloc_state->num);
+       if (ipalloc_state->noiptakeover == NULL) {
+               DEBUG(DEBUG_ERR, (__location__ " Out of memory\n"));
+               goto fail;
+       }
+       ipalloc_state->noiphost =
+               talloc_zero_array(ipalloc_state,
+                                 bool,
+                                 ipalloc_state->num);
+       if (ipalloc_state->noiphost == NULL) {
+               DEBUG(DEBUG_ERR, (__location__ " Out of memory\n"));
+               goto fail;
+       }
+
+       ipalloc_state->algorithm = algorithm;
+       ipalloc_state->no_ip_failback = no_ip_failback;
+       ipalloc_state->force_rebalance_nodes = force_rebalance_nodes;
+
+       return ipalloc_state;
+fail:
+       talloc_free(ipalloc_state);
+       return NULL;
+}
+
 static void *add_ip_callback(void *parm, void *data)
 {
        struct public_ip_list *this_ip = parm;