mirror of
https://github.com/BeamMP/BeamMP-Launcher.git
synced 2026-04-03 06:16:15 +00:00
Replaced cURL with evpp
This commit is contained in:
Anonymous-275
@@ -6,139 +6,113 @@
|
||||
/// Created by Anonymous275 on 7/18/2020
|
||||
///
|
||||
|
||||
#include <curl/curl.h>
|
||||
#include <evpp/event_loop_thread.h>
|
||||
#include <evpp/httpc/conn_pool.h>
|
||||
#include <iostream>
|
||||
#include <Logger.h>
|
||||
#include <fstream>
|
||||
#include "http.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;
|
||||
};
|
||||
#include "winmain-inl.h"
|
||||
|
||||
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;
|
||||
static bool responded;
|
||||
std::string HTTP::Res_{};
|
||||
void HTTP::Response(const std::shared_ptr<evpp::httpc::Response>& response, evpp::httpc::Request* request){
|
||||
if(response->http_code() == 200){
|
||||
Res_ = std::move(response->body().ToString());
|
||||
} else {
|
||||
std::cout << "\n";
|
||||
error("Failed request! http code " + std::to_string(response->http_code()));
|
||||
}
|
||||
responded = true;
|
||||
assert(request == response->request());
|
||||
delete request;
|
||||
}
|
||||
|
||||
std::string HTTP_REQUEST(const std::string& IP,int port){
|
||||
static thread_local CurlManager M;
|
||||
std::string HTTP::Get(const std::string &IP) {
|
||||
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";
|
||||
responded = false;
|
||||
Res_.clear();
|
||||
evpp::EventLoopThread t;
|
||||
t.Start(true);
|
||||
|
||||
auto* r = new evpp::httpc::GetRequest(t.loop(), IP, evpp::Duration(10.0));
|
||||
r->Execute(Response);
|
||||
|
||||
while (!responded) {
|
||||
usleep(1);
|
||||
}
|
||||
return readBuffer;
|
||||
|
||||
t.Stop(true);
|
||||
return Res_;
|
||||
}
|
||||
|
||||
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;
|
||||
std::string HTTP::Post(const std::string& IP, const std::string& Fields) {
|
||||
static std::mutex Lock;
|
||||
std::scoped_lock Guard(Lock);
|
||||
CURL* curl = M.Get();
|
||||
CURLcode res;
|
||||
std::string readBuffer;
|
||||
responded = false;
|
||||
Res_.clear();
|
||||
evpp::EventLoopThread t;
|
||||
t.Start(true);
|
||||
|
||||
auto* r = new evpp::httpc::PostRequest(t.loop(), IP, Fields, evpp::Duration(10.0));
|
||||
r->AddHeader("Content-Type","application/json");
|
||||
r->Execute(Response);
|
||||
|
||||
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";
|
||||
while (!responded) {
|
||||
usleep(1);
|
||||
}
|
||||
return readBuffer;
|
||||
}
|
||||
|
||||
t.Stop(true);
|
||||
if(Res_.empty())return "-1";
|
||||
else return Res_;
|
||||
}
|
||||
|
||||
void ProgressBar(size_t d, size_t t){
|
||||
static double last_progress, progress_bar_adv;
|
||||
progress_bar_adv = round(d/double(t)*25);
|
||||
std::cout << "\r";
|
||||
std::cout << "Progress : [ ";
|
||||
std::cout << round(d/double(t)*100);
|
||||
std::cout << "% ] [";
|
||||
int i;
|
||||
for(i = 0; i <= progress_bar_adv; i++)std::cout<<"#";
|
||||
for(i = 0; i < 25 - progress_bar_adv; i++)std::cout<<".";
|
||||
std::cout << "]";
|
||||
last_progress = round(d/double(t)*100);
|
||||
}
|
||||
|
||||
bool HTTP::Download(const std::string &IP, const std::string &Path) {
|
||||
static std::mutex Lock;
|
||||
std::scoped_lock Guard(Lock);
|
||||
responded = false;
|
||||
Res_.clear();
|
||||
evpp::EventLoopThread t;
|
||||
t.Start(true);
|
||||
auto* r = new evpp::httpc::GetRequest(t.loop(), IP, evpp::Duration(10.0));
|
||||
r->set_progress_callback(ProgressBar);
|
||||
r->Execute(Response);
|
||||
|
||||
while (!responded) {
|
||||
usleep(1);
|
||||
}
|
||||
|
||||
t.Stop(true);
|
||||
|
||||
if(Res_.empty())return false;
|
||||
|
||||
std::ofstream File(Path, std::ios::binary);
|
||||
if(File.is_open()) {
|
||||
File << Res_;
|
||||
File.close();
|
||||
std::cout << "\n";
|
||||
}else{
|
||||
error("Failed to open file directory: " + Path);
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user