ctdb-daemon: Rename struct ctdb_ban_time to ctdb_ban_state
[obnox/samba/samba-obnox.git] / ctdb / server / ctdb_control.c
index ee69270c784899a1fc6d52e5f7212a832551b60b..1ba1ef796eefee21c25a32595dd915b9368baabb 100644 (file)
    You should have received a copy of the GNU General Public License
    along with this program; if not, see <http://www.gnu.org/licenses/>.
 */
-#include "includes.h"
-#include "tdb.h"
+#include "replace.h"
 #include "system/network.h"
 #include "system/filesys.h"
 #include "system/wait.h"
-#include "../include/ctdb_private.h"
-#include "lib/util/dlinklist.h"
+
+#include <talloc.h>
+#include <tevent.h>
+
 #include "lib/tdb_wrap/tdb_wrap.h"
+#include "lib/util/dlinklist.h"
+#include "lib/util/debug.h"
+#include "lib/util/samba_util.h"
 #include "lib/util/talloc_report.h"
 
+#include "ctdb_private.h"
+#include "ctdb_client.h"
+#include "ctdb_logging.h"
+
+#include "common/reqid.h"
+#include "common/common.h"
+
 
 struct ctdb_control_state {
        struct ctdb_context *ctdb;
@@ -80,7 +91,7 @@ static int32_t control_not_implemented(const char *unsupported,
   process a control request
  */
 static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb, 
-                                    struct ctdb_req_control *c,
+                                    struct ctdb_req_control_old *c,
                                     TDB_DATA indata,
                                     TDB_DATA *outdata, uint32_t srcnode,
                                     const char **errormsg,
@@ -110,16 +121,10 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
        }
 
        case CTDB_CONTROL_STATISTICS: {
-               int i;
                CHECK_CONTROL_DATA_SIZE(0);
                ctdb->statistics.memory_used = talloc_total_size(NULL);
                ctdb->statistics.num_clients = ctdb->num_clients;
-               ctdb->statistics.frozen = 0;
-               for (i=1; i<= NUM_DB_PRIORITIES; i++) {
-                       if (ctdb->freeze_mode[i] == CTDB_FREEZE_FROZEN) {
-                               ctdb->statistics.frozen = 1;
-                       }
-               }
+               ctdb->statistics.frozen = (ctdb_db_all_frozen(ctdb) ? 1 : 0);
                ctdb->statistics.recovering = (ctdb->recovery_mode == CTDB_RECOVERY_ACTIVE);
                ctdb->statistics.statistics_current_time = timeval_current();
 
@@ -370,7 +375,7 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
                return ctdb_control_get_public_ips(ctdb, c, outdata);
 
        case CTDB_CONTROL_TCP_CLIENT:
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_tcp_addr));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_connection));
                return ctdb_control_tcp_client(ctdb, client_id, indata);
 
        case CTDB_CONTROL_STARTUP: 
@@ -378,15 +383,15 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
                return ctdb_control_startup(ctdb, srcnode);
 
        case CTDB_CONTROL_TCP_ADD: 
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_tcp_connection));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_connection));
                return ctdb_control_tcp_add(ctdb, indata, false);
 
        case CTDB_CONTROL_TCP_ADD_DELAYED_UPDATE: 
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_tcp_connection));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_connection));
                return ctdb_control_tcp_add(ctdb, indata, true);
 
        case CTDB_CONTROL_TCP_REMOVE: 
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_tcp_connection));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_connection));
                return ctdb_control_tcp_remove(ctdb, indata);
 
        case CTDB_CONTROL_SET_TUNABLE:
@@ -403,7 +408,7 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
                return ctdb_control_modflags(ctdb, indata);
 
        case CTDB_CONTROL_KILL_TCP: 
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_killtcp));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_connection));
                return ctdb_control_kill_tcp(ctdb, indata);
 
        case CTDB_CONTROL_GET_TCP_TICKLE_LIST:
