--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/daemon/dtnd.cc Thu Apr 21 14:57:45 2011 +0100
@@ -0,0 +1,264 @@
+/*
+ * Copyright 2004-2006 Intel Corporation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifdef HAVE_CONFIG_H
+# include <dtn-config.h>
+#endif
+
+#include <errno.h>
+#include <string>
+#include <sys/time.h>
+
+#include <oasys/debug/Log.h>
+#include <oasys/io/NetUtils.h>
+#include <oasys/tclcmd/ConsoleCommand.h>
+#include <oasys/tclcmd/TclCommand.h>
+#include <oasys/thread/Timer.h>
+#include <oasys/util/App.h>
+#include <oasys/util/Getopt.h>
+#include <oasys/util/StringBuffer.h>
+
+#include "applib/APIServer.h"
+#include "cmd/TestCommand.h"
+#include "servlib/DTNServer.h"
+#include "storage/DTNStorageConfig.h"
+#include "bundling/BundleDaemon.h"
+
+#include "bundling/S10Logger.h"
+
+extern const char* dtn_version;
+
+/**
+ * Namespace for the dtn daemon source code.
+ */
+namespace dtn {
+
+/**
+ * Thin class that implements the daemon itself.
+ */
+class DTND : public oasys::App {
+public:
+ DTND();
+ int main(int argc, char* argv[]);
+
+protected:
+ TestCommand* testcmd_;
+ oasys::ConsoleCommand* consolecmd_;
+ DTNStorageConfig storage_config_;
+
+ // virtual from oasys::App
+ void fill_options();
+
+ void init_testcmd(int argc, char* argv[]);
+ void run_console();
+};
+
+//----------------------------------------------------------------------
+DTND::DTND()
+ : App("DTND", "dtnd", dtn_version),
+ testcmd_(NULL),
+ consolecmd_(NULL),
+ storage_config_("storage", // command name
+ "berkeleydb", // storage type
+ "DTN", // DB name
+ INSTALL_LOCALSTATEDIR "/dtn/db") // DB directory
+{
+ // override default logging settings
+ loglevel_ = oasys::LOG_NOTICE;
+ debugpath_ = "~/.dtndebug";
+
+ // override defaults from oasys storage config
+ storage_config_.db_max_tx_ = 1000;
+
+ testcmd_ = new TestCommand();
+ consolecmd_ = new oasys::ConsoleCommand("dtn% ");
+}
+
+//----------------------------------------------------------------------
+void
+DTND::fill_options()
+{
+ fill_default_options(DAEMONIZE_OPT | CONF_FILE_OPT);
+
+ opts_.addopt(
+ new oasys::BoolOpt('t', "tidy", &storage_config_.tidy_,
+ "clear database and initialize tables on startup"));
+
+ opts_.addopt(
+ new oasys::BoolOpt(0, "init-db", &storage_config_.init_,
+ "initialize database on startup"));
+
+ opts_.addopt(
+ new oasys::InAddrOpt(0, "console-addr", &consolecmd_->addr_, "<addr>",
+ "set the console listening addr (default off)"));
+
+ opts_.addopt(
+ new oasys::UInt16Opt(0, "console-port", &consolecmd_->port_, "<port>",
+ "set the console listening port (default off)"));
+
+ opts_.addopt(
+ new oasys::IntOpt('i', 0, &testcmd_->id_, "<id>",
+ "set the test id"));
+}
+
+//----------------------------------------------------------------------
+void
+DTND::init_testcmd(int argc, char* argv[])
+{
+ for (int i = 0; i < argc; ++i) {
+ testcmd_->argv_.append(argv[i]);
+ testcmd_->argv_.append(" ");
+ }
+
+ testcmd_->bind_vars();
+ oasys::TclCommandInterp::instance()->reg(testcmd_);
+}
+
+//----------------------------------------------------------------------
+void
+DTND::run_console()
+{
+ // launch the console server
+ if (consolecmd_->port_ != 0) {
+ log_info_p("/dtnd", "starting console on %s:%d",
+ intoa(consolecmd_->addr_), consolecmd_->port_);
+
+ oasys::TclCommandInterp::instance()->
+ command_server(consolecmd_->prompt_.c_str(),
+ consolecmd_->addr_, consolecmd_->port_);
+ }
+
+ if (daemonize_ || (consolecmd_->stdio_ == false)) {
+ oasys::TclCommandInterp::instance()->event_loop();
+ } else {
+ oasys::TclCommandInterp::instance()->
+ command_loop(consolecmd_->prompt_.c_str());
+ }
+}
+
+//----------------------------------------------------------------------
+int
+DTND::main(int argc, char* argv[])
+{
+ init_app(argc, argv);
+
+ log_notice_p("/dtnd", "DTN daemon starting up... (pid %d)", getpid());
+
+
+ if (oasys::TclCommandInterp::init(argv[0], "/dtn/tclcmd") != 0)
+ {
+ log_crit_p("/dtnd", "Can't init TCL");
+ notify_and_exit(1);
+ }
+
+ // stop thread creation b/c of initialization dependencies
+ oasys::Thread::activate_start_barrier();
+
+ DTNServer* dtnserver = new DTNServer("/dtnd", &storage_config_);
+ APIServer* apiserver = new APIServer();
+
+ dtnserver->init();
+
+ oasys::TclCommandInterp::instance()->reg(consolecmd_);
+ init_testcmd(argc, argv);
+
+ if (! dtnserver->parse_conf_file(conf_file_, conf_file_set_)) {
+ log_err_p("/dtnd", "error in configuration file, exiting...");
+ notify_and_exit(1);
+ }
+
+ if (storage_config_.init_)
+ {
+ log_notice_p("/dtnd", "initializing persistent data store");
+ }
+
+ if (! dtnserver->init_datastore()) {
+ log_err_p("/dtnd", "error initializing data store, exiting...");
+ notify_and_exit(1);
+ }
+
+ // If we're running as --init-db, make an empty database and exit
+ if (storage_config_.init_ && !storage_config_.tidy_)
+ {
+ dtnserver->close_datastore();
+ log_info_p("/dtnd", "database initialization complete.");
+ notify_and_exit(0);
+ }
+
+ if (BundleDaemon::instance()->local_eid().equals(EndpointID::NULL_EID()))
+ {
+ log_err_p("/dtnd", "no local eid specified; use the 'route local_eid' command");
+ notify_and_exit(1);
+ }
+ s10_setlocal(BundleDaemon::instance()->local_eid().c_str());
+ s10_daemon(S10_STARTING);
+
+ // if we've daemonized, now is the time to notify our parent
+ // process that we've successfully initialized
+ if (daemonize_) {
+ daemonizer_.notify_parent(0);
+ }
+
+ dtnserver->start();
+ if (apiserver->enabled()) {
+ apiserver->bind_listen_start(apiserver->local_addr(),
+ apiserver->local_port());
+ }
+ oasys::Thread::release_start_barrier(); // run blocked threads
+
+ // if the test script specified something to run for the test,
+ // then execute it now
+ if (testcmd_->initscript_.length() != 0) {
+ oasys::TclCommandInterp::instance()->
+ exec_command(testcmd_->initscript_.c_str());
+ }
+
+ // allow startup messages to be flushed to standard-out before
+ // the prompt is displayed
+ oasys::Thread::yield();
+ usleep(500000);
+
+ run_console();
+
+ log_notice_p("/dtnd", "command loop exited... shutting down daemon");
+
+ apiserver->stop();
+
+ oasys::TclCommandInterp::shutdown();
+ dtnserver->shutdown();
+
+ // close out servers
+ delete dtnserver;
+ // don't delete apiserver; keep it around so slow APIClients can finish
+
+ // give other threads (like logging) a moment to catch up before exit
+ oasys::Thread::yield();
+ sleep(1);
+
+ // kill logging
+ oasys::Log::shutdown();
+
+ return 0;
+}
+
+} // namespace dtn
+
+int
+main(int argc, char* argv[])
+{
+ dtn::DTND dtnd;
+ dtnd.main(argc, argv);
+}