server/ctdb_call.c

   1 /*
   2    ctdb_call protocol code
   3
   4    Copyright (C) Andrew Tridgell  2006
   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 /*
  20   see http://wiki.samba.org/index.php/Samba_%26_Clustering for
  21   protocol design and packet details
  22 */
  23 #include "includes.h"
  24 #include "lib/events/events.h"
  25 #include "lib/tdb/include/tdb.h"
  26 #include "lib/util/dlinklist.h"
  27 #include "system/network.h"
  28 #include "system/filesys.h"
  29 #include "../include/ctdb_private.h"
  30
  31 /*
  32   find the ctdb_db from a db index
  33  */
  34  struct ctdb_db_context *find_ctdb_db(struct ctdb_context *ctdb, uint32_t id)
  35 {
  36         struct ctdb_db_context *ctdb_db;
  37
  38         for (ctdb_db=ctdb->db_list; ctdb_db; ctdb_db=ctdb_db->next) {
  39                 if (ctdb_db->db_id == id) {
  40                         break;
  41                 }
  42         }
  43         return ctdb_db;
  44 }
  45
  46
  47 /*
  48   a varient of input packet that can be used in lock requeue
  49 */
  50 static void ctdb_call_input_pkt(void *p, struct ctdb_req_header *hdr)
  51 {
  52         struct ctdb_context *ctdb = talloc_get_type(p, struct ctdb_context);
  53         ctdb_input_pkt(ctdb, hdr);
  54 }
  55
  56
  57 /*
  58   send an error reply
  59 */
  60 static void ctdb_send_error(struct ctdb_context *ctdb,
  61                             struct ctdb_req_header *hdr, uint32_t status,
  62                             const char *fmt, ...) PRINTF_ATTRIBUTE(4,5);
  63 static void ctdb_send_error(struct ctdb_context *ctdb,
  64                             struct ctdb_req_header *hdr, uint32_t status,
  65                             const char *fmt, ...)
  66 {
  67         va_list ap;
  68         struct ctdb_reply_error *r;
  69         char *msg;
  70         int msglen, len;
  71
  72         va_start(ap, fmt);
  73         msg = talloc_vasprintf(ctdb, fmt, ap);
  74         if (msg == NULL) {
  75                 ctdb_fatal(ctdb, "Unable to allocate error in ctdb_send_error\n");
  76         }
  77         va_end(ap);
  78
  79         msglen = strlen(msg)+1;
  80         len = offsetof(struct ctdb_reply_error, msg);
  81         r = ctdb_transport_allocate(ctdb, msg, CTDB_REPLY_ERROR, len + msglen,
  82                                     struct ctdb_reply_error);
  83         CTDB_NO_MEMORY_FATAL(ctdb, r);
  84
  85         r->hdr.destnode  = hdr->srcnode;
  86         r->hdr.reqid     = hdr->reqid;
  87         r->status        = status;
  88         r->msglen        = msglen;
  89         memcpy(&r->msg[0], msg, msglen);
  90
  91         ctdb_queue_packet(ctdb, &r->hdr);
  92
  93         talloc_free(msg);
  94 }
  95
  96
  97 /*
  98   send a redirect reply
  99 */
 100 static void ctdb_call_send_redirect(struct ctdb_context *ctdb,
 101                                     TDB_DATA key,
 102                                     struct ctdb_req_call *c,
 103                                     struct ctdb_ltdb_header *header)
 104 {
 105
 106         uint32_t lmaster = ctdb_lmaster(ctdb, &key);
 107         if (ctdb->pnn == lmaster) {
 108                 c->hdr.destnode = header->dmaster;
 109         } else if ((c->hopcount % ctdb->tunable.max_redirect_count) == 0) {
 110                 c->hdr.destnode = lmaster;
 111         } else {
 112                 c->hdr.destnode = header->dmaster;
 113         }
 114         c->hopcount++;
 115         ctdb_queue_packet(ctdb, &c->hdr);
 116 }
 117
 118
 119 /*
 120   send a dmaster reply
 121
 122   caller must have the chainlock before calling this routine. Caller must be
 123   the lmaster
 124 */
 125 static void ctdb_send_dmaster_reply(struct ctdb_db_context *ctdb_db,
 126                                     struct ctdb_ltdb_header *header,
 127                                     TDB_DATA key, TDB_DATA data,
 128                                     uint32_t new_dmaster,
 129                                     uint32_t reqid)
 130 {
 131         struct ctdb_context *ctdb = ctdb_db->ctdb;
 132         struct ctdb_reply_dmaster *r;
 133         int ret, len;
 134         TALLOC_CTX *tmp_ctx;
 135
 136         if (ctdb->pnn != ctdb_lmaster(ctdb, &key)) {
 137                 DEBUG(DEBUG_ALERT,(__location__ " Caller is not lmaster!\n"));
 138                 return;
 139         }
 140
 141         header->dmaster = new_dmaster;
 142         ret = ctdb_ltdb_store(ctdb_db, key, header, data);
 143         if (ret != 0) {
 144                 ctdb_fatal(ctdb, "ctdb_req_dmaster unable to update dmaster");
 145                 return;
 146         }
 147
 148         /* put the packet on a temporary context, allowing us to safely free
 149            it below even if ctdb_reply_dmaster() has freed it already */
 150         tmp_ctx = talloc_new(ctdb);
 151
 152         /* send the CTDB_REPLY_DMASTER */
 153         len = offsetof(struct ctdb_reply_dmaster, data) + key.dsize + data.dsize;
 154         r = ctdb_transport_allocate(ctdb, tmp_ctx, CTDB_REPLY_DMASTER, len,
 155                                     struct ctdb_reply_dmaster);
 156         CTDB_NO_MEMORY_FATAL(ctdb, r);
 157
 158         r->hdr.destnode  = new_dmaster;
 159         r->hdr.reqid     = reqid;
 160         r->rsn           = header->rsn;
 161         r->keylen        = key.dsize;
 162         r->datalen       = data.dsize;
 163         r->db_id         = ctdb_db->db_id;
 164         memcpy(&r->data[0], key.dptr, key.dsize);
 165         memcpy(&r->data[key.dsize], data.dptr, data.dsize);
 166
 167         ctdb_queue_packet(ctdb, &r->hdr);
 168
 169         talloc_free(tmp_ctx);
 170 }
 171
 172 /*
 173   send a dmaster request (give another node the dmaster for a record)
 174
 175   This is always sent to the lmaster, which ensures that the lmaster
 176   always knows who the dmaster is. The lmaster will then send a
 177   CTDB_REPLY_DMASTER to the new dmaster
 178 */
 179 static void ctdb_call_send_dmaster(struct ctdb_db_context *ctdb_db,
 180                                    struct ctdb_req_call *c,
 181                                    struct ctdb_ltdb_header *header,
 182                                    TDB_DATA *key, TDB_DATA *data)
 183 {
 184         struct ctdb_req_dmaster *r;
 185         struct ctdb_context *ctdb = ctdb_db->ctdb;
 186         int len;
 187         uint32_t lmaster = ctdb_lmaster(ctdb, key);
 188
 189         if (lmaster == ctdb->pnn) {
 190                 ctdb_send_dmaster_reply(ctdb_db, header, *key, *data,
 191                                         c->hdr.srcnode, c->hdr.reqid);
 192                 return;
 193         }
 194
 195         len = offsetof(struct ctdb_req_dmaster, data) + key->dsize + data->dsize;
 196         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REQ_DMASTER, len,
 197                                     struct ctdb_req_dmaster);
 198         CTDB_NO_MEMORY_FATAL(ctdb, r);
 199         r->hdr.destnode  = lmaster;
 200         r->hdr.reqid     = c->hdr.reqid;
 201         r->db_id         = c->db_id;
 202         r->rsn           = header->rsn;
 203         r->dmaster       = c->hdr.srcnode;
 204         r->keylen        = key->dsize;
 205         r->datalen       = data->dsize;
 206         memcpy(&r->data[0], key->dptr, key->dsize);
 207         memcpy(&r->data[key->dsize], data->dptr, data->dsize);
 208
 209         header->dmaster = c->hdr.srcnode;
 210         if (ctdb_ltdb_store(ctdb_db, *key, header, *data) != 0) {
 211                 ctdb_fatal(ctdb, "Failed to store record in ctdb_call_send_dmaster");
 212         }
 213
 214         ctdb_queue_packet(ctdb, &r->hdr);
 215
 216         talloc_free(r);
 217 }
 218
 219 /*
 220   called when a CTDB_REPLY_DMASTER packet comes in, or when the lmaster
 221   gets a CTDB_REQUEST_DMASTER for itself. We become the dmaster.
 222
 223   must be called with the chainlock held. This function releases the chainlock
 224 */
 225 static void ctdb_become_dmaster(struct ctdb_db_context *ctdb_db,
 226                                 struct ctdb_req_header *hdr,
 227                                 TDB_DATA key, TDB_DATA data,
 228                                 uint64_t rsn)
 229 {
 230         struct ctdb_call_state *state;
 231         struct ctdb_context *ctdb = ctdb_db->ctdb;
 232         struct ctdb_ltdb_header header;
 233
 234         DEBUG(DEBUG_INFO,("pnn %u dmaster response %08x\n", ctdb->pnn, ctdb_hash(&key)));
 235
 236         ZERO_STRUCT(header);
 237         header.rsn = rsn + 1;
 238         header.dmaster = ctdb->pnn;
 239
 240         if (ctdb_ltdb_store(ctdb_db, key, &header, data) != 0) {
 241                 ctdb_fatal(ctdb, "ctdb_reply_dmaster store failed\n");
 242                 ctdb_ltdb_unlock(ctdb_db, key);
 243                 return;
 244         }
 245
 246         state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_call_state);
 247
 248         if (state == NULL) {
 249                 DEBUG(DEBUG_ERR,("pnn %u Invalid reqid %u in ctdb_become_dmaster from node %u\n",
 250                          ctdb->pnn, hdr->reqid, hdr->srcnode));
 251                 ctdb_ltdb_unlock(ctdb_db, key);
 252                 return;
 253         }
 254
 255         if (hdr->reqid != state->reqid) {
 256                 /* we found a record  but it was the wrong one */
 257                 DEBUG(DEBUG_ERR, ("Dropped orphan in ctdb_become_dmaster with reqid:%u\n from node %u", hdr->reqid, hdr->srcnode));
 258                 ctdb_ltdb_unlock(ctdb_db, key);
 259                 return;
 260         }
 261
 262         ctdb_call_local(ctdb_db, state->call, &header, state, &data, ctdb->pnn);
 263
 264         ctdb_ltdb_unlock(ctdb_db, state->call->key);
 265
 266         state->state = CTDB_CALL_DONE;
 267         if (state->async.fn) {
 268                 state->async.fn(state);
 269         }
 270 }
 271
 272
 273
 274 /*
 275   called when a CTDB_REQ_DMASTER packet comes in
 276
 277   this comes into the lmaster for a record when the current dmaster
 278   wants to give up the dmaster role and give it to someone else
 279 */
 280 void ctdb_request_dmaster(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 281 {
 282         struct ctdb_req_dmaster *c = (struct ctdb_req_dmaster *)hdr;
 283         TDB_DATA key, data, data2;
 284         struct ctdb_ltdb_header header;
 285         struct ctdb_db_context *ctdb_db;
 286         int ret;
 287
 288         key.dptr = c->data;
 289         key.dsize = c->keylen;
 290         data.dptr = c->data + c->keylen;
 291         data.dsize = c->datalen;
 292
 293         ctdb_db = find_ctdb_db(ctdb, c->db_id);
 294         if (!ctdb_db) {
 295                 ctdb_send_error(ctdb, hdr, -1,
 296                                 "Unknown database in request. db_id==0x%08x",
 297                                 c->db_id);
 298                 return;
 299         }
 300
 301         /* fetch the current record */
 302         ret = ctdb_ltdb_lock_fetch_requeue(ctdb_db, key, &header, hdr, &data2,
 303                                            ctdb_call_input_pkt, ctdb, False);
 304         if (ret == -1) {
 305                 ctdb_fatal(ctdb, "ctdb_req_dmaster failed to fetch record");
 306                 return;
 307         }
 308         if (ret == -2) {
 309                 DEBUG(DEBUG_INFO,(__location__ " deferring ctdb_request_dmaster\n"));
 310                 return;
 311         }
 312
 313         if (ctdb_lmaster(ctdb, &key) != ctdb->pnn) {
 314                 DEBUG(DEBUG_ALERT,("pnn %u dmaster request to non-lmaster lmaster=%u gen=%u curgen=%u\n",
 315                          ctdb->pnn, ctdb_lmaster(ctdb, &key),
 316                          hdr->generation, ctdb->vnn_map->generation));
 317                 ctdb_fatal(ctdb, "ctdb_req_dmaster to non-lmaster");
 318         }
 319
 320         DEBUG(DEBUG_INFO,("pnn %u dmaster request on %08x for %u from %u\n",
 321                  ctdb->pnn, ctdb_hash(&key), c->dmaster, c->hdr.srcnode));
 322
 323         /* its a protocol error if the sending node is not the current dmaster */
 324         if (header.dmaster != hdr->srcnode) {
 325                 DEBUG(DEBUG_ALERT,("pnn %u dmaster request for new-dmaster %u from non-master %u real-dmaster=%u key %08x dbid 0x%08x gen=%u curgen=%u c->rsn=%llu header.rsn=%llu reqid=%u keyval=0x%08x\n",
 326                          ctdb->pnn, c->dmaster, hdr->srcnode, header.dmaster, ctdb_hash(&key),
 327                          ctdb_db->db_id, hdr->generation, ctdb->vnn_map->generation,
 328                          (unsigned long long)c->rsn, (unsigned long long)header.rsn, c->hdr.reqid,
 329                          (key.dsize >= 4)?(*(uint32_t *)key.dptr):0));
 330                 if (header.rsn != 0 || header.dmaster != ctdb->pnn) {
 331                         ctdb_fatal(ctdb, "ctdb_req_dmaster from non-master");
 332                         return;
 333                 }
 334         }
 335
 336         if (header.rsn > c->rsn) {
 337                 DEBUG(DEBUG_ALERT,("pnn %u dmaster request with older RSN new-dmaster %u from %u real-dmaster=%u key %08x dbid 0x%08x gen=%u curgen=%u c->rsn=%llu header.rsn=%llu reqid=%u\n",
 338                          ctdb->pnn, c->dmaster, hdr->srcnode, header.dmaster, ctdb_hash(&key),
 339                          ctdb_db->db_id, hdr->generation, ctdb->vnn_map->generation,
 340                          (unsigned long long)c->rsn, (unsigned long long)header.rsn, c->hdr.reqid));
 341         }
 342
 343         /* use the rsn from the sending node */
 344         header.rsn = c->rsn;
 345
 346         /* check if the new dmaster is the lmaster, in which case we
 347            skip the dmaster reply */
 348         if (c->dmaster == ctdb->pnn) {
 349                 ctdb_become_dmaster(ctdb_db, hdr, key, data, c->rsn);
 350         } else {
 351                 ctdb_send_dmaster_reply(ctdb_db, &header, key, data, c->dmaster, hdr->reqid);
 352                 ctdb_ltdb_unlock(ctdb_db, key);
 353         }
 354 }
 355
 356
 357 /*
 358   called when a CTDB_REQ_CALL packet comes in
 359 */
 360 void ctdb_request_call(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 361 {
 362         struct ctdb_req_call *c = (struct ctdb_req_call *)hdr;
 363         TDB_DATA data;
 364         struct ctdb_reply_call *r;
 365         int ret, len;
 366         struct ctdb_ltdb_header header;
 367         struct ctdb_call *call;
 368         struct ctdb_db_context *ctdb_db;
 369
 370         ctdb_db = find_ctdb_db(ctdb, c->db_id);
 371         if (!ctdb_db) {
 372                 ctdb_send_error(ctdb, hdr, -1,
 373                                 "Unknown database in request. db_id==0x%08x",
 374                                 c->db_id);
 375                 return;
 376         }
 377
 378         call = talloc(hdr, struct ctdb_call);
 379         CTDB_NO_MEMORY_FATAL(ctdb, call);
 380
 381         call->call_id  = c->callid;
 382         call->key.dptr = c->data;
 383         call->key.dsize = c->keylen;
 384         call->call_data.dptr = c->data + c->keylen;
 385         call->call_data.dsize = c->calldatalen;
 386
 387         /* determine if we are the dmaster for this key. This also
 388            fetches the record data (if any), thus avoiding a 2nd fetch of the data
 389            if the call will be answered locally */
 390
 391         ret = ctdb_ltdb_lock_fetch_requeue(ctdb_db, call->key, &header, hdr, &data,
 392                                            ctdb_call_input_pkt, ctdb, False);
 393         if (ret == -1) {
 394                 ctdb_send_error(ctdb, hdr, ret, "ltdb fetch failed in ctdb_request_call");
 395                 return;
 396         }
 397         if (ret == -2) {
 398                 DEBUG(DEBUG_INFO,(__location__ " deferred ctdb_request_call\n"));
 399                 return;
 400         }
 401
 402         /* if we are not the dmaster, then send a redirect to the
 403            requesting node */
 404         if (header.dmaster != ctdb->pnn) {
 405                 talloc_free(data.dptr);
 406                 ctdb_call_send_redirect(ctdb, call->key, c, &header);
 407                 ctdb_ltdb_unlock(ctdb_db, call->key);
 408                 return;
 409         }
 410
 411         if (c->hopcount > ctdb->statistics.max_hop_count) {
 412                 ctdb->statistics.max_hop_count = c->hopcount;
 413         }
 414
 415         /* if this nodes has done enough consecutive calls on the same record
 416            then give them the record
 417            or if the node requested an immediate migration
 418         */
 419         if ( c->hdr.srcnode != ctdb->pnn &&
 420              ((header.laccessor == c->hdr.srcnode
 421                && header.lacount >= ctdb->tunable.max_lacount)
 422               || (c->flags & CTDB_IMMEDIATE_MIGRATION)) ) {
 423                 DEBUG(DEBUG_INFO,("pnn %u starting migration of %08x to %u\n",
 424                          ctdb->pnn, ctdb_hash(&(call->key)), c->hdr.srcnode));
 425                 ctdb_call_send_dmaster(ctdb_db, c, &header, &(call->key), &data);
 426                 talloc_free(data.dptr);
 427                 ctdb_ltdb_unlock(ctdb_db, call->key);
 428                 return;
 429         }
 430
 431         ctdb_call_local(ctdb_db, call, &header, hdr, &data, c->hdr.srcnode);
 432
 433         ctdb_ltdb_unlock(ctdb_db, call->key);
 434
 435         len = offsetof(struct ctdb_reply_call, data) + call->reply_data.dsize;
 436         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REPLY_CALL, len,
 437                                     struct ctdb_reply_call);
 438         CTDB_NO_MEMORY_FATAL(ctdb, r);
 439         r->hdr.destnode  = hdr->srcnode;
 440         r->hdr.reqid     = hdr->reqid;
 441         r->status        = call->status;
 442         r->datalen       = call->reply_data.dsize;
 443         if (call->reply_data.dsize) {
 444                 memcpy(&r->data[0], call->reply_data.dptr, call->reply_data.dsize);
 445         }
 446
 447         ctdb_queue_packet(ctdb, &r->hdr);
 448
 449         talloc_free(r);
 450 }
 451
 452 /*
 453   called when a CTDB_REPLY_CALL packet comes in
 454
 455   This packet comes in response to a CTDB_REQ_CALL request packet. It
 456   contains any reply data from the call
 457 */
 458 void ctdb_reply_call(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 459 {
 460         struct ctdb_reply_call *c = (struct ctdb_reply_call *)hdr;
 461         struct ctdb_call_state *state;
 462
 463         state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_call_state);
 464         if (state == NULL) {
 465                 DEBUG(DEBUG_ERR, (__location__ " reqid %u not found\n", hdr->reqid));
 466                 return;
 467         }
 468
 469         if (hdr->reqid != state->reqid) {
 470                 /* we found a record  but it was the wrong one */
 471                 DEBUG(DEBUG_ERR, ("Dropped orphaned call reply with reqid:%u\n",hdr->reqid));
 472                 return;
 473         }
 474
 475         state->call->reply_data.dptr = c->data;
 476         state->call->reply_data.dsize = c->datalen;
 477         state->call->status = c->status;
 478
 479         talloc_steal(state, c);
 480
 481         state->state = CTDB_CALL_DONE;
 482         if (state->async.fn) {
 483                 state->async.fn(state);
 484         }
 485 }
 486
 487
 488 /*
 489   called when a CTDB_REPLY_DMASTER packet comes in
 490
 491   This packet comes in from the lmaster response to a CTDB_REQ_CALL
 492   request packet. It means that the current dmaster wants to give us
 493   the dmaster role
 494 */
 495 void ctdb_reply_dmaster(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 496 {
 497         struct ctdb_reply_dmaster *c = (struct ctdb_reply_dmaster *)hdr;
 498         struct ctdb_db_context *ctdb_db;
 499         TDB_DATA key, data;
 500         int ret;
 501
 502         ctdb_db = find_ctdb_db(ctdb, c->db_id);
 503         if (ctdb_db == NULL) {
 504                 DEBUG(DEBUG_ERR,("Unknown db_id 0x%x in ctdb_reply_dmaster\n", c->db_id));
 505                 return;
 506         }
 507
 508         key.dptr = c->data;
 509         key.dsize = c->keylen;
 510         data.dptr = &c->data[key.dsize];
 511         data.dsize = c->datalen;
 512
 513         ret = ctdb_ltdb_lock_requeue(ctdb_db, key, hdr,
 514                                      ctdb_call_input_pkt, ctdb, False);
 515         if (ret == -2) {
 516                 return;
 517         }
 518         if (ret != 0) {
 519                 DEBUG(DEBUG_ERR,(__location__ " Failed to get lock in ctdb_reply_dmaster\n"));
 520                 return;
 521         }
 522
 523         ctdb_become_dmaster(ctdb_db, hdr, key, data, c->rsn);
 524 }
 525
 526
 527 /*
 528   called when a CTDB_REPLY_ERROR packet comes in
 529 */
 530 void ctdb_reply_error(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 531 {
 532         struct ctdb_reply_error *c = (struct ctdb_reply_error *)hdr;
 533         struct ctdb_call_state *state;
 534
 535         state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_call_state);
 536         if (state == NULL) {
 537                 DEBUG(DEBUG_ERR,("pnn %u Invalid reqid %u in ctdb_reply_error\n",
 538                          ctdb->pnn, hdr->reqid));
 539                 return;
 540         }
 541
 542         if (hdr->reqid != state->reqid) {
 543                 /* we found a record  but it was the wrong one */
 544                 DEBUG(DEBUG_ERR, ("Dropped orphaned error reply with reqid:%u\n",hdr->reqid));
 545                 return;
 546         }
 547
 548         talloc_steal(state, c);
 549
 550         state->state  = CTDB_CALL_ERROR;
 551         state->errmsg = (char *)c->msg;
 552         if (state->async.fn) {
 553                 state->async.fn(state);
 554         }
 555 }
 556
 557
 558 /*
 559   destroy a ctdb_call
 560 */
 561 static int ctdb_call_destructor(struct ctdb_call_state *state)
 562 {
 563         DLIST_REMOVE(state->ctdb_db->ctdb->pending_calls, state);
 564         ctdb_reqid_remove(state->ctdb_db->ctdb, state->reqid);
 565         return 0;
 566 }
 567
 568
 569 /*
 570   called when a ctdb_call needs to be resent after a reconfigure event
 571 */
 572 static void ctdb_call_resend(struct ctdb_call_state *state)
 573 {
 574         struct ctdb_context *ctdb = state->ctdb_db->ctdb;
 575
 576         state->generation = ctdb->vnn_map->generation;
 577
 578         /* use a new reqid, in case the old reply does eventually come in */
 579         ctdb_reqid_remove(ctdb, state->reqid);
 580         state->reqid = ctdb_reqid_new(ctdb, state);
 581         state->c->hdr.reqid = state->reqid;
 582
 583         /* update the generation count for this request, so its valid with the new vnn_map */
 584         state->c->hdr.generation = state->generation;
 585
 586         /* send the packet to ourselves, it will be redirected appropriately */
 587         state->c->hdr.destnode = ctdb->pnn;
 588
 589         ctdb_queue_packet(ctdb, &state->c->hdr);
 590         DEBUG(DEBUG_NOTICE,("resent ctdb_call\n"));
 591 }
 592
 593 /*
 594   resend all pending calls on recovery
 595  */
 596 void ctdb_call_resend_all(struct ctdb_context *ctdb)
 597 {
 598         struct ctdb_call_state *state, *next;
 599         for (state=ctdb->pending_calls;state;state=next) {
 600                 next = state->next;
 601                 ctdb_call_resend(state);
 602         }
 603 }
 604
 605 /*
 606   this allows the caller to setup a async.fn
 607 */
 608 static void call_local_trigger(struct event_context *ev, struct timed_event *te,
 609                        struct timeval t, void *private_data)
 610 {
 611         struct ctdb_call_state *state = talloc_get_type(private_data, struct ctdb_call_state);
 612         if (state->async.fn) {
 613                 state->async.fn(state);
 614         }
 615 }
 616
 617
 618 /*
 619   construct an event driven local ctdb_call
 620
 621   this is used so that locally processed ctdb_call requests are processed
 622   in an event driven manner
 623 */
 624 struct ctdb_call_state *ctdb_call_local_send(struct ctdb_db_context *ctdb_db,
 625                                              struct ctdb_call *call,
 626                                              struct ctdb_ltdb_header *header,
 627                                              TDB_DATA *data)
 628 {
 629         struct ctdb_call_state *state;
 630         struct ctdb_context *ctdb = ctdb_db->ctdb;
 631         int ret;
 632
 633         state = talloc_zero(ctdb_db, struct ctdb_call_state);
 634         CTDB_NO_MEMORY_NULL(ctdb, state);
 635
 636         talloc_steal(state, data->dptr);
 637
 638         state->state = CTDB_CALL_DONE;
 639         state->call  = talloc(state, struct ctdb_call);
 640         CTDB_NO_MEMORY_NULL(ctdb, state->call);
 641         *(state->call) = *call;
 642         state->ctdb_db = ctdb_db;
 643
 644         ret = ctdb_call_local(ctdb_db, state->call, header, state, data, ctdb->pnn);
 645
 646         event_add_timed(ctdb->ev, state, timeval_zero(), call_local_trigger, state);
 647
 648         return state;
 649 }
 650
 651
 652 /*
 653   make a remote ctdb call - async send. Called in daemon context.
 654
 655   This constructs a ctdb_call request and queues it for processing.
 656   This call never blocks.
 657 */
 658 struct ctdb_call_state *ctdb_daemon_call_send_remote(struct ctdb_db_context *ctdb_db,
 659                                                      struct ctdb_call *call,
 660                                                      struct ctdb_ltdb_header *header)
 661 {
 662         uint32_t len;
 663         struct ctdb_call_state *state;
 664         struct ctdb_context *ctdb = ctdb_db->ctdb;
 665
 666         state = talloc_zero(ctdb_db, struct ctdb_call_state);
 667         CTDB_NO_MEMORY_NULL(ctdb, state);
 668         state->call = talloc(state, struct ctdb_call);
 669         CTDB_NO_MEMORY_NULL(ctdb, state->call);
 670
 671         state->reqid = ctdb_reqid_new(ctdb, state);
 672         state->ctdb_db = ctdb_db;
 673         talloc_set_destructor(state, ctdb_call_destructor);
 674
 675         len = offsetof(struct ctdb_req_call, data) + call->key.dsize + call->call_data.dsize;
 676         state->c = ctdb_transport_allocate(ctdb, state, CTDB_REQ_CALL, len,
 677                                            struct ctdb_req_call);
 678         CTDB_NO_MEMORY_NULL(ctdb, state->c);
 679         state->c->hdr.destnode  = header->dmaster;
 680
 681         /* this limits us to 16k outstanding messages - not unreasonable */
 682         state->c->hdr.reqid     = state->reqid;
 683         state->c->flags         = call->flags;
 684         state->c->db_id         = ctdb_db->db_id;
 685         state->c->callid        = call->call_id;
 686         state->c->hopcount      = 0;
 687         state->c->keylen        = call->key.dsize;
 688         state->c->calldatalen   = call->call_data.dsize;
 689         memcpy(&state->c->data[0], call->key.dptr, call->key.dsize);
 690         memcpy(&state->c->data[call->key.dsize],
 691                call->call_data.dptr, call->call_data.dsize);
 692         *(state->call)              = *call;
 693         state->call->call_data.dptr = &state->c->data[call->key.dsize];
 694         state->call->key.dptr       = &state->c->data[0];
 695
 696         state->state  = CTDB_CALL_WAIT;
 697         state->generation = ctdb->vnn_map->generation;
 698
 699         DLIST_ADD(ctdb->pending_calls, state);
 700
 701         ctdb_queue_packet(ctdb, &state->c->hdr);
 702
 703         return state;
 704 }
 705
 706 /*
 707   make a remote ctdb call - async recv - called in daemon context
 708
 709   This is called when the program wants to wait for a ctdb_call to complete and get the
 710   results. This call will block unless the call has already completed.
 711 */
 712 int ctdb_daemon_call_recv(struct ctdb_call_state *state, struct ctdb_call *call)
 713 {
 714         while (state->state < CTDB_CALL_DONE) {
 715                 event_loop_once(state->ctdb_db->ctdb->ev);
 716         }
 717         if (state->state != CTDB_CALL_DONE) {
 718                 ctdb_set_error(state->ctdb_db->ctdb, "%s", state->errmsg);
 719                 talloc_free(state);
 720                 return -1;
 721         }
 722
 723         if (state->call->reply_data.dsize) {
 724                 call->reply_data.dptr = talloc_memdup(call,
 725                                                       state->call->reply_data.dptr,
 726                                                       state->call->reply_data.dsize);
 727                 call->reply_data.dsize = state->call->reply_data.dsize;
 728         } else {
 729                 call->reply_data.dptr = NULL;
 730                 call->reply_data.dsize = 0;
 731         }
 732         call->status = state->call->status;
 733         talloc_free(state);
 734         return 0;
 735 }
 736
 737
 738 /*
 739    send a keepalive packet to the other node
 740 */
 741 void ctdb_send_keepalive(struct ctdb_context *ctdb, uint32_t destnode)
 742 {
 743         struct ctdb_req_keepalive *r;
 744
 745         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REQ_KEEPALIVE,
 746                                     sizeof(struct ctdb_req_keepalive),
 747                                     struct ctdb_req_keepalive);
 748         CTDB_NO_MEMORY_FATAL(ctdb, r);
 749         r->hdr.destnode  = destnode;
 750         r->hdr.reqid     = 0;
 751
 752         ctdb->statistics.keepalive_packets_sent++;
 753
 754         ctdb_queue_packet(ctdb, &r->hdr);
 755
 756         talloc_free(r);
 757 }