Curl.cpp

#include "Global.h"
#include "Curl.h"

#include "curl/curl.h"
#include "Util.h"

string Curl::longpoll_url;
bool Curl::longpoll_active = false;

void Curl::GlobalInit()
{
	curl_global_init(CURL_GLOBAL_ALL);
}

void Curl::Init()
{
	curl = curl_easy_init();
	if (curl == NULL)
	{
		throw string("libcurl initialization failure");
	}
}

void Curl::Quit()
{
	if (curl != NULL)
	{
		curl_easy_cleanup(curl);
	}
	curl_global_cleanup();
}

size_t ResponseCallback(void *ptr, size_t size, size_t nmemb, void *data)
{
	size_t bytes = size * nmemb;
	string* getworksentdata = (string*)data;
	try
	{
		std::string sentdata((char *)ptr, bytes);
		getworksentdata->append(sentdata);
	}
	catch(std::exception s)
	{
		cout << "(1) Error: " << s.what() << endl;
	}
	return bytes;
}

size_t HeaderCallback( void *ptr, size_t size, size_t nmemb, void *userdata)
{
	size_t bytes = size * nmemb;
	std::string hdr((char *)ptr, bytes);
	if (!Curl::longpoll_active && hdr.length() >= 0x10 && hdr.substr(0,0xF) == "X-Long-Polling:")
	{
		Curl::longpoll_url = hdr.substr(0x10);
		Curl::longpoll_url = Curl::longpoll_url.substr(0, Curl::longpoll_url.length()-2);
		//cout << "Longpoll url -->" << Curl::longpoll_url << "<-- " << endl;
		Curl::longpoll_active = true;
	}
	return bytes;
}

string Curl::GetWork_LP(string path, uint timeout)
{
	return Execute(GETWORK_LP, "", path, timeout);
}

string Curl::GetWork(string path, uint timeout)
{
	return Execute(GETWORK, "", path, timeout);
}

string Curl::TestWork(string work)
{
	return Execute(TESTWORK, work, "", 1);
}

string Curl::Execute(Curl::EXEC_TYPE type, string work, string path, uint timeout)
{
	string responsedata;

	curl_easy_setopt(curl, CURLOPT_URL, ("http://" + host + path).c_str());
	curl_easy_setopt(curl, CURLOPT_USERPWD, (username + ":" + password).c_str());
	curl_easy_setopt(curl, CURLOPT_PORT, port);

	curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, ResponseCallback);
	curl_easy_setopt(curl, CURLOPT_WRITEDATA, &responsedata);

	curl_easy_setopt(curl, CURLOPT_HEADERFUNCTION, HeaderCallback);

	if (proxy != "")
		curl_easy_setopt(curl, CURLOPT_PROXY, proxy.c_str());

	curl_slist* headerlist = NULL;
	headerlist = curl_slist_append(headerlist, "Content-Type: application/json");
	headerlist = curl_slist_append(headerlist, "Accept: application/json");
	headerlist = curl_slist_append(headerlist, "User-Agent: reaper/" REAPER_VERSION);

	curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headerlist);

	curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, timeout);
	
	if (type == GETWORK_LP) 
	{
		curl_easy_setopt(curl, CURLOPT_POSTFIELDS, NULL);
		curl_easy_setopt(curl, CURLOPT_POST, 0);
	}
	else if (type == GETWORK)
	{
		curl_easy_setopt(curl, CURLOPT_POST, 1);
		curl_easy_setopt(curl, CURLOPT_COPYPOSTFIELDS, "{\"method\":\"sc_getwork\",\"params\":[],\"id\":1}");
	}
	else if (type == TESTWORK)
	{
		curl_easy_setopt(curl, CURLOPT_POST, 1);
		curl_easy_setopt(curl, CURLOPT_COPYPOSTFIELDS, ("{\"method\":\"sc_testwork\",\"id\":\"1\",\"params\":[\"" + work + "\"]}").c_str());
	}
	else
	{
		cout << "ERMYYR" << endl;
	}

	CURLcode code = curl_easy_perform(curl);
	if(code != CURLE_OK)
	{
		if (code == CURLE_COULDNT_CONNECT)
		{
			cout << humantime() << "Could not connect. Server down?" << endl;
		}
		else
		{
			cout << humantime() << "Error " << code << " getting work. See http://curl.haxx.se/libcurl/c/libcurl-errors.html for error code explanations." << endl;
		}
	}
	curl_slist_free_all(headerlist);
	return responsedata;
}

#undef SetPort
void Curl::SetPort(string port_) 
{ 
	port = FromString<unsigned short>(port_); 
}

string Curl::GetPort() 
{ 
	return ToString(port); 
}