servlets.h

/*NB_REVISION*/
 
/*NB_COPYRIGHT*/
 
#ifndef _NB_SERVLET_H
#define _NB_SERVLET_H
 
#include <iosys.h>
#include <nbrtos.h>
#include <webclient/http_funcs.h>
#include <iosys.h>
#include <nbstring.h>
#include <dns.h>
#include <nettimer.h>
 
 
 
 
class servlet_list;
 
class servlet
{
protected:
    servlet *pNext;
    servlet_list *pOwner;
 
   public:
    servlet(servlet_list *whereToAdd);
    servlet();
    ~servlet();
 
    // Also returns the number of seconds needed for timeout/recycle...
    virtual int AddToSelectSet(fd_set &rd_set, fd_set &wr_set, fd_set &er_set) = 0;
    virtual void ProcessSelectResult(fd_set &rd_set, fd_set &wr_set, fd_set &er_set) = 0;
    friend class servlet_list;
};
 
class servlet_list
{
    servlet *pHead;
    OS_CRIT ListCritical;
    OS_SEM m_Sem;
 
   public:
    servlet_list();
    ~servlet_list();
    void add(servlet *ps);
    void remove(servlet *ps);
    void run_once_through_select_loop(int max_timeout_ticks);
};
 
 
typedef enum {
     eReadingHeader,
     eChunkSize,
     eBody
}eWebResponse_t;
 
typedef enum {
    eFirstValidInterface,
    eTransactionComplete,
    eTransactionDnsFail,
    eTransactionFail,
    eTransactionTimeOut,
    eReconnectTime,
    eReconnectNow,
} eWebClientAction_t;
 
typedef enum {
    eIdle,
    eWaitingIntf,
    eDoingDns,
    eSendingRequest,
    eSendingPayload,
    eWaitingForResponse,
    eSleeping,
    eDisconnected,
} eWebClientState_t;
 
const int HEADER_BUFFER_SIZE=256;
 
class WebClientServlet: public servlet, public TimeOutElement
{
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();
 
 int m_fd;         //fd to hold DNS Requests and tcp send/rx
 eWebClientState_t  m_cur_state; //State variable
 eWebResponse_t     m_resp_state;
 ParsedURI m_cur_request; //Current URI
 buffer_object * pDestinationBuffer;
 bool m_bChunk;
 NBString PostPayload;
 NBString AdditionalHeaders;
 NBString PostType;
 
 
 char HeaderBuffer[HEADER_BUFFER_SIZE];
 int m_HeaderPos;
 int m_rem_content;
 
 uint32_t m_ResultCode;
 
 TickTimeout m_waketime;
 TickTimeout m_responsetimeout;
 
 
 
 servlet_list * m_pWhoToReJoin;
 
 
//Fucntion to handle data to read
//returns true if the function is complete.
virtual bool ResponseReader(int fd);
 
virtual void StartQuery();
 
 
//Called when things happen 
virtual void ActionComplete(eWebClientAction_t action);
 
protected:
    const char* ActionText(eWebClientAction_t action);
 
 
 
void DisconnectTil(TickTimeout & tt); //Maximum interval ~ 3 yrs with default ticks.
void ReConnectNow();
 
void CoreStart(const char* pUrl,buffer_object * result_buffer, uint32_t timeout); 
 
public:
 
// If List_To_Join is null joins config server
WebClientServlet(servlet_list * List_To_Join=0);
 
uint32_t GetResultCode(){return m_ResultCode; };
 
void    NewGet(const char* pUrl,buffer_object * result_buffer, uint32_t timeout=0,const char* additional_header=0);
void    NewPost(const char * pUrl,const char* pType, NBString & payload,buffer_object * result_buffer, uint32_t timeout=0,const char* additional_header=0);
void    NewPost(const char * pUrl,const char* pType, const char* payload,buffer_object * result_buffer, uint32_t timeout=0,const char* additional_header=0);
};
 
 
 
#endif