@@ -415,15 +420,15 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
                return ctdb_control_set_tcp_tickle_list(ctdb, indata);
 
        case CTDB_CONTROL_REGISTER_SERVER_ID: 
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_server_id));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_client_id));
                return ctdb_control_register_server_id(ctdb, client_id, indata);
 
        case CTDB_CONTROL_UNREGISTER_SERVER_ID: 
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_server_id));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_client_id));
                return ctdb_control_unregister_server_id(ctdb, indata);
 
        case CTDB_CONTROL_CHECK_SERVER_ID: 
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_server_id));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_client_id));
                return ctdb_control_check_server_id(ctdb, indata);
 
        case CTDB_CONTROL_GET_SERVER_ID_LIST:
@@ -448,7 +453,7 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
                return ctdb_control_transaction_commit(ctdb, *(uint32_t *)indata.dptr);
 
        case CTDB_CONTROL_WIPE_DATABASE:
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_wipe_database));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_transdb));
                return ctdb_control_wipe_database(ctdb, indata);
 
        case CTDB_CONTROL_UPTIME:
@@ -599,7 +604,7 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
                return ctdb_control_disable_script(ctdb, indata);
 
        case CTDB_CONTROL_SET_BAN_STATE:
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_ban_time));
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_ban_state));
                return ctdb_control_set_ban_state(ctdb, indata);
 
        case CTDB_CONTROL_GET_BAN_STATE:
@@ -629,7 +634,7 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
                return ctdb_control_register_notify(ctdb, client_id, indata);
 
        case CTDB_CONTROL_DEREGISTER_NOTIFY:
-               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_client_notify_deregister));
+               CHECK_CONTROL_DATA_SIZE(sizeof(uint64_t));
                return ctdb_control_deregister_notify(ctdb, client_id, indata);
 
        case CTDB_CONTROL_GET_LOG:
@@ -689,6 +694,27 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
        case CTDB_CONTROL_DB_DETACH:
                return ctdb_control_db_detach(ctdb, indata, client_id);
 
+       case CTDB_CONTROL_DB_FREEZE:
+               CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
+               return ctdb_control_db_freeze(ctdb, c, *(uint32_t *)indata.dptr,
+                                             async_reply);
+
+       case CTDB_CONTROL_DB_THAW:
+               CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
+               return ctdb_control_db_thaw(ctdb, *(uint32_t *)indata.dptr);
+
+       case CTDB_CONTROL_DB_TRANSACTION_START:
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_transdb));
+               return ctdb_control_db_transaction_start(ctdb, indata);
+
+       case CTDB_CONTROL_DB_TRANSACTION_COMMIT:
+               CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_transdb));
+               return ctdb_control_db_transaction_commit(ctdb, indata);
+
+       case CTDB_CONTROL_DB_TRANSACTION_CANCEL:
+               CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
+               return ctdb_control_db_transaction_cancel(ctdb, indata);
+
        default:
                DEBUG(DEBUG_CRIT,(__location__ " Unknown CTDB control opcode %u\n", opcode));
                return -1;
@@ -698,10 +724,10 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
 /*
   send a reply for a ctdb control
  */
