87010e464cbb056f2b291f1c8fcbef0083fdd51f
[vlendec/samba-autobuild/.git] / ctdb / common / ctdb_message.c
1 /* 
2    ctdb_message protocol code
3
4    Copyright (C) Andrew Tridgell  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 2 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, write to the Free Software
18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19 */
20 /*
21   see http://wiki.samba.org/index.php/Samba_%26_Clustering for
22   protocol design and packet details
23 */
24 #include "includes.h"
25 #include "lib/events/events.h"
26 #include "lib/tdb/include/tdb.h"
27 #include "system/network.h"
28 #include "system/filesys.h"
29 #include "../include/ctdb_private.h"
30 #include "lib/util/dlinklist.h"
31
32 /*
33   this dispatches the messages to the registered ctdb message handler
34 */
35 int ctdb_dispatch_message(struct ctdb_context *ctdb, uint64_t srvid, TDB_DATA data)
36 {
37         struct ctdb_message_list *ml;
38
39         for (ml=ctdb->message_list;ml;ml=ml->next) {
40                 if (ml->srvid == srvid || ml->srvid == CTDB_SRVID_ALL) {
41                         ml->message_handler(ctdb, srvid, data, ml->message_private);
42                 }
43         }
44
45         return 0;
46 }
47
48 /*
49   called when a CTDB_REQ_MESSAGE packet comes in
50 */
51 void ctdb_request_message(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
52 {
53         struct ctdb_req_message *c = (struct ctdb_req_message *)hdr;
54         TDB_DATA data;
55
56         data.dptr = &c->data[0];
57         data.dsize = c->datalen;
58
59         ctdb_dispatch_message(ctdb, c->srvid, data);
60 }
61
62
63 /*
64   when a client goes away, we need to remove its srvid handler from the list
65  */
66 static int message_handler_destructor(struct ctdb_message_list *m)
67 {
68         DLIST_REMOVE(m->ctdb->message_list, m);
69         return 0;
70 }
71
72 /*
73   setup handler for receipt of ctdb messages from ctdb_send_message()
74 */
75 int ctdb_register_message_handler(struct ctdb_context *ctdb, 
76                                   TALLOC_CTX *mem_ctx,
77                                   uint64_t srvid,
78                                   ctdb_message_fn_t handler,
79                                   void *private_data)
80 {
81         struct ctdb_message_list *m;
82
83         m = talloc(mem_ctx, struct ctdb_message_list);
84         CTDB_NO_MEMORY(ctdb, m);
85
86         m->ctdb            = ctdb;
87         m->srvid           = srvid;
88         m->message_handler = handler;
89         m->message_private = private_data;
90         
91         DLIST_ADD(ctdb->message_list, m);
92
93         talloc_set_destructor(m, message_handler_destructor);
94
95         return 0;
96 }
97
98
99 /*
100   setup handler for receipt of ctdb messages from ctdb_send_message()
101 */
102 int ctdb_deregister_message_handler(struct ctdb_context *ctdb, uint64_t srvid, void *private_data)
103 {
104         struct ctdb_message_list *m;
105
106         for (m=ctdb->message_list;m;m=m->next) {
107                 if (m->srvid == srvid && m->message_private == private_data) {
108                         talloc_free(m);
109                         return 0;
110                 }
111         }
112         return -1;
113 }