102 lines
3.4 KiB
C
102 lines
3.4 KiB
C
/*
|
|
* This program is free software: you can redistribute it and/or modify it under
|
|
* the terms of the GNU Lesser General Public License as published by the Free
|
|
* Software Foundation, either version 3 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 Lesser General Public License for more
|
|
* details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "mastodont_fetch.h"
|
|
|
|
/* For use with libcurl */
|
|
size_t mstdnt_curl_write_callback(char* ptr, size_t _size, size_t nmemb, void* _content)
|
|
{
|
|
size_t size = nmemb * _size; /* Mostly pointless, but portable */
|
|
struct mstdnt_fetch_results* res = _content; /* Cast */
|
|
char* data;
|
|
|
|
if ((data = realloc(res->response, res->size + size + 1)) == NULL)
|
|
{
|
|
perror("realloc");
|
|
return 0;
|
|
}
|
|
|
|
res->response = data;
|
|
memcpy(&(res->response[res->size]), ptr, size);
|
|
res->size += size;
|
|
res->response[res->size] = 0;
|
|
|
|
return size;
|
|
}
|
|
|
|
void mastodont_fetch_results_cleanup(struct mstdnt_fetch_results* res)
|
|
{
|
|
free(res->response);
|
|
}
|
|
|
|
#define TOKEN_STR_SIZE 512
|
|
int mastodont_fetch_curl(mastodont_t* mstdnt,
|
|
struct mstdnt_args* m_args,
|
|
char* _url,
|
|
struct mstdnt_fetch_results* results,
|
|
CURLoption request_t,
|
|
char* request_t_custom)
|
|
{
|
|
#define is_custom request_t_custom && request_t == CURLOPT_CUSTOMREQUEST
|
|
int res = 3;
|
|
char token[TOKEN_STR_SIZE] = { 0 };
|
|
struct curl_slist* list = NULL;
|
|
|
|
/* Setup URL */
|
|
char url[MSTDNT_URLSIZE] = { 0 };
|
|
strncpy(url, m_args->url, MSTDNT_URLSIZE-1);
|
|
strncat(url, _url, MSTDNT_URLSIZE-1);
|
|
|
|
/* Setup token */
|
|
if (m_args->token)
|
|
{
|
|
snprintf(token, TOKEN_STR_SIZE, "Authorization: Bearer %s",
|
|
m_args->token);
|
|
list = curl_slist_append(list, token);
|
|
curl_easy_setopt(mstdnt->curl, CURLOPT_HTTPHEADER, list);
|
|
}
|
|
|
|
/* Set options */
|
|
curl_easy_setopt(mstdnt->curl, CURLOPT_URL, url);
|
|
curl_easy_setopt(mstdnt->curl, CURLOPT_WRITEFUNCTION, mstdnt_curl_write_callback);
|
|
curl_easy_setopt(mstdnt->curl, CURLOPT_WRITEDATA, results);
|
|
/* Should we verify the peer's SSL cert? */
|
|
curl_easy_setopt(mstdnt->curl, CURLOPT_SSL_VERIFYPEER,
|
|
!MSTDNT_T_FLAG_ISSET(m_args, MSTDNT_FLAG_SSL_UNVERIFIED));
|
|
curl_easy_setopt(mstdnt->curl, CURLOPT_SSL_VERIFYHOST,
|
|
!MSTDNT_T_FLAG_ISSET(m_args, MSTDNT_FLAG_SSL_UNVERIFIED));
|
|
/* PUT, POST, GET, Custom */
|
|
/* Mimes are expected to be set beforehand manually */
|
|
if (is_custom)
|
|
curl_easy_setopt(mstdnt->curl, request_t, request_t_custom);
|
|
else if (request_t != CURLOPT_MIMEPOST)
|
|
curl_easy_setopt(mstdnt->curl, request_t, 1);
|
|
|
|
res = curl_easy_perform(mstdnt->curl);
|
|
|
|
// Reset values that are optional
|
|
// Reset if custom
|
|
if (is_custom)
|
|
curl_easy_setopt(mstdnt->curl, request_t, NULL);
|
|
|
|
curl_easy_setopt(mstdnt->curl, CURLOPT_HTTPHEADER, NULL);
|
|
|
|
if (list) curl_slist_free_all(list);
|
|
|
|
return res;
|
|
}
|