Search  

---

C:/temp/src/j2k/Net/TFTP/TFTP.cpp

Go to the documentation of this file.

#ifndef __J2K__TFTP_CPP__
#define __J2K__TFTP_CPP__

#include <j2k/Net/TFTP/TFTP.hpp>
#include <j2k/Fred/JFile.cpp>
#include <j2k/Fred/Error/JError.cpp>
#include <j2k/Net/TFTP/TFTP_Project.cpp>

TFTP::TFTP( int argc, char* argv[] )
{
  parser = NULL;
  option.ArgParser( argc, argv );
  option.createSocket();

   add_TFTP_ErrorHandlers();
  parser = new TFTP_Parser();
  option.sendFactory       = new TFTP_Sender();
  option.dataFactory       = new DataFactory();
  option.fileFactory       = new FileFactory();
  option.data_queue        = new DList_pPacket();
  option.waitingAck_queue  = new DList_pPacket();
  option.timeoutThread     = new TFTP_Timeout();   

  switch( option.pgmType )
  {
     case TFTP_Server:  server(); break;

     case TFTP_Client:  // Shell or prompt a R/W request
        if ( option.RW_command == 1 ) { read();   break; }
        if ( option.RW_command == 2 ) { write();  break; }
        option.RW_command = 0;

#ifndef SHELL
        char rw;
        char f[161];
        cout << "\n\nDo you want to (R)ead or (W)rite a file ? ";
        cin >> rw;
        cout << "\n\nEnter the name of the file: ";
        cin >> f;
        cout << "\n\nInitializing...\n";
        option.setFileName( f );

        if ( rw == 'r' || rw == 'R' || rw == 'd' || rw == 'D' ) {
          read(); break;
        } else {
          write(); break;
        }
#else
        TFTP_client_shell();        
#endif

       break;
       
     case TFTP_ClientRead:  read();  TFTP_client_shell(); break;
     case TFTP_ClientWrite: write(); TFTP_client_shell(); break;

     default:
        cout << "\n\nError: Client/Server not specified. \n";
        exit( 1 );
  }
}

inline TFTP::~TFTP() {
  if ( parser != NULL )  delete parser;
}

void TFTP::read() {
    printf("Reading\n");
    option.state = TFTP_Connected;  
    option.setMode( "octet" );
    option.build_OptionPacket();
    option.build_RRQ_WRQ_TemplatePacket();
    option.openFile( "wb" );

    option.showOptions();

    option.timeoutThread->start();
    TFTP_Packet pkt;
    //I'm just going to try to send a read request and send an ack back. 
    pkt.sendRRQ();
    option.state = TFTP_Reading;
    option.showOptions();
}

void TFTP::write() {
    printf("Writing\n");
    option.state = TFTP_Connected;
    option.build_OptionPacket();
    option.build_RRQ_WRQ_TemplatePacket();
    option.openFile( "rb" );
    option.showOptions();

    option.timeoutThread->start();
    TFTP_Packet pkt;
    //I'm just going to try to send a read request and send an ack back. 
    pkt.sendWRQ();
    option.state = TFTP_Writing;
    option.showOptions();
}

void TFTP::server() {
  printf("Server\n");
  option.state = TFTP_Connected;
  option.build_OptionPacket();
  
    option.showOptions();

    option.timeoutThread->start();

  //First receive request.
  //Send Ack
}

int main( int argc, char* argv[] )
{
  pthread_attr_t attr;
  sched_param    param;

  memset( &attr,  0, sizeof( attr  ) );
  memset( &param, 0, sizeof( param ) );

  pthread_attr_init( &attr );
  pthread_attr_setscope(        &attr, PTHREAD_SCOPE_SYSTEM );
  pthread_attr_setdetachstate(  &attr, PTHREAD_CREATE_JOINABLE );
  pthread_attr_setinheritsched( &attr, PTHREAD_EXPLICIT_SCHED );
  pthread_attr_setschedparam(   &attr, &param );
 
  param.sched_priority = 15; 
  pthread_setschedparam( pthread_self(), SCHED_RR, &param );

  size_t q = 0;
  TFTP( argc, argv );
  while( option.state != TFTP_Finished && !option.isEnding() )
  {
//    q++;
//    if ( q > 100 ) {
//      printf("."); 
//      q = 0;
//    }
    sched_yield();  // I'm lazy, let bright children run please.
  }
  return 0;
}
#endif

---