eiennet_curl.hpp

浏览该文件的文档.

#ifndef __CURL_HPP__
#define __CURL_HPP__

// curl.h的一些类型
typedef void CURL;
struct curl_slist;
struct curl_httppost;

namespace eiennet
{

class EIENNET_DLL CUrlError : public winux::Error
{
public:
    CUrlError( int errNo, char const * err ) : winux::Error( errNo, err ) { }
};

class EIENNET_DLL CUrlLib
{
public:
    static char const * GetVersion();

    CUrlLib();
    ~CUrlLib();
    DISABLE_OBJECT_COPY(CUrlLib)
};

class EIENNET_DLL SList
{
    struct curl_slist * _slist;
public:
    SList();
    virtual ~SList();
    void free();
    void append( char const * s );

    operator struct curl_slist * () const { return _slist; }
    DISABLE_OBJECT_COPY(SList)
};

class EIENNET_DLL PostMultipart
{
    struct curl_httppost * _first;
    struct curl_httppost * _last;
public:
    PostMultipart();
    virtual ~PostMultipart();
    void free();
    void addVar( winux::String const & name, winux::String const & value );
    void addFile( winux::String const & name, winux::String const & filename, winux::String const & showfilename = "" );
    void addFile( winux::String const & name, char const * data, long size, winux::String const & showfilename );

    operator struct curl_httppost * () const { return _first; }
    DISABLE_OBJECT_COPY(PostMultipart)
};

class EIENNET_DLL CUrl
{
private:
    CURL * _curl;
    int _errNo;
    winux::AnsiString _errBuf;
protected:
    // '写'回调函数，回调函数将转到成员虚函数进行响应
    static size_t WriteCallback( char * buf, size_t itemSize, size_t count, void * data );
    // 当发生'写'动作
    virtual size_t OnWrite( char * buf, size_t itemSize, size_t count );

    // '读'回调函数
    static size_t ReadCallback( char * buf, size_t itemSize, size_t count, void * data );
    // 当发生'读'动作
    virtual size_t OnRead( char * buf, size_t itemSize, size_t count );

    // '头部'数据回调函数
    static size_t HeaderCallback( char * buf, size_t itemSize, size_t count, void * data );
    // 当'头部'数据到
    virtual size_t OnHeader( char * buf, size_t itemSize, size_t count );

    // '进度'回调函数
    static int ProgressCallback( void * data, double dltotal, double dlnow, double ultotal, double ulnow );
    // 当'下载进度'操作
    virtual int OnDownloadProgress( double dltotal, double dlnow );
    // 当'上传进度'操作
    virtual int OnUploadProgress( double ultotal, double ulnow );
public:
    // 错误码转成文本信息
    //static char const * getErrNoStr( int errNo );

    CUrl( bool isInit = true );
    CUrl( CUrl const & other );
    virtual ~CUrl();

    CUrl & operator = ( CUrl const & other );

    void init();
    void cleanup();

    bool perform();

    void reset();

    // set options ------------------------------------------------
    void setHttpGet( bool b = true );

    void setHttpPost( bool b = true );

    void setUrl( winux::String const & url );

    void setTimeout( winux::ulong timeout );

    void setErrorBuffer( char * errBuf );

    void setNoProgress( bool b );

    void setUsername( winux::String const & username );

    void setPassword( winux::String const & password );

    void setPostFields( char const * data );

    void setPostFieldSize( long size );

    void setPostMultipart( PostMultipart const & data );

    void setHttpHeader( SList const & headers );

    void setVerbose( bool b );

    void setCookieJar( winux::String const & filename );

    void setSslVerifyPeer( bool b );

    void setSslVerifyHost( bool b );

    // set handlers -----------------------------------------------
    typedef size_t (* WriteFunction)( char * buf, size_t itemSize, size_t count, void * data );
    void setWriteHandler( WriteFunction handler );
    void setWriteHandlerData( void * data );

    typedef size_t (* ReadFunction)( char * buf, size_t itemSize, size_t count, void * data );
    void setReadHandler( ReadFunction handler );
    void setReadHandlerData( void * data );

    typedef size_t (* HeaderFunction)( char * buf, size_t itemSize, size_t count, void * data );
    void setHeaderHandler( HeaderFunction handler );
    void setHeaderHandlerData( void * data );

    typedef int (* ProgressFunction)( void * data, double dltotal, double dlnow, double ultotal, double ulnow );
    void setProgressHandler( ProgressFunction handler );
    void setProgressHandlerData( void * data );

    // attributes -------------------------------------------------
    operator CURL * () const { return _curl; }

    int errNo() const { return _errNo; }

    char const * errNoStr() const;

    char const * error() const;

};

class EIENNET_DLL HttpCUrl : public CUrl
{
protected:
    // 当发生'写'动作
    virtual size_t OnWrite( char * buf, size_t itemSize, size_t count );

private:
    winux::GrowBuffer _response;
public:
    HttpCUrl();
    virtual ~HttpCUrl();

    winux::Buffer const & getResponse() const;

    winux::ulong getResponseSize() const;

    char const * getResponseStr( winux::ulong * size = NULL ) const;

    bool get( winux::String const & url, http::Header const & headers = http::Header() );
    bool post( winux::String const & url, winux::Mixed const & postVars = winux::Mixed(), http::Header const & headers = http::Header() );
    bool post( winux::String const & url, PostMultipart const & multipart, http::Header const & headers = http::Header() );

};


} // namespace eiennet

#endif // __CURL_HPP__