3 files changed, 36 insertions, 3 deletions
diff --git a/src/main.c b/src/main.c
index e9d22f1..ecd9381 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,10 +1,12 @@
 #include <GL/glew.h>
 #include <GLFW/glfw3.h>
 #include <curl/curl.h>
+#include <limits.h>
 #include <locale.h>
 #include <stdio.h>
 #include <stdlib.h>
 
+#include "toml.h"
 #define MAX_VERTEX_BUFFER 512 * 1024
 #define MAX_ELEMENT_BUFFER 128 * 1024
 #define NK_INCLUDE_FIXED_TYPES
@@ -29,6 +31,7 @@
 #include "unifont.h"
 
 extern int win_width, win_height;
+static char *cookie;
 extern void load_ui(struct ui_struct *);
 static void error_callback(int e, const char *d) {
   printf("Error %d: %s\n", e, d);
@@ -40,8 +43,34 @@ int main(void) {
   /* Set locale*/
   setlocale(LC_ALL, ".UTF-8");
 
+  /* Parse config */
+  // NOTICE: string parsed from toml should always be freed afterwards!!!
+  {
+    FILE *fp;
+    const char *config_file = "config.toml";
+    char errbuf[UCHAR_MAX];
+
+    fp = fopen(config_file, "r");
+    if (!fp) {
+      fprintf(stderr, "[tomlc99] Cannot open %s, applying default config...\n",
+              config_file);
+    } else {
+      toml_table_t *conf = toml_parse_file(fp, errbuf, sizeof(errbuf));
+      fclose(fp);
+
+      if (!conf) {
+        fprintf(stderr, "[tomlc99] Cannot parse %s\n", config_file);
+      }
+      toml_datum_t cookie_datum = toml_string_in(conf, "cookie");
+      if (cookie_datum.ok) {
+        printf("[tomlc99] Found cookie string in config.toml, applying...\n");
+        cookie = cookie_datum.u.s;
+      }
+    }
+  }
+
   /* Curl */
-  curl_init();
+  curl_init(cookie);
 
   /* GLFW */
   struct nk_glfw glfw = {0};
diff --git a/src/process_url.c b/src/process_url.c
index bf2edfd..e51da98 100644
--- a/src/process_url.c
+++ b/src/process_url.c
@@ -39,6 +39,7 @@ mtx_t mtx;
 cnd_t cnd;
 bool corrupted;
 static const char *outdir_g, *referer_g;
+static char *cookie_g;
 static callback_t callback_g;
 static callback_struct_t *p_callback_struct_g;
 static CURLU *h;
@@ -113,6 +114,7 @@ static void curl_easy_setcommonopts(CURL *curl) {
   /* enable all supported built-in compressions,
    * since serveral sites enable gzip encoding */
   curl_easy_setopt(curl, CURLOPT_ACCEPT_ENCODING, "");
+  curl_easy_setopt(curl, CURLOPT_COOKIE, cookie_g);
 }
 
 static int progress_callback(void *clientp, curl_off_t dltotal,
@@ -440,10 +442,11 @@ static char *callback_struct_convert_fullpath(char *filename) {
   return tmp;
 }
 
-void curl_init(curl_conf_t *curl) {
+void curl_init(char *cookie) {
   curl_global_init(CURL_GLOBAL_ALL);
   h = curl_url();
   dl_queue = create_queue();
+  cookie_g = cookie;
   mtx_init(&mtx, mtx_plain);
   cnd_init(&cnd);
 }
@@ -467,6 +470,7 @@ void curl_cleanup(status_t *stat) {
     cnd_destroy(&cnd);
   }
   free_queue(&dl_queue);
+  free_and_nullify(cookie_g);
   curl_url_cleanup(h);
   curl_global_cleanup();
 }
diff --git a/src/process_url.h b/src/process_url.h
index 25caef6..5343c9e 100644
--- a/src/process_url.h
+++ b/src/process_url.h
@@ -52,7 +52,7 @@ typedef struct callback_struct {
 
 typedef int (*callback_t)(callback_struct_t *);
 
-void curl_init();
+void curl_init(char *);
 
 void curl_cleanup(status_t *);