You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
4.5 KiB
139 lines
4.5 KiB
/* Copyright StrongLoop, Inc. All rights reserved. |
|
* |
|
* Permission is hereby granted, free of charge, to any person obtaining a copy |
|
* of this software and associated documentation files (the "Software"), to |
|
* deal in the Software without restriction, including without limitation the |
|
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or |
|
* sell copies of the Software, and to permit persons to whom the Software is |
|
* furnished to do so, subject to the following conditions: |
|
* |
|
* The above copyright notice and this permission notice shall be included in |
|
* all copies or substantial portions of the Software. |
|
* |
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
|
* IN THE SOFTWARE. |
|
*/ |
|
|
|
#ifndef DEFS_H_ |
|
#define DEFS_H_ |
|
|
|
#include "s5.h" |
|
#include "uv.h" |
|
|
|
#include <assert.h> |
|
#include <netinet/in.h> /* sockaddr_in, sockaddr_in6 */ |
|
#include <stddef.h> /* size_t, ssize_t */ |
|
#include <stdint.h> |
|
#include <sys/socket.h> /* sockaddr */ |
|
|
|
struct client_ctx; |
|
|
|
typedef struct { |
|
const char *bind_host; |
|
unsigned short bind_port; |
|
unsigned int idle_timeout; |
|
} server_config; |
|
|
|
typedef struct { |
|
unsigned int idle_timeout; /* Connection idle timeout in ms. */ |
|
uv_tcp_t tcp_handle; |
|
uv_loop_t *loop; |
|
} server_ctx; |
|
|
|
typedef struct { |
|
unsigned char rdstate; |
|
unsigned char wrstate; |
|
unsigned int idle_timeout; |
|
struct client_ctx *client; /* Backlink to owning client context. */ |
|
ssize_t result; |
|
union { |
|
uv_handle_t handle; |
|
uv_stream_t stream; |
|
uv_tcp_t tcp; |
|
uv_udp_t udp; |
|
} handle; |
|
uv_timer_t timer_handle; /* For detecting timeouts. */ |
|
uv_write_t write_req; |
|
/* We only need one of these at a time so make them share memory. */ |
|
union { |
|
uv_getaddrinfo_t addrinfo_req; |
|
uv_connect_t connect_req; |
|
uv_req_t req; |
|
struct sockaddr_in6 addr6; |
|
struct sockaddr_in addr4; |
|
struct sockaddr addr; |
|
char buf[2048]; /* Scratch space. Used to read data into. */ |
|
} t; |
|
} conn; |
|
|
|
typedef struct client_ctx { |
|
unsigned int state; |
|
server_ctx *sx; /* Backlink to owning server context. */ |
|
s5_ctx parser; /* The SOCKS protocol parser. */ |
|
conn incoming; /* Connection with the SOCKS client. */ |
|
conn outgoing; /* Connection with upstream. */ |
|
} client_ctx; |
|
|
|
/* server.c */ |
|
int server_run(const server_config *cf, uv_loop_t *loop); |
|
int can_auth_none(const server_ctx *sx, const client_ctx *cx); |
|
int can_auth_passwd(const server_ctx *sx, const client_ctx *cx); |
|
int can_access(const server_ctx *sx, |
|
const client_ctx *cx, |
|
const struct sockaddr *addr); |
|
|
|
/* client.c */ |
|
void client_finish_init(server_ctx *sx, client_ctx *cx); |
|
|
|
/* util.c */ |
|
#if defined(__GNUC__) |
|
# define ATTRIBUTE_FORMAT_PRINTF(a, b) __attribute__((format(printf, a, b))) |
|
#else |
|
# define ATTRIBUTE_FORMAT_PRINTF(a, b) |
|
#endif |
|
void pr_info(const char *fmt, ...) ATTRIBUTE_FORMAT_PRINTF(1, 2); |
|
void pr_warn(const char *fmt, ...) ATTRIBUTE_FORMAT_PRINTF(1, 2); |
|
void pr_err(const char *fmt, ...) ATTRIBUTE_FORMAT_PRINTF(1, 2); |
|
void *xmalloc(size_t size); |
|
|
|
/* main.c */ |
|
const char *_getprogname(void); |
|
|
|
/* getopt.c */ |
|
#if !HAVE_UNISTD_H |
|
extern char *optarg; |
|
int getopt(int argc, char **argv, const char *options); |
|
#endif |
|
|
|
/* ASSERT() is for debug checks, CHECK() for run-time sanity checks. |
|
* DEBUG_CHECKS is for expensive debug checks that we only want to |
|
* enable in debug builds but still want type-checked by the compiler |
|
* in release builds. |
|
*/ |
|
#if defined(NDEBUG) |
|
# define ASSERT(exp) |
|
# define CHECK(exp) do { if (!(exp)) abort(); } while (0) |
|
# define DEBUG_CHECKS (0) |
|
#else |
|
# define ASSERT(exp) assert(exp) |
|
# define CHECK(exp) assert(exp) |
|
# define DEBUG_CHECKS (1) |
|
#endif |
|
|
|
#define UNREACHABLE() CHECK(!"Unreachable code reached.") |
|
|
|
/* This macro looks complicated but it's not: it calculates the address |
|
* of the embedding struct through the address of the embedded struct. |
|
* In other words, if struct A embeds struct B, then we can obtain |
|
* the address of A by taking the address of B and subtracting the |
|
* field offset of B in A. |
|
*/ |
|
#define CONTAINER_OF(ptr, type, field) \ |
|
((type *) ((char *) (ptr) - ((char *) &((type *) 0)->field))) |
|
|
|
#endif /* DEFS_H_ */
|
|
|