Search  

---

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

Go to the documentation of this file.

#ifndef __J2K__File_Factory_CPP__
#define __J2K__File_Factory_CPP__

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

FileFactory::FileFactory()
 : Basic_PThread(),
  expectedBlockNo( 1 ), initialized( FALSE ),
  sz( 0 ), rc( 0 ), length( 0 ), block( 0 ), empty( TRUE )
{
  blocksize = option.getBlockSizeValue();
}

void FileFactory::init()
{
  if ( initialized == TRUE ) return;
  initialized = TRUE;
  expectedBlockNo = 1;
  sz     = 0;
  rc     = 0;
  length = 0;
  block  = 0;
  empty  = TRUE;

  printf( "FileFactory::init\n");
  fflush( stdout );

  option.openFile( "wb" );
  option.rwlock2_file.read();
  option.myFile->setBegin();
  option.rwlock2_file.releaseRead();
}

void FileFactory::checkIfEmpty()
{
  while ( empty == TRUE ) 
  { 
    option.rwlock5_data_queue.read();
    empty = option.data_queue->isEmpty();
    option.rwlock5_data_queue.releaseRead();
   
    if ( empty == TRUE )
    {
        printf("Empty Packet\n");
        sched_yield();  // it's empty, yield.
    }
  }
}

void FileFactory::emptyPacket()
{
    option.rwlock5_data_queue.read();

#ifdef DEBUG
  printf( "Packet null in list, size=%d\n", option.data_queue->size() );
  option.data_queue->display();
#endif

  option.rwlock5_data_queue.releaseRead();
  fflush( stdout );
  sched_yield();
}

void FileFactory::directWrite( TFTP_Packet* pkt )
{
  init();
  packet = pkt;
  validPacket();
}

void FileFactory::validPacket()
{
  block = packet->getBlockNo();

  if ( block == expectedBlockNo ) 
  {
    printf( "\nThis packet was in order: %d / %d\n", block, expectedBlockNo );
    fflush( stdout );

    option.rwlock2_file.read(); // Read filedes to write on file
    length = packet->getpktLength() - 4;
    if ( length > 0 ) 
    { 
      printf( "\nWriting to file: #%d, length %d\n", block, length );
      fflush( stdout );

#ifdef DEBUG
      option.showChar( "Data",  packet->getData(), length );
#endif

      sz = option.myFile->putData( packet->getData(), length, option.getBlockSizeValue() );

    } else if (length == 0)
    {
        sz = 0;
        printf( "\nNothing to write to file: #%d, length %d\n", block, length );
        fflush( stdout );
    } else 
    {   
      printf( "\nDid NOT write to file: #%d, length %d\n", block, length );
      fflush( stdout );
    }

    option.rwlock2_file.releaseRead();
    expectedBlockNo++; 

    // Last Packet;
    if ( sz < option.getBlockSizeValue() ) 
    {
      printf( "\nThis shall be the last packet of size %d out of %d #%d\n", \
              sz, option.getBlockSizeValue(), block );

      if ( option.myFile != NULL ) {
        delete option.myFile;
        option.myFile = NULL;
      }
      fflush( stdout );

      if ( option.isReading() ) {
        option.state = TFTP_EndofRead;
      } else if ( option.isWriting() ) {
        option.state = TFTP_EndofWrite;
      } else {
        option.state = TFTP_Finished;
      }
      return;
    }
    else
    {
        printf("This aint the last packet\n");
        printf("The size is %d, Blocksize is %d\n",sz,option.getBlockSizeValue());
    }
    
    // It's 1 NOT 0, since 0 means we started!
    // Loop the Window size.
    if ( (expectedBlockNo-1) > option.getWindowSizeValue() ) 
    {
        //No need for this
     // expectedBlockNo = 1;   
    }

    } else   { // Trash Packet.
     printf( "\nTrashing Data Packet# %d / %d\n", block, expectedBlockNo );
     fflush( stdout );
     packet = NULL;
     delete packet;
  }
}

void FileFactory::run() 
{
  init(); 
  
  while ( option.isConnected() || !option.data_queue->isEmpty() )
  {
    checkIfEmpty();

    // Read data in Write mode (modifying structure) 
    option.rwlock5_data_queue.write();
    packet = option.data_queue->dequeue();        
    option.rwlock5_data_queue.releaseWrite();
    
    if ( packet != NULL ) 
    {
      validPacket();

    } else {  // List is empty ?!

      if ( option.isEnding() ) return;

      emptyPacket();
      //      sched_yield();
    }

  } // end while()
}

#endif

---