Search  

---

C:/temp/src/j2k/Net/TFTP/TFTP_Option.hpp

Go to the documentation of this file.

#ifndef __J2K__TFTP_Option_HPP__
#define __J2K__TFTP_Option_HPP__

#include <j2k/Fred/Standard.hpp>
#include <j2k/Fred/JFile.hpp>
#include <j2k/Net/TFTP/TFTP_header.hpp>
#include <j2k/Posix/RWLock.hpp>
#include <j2k/Net/TFTP/DataFactory.hpp>
#include <j2k/Net/TFTP/FileFactory.hpp>

// Needed for str* functions not included with g++ <string.h>
#include <j2k/Fred/String/Str-fn.hpp>

#define  DEFAULT_WINDOWSIZE "1"
#define  DEFAULT_BLOCKSIZE  "512"
#define  DEFAULT_TIMEOUT    "5"

//#define  DEFAULT_WINDOWSIZE "1"
//#define  DEFAULT_BLOCKSIZE  "512"
//#define  DEFAULT_TIMEOUT    "5"

#define  DEFAULT_MODE       "octet"
#define  DEFAULT_HOSTNAME   "fbi.ece.concordia.ca"
#define  DEFAULT_PORT       2000

#define    FILENAME_LENGTH   161
#define    FILEPATH_LENGTH   161
#define    HOSTNAME_LENGTH   161

#define    FILESIZE_LENGTH   15
#define   BLOCKSIZE_LENGTH    6
#define     TIMEOUT_LENGTH    4
#define        MODE_LENGTH    9
#define  WINDOWSIZE_LENGTH   12  

#define  RRQ_WRQ_TEMPLATE_LENGTH  (FILENAME_LENGTH + MODE_LENGTH + 2)
#define   OPTION_TEMPLATE_LENGTH   2

class TFTP_Option 
{
public:
   TFTP_Option();

   virtual ~TFTP_Option();

   // Parse the arguments passed on the command line.
   void ArgParser( int argc, char* argv[] );

  // Detect what kind of argument it is
  // and initialize variables in consequence to that argument.
  void ArgDetect( const char* s, char* o );

  // Config hostname and port for client using string option
  void ConfigClient( char* opt );

  // Config port for server
  void TFTP_Option::ConfigServer( char* opt );

  // Build a template char[] that will be used by RRQ and WRQ
  // in Packet class.
  void build_RRQ_WRQ_TemplatePacket();

  void build_OptionPacket();

  // Set the filename
  inline void setFileName( const char* fname );

  // Get the filename
  const char* getFileName();

  // Set the blocksize
  inline void setBlockSize( size_t bsize );

  // Requested blocksize
  inline void requestedBlockSize( size_t bsize );

  // Get the blocksize
  inline size_t getBlockSize();


  // Set Port Number
  inline void setPort( size_t portNo );

  // Set the HostName
  inline void setHostName( const char* host );

  // Set the windowsize
  inline void setWindowSize( size_t wsize );

  // Requested windowsize
  inline void requestedWindowSize( size_t wsize );

  // Get the windowsize
  inline size_t getWindowSize();

  // Set the filesize
  inline void setFileSize( size_t fsize );

  // Requested windowsize
  inline void requestedFileSize( size_t fsize );

  // Get the filesize    
  inline const char* getFileSize();

  // Set the mode
  inline void setMode( char* m );

  // Requested mode
  inline void requestedMode( const char* m );

  // Get the mode  
  inline const char* getMode();

  // Set the timeout
  inline void setTimeOut( size_t t );

  // Requested timeout
  inline void requestedTimeOut( size_t t );

  // Get the timeout
  inline const char* getTimeOut();

  // Get the timeout value
  inline ULONG getTimeOutValue();

  // Get the blocksize value
  inline ULONG getBlockSizeValue();

  // Get the windowsize value
  inline ULONG getWindowSizeValue();

  // "rb" for read, "wb" for write
  void  openFile( char* mode = "rb" );

  // Ask user to select Server or client.     
  void ServiceSelection();

/**************************************************************************
 ***  Create a socket with the given parsed parameters                  ***
 **************************************************************************/
    void createSocket();
    
    void showStr( const char* n, const char* s );    
    void showVal( const char* n, size_t s );
    void showChar( const char*n, const char* s, size_t sz );

    void showOptions();

    /*
  bit 0 Init
  bit 1 Connected
  bit 2 Ack for RRQ/WRQ
  bit 3 Reading
  bit 4 Writing
  bit 5 End of Read/Write
  bit 6 Error
  bit 7 Finished, program exit signal

  Disconnect means that we are not connected and all data
  in TFTP_Option are initialized.
*/

  inline int isInitialized();
  inline int isConnected();
  inline int isWaitingForAck();
  inline int isReading();
  inline int isWriting();
  inline int isEnding();
  inline int hasError();
  inline int isKilling();
 
  inline int isRRQ_Valid();
  inline int isWRQ_Valid();
  inline int isDATA_Valid();
  inline int isACK_Valid();
  inline int isOACK_Valid();

  inline int isServer();
  inline int isClient();
 
private:        
  char*   fpath;     // Full path          "/home/fred/test.txt"
  char*   filename;  // Only the filename  "test.txt"
  char*   filesize;
  char*   timeout;
  char*   blocksize;
  char*   mode;
  char*   windowsize;

  USHORT  port;
  char*   hostname;
  size_t  hostnameLength;

  size_t       fpathLength;
  size_t    filenameLength;
  size_t    filesizeLength; 
  size_t   blocksizeLength;
  size_t  windowsizeLength;
  size_t     timeoutLength;
  size_t        modeLength;

  char*    openedFile;

// Shared variables...
public:
  JFile*        myFile;
  UDP_Socket*   sender;
  DataFactory*  dataFactory;
  FileFactory*  fileFactory;

  char*  RRQ_WRQ_template;
  size_t RRQ_WRQ_templateLength;

  char*  Option_template;
  size_t Option_templateLength;

  RWLock  rwlock1_global;
  RWLock  rwlock2_file;
  RWLock  rwlock3_templates;

  enum TFTP_Status_t  state;
  enum TFTP_Type_t    pgmType;

  int  RW_command;
  BOOL OACK_arg;

  RWLock          rwlock5_data_queue;
  DList_pPacket*  data_queue;

  RWLock          rwlock6_wait_queue;
  DList_pPacket*  waitingAck_queue;

  BOOL            Done;
  size_t          LastData;

  double          currentTime;
  size_t          windowsizeLeft;

  size_t          TopAck;

  TFTP_Timeout*   timeoutThread;
  TFTP_Sender*    sendFactory;

  double         debug;

  friend TFTP_OptionSaved;
};

#include <j2k/Net/TFTP/TFTP_Sender.hpp>

#endif

---