clean out some more cruft
[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   this local messaging handler is ugly, but is needed to prevent
64   recursion in ctdb_send_message() when the destination node is the
65   same as the source node
66  */
67 struct ctdb_local_message {
68         struct ctdb_context *ctdb;
69         uint64_t srvid;
70         TDB_DATA data;
71 };
72
73 static void ctdb_local_message_trigger(struct event_context *ev, struct timed_event *te, 
74                                        struct timeval t, void *private_data)
75 {
76         struct ctdb_local_message *m = talloc_get_type(private_data, 
77                                                        struct ctdb_local_message);
78         int res;
79
80         res = ctdb_dispatch_message(m->ctdb, m->srvid, m->data);
81         if (res != 0) {
82                 DEBUG(0, (__location__ " Failed to dispatch message for srvid=%llu\n", 
83                           (unsigned long long)m->srvid));
84         }
85         talloc_free(m);
86 }
87
88 static int ctdb_local_message(struct ctdb_context *ctdb, uint64_t srvid, TDB_DATA data)
89 {
90         struct ctdb_local_message *m;
91         m = talloc(ctdb, struct ctdb_local_message);
92         CTDB_NO_MEMORY(ctdb, m);
93
94         m->ctdb = ctdb;
95         m->srvid = srvid;
96         m->data  = data;
97         m->data.dptr = talloc_memdup(m, m->data.dptr, m->data.dsize);
98         if (m->data.dptr == NULL) {
99                 talloc_free(m);
100                 return -1;
101         }
102
103         /* this needs to be done as an event to prevent recursion */
104         event_add_timed(ctdb->ev, m, timeval_zero(), ctdb_local_message_trigger, m);
105         return 0;
106 }
107
108 /*
109   send a ctdb message
110 */
111 int ctdb_daemon_send_message(struct ctdb_context *ctdb, uint32_t vnn,
112                              uint64_t srvid, TDB_DATA data)
113 {
114         struct ctdb_req_message *r;
115         int len;
116
117         /* see if this is a message to ourselves */
118         if (vnn == ctdb->vnn) {
119                 return ctdb_local_message(ctdb, srvid, data);
120         }
121
122         len = offsetof(struct ctdb_req_message, data) + data.dsize;
123         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REQ_MESSAGE, len,
124                                     struct ctdb_req_message);
125         CTDB_NO_MEMORY(ctdb, r);
126
127         r->hdr.destnode  = vnn;
128         r->srvid         = srvid;
129         r->datalen       = data.dsize;
130         memcpy(&r->data[0], data.dptr, data.dsize);
131
132         ctdb_queue_packet(ctdb, &r->hdr);
133
134         talloc_free(r);
135         return 0;
136 }
137
138
139 /*
140   when a client goes away, we need to remove its srvid handler from the list
141  */
142 static int message_handler_destructor(struct ctdb_message_list *m)
143 {
144         DLIST_REMOVE(m->ctdb->message_list, m);
145         return 0;
146 }
147
148 /*
149   setup handler for receipt of ctdb messages from ctdb_send_message()
150 */
151 int ctdb_register_message_handler(struct ctdb_context *ctdb, 
152                                   TALLOC_CTX *mem_ctx,
153                                   uint64_t srvid,
154                                   ctdb_message_fn_t handler,
155                                   void *private_data)
156 {
157         struct ctdb_message_list *m;
158
159         m = talloc(mem_ctx, struct ctdb_message_list);
160         CTDB_NO_MEMORY(ctdb, m);
161
162         m->ctdb            = ctdb;
163         m->srvid           = srvid;
164         m->message_handler = handler;
165         m->message_private = private_data;
166         
167         DLIST_ADD(ctdb->message_list, m);
168
169         talloc_set_destructor(m, message_handler_destructor);
170
171         return 0;
172 }
173
174
175 /*
176   setup handler for receipt of ctdb messages from ctdb_send_message()
177 */
178 int ctdb_deregister_message_handler(struct ctdb_context *ctdb, uint64_t srvid, void *private_data)
179 {
180         struct ctdb_message_list *m;
181
182         for (m=ctdb->message_list;m;m=m->next) {
183                 if (m->srvid == srvid && m->message_private == private_data) {
184                         talloc_free(m);
185                         return 0;
186                 }
187         }
188         return -1;
189 }