86 lines
3.4 KiB
C++
86 lines
3.4 KiB
C++
/*****************************************************************************
|
|
* PokerTH - The open source texas holdem engine *
|
|
* Copyright (C) 2006-2011 Felix Hammer, Florian Thauer, Lothar May *
|
|
* *
|
|
* This program is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the GNU Affero General Public License as *
|
|
* published by the Free Software Foundation, either version 3 of the *
|
|
* License, or (at your option) any later version. *
|
|
* *
|
|
* 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 Affero General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Affero General Public License *
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
|
*****************************************************************************/
|
|
|
|
#include <net/socket_helper.h>
|
|
#include <net/uploadhelper.h>
|
|
#include <net/netexception.h>
|
|
#include <net/socket_msg.h>
|
|
#include <net/transferdata.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include <cstdio>
|
|
|
|
using namespace std;
|
|
|
|
|
|
size_t
|
|
readFunction(char *bufptr, size_t size, size_t nitems, void *userp)
|
|
{
|
|
return fread(bufptr, size, nitems, (FILE *)userp);
|
|
}
|
|
|
|
size_t
|
|
writeFunction(char *bufptr, size_t size, size_t nitems, void *userp)
|
|
{
|
|
string msgPart(bufptr, size * nitems);
|
|
((string *)userp)->append(msgPart);
|
|
return size * nitems;
|
|
}
|
|
|
|
UploadHelper::UploadHelper()
|
|
{
|
|
}
|
|
|
|
UploadHelper::~UploadHelper()
|
|
{
|
|
}
|
|
|
|
void
|
|
UploadHelper::InternalInit(const string &/*url*/, const string &targetFileName, const string &user, const string &password, size_t filesize, const string &httpPost)
|
|
{
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_SSL_VERIFYPEER, 0L);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_SSL_VERIFYHOST, 0L);
|
|
|
|
if (!user.empty() || !password.empty()) {
|
|
GetData()->userCredentials = user + ":" + password;
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_USERPWD, GetData()->userCredentials.c_str());
|
|
}
|
|
if (httpPost.empty()) {
|
|
// Open target file for reading.
|
|
GetData()->targetFile = fopen(targetFileName.c_str(), "rb");
|
|
if (!GetData()->targetFile)
|
|
throw NetException(__FILE__, __LINE__, ERR_SOCK_TRANSFER_OPEN_FAILED, 0);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_READFUNCTION, readFunction);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_READDATA, GetData()->targetFile);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_PUT, 1L);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_UPLOAD, 1L);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_INFILESIZE, filesize);
|
|
} else {
|
|
// Curl will handle file I/O.
|
|
struct curl_httppost *last = NULL;
|
|
curl_formadd(&GetData()->post, &last,
|
|
CURLFORM_COPYNAME, httpPost.c_str(),
|
|
CURLFORM_FILE, targetFileName.c_str(),
|
|
CURLFORM_END);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_HTTPPOST, GetData()->post);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_WRITEFUNCTION, writeFunction);
|
|
curl_easy_setopt(GetData()->curlHandle, CURLOPT_WRITEDATA, &GetData()->returnMessage);
|
|
}
|
|
}
|
|
|