406 lines
12 KiB
C
406 lines
12 KiB
C
#include "network.h"
|
|
|
|
#include "cache.h"
|
|
|
|
#include <openssl/crypto.h>
|
|
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
|
|
#define DEFAULT_NETWORK_MAX_CONNS 10
|
|
#define DEFAULT_HTTP_429_WAIT 5
|
|
|
|
/* ----------------- External variables ---------------------- */
|
|
CURLSH *CURL_SHARE;
|
|
NetworkConfigStruct NETWORK_CONFIG;
|
|
int HTTP_429_WAIT = DEFAULT_HTTP_429_WAIT;
|
|
|
|
/* ----------------- Static variable ----------------------- */
|
|
/** \brief curl multi interface handle */
|
|
static CURLM *curl_multi;
|
|
/** \brief mutex for transfer functions */
|
|
static pthread_mutex_t transfer_lock;
|
|
/** \brief the lock array for cryptographic functions */
|
|
static pthread_mutex_t *crypto_lockarray;
|
|
/** \brief mutex for curl share interface itself */
|
|
static pthread_mutex_t curl_lock;
|
|
|
|
/* -------------------- Functions -------------------------- */
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wunused-function"
|
|
/**
|
|
* \brief OpenSSL 1.02 cryptography callback function
|
|
* \details Required for OpenSSL 1.02, but not OpenSSL 1.1
|
|
*/
|
|
static void crypto_lock_callback(int mode, int type, char *file, int line)
|
|
{
|
|
(void) file;
|
|
(void) line;
|
|
if (mode & CRYPTO_LOCK) {
|
|
PTHREAD_MUTEX_LOCK(&(crypto_lockarray[type]));
|
|
} else {
|
|
PTHREAD_MUTEX_UNLOCK(&(crypto_lockarray[type]));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* \brief OpenSSL 1.02 thread ID function
|
|
* \details Required for OpenSSL 1.02, but not OpenSSL 1.1
|
|
*/
|
|
static unsigned long thread_id(void)
|
|
{
|
|
unsigned long ret;
|
|
|
|
ret = (unsigned long)pthread_self();
|
|
return ret;
|
|
}
|
|
#pragma GCC diagnostic pop
|
|
|
|
static void crypto_lock_init(void)
|
|
{
|
|
int i;
|
|
|
|
crypto_lockarray = (pthread_mutex_t *)OPENSSL_malloc(CRYPTO_num_locks() *
|
|
sizeof(pthread_mutex_t));
|
|
for(i = 0; i<CRYPTO_num_locks(); i++) {
|
|
pthread_mutex_init(&(crypto_lockarray[i]), NULL);
|
|
}
|
|
|
|
CRYPTO_set_id_callback((unsigned long (*)())thread_id);
|
|
CRYPTO_set_locking_callback((void (*)())crypto_lock_callback);
|
|
}
|
|
|
|
/**
|
|
* \brief Curl share handle callback function
|
|
* \details Adapted from:
|
|
* https://curl.haxx.se/libcurl/c/threaded-shared-conn.html
|
|
*/
|
|
static void curl_callback_lock(CURL *handle, curl_lock_data data,
|
|
curl_lock_access access, void *userptr)
|
|
{
|
|
(void)access; /* unused */
|
|
(void)userptr; /* unused */
|
|
(void)handle; /* unused */
|
|
(void)data; /* unused */
|
|
PTHREAD_MUTEX_LOCK(&curl_lock);
|
|
}
|
|
|
|
static void curl_callback_unlock(CURL *handle, curl_lock_data data,
|
|
void *userptr)
|
|
{
|
|
(void)userptr; /* unused */
|
|
(void)handle; /* unused */
|
|
(void)data; /* unused */
|
|
PTHREAD_MUTEX_UNLOCK(&curl_lock);
|
|
}
|
|
|
|
/**
|
|
* \brief Process a curl message
|
|
* \details Adapted from:
|
|
* https://curl.haxx.se/libcurl/c/10-at-a-time.html
|
|
*/
|
|
static void curl_process_msgs(CURLMsg *curl_msg, int n_running_curl,
|
|
int n_mesgs)
|
|
{
|
|
(void) n_running_curl;
|
|
(void) n_mesgs;
|
|
static volatile int slept = 0;
|
|
if (curl_msg->msg == CURLMSG_DONE) {
|
|
TransferStruct *transfer;
|
|
CURL *curl = curl_msg->easy_handle;
|
|
curl_easy_getinfo(curl_msg->easy_handle, CURLINFO_PRIVATE,
|
|
&transfer);
|
|
transfer->transferring = 0;
|
|
char *url = NULL;
|
|
curl_easy_getinfo(curl, CURLINFO_EFFECTIVE_URL, &url);
|
|
|
|
/* Wait for 5 seconds if we get HTTP 429 */
|
|
long http_resp = 0;
|
|
curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &http_resp);
|
|
if (http_resp == HTTP_TOO_MANY_REQUESTS) {
|
|
if (!slept) {
|
|
fprintf(stderr,
|
|
"curl_process_msgs(): HTTP 429, sleeping for %d sec\n",
|
|
HTTP_429_WAIT);
|
|
sleep(HTTP_429_WAIT);
|
|
slept = 1;
|
|
}
|
|
} else {
|
|
slept = 0;
|
|
}
|
|
|
|
if (!curl_msg->data.result) {
|
|
/* Transfer successful, set the file size */
|
|
if (transfer->type == FILESTAT) {
|
|
Link_set_file_stat(transfer->link, curl);
|
|
}
|
|
} else {
|
|
fprintf(stderr, "curl_process_msgs(): %d - %s <%s>\n",
|
|
curl_msg->data.result,
|
|
curl_easy_strerror(curl_msg->data.result),
|
|
url);
|
|
}
|
|
curl_multi_remove_handle(curl_multi, curl);
|
|
/* clean up the handle, if we are querying the file size */
|
|
if (transfer->type == FILESTAT) {
|
|
curl_easy_cleanup(curl);
|
|
free(transfer);
|
|
}
|
|
} else {
|
|
fprintf(stderr, "curl_process_msgs(): curl_msg->msg: %d\n",
|
|
curl_msg->msg);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* \details effectively based on
|
|
* https://curl.haxx.se/libcurl/c/multi-double.html
|
|
*/
|
|
int curl_multi_perform_once()
|
|
{
|
|
#ifdef NETWORK_LOCK_DEBUG
|
|
fprintf(stderr,
|
|
"curl_multi_perform_once(): thread %lu: locking transfer_lock;\n",
|
|
pthread_self());
|
|
#endif
|
|
PTHREAD_MUTEX_LOCK(&transfer_lock);
|
|
/* Get curl multi interface to perform pending tasks */
|
|
int n_running_curl;
|
|
CURLMcode mc = curl_multi_perform(curl_multi, &n_running_curl);
|
|
if(mc > 0) {
|
|
fprintf(stderr, "curl_multi_perform(): %s\n", curl_multi_strerror(mc));
|
|
}
|
|
|
|
fd_set fdread;
|
|
fd_set fdwrite;
|
|
fd_set fdexcep;
|
|
int maxfd = -1;
|
|
|
|
long curl_timeo = -1;
|
|
|
|
FD_ZERO(&fdread);
|
|
FD_ZERO(&fdwrite);
|
|
FD_ZERO(&fdexcep);
|
|
|
|
/* set a default timeout for select() */
|
|
struct timeval timeout;
|
|
timeout.tv_sec = 1;
|
|
timeout.tv_usec = 0;
|
|
|
|
curl_multi_timeout(curl_multi, &curl_timeo);
|
|
/* We effectively cap timeout to 1 sec */
|
|
if (curl_timeo >= 0) {
|
|
timeout.tv_sec = curl_timeo / 1000;
|
|
if (timeout.tv_sec > 1) {
|
|
timeout.tv_sec = 1;
|
|
} else {
|
|
timeout.tv_usec = (curl_timeo % 1000) * 1000;
|
|
}
|
|
}
|
|
|
|
/* get file descriptors from the transfers */
|
|
mc = curl_multi_fdset(curl_multi, &fdread, &fdwrite, &fdexcep, &maxfd);
|
|
|
|
if (mc > 0) {
|
|
fprintf(stderr, "curl_multi_fdset(): %s.\n", curl_multi_strerror(mc));
|
|
}
|
|
|
|
if (maxfd == -1) {
|
|
usleep(100*1000);
|
|
} else {
|
|
if (select(maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout) < 0) {
|
|
fprintf(stderr, "curl_multi_perform_once(): select(): %s.\n",
|
|
strerror(errno));
|
|
}
|
|
}
|
|
|
|
/* Process the message queue */
|
|
int n_mesgs;
|
|
CURLMsg *curl_msg;
|
|
while((curl_msg = curl_multi_info_read(curl_multi, &n_mesgs))) {
|
|
curl_process_msgs(curl_msg, n_running_curl, n_mesgs);
|
|
}
|
|
#ifdef NETWORK_LOCK_DEBUG
|
|
fprintf(stderr,
|
|
"curl_multi_perform_once(): thread %lu: unlocking transfer_lock;\n",
|
|
pthread_self());
|
|
#endif
|
|
PTHREAD_MUTEX_UNLOCK(&transfer_lock);
|
|
return n_running_curl;
|
|
}
|
|
|
|
void network_config_init()
|
|
{
|
|
NETWORK_CONFIG.username = NULL;
|
|
NETWORK_CONFIG.password = NULL;
|
|
NETWORK_CONFIG.proxy = NULL;
|
|
NETWORK_CONFIG.proxy_user = NULL;
|
|
NETWORK_CONFIG.proxy_pass = NULL;
|
|
NETWORK_CONFIG.max_conns = DEFAULT_NETWORK_MAX_CONNS;
|
|
NETWORK_CONFIG.user_agent = "HTTPDirFS";
|
|
NETWORK_CONFIG.cache_enabled = 0;
|
|
NETWORK_CONFIG.cache_dir = NULL;
|
|
}
|
|
|
|
LinkTable *network_init(const char *url)
|
|
{
|
|
/* ------- Global related ----------*/
|
|
if (curl_global_init(CURL_GLOBAL_ALL)) {
|
|
fprintf(stderr, "network_init(): curl_global_init() failed!\n");
|
|
exit_failure();
|
|
}
|
|
|
|
/* -------- Share related ----------*/
|
|
CURL_SHARE = curl_share_init();
|
|
if (!(CURL_SHARE)) {
|
|
fprintf(stderr, "network_init(): curl_share_init() failed!\n");
|
|
exit_failure();
|
|
}
|
|
|
|
curl_share_setopt(CURL_SHARE, CURLSHOPT_SHARE, CURL_LOCK_DATA_COOKIE);
|
|
curl_share_setopt(CURL_SHARE, CURLSHOPT_SHARE, CURL_LOCK_DATA_DNS);
|
|
curl_share_setopt(CURL_SHARE, CURLSHOPT_SHARE, CURL_LOCK_DATA_SSL_SESSION);
|
|
|
|
if (pthread_mutex_init(&curl_lock, NULL) != 0) {
|
|
fprintf(stderr, "network_init(): curl_lock initialisation failed!\n");
|
|
exit_failure();
|
|
}
|
|
curl_share_setopt(CURL_SHARE, CURLSHOPT_LOCKFUNC, curl_callback_lock);
|
|
curl_share_setopt(CURL_SHARE, CURLSHOPT_UNLOCKFUNC, curl_callback_unlock);
|
|
|
|
/* ------------- Multi related -----------*/
|
|
curl_multi = curl_multi_init();
|
|
if (!curl_multi) {
|
|
fprintf(stderr, "network_init(): curl_multi_init() failed!\n");
|
|
exit_failure();
|
|
}
|
|
curl_multi_setopt(curl_multi, CURLMOPT_MAX_TOTAL_CONNECTIONS,
|
|
NETWORK_CONFIG.max_conns);
|
|
curl_multi_setopt(curl_multi, CURLMOPT_MAX_HOST_CONNECTIONS,
|
|
NETWORK_CONFIG.max_conns);
|
|
|
|
/* ------------ Initialise locks ---------*/
|
|
if (pthread_mutex_init(&transfer_lock, NULL)) {
|
|
fprintf(stderr,
|
|
"network_init(): transfer_lock initialisation failed!\n");
|
|
exit_failure();
|
|
}
|
|
|
|
/*
|
|
* cryptographic lock functions were shamelessly copied from
|
|
* https://curl.haxx.se/libcurl/c/threaded-ssl.html
|
|
*/
|
|
crypto_lock_init();
|
|
|
|
/* --------- Print off SSL engine version stream --------- */
|
|
curl_version_info_data *data = curl_version_info(CURLVERSION_NOW);
|
|
fprintf(stderr, "libcurl SSL engine: %s\n", data->ssl_version);
|
|
|
|
/* --------- Set the length of the root link ----------- */
|
|
/* This is where the '/' should be */
|
|
ROOT_LINK_OFFSET = strnlen(url, MAX_PATH_LEN) - 1;
|
|
if (url[ROOT_LINK_OFFSET] != '/') {
|
|
/*
|
|
* If '/' is not there, it is automatically added, so we need to skip 2
|
|
* characters
|
|
*/
|
|
ROOT_LINK_OFFSET += 2;
|
|
} else {
|
|
/* If '/' is there, we need to skip it */
|
|
ROOT_LINK_OFFSET += 1;
|
|
}
|
|
|
|
/* ----------- Enable cache system --------------------*/
|
|
if (NETWORK_CONFIG.cache_enabled) {
|
|
if (NETWORK_CONFIG.cache_dir) {
|
|
CacheSystem_init(NETWORK_CONFIG.cache_dir, 0);
|
|
} else {
|
|
CacheSystem_init(url, 1);
|
|
}
|
|
}
|
|
|
|
/* ----------- Create the root link table --------------*/
|
|
ROOT_LINK_TBL = LinkTable_new(url);
|
|
return ROOT_LINK_TBL;
|
|
}
|
|
|
|
void transfer_blocking(CURL *curl)
|
|
{
|
|
/*
|
|
* We don't need to malloc here, as the transfer is finished before
|
|
* the variable gets popped from the stack
|
|
*/
|
|
volatile TransferStruct transfer;
|
|
transfer.type = DATA;
|
|
transfer.transferring = 1;
|
|
curl_easy_setopt(curl, CURLOPT_PRIVATE, &transfer);
|
|
#ifdef NETWORK_LOCK_DEBUG
|
|
fprintf(stderr,
|
|
"transfer_blocking(): thread %lu: locking transfer_lock;\n",
|
|
pthread_self());
|
|
#endif
|
|
PTHREAD_MUTEX_LOCK(&transfer_lock);
|
|
CURLMcode res = curl_multi_add_handle(curl_multi, curl);
|
|
#ifdef NETWORK_LOCK_DEBUG
|
|
fprintf(stderr,
|
|
"transfer_blocking(): thread %lu: unlocking transfer_lock;\n",
|
|
pthread_self());
|
|
#endif
|
|
PTHREAD_MUTEX_UNLOCK(&transfer_lock);
|
|
|
|
if(res > 0) {
|
|
fprintf(stderr, "transfer_blocking(): %d, %s\n",
|
|
res, curl_multi_strerror(res));
|
|
exit_failure();
|
|
}
|
|
|
|
while (transfer.transferring) {
|
|
curl_multi_perform_once();
|
|
}
|
|
}
|
|
|
|
void transfer_nonblocking(CURL *curl)
|
|
{
|
|
#ifdef NETWORK_LOCK_DEBUG
|
|
fprintf(stderr,
|
|
"transfer_nonblocking(): thread %lu: locking transfer_lock;\n",
|
|
pthread_self());
|
|
#endif
|
|
PTHREAD_MUTEX_LOCK(&transfer_lock);
|
|
CURLMcode res = curl_multi_add_handle(curl_multi, curl);
|
|
#ifdef NETWORK_LOCK_DEBUG
|
|
fprintf(stderr,
|
|
"transfer_nonblocking(): thread %lu: unlocking transfer_lock;\n",
|
|
pthread_self());
|
|
#endif
|
|
PTHREAD_MUTEX_UNLOCK(&transfer_lock);
|
|
|
|
if(res > 0) {
|
|
fprintf(stderr, "transfer_nonblocking(): %s\n",
|
|
curl_multi_strerror(res));
|
|
}
|
|
}
|
|
|
|
size_t write_memory_callback(void *contents, size_t size, size_t nmemb,
|
|
void *userp)
|
|
{
|
|
size_t realsize = size * nmemb;
|
|
MemoryStruct *mem = (MemoryStruct *)userp;
|
|
|
|
mem->memory = realloc(mem->memory, mem->size + realsize + 1);
|
|
if(!mem->memory) {
|
|
/* out of memory! */
|
|
fprintf(stderr, "write_memory_callback(): realloc failure!\n");
|
|
exit_failure();
|
|
return 0;
|
|
}
|
|
|
|
memmove(&mem->memory[mem->size], contents, realsize);
|
|
mem->size += realsize;
|
|
mem->memory[mem->size] = 0;
|
|
|
|
return realsize;
|
|
}
|