Search  

---

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

Go to the documentation of this file.

/*

NOTICE:  Block Number ARE ALWAYS FROM 1...WindowSize

Block Number 0 means we STARTED !

Someone must call this guy... Client/Server UDP... Done!

*/

#ifndef __J2K__TFTP_Parser_CPP__
#define __J2K__TFTP_Parser_CPP__

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

TFTP_Parser::TFTP_Parser()
{
  OACK_arg = FALSE;
  array_sz = 0;
  expectedBlockSize = option.getBlockSizeValue();

   blocksizeDefault = 1;
  windowsizeDefault = 1;
     timeoutDefault = 1;

  opTable = new TFTP_ParserCallback[6];

  opTable[0] = &TFTP_Parser::RRQ_driver;
  opTable[1] = &TFTP_Parser::WRQ_driver;
  opTable[2] = &TFTP_Parser::DATA_driver;
  opTable[3] = &TFTP_Parser::ACK_driver;
  opTable[4] = &TFTP_Parser::ERROR_driver;
  opTable[5] = &TFTP_Parser::OACK_driver;
}
 
TFTP_Parser::~TFTP_Parser()
{
  if ( opTable != NULL ) delete[] opTable;
}

void TFTP_Parser::parseOptions( char* str, size_t length, size_t offset )
{
  // CORRUPTION OF THE STACK, PATCH THE STRING BY BACKUPing!
  char* tmp = new char[ length ];
  memcpy( tmp, str, length );
  delete [] str;

  array_sz = 0;
  
   blocksizeDefault = 1;
  windowsizeDefault = 1;
     timeoutDefault = 1;

  getParams( tmp, offset );

  printf( "%d", array_sz );
  fflush( stdout );
  for( int i = 0; i < array_sz; i++ ) {
    safePrint( array[i] );
    printf( "\t%d-%d\n", i, array_sz );
    fflush( stdout );
    i += select( i );
  }

  // If not defined, set TFTP default options.
  if (  blocksizeDefault ) {
    option.setBlockSize( atol( DEFAULT_BLOCKSIZE ) );
  }

  if ( windowsizeDefault ) {
    option.setWindowSize( atol( DEFAULT_WINDOWSIZE ) );
  }

  if (    timeoutDefault ) {
    option.setTimeOut( atol( DEFAULT_TIMEOUT ) );
  }

  expectedBlockSize = option.getBlockSizeValue();

  str = new char[ length ];
  memcpy( str, tmp, length );
}

char* TFTP_Parser::scanParam( char* str, size_t* len )
{
  size_t l = strlen( str );
  char*  t = str + l + 1;

  *len = l;
  return t;
}

void TFTP_Parser::getParams( char* str, size_t offset )
{
   size_t l = 1;
   char*  f = str + offset;
   char*  p = f;
   char*  q = p;

   int k = 0;
   while( l != 0 )
   {
     // q = scanParam( p, &l );
     l = strlen( p );
     q = p + l + 1;
     array[ k ] = new char[ l ];
     memcpy( array[ k ], p, l );
     k++;
     p = q;
   }
   array_sz = k - 1;
}    

// Return 0, if the param has no associated value
// Return 1, if the next param is taken as a value
int TFTP_Parser::select( int i ) 
{
     char* s2  = array[i];

     if ( s2 != NULL ) s2 = strlwr( s2 );

     char* opt = NULL;

     if ( (i+1) <= array_sz ) {
       opt = strlwr( array[i+1] );
     }

     if (  !strcmp( s2, "blksize" ) || !strcmp( s2, "blocksize" ) )
     {
       long bsize = atol( opt );
       printf("BlockSize[%d]", bsize );
       fflush( stdout );
//       OACK_arg = TRUE;
       option.requestedBlockSize( bsize );  
       blocksizeDefault = 0;
       return 1;   
     } else
     if ( !strcmp( s2, "timeout" ) )
     {
       size_t t = atol( opt );
       printf("Timeout[%d]", t ); 
       fflush( stdout );
//       OACK_arg = TRUE;
       option.requestedTimeOut( t );
       timeoutDefault = 0;
       return 1;   
     } else
     if ( !strcmp( s2, "tsize" ) )
     {
       size_t sz = atol( opt );
       printf("FileSize[%d]", sz ); 
       fflush( stdout );
//       OACK_arg = TRUE;
       option.requestedFileSize( sz );
       return 1;   
     } else 
     if (  !strcmp( s2, "wsize" ) || !strcmp( s2, "windowsize" ) \
        || !strcmp( s2, "window" ) )
     {
       long wsize = atol( opt );
       printf("WindowSize[%d]", wsize ); 
       fflush( stdout );
//       OACK_arg = TRUE;
       option.requestedWindowSize( wsize );     
       windowsizeDefault = 0;
       return 1;   
     }
    
     if ( i == 0 ) {
       printf("Filename[%s]\n", array[i] ); 
       fflush( stdout );
       option.setFileName( array[i] );
       return 0;   
     }

     if ( i == 1 ) {
       printf("Mode[%s]\n", array[i] ); 
       fflush( stdout );
       option.requestedMode( array[i] );
       return 0;   
     }
 
  return 0;   
}

#endif

---