MCFTPid.h

Go to the documentation of this file.

// MCFTPid.h
//////////////////////////////////////////////////////////////////////
//          (c) Copyright 2002 Brown Deer Technology, LLC.
//                        All rights reserved.
//////////////////////////////////////////////////////////////////////

#ifndef MCFTPID_H
#define MCFTPID_H

#ifdef USE_STD
#include <iostream>
#include <string>
using namespace std;
#else
#include <iostream.h>
#include <string.h>
#endif

#include "utility.h"
#include "BufferBase.h"

struct MCFTPid
{
   public:

      MCFTPid();
      MCFTPid( const MCFTPid& );
      MCFTPid( 
        unsigned int dst_ip, unsigned int dst_port, 
        unsigned int src_ip, unsigned int src_port, 
        unsigned int mssgnum 
      );

      MCFTPid& operator () ();
      MCFTPid& operator () ( const MCFTPid& );
      MCFTPid& operator () ( 
        unsigned int dst_ip, unsigned int dst_port, 
        unsigned int src_ip, unsigned int src_port, 
        unsigned int mssgnum 
      );

      MCFTPid& copy( const MCFTPid id );
      MCFTPid& operator = ( const MCFTPid& rhs );

      bool is_equal( const MCFTPid& rhs );
      bool operator == ( const MCFTPid& rhs );
      bool operator != ( const MCFTPid& rhs );

      void write( BufferBase& );
      MCFTPid& operator << ( BufferBase& );
      void read( BufferBase& );
      MCFTPid& operator >> ( BufferBase& );

      void write( istream& );
//      MCFTPid& operator << ( istream& );
      void read( ostream& );
//      MCFTPid& operator >> ( ostream& );

      void set_dst_ip( unsigned int dst_ip );
      void set_dst_ip( 
        unsigned char ip3, 
        unsigned char ip2, 
        unsigned char ip1, 
        unsigned char ip0 
      ); // address = ip3.ip2.ip1.ip0
      void set_dst_port( unsigned int dst_port );
      void set_src_ip( unsigned int src_ip );
      void set_src_ip( 
        unsigned char ip3, 
        unsigned char ip2, 
        unsigned char ip1, 
        unsigned char ip0 
      ); // adddress = ip3.ip2.ip1.ip0
      void set_src_port( unsigned int src_port );
      void set_mssgnum( unsigned int mssgnum );

      unsigned int dst_ip();
      unsigned char dst_ip( int b );
      unsigned int dst_port();
      unsigned int src_ip();
      unsigned char src_ip( int b );
      unsigned int src_port();
      unsigned int mssgnum();

      string hex();
      string filename();

      void report( ostream& = cout );

   private:
      unsigned int _dst_ip;
      unsigned int _dst_port;
      unsigned int _src_ip;
      unsigned int _src_port;
      unsigned int _mssgnum;

   friend BufferBase& operator >> ( BufferBase&, MCFTPid& );
   friend BufferBase& operator << ( BufferBase&, MCFTPid& );

   friend istream& operator >> ( istream&, MCFTPid& );
   friend ostream& operator << ( ostream&, MCFTPid& );

};

inline
MCFTPid::MCFTPid()
  : _dst_ip(0), _dst_port(0), 
    _src_ip(0), _src_port(0), 
    _mssgnum(0)
{}

inline
MCFTPid::MCFTPid( const MCFTPid& id )
  : _dst_ip(id._dst_ip), _dst_port(id._dst_port), 
    _src_ip(id._src_ip), _src_port(id._src_port), 
    _mssgnum(id._mssgnum)
{}

inline
MCFTPid::MCFTPid( 
  unsigned int dst_ip, unsigned int dst_port, 
  unsigned int src_ip, unsigned int src_port, 
  unsigned int mssgnum 
) : _dst_ip(dst_ip), _dst_port(dst_port), 
    _src_ip(src_ip), _src_port(src_port), 
    _mssgnum(mssgnum)
{}

inline
MCFTPid& MCFTPid::operator () ()
{
   _dst_ip = 0;
   _dst_port = 0;
   _src_ip = 0;
   _src_port = 0;
   _mssgnum = 0;
   return *this;
}

