tftp_server.h

/*NB_REVISION*/
 
/*NB_COPYRIGHT*/


 
#ifndef _NB_TFTP_SERVER
#define _NB_TFTP_SERVER 
 
#include <config_server.h>
#include <servlets.h>
#include <nettimer.h>
#include <nbstring.h>
 

 
 
class TFTP_active_Records;
class TFTP_Server_source_File;

class TFTP_Server: public servlet, public TimeOutElement
{
    int udp_fd;
    uint16_t m_port;
    TFTP_active_Records * m_pRecords;
    TFTP_Server_source_File * m_pSources;
 
    protected:
        //Servlet virtual functions
        virtual int AddToSelectSet(fd_set &rd_set, fd_set &wr_set, fd_set &er_set);
        virtual void ProcessSelectResult(fd_set &rd_set, fd_set &wr_set, fd_set &er_set);
        //Timeout element virtual function
        virtual void TimeElementEvent();
 
    public:
        TFTP_Server( uint16_t port = 69 );

        void Add(TFTP_Server_source_File & sf);
 
    friend class TFTP_active_Records;
};
 

class TFTP_Server_source_File
{
    TFTP_Server_source_File * pNext_file;
    NBString m_fname;
    const uint8_t *pData;
    uint32_t len;
 
    public:
        virtual int GetNextChunk(uint32_t offset, uint32_t max_len, puint8_t destbuffer);
        //Returns the chunk size, any thing less than requested means end of file.
        TFTP_Server_source_File(TFTP_Server & server,const char * fname, const uint8_t * buf, uint32_t len);
 
    friend class TFTP_Server;
    friend class TFTP_active_Records;
};
 
#endif