servlib/discovery/IPAnnounce.cc
changeset 0 2b3e5ec03512
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/servlib/discovery/IPAnnounce.cc	Thu Apr 21 14:57:45 2011 +0100
@@ -0,0 +1,105 @@
+/*
+ *    Copyright 2006 Baylor University
+ * 
+ *    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 <oasys/util/OptParser.h>
+#include "bundling/BundleDaemon.h"
+#include "IPAnnounce.h"
+#include "conv_layers/IPConvergenceLayerUtils.h"
+#include "conv_layers/TCPConvergenceLayer.h"
+#include "conv_layers/UDPConvergenceLayer.h"
+
+namespace dtn {
+
+IPAnnounce::IPAnnounce()
+    : cl_addr_(INADDR_ANY), cl_port_(TCPConvergenceLayer::TCPCL_DEFAULT_PORT)
+{
+}
+
+bool
+IPAnnounce::configure(const std::string& name, ConvergenceLayer* cl, 
+                      int argc, const char* argv[])
+{
+    if (cl == NULL) return false;
+
+    cl_ = cl;
+    name_ = name;
+    type_.assign(cl->name());
+
+    // validate convergence layer details
+    if (strncmp(cl_->name(),"tcp",3) != 0 &&
+        strncmp(cl_->name(),"udp",3) != 0)
+    {
+        log_err("ip announce does not support cl type %s",
+                cl_->name());
+        return false;
+    }
+
+    // parse the options
+    oasys::OptParser p;
+    p.addopt(new oasys::InAddrOpt("cl_addr",&cl_addr_));
+    p.addopt(new oasys::UInt16Opt("cl_port",&cl_port_));
+    p.addopt(new oasys::UIntOpt("interval",&interval_));
+
+    const char* invalid;
+    if (! p.parse(argc, argv, &invalid))
+    {
+        log_err("bad parameter %s",invalid);
+        return false;
+    }
+
+    if (interval_ == 0)
+    {
+        log_err("interval must be greater than 0");
+        return false;
+    }
+
+    // convert from seconds to ms
+    interval_ *= 1000;
+
+    oasys::StringBuffer buf("%s:%d",intoa(cl_addr_),cl_port_);
+    local_.assign(buf.c_str());
+    return true;
+}
+
+size_t
+IPAnnounce::format_advertisement(u_char* bp, size_t len)
+{
+    EndpointID local(BundleDaemon::instance()->local_eid());
+    size_t length = FOUR_BYTE_ALIGN(local.length() + sizeof(DiscoveryHeader));
+
+    if (len <= length)
+        return 0;
+
+    DiscoveryHeader* hdr = (DiscoveryHeader*) bp;
+    memset(hdr,0,len);
+
+    hdr->cl_type = IPDiscovery::str_to_type(type().c_str());
+    hdr->interval = interval_ / 100;
+    hdr->length = htons(length);
+    hdr->inet_addr = cl_addr_;
+    hdr->inet_port = htons(cl_port_);
+    hdr->name_len = htons(local.length());
+
+    memcpy(hdr->sender_name,local.c_str(),local.length());
+    ::gettimeofday(&data_sent_,0);
+    return length;
+}
+
+} // dtn