server/ctdb_serverids.c

   1 /*
   2    ctdb_control protocol code to manage server ids
   3
   4    Copyright (C) Ronnie Sahlberg 2007
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, see <http://www.gnu.org/licenses/>.
  18 */
  19 #include "includes.h"
  20 #include "../include/ctdb_private.h"
  21 #include "../common/rb_tree.h"
  22
  23
  24 #define SERVER_ID_KEY_SIZE 3
  25 static uint32_t *get_server_id_key(struct ctdb_server_id *server_id)
  26 {
  27         static uint32_t key[SERVER_ID_KEY_SIZE];
  28
  29         key[0] = server_id->type;
  30         key[1] = server_id->vnn;
  31         key[2] = server_id->server_id;
  32
  33         return &key[0];
  34 }
  35
  36 /* add a server_id to the tree.
  37    if we had already 'data' in the tree then this is a duplicate and we can
  38    just talloc_free the structure in parm and leave data in the tree.
  39    othervise if this is a new node we return parm and that is inserted
  40    into the tree.
  41 */
  42 static void *add_server_id_callback(void *parm, void *data)
  43 {
  44         if (data) {
  45                 talloc_free(parm);
  46                 return data;
  47         }
  48         return parm;
  49 }
  50
  51 /*
  52   register a server id
  53   a serverid that is registered with ctdb will be automatically unregistered
  54   once the client domain socket dissappears.
  55  */
  56 int32_t ctdb_control_register_server_id(struct ctdb_context *ctdb,
  57                                  uint32_t client_id,
  58                                  TDB_DATA indata)
  59 {
  60         struct ctdb_server_id *server_id;
  61         struct ctdb_client *client = ctdb_reqid_find(ctdb, client_id, struct ctdb_client);
  62
  63
  64         if (client == NULL) {
  65                 DEBUG(0,(__location__ " Could not find client parent structure. You can not send this control to a remote node\n"));
  66                 return 1;
  67         }
  68
  69         /* hang the server_id structure off client before storing it in the
  70            tree so that is will be automatically destroyed when client
  71            is destroyed.
  72            when the structure is free'd it will be automatically
  73            removed from the tree
  74         */
  75         server_id = talloc_memdup(client, indata.dptr, indata.dsize);
  76         CTDB_NO_MEMORY(ctdb, server_id);
  77
  78         trbt_insertarray32_callback(ctdb->server_ids, SERVER_ID_KEY_SIZE,
  79                 get_server_id_key(server_id),
  80                 add_server_id_callback, server_id);
  81
  82         return 0;
  83 }
  84
  85
  86 /*
  87   check whether a server id exists
  88  */
  89 int32_t ctdb_control_check_server_id(struct ctdb_context *ctdb,
  90                                  TDB_DATA indata)
  91 {
  92         struct ctdb_server_id *server_id = (struct ctdb_server_id *)indata.dptr;
  93
  94         return (int32_t)trbt_lookuparray32(ctdb->server_ids,
  95                                 SERVER_ID_KEY_SIZE,
  96                                 get_server_id_key(server_id));
  97 }
  98
  99 /*
 100   unregisters a server id
 101  */
 102 int32_t ctdb_control_unregister_server_id(struct ctdb_context *ctdb,
 103                                  TDB_DATA indata)
 104 {
 105         struct ctdb_server_id *server_id = (struct ctdb_server_id *)indata.dptr;
 106
 107         talloc_free(trbt_lookuparray32(ctdb->server_ids,
 108                         SERVER_ID_KEY_SIZE,
 109                         get_server_id_key(server_id)));
 110         return 0;
 111 }
 112
 113