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 */