-void ctdb_request_control_reply(struct ctdb_context *ctdb, struct ctdb_req_control *c,
+void ctdb_request_control_reply(struct ctdb_context *ctdb, struct ctdb_req_control_old *c,
                                TDB_DATA *outdata, int32_t status, const char *errormsg)
 {
-       struct ctdb_reply_control *r;
+       struct ctdb_reply_control_old *r;
        size_t len;
        
        /* some controls send no reply */
@@ -709,11 +735,11 @@ void ctdb_request_control_reply(struct ctdb_context *ctdb, struct ctdb_req_contr
                return;
        }
 
-       len = offsetof(struct ctdb_reply_control, data) + (outdata?outdata->dsize:0);
+       len = offsetof(struct ctdb_reply_control_old, data) + (outdata?outdata->dsize:0);
        if (errormsg) {
                len += strlen(errormsg);
        }
-       r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REPLY_CONTROL, len, struct ctdb_reply_control);
+       r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REPLY_CONTROL, len, struct ctdb_reply_control_old);
        if (r == NULL) {
                DEBUG(DEBUG_ERR,(__location__ "Unable to allocate transport - OOM or transport is down\n"));
                return;
@@ -741,7 +767,7 @@ void ctdb_request_control_reply(struct ctdb_context *ctdb, struct ctdb_req_contr
 */
 void ctdb_request_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 {
-       struct ctdb_req_control *c = (struct ctdb_req_control *)hdr;
+       struct ctdb_req_control_old *c = (struct ctdb_req_control_old *)hdr;
        TDB_DATA data, *outdata;
        int32_t status;
        bool async_reply = false;
@@ -765,12 +791,12 @@ void ctdb_request_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr
 */
 void ctdb_reply_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 {
-       struct ctdb_reply_control *c = (struct ctdb_reply_control *)hdr;
+       struct ctdb_reply_control_old *c = (struct ctdb_reply_control_old *)hdr;
        TDB_DATA data;
        struct ctdb_control_state *state;
        const char *errormsg = NULL;
 
-       state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_control_state);
+       state = reqid_find(ctdb->idr, hdr->reqid, struct ctdb_control_state);
        if (state == NULL) {
                DEBUG(DEBUG_ERR,("pnn %u Invalid reqid %u in ctdb_reply_control\n",
                         ctdb->pnn, hdr->reqid));
@@ -799,15 +825,16 @@ void ctdb_reply_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 
 static int ctdb_control_destructor(struct ctdb_control_state *state)
 {
-       ctdb_reqid_remove(state->ctdb, state->reqid);
+       reqid_remove(state->ctdb->idr, state->reqid);
        return 0;
 }
 
 /*
   handle a timeout of a control
  */
-static void ctdb_control_timeout(struct event_context *ev, struct timed_event *te, 
-                      struct timeval t, void *private_data)
+static void ctdb_control_timeout(struct tevent_context *ev,
+                                struct tevent_timer *te,
+                                struct timeval t, void *private_data)
 {
        struct ctdb_control_state *state = talloc_get_type(private_data, struct ctdb_control_state);
        TALLOC_CTX *tmp_ctx = talloc_new(ev);
@@ -833,7 +860,7 @@ int ctdb_daemon_send_control(struct ctdb_context *ctdb, uint32_t destnode,
                             ctdb_control_callback_fn_t callback,
                             void *private_data)
 {
-       struct ctdb_req_control *c;
+       struct ctdb_req_control_old *c;
        struct ctdb_control_state *state;
        size_t len;
 
@@ -866,7 +893,7 @@ int ctdb_daemon_send_control(struct ctdb_context *ctdb, uint32_t destnode,
        state = talloc(private_data?private_data:ctdb, struct ctdb_control_state);
        CTDB_NO_MEMORY(ctdb, state);
 
-       state->reqid = ctdb_reqid_new(ctdb, state);
+       state->reqid = reqid_new(ctdb->idr, state);
        state->callback = callback;
        state->private_data = private_data;
        state->ctdb = ctdb;
@@ -874,9 +901,9 @@ int ctdb_daemon_send_control(struct ctdb_context *ctdb, uint32_t destnode,
 
        talloc_set_destructor(state, ctdb_control_destructor);
 
-       len = offsetof(struct ctdb_req_control, data) + data.dsize;
+       len = offsetof(struct ctdb_req_control_old, data) + data.dsize;
        c = ctdb_transport_allocate(ctdb, state, CTDB_REQ_CONTROL, len, 
-                                   struct ctdb_req_control);
+                                   struct ctdb_req_control_old);
        CTDB_NO_MEMORY(ctdb, c);
        talloc_set_name_const(c, "ctdb_req_control packet");
 
@@ -899,9 +926,9 @@ int ctdb_daemon_send_control(struct ctdb_context *ctdb, uint32_t destnode,
        }
 
        if (ctdb->tunable.control_timeout) {
-               event_add_timed(ctdb->ev, state, 
-                               timeval_current_ofs(ctdb->tunable.control_timeout, 0), 
-                               ctdb_control_timeout, state);
+               tevent_add_timer(ctdb->ev, state,
+                                timeval_current_ofs(ctdb->tunable.control_timeout, 0),
+                                ctdb_control_timeout, state);
        }
 
        talloc_free(c);