sink.h

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
 
 
/*
* Copyright (c) 2020 DLTLT 
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation;
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*
* Corresponding author: Niki Hrovatin <niki.hrovatin@famnit.upr.si>
*/
 
 
#ifndef SINK_H
#define SINK_H
 
#include <sodium.h>
 
#include <assert.h> /* assert */
#include <iostream>
#include <map>
 
#include "ns3/wsn_node.h"
#include "ns3/proto-packet.pb.h"
#include "ns3/serializationwrapper.h"
#include "ns3/network-module.h"
#include "ns3/enums.h"
 
namespace ns3 {
 
class Sink : public Wsn_node
{
public:
  static TypeId GetTypeId (void);
 
  Sink ();
 
  virtual ~Sink ();
 
  void Setup (uint16_t *onionPathlengths, uint16_t numOnionLengths, int repeateTimes);
 
  void Accept (Ptr<Socket> socket, const ns3::Address &from);
 
  void SinkTasks ();
 
  void CheckOnion (void);
 
private:
  virtual void StartApplication (void);
 
  virtual void StopApplication (void);
 
  void ReceivePacket (Ptr<Socket> socket);
 
  void RecvHandshake (protomessage::ProtoPacket_Handshake *handshake_data, InetSocketAddress from);
 
  void RecvOnion (protomessage::ProtoPacket_OnionBody *onion_body);
 
  void SelectRoute (int *route, int routeLen);
 
  void PrepareOnion (int *route, int routeLen);
 
  void SendOnion (uint32_t firstHop, int routeLen, unsigned char *cipher, int cipherLen);
 
  uint16_t m_numnodes; 
  uint32_t m_onionDelay; 
  std::map<uint32_t, std::string>
      m_nodeManager; 
  uint32_t m_decoyNum =
      1203; 
  bool
      m_fixedOnionSize; 
  enum BodyOptions m_bodyOptions; //<!  Specifies how the onion body must behave
  uint16_t m_bodySize; //<! the onion body size in bytes if ns3::BodyOptions is set appropriately
 
  //simulation task
  int m_repeateTimes; 
  int m_repeateCount = 0; 
  int m_onionLengthIndex = 0; 
  uint16_t *m_onionPathLengths; 
  uint16_t m_numOnionLengths; 
 
  //onion sequence number
  int m_onionId = 1; 
 
  //VErbosity of onion
  std::stringstream m_onionStream; 
 
  //sink node keys
  std::string m_publickey; 
  std::string m_secretkey; 
};
 
} // namespace ns3
 
#endif /* SINK_H */