Search  

---

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

Go to the documentation of this file.

#ifndef __J2K__TFTP_ParserStatic_HPP__
#define __J2K__TFTP_ParserStatic_HPP__

// [  2 |     n+1     |   n+1   |      n+1      | ... ]
// [ 01 | Filename \0 | Mode \0 | extensions \0 | ... ]
void RRQ_driver( char* rawPacket, int sz )
{
  // %0000,0011  && Server
  // We are the Server and the client wants to read a file from the server.
  if (  option.isRRQ_Valid()  ) 
  {
    option.state = TFTP_Reading;

    // Save default, for future request from other clients.
    savedOptions.save();

    // Parse RRQ args
    parseOptions( rawPacket, sz, 2 );

    option.showOptions();
    fflush( stdout );

    option.openFile( "rb" );
    fflush( stdout );

    if ( option.dataFactory == NULL ) {
      option.dataFactory = new DataFactory();
    }

    // Cheap scheduling without threads
    option.dataFactory->run();
   
    // Let Data Factory the time to produce ! =)
    //    sched_yield();

    //option.build_RRQ_WRQ_TemplatePacket();
    option.build_OptionPacket();

    printf("back\n");
    option.showOptions();
    fflush( stdout );

    OACK_arg = FALSE;         
    // Send Data/OACK...
    if ( OACK_arg == TRUE ) 
    {
        TFTP_Packet* pkt = new TFTP_Packet();
        pkt->sendOAck();

        // Ack#0 will start sendFactory then...

    } else {

        if ( option.sendFactory == NULL ) {
          option.sendFactory = new TFTP_Sender();
        }

        option.sendFactory->start();

        // Let Send Factory the time to Send ! =)
        sched_yield(); 

        // Send data #1
        // Start the DataFactory & SendFactory
        // which will create data packet and send them.
    }

  } else {
 
    option.state = TFTP_Error;

    // We are in a situation where a RRQ packet is not expected,
    // in any given way and therefore makes no sense.

    // Send Error #4 packet.
    SendIllegalOpError( "RRQ Rejected: Your Read Request makes no sense. \n" );
  }
}

// [  2 |     n+1     |   n+1   |      n+1      | ... ]
// [ 02 | Filename \0 | Mode \0 | extensions \0 | ... ]
void WRQ_driver( char* rawPacket, int sz )
{
  // %0000,0011  && Server
  // We are the Server and the client wants to write
  // a file on the server side.
  if (  option.isWRQ_Valid()  )
  {
    option.state = TFTP_Writing;

    // Save default, for future request from other clients.
    savedOptions.save();

    // Parse WRQ args
    parseOptions( rawPacket, sz, 2 );
    option.showOptions();
    option.openFile( "wb" );
    
    if ( option.fileFactory != NULL ) {
      option.fileFactory = new FileFactory();
    }




//    option.fileFactory->start(28);

    option.build_RRQ_WRQ_TemplatePacket();
    option.build_OptionPacket();

    // Send Data/OACK...
    if ( OACK_arg == TRUE ) 
    {
        // Send OAck
        TFTP_Packet* pkt = new TFTP_Packet();
        pkt->sendOAck();

    } else {

        TFTP_Packet* pkt = new TFTP_Packet();
        pkt->sendAck( 0 );
        // Send Ack Block#0
    }
  } else {
 
    option.state = TFTP_Error;

    // We are in a situation where a WRQ packet is not expected,
    // in any given way and therefore makes no sense.

    // Send Error #4 packet.
    SendIllegalOpError( "WRQ Rejected: Your Write Request makes no sense. \n" );
  }
}

int TestTimeout( short blockNo )
{
return 1;

    if ( blockNo == 2 ) 
    {
        if ( option.debug == 0 ) 
        {
            option.debug = option.currentTime + 3;
        } else 
        {
            if ( option.debug > option.currentTime ) {
         printf( "\nLet NOT TIMEOUT !\n");
              fflush( stdout );
              return 1;
            }
        }
        printf( "\nTIMEOUT TEST!\n");
        fflush( stdout );
        return 0;
    }
    
    return 1;    
}

// Might act as an ack for poor type TFTP

// Data #1  as ack for RRQ on the client side
// Data #1  after OACK received for RRQ on the client side
// Data > 1 for reading on the client side
// Data received on the server side for WRQ/Writing

// [  2 |    2     |    n     ]
// [ 03 | Block No |   Data   ]
void DATA_driver( char* rawPacket, int sz )
{
  if ( option.isDATA_Valid()  )
  {
    TFTP_Packet* pktData = new TFTP_Packet();
    pktData->createData( rawPacket, sz );

//    option.rwlock5_data_queue.write();
//    option.data_queue->enqueue( pktData );
//    option.rwlock5_data_queue.releaseWrite();

    TFTP_Packet* pktAck = new TFTP_Packet();
    short blockNo = 0;
    memcpy( (char*)&blockNo, rawPacket + 2, 2 );

//    option.fileFactory->start(28);

    option.fileFactory->directWrite( pktData );

    sched_yield();

    if ( TestTimeout( blockNo ) )
    {
        printf("Sending Ack for block number %d\n",blockNo);
        fflush(stdout);
        pktAck->sendAck( blockNo );  // Auto-Respond.
    }
    
    if (option.isEnding())
    {
        exit( 0);
    }    
  } else {
 
    option.state = TFTP_Error;

    // Send Error 4.
    SendIllegalOpError( "DATA Rejected: Your NOT allowed yet to send data. \n" );
  }
}

