mirror of
https://github.com/obsproject/obs-studio.git
synced 2024-09-20 04:42:18 +02:00
710d99ef4d
When a source file contains an explicit include with a filename following the "moc_<actual-filename>.cpp" pattern, then CMake's AUTOMOC generation tool will recognize the matching pair and generate the replacement header file and add the required include directory entries. For all files which do contain Q_OBJECT or similar declarations but do not have an explicit include directive, the global mocs_compilation.cpp file will still be generated (which groups all "missing" generated headers). The larger this global file is, the more expensive incremental compilation will be as this file (and all its contained generated headers) will be re-generated regardless of whether actual changes occurred.
227 lines
6.1 KiB
C++
227 lines
6.1 KiB
C++
/******************************************************************************
|
|
Copyright (C) 2023 by Lain Bailey <lain@obsproject.com>
|
|
|
|
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 2 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
******************************************************************************/
|
|
|
|
#include <util/curl/curl-helper.h>
|
|
#include <qt-wrappers.hpp>
|
|
#include "obs-app.hpp"
|
|
#include "moc_remote-text.cpp"
|
|
|
|
using namespace std;
|
|
|
|
static auto curl_deleter = [](CURL *curl) {
|
|
curl_easy_cleanup(curl);
|
|
};
|
|
using Curl = unique_ptr<CURL, decltype(curl_deleter)>;
|
|
|
|
static size_t string_write(char *ptr, size_t size, size_t nmemb, string &str)
|
|
{
|
|
size_t total = size * nmemb;
|
|
if (total)
|
|
str.append(ptr, total);
|
|
|
|
return total;
|
|
}
|
|
|
|
void RemoteTextThread::run()
|
|
{
|
|
char error[CURL_ERROR_SIZE];
|
|
CURLcode code;
|
|
|
|
string versionString("User-Agent: obs-basic ");
|
|
versionString += App()->GetVersionString();
|
|
|
|
string contentTypeString;
|
|
if (!contentType.empty()) {
|
|
contentTypeString += "Content-Type: ";
|
|
contentTypeString += contentType;
|
|
}
|
|
|
|
Curl curl{curl_easy_init(), curl_deleter};
|
|
if (curl) {
|
|
struct curl_slist *header = nullptr;
|
|
string str;
|
|
|
|
header = curl_slist_append(header, versionString.c_str());
|
|
|
|
if (!contentTypeString.empty()) {
|
|
header = curl_slist_append(header,
|
|
contentTypeString.c_str());
|
|
}
|
|
|
|
for (std::string &h : extraHeaders)
|
|
header = curl_slist_append(header, h.c_str());
|
|
|
|
curl_easy_setopt(curl.get(), CURLOPT_URL, url.c_str());
|
|
curl_easy_setopt(curl.get(), CURLOPT_ACCEPT_ENCODING, "");
|
|
curl_easy_setopt(curl.get(), CURLOPT_HTTPHEADER, header);
|
|
curl_easy_setopt(curl.get(), CURLOPT_ERRORBUFFER, error);
|
|
curl_easy_setopt(curl.get(), CURLOPT_FAILONERROR, 1L);
|
|
curl_easy_setopt(curl.get(), CURLOPT_WRITEFUNCTION,
|
|
string_write);
|
|
curl_easy_setopt(curl.get(), CURLOPT_WRITEDATA, &str);
|
|
curl_obs_set_revoke_setting(curl.get());
|
|
|
|
if (timeoutSec)
|
|
curl_easy_setopt(curl.get(), CURLOPT_TIMEOUT,
|
|
timeoutSec);
|
|
|
|
if (!postData.empty()) {
|
|
curl_easy_setopt(curl.get(), CURLOPT_POSTFIELDS,
|
|
postData.c_str());
|
|
}
|
|
|
|
code = curl_easy_perform(curl.get());
|
|
if (code != CURLE_OK) {
|
|
blog(LOG_WARNING,
|
|
"RemoteTextThread: HTTP request failed. %s",
|
|
strlen(error) ? error : curl_easy_strerror(code));
|
|
emit Result(QString(), QT_UTF8(error));
|
|
} else {
|
|
emit Result(QT_UTF8(str.c_str()), QString());
|
|
}
|
|
|
|
curl_slist_free_all(header);
|
|
}
|
|
}
|
|
|
|
static size_t header_write(char *ptr, size_t size, size_t nmemb,
|
|
vector<string> &list)
|
|
{
|
|
string str;
|
|
|
|
size_t total = size * nmemb;
|
|
if (total)
|
|
str.append(ptr, total);
|
|
|
|
if (str.back() == '\n')
|
|
str.resize(str.size() - 1);
|
|
if (str.back() == '\r')
|
|
str.resize(str.size() - 1);
|
|
|
|
list.push_back(std::move(str));
|
|
return total;
|
|
}
|
|
|
|
bool GetRemoteFile(const char *url, std::string &str, std::string &error,
|
|
long *responseCode, const char *contentType,
|
|
std::string request_type, const char *postData,
|
|
std::vector<std::string> extraHeaders,
|
|
std::string *signature, int timeoutSec, bool fail_on_error,
|
|
int postDataSize)
|
|
{
|
|
vector<string> header_in_list;
|
|
char error_in[CURL_ERROR_SIZE];
|
|
CURLcode code = CURLE_FAILED_INIT;
|
|
|
|
error_in[0] = 0;
|
|
|
|
string versionString("User-Agent: obs-basic ");
|
|
versionString += App()->GetVersionString();
|
|
|
|
string contentTypeString;
|
|
if (contentType) {
|
|
contentTypeString += "Content-Type: ";
|
|
contentTypeString += contentType;
|
|
}
|
|
|
|
Curl curl{curl_easy_init(), curl_deleter};
|
|
if (curl) {
|
|
struct curl_slist *header = nullptr;
|
|
|
|
header = curl_slist_append(header, versionString.c_str());
|
|
|
|
if (!contentTypeString.empty()) {
|
|
header = curl_slist_append(header,
|
|
contentTypeString.c_str());
|
|
}
|
|
|
|
for (std::string &h : extraHeaders)
|
|
header = curl_slist_append(header, h.c_str());
|
|
|
|
curl_easy_setopt(curl.get(), CURLOPT_URL, url);
|
|
curl_easy_setopt(curl.get(), CURLOPT_ACCEPT_ENCODING, "");
|
|
curl_easy_setopt(curl.get(), CURLOPT_HTTPHEADER, header);
|
|
curl_easy_setopt(curl.get(), CURLOPT_ERRORBUFFER, error_in);
|
|
if (fail_on_error)
|
|
curl_easy_setopt(curl.get(), CURLOPT_FAILONERROR, 1L);
|
|
curl_easy_setopt(curl.get(), CURLOPT_WRITEFUNCTION,
|
|
string_write);
|
|
curl_easy_setopt(curl.get(), CURLOPT_WRITEDATA, &str);
|
|
curl_obs_set_revoke_setting(curl.get());
|
|
|
|
if (signature) {
|
|
curl_easy_setopt(curl.get(), CURLOPT_HEADERFUNCTION,
|
|
header_write);
|
|
curl_easy_setopt(curl.get(), CURLOPT_HEADERDATA,
|
|
&header_in_list);
|
|
}
|
|
|
|
if (timeoutSec)
|
|
curl_easy_setopt(curl.get(), CURLOPT_TIMEOUT,
|
|
timeoutSec);
|
|
|
|
if (!request_type.empty()) {
|
|
if (request_type != "GET")
|
|
curl_easy_setopt(curl.get(),
|
|
CURLOPT_CUSTOMREQUEST,
|
|
request_type.c_str());
|
|
|
|
// Special case of "POST"
|
|
if (request_type == "POST") {
|
|
curl_easy_setopt(curl.get(), CURLOPT_POST, 1);
|
|
if (!postData)
|
|
curl_easy_setopt(curl.get(),
|
|
CURLOPT_POSTFIELDS,
|
|
"{}");
|
|
}
|
|
}
|
|
if (postData) {
|
|
if (postDataSize > 0) {
|
|
curl_easy_setopt(curl.get(),
|
|
CURLOPT_POSTFIELDSIZE,
|
|
(long)postDataSize);
|
|
}
|
|
curl_easy_setopt(curl.get(), CURLOPT_POSTFIELDS,
|
|
postData);
|
|
}
|
|
|
|
code = curl_easy_perform(curl.get());
|
|
if (responseCode)
|
|
curl_easy_getinfo(curl.get(), CURLINFO_RESPONSE_CODE,
|
|
responseCode);
|
|
|
|
if (code != CURLE_OK) {
|
|
error = strlen(error_in) ? error_in
|
|
: curl_easy_strerror(code);
|
|
} else if (signature) {
|
|
for (string &h : header_in_list) {
|
|
string name = h.substr(0, 13);
|
|
// HTTP headers are technically case-insensitive
|
|
if (name == "X-Signature: " ||
|
|
name == "x-signature: ") {
|
|
*signature = h.substr(13);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
curl_slist_free_all(header);
|
|
}
|
|
|
|
return code == CURLE_OK;
|
|
}
|