webclient.h

Go to the documentation of this file.

 
#ifndef WEBCLIENT_H_
#define WEBCLIENT_H_
 
#include <vector>
#include <string>
#include <functional>
 
#define USE_OPENSSL 1
#define OPENSSL_NO_SHA1 1
 
class WebContext;
class WebClient;
 
class WebClientHandler {
public:
    virtual ~WebClientHandler() = default;
    virtual void checkHeader(WebClient *wc)
    {
        (void)wc;
    }
    virtual void checkContent(char const *ptr, size_t len)
    {
        (void)ptr;
        (void)len;
    }
};
 
class WebProxy {
public:
    std::string server;
    bool empty() const
    {
        return server.empty();
    }
};
 
struct ResponseHeader;
 
class WebClient {
public:
    class ContentType {
    public:
        static constexpr const char *APPLICATION_OCTET_STREAM = "application/octet-stream";
        static constexpr const char *APPLICATION_X_WWW_FORM_URLENCODED = "application/x-www-form-urlencoded";
        static constexpr const char *MULTIPART_FORM_DATA = "multipart/form-data";
    };
    class URL {
        friend class WebClient;
    private:
        struct Data {
            std::string full_request;
            std::string scheme;
            std::string host;
            int port = 0;
            std::string path;
        } data;
    public:
        URL() = default;
        URL(std::string const &addr);
        std::string const &scheme() const { return data.scheme; }
        std::string const &host() const { return data.host; }
        int port() const { return data.port; }
        std::string const &path() const { return data.path; }
        bool isssl() const;
    };
    enum HttpVersion {
        HTTP_1_0,
        HTTP_1_1,
    };
    struct Authorization {
        enum Type {
            None,
            Basic,
        } type = None;
        std::string uid;
        std::string pwd;
    };
    class Request {
        friend class ::WebClient;
    private:
        URL url;
        Authorization auth;
        std::vector<std::string> headers;
    public:
        Request() = default;
        Request(std::string const &loc, std::vector<std::string> const &headers = {})
            : url(loc)
            , headers(headers)
        {
        }
        void set_location(std::string const &loc)
        {
            url = URL(loc);
        }
        void set_authorization(Authorization::Type type, std::string const &uid, std::string const &pwd)
        {
            auth.type = type;
            auth.uid = uid;
            auth.pwd = pwd;
        }
        void set_basic_authorization(std::string const &uid, std::string const &pwd)
        {
            set_authorization(WebClient::Authorization::Basic, uid, pwd);
        }
        void add_header(std::string const &s)
        {
            headers.push_back(s);
        }
    };
    class Error {
    private:
        std::string msg_;
    public:
        Error() = default;
        Error(std::string const &message)
            : msg_(message)
        {
        }
        virtual ~Error() = default;
        std::string what() const
        {
            return msg_;
        }
    };
    struct Response {
        int code = 0;
        struct Version {
            unsigned int hi = 0;
            unsigned int lo = 0;
        } version;
        std::vector<std::string> header;
        std::vector<char> content;
    };
    struct Post {
        std::string content_type;
        std::string boundary;
        std::vector<char> data;
    };
    struct ContentDisposition {
        std::string type;
        std::string name;
        std::string filename;
    };
    struct Part {
        char const *data = nullptr;
        size_t size = 0;
        std::string content_type;
        ContentDisposition content_disposition;
        std::string content_transfer_encoding;
        Part() = default;
        Part(char const *data, size_t size, std::string const &content_type = {})
            : data(data)
            , size(size)
            , content_type(content_type)
        {
        }
        void set_content_disposition(ContentDisposition const &cd)
        {
            content_disposition = cd;
        }
    };
    struct RequestOption {
        WebClientHandler *handler = nullptr;
        bool keep_alive = true;
    };
private:
    struct Private;
    Private *m;
    void clear_error();
    static int get_port(URL const *url, char const *scheme, char const *protocol);
    void set_default_header(const Request &url, Post const *post, const RequestOption &opt);
    std::string make_http_request(const Request &url, Post const *post, const WebProxy *proxy, bool https);
    void parse_http_header(char const *begin, char const *end, std::vector<std::string> *header);
    void parse_http_header(char const *begin, char const *end, Response *out);
    bool http_get(const Request &request_req, Post const *post, RequestOption const &opt, ResponseHeader *rh, std::vector<char> *out);
    bool https_get(const Request &request_url, Post const *post, RequestOption const &opt, ResponseHeader *rh, std::vector<char> *out);
    bool get(const Request &req, Post const *post, Response *out, WebClientHandler *handler);
    static void parse_header(std::vector<std::string> const *header, WebClient::Response *res);
    static std::string header_value(std::vector<std::string> const *header, std::string const &name);
    void append(char const *ptr, size_t len, std::vector<char> *out, WebClientHandler *handler);
    void on_end_header(const std::vector<char> *vec, WebClientHandler *handler);
    void receive_(const RequestOption &opt, std::function<int (char *, int)> const &, ResponseHeader *rh, std::vector<char> *out);
    void output_debug_string(char const *str);
    void output_debug_strings(const std::vector<std::string> &vec);
    static void cleanup();
    void reset();
public:
    static void initialize();
    WebClient(WebContext *webcx);
    ~WebClient();
    WebClient(WebClient const &) = delete;
    void operator = (WebClient const &) = delete;
 
    void set_http_version(HttpVersion httpver);
 
    Error const &error() const;
    int get(const Request &req, WebClientHandler *handler = nullptr);
    int post(const Request &req, Post const *post, WebClientHandler *handler = nullptr);
    void close();
    void add_header(std::string const &text);
    Response const &response() const;
    std::string header_value(std::string const &name) const;
    std::string content_type() const;
    size_t content_length() const;
    char const *content_data() const;
 
    static void make_application_www_form_urlencoded(char const *begin, char const *end, WebClient::Post *out);
    static void make_multipart_form_data(const std::vector<Part> &parts, WebClient::Post *out, std::string const &boundary);
    static void make_multipart_form_data(char const *data, size_t size, WebClient::Post *out, std::string const &boundary);
 
    static std::string quick_get(const std::string &url);
    static std::string checkip();
};
 
class WebContext {
    friend class WebClient;
public:
private:
    struct Private;
    Private *m;
public:
    WebContext(WebClient::HttpVersion httpver);
    ~WebContext();
    WebContext(WebContext const &r) = delete;
    void operator = (WebContext const &r) = delete;
 
    void set_http_version(WebClient::HttpVersion httpver);
    void set_keep_alive_enabled(bool f);
 
    void set_http_proxy(std::string const &proxy);
    void set_https_proxy(std::string const &proxy);
    WebProxy const *http_proxy() const;
    WebProxy const *https_proxy() const;
 
    bool load_cacert(char const *path);
 
    void notify_broken_pipe();
};
 
#endif