those who commitHEADmaster

1 files changed, 141 insertions, 125 deletions

diff --git a/src/Proxy/Proxy.c b/src/Proxy/Proxy.c
index 9908350..d2004aa 100644
--- a/src/Proxy/Proxy.c
+++ b/src/Proxy/Proxy.c
@@ -1,9 +1,9 @@
 #include "Proxy.h"
+#include <pthread.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <time.h>
-#include <pthread.h>
 
 Proxy *proxy_list = NULL;
 int proxy_count = 0;
@@ -13,76 +13,85 @@ int randomize_password = 0;
 char proxy_url[512] = {0};
 static pthread_mutex_t proxy_mutex = PTHREAD_MUTEX_INITIALIZER;
 
-static const char RAND_CHARS[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
+static const char RAND_CHARS[] =
+    "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
 
 static void generate_random_string(char *buf, size_t len) {
   for (size_t i = 0; i < len - 1; i++) {
-  buf[i] = RAND_CHARS[rand() % (sizeof(RAND_CHARS) - 1)];
+    buf[i] = RAND_CHARS[rand() % (sizeof(RAND_CHARS) - 1)];
   }
   buf[len - 1] = '\0';
 }
 
 void set_proxy_config(const char *proxy_str, int rand_user, int rand_pass) {
   if (proxy_str && proxy_str[0]) {
-  strncpy(proxy_url, proxy_str, sizeof(proxy_url) - 1);
-  proxy_url[sizeof(proxy_url) - 1] = '\0';
+    strncpy(proxy_url, proxy_str, sizeof(proxy_url) - 1);
+    proxy_url[sizeof(proxy_url) - 1] = '\0';
   }
   randomize_username = rand_user;
   randomize_password = rand_pass;
 }
 
 static Proxy parse_proxy_line(const char *line) {
-  Proxy proxy = {.type = PROXY_SOCKS5, .port = 0, .username[0] = '\0', .password[0] = '\0', .failures = 0};
+  Proxy proxy = {.type = PROXY_SOCKS5,
+                 .port = 0,
+                 .username[0] = '\0',
+                 .password[0] = '\0',
+                 .failures = 0};
   const char *host_start = NULL;
   const char *port_start = NULL;
 
   size_t len = strlen(line);
-  if (len == 0) return proxy;
+  if (len == 0)
+    return proxy;
 
   if (strncmp(line, "http://", 7) == 0) {
-  proxy.type = PROXY_HTTP;
-  host_start = line + 7;
+    proxy.type = PROXY_HTTP;
+    host_start = line + 7;
   } else if (strncmp(line, "socks5://", 9) == 0) {
-  proxy.type = PROXY_SOCKS5;
-  host_start = line + 9;
+    proxy.type = PROXY_SOCKS5;
+    host_start = line + 9;
   } else if (strncmp(line, "socks4://", 9) == 0) {
-  proxy.type = PROXY_SOCKS4;
-  host_start = line + 9;
+    proxy.type = PROXY_SOCKS4;
+    host_start = line + 9;
   } else {
-  host_start = line;
+    host_start = line;
   }
 
   const char *at = strchr(host_start, '@');
   if (at) {
-  char cred_buf[128];
-  size_t cred_len = at - host_start;
-  if (cred_len >= sizeof(cred_buf)) cred_len = sizeof(cred_buf) - 1;
-  strncpy(cred_buf, host_start, cred_len);
-  cred_buf[cred_len] = '\0';
-  
-  char *colon = strchr(cred_buf, ':');
-  if (colon) {
-    size_t user_len = colon - cred_buf;
-    if (user_len >= sizeof(proxy.username)) user_len = sizeof(proxy.username) - 1;
-    strncpy(proxy.username, cred_buf, user_len);
-    proxy.username[user_len] = '\0';
-    strncpy(proxy.password, colon + 1, sizeof(proxy.password) - 1);
-    proxy.password[sizeof(proxy.password) - 1] = '\0';
-  }
-  host_start = at + 1;
+    char cred_buf[128];
+    size_t cred_len = at - host_start;
+    if (cred_len >= sizeof(cred_buf))
+      cred_len = sizeof(cred_buf) - 1;
+    strncpy(cred_buf, host_start, cred_len);
+    cred_buf[cred_len] = '\0';
+
+    char *colon = strchr(cred_buf, ':');
+    if (colon) {
+      size_t user_len = colon - cred_buf;
+      if (user_len >= sizeof(proxy.username))
+        user_len = sizeof(proxy.username) - 1;
+      strncpy(proxy.username, cred_buf, user_len);
+      proxy.username[user_len] = '\0';
+      strncpy(proxy.password, colon + 1, sizeof(proxy.password) - 1);
+      proxy.password[sizeof(proxy.password) - 1] = '\0';
+    }
+    host_start = at + 1;
   }
 
   port_start = strchr(host_start, ':');
   if (port_start) {
-  char host_buf[256];
-  size_t host_len = port_start - host_start;
-  if (host_len >= sizeof(host_buf)) host_len = sizeof(host_buf) - 1;
-  strncpy(host_buf, host_start, host_len);
-  host_buf[host_len] = '\0';
-  snprintf(proxy.host, sizeof(proxy.host), "%.*s", (int)host_len, host_buf);
-  proxy.port = atoi(port_start + 1);
+    char host_buf[256];
+    size_t host_len = port_start - host_start;
+    if (host_len >= sizeof(host_buf))
+      host_len = sizeof(host_buf) - 1;
+    strncpy(host_buf, host_start, host_len);
+    host_buf[host_len] = '\0';
+    snprintf(proxy.host, sizeof(proxy.host), "%.*s", (int)host_len, host_buf);
+    proxy.port = atoi(port_start + 1);
   } else {
-  snprintf(proxy.host, sizeof(proxy.host), "%s", host_start);
+    snprintf(proxy.host, sizeof(proxy.host), "%s", host_start);
   }
 
   return proxy;
@@ -90,71 +99,73 @@ static Proxy parse_proxy_line(const char *line) {
 
 int load_proxy_list(const char *filename) {
   if (!filename || filename[0] == '\0') {
-  return 0;
+    return 0;
   }
 
   pthread_mutex_lock(&proxy_mutex);
 
   if (proxy_list) {
-  free(proxy_list);
-  proxy_list = NULL;
+    free(proxy_list);
+    proxy_list = NULL;
   }
   proxy_count = 0;
 
   FILE *file = fopen(filename, "r");
   if (!file) {
-  pthread_mutex_unlock(&proxy_mutex);
-  fprintf(stderr, "[WARN] Could not open proxy list file: %s\n", filename);
-  return -1;
+    pthread_mutex_unlock(&proxy_mutex);
+    fprintf(stderr, "[WARN] Could not open proxy list file: %s\n", filename);
+    return -1;
   }
 
   int capacity = 16;
   proxy_list = (Proxy *)malloc(capacity * sizeof(Proxy));
   if (!proxy_list) {
-  fclose(file);
-  return -1;
+    fclose(file);
+    return -1;
   }
   proxy_count = 0;
 
   char line[512];
   while (fgets(line, sizeof(line), file)) {
-  line[strcspn(line, "\r\n")] = 0;
+    line[strcspn(line, "\r\n")] = 0;
 
-  if (line[0] == '\0' || line[0] == '#') {
-    continue;
-  }
+    if (line[0] == '\0' || line[0] == '#') {
+      continue;
+    }
 
-  char *p = line;
-  while (*p == ' ' || *p == '\t') p++;
+    char *p = line;
+    while (*p == ' ' || *p == '\t')
+      p++;
 
-  char *end = p + strlen(p) - 1;
-  while (end > p && (*end == ' ' || *end == '\t')) {
-    *end = '\0';
-    end--;
-  }
+    char *end = p + strlen(p) - 1;
+    while (end > p && (*end == ' ' || *end == '\t')) {
+      *end = '\0';
+      end--;
+    }
 
-  if (p[0] == '\0') continue;
+    if (p[0] == '\0')
+      continue;
 
-  Proxy proxy = parse_proxy_line(p);
-  if (proxy.port == 0) {
-    continue;
-  }
+    Proxy proxy = parse_proxy_line(p);
+    if (proxy.port == 0) {
+      continue;
+    }
 
-  if (proxy_count >= capacity) {
-    capacity *= 2;
-    Proxy *new_list = (Proxy *)realloc(proxy_list, capacity * sizeof(Proxy));
-    if (!new_list) {
-    free(proxy_list);
-    proxy_list = NULL;
-    proxy_count = 0;
-    fclose(file);
-    pthread_mutex_unlock(&proxy_mutex);
-    return -1;
+    if (proxy_count >= capacity) {
+      capacity *= 2;
+      Proxy *new_list = (Proxy *)realloc(proxy_list, capacity * sizeof(Proxy));
+      if (!new_list) {
+        free(proxy_list);
+        proxy_list = NULL;
+        proxy_count = 0;
+        fclose(file);
+        pthread_mutex_unlock(&proxy_mutex);
+        return -1;
+      }
+      proxy_list = new_list;
     }
-    proxy_list = new_list;
-  }
 
-  proxy_list[proxy_count++] = proxy;
+    proxy_list[proxy_count++] = proxy;
   }
 
   fclose(file);
@@ -166,8 +177,8 @@ int load_proxy_list(const char *filename) {
 void free_proxy_list(void) {
   pthread_mutex_lock(&proxy_mutex);
   if (proxy_list) {
-  free(proxy_list);
-  proxy_list = NULL;
+    free(proxy_list);
+    proxy_list = NULL;
   }
   proxy_count = 0;
   pthread_mutex_unlock(&proxy_mutex);
@@ -176,8 +187,8 @@ void free_proxy_list(void) {
 Proxy *get_random_proxy(void) {
   pthread_mutex_lock(&proxy_mutex);
   if (proxy_count == 0) {
-  pthread_mutex_unlock(&proxy_mutex);
-  return NULL;
+    pthread_mutex_unlock(&proxy_mutex);
+    return NULL;
   }
 
   int start = rand() % proxy_count;
@@ -185,19 +196,19 @@ Proxy *get_random_proxy(void) {
   Proxy *selected = NULL;
 
   while (checked < proxy_count) {
-  int idx = (start + checked) % proxy_count;
-  if (proxy_list[idx].failures < max_proxy_retries) {
-    selected = &proxy_list[idx];
-    break;
-  }
-  checked++;
+    int idx = (start + checked) % proxy_count;
+    if (proxy_list[idx].failures < max_proxy_retries) {
+      selected = &proxy_list[idx];
+      break;
+    }
+    checked++;
   }
 
   if (!selected) {
-  for (int i = 0; i < proxy_count; i++) {
-    proxy_list[i].failures = 0;
-  }
-  selected = &proxy_list[rand() % proxy_count];
+    for (int i = 0; i < proxy_count; i++) {
+      proxy_list[i].failures = 0;
+    }
+    selected = &proxy_list[rand() % proxy_count];
   }
 
   pthread_mutex_unlock(&proxy_mutex);
@@ -205,7 +216,8 @@ Proxy *get_random_proxy(void) {
 }
 
 void record_proxy_failure(Proxy *proxy) {
-  if (!proxy) return;
+  if (!proxy)
+    return;
   pthread_mutex_lock(&proxy_mutex);
   proxy->failures++;
   pthread_mutex_unlock(&proxy_mutex);
@@ -213,45 +225,49 @@ void record_proxy_failure(Proxy *proxy) {
 
 void apply_proxy_settings(CURL *curl) {
   if (proxy_url[0] != '\0') {
-  curl_easy_setopt(curl, CURLOPT_PROXY, proxy_url);
-  if (strncmp(proxy_url, "socks5://", 9) == 0) {
-    curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS5);
-  } else if (strncmp(proxy_url, "socks4://", 9) == 0) {
-    curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS4A);
-  } else {
-    curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_HTTP);
-  }
+    curl_easy_setopt(curl, CURLOPT_PROXY, proxy_url);
+    if (strncmp(proxy_url, "socks5://", 9) == 0) {
+      curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS5);
+    } else if (strncmp(proxy_url, "socks4://", 9) == 0) {
+      curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS4A);
+    } else {
+      curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_HTTP);
+    }
 
-  if (randomize_username || randomize_password) {
-    char userpwd[256];
-    char username[32] = {0};
-    char password[32] = {0};
+    if (randomize_username || randomize_password) {
+      char userpwd[256];
+      char username[32] = {0};
+      char password[32] = {0};
 
-    if (randomize_username) generate_random_string(username, sizeof(username));
-    if (randomize_password) generate_random_string(password, sizeof(password));
+      if (randomize_username)
+        generate_random_string(username, sizeof(username));
+      if (randomize_password)
+        generate_random_string(password, sizeof(password));
 
-    snprintf(userpwd, sizeof(userpwd), "%s:%s", username, password);
-    curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD, userpwd);
-  }
-  } else if (proxy_count > 0) {
-  Proxy *proxy = get_random_proxy();
-  if (proxy) {
-    char proxy_url_buf[512];
-    snprintf(proxy_url_buf, sizeof(proxy_url_buf), "%s:%d", proxy->host, proxy->port);
-    curl_easy_setopt(curl, CURLOPT_PROXY, proxy_url_buf);
-    if (proxy->type == PROXY_HTTP) {
-    curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_HTTP);
-    } else if (proxy->type == PROXY_SOCKS4) {
-    curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS4A);
-    } else {
-    curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS5);
+      snprintf(userpwd, sizeof(userpwd), "%s:%s", username, password);
+      curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD, userpwd);
     }
-
-    if (proxy->username[0] != '\0' || proxy->password[0] != '\0') {
-    char userpwd[128];
-    snprintf(userpwd, sizeof(userpwd), "%s:%s", proxy->username, proxy->password);
-    curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD, userpwd);
+  } else if (proxy_count > 0) {
+    Proxy *proxy = get_random_proxy();
+    if (proxy) {
+      char proxy_url_buf[512];
+      snprintf(proxy_url_buf, sizeof(proxy_url_buf), "%s:%d", proxy->host,
+               proxy->port);
+      curl_easy_setopt(curl, CURLOPT_PROXY, proxy_url_buf);
+      if (proxy->type == PROXY_HTTP) {
+        curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_HTTP);
+      } else if (proxy->type == PROXY_SOCKS4) {
+        curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS4A);
+      } else {
+        curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS5);
+      }
+
+      if (proxy->username[0] != '\0' || proxy->password[0] != '\0') {
+        char userpwd[128];
+        snprintf(userpwd, sizeof(userpwd), "%s:%s", proxy->username,
+                 proxy->password);
+        curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD, userpwd);
+      }
     }
   }
-  }
 }