---

sue_sess.hpp

Go to the documentation of this file.

#ifndef SENTRY_SUE_SESS_HPP
#define SENTRY_SUE_SESS_HPP


#ifndef SENTRY_SUE_SEL_HPP
#include "sue_sel.hpp"
#endif


class SUEBuffer {
    char *data;
    int datalen;
    int maxlen;
public:
    SUEBuffer();
    ~SUEBuffer();
    

    void AddData(const char *buf, int size);
    

    int GetData(char *buf, int bufsize);


    void DropData(int len);

    void EraseData(int index, int len);

    void DropAll() { datalen = 0; } 
    
    void AddChar(char c);


    void AddString(const char *str);


    int ReadLine(char *buf, int bufsize);

    bool ReadLine(SUEBuffer &buf);
    

    int FindLineMarker(const char *marker) const;
    int ReadUntilLineMarker(const char *marker, char *buf, int bufsize);
    bool ReadUntilLineMarker(const char *marker, SUEBuffer &dest);
    

    bool ContainsExactText(const char *str) const;
    

    const char *GetBuffer() const { return data; }

    int Length() const { return datalen; }

    char& operator[](int i) const { return *(data + i); }

private:
    void ProvideMaxLen(int n);
};



class SUEGenericDuplexSession : protected SUEFdHandler, 
                                private SUETimeoutHandler 
{

    long timeout_sec; 
    long timeout_usec; 

    bool scheduleddown; 

protected: 

    SUEEventSelector *the_selector;

    SUEBuffer inputbuffer;
    SUEBuffer outputbuffer;


    bool inputresetstimeout;
    bool outputresetstimeout;


    virtual void FdHandle(bool a_r, bool a_w, bool a_ex);


    virtual void TimeoutHandle();
  
      
    virtual bool WantRead() const { return true; }
    virtual bool WantWrite() const { return outputbuffer.Length() > 0; }
    virtual bool WantExcept() const { return false; }

public:

    SUEGenericDuplexSession(int a_timeout = 0,
                            const char *a_greeting = 0);    
    virtual ~SUEGenericDuplexSession();


    void SetTimeout(int sec, int usec = 0);


    void Startup(SUEEventSelector *a_selector, 
                 int a_fd, const char *a_greeting = 0);


    void Shutdown();


    void GracefulShutdown();


    virtual void HandleNewInput() = 0; 


    virtual void ShutdownHook() {} 


    virtual void HandleSessionTimeout() { Shutdown(); }


    virtual void HandleReadError() { Shutdown(); }


    virtual void HandleWriteError() { Shutdown(); }


    virtual void HandleRemoteClosing() { Shutdown(); }

private:
    void ResetTimeout();
};


#endif //sentry

---