summaryrefslogtreecommitdiffstats
path: root/src/test/testmsgr.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/testmsgr.cc')
-rw-r--r--src/test/testmsgr.cc144
1 files changed, 144 insertions, 0 deletions
diff --git a/src/test/testmsgr.cc b/src/test/testmsgr.cc
new file mode 100644
index 000000000..ae162a290
--- /dev/null
+++ b/src/test/testmsgr.cc
@@ -0,0 +1,144 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2004-2006 Sage Weil <sage@newdream.net>
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License version 2.1, as published by the Free Software
+ * Foundation. See file COPYING.
+ *
+ */
+
+#include <sys/stat.h>
+#include <iostream>
+#include <string>
+using namespace std;
+
+#include "common/config.h"
+
+#include "mon/MonMap.h"
+#include "mon/MonClient.h"
+#include "msg/Messenger.h"
+#include "messages/MPing.h"
+
+#include "common/Timer.h"
+#include "global/global_init.h"
+#include "common/ceph_argparse.h"
+
+#include <sys/types.h>
+#include <fcntl.h>
+
+#define dout_subsys ceph_subsys_ms
+
+Messenger *messenger = 0;
+
+ceph::mutex test_lock = ceph::make_mutex("mylock");
+ceph::condition_variable cond;
+
+uint64_t received = 0;
+
+class Admin : public Dispatcher {
+public:
+ Admin()
+ : Dispatcher(g_ceph_context)
+ {
+ }
+private:
+ bool ms_dispatch(Message *m) {
+
+ //cerr << "got ping from " << m->get_source() << std::endl;
+ dout(0) << "got ping from " << m->get_source() << dendl;
+ test_lock.lock();
+ ++received;
+ cond.notify_all();
+ test_lock.unlock();
+
+ m->put();
+ return true;
+ }
+
+ bool ms_handle_reset(Connection *con) { return false; }
+ void ms_handle_remote_reset(Connection *con) {}
+ bool ms_handle_refused(Connection *con) { return false; }
+
+} dispatcher;
+
+
+int main(int argc, const char **argv, const char *envp[]) {
+
+ auto args = argv_to_vec(argc, argv);
+
+ auto cct = global_init(NULL, args, CEPH_ENTITY_TYPE_CLIENT,
+ CODE_ENVIRONMENT_UTILITY,
+ CINIT_FLAG_NO_DEFAULT_CONFIG_FILE);
+ common_init_finish(g_ceph_context);
+
+ dout(0) << "i am mon " << args[0] << dendl;
+
+ // get monmap
+ MonClient mc(g_ceph_context);
+ if (mc.build_initial_monmap() < 0)
+ return -1;
+
+ // start up network
+ int whoami = mc.monmap.get_rank(args[0]);
+ ceph_assert(whoami >= 0);
+ ostringstream ss;
+ ss << mc.monmap.get_addr(whoami);
+ std::string sss(ss.str());
+ g_ceph_context->_conf.set_val("public_addr", sss.c_str());
+ g_ceph_context->_conf.apply_changes(nullptr);
+ std::string public_msgr_type = g_conf()->ms_public_type.empty() ? g_conf().get_val<std::string>("ms_type") : g_conf()->ms_public_type;
+ Messenger *rank = Messenger::create(g_ceph_context,
+ public_msgr_type,
+ entity_name_t::MON(whoami), "tester",
+ getpid());
+ int err = rank->bind(g_ceph_context->_conf->public_addr);
+ if (err < 0)
+ return 1;
+
+ // start monitor
+ messenger = rank;
+ messenger->set_default_send_priority(CEPH_MSG_PRIO_HIGH);
+ messenger->add_dispatcher_head(&dispatcher);
+
+ rank->start();
+
+ int isend = 0;
+ if (whoami == 0)
+ isend = 100;
+
+ std::unique_lock l{test_lock};
+ uint64_t sent = 0;
+ while (1) {
+ while (received + isend <= sent) {
+ //cerr << "wait r " << received << " s " << sent << " is " << isend << std::endl;
+ dout(0) << "wait r " << received << " s " << sent << " is " << isend << dendl;
+ cond.wait(l);
+ }
+
+ int t = rand() % mc.get_num_mon();
+ if (t == whoami)
+ continue;
+
+ if (rand() % 10 == 0) {
+ //cerr << "mark_down " << t << std::endl;
+ dout(0) << "mark_down " << t << dendl;
+ messenger->mark_down_addrs(mc.get_mon_addrs(t));
+ }
+ //cerr << "pinging " << t << std::endl;
+ dout(0) << "pinging " << t << dendl;
+ messenger->send_to_mon(new MPing, mc.get_mon_addrs(t));
+ cerr << isend << "\t" << ++sent << "\t" << received << "\r";
+ }
+ l.unlock();
+
+ // wait for messenger to finish
+ rank->wait();
+
+ return 0;
+}
+