httpclienttask.h

// Copyright (c) 2018 IIS (The Internet Foundation in Sweden)
// Written by Göran Andersson <initgoran@gmail.com>
 
#pragma once
 
#include "httptask.h"
#include "cookiemanager.h"
#include "httpclientconnection.h"
 
class HttpClientTask : public HttpTask {
public:
    HttpClientTask(const std::string &name, const HttpHost &httpserver);
 
    virtual void newRequest(HttpClientConnection *) {
    }
 
    virtual bool headerComplete(HttpClientConnection *);
 
    virtual bool websocketUpgrade(HttpClientConnection *);
 
    virtual bool requestComplete(HttpClientConnection *) = 0;
 
    virtual void payload(HttpClientConnection *, char *, size_t ) {
    }
 
    virtual size_t doPost(HttpClientConnection *conn, size_t len);
 
    std::string serverHost() const {
        return peer.hostname;
    }
 
    virtual std::string httpHeaderLines(const std::string &uri) const {
        if (!peer.cmgr)
            return user_agent_header;
        return peer.cmgr->httpHeaderLine(peer.hostname, uri) +
            user_agent_header;
    }
 
    void setCookie(const std::string &header_line, const std::string &uri) {
        if (peer.cmgr)
            peer.cmgr->setCookie(header_line, peer.hostname, uri);
    }
    virtual std::string cacheLabel() {
        return peer.hostname + std::to_string(peer.port);
    }
    void setUserAgentString(const std::string &s) {
        if (s.find_first_of("\r\n") == std::string::npos)
            user_agent_header = "User-Agent: " + s + "\r\n";
    }
 
    static bool setLocalAddress(const std::string &addr, uint16_t iptype);
 
    static const std::string &getLocalAddress() {
        return local_address;
    }
protected:
    bool createNewConnection();
 
    std::string proxyHost() const {
        return peer.proxyHost;
    }
    uint32_t proxyPort() const {
        return peer.proxyPort;
    }
    CookieManager *cookieMgr() const {
        return peer.cmgr;
    }
    void setServer(const std::string hostname, uint16_t port_number = 80) {
        peer.hostname = hostname;
        peer.port = port_number;
    }
private:
    HttpHost peer;
    std::string user_agent_header;
    std::string real_server;
    static std::string local_address;
    static struct addrinfo *local_ip;
};