inline
MCFTPid& MCFTPid::operator () ( const MCFTPid& id )
{
   _dst_ip = id._dst_ip;
   _dst_port = id._dst_port;
   _src_ip = id._src_ip;
   _src_port = id._src_port;
   _mssgnum = id._mssgnum;
   return *this;
}

inline
MCFTPid& MCFTPid::operator() ( 
  unsigned int dst_ip, unsigned int dst_port, 
  unsigned int src_ip, unsigned int src_port, 
  unsigned int mssgnum 
) 
{
   _dst_ip = dst_ip;
   _dst_port = dst_port;
   _src_ip = src_ip;
   _src_port = src_port;
   _mssgnum = mssgnum;
   return *this;
}

inline
MCFTPid& MCFTPid::copy( const MCFTPid id )
{
   _dst_ip = id._dst_ip;
   _dst_port = id._dst_port;
   _src_ip = id._src_ip;
   _src_port = id._src_port;
   _mssgnum = id._mssgnum;
   return *this;
}

inline
MCFTPid& MCFTPid::operator = ( const MCFTPid& rhs ) 
{ 
   return copy(rhs); 
}

inline
bool MCFTPid::is_equal( const MCFTPid& id )
{
   if ( _dst_ip != id._dst_ip ) return false;
   if ( _dst_port != id._dst_port ) return false;
   if ( _src_ip != id._src_ip ) return false;
   if ( _src_port != id._src_port ) return false;
   if ( _mssgnum != id._mssgnum ) return false;
   return true;
}

inline
bool MCFTPid::operator == ( const MCFTPid& rhs ) 
{ 
   return is_equal(rhs); 
}

inline
bool MCFTPid::operator != ( const MCFTPid& rhs ) 
{ 
   return !is_equal(rhs); 
}

inline
void MCFTPid::write( BufferBase& buffer )
{
   bytepack bp;
   buffer >> bp( 0, 4 ); _dst_ip = int(bp);
   buffer >> bp( 0, 2 ); _dst_port = int(bp);
   buffer >> bp( 0, 4 ); _src_ip = int(bp);
   buffer >> bp( 0, 2 ); _src_port = int(bp);
   buffer >> bp( 0, 4 ); _mssgnum = int(bp);
}

inline
MCFTPid& MCFTPid::operator << ( BufferBase& buffer )
{
   write(buffer);
   return *this;
}

inline
void MCFTPid::read( BufferBase& buffer )
{
   bytepack bp;
   buffer << bp( int(_dst_ip), 4 );
   buffer << bp( int(_dst_port), 2 );
   buffer << bp( int(_src_ip), 4 );
   buffer << bp( int(_src_port), 2 );
   buffer << bp( int(_mssgnum), 4 );
}

inline
MCFTPid& MCFTPid::operator >> ( BufferBase& buffer )
{
   read(buffer);
   return *this;
}

inline
void MCFTPid::write( istream& is )
{
   bytepack bp;
   is >> bp( 0, 4 ); _dst_ip = int(bp);
   is >> bp( 0, 2 ); _dst_port = int(bp);
   is >> bp( 0, 4 ); _src_ip = int(bp);
   is >> bp( 0, 2 ); _src_port = int(bp);
   is >> bp( 0, 4 ); _mssgnum = int(bp);
}

//inline
//MCFTPid& MCFTPid::operator << ( istream& is )
//{
//   write(is);
//   return *this;
//}

inline
void MCFTPid::read( ostream& os )
{
   bytepack bp;
   os << bp( int(_dst_ip), 4 );
   os << bp( int(_dst_port), 2 );
   os << bp( int(_src_ip), 4 );
   os << bp( int(_src_port), 2 );
   os << bp( int(_mssgnum), 4 );
}

//inline
//MCFTPid& MCFTPid::operator >> ( ostream& os )
//{
//   read(os);
//   return *this;
//}

inline
void MCFTPid::set_dst_ip( unsigned int dst_ip ) { _dst_ip = dst_ip; }

inline
void MCFTPid::set_dst_ip( 
  unsigned char ip3, unsigned char ip2, unsigned char ip1, unsigned char ip0 
) 
{ 
   _dst_ip = ip3;
   _dst_ip = _dst_ip*256 + ip2;
   _dst_ip = _dst_ip*256 + ip1;
   _dst_ip = _dst_ip*256 + ip0;
}

