/home/pkhodos/work/trunk/ToolLibs/CurlClient/curlclient.h

Go to the documentation of this file.

#ifndef _CURLCLIENT_H
#define _CURLCLIENT_H

#ifdef _MSC_VER
#       ifdef curlclient_EXPORTS
#               define CURLCLIENTLIB_DECL __declspec(dllexport)
#       else
#               define CURLCLIENTLIB_DECL __declspec(dllimport)
#       endif
#else
#       define CURLCLIENTLIB_DECL
#endif


#include <string>
#include <list>
#include <fstream>
#include <sstream>
#include <vector>
#include <list>

#define DEFAULT_TIMEOUT 60
#define DEFAULT_CONNECT_TIMEOUT 10

/*
  CURL and OpenSSL initializaers are not thread safe. We MUST explicitly call
  initializer during library loading when no more then 1 thread are running
*/

class CURLCLIENTLIB_DECL CurlForm {
  private:
    struct curl_httppost* post;
    struct curl_httppost* last;

    std::list<std::string> bufPart;
    std::list<std::string> bufName;

  public:
    CurlForm();
    ~CurlForm();

    void addContent(const std::string &part, const std::string &name, const std::string &ctype);
    void addFile(const std::string &part, const std::string &name, const std::string &ctype);

    const curl_httppost* getPost(void) const {return post;};
    enum CurlFormType {
      CurlFormTypeContent,
      CurlFormTypeFile
    };

};



/* forward declaration to allow compile w/o libcurl */
typedef void CURL;

class CURLCLIENTLIB_DECL CurlClient {
private:
  /* CURL* local instance */
  CURL * instance;
  void initInstance();
  void setInstance();
//  void performInstance();
  void cleanInstance();

  /* data stored here by callbacks */
  std::string bufferHeader;
  //std::string errorBuffer;

  /* CURL interlan buffer to store error and error string */
  char *errorBuffer;
  int errorCode;

  /* linked list to store headers */
  struct curl_slist* pcsHeaders;
 
  /* input data parameters */
  std::string url;
  int timeOut;
  int connectTimeOut;
  std::string contentType;
  std::list<std::string> lsHeader;
  /* Flag indicates force to use SSLv3 instead of TLSv1 */
  bool useSSLv3;

  /* client certificates and private keys */
  std::string clientCert;
  std::string clientCertType;//Supported formats are "PEM" and "DER"
  std::string clientKey;
  std::string clientKeyType;//Supported formats are "PEM", "DER" and "ENG".

  /* authorization parameters */
  std::string username;
  std::string password;

  /* output data parameter */
  int httpStatus;
  void getInfo();

  static size_t callbackWriteFunction(char *data, size_t size, size_t nmemb, std::string *stream);
  static size_t callbackReadFunction(char *data, size_t size, size_t nmemb, std::stringstream *stream);


  void clean();

public:

  CurlClient();

  virtual ~CurlClient();

  void setUrl(const std::string &sUrl);

  void setTimeOut(int iTimeOut);

  void setConnectTimeOut(int iConnectTimeOut);

  void setContentType(const std::string &sContentType);

  void setHeader(const std::string &sHeader);

  void setUseSSLv3();

  void setClientCert(const std::string &cert);
  
  void setClientCertType(const std::string &type);


  void setClientKey(const std::string &key);

  void setClientKeyType(const std::string &type);

  void setHttpAuth(const std::string &s_username, const std::string &s_password);

  void clearHeaders();

  /* public result getters */
  std::string getResultHeader() const;

  const std::string getErrorBuffer() const;

  int getErrorCode() const;

  int getHttpStatus() const;

  const std::string callGet();

  const std::string callPost(const std::string &postData);

  void uploadToFTP(std::stringstream &in);

  const std::string callHttpPost(const CurlForm &data);

  const std::string downloadFromFTP(void);

  static std::string escapeUrl(const std::string &url);
  
  static std::string unescapeUrl(const std::string &url);

  const std::string getFullErrorMessage(void) const;

};

#endif