4 Copyright (C) Amitay Isaacs 2015
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/>.
21 #include "system/network.h"
22 #include "system/filesys.h"
28 #include "lib/util/tevent_unix.h"
30 #include "common/reqid.h"
31 #include "common/srvid.h"
32 #include "common/comm.h"
34 #include "protocol/protocol.h"
35 #include "protocol/protocol_api.h"
37 #include "client/client_private.h"
38 #include "client/client.h"
45 struct ctdb_client_message_state
{
46 struct ctdb_client_context
*client
;
50 static int ctdb_client_message_state_destructor(
51 struct ctdb_client_message_state
*state
);
52 static void ctdb_client_message_done(struct tevent_req
*subreq
);
54 struct tevent_req
*ctdb_client_message_send(TALLOC_CTX
*mem_ctx
,
55 struct tevent_context
*ev
,
56 struct ctdb_client_context
*client
,
58 struct ctdb_req_message
*message
)
60 struct tevent_req
*req
, *subreq
;
61 struct ctdb_client_message_state
*state
;
62 struct ctdb_req_header h
;
68 req
= tevent_req_create(mem_ctx
, &state
,
69 struct ctdb_client_message_state
);
74 reqid
= reqid_new(client
->idr
, state
);
75 if (reqid
== REQID_INVALID
) {
80 state
->client
= client
;
83 talloc_set_destructor(state
, ctdb_client_message_state_destructor
);
85 ctdb_req_header_fill(&h
, 0, CTDB_REQ_MESSAGE
, destnode
,
88 ret
= ctdb_req_message_push(&h
, message
, state
, &buf
, &buflen
);
90 tevent_req_error(req
, ret
);
91 return tevent_req_post(req
, ev
);
94 subreq
= comm_write_send(state
, ev
, client
->comm
, buf
, buflen
);
95 if (tevent_req_nomem(subreq
, req
)) {
96 return tevent_req_post(req
, ev
);
98 tevent_req_set_callback(subreq
, ctdb_client_message_done
, req
);
103 static int ctdb_client_message_state_destructor(
104 struct ctdb_client_message_state
*state
)
106 reqid_remove(state
->client
->idr
, state
->reqid
);
110 static void ctdb_client_message_done(struct tevent_req
*subreq
)
112 struct tevent_req
*req
= tevent_req_callback_data(
113 subreq
, struct tevent_req
);
117 status
= comm_write_recv(subreq
, &ret
);
120 tevent_req_error(req
, ret
);
124 tevent_req_done(req
);
127 bool ctdb_client_message_recv(struct tevent_req
*req
, int *perr
)
131 if (tevent_req_is_unix_error(req
, &err
)) {
141 void ctdb_client_req_message(struct ctdb_client_context
*client
,
142 uint8_t *buf
, size_t buflen
, uint32_t reqid
)
144 struct ctdb_req_header h
;
145 struct ctdb_req_message_data message
;
146 TALLOC_CTX
*tmp_ctx
= talloc_new(client
);
149 ret
= ctdb_req_message_data_pull(buf
, buflen
, &h
, tmp_ctx
, &message
);
154 srvid_dispatch(client
->srv
, message
.srvid
, CTDB_SRVID_ALL
,
156 talloc_free(tmp_ctx
);
160 * sync version of message send
163 int ctdb_client_message(TALLOC_CTX
*mem_ctx
, struct tevent_context
*ev
,
164 struct ctdb_client_context
*client
,
165 uint32_t destnode
, struct ctdb_req_message
*message
)
168 struct tevent_req
*req
;
172 tmp_ctx
= talloc_new(client
);
173 if (tmp_ctx
== NULL
) {
177 req
= ctdb_client_message_send(tmp_ctx
, ev
, client
, destnode
, message
);
179 talloc_free(tmp_ctx
);
183 tevent_req_poll(req
, ev
);
185 status
= ctdb_client_message_recv(req
, &ret
);
187 talloc_free(tmp_ctx
);
191 talloc_free(tmp_ctx
);
195 struct ctdb_client_set_message_handler_state
{
196 struct ctdb_client_context
*client
;
198 srvid_handler_fn handler
;
202 static void ctdb_client_set_message_handler_done(struct tevent_req
*subreq
);
204 struct tevent_req
*ctdb_client_set_message_handler_send(
206 struct tevent_context
*ev
,
207 struct ctdb_client_context
*client
,
209 srvid_handler_fn handler
,
212 struct tevent_req
*req
, *subreq
;
213 struct ctdb_client_set_message_handler_state
*state
;
214 struct ctdb_req_control request
;
216 req
= tevent_req_create(mem_ctx
, &state
,
217 struct ctdb_client_set_message_handler_state
);
222 state
->client
= client
;
223 state
->srvid
= srvid
;
224 state
->handler
= handler
;
225 state
->private_data
= private_data
;
227 ctdb_req_control_register_srvid(&request
, srvid
);
228 subreq
= ctdb_client_control_send(state
, ev
, client
, client
->pnn
,
229 tevent_timeval_zero(), &request
);
230 if (tevent_req_nomem(subreq
, req
)) {
231 return tevent_req_post(req
, ev
);
233 tevent_req_set_callback(subreq
, ctdb_client_set_message_handler_done
,
239 static void ctdb_client_set_message_handler_done(struct tevent_req
*subreq
)
241 struct tevent_req
*req
= tevent_req_callback_data(
242 subreq
, struct tevent_req
);
243 struct ctdb_client_set_message_handler_state
*state
= tevent_req_data(
244 req
, struct ctdb_client_set_message_handler_state
);
245 struct ctdb_reply_control
*reply
;
249 status
= ctdb_client_control_recv(subreq
, &ret
, state
, &reply
);
252 tevent_req_error(req
, ret
);
256 ret
= ctdb_reply_control_register_srvid(reply
);
259 tevent_req_error(req
, ret
);
263 ret
= srvid_register(state
->client
->srv
, state
->client
, state
->srvid
,
264 state
->handler
, state
->private_data
);
266 tevent_req_error(req
, ret
);
270 tevent_req_done(req
);
273 bool ctdb_client_set_message_handler_recv(struct tevent_req
*req
, int *perr
)
277 if (tevent_req_is_unix_error(req
, &err
)) {
286 struct ctdb_client_remove_message_handler_state
{
287 struct ctdb_client_context
*client
;
292 static void ctdb_client_remove_message_handler_done(struct tevent_req
*subreq
);
294 struct tevent_req
*ctdb_client_remove_message_handler_send(
296 struct tevent_context
*ev
,
297 struct ctdb_client_context
*client
,
301 struct tevent_req
*req
, *subreq
;
302 struct ctdb_client_remove_message_handler_state
*state
;
303 struct ctdb_req_control request
;
305 req
= tevent_req_create(mem_ctx
, &state
,
306 struct ctdb_client_remove_message_handler_state
);
311 state
->client
= client
;
312 state
->srvid
= srvid
;
313 state
->private_data
= private_data
;
315 ctdb_req_control_deregister_srvid(&request
, srvid
);
316 subreq
= ctdb_client_control_send(state
, ev
, client
, client
->pnn
,
317 tevent_timeval_zero(), &request
);
318 if (tevent_req_nomem(subreq
, req
)) {
319 return tevent_req_post(req
, ev
);
321 tevent_req_set_callback(subreq
,
322 ctdb_client_remove_message_handler_done
, req
);
327 static void ctdb_client_remove_message_handler_done(struct tevent_req
*subreq
)
329 struct tevent_req
*req
= tevent_req_callback_data(
330 subreq
, struct tevent_req
);
331 struct ctdb_client_remove_message_handler_state
*state
= tevent_req_data(
332 req
, struct ctdb_client_remove_message_handler_state
);
333 struct ctdb_reply_control
*reply
;
337 status
= ctdb_client_control_recv(subreq
, &ret
, state
, &reply
);
340 tevent_req_error(req
, ret
);
344 ret
= ctdb_reply_control_deregister_srvid(reply
);
347 tevent_req_error(req
, ret
);
351 ret
= srvid_deregister(state
->client
->srv
, state
->srvid
,
352 state
->private_data
);
354 tevent_req_error(req
, ret
);
358 tevent_req_done(req
);
361 bool ctdb_client_remove_message_handler_recv(struct tevent_req
*req
, int *perr
)
365 if (tevent_req_is_unix_error(req
, &err
)) {
374 int ctdb_client_set_message_handler(TALLOC_CTX
*mem_ctx
,
375 struct tevent_context
*ev
,
376 struct ctdb_client_context
*client
,
377 uint64_t srvid
, srvid_handler_fn handler
,
382 ret
= ctdb_ctrl_register_srvid(mem_ctx
, ev
, client
, client
->pnn
,
383 tevent_timeval_zero(), srvid
);
388 return srvid_register(client
->srv
, client
, srvid
,
389 handler
, private_data
);
392 int ctdb_client_remove_message_handler(TALLOC_CTX
*mem_ctx
,
393 struct tevent_context
*ev
,
394 struct ctdb_client_context
*client
,
395 uint64_t srvid
, void *private_data
)
399 ret
= ctdb_ctrl_deregister_srvid(mem_ctx
, ev
, client
, client
->pnn
,
400 tevent_timeval_zero(), srvid
);
405 return srvid_deregister(client
->srv
, srvid
, private_data
);