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/tevent/tevent.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         if (ctdb->methods == NULL) {
  73                 DEBUG(DEBUG_INFO,(__location__ " Failed to send error. Transport is DOWN\n"));
  74                 return;
  75         }
  76
  77         va_start(ap, fmt);
  78         msg = talloc_vasprintf(ctdb, fmt, ap);
  79         if (msg == NULL) {
  80                 ctdb_fatal(ctdb, "Unable to allocate error in ctdb_send_error\n");
  81         }
  82         va_end(ap);
  83
  84         msglen = strlen(msg)+1;
  85         len = offsetof(struct ctdb_reply_error, msg);
  86         r = ctdb_transport_allocate(ctdb, msg, CTDB_REPLY_ERROR, len + msglen,
  87                                     struct ctdb_reply_error);
  88         CTDB_NO_MEMORY_FATAL(ctdb, r);
  89
  90         r->hdr.destnode  = hdr->srcnode;
  91         r->hdr.reqid     = hdr->reqid;
  92         r->status        = status;
  93         r->msglen        = msglen;
  94         memcpy(&r->msg[0], msg, msglen);
  95
  96         ctdb_queue_packet(ctdb, &r->hdr);
  97
  98         talloc_free(msg);
  99 }
 100
 101
 102 /**
 103  * send a redirect reply
 104  *
 105  * The logic behind this function is this:
 106  *
 107  * A client wants to grab a record and sends a CTDB_REQ_CALL packet
 108  * to its local ctdb (ctdb_request_call). If the node is not itself
 109  * the record's DMASTER, it first redirects the packet to  the
 110  * record's LMASTER. The LMASTER then redirects the call packet to
 111  * the current DMASTER. But there is a race: The record may have
 112  * been migrated off the DMASTER while the redirected packet is
 113  * on the wire (or in the local queue). So in case the record has
 114  * migrated off the new destinaton of the call packet, instead of
 115  * going back to the LMASTER to get the new DMASTER, we try to
 116  * reduce rountrips by fist chasing the record a couple of times
 117  * before giving up the direct chase and finally going back to the
 118  * LMASTER (again). Note that this works because auf this: When
 119  * a record is migrated off a node, then the new DMASTER is stored
 120  * in the record's copy on the former DMASTER.
 121  *
 122  * The maxiumum number of attempts for direct chase to make before
 123  * going back to the LMASTER is configurable by the tunable
 124  * "MaxRedirectCount".
 125  */
 126 static void ctdb_call_send_redirect(struct ctdb_context *ctdb,
 127                                     TDB_DATA key,
 128                                     struct ctdb_req_call *c,
 129                                     struct ctdb_ltdb_header *header)
 130 {
 131
 132         uint32_t lmaster = ctdb_lmaster(ctdb, &key);
 133         if (ctdb->pnn == lmaster) {
 134                 c->hdr.destnode = header->dmaster;
 135         } else if ((c->hopcount % ctdb->tunable.max_redirect_count) == 0) {
 136                 c->hdr.destnode = lmaster;
 137         } else {
 138                 c->hdr.destnode = header->dmaster;
 139         }
 140         c->hopcount++;
 141         ctdb_queue_packet(ctdb, &c->hdr);
 142 }
 143
 144
 145 /*
 146   send a dmaster reply
 147
 148   caller must have the chainlock before calling this routine. Caller must be
 149   the lmaster
 150 */
 151 static void ctdb_send_dmaster_reply(struct ctdb_db_context *ctdb_db,
 152                                     struct ctdb_ltdb_header *header,
 153                                     TDB_DATA key, TDB_DATA data,
 154                                     uint32_t new_dmaster,
 155                                     uint32_t reqid)
 156 {
 157         struct ctdb_context *ctdb = ctdb_db->ctdb;
 158         struct ctdb_reply_dmaster *r;
 159         int ret, len;
 160         TALLOC_CTX *tmp_ctx;
 161
 162         if (ctdb->pnn != ctdb_lmaster(ctdb, &key)) {
 163                 DEBUG(DEBUG_ALERT,(__location__ " Caller is not lmaster!\n"));
 164                 return;
 165         }
 166
 167         header->dmaster = new_dmaster;
 168         ret = ctdb_ltdb_store(ctdb_db, key, header, data);
 169         if (ret != 0) {
 170                 ctdb_fatal(ctdb, "ctdb_send_dmaster_reply unable to update dmaster");
 171                 return;
 172         }
 173
 174         if (ctdb->methods == NULL) {
 175                 ctdb_fatal(ctdb, "ctdb_send_dmaster_reply cant update dmaster since transport is down");
 176                 return;
 177         }
 178
 179         /* put the packet on a temporary context, allowing us to safely free
 180            it below even if ctdb_reply_dmaster() has freed it already */
 181         tmp_ctx = talloc_new(ctdb);
 182
 183         /* send the CTDB_REPLY_DMASTER */
 184         len = offsetof(struct ctdb_reply_dmaster, data) + key.dsize + data.dsize + sizeof(uint32_t);
 185         r = ctdb_transport_allocate(ctdb, tmp_ctx, CTDB_REPLY_DMASTER, len,
 186                                     struct ctdb_reply_dmaster);
 187         CTDB_NO_MEMORY_FATAL(ctdb, r);
 188
 189         r->hdr.destnode  = new_dmaster;
 190         r->hdr.reqid     = reqid;
 191         r->rsn           = header->rsn;
 192         r->keylen        = key.dsize;
 193         r->datalen       = data.dsize;
 194         r->db_id         = ctdb_db->db_id;
 195         memcpy(&r->data[0], key.dptr, key.dsize);
 196         memcpy(&r->data[key.dsize], data.dptr, data.dsize);
 197         memcpy(&r->data[key.dsize+data.dsize], &header->flags, sizeof(uint32_t));
 198
 199         ctdb_queue_packet(ctdb, &r->hdr);
 200
 201         talloc_free(tmp_ctx);
 202 }
 203
 204 /*
 205   send a dmaster request (give another node the dmaster for a record)
 206
 207   This is always sent to the lmaster, which ensures that the lmaster
 208   always knows who the dmaster is. The lmaster will then send a
 209   CTDB_REPLY_DMASTER to the new dmaster
 210 */
 211 static void ctdb_call_send_dmaster(struct ctdb_db_context *ctdb_db,
 212                                    struct ctdb_req_call *c,
 213                                    struct ctdb_ltdb_header *header,
 214                                    TDB_DATA *key, TDB_DATA *data)
 215 {
 216         struct ctdb_req_dmaster *r;
 217         struct ctdb_context *ctdb = ctdb_db->ctdb;
 218         int len;
 219         uint32_t lmaster = ctdb_lmaster(ctdb, key);
 220
 221         if (ctdb->methods == NULL) {
 222                 ctdb_fatal(ctdb, "Failed ctdb_call_send_dmaster since transport is down");
 223                 return;
 224         }
 225
 226         if (data->dsize != 0) {
 227                 header->flags |= CTDB_REC_FLAG_MIGRATED_WITH_DATA;
 228         }
 229
 230         if (lmaster == ctdb->pnn) {
 231                 ctdb_send_dmaster_reply(ctdb_db, header, *key, *data,
 232                                         c->hdr.srcnode, c->hdr.reqid);
 233                 return;
 234         }
 235
 236         len = offsetof(struct ctdb_req_dmaster, data) + key->dsize + data->dsize
 237                         + sizeof(uint32_t);
 238         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REQ_DMASTER, len,
 239                                     struct ctdb_req_dmaster);
 240         CTDB_NO_MEMORY_FATAL(ctdb, r);
 241         r->hdr.destnode  = lmaster;
 242         r->hdr.reqid     = c->hdr.reqid;
 243         r->db_id         = c->db_id;
 244         r->rsn           = header->rsn;
 245         r->dmaster       = c->hdr.srcnode;
 246         r->keylen        = key->dsize;
 247         r->datalen       = data->dsize;
 248         memcpy(&r->data[0], key->dptr, key->dsize);
 249         memcpy(&r->data[key->dsize], data->dptr, data->dsize);
 250         memcpy(&r->data[key->dsize + data->dsize], &header->flags, sizeof(uint32_t));
 251
 252         header->dmaster = c->hdr.srcnode;
 253         if (ctdb_ltdb_store(ctdb_db, *key, header, *data) != 0) {
 254                 ctdb_fatal(ctdb, "Failed to store record in ctdb_call_send_dmaster");
 255         }
 256
 257         ctdb_queue_packet(ctdb, &r->hdr);
 258
 259         talloc_free(r);
 260 }
 261
 262 /*
 263   called when a CTDB_REPLY_DMASTER packet comes in, or when the lmaster
 264   gets a CTDB_REQUEST_DMASTER for itself. We become the dmaster.
 265
 266   must be called with the chainlock held. This function releases the chainlock
 267 */
 268 static void ctdb_become_dmaster(struct ctdb_db_context *ctdb_db,
 269                                 struct ctdb_req_header *hdr,
 270                                 TDB_DATA key, TDB_DATA data,
 271                                 uint64_t rsn, uint32_t record_flags)
 272 {
 273         struct ctdb_call_state *state;
 274         struct ctdb_context *ctdb = ctdb_db->ctdb;
 275         struct ctdb_ltdb_header header;
 276         int ret;
 277
 278         DEBUG(DEBUG_DEBUG,("pnn %u dmaster response %08x\n", ctdb->pnn, ctdb_hash(&key)));
 279
 280         ZERO_STRUCT(header);
 281         header.rsn = rsn + 1;
 282         header.dmaster = ctdb->pnn;
 283         header.flags = record_flags;
 284
 285         state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_call_state);
 286
 287         if (state) {
 288                 if (state->call->flags & CTDB_CALL_FLAG_VACUUM_MIGRATION) {
 289                         /*
 290                          * We temporarily add the VACUUM_MIGRATED flag to
 291                          * the record flags, so that ctdb_ltdb_store can
 292                          * decide whether the record should be stored or
 293                          * deleted.
 294                          */
 295                         header.flags |= CTDB_REC_FLAG_VACUUM_MIGRATED;
 296                 }
 297         }
 298
 299         if (ctdb_ltdb_store(ctdb_db, key, &header, data) != 0) {
 300                 ctdb_fatal(ctdb, "ctdb_reply_dmaster store failed\n");
 301
 302                 ret = ctdb_ltdb_unlock(ctdb_db, key);
 303                 if (ret != 0) {
 304                         DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 305                 }
 306                 return;
 307         }
 308
 309
 310         if (state == NULL) {
 311                 DEBUG(DEBUG_ERR,("pnn %u Invalid reqid %u in ctdb_become_dmaster from node %u\n",
 312                          ctdb->pnn, hdr->reqid, hdr->srcnode));
 313
 314                 ret = ctdb_ltdb_unlock(ctdb_db, key);
 315                 if (ret != 0) {
 316                         DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 317                 }
 318                 return;
 319         }
 320
 321         if (key.dsize != state->call->key.dsize || memcmp(key.dptr, state->call->key.dptr, key.dsize)) {
 322                 DEBUG(DEBUG_ERR, ("Got bogus DMASTER packet reqid:%u from node %u. Key does not match key held in matching idr.\n", hdr->reqid, hdr->srcnode));
 323
 324                 ret = ctdb_ltdb_unlock(ctdb_db, key);
 325                 if (ret != 0) {
 326                         DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 327                 }
 328                 return;
 329         }
 330
 331         if (hdr->reqid != state->reqid) {
 332                 /* we found a record  but it was the wrong one */
 333                 DEBUG(DEBUG_ERR, ("Dropped orphan in ctdb_become_dmaster with reqid:%u\n from node %u", hdr->reqid, hdr->srcnode));
 334
 335                 ret = ctdb_ltdb_unlock(ctdb_db, key);
 336                 if (ret != 0) {
 337                         DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 338                 }
 339                 return;
 340         }
 341
 342         ctdb_call_local(ctdb_db, state->call, &header, state, &data);
 343
 344         ret = ctdb_ltdb_unlock(ctdb_db, state->call->key);
 345         if (ret != 0) {
 346                 DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 347         }
 348
 349         state->state = CTDB_CALL_DONE;
 350         if (state->async.fn) {
 351                 state->async.fn(state);
 352         }
 353 }
 354
 355
 356
 357 /*
 358   called when a CTDB_REQ_DMASTER packet comes in
 359
 360   this comes into the lmaster for a record when the current dmaster
 361   wants to give up the dmaster role and give it to someone else
 362 */
 363 void ctdb_request_dmaster(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 364 {
 365         struct ctdb_req_dmaster *c = (struct ctdb_req_dmaster *)hdr;
 366         TDB_DATA key, data, data2;
 367         struct ctdb_ltdb_header header;
 368         struct ctdb_db_context *ctdb_db;
 369         uint32_t record_flags = 0;
 370         size_t len;
 371         int ret;
 372
 373         key.dptr = c->data;
 374         key.dsize = c->keylen;
 375         data.dptr = c->data + c->keylen;
 376         data.dsize = c->datalen;
 377         len = offsetof(struct ctdb_req_dmaster, data) + key.dsize + data.dsize
 378                         + sizeof(uint32_t);
 379         if (len <= c->hdr.length) {
 380                 record_flags = *(uint32_t *)&c->data[c->keylen + c->datalen];
 381         }
 382
 383         ctdb_db = find_ctdb_db(ctdb, c->db_id);
 384         if (!ctdb_db) {
 385                 ctdb_send_error(ctdb, hdr, -1,
 386                                 "Unknown database in request. db_id==0x%08x",
 387                                 c->db_id);
 388                 return;
 389         }
 390
 391         /* fetch the current record */
 392         ret = ctdb_ltdb_lock_fetch_requeue(ctdb_db, key, &header, hdr, &data2,
 393                                            ctdb_call_input_pkt, ctdb, False);
 394         if (ret == -1) {
 395                 ctdb_fatal(ctdb, "ctdb_req_dmaster failed to fetch record");
 396                 return;
 397         }
 398         if (ret == -2) {
 399                 DEBUG(DEBUG_INFO,(__location__ " deferring ctdb_request_dmaster\n"));
 400                 return;
 401         }
 402
 403         if (ctdb_lmaster(ctdb, &key) != ctdb->pnn) {
 404                 DEBUG(DEBUG_ALERT,("pnn %u dmaster request to non-lmaster lmaster=%u gen=%u curgen=%u\n",
 405                          ctdb->pnn, ctdb_lmaster(ctdb, &key),
 406                          hdr->generation, ctdb->vnn_map->generation));
 407                 ctdb_fatal(ctdb, "ctdb_req_dmaster to non-lmaster");
 408         }
 409
 410         DEBUG(DEBUG_DEBUG,("pnn %u dmaster request on %08x for %u from %u\n",
 411                  ctdb->pnn, ctdb_hash(&key), c->dmaster, c->hdr.srcnode));
 412
 413         /* its a protocol error if the sending node is not the current dmaster */
 414         if (header.dmaster != hdr->srcnode) {
 415                 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",
 416                          ctdb->pnn, c->dmaster, hdr->srcnode, header.dmaster, ctdb_hash(&key),
 417                          ctdb_db->db_id, hdr->generation, ctdb->vnn_map->generation,
 418                          (unsigned long long)c->rsn, (unsigned long long)header.rsn, c->hdr.reqid,
 419                          (key.dsize >= 4)?(*(uint32_t *)key.dptr):0));
 420                 if (header.rsn != 0 || header.dmaster != ctdb->pnn) {
 421                         DEBUG(DEBUG_ERR,("ctdb_req_dmaster from non-master. Force a recovery.\n"));
 422
 423                         ctdb->recovery_mode = CTDB_RECOVERY_ACTIVE;
 424                         ctdb_ltdb_unlock(ctdb_db, key);
 425                         return;
 426                 }
 427         }
 428
 429         if (header.rsn > c->rsn) {
 430                 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",
 431                          ctdb->pnn, c->dmaster, hdr->srcnode, header.dmaster, ctdb_hash(&key),
 432                          ctdb_db->db_id, hdr->generation, ctdb->vnn_map->generation,
 433                          (unsigned long long)c->rsn, (unsigned long long)header.rsn, c->hdr.reqid));
 434         }
 435
 436         /* use the rsn from the sending node */
 437         header.rsn = c->rsn;
 438
 439         /* store the record flags from the sending node */
 440         header.flags = record_flags;
 441
 442         /* check if the new dmaster is the lmaster, in which case we
 443            skip the dmaster reply */
 444         if (c->dmaster == ctdb->pnn) {
 445                 ctdb_become_dmaster(ctdb_db, hdr, key, data, c->rsn, record_flags);
 446         } else {
 447                 ctdb_send_dmaster_reply(ctdb_db, &header, key, data, c->dmaster, hdr->reqid);
 448
 449                 ret = ctdb_ltdb_unlock(ctdb_db, key);
 450                 if (ret != 0) {
 451                         DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 452                 }
 453         }
 454 }
 455
 456
 457 /*
 458   called when a CTDB_REQ_CALL packet comes in
 459 */
 460 void ctdb_request_call(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 461 {
 462         struct ctdb_req_call *c = (struct ctdb_req_call *)hdr;
 463         TDB_DATA data;
 464         struct ctdb_reply_call *r;
 465         int ret, len;
 466         struct ctdb_ltdb_header header;
 467         struct ctdb_call *call;
 468         struct ctdb_db_context *ctdb_db;
 469
 470         if (ctdb->methods == NULL) {
 471                 DEBUG(DEBUG_INFO,(__location__ " Failed ctdb_request_call. Transport is DOWN\n"));
 472                 return;
 473         }
 474
 475
 476         ctdb_db = find_ctdb_db(ctdb, c->db_id);
 477         if (!ctdb_db) {
 478                 ctdb_send_error(ctdb, hdr, -1,
 479                                 "Unknown database in request. db_id==0x%08x",
 480                                 c->db_id);
 481                 return;
 482         }
 483
 484         call = talloc(hdr, struct ctdb_call);
 485         CTDB_NO_MEMORY_FATAL(ctdb, call);
 486
 487         call->call_id  = c->callid;
 488         call->key.dptr = c->data;
 489         call->key.dsize = c->keylen;
 490         call->call_data.dptr = c->data + c->keylen;
 491         call->call_data.dsize = c->calldatalen;
 492
 493         /* determine if we are the dmaster for this key. This also
 494            fetches the record data (if any), thus avoiding a 2nd fetch of the data
 495            if the call will be answered locally */
 496
 497         ret = ctdb_ltdb_lock_fetch_requeue(ctdb_db, call->key, &header, hdr, &data,
 498                                            ctdb_call_input_pkt, ctdb, False);
 499         if (ret == -1) {
 500                 ctdb_send_error(ctdb, hdr, ret, "ltdb fetch failed in ctdb_request_call");
 501                 return;
 502         }
 503         if (ret == -2) {
 504                 DEBUG(DEBUG_INFO,(__location__ " deferred ctdb_request_call\n"));
 505                 return;
 506         }
 507
 508         /* if we are not the dmaster, then send a redirect to the
 509            requesting node */
 510         if (header.dmaster != ctdb->pnn) {
 511                 talloc_free(data.dptr);
 512                 ctdb_call_send_redirect(ctdb, call->key, c, &header);
 513
 514                 ret = ctdb_ltdb_unlock(ctdb_db, call->key);
 515                 if (ret != 0) {
 516                         DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 517                 }
 518                 return;
 519         }
 520
 521         CTDB_UPDATE_STAT(ctdb, max_hop_count, c->hopcount);
 522
 523         /* Try if possible to migrate the record off to the caller node.
 524          * From the clients perspective a fetch of the data is just as
 525          * expensive as a migration.
 526          */
 527         if (c->hdr.srcnode != ctdb->pnn) {
 528                 if (ctdb_db->transaction_active) {
 529                         DEBUG(DEBUG_INFO, (__location__ " refusing migration"
 530                               " of key %s while transaction is active\n",
 531                               (char *)call->key.dptr));
 532                 } else {
 533                         DEBUG(DEBUG_DEBUG,("pnn %u starting migration of %08x to %u\n",
 534                                  ctdb->pnn, ctdb_hash(&(call->key)), c->hdr.srcnode));
 535                         ctdb_call_send_dmaster(ctdb_db, c, &header, &(call->key), &data);
 536                         talloc_free(data.dptr);
 537
 538                         ret = ctdb_ltdb_unlock(ctdb_db, call->key);
 539                         if (ret != 0) {
 540                                 DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 541                         }
 542                         return;
 543                 }
 544         }
 545
 546         ctdb_call_local(ctdb_db, call, &header, hdr, &data);
 547
 548         ret = ctdb_ltdb_unlock(ctdb_db, call->key);
 549         if (ret != 0) {
 550                 DEBUG(DEBUG_ERR,(__location__ " ctdb_ltdb_unlock() failed with error %d\n", ret));
 551         }
 552
 553         len = offsetof(struct ctdb_reply_call, data) + call->reply_data.dsize;
 554         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REPLY_CALL, len,
 555                                     struct ctdb_reply_call);
 556         CTDB_NO_MEMORY_FATAL(ctdb, r);
 557         r->hdr.destnode  = hdr->srcnode;
 558         r->hdr.reqid     = hdr->reqid;
 559         r->status        = call->status;
 560         r->datalen       = call->reply_data.dsize;
 561         if (call->reply_data.dsize) {
 562                 memcpy(&r->data[0], call->reply_data.dptr, call->reply_data.dsize);
 563         }
 564
 565         ctdb_queue_packet(ctdb, &r->hdr);
 566
 567         talloc_free(r);
 568 }
 569
 570 /*
 571   called when a CTDB_REPLY_CALL packet comes in
 572
 573   This packet comes in response to a CTDB_REQ_CALL request packet. It
 574   contains any reply data from the call
 575 */
 576 void ctdb_reply_call(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 577 {
 578         struct ctdb_reply_call *c = (struct ctdb_reply_call *)hdr;
 579         struct ctdb_call_state *state;
 580
 581         state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_call_state);
 582         if (state == NULL) {
 583                 DEBUG(DEBUG_ERR, (__location__ " reqid %u not found\n", hdr->reqid));
 584                 return;
 585         }
 586
 587         if (hdr->reqid != state->reqid) {
 588                 /* we found a record  but it was the wrong one */
 589                 DEBUG(DEBUG_ERR, ("Dropped orphaned call reply with reqid:%u\n",hdr->reqid));
 590                 return;
 591         }
 592
 593         state->call->reply_data.dptr = c->data;
 594         state->call->reply_data.dsize = c->datalen;
 595         state->call->status = c->status;
 596
 597         talloc_steal(state, c);
 598
 599         state->state = CTDB_CALL_DONE;
 600         if (state->async.fn) {
 601                 state->async.fn(state);
 602         }
 603 }
 604
 605
 606 /*
 607   called when a CTDB_REPLY_DMASTER packet comes in
 608
 609   This packet comes in from the lmaster response to a CTDB_REQ_CALL
 610   request packet. It means that the current dmaster wants to give us
 611   the dmaster role
 612 */
 613 void ctdb_reply_dmaster(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 614 {
 615         struct ctdb_reply_dmaster *c = (struct ctdb_reply_dmaster *)hdr;
 616         struct ctdb_db_context *ctdb_db;
 617         TDB_DATA key, data;
 618         uint32_t record_flags = 0;
 619         size_t len;
 620         int ret;
 621
 622         ctdb_db = find_ctdb_db(ctdb, c->db_id);
 623         if (ctdb_db == NULL) {
 624                 DEBUG(DEBUG_ERR,("Unknown db_id 0x%x in ctdb_reply_dmaster\n", c->db_id));
 625                 return;
 626         }
 627
 628         key.dptr = c->data;
 629         key.dsize = c->keylen;
 630         data.dptr = &c->data[key.dsize];
 631         data.dsize = c->datalen;
 632         len = offsetof(struct ctdb_reply_dmaster, data) + key.dsize + data.dsize
 633                 + sizeof(uint32_t);
 634         if (len <= c->hdr.length) {
 635                 record_flags = *(uint32_t *)&c->data[c->keylen + c->datalen];
 636         }
 637
 638         ret = ctdb_ltdb_lock_requeue(ctdb_db, key, hdr,
 639                                      ctdb_call_input_pkt, ctdb, False);
 640         if (ret == -2) {
 641                 return;
 642         }
 643         if (ret != 0) {
 644                 DEBUG(DEBUG_ERR,(__location__ " Failed to get lock in ctdb_reply_dmaster\n"));
 645                 return;
 646         }
 647
 648         ctdb_become_dmaster(ctdb_db, hdr, key, data, c->rsn, record_flags);
 649 }
 650
 651
 652 /*
 653   called when a CTDB_REPLY_ERROR packet comes in
 654 */
 655 void ctdb_reply_error(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 656 {
 657         struct ctdb_reply_error *c = (struct ctdb_reply_error *)hdr;
 658         struct ctdb_call_state *state;
 659
 660         state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_call_state);
 661         if (state == NULL) {
 662                 DEBUG(DEBUG_ERR,("pnn %u Invalid reqid %u in ctdb_reply_error\n",
 663                          ctdb->pnn, hdr->reqid));
 664                 return;
 665         }
 666
 667         if (hdr->reqid != state->reqid) {
 668                 /* we found a record  but it was the wrong one */
 669                 DEBUG(DEBUG_ERR, ("Dropped orphaned error reply with reqid:%u\n",hdr->reqid));
 670                 return;
 671         }
 672
 673         talloc_steal(state, c);
 674
 675         state->state  = CTDB_CALL_ERROR;
 676         state->errmsg = (char *)c->msg;
 677         if (state->async.fn) {
 678                 state->async.fn(state);
 679         }
 680 }
 681
 682
 683 /*
 684   destroy a ctdb_call
 685 */
 686 static int ctdb_call_destructor(struct ctdb_call_state *state)
 687 {
 688         DLIST_REMOVE(state->ctdb_db->ctdb->pending_calls, state);
 689         ctdb_reqid_remove(state->ctdb_db->ctdb, state->reqid);
 690         return 0;
 691 }
 692
 693
 694 /*
 695   called when a ctdb_call needs to be resent after a reconfigure event
 696 */
 697 static void ctdb_call_resend(struct ctdb_call_state *state)
 698 {
 699         struct ctdb_context *ctdb = state->ctdb_db->ctdb;
 700
 701         state->generation = ctdb->vnn_map->generation;
 702
 703         /* use a new reqid, in case the old reply does eventually come in */
 704         ctdb_reqid_remove(ctdb, state->reqid);
 705         state->reqid = ctdb_reqid_new(ctdb, state);
 706         state->c->hdr.reqid = state->reqid;
 707
 708         /* update the generation count for this request, so its valid with the new vnn_map */
 709         state->c->hdr.generation = state->generation;
 710
 711         /* send the packet to ourselves, it will be redirected appropriately */
 712         state->c->hdr.destnode = ctdb->pnn;
 713
 714         ctdb_queue_packet(ctdb, &state->c->hdr);
 715         DEBUG(DEBUG_NOTICE,("resent ctdb_call\n"));
 716 }
 717
 718 /*
 719   resend all pending calls on recovery
 720  */
 721 void ctdb_call_resend_all(struct ctdb_context *ctdb)
 722 {
 723         struct ctdb_call_state *state, *next;
 724         for (state=ctdb->pending_calls;state;state=next) {
 725                 next = state->next;
 726                 ctdb_call_resend(state);
 727         }
 728 }
 729
 730 /*
 731   this allows the caller to setup a async.fn
 732 */
 733 static void call_local_trigger(struct event_context *ev, struct timed_event *te,
 734                        struct timeval t, void *private_data)
 735 {
 736         struct ctdb_call_state *state = talloc_get_type(private_data, struct ctdb_call_state);
 737         if (state->async.fn) {
 738                 state->async.fn(state);
 739         }
 740 }
 741
 742
 743 /*
 744   construct an event driven local ctdb_call
 745
 746   this is used so that locally processed ctdb_call requests are processed
 747   in an event driven manner
 748 */
 749 struct ctdb_call_state *ctdb_call_local_send(struct ctdb_db_context *ctdb_db,
 750                                              struct ctdb_call *call,
 751                                              struct ctdb_ltdb_header *header,
 752                                              TDB_DATA *data)
 753 {
 754         struct ctdb_call_state *state;
 755         struct ctdb_context *ctdb = ctdb_db->ctdb;
 756         int ret;
 757
 758         state = talloc_zero(ctdb_db, struct ctdb_call_state);
 759         CTDB_NO_MEMORY_NULL(ctdb, state);
 760
 761         talloc_steal(state, data->dptr);
 762
 763         state->state = CTDB_CALL_DONE;
 764         state->call  = talloc(state, struct ctdb_call);
 765         CTDB_NO_MEMORY_NULL(ctdb, state->call);
 766         *(state->call) = *call;
 767         state->ctdb_db = ctdb_db;
 768
 769         ret = ctdb_call_local(ctdb_db, state->call, header, state, data);
 770
 771         event_add_timed(ctdb->ev, state, timeval_zero(), call_local_trigger, state);
 772
 773         return state;
 774 }
 775
 776
 777 /*
 778   make a remote ctdb call - async send. Called in daemon context.
 779
 780   This constructs a ctdb_call request and queues it for processing.
 781   This call never blocks.
 782 */
 783 struct ctdb_call_state *ctdb_daemon_call_send_remote(struct ctdb_db_context *ctdb_db,
 784                                                      struct ctdb_call *call,
 785                                                      struct ctdb_ltdb_header *header)
 786 {
 787         uint32_t len;
 788         struct ctdb_call_state *state;
 789         struct ctdb_context *ctdb = ctdb_db->ctdb;
 790
 791         if (ctdb->methods == NULL) {
 792                 DEBUG(DEBUG_INFO,(__location__ " Failed send packet. Transport is down\n"));
 793                 return NULL;
 794         }
 795
 796         state = talloc_zero(ctdb_db, struct ctdb_call_state);
 797         CTDB_NO_MEMORY_NULL(ctdb, state);
 798         state->call = talloc(state, struct ctdb_call);
 799         CTDB_NO_MEMORY_NULL(ctdb, state->call);
 800
 801         state->reqid = ctdb_reqid_new(ctdb, state);
 802         state->ctdb_db = ctdb_db;
 803         talloc_set_destructor(state, ctdb_call_destructor);
 804
 805         len = offsetof(struct ctdb_req_call, data) + call->key.dsize + call->call_data.dsize;
 806         state->c = ctdb_transport_allocate(ctdb, state, CTDB_REQ_CALL, len,
 807                                            struct ctdb_req_call);
 808         CTDB_NO_MEMORY_NULL(ctdb, state->c);
 809         state->c->hdr.destnode  = header->dmaster;
 810
 811         /* this limits us to 16k outstanding messages - not unreasonable */
 812         state->c->hdr.reqid     = state->reqid;
 813         state->c->flags         = call->flags;
 814         state->c->db_id         = ctdb_db->db_id;
 815         state->c->callid        = call->call_id;
 816         state->c->hopcount      = 0;
 817         state->c->keylen        = call->key.dsize;
 818         state->c->calldatalen   = call->call_data.dsize;
 819         memcpy(&state->c->data[0], call->key.dptr, call->key.dsize);
 820         memcpy(&state->c->data[call->key.dsize],
 821                call->call_data.dptr, call->call_data.dsize);
 822         *(state->call)              = *call;
 823         state->call->call_data.dptr = &state->c->data[call->key.dsize];
 824         state->call->key.dptr       = &state->c->data[0];
 825
 826         state->state  = CTDB_CALL_WAIT;
 827         state->generation = ctdb->vnn_map->generation;
 828
 829         DLIST_ADD(ctdb->pending_calls, state);
 830
 831         ctdb_queue_packet(ctdb, &state->c->hdr);
 832
 833         return state;
 834 }
 835
 836 /*
 837   make a remote ctdb call - async recv - called in daemon context
 838
 839   This is called when the program wants to wait for a ctdb_call to complete and get the
 840   results. This call will block unless the call has already completed.
 841 */
 842 int ctdb_daemon_call_recv(struct ctdb_call_state *state, struct ctdb_call *call)
 843 {
 844         while (state->state < CTDB_CALL_DONE) {
 845                 event_loop_once(state->ctdb_db->ctdb->ev);
 846         }
 847         if (state->state != CTDB_CALL_DONE) {
 848                 ctdb_set_error(state->ctdb_db->ctdb, "%s", state->errmsg);
 849                 talloc_free(state);
 850                 return -1;
 851         }
 852
 853         if (state->call->reply_data.dsize) {
 854                 call->reply_data.dptr = talloc_memdup(call,
 855                                                       state->call->reply_data.dptr,
 856                                                       state->call->reply_data.dsize);
 857                 call->reply_data.dsize = state->call->reply_data.dsize;
 858         } else {
 859                 call->reply_data.dptr = NULL;
 860                 call->reply_data.dsize = 0;
 861         }
 862         call->status = state->call->status;
 863         talloc_free(state);
 864         return 0;
 865 }
 866
 867
 868 /*
 869    send a keepalive packet to the other node
 870 */
 871 void ctdb_send_keepalive(struct ctdb_context *ctdb, uint32_t destnode)
 872 {
 873         struct ctdb_req_keepalive *r;
 874
 875         if (ctdb->methods == NULL) {
 876                 DEBUG(DEBUG_INFO,(__location__ " Failed to send keepalive. Transport is DOWN\n"));
 877                 return;
 878         }
 879
 880         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REQ_KEEPALIVE,
 881                                     sizeof(struct ctdb_req_keepalive),
 882                                     struct ctdb_req_keepalive);
 883         CTDB_NO_MEMORY_FATAL(ctdb, r);
 884         r->hdr.destnode  = destnode;
 885         r->hdr.reqid     = 0;
 886
 887         CTDB_INCREMENT_STAT(ctdb, keepalive_packets_sent);
 888
 889         ctdb_queue_packet(ctdb, &r->hdr);
 890
 891         talloc_free(r);
 892 }