// When do we receive ack ?
// [  2 |    2     ]
// [ 04 | Block No ]
void ACK_driver( char* rawPacket, int sz )
{           
  if ( option.isACK_Valid() )
  {
    short blockNo = 0;
    memcpy( (char*)&blockNo, rawPacket+2, 2 );

    // Scan waitingForAck SList and remove packets from Timeout Queue.
    if (  option.isClient() && option.state == TFTP_Writing ) 
    {
      if ( option.dataFactory == NULL ) {
        option.dataFactory = new DataFactory();
     }

      // Cheap scheduling without threads
      option.dataFactory->run();  
    }
        
    if ( blockNo == 0 && \
       ( (  option.isServer() && option.state == TFTP_Reading ) || \
         (  option.isClient() && option.state == TFTP_Writing ) ) )
    {
        if ( option.sendFactory != NULL ) {
          option.sendFactory = new TFTP_Sender();
        }

        option.sendFactory->start();
        // Let Send Factory the time to Send ! =)
        // sched_yield(); 

        // Send data #1
        // Start the DataFactory & TFTP_DataSender
        // which will create data packet and send them.
    }
    
    
    
    if (blockNo == option.LastData && blockNo != 0 )
    {
        printf("Got the ACK of the last packet. \n");
        option.state= TFTP_Finished;
        //exit(0);
    }
    
    if ( blockNo > 0 ) {
        TFTP_DataAck clean( blockNo );  
        clean.run();
        option.windowsizeLeft++;
        printf("I received an acknowledgment, for block number %d\n",blockNo);

        // printf("The windowsizeLeft is now %d\n",option.windowsizeLeft);

      // Should be auto-destroyed after this if block
    }
    
    
    if ( option.isClient() && option.state == TFTP_WritingNotAck ) {
      // The server refused all extension and used default TFTP.

    }

    if ( option.isServer() && option.isReading() ) {
      // The client approved server OACK and replied.
    }

   }else {
       option.state = TFTP_Error;
    // Send Error 4.
       SendIllegalOpError( "ACK Rejected ?! \n" );
  }
}

// [  2 |    2     |     n+1      ]
// [ 05 | Error No | Error Msg \0 ]
void ERROR_driver( char* rawPacket, int sz )
{
  // Allow DList to complete ?
  option.state = TFTP_Finished;
  exit( 0 );
}

// When do we receive OAck ?
// [  2 |      n+1      | ... ]
// [ 06 | extensions \0 | ... ]
void OACK_driver( char* rawPacket, int sz )
{
  if (  ( option.state == TFTP_ReadingNotAck ||  \
          option.state == TFTP_WritingNotAck ) && \
          option.isClient()  )
  {
     // Okay.

    // Save default, for future request to other servers.
    savedOptions.save();

    // Parse OACK args
    parseOptions( rawPacket, sz, 2 );

    if ( option.state == TFTP_ReadingNotAck ) {
      option.state = TFTP_Reading;
      // option.openFile( "rb" );      
      TFTP_Packet* pkt = new TFTP_Packet();
      pkt->sendAck( 0 );
    }

    if ( option.state == TFTP_WritingNotAck ) { 
      option.state = TFTP_Writing;
        // option.openFile( "wb" );
        // pkt.sendData( 1 );

        if ( option.dataFactory != NULL ) {
          option.dataFactory = new DataFactory();
        }

        option.dataFactory->run();

        // Let Data Factory the time to produce ! =)
        sched_yield();

        if ( option.sendFactory != NULL ) {
          option.sendFactory = new TFTP_Sender();
        }

        option.sendFactory->run();

        // Let Send Factory the time to Send ! =)
        // sched_yield(); 

        // Send data #1
        // Start the DataFactory & TFTP_DataSender
        // which will create data packet and send them.
    }

  } else {
 
    option.state = TFTP_Error;
    // Send Error 4.
    SendIllegalOpError( "OACK Rejected. \n" );
  }
}

void opcodeDispatcher( char* rawPacket, int sz )
{
    //printf("\nPDU = {");
    //for( int i = 0; i < sz; i++ )
    //    printf("[%d]%c)", rawPacket[i], rawPacket[i] );

    //printf("};\n");
    //option.showOptions();
    //fflush( stdout );
    short op;
    
    memcpy( (char*)&op, rawPacket, 2 );

    if ( op >= 1 && op <= 6 ) 
    {
      char* rawCopy = new char[ sz + 1 ];
      memcpy( rawCopy, rawPacket, sz );

      // opTable start at 0, not 1
      switch( op ) 
      {
        case 1: RRQ_driver(   rawCopy, sz ); break;
        case 2: WRQ_driver(   rawCopy, sz ); break;
        case 3: DATA_driver(  rawCopy, sz ); break;
        case 4: ACK_driver(   rawCopy, sz ); break;
        case 5: ERROR_driver( rawCopy, sz ); break;
        case 6: OACK_driver(  rawCopy, sz ); break;
      }
      
      //(this->*( opTable[ op-1 ] ))( rawCopy, sz ); 

    } else {

        char buf[80];
        sprintf( buf, "Invalid Opcode: %d\n", op );
        SendIllegalOpError( buf );        
    }    
}

#endif

---