2 ctdb_control protocol code to manage server ids
4 Copyright (C) Ronnie Sahlberg 2007
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.
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.
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/>.
20 #include "../include/ctdb_private.h"
21 #include "../common/rb_tree.h"
24 #define SERVER_ID_KEY_SIZE 3
25 static uint32_t *get_server_id_key(struct ctdb_server_id *server_id)
27 static uint32_t key[SERVER_ID_KEY_SIZE];
29 key[0] = server_id->type;
30 key[1] = server_id->vnn;
31 key[2] = server_id->server_id;
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
42 static void *add_server_id_callback(void *parm, void *data)
53 a serverid that is registered with ctdb will be automatically unregistered
54 once the client domain socket dissappears.
56 int32_t ctdb_control_register_server_id(struct ctdb_context *ctdb,
60 struct ctdb_server_id *server_id;
61 struct ctdb_client *client = ctdb_reqid_find(ctdb, client_id, struct ctdb_client);
65 DEBUG(0,(__location__ " Could not find client parent structure. You can not send this control to a remote node\n"));
69 /* hang the server_id structure off client before storing it in the
70 tree so that is will be automatically destroyed when client
72 when the structure is free'd it will be automatically
75 server_id = talloc_zero(client, struct ctdb_server_id);
76 CTDB_NO_MEMORY(ctdb, server_id);
77 memcpy(server_id, indata.dptr, sizeof(struct ctdb_server_id));
79 trbt_insertarray32_callback(ctdb->server_ids, SERVER_ID_KEY_SIZE,
80 get_server_id_key(server_id),
81 add_server_id_callback, server_id);
88 check whether a server id exists
90 int32_t ctdb_control_check_server_id(struct ctdb_context *ctdb,
93 struct ctdb_server_id *server_id = (struct ctdb_server_id *)indata.dptr;
95 return (int32_t)trbt_lookuparray32(ctdb->server_ids,
97 get_server_id_key(server_id));
101 unregisters a server id
103 int32_t ctdb_control_unregister_server_id(struct ctdb_context *ctdb,
106 struct ctdb_server_id *server_id = (struct ctdb_server_id *)indata.dptr;
108 talloc_free(trbt_lookuparray32(ctdb->server_ids,
110 get_server_id_key(server_id)));
117 struct count_server_ids {
119 struct ctdb_server_id_list *list;
122 static void server_id_count(void *param, void *data)
124 struct count_server_ids *svid = talloc_get_type(param,
125 struct count_server_ids);
128 DEBUG(0, (__location__ " Got null pointer for svid\n"));
135 static void server_id_store(void *param, void *data)
137 struct count_server_ids *svid = talloc_get_type(param,
138 struct count_server_ids);
139 struct ctdb_server_id *server_id = talloc_get_type(data,
140 struct ctdb_server_id);
143 DEBUG(0, (__location__ " Got null pointer for svid\n"));
147 if (svid->count >= svid->list->num) {
148 DEBUG(0, (__location__ " size of server id tree changed during traverse\n"));
152 memcpy(&svid->list->server_ids[svid->count], server_id, sizeof(struct ctdb_server_id));
157 returns a list of all registered server ids for a node
159 int32_t ctdb_control_get_server_id_list(struct ctdb_context *ctdb, TDB_DATA *outdata)
161 struct count_server_ids *svid;
164 svid = talloc_zero(outdata, struct count_server_ids);
165 CTDB_NO_MEMORY(ctdb, svid);
168 /* first we must count how many entries we have */
169 trbt_traversearray32(ctdb->server_ids, SERVER_ID_KEY_SIZE,
170 server_id_count, svid);
173 outdata->dsize = offsetof(struct ctdb_server_id_list,
175 + sizeof(struct ctdb_server_id) * svid->count;
176 outdata->dptr = talloc_size(outdata, outdata->dsize);
177 CTDB_NO_MEMORY(ctdb, outdata->dptr);
180 /* now fill the structure in */
181 svid->list = (struct ctdb_server_id_list *)(outdata->dptr);
182 svid->list->num = svid->count;
184 trbt_traversearray32(ctdb->server_ids, SERVER_ID_KEY_SIZE,
185 server_id_store, svid);