inline
void MCFTPid::set_dst_port( unsigned int dst_port ) { _dst_port = dst_port; }

inline
void MCFTPid::set_src_ip( unsigned int src_ip ) { _src_ip = src_ip; }

inline
void MCFTPid::set_src_ip( 
  unsigned char ip3, unsigned char ip2, unsigned char ip1, unsigned char ip0 
) 
{ 
   _src_ip = ip3;
   _src_ip = _src_ip*256 + ip2;
   _src_ip = _src_ip*256 + ip1;
   _src_ip = _src_ip*256 + ip0;
}

inline
void MCFTPid::set_src_port( unsigned int src_port ) { _src_port = src_port; }

inline
void MCFTPid::set_mssgnum( unsigned int mssgnum ) { _mssgnum = mssgnum; }

inline
unsigned int MCFTPid::dst_ip() { return _dst_ip; }

inline
unsigned char MCFTPid::dst_ip( int b ) 
{
   int itmp = 0;
   switch ( b ) {
      case 0:
         itmp = _dst_ip & 0x000000FF;
         break;
      case 1:
         itmp = (_dst_ip & 0x0000FF00) >> 8;
         break;
      case 2:
         itmp = (_dst_ip & 0x00FF0000) >> 16;
         break;
      case 3:
         itmp = (_dst_ip & 0xFF000000) >> 24;
         break;
   }
   return (unsigned char)(itmp);
}

inline
unsigned int MCFTPid::dst_port() { return _dst_port; }

inline
unsigned int MCFTPid::src_ip() { return _src_ip; }

inline
unsigned char MCFTPid::src_ip( int b ) 
{
   int itmp = 0;
   switch ( b ) {
      case 0:
         itmp = _src_ip & 0x000000FF;
         break;
      case 1:
         itmp = (_src_ip & 0x0000FF00) >> 8;
         break;
      case 2:
         itmp = (_src_ip & 0x00FF0000) >> 16;
         break;
      case 3:
         itmp = (_src_ip & 0xFF000000) >> 24;
         break;
   }
   return (unsigned char)(itmp);
}

inline
unsigned int MCFTPid::src_port() { return _src_port; }

inline
unsigned int MCFTPid::mssgnum() { return _mssgnum; }

inline
string MCFTPid::hex()
{
   string s = "";

   char* tmp2 = new char[4+1];
   char* tmp4 = new char[8+1];

   inttohex( int(_dst_ip), tmp4, 4 ); s += tmp4;
   inttohex( _dst_port, tmp2, 2 ); s += tmp2;
   inttohex( _src_ip, tmp4, 4 ); s += tmp4;
   inttohex( _src_port, tmp2, 2 ); s += tmp2;
   inttohex( _mssgnum, tmp4, 4 ); s += tmp4;

   return s;
}

inline
string MCFTPid::filename()
{
   return hex() + ".mcftp";
}

inline
void MCFTPid::report( ostream& os )
{
   os << "MCFTPid::report(){\n";
   os << "dst_ip = " << _dst_ip << " = " 
        << int(dst_ip(3)) << '.' << int(dst_ip(2)) << '.' 
        << int(dst_ip(1)) << '.' << int(dst_ip(0)) << '\n';
   os << "dst_port = " << _dst_port << '\n';
   os << "src_ip = " << _src_ip << '\n';
   os << "src_ip = " << _src_ip << " = " 
        << int(src_ip(3)) << '.' << int(src_ip(2)) << '.' 
        << int(src_ip(1)) << '.' << int(src_ip(0)) << '\n';
   os << "src_port = " << _src_port << '\n';
   os << "mssgnum = " << _mssgnum << '\n';
   os << "hex = " << hex() << '\n';
   os << "}\n";
}

// ##### friends #####

inline
BufferBase& operator << ( BufferBase& buffer, MCFTPid& id )
{
   id.read(buffer);
   return buffer;
}

inline
BufferBase& operator >> ( BufferBase& buffer, MCFTPid& id )
{
   id.write(buffer);
   return buffer;
}

inline
ostream& operator << ( ostream& os, MCFTPid& id )
{
   id.read(os);
   return os;
}

inline
istream& operator >> ( istream& is, MCFTPid& id )
{
   id.write(is);
   return is;
}

#endif

---