summaryrefslogtreecommitdiffstats
path: root/fb-upload.c.in
diff options
context:
space:
mode:
authorFlorian Pritz <bluewind@xinu.at>2011-06-04 16:01:39 +0200
committerFlorian Pritz <bluewind@xinu.at>2011-06-04 16:01:39 +0200
commitb6189f1907e22aad1cda7ecf236529bd7b270734 (patch)
treeae7b218e5ec065e8232a9c0986635025e02ed3d2 /fb-upload.c.in
parent772c2fdc00319ce312845eba3df85115567f6aa5 (diff)
remove dependency on curl binary
fb-upload is now called fb-helper which supports downloading and uploading Signed-off-by: Florian Pritz <bluewind@xinu.at>
Diffstat (limited to 'fb-upload.c.in')
-rw-r--r--fb-upload.c.in237
1 files changed, 0 insertions, 237 deletions
diff --git a/fb-upload.c.in b/fb-upload.c.in
deleted file mode 100644
index b0a44ee..0000000
--- a/fb-upload.c.in
+++ /dev/null
@@ -1,237 +0,0 @@
-/*
- * Description: This is intended as a helper script for fb only.
- *
- * Synopsis: ./fb-upload <pastebin URL> <file>
- *
- * Author: Florian "Bluewind" Pritz <flo@xssn.at>
- *
- * Licensed under GPLv3
- * (see COPYING for full license text)
- */
-
-#include <stdio.h>
-#include <sys/time.h>
-#include <sys/stat.h>
-#include <sys/mman.h>
-#include <libgen.h>
-#include <string.h>
-#include <stdlib.h>
-#include <math.h>
-
-#include <curl/curl.h>
-#include <curl/types.h>
-#include <curl/easy.h>
-
-#define FORMAT_ARRAY_SIZE 5
-
-/* struct which holds the persistent data for progress_callback */
-struct progressData {
- struct timeval starttime;
- struct timeval last;
- double ullast;
- int lastStringLength;
-};
-
-/* load the contents of file fn into data */
-int load_file(const char *fn, char **data, size_t *data_size)
-{
- FILE *fp;
- size_t buf_size = 1024*1024; /* use 1MiB chunks */
- int size = 0;
-
- fp = fopen(fn, "rb");
- if (fp == NULL) {
- perror("load_file");
- return 1;
- }
-
- /* read the file in buf_size chunks and appened the data to *data */
- while (!feof(fp)) {
- *data = realloc(*data, *data_size + buf_size);
- if (*data == NULL) {
- perror("load_file");
- return 1;
- }
- size = fread(*data + *data_size, sizeof(char), buf_size, fp);
- *data_size += size;
- }
-
- fclose(fp);
-
- return 0;
-}
-
-void format_bytes(double bytes, char *buf)
-{
- double base = 0;
- int suffix_pos = 0;
- char suffix[FORMAT_ARRAY_SIZE][4] = {"B", "KiB", "MiB", "GiB", "TiB"};
-
- base = log(bytes) / log(1024);
- suffix_pos = (int)floor(base);
-
- if (suffix_pos >= FORMAT_ARRAY_SIZE)
- suffix_pos = FORMAT_ARRAY_SIZE-1;
-
- if (suffix_pos > 0) {
- snprintf(buf, 64, "%.2f%s", pow(1024, base - suffix_pos), suffix[suffix_pos]);
- } else {
- snprintf(buf, 64, "%.2fB", bytes);
- }
-}
-
-int progress_callback(void *cb_data,
- double dltotal, double dlnow,
- double ultotal, double ulnow)
-{
- struct timeval now;
- struct progressData *data = (struct progressData *)cb_data;
- double timeSpent = 0;
- int printed = 0;
- char speed[64];
- char total[64];
-
- if (0 == ulnow)
- return 0;
-
- /* upload complete; clean up */
- if (ulnow >= ultotal) {
- fprintf(stderr, "%*s\r", data->lastStringLength + 1, "");
- return 0;
- }
-
- gettimeofday(&now, NULL);
-
- /* only refresh once a second */
- if (data->last.tv_sec == now.tv_sec)
- return 0;
-
- /* calculate time between this and the last call in seconds */
- timeSpent =
- (double)(now.tv_sec - data->last.tv_sec) +
- (double)(now.tv_usec - data->last.tv_usec) / 1000000.0;
-
- format_bytes((ulnow - data->ullast) / timeSpent, (char *)&speed);
- format_bytes(ulnow, (char *)&total);
-
- /* print the progress */
- printed = fprintf(stderr,
- "\r%s/s uploaded: %.1f%% = %s",
- speed, /* upload speed */
- ulnow * 100.0 / ultotal, /* percent uploaded */
- total); /* total data uploaded */
-
- /* pad the string if the last one was longer to remove left over characters */
- if (data->lastStringLength > printed)
- fprintf(stderr, "%*s", data->lastStringLength - printed, "");
-
- /* save current values for the next run */
- data->ullast = ulnow;
- data->last = now;
- data->lastStringLength = printed;
-
- return 0;
-}
-
-int main(int argc, char *argv[])
-{
- CURL *curl;
- CURLcode res;
-
- char *userAgent = "fb-client/@VERSION@";
-
- struct progressData cb_data = {
- .starttime = {.tv_sec = 0, .tv_usec = 0},
- .last = {.tv_sec = 0, .tv_usec = 0},
- .ullast = 0.0,
- .lastStringLength = 0
- };
- struct stat statbuf;
-
- struct curl_httppost *formpost=NULL;
- struct curl_httppost *lastptr=NULL;
- struct curl_slist *headerlist=NULL;
- static const char buf[] = "Expect:";
- struct curl_forms forms[4];
-
- char *data = NULL;
- size_t data_size = 0;
-
- int ret = 0;
-
- /* simple arg check */
- if(argc != 3)
- return 1;
-
- if (curl_global_init(CURL_GLOBAL_ALL) != 0) {
- fprintf(stderr, "Error initializing curl");
- return 10;
- }
-
- if(stat(argv[2], &statbuf) == -1) {
- perror("fb-upload");
- return 1;
- }
-
- /* load files with 0 size (/proc files for example) into memory so we can
- * determine their real length */
- if (statbuf.st_size == 0) {
- if (load_file(argv[2], &data, &data_size) != 0) {
- return 1;
- }
- forms[0].option = CURLFORM_BUFFER;
- forms[0].value = basename(argv[2]);
- forms[1].option = CURLFORM_BUFFERPTR;
- forms[1].value = data;
- forms[2].option = CURLFORM_BUFFERLENGTH;
- forms[2].value = (char *)data_size;
- forms[3].option = CURLFORM_END;
- } else {
- forms[0].option = CURLFORM_FILE;
- forms[0].value = argv[2];
- forms[1].option = CURLFORM_END;
- }
-
- /* Fill in the file upload field */
- curl_formadd(&formpost,
- &lastptr,
- CURLFORM_COPYNAME, "file",
- CURLFORM_ARRAY, forms,
- CURLFORM_END);
-
- curl = curl_easy_init();
- /* initialize custom header list (stating that Expect: 100-continue is not
- wanted */
- headerlist = curl_slist_append(headerlist, buf);
- if(curl) {
- /* what URL that receives this POST */
- curl_easy_setopt(curl, CURLOPT_URL, argv[1]);
- curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headerlist);
- curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0);
- curl_easy_setopt(curl, CURLOPT_NETRC, CURL_NETRC_OPTIONAL);
- curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, &cb_data);
- curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);
- curl_easy_setopt(curl, CURLOPT_USERAGENT, userAgent);
- curl_easy_setopt(curl, CURLOPT_HTTPPOST, formpost);
- curl_easy_setopt(curl, CURLOPT_LOW_SPEED_LIMIT, (long)1);
- curl_easy_setopt(curl, CURLOPT_LOW_SPEED_TIME, (long)30);
- gettimeofday(&cb_data.starttime, NULL);
- res = curl_easy_perform(curl);
-
- if (res != 0) {
- fprintf(stderr, "\n%s\n", curl_easy_strerror(res));
- ret = 1;
- }
-
- /* cleanup */
- curl_easy_cleanup(curl);
- curl_formfree(formpost);
- curl_slist_free_all (headerlist);
- curl_global_cleanup();
- free(data);
- } else {
- fprintf(stderr, "Error initializing curl");
- ret = 1;
- }
- return ret;
-}