From 6c7adadf6d5c85d5e280e965d4dee1563bf46821 Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Sun, 1 Dec 2019 18:05:16 +0100
Subject: Add 4chan posting

---
 src/DownloadUtils.cpp | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 src/DownloadUtils.cpp

(limited to 'src/DownloadUtils.cpp')

diff --git a/src/DownloadUtils.cpp b/src/DownloadUtils.cpp
new file mode 100644
index 0000000..a61d0a1
--- /dev/null
+++ b/src/DownloadUtils.cpp
@@ -0,0 +1,53 @@
+#include "../include/DownloadUtils.hpp"
+#include "../include/Program.h"
+#include <SFML/System/Clock.hpp>
+
+static int accumulate_string(char *data, int size, void *userdata) {
+    std::string *str = (std::string*)userdata;
+    str->append(data, size);
+    return 0;
+}
+
+namespace QuickMedia {
+    // TODO: Add timeout
+    DownloadResult download_to_string(const std::string &url, std::string &result, const std::vector<CommandArg> &additional_args, bool use_tor) {
+        sf::Clock timer;
+        std::vector<const char*> args;
+        if(use_tor)
+            args.push_back("torsocks");
+        args.insert(args.end(), { "curl", "-f", "-H", "Accept-Language: en-US,en;q=0.5", "--compressed", "-s", "-L" });
+        for(const CommandArg &arg : additional_args) {
+            args.push_back(arg.option.c_str());
+            args.push_back(arg.value.c_str());
+        }
+        args.push_back("--");
+        args.push_back(url.c_str());
+        args.push_back(nullptr);
+        if(exec_program(args.data(), accumulate_string, &result) != 0)
+            return DownloadResult::NET_ERR;
+        fprintf(stderr, "Download duration for %s: %d ms\n", url.c_str(), timer.getElapsedTime().asMilliseconds());
+        return DownloadResult::OK;
+    }
+
+    std::vector<CommandArg> create_command_args_from_form_data(const std::vector<FormData> &form_data) {
+        // TODO: This boundary value might need to change, depending on the content. What if the form data contains the boundary value?
+        const std::string boundary = "-----------------------------119561554312148213571335532670";
+        std::string form_data_str;
+        for(const FormData &form_data_item : form_data) {
+            form_data_str += boundary;
+            form_data_str += "\r\n";
+            // TODO: What if the form key contains " or \r\n ?
+            form_data_str += "Content-Disposition: form-data; name=\"" + form_data_item.key + "\"";
+            form_data_str += "\r\n\r\n";
+            // TODO: What is the value contains \r\n ?
+            form_data_str += form_data_item.value;
+            form_data_str += "\r\n";
+        }
+        // TODO: Verify if this should only be done also if the form data is empty
+        form_data_str += boundary + "--";
+        return {
+            CommandArg{"-H", "Content-Type: multipart/form-data; boundary=" + boundary},
+            CommandArg{"--data-binary", std::move(form_data_str)}
+        };
+    }
+}
\ No newline at end of file
-- 
cgit v1.2.3