Attachment #766 for bug #407

View | Details | Raw Unified | Return to bug 407
Collapse All | Expand All




OlsrHelper::OlsrHelper (const OlsrHelper &o)
  : m_agentFactory (o.m_agentFactory)
{
  m_interfaceExclusions = o.m_interfaceExclusions;
}

OlsrHelper* 

  return new OlsrHelper (*this); 
}

void
OlsrHelper::ExcludeInterface (Ptr<Node> node, uint32_t interface)
{
  std::map< Ptr<Node>, std::set<uint32_t> >::iterator it = m_interfaceExclusions.find (node);

  if(it == m_interfaceExclusions.end ())
    {
      std::set<uint32_t> interfaces;
      interfaces.insert (interface);
    
      m_interfaceExclusions.insert (std::make_pair (node, std::set<uint32_t> (interfaces) ));
    }
  else
    {
      it->second.insert (interface);
    }
}

Ptr<Ipv4RoutingProtocol> 
OlsrHelper::Create (Ptr<Node> node) const
{
  Ptr<olsr::RoutingProtocol> agent = m_agentFactory.Create<olsr::RoutingProtocol> ();

  std::map<Ptr<Node>, std::set<uint32_t> >::const_iterator it = m_interfaceExclusions.find (node);

  if(it != m_interfaceExclusions.end ())
    {
      agent->SetInterfaceExclusions (it->second);
    }

  node->AggregateObject (agent);
  return agent;
}

(-)a/src/helper/olsr-helper.h (+13 lines)

Lines 24-29	Link Here

#include "ns3/node.h"

#include "ns3/node.h"

#include "node-container.h"

#include "node-container.h"

#include "ipv4-routing-helper.h"

#include "ipv4-routing-helper.h"

#include <map>

#include <set>

namespace ns3 {

namespace ns3 {

Lines 57-62	Link Here

*/

*/

  OlsrHelper* Copy (void) const;

  OlsrHelper* Copy (void) const;

/**

   * \param node the node for which an exception is to be defined

   * \param interface an interface of node on which OLSR is not to be installed

   * This method allows the user to specify an interface on which OLSR is not to be installed on

*/

  void ExcludeInterface (Ptr<Node> node, uint32_t interface);

/**

/**

   * \param node the node on which the routing protocol will run

   * \param node the node on which the routing protocol will run

   * \returns a newly-created routing protocol

   * \returns a newly-created routing protocol

Lines 72-77	Link Here

   * This method controls the attributes of ns3::olsr::RoutingProtocol

   * This method controls the attributes of ns3::olsr::RoutingProtocol

*/

*/

  void Set (std::string name, const AttributeValue &value);

  void Set (std::string name, const AttributeValue &value);

private:

private:

/**

/**

   * \internal

   * \internal

Lines 80-85	Link Here

*/

*/

  OlsrHelper &operator = (const OlsrHelper &o);

  OlsrHelper &operator = (const OlsrHelper &o);

  ObjectFactory m_agentFactory;

  ObjectFactory m_agentFactory;

  std::map< Ptr<Node>, std::set<uint32_t> > m_interfaceExclusions;

};

};

} // namespace ns3

} // namespace ns3

(-)a/src/routing/olsr/olsr-routing-protocol.cc (-4 / +18 lines)

Lines 251-256	Link Here

251

  NS_LOG_DEBUG ("Starting OLSR on node " << m_mainAddress);

251

  NS_LOG_DEBUG ("Starting OLSR on node " << m_mainAddress);

252

253

  Ipv4Address loopback ("127.0.0.1");

253

  Ipv4Address loopback ("127.0.0.1");

254

255

  bool canRunOlsr = false;

254

  for (uint32_t i = 0; i < m_ipv4->GetNInterfaces (); i++)

256

  for (uint32_t i = 0; i < m_ipv4->GetNInterfaces (); i++)

255

257

256

      Ipv4Address addr = m_ipv4->GetAddress (i, 0).GetLocal ();

258

      Ipv4Address addr = m_ipv4->GetAddress (i, 0).GetLocal ();

Lines 269-274	Link Here

269

          NS_ASSERT (GetMainAddress (addr) == m_mainAddress);

271

          NS_ASSERT (GetMainAddress (addr) == m_mainAddress);

270

272

271

273

274

      if(m_interfaceExclusions.find (i) != m_interfaceExclusions.end ())

275

        continue;

276

272

      // Create a socket to listen only on this interface

277

      // Create a socket to listen only on this interface

273

      Ptr<Socket> socket = Socket::CreateSocket (GetObject<Node> (),

278

      Ptr<Socket> socket = Socket::CreateSocket (GetObject<Node> (),

274

        UdpSocketFactory::GetTypeId());

279

        UdpSocketFactory::GetTypeId());

Lines 279-291	Link Here

279

284

280

      socket->Connect (InetSocketAddress (Ipv4Address (0xffffffff), OLSR_PORT_NUMBER));

285

      socket->Connect (InetSocketAddress (Ipv4Address (0xffffffff), OLSR_PORT_NUMBER));

281

      m_socketAddresses[socket] = m_ipv4->GetAddress (i, 0);

286

      m_socketAddresses[socket] = m_ipv4->GetAddress (i, 0);

287

288

      canRunOlsr = true;

282

289

283

290

284

  HelloTimerExpire ();

291

  if(canRunOlsr)

285

  TcTimerExpire ();

292

286

  MidTimerExpire ();

293

      HelloTimerExpire ();

294

      TcTimerExpire ();

295

      MidTimerExpire ();

287

296

288

  NS_LOG_DEBUG ("OLSR on node " << m_mainAddress << " started");

297

      NS_LOG_DEBUG ("OLSR on node " << m_mainAddress << " started");

298

289

299

290

300

291

void RoutingProtocol::SetMainInterface (uint32_t interface)

301

void RoutingProtocol::SetMainInterface (uint32_t interface)

Lines 293-298	Link Here

293

  m_mainAddress = m_ipv4->GetAddress (interface, 0).GetLocal ();

303

  m_mainAddress = m_ipv4->GetAddress (interface, 0).GetLocal ();

294

304

295

305

306

void RoutingProtocol::SetInterfaceExclusions (std::set<uint32_t> exceptions)

307

308

  m_interfaceExclusions = exceptions;

309

296

310

297

//

311

//

298

// \brief Processes an incoming %OLSR packet following RFC 3626 specification.

312

// \brief Processes an incoming %OLSR packet following RFC 3626 specification.




   **/
  std::vector<RoutingTableEntry> GetRoutingTableEntries () const;

private:
  std::set<uint32_t> m_interfaceExclusions;

public:
  std::set<uint32_t> GetInterfaceExclusions () const
    {
      return m_interfaceExclusions;
    }
  void SetInterfaceExclusions (std::set<uint32_t> exceptions);

protected:
  virtual void DoStart (void);
private:

Return to bug 407