cantyonion/LiveParise
Archived
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add curl_custom_init instead of init curl directly

Browse Source
This commit is contained in:
Jeffrey Hsu
2024-09-09 15:06:54 +08:00
parent 1e3d3e5a83
commit 426db0ede6
1 changed files with 35 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
main.c
View File

@@ -38,10 +38,9 @@ struct enc
long long timestamp; long long timestamp;
}; };
CURLcode curl_get(const char *, struct memory *); CURLcode curl_get(CURL *, const char *, struct memory *);
long long current_timestamp();
size_t cb(char *, size_t, size_t, void *); size_t cb(char *, size_t, size_t, void *);
struct enc get_enc(); struct enc get_enc(CURL *);
void parise(void *); void parise(void *);
void exit_handler(); void exit_handler();
@@ -88,34 +87,37 @@ int main(int argc, char *argv[])
return 0; return 0;
} }
CURLcode curl_get(const char *const url, struct memory *chunk) CURLcode curl_custom_init(CURL **curl, struct curl_slist **headers)
{ {
int status; *curl = curl_easy_init();
CURL *curl = NULL;
struct curl_slist *headers = NULL;
curl = curl_easy_init(); if (*curl == NULL)
if (curl == NULL)
return CURLE_FAILED_INIT; return CURLE_FAILED_INIT;
headers = curl_slist_append(headers, "Accept: application/json, text/javascript, */*; q=0.01"); *headers = curl_slist_append(*headers, "Accept: application/json, text/javascript, */*; q=0.01");
headers = curl_slist_append(headers, "User-Agent: Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like " *headers =
"Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0"); curl_slist_append(*headers, "User-Agent: Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like "
"Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0");
curl_easy_setopt(*curl, CURLOPT_HTTPHEADER, headers);
curl_easy_setopt(*curl, CURLOPT_DNS_SERVERS, DNS);
curl_easy_setopt(*curl, CURLOPT_USE_SSL, true);
curl_easy_setopt(*curl, CURLOPT_WRITEFUNCTION, cb);
return CURLE_OK;
}
CURLcode curl_get(CURL *curl, const char *const url, struct memory *chunk)
{
int status;
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
curl_easy_setopt(curl, CURLOPT_DNS_SERVERS, DNS);
curl_easy_setopt(curl, CURLOPT_URL, url); curl_easy_setopt(curl, CURLOPT_URL, url);
curl_easy_setopt(curl, CURLOPT_USE_SSL, true);
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, cb);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)chunk); curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)chunk);
status = curl_easy_perform(curl); status = curl_easy_perform(curl);
if (status != CURLE_OK) if (status != CURLE_OK)
fprintf(stderr, "curl_easy_perform() failed: %s\n", curl_easy_strerror(status)); fprintf(stderr, "curl_easy_perform() failed: %s\n", curl_easy_strerror(status));
curl_easy_cleanup(curl);
curl_slist_free_all(headers);
return status; return status;
} }
@@ -143,7 +145,7 @@ size_t cb(char *data, size_t size, size_t nmemb, void *clientp)
return realsize; return realsize;
} }
struct enc get_enc() struct enc get_enc(CURL *curl)
{ {
char url[200]; char url[200];
struct enc ret = {0}; struct enc ret = {0};
@@ -155,7 +157,7 @@ struct enc get_enc()
REGEN_ENC_URL_F(url, ret.timestamp); REGEN_ENC_URL_F(url, ret.timestamp);
if (curl_get(url, &data) != CURLE_OK) if (curl_get(curl, url, &data) != CURLE_OK)
return ret; return ret;
DPRINT("%s\n", data.response); DPRINT("%s\n", data.response);
@@ -176,6 +178,14 @@ struct enc get_enc()
void parise(void *arg) void parise(void *arg)
{ {
const int idx = (int)arg; const int idx = (int)arg;
CURL *curl = NULL;
struct curl_slist *headers = NULL;
if (curl_custom_init(&curl, &headers) != CURLE_OK)
{
fprintf(stderr, "T%d: curl_custom_init() failed\n", idx);
return;
}
while (flag) while (flag)
{ {
@@ -187,13 +197,12 @@ void parise(void *arg)
const cJSON *data = NULL; const cJSON *data = NULL;
const cJSON *errorMsg = NULL; const cJSON *errorMsg = NULL;
cJSON *json = NULL; cJSON *json = NULL;
if ((enc = get_enc(curl)).enc == NULL)
if ((enc = get_enc()).enc == NULL)
continue; continue;
REGEN_URL_F(url, enc.enc, enc.timestamp); REGEN_URL_F(url, enc.enc, enc.timestamp);
if (curl_get(url, &json_c) != CURLE_OK) if (curl_get(curl, url, &json_c) != CURLE_OK)
goto clean; goto clean;
json = cJSON_Parse(json_c.response); json = cJSON_Parse(json_c.response);
@@ -214,6 +223,8 @@ void parise(void *arg)
cJSON_Delete(json); cJSON_Delete(json);
} }
curl_easy_cleanup(curl);
curl_slist_free_all(headers);
printf("\rSignal Received: exit. T%d exiting\n", idx); printf("\rSignal Received: exit. T%d exiting\n", idx);
} }