Utility.cpp

Go to the documentation of this file.

//-----------------------------------------------------------------------------------
//
// Pro-Vocation Light Engine (PVLE)
// Copyright (C) 2007-2009  Sukender, KinoX & Buzib
// For more information, contact us : sukender@free.fr
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 3 of the License, or
// (at your option) any later version.
//
// For any use that is not compatible with the terms of the GNU 
// General Public License, please contact the authors for alternative
// licensing options.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
//-----------------------------------------------------------------------------------

#include <PVLE/Network/HTTP/Utility.h>

#ifdef PVLE_URL

#include <PVLE/Util/Log.h>
//#include <curl/curl.h>
#include <boost/algorithm/string/case_conv.hpp>

namespace Net {

void URLOptions::setProxyType(std::string strType) {
    boost::algorithm::to_lower(strType);
    if (strType == "http") proxyType = PROXY_HTTP;
    else if (strType == "socks4") proxyType = PROXY_SOCKS4;
    else if (strType == "socks5") proxyType = PROXY_SOCKS5;
    else if (strType == "socks4a") proxyType = PROXY_SOCKS4A;
    else if (strType == "socks5h" || strType == "socks5hostname") proxyType = PROXY_SOCKS5_HOSTNAME;
    else LOG_WARN << "Could not interpret proxy type: " << strType;
}


CURLHolder::CURLHolder() {
    handle = curl_easy_init();
#ifdef WIN32
    curl_global_init(CURL_GLOBAL_WIN32);
#else
    curl_global_init(CURL_GLOBAL_NOTHING);
#endif
}

CURLHolder::~CURLHolder() {
    curl_easy_cleanup(handle);
}


void applyOptions(const URLOptions & options) {
    CURL * curlHandle = CURLHolder::instance();

    curl_easy_setopt(curlHandle, CURLOPT_FOLLOWLOCATION, 1);
    curl_easy_setopt(curlHandle, CURLOPT_USERAGENT, "PVLE C++ library (uses libcurl)");
    curl_easy_setopt(curlHandle, CURLOPT_FAILONERROR, 1);
#ifdef _DEBUG
    //curl_easy_setopt(curlHandle, CURLOPT_VERBOSE, 1);
#endif

    if (options.usesProxy()) {
        curl_easy_setopt(curlHandle, CURLOPT_PROXYPORT, options.getProxyPort());
        curl_easy_setopt(curlHandle, CURLOPT_PROXY, options.getProxyAddress().c_str());
        curl_easy_setopt(curlHandle, CURLOPT_PROXYTYPE, options.getProxyType());
    } else {
        curl_easy_setopt(curlHandle, CURLOPT_PROXY, "");
    }
    curl_easy_setopt(curlHandle, CURLOPT_TIMEOUT, options.getTimeout());

    //curl_easy_setopt(curlHandle, CURLOPT_NOSIGNAL, 1);
}


std::size_t readURLAsStringCBf(void *ptr, std::size_t size, std::size_t nmemb, void *stream) {
    *reinterpret_cast<std::string *>(stream) += static_cast<const char *>(ptr);
    return nmemb;
}


std::string readURLAsString(const std::string & url, const URLOptions & options) {
    //const bool includeHeader = true;
    CURL * curlHandle = CURLHolder::instance();

    std::string result;
    curl_easy_setopt(curlHandle, CURLOPT_URL, url.c_str());
    //curl_easy_setopt(curlHandle, CURLOPT_WRITEFUNCTION, boost::bind<std::size_t>(&readURLAsStringCB::operator(), &cb, _1, _2, _3, _4));
    curl_easy_setopt(curlHandle, CURLOPT_WRITEFUNCTION, &readURLAsStringCBf);
    curl_easy_setopt(curlHandle, CURLOPT_WRITEDATA, &result);
    //curl_easy_setopt(curlHandle, CURLOPT_HEADER, includeHeader ? 1 : 0);

    applyOptions(options);

    curl_easy_perform(curlHandle);
    //return cb.result;

    long response = 0;
    curl_easy_getinfo(curlHandle, CURLINFO_RESPONSE_CODE, &response);
    return response>=400 ? "" : result;
}


}  // namespace Net

#endif  // PVLE_URL

---