1 files changed, 124 insertions, 0 deletions
diff --git a/src/utils.c b/src/utils.c
new file mode 100644
index 0000000..0523451
--- /dev/null
+++ b/src/utils.c
@@ -0,0 +1,124 @@
+#include <ctype.h>
+#include <fcntl.h>
+#include <signal.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#if defined(__linux__) && !defined(__ANDROID__)
+// https://github.com/karelzak/util-linux/blob/master/misc-utils/kill.c
+const char *sys_signame[NSIG] = {
+    "zero", "HUP",   "INT",  "QUIT", "ILL",  "TRAP", "ABRT", "UNUSED", "FPE",
+    "KILL", "USR1",  "SEGV", "USR2", "PIPE", "ALRM", "TERM", "STKFLT", "CHLD",
+    "CONT", "STOP",  "TSTP", "TTIN", "TTOU", "URG",  "XCPU", "XFSZ",   "VTALRM",
+    "PROF", "WINCH", "IO",   "PWR",  "SYS",  NULL};
+#endif
+
+#if defined(_WIN32) || defined(__CYGWIN__)
+#include <shellapi.h>
+#include <windows.h>
+// https://github.com/mirror/newlib-cygwin/blob/master/winsup/cygwin/strsig.cc
+#ifndef NSIG
+#define NSIG 33
+#endif
+const char *sys_signame[NSIG] = {
+    "zero", "HUP",   "INT",  "QUIT", "ILL",  "TRAP", "IOT",  "EMT",  "FPE",
+    "KILL", "BUS",   "SEGV", "SYS",  "PIPE", "ALRM", "TERM", "URG",  "STOP",
+    "TSTP", "CONT",  "CHLD", "TTIN", "TTOU", "IO",   "XCPU", "XFSZ", "VTALRM",
+    "PROF", "WINCH", "PWR",  "USR1", "USR2", NULL};
+#endif
+
+void *xmalloc(size_t size) {
+  if (size == 0) return NULL;
+  void *p = malloc(size);
+  if (!p) abort();
+  return p;
+}
+
+void *xrealloc(void *p, size_t size) {
+  if ((size == 0) && (p == NULL)) return NULL;
+  p = realloc(p, size);
+  if (!p) abort();
+  return p;
+}
+
+char *uppercase(char *str) {
+  int i = 0;
+  do {
+    str[i] = (char)toupper(str[i]);
+  } while (str[i++] != '\0');
+  return str;
+}
+
+bool endswith(const char *str, const char *suffix) {
+  size_t str_len = strlen(str);
+  size_t suffix_len = strlen(suffix);
+  return str_len > suffix_len && !strcmp(str + (str_len - suffix_len), suffix);
+}
+
+int get_sig_name(int sig, char *buf, size_t len) {
+  int n =
+      snprintf(buf, len, "SIG%s", sig < NSIG ? sys_signame[sig] : "unknown");
+  uppercase(buf);
+  return n;
+}
+
+int get_sig(const char *sig_name) {
+  for (int sig = 1; sig < NSIG; sig++) {
+    const char *name = sys_signame[sig];
+    if (name != NULL && (strcasecmp(name, sig_name) == 0 ||
+                         strcasecmp(name, sig_name + 3) == 0))
+      return sig;
+  }
+  return atoi(sig_name);
+}
+
+bool fd_set_cloexec(const int fd) {
+  int flags = fcntl(fd, F_GETFD);
+  if (flags < 0) return false;
+  return (flags & FD_CLOEXEC) == 0 ||
+         fcntl(fd, F_SETFD, flags | FD_CLOEXEC) != -1;
+}
+
+int open_uri(char *uri) {
+#ifdef __APPLE__
+  char command[256];
+  sprintf(command, "open %s > /dev/null 2>&1", uri);
+  return system(command);
+#elif defined(_WIN32) || defined(__CYGWIN__)
+  return ShellExecute(0, 0, uri, 0, 0, SW_SHOW) > 32 ? 0 : 1;
+#else
+  // check if X server is running
+  if (system("xset -q > /dev/null 2>&1")) return 1;
+  char command[256];
+  sprintf(command, "xdg-open %s > /dev/null 2>&1", uri);
+  return system(command);
+#endif
+}
+
+// https://github.com/darkk/redsocks/blob/master/base64.c
+char *base64_encode(const unsigned char *buffer, size_t length) {
+  static const char b64[] =
+      "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+  char *ret, *dst;
+  unsigned i_bits = 0;
+  int i_shift = 0;
+  int bytes_remaining = (int)length;
+
+  ret = dst = xmalloc((size_t)(((length + 2) / 3 * 4) + 1));
+  while (bytes_remaining) {
+    i_bits = (i_bits << 8) + *buffer++;
+    bytes_remaining--;
+    i_shift += 8;
+
+    do {
+      *dst++ = b64[(i_bits << 6 >> i_shift) & 0x3f];
+      i_shift -= 6;
+    } while (i_shift > 6 || (bytes_remaining == 0 && i_shift > 0));
+  }
+  while ((dst - ret) & 3) *dst++ = '=';
+  *dst = '\0';
+
+  return ret;
+}