ctdb-daemon: Rename struct ctdb_server_id to ctdb_client_id
[obnox/samba/samba-obnox.git] / ctdb / server / ctdb_control.c
index fda4c29134a1a9e97050d036a480c78440c4c907..802289f31ced733e3e5debcf25c72c8defe0a8b3 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 {
@@ -40,34 +52,23 @@ struct ctdb_control_state {
  */
 int32_t ctdb_dump_memory(struct ctdb_context *ctdb, TDB_DATA *outdata)
 {
-       /* dump to a file, then send the file as a blob */
-       FILE *f;
-       long fsize;
-       f = tmpfile();
-       if (f == NULL) {
-               DEBUG(DEBUG_ERR,(__location__ " Unable to open tmpfile - %s\n", strerror(errno)));
-               return -1;
-       }
-       talloc_report_full(NULL, f);
-       fsize = ftell(f);
-       if (fsize == -1) {
-               DEBUG(DEBUG_ERR, (__location__ " Unable to get file size - %s\n",
-                                 strerror(errno)));
-               fclose(f);
+       char *report;
+       size_t reportlen;
+
+       report = talloc_report_str(outdata, NULL);
+       if (report == NULL) {
+               DEBUG(DEBUG_ERR,
+                     (__location__ " talloc_report_str failed\n"));
                return -1;
        }
-       rewind(f);
-       outdata->dptr = talloc_size(outdata, fsize);
-       if (outdata->dptr == NULL) {
-               fclose(f);
-               CTDB_NO_MEMORY(ctdb, outdata->dptr);
-       }
-       outdata->dsize = fread(outdata->dptr, 1, fsize, f);
-       fclose(f);
-       if (outdata->dsize != fsize) {
-               DEBUG(DEBUG_ERR,(__location__ " Unable to read tmpfile\n"));
-               return -1;
+       reportlen = talloc_get_size(report);
+
+       if (reportlen > 0) {
+               reportlen -= 1; /* strip trailing zero */
        }
+
+       outdata->dptr = (uint8_t *)report;
+       outdata->dsize = reportlen;
        return 0;
 }
 
@@ -90,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,
@@ -120,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();
 
@@ -151,8 +146,15 @@ static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
        }
 
        case CTDB_CONTROL_STATISTICS_RESET: {
+               struct ctdb_db_context *ctdb_db;
+
                CHECK_CONTROL_DATA_SIZE(0);
                ZERO_STRUCT(ctdb->statistics);
+               for (ctdb_db = ctdb->db_list;
+                    ctdb_db != NULL;
+                    ctdb_db = ctdb_db->next) {
+                       ctdb_db_statistics_reset(ctdb_db);
+               }
                ctdb->statistics.statistics_start_time = timeval_current();
                return 0;
        }
@@ -406,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_tcp_connection));
                return ctdb_control_kill_tcp(ctdb, indata);
 
        case CTDB_CONTROL_GET_TCP_TICKLE_LIST:
@@ -418,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:
@@ -451,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:
@@ -632,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:
@@ -692,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;
@@ -701,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 */
@@ -712,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;
@@ -744,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;
@@ -768,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));
@@ -802,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);
@@ -836,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;
 
@@ -869,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;
@@ -877,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");
 
@@ -902,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);