mirror of
https://github.com/BeamMP/BeamMP-Launcher.git
synced 2026-04-03 06:16:15 +00:00
Mingw Compatible
This commit is contained in:
Anonymous-275
282
src/Network/Http.cpp
Normal file → Executable file
282
src/Network/Http.cpp
Normal file → Executable file
@@ -1,140 +1,144 @@
|
||||
// Copyright (c) 2019-present Anonymous275.
|
||||
// BeamMP Launcher code is not in the public domain and is not free software.
|
||||
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
|
||||
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
|
||||
///
|
||||
/// Created by Anonymous275 on 7/18/2020
|
||||
///
|
||||
|
||||
#include <curl/curl.h>
|
||||
#include <iostream>
|
||||
#include <Logger.h>
|
||||
#include <mutex>
|
||||
|
||||
class CurlManager{
|
||||
public:
|
||||
CurlManager(){
|
||||
curl = curl_easy_init();
|
||||
}
|
||||
~CurlManager(){
|
||||
curl_easy_cleanup(curl);
|
||||
}
|
||||
inline CURL* Get(){
|
||||
return curl;
|
||||
}
|
||||
private:
|
||||
CURL *curl;
|
||||
};
|
||||
|
||||
static size_t WriteCallback(void *contents, size_t size, size_t nmemb, void *userp){
|
||||
((std::string*)userp)->append((char*)contents, size * nmemb);
|
||||
return size * nmemb;
|
||||
}
|
||||
|
||||
std::string HTTP_REQUEST(const std::string& IP,int port){
|
||||
static thread_local CurlManager M;
|
||||
static std::mutex Lock;
|
||||
std::scoped_lock Guard(Lock);
|
||||
CURL *curl = M.Get();
|
||||
CURLcode res;
|
||||
std::string readBuffer;
|
||||
if(curl) {
|
||||
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 2L);
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 1L);
|
||||
curl_easy_setopt(curl, CURLOPT_PORT, port);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer);
|
||||
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10);
|
||||
res = curl_easy_perform(curl);
|
||||
if(res != CURLE_OK)return "-1";
|
||||
}
|
||||
return readBuffer;
|
||||
}
|
||||
|
||||
int nb_bar;
|
||||
double last_progress, progress_bar_adv;
|
||||
bool Downloaded;
|
||||
int progress_bar (void *bar, double t, double d){
|
||||
if(t > 0 && last_progress != round(d/t*100)){
|
||||
nb_bar = 25;
|
||||
progress_bar_adv = round(d/t*nb_bar);
|
||||
std::cout<<"\r";
|
||||
std::cout<< "Progress : [ ";
|
||||
std::cout<<round(d/t*100);
|
||||
std::cout << "% ] [";
|
||||
int i;
|
||||
for(i = 0; i <= progress_bar_adv; i++)std::cout<<"#";
|
||||
for(i = 0; i < nb_bar - progress_bar_adv; i++)std::cout<<".";
|
||||
std::cout<<"]";
|
||||
last_progress = round(d/t*100);
|
||||
if(round(d/t*100) == 100){
|
||||
Downloaded = true;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
struct File {
|
||||
const char *filename;
|
||||
FILE *stream;
|
||||
};
|
||||
static size_t my_fwrite(void *buffer,size_t size,size_t nmemb,void *stream){
|
||||
auto *out = (struct File*)stream;
|
||||
if(!out->stream) {
|
||||
fopen_s(&out->stream,out->filename,"wb");
|
||||
if(!out->stream)return -1;
|
||||
}
|
||||
return fwrite(buffer, size, nmemb, out->stream);
|
||||
}
|
||||
int Download(const std::string& URL,const std::string& Path,bool close){
|
||||
static thread_local CurlManager M;
|
||||
CURL *curl = M.Get();
|
||||
CURLcode res;
|
||||
struct File file = {Path.c_str(),nullptr};
|
||||
Downloaded = false;
|
||||
if(curl){
|
||||
curl_easy_setopt(curl, CURLOPT_URL,URL.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 2L);
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 1L);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, my_fwrite);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &file);
|
||||
curl_easy_setopt(curl, CURLOPT_NOPROGRESS, FALSE);
|
||||
curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_bar);
|
||||
curl_easy_setopt(curl, CURLOPT_USE_SSL, CURLUSESSL_ALL);
|
||||
res = curl_easy_perform(curl);
|
||||
if(res != CURLE_OK){
|
||||
return res;
|
||||
}
|
||||
}
|
||||
if(file.stream)fclose(file.stream);
|
||||
if(!Downloaded){
|
||||
remove(Path.c_str());
|
||||
fatal("Failed to download please try again later!");
|
||||
}
|
||||
std::cout << std::endl;
|
||||
return -1;
|
||||
}
|
||||
std::string PostHTTP(const std::string& IP, const std::string& Fields) {
|
||||
static auto *header = new curl_slist{(char*)"Content-Type: application/json"};
|
||||
static thread_local CurlManager M;
|
||||
static std::mutex Lock;
|
||||
std::scoped_lock Guard(Lock);
|
||||
CURL* curl = M.Get();
|
||||
CURLcode res;
|
||||
std::string readBuffer;
|
||||
|
||||
if (curl) {
|
||||
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, header);
|
||||
curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, Fields.size());
|
||||
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, Fields.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer);
|
||||
curl_easy_setopt(curl, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4);
|
||||
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10);
|
||||
res = curl_easy_perform(curl);
|
||||
if (res != CURLE_OK)
|
||||
return "-1";
|
||||
}
|
||||
return readBuffer;
|
||||
// Copyright (c) 2019-present Anonymous275.
|
||||
// BeamMP Launcher code is not in the public domain and is not free software.
|
||||
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
|
||||
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
|
||||
///
|
||||
/// Created by Anonymous275 on 7/18/2020
|
||||
///
|
||||
|
||||
#include <curl/curl.h>
|
||||
#include <iostream>
|
||||
#include <Logger.h>
|
||||
#include <mutex>
|
||||
#include <cmath>
|
||||
|
||||
class CurlManager{
|
||||
public:
|
||||
CurlManager(){
|
||||
curl = curl_easy_init();
|
||||
}
|
||||
~CurlManager(){
|
||||
curl_easy_cleanup(curl);
|
||||
}
|
||||
inline CURL* Get(){
|
||||
return curl;
|
||||
}
|
||||
private:
|
||||
CURL *curl;
|
||||
};
|
||||
|
||||
static size_t WriteCallback(void *contents, size_t size, size_t nmemb, void *userp){
|
||||
((std::string*)userp)->append((char*)contents, size * nmemb);
|
||||
return size * nmemb;
|
||||
}
|
||||
|
||||
std::string HTTP_REQUEST(const std::string& IP,int port){
|
||||
static thread_local CurlManager M;
|
||||
static std::mutex Lock;
|
||||
std::scoped_lock Guard(Lock);
|
||||
CURL *curl = M.Get();
|
||||
CURLcode res;
|
||||
std::string readBuffer;
|
||||
if(curl) {
|
||||
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 2L);
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 1L);
|
||||
curl_easy_setopt(curl, CURLOPT_PORT, port);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer);
|
||||
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10);
|
||||
res = curl_easy_perform(curl);
|
||||
if(res != CURLE_OK)return "-1";
|
||||
}
|
||||
return readBuffer;
|
||||
}
|
||||
|
||||
int nb_bar;
|
||||
double last_progress, progress_bar_adv;
|
||||
bool Downloaded;
|
||||
int progress_bar (void *bar, double t, double d){
|
||||
if(t > 0 && last_progress != round(d/t*100)){
|
||||
nb_bar = 25;
|
||||
progress_bar_adv = round(d/t*nb_bar);
|
||||
std::cout<<"\r";
|
||||
std::cout<< "Progress : [ ";
|
||||
std::cout<<round(d/t*100);
|
||||
std::cout << "% ] [";
|
||||
int i;
|
||||
for(i = 0; i <= progress_bar_adv; i++)std::cout<<"#";
|
||||
for(i = 0; i < nb_bar - progress_bar_adv; i++)std::cout<<".";
|
||||
std::cout<<"]";
|
||||
last_progress = round(d/t*100);
|
||||
if(round(d/t*100) == 100){
|
||||
Downloaded = true;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
struct File {
|
||||
const char *filename;
|
||||
FILE *stream;
|
||||
};
|
||||
static size_t my_fwrite(void *buffer,size_t size,size_t nmemb,void *stream){
|
||||
auto *out = (struct File*)stream;
|
||||
if(!out->stream) {
|
||||
fopen_s(&out->stream,out->filename,"wb");
|
||||
if(!out->stream)return -1;
|
||||
}
|
||||
return fwrite(buffer, size, nmemb, out->stream);
|
||||
}
|
||||
int Download(const std::string& URL,const std::string& Path,bool close){
|
||||
static thread_local CurlManager M;
|
||||
CURL *curl = M.Get();
|
||||
CURLcode res;
|
||||
struct File file = {Path.c_str(),nullptr};
|
||||
Downloaded = false;
|
||||
if(curl){
|
||||
curl_easy_setopt(curl, CURLOPT_URL,URL.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 2L);
|
||||
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 1L);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, my_fwrite);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &file);
|
||||
curl_easy_setopt(curl, CURLOPT_NOPROGRESS, FALSE);
|
||||
curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_bar);
|
||||
curl_easy_setopt(curl, CURLOPT_USE_SSL, CURLUSESSL_ALL);
|
||||
res = curl_easy_perform(curl);
|
||||
if(res != CURLE_OK){
|
||||
return res;
|
||||
}
|
||||
}
|
||||
if(file.stream)fclose(file.stream);
|
||||
if(!Downloaded){
|
||||
remove(Path.c_str());
|
||||
fatal("Failed to download please try again later!");
|
||||
}
|
||||
std::cout << std::endl;
|
||||
return -1;
|
||||
}
|
||||
std::string PostHTTP(const std::string& IP, const std::string& Fields) {
|
||||
static auto *header = new curl_slist{(char*)"Content-Type: application/json"};
|
||||
static thread_local CurlManager M;
|
||||
static std::mutex Lock;
|
||||
std::scoped_lock Guard(Lock);
|
||||
CURL* curl = M.Get();
|
||||
CURLcode res;
|
||||
std::string readBuffer;
|
||||
|
||||
|
||||
if (curl) {
|
||||
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, header);
|
||||
curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, Fields.size());
|
||||
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, Fields.c_str());
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
|
||||
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer);
|
||||
//curl_easy_setopt(curl, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4);
|
||||
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10);
|
||||
|
||||
res = curl_easy_perform(curl);
|
||||
|
||||
if (res != CURLE_OK)
|
||||
return "-1";
|
||||
}
|
||||
return readBuffer;
|
||||
}
|
||||
Reference in New Issue
Block a user