CurlSock.h

Go to the documentation of this file.

 
#ifndef HAVE_CURLSOCK_H
#define HAVE_CURLSOCK_H
 
#include "Csocket.h"
#include <curl/curl.h>
 
#ifndef _NO_CSOCKET_NS
namespace Csocket
{
#endif /* _NO_CSOCKET_NS */
 
class CS_EXPORT CCurlSock : public CSMonitorFD
{
public:
        CCurlSock();
        virtual ~CCurlSock();
 
        virtual bool GatherFDsForSelect( std::map< int, int16_t > & miiReadyFds, long & iTimeoutMS );
        virtual bool CheckFDs( const std::map< int, int16_t > & miiReadyFds ) { return( m_bEnabled ); }
 
        CURL * Retr( const CS_STRING & sURL, const CS_STRING & sReferrer = "" );
 
protected:
        virtual void OnCURLComplete( CURL * pCURL ) = 0;
 
        virtual size_t OnHeader( CURL * pCURL, const char * pData, size_t uBytes ) { return( uBytes ); }
        virtual size_t OnBody( CURL * pCURL, const char * pData, size_t uBytes ) { return( uBytes ); }
 
private:
        static size_t WriteData( void * pData, size_t uSize, size_t uNemb, void * pStream );
        static size_t WriteHeader( void * pData, size_t uSize, size_t uNemb, void * pStream );
        static int SetupSock( CURL * pCurlHandle, curl_socket_t iFD, int iWhat, void * pcbPtr, void * pSockPtr );
        static int SetupTimer( CURLM * pMulti, long iTimeout_ms, void * pcdPtr );
 
        void SetTimeoutMS( long iTimeoutMS )
        {
                m_iTimeout_ms = iTimeoutMS;
        }
 
        long            m_iTimeout_ms;  
        CURLM *         m_pMultiHandle; 
        std::map< CURL *, bool > m_pcbCurlHandles; 
};
 
#ifndef _NO_CSOCKET_NS
};
#endif /* _NO_CSOCKET_NS */
 
#endif /* HAVE_CURLSOCK_H */