onion-routing.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
*
* Author: Niki Hrovatin <niki.hrovatin@famnit.upr.si>
*/
 
#ifndef ONION_ROUTING_H
#define ONION_ROUTING_H
 
#include "ns3/core-module.h"
#include "ns3/ipv4-l3-protocol.h"
#include "ns3/ipv6-l3-protocol.h"
 
namespace ns3 {
 
struct orLayer
{
  uint8_t *nextHopIP; 
  uint8_t *innerLayer; 
  uint16_t innerLayerLen; 
};
 
class OnionRouting : public Object
{
public:
  static TypeId GetTypeId (void);
 
  enum OnionErrno {
    ERROR_NOTERROR,
    ERROR_PROT_NUMBER,
    ERROR_ROUTE_TO_SHORT,
    ERROR_ENCRYPTION,
    ERROR_DECRYPTION
  };
 
  OnionRouting (uint16_t sealPadding, const uint16_t protocolNumber);
 
  void BuildOnion (uint8_t *cipher, uint8_t **route, uint8_t **keys, uint16_t routeLen);
  void BuildOnion (uint8_t *cipher, uint8_t **route, uint8_t **keys, uint16_t routeLen,
                   uint8_t *endContent, uint16_t endContentLen);
  void BuildOnion (uint8_t *cipher, uint8_t **route, uint8_t **keys, uint8_t **layerContent,
                   uint16_t layerContentLen, uint16_t routeLen);
  void BuildOnion (uint8_t *cipher, uint8_t **route, uint8_t **keys, uint8_t **layerContent,
                   uint16_t layerContentLen, uint16_t routeLen, uint8_t *endContent,
                   uint16_t endContentLen); //setup creation of the onion
 
  void CreateOnion (uint8_t *cipher, uint8_t **route, uint8_t **keys, uint16_t index,
                    uint16_t routeLen, uint8_t **layerContent, uint16_t layerContentLen,
                    uint8_t *endContent, uint16_t endContentLen); //create the onion
 
  orLayer *PeelOnion (uint8_t *onion, uint16_t onionLen, uint8_t *publicKey,
                      uint8_t *secretKey); //peel one layer of the onion
 
  virtual void EncryptLayer (uint8_t *ciphertext, uint8_t *plaintext, int len,
                             uint8_t *key) const = 0;
 
  virtual void DecryptLayer (uint8_t *plaintext, uint8_t *ciphertext, uint16_t len,
                             uint8_t *publicKey, uint8_t *secretKey) const = 0;
 
  uint16_t OnionLength (uint16_t routeLen, uint16_t layerContentLen, uint16_t endContentLen);
 
  void AddressToStream (uint8_t *ip);
 
  enum OnionErrno GetErrno (void);
 
  uint16_t
      m_sealPadding; 
  uint16_t m_addressSize; 
  std::stringstream m_onionStream; 
  mutable enum OnionErrno m_errno; 
};
 
class OnionRoutingDummyEncryption : public OnionRouting
{
public:
  static TypeId GetTypeId (void);
 
  OnionRoutingDummyEncryption (uint16_t sealPadding, const uint16_t protocolNumber);
 
  void GenerateNewKey (void);
 
  uint8_t *GetEncryptionKey (void);
 
  virtual void EncryptLayer (uint8_t *ciphertext, uint8_t *message, int len, uint8_t *key) const;
  virtual void DecryptLayer (uint8_t *innerLayer, uint8_t *onion, uint16_t onionLen, uint8_t *pk,
                             uint8_t *sk) const;
 
  uint8_t m_encryptionkey[4]; 
};
 
} // namespace ns3
 
#endif /* ONION_ROUTING_H */