sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-01-10 01:18:15 +03:00
Code

ctdb-tests: Implement transaction control in fake_ctdbd

Browse Source 
Signed-off-by: Amitay Isaacs <amitay@gmail.com>
Reviewed-by: Martin Schwenke <martin@meltin.net>
This commit is contained in:
Amitay Isaacs 2018-02-05 14:03:27 +11:00 committed by Amitay Isaacs
parent 7c30d4791d
commit 8250956c92
1 changed files with 109 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

109
ctdb/tests/src/fake_ctdbd.c
View File

@ -969,6 +969,68 @@ static int database_seqnum(struct database *db, uint64_t *seqnum)
return ret;
}
static int ltdb_transaction_update(uint32_t reqid,
struct ctdb_ltdb_header *no_header,
TDB_DATA key, TDB_DATA data,
void *private_data)
{
struct database *db = (struct database *)private_data;
TALLOC_CTX *tmp_ctx = talloc_new(db);
struct ctdb_ltdb_header header = { 0 }, oldheader;
TDB_DATA olddata;
int ret;
if (db->tdb == NULL) {
return EINVAL;
}
ret = ctdb_ltdb_header_extract(&data, &header);
if (ret != 0) {
return ret;
}
ret = ltdb_fetch(db, key, &oldheader, tmp_ctx, &olddata);
if (ret != 0) {
return ret;
}
if (olddata.dsize > 0) {
if (oldheader.rsn > header.rsn ||
(oldheader.rsn == header.rsn &&
olddata.dsize != data.dsize)) {
return -1;
}
}
talloc_free(tmp_ctx);
ret = ltdb_store(db, key, &header, data);
return ret;
}
static int ltdb_transaction(struct database *db,
struct ctdb_rec_buffer *recbuf)
{
int ret;
if (db->tdb == NULL) {
return EINVAL;
}
ret = tdb_transaction_start(db->tdb);
if (ret == -1) {
return ret;
}
ret = ctdb_rec_buffer_traverse(recbuf, ltdb_transaction_update, db);
if (ret != 0) {
tdb_transaction_cancel(db->tdb);
}
ret = tdb_transaction_commit(db->tdb);
return ret;
}
static bool public_ips_parse(struct ctdbd_context *ctdb,
uint32_t numnodes)
{
@ -2780,6 +2842,49 @@ fail:
reply.errmsg = "Failed to ban node";
}
static void control_trans3_commit(TALLOC_CTX *mem_ctx,
struct tevent_req *req,
struct ctdb_req_header *header,
struct ctdb_req_control *request)
{
struct client_state *state = tevent_req_data(
req, struct client_state);
struct ctdbd_context *ctdb = state->ctdb;
struct ctdb_reply_control reply;
struct database *db;
int ret;
reply.rdata.opcode = request->opcode;
db = database_find(ctdb->db_map, request->rdata.data.recbuf->db_id);
if (db == NULL) {
reply.status = -1;
reply.errmsg = "Unknown database";
client_send_control(req, header, &reply);
return;
}
if (! (db->flags &
(CTDB_DB_FLAGS_PERSISTENT|CTDB_DB_FLAGS_REPLICATED))) {
reply.status = -1;
reply.errmsg = "Transactions on volatile database";
client_send_control(req, header, &reply);
return;
}
ret = ltdb_transaction(db, request->rdata.data.recbuf);
if (ret != 0) {
reply.status = -1;
reply.errmsg = "Transaction failed";
client_send_control(req, header, &reply);
return;
}
reply.status = 0;
reply.errmsg = NULL;
client_send_control(req, header, &reply);
}
static void control_get_db_seqnum(TALLOC_CTX *mem_ctx,
struct tevent_req *req,
struct ctdb_req_header *header,
@ -3923,6 +4028,10 @@ static void client_process_control(struct tevent_req *req,
control_set_ban_state(mem_ctx, req, &header, &request);
break;
case CTDB_CONTROL_TRANS3_COMMIT:
control_trans3_commit(mem_ctx, req, &header, &request);
break;
case CTDB_CONTROL_GET_DB_SEQNUM:
control_get_db_seqnum(mem_ctx, req, &header, &request);
break;