diego
d5da9a9574
being unreadable. Hopefully this makes the thing more clear and readable. Some improvements could be done to the upload_file_* thing. As there's really no need to pass argc and argv to those. And passing argv to that is an overkill Solution: To create a const char* variable and save the filename in there and pass that to the functions. Simple as that. But i'm a lazy motherfucker and I'm tired. Boss.
152 lines
3.2 KiB
C
152 lines
3.2 KiB
C
#include <curl/system.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <libconfig.h>
|
|
#include <curl/curl.h>
|
|
#include <stdbool.h>
|
|
#include <sys/stat.h>
|
|
#include <getopt.h>
|
|
#include "sakisafecli.h"
|
|
#include "config.h"
|
|
|
|
size_t
|
|
write_data(void *buffer, size_t size, size_t nmemb, void *userp)
|
|
{
|
|
memcpy(userp, buffer, nmemb * size);
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
print_usage()
|
|
{
|
|
printf("USAGE: sakisafecli [-p socks proxy|-P http proxy] [-x] [-s] "
|
|
"[-6|-4] [--server] FILE\n");
|
|
return;
|
|
}
|
|
|
|
void
|
|
print_help()
|
|
{
|
|
printf("-s|--server: specifies the sakisafe "
|
|
"server\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s",
|
|
"-t|--token: Authentication token (https://u.kalli.st)",
|
|
"-P|--http-proxy: http proxy to use e.g. http://127.0.0.1:4444",
|
|
"-p|--socks-proxy: SOCK proxy to use e.g. 127.0.0.1:9050",
|
|
"-6|--ipv6: uses IPv6 only",
|
|
"-4|--ipv6: uses IPv4 only",
|
|
"-S|--silent: doesn't print progress",
|
|
"-x|--paste: read file from stdin",
|
|
"-C: print current settings",
|
|
"-h|--help: print this message.\n");
|
|
return;
|
|
}
|
|
|
|
size_t
|
|
progress(void *clientp,
|
|
curl_off_t dltotal,
|
|
curl_off_t dlnow,
|
|
curl_off_t ultotal,
|
|
curl_off_t ulnow)
|
|
{
|
|
/* I don't know why the fuck I have to do this */
|
|
if(ultotal == 0) {
|
|
ultotal++;
|
|
}
|
|
printf("\r%li uploaded of %li (\033[32;1m%li%%\033[30;0m)",
|
|
ulnow,
|
|
ultotal,
|
|
ulnow * 100 / ultotal);
|
|
fflush(stdout);
|
|
return 0;
|
|
}
|
|
int
|
|
get_protocol(char *server)
|
|
{
|
|
if(strstr(server, "http://") != NULL || strstr(server, "https://"))
|
|
return CURLPROTO_HTTP;
|
|
else if(strstr(server, "scp://") != NULL)
|
|
return CURLPROTO_SCP;
|
|
else
|
|
return -1;
|
|
}
|
|
void
|
|
die(const char *msg)
|
|
{
|
|
fprintf(stderr, "%i: %s", errno, msg);
|
|
}
|
|
|
|
void
|
|
init_sakisafe_options()
|
|
{
|
|
ipv6_flag = http_proxy_flag,
|
|
socks_proxy_flag = paste_flag = silent_flag = false;
|
|
|
|
ipv4_flag = true;
|
|
|
|
ssh_key_path = NULL;
|
|
server = "https://meth.cat";
|
|
path = ".cache/sakisafelinks";
|
|
}
|
|
|
|
int
|
|
upload_file_http(int argc, char **argv)
|
|
{
|
|
|
|
curl_mime *mime;
|
|
mime = curl_mime_init(easy_handle);
|
|
|
|
curl_easy_setopt(easy_handle, CURLOPT_MIMEPOST, mime);
|
|
if(!mime) {
|
|
fprintf(stderr, "Error initializing curl_mime\n");
|
|
return -1;
|
|
}
|
|
|
|
curl_mimepart *file_data;
|
|
file_data = curl_mime_addpart(mime);
|
|
char *filename = argv[optind - 1];
|
|
/* Get file from stdin */
|
|
if(paste_flag)
|
|
filename = "/dev/stdin";
|
|
curl_mime_filedata(file_data, filename);
|
|
curl_mime_name(file_data, "file");
|
|
if(paste_flag)
|
|
curl_mime_filename(file_data, "-");
|
|
curl_easy_perform(easy_handle);
|
|
if(!silent_flag)
|
|
putchar('\n');
|
|
puts(buffer);
|
|
curl_mime_free(mime);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
upload_file_scp(int argc, char **argv)
|
|
{
|
|
curl_easy_setopt(easy_handle, CURLOPT_SSH_PRIVATE_KEYFILE, ssh_key_path);
|
|
|
|
char path[256];
|
|
char *filename = argv[optind];
|
|
|
|
curl_easy_setopt(easy_handle, CURLOPT_UPLOAD, true);
|
|
FILE *fp = fopen(filename, "r");
|
|
if(fp == NULL) {
|
|
fprintf(stderr, "%s", strerror(errno));
|
|
exit(-1);
|
|
}
|
|
|
|
struct stat st;
|
|
stat(argv[optind], &st);
|
|
snprintf(path, 256, "%s/%s", server, filename);
|
|
curl_easy_setopt(easy_handle, CURLOPT_READDATA, fp);
|
|
curl_easy_setopt(
|
|
easy_handle, CURLOPT_INFILESIZE_LARGE, (curl_off_t)st.st_size);
|
|
|
|
int ret = curl_easy_perform(easy_handle);
|
|
putchar('\n');
|
|
if(ret != 0) {
|
|
fprintf(stderr, "%i: %s\n", ret, curl_easy_strerror(ret));
|
|
}
|
|
fclose(fp);
|
|
return 0;
|
|
}
|