main.c

   1 #include <netinet/in.h>
   2 #include <stdio.h>
   3 #include <string.h>
   4 #include <event2/event.h>
   5 #include <event2/dns.h>
   6 #include <event2/util.h>
   7 #include "proxy.h"
   8 #include "conn.h"
   9 #include "log.h"
  10 #include "util.h"
  11
  12 int
  13 main(int argc, char **argv)
  14 {
  15         struct sockaddr_in sin;
  16         struct event_base *base;
  17         struct evdns_base *dns;
  18
  19         base = event_base_new();
  20         dns = evdns_base_new(base, 1);
  21         log_set_min_level(LOG_DEBUG);
  22         log_set_file(NULL);
  23
  24         if (argc >= 2) {
  25                 struct url *url;
  26                 enum socks_ver socksv;
  27
  28                 url = url_tokenize(argv[1]);
  29                 if (!url) {
  30                         log_error("shim: bad socks server, %s", argv[1]);
  31                         return 0;
  32                 }
  33                 if (url->port < 0)
  34                         url->port = 1080;
  35
  36                 if (!evutil_ascii_strcasecmp(url->scheme, "socks4"))
  37                         socksv = SOCKS_4;
  38                 else if (!evutil_ascii_strcasecmp(url->scheme, "socks4a"))
  39                         socksv = SOCKS_4a;
  40                 else {
  41                         log_error("shim: unknown socks version, %s",
  42                                   url->scheme);
  43                         return 0;
  44                 }
  45
  46                 if (conn_set_socks_server(url->host, url->port, socksv) < 0)
  47                         return 0;
  48
  49                 url_free(url);
  50         }
  51
  52         memset(&sin, 0, sizeof(sin));
  53         sin.sin_family = AF_INET;
  54         sin.sin_port = htons(8888);
  55
  56         proxy_init(base, dns, (struct sockaddr *)&sin, sizeof(sin));
  57
  58         event_base_dispatch(base);
  59
  60         return 0;
  61 }