2019-06-05 15:57:52 +02:00
|
|
|
/* socks5.c -- generated by Trunnel v1.5.3.
|
2018-05-13 14:13:15 +02:00
|
|
|
* https://gitweb.torproject.org/trunnel.git
|
|
|
|
* You probably shouldn't edit this file.
|
|
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include "trunnel-impl.h"
|
|
|
|
|
|
|
|
#include "socks5.h"
|
|
|
|
|
|
|
|
#define TRUNNEL_SET_ERROR_CODE(obj) \
|
|
|
|
do { \
|
|
|
|
(obj)->trunnel_error_code_ = 1; \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
#if defined(__COVERITY__) || defined(__clang_analyzer__)
|
|
|
|
/* If we're running a static analysis tool, we don't want it to complain
|
|
|
|
* that some of our remaining-bytes checks are dead-code. */
|
|
|
|
int socks_deadcode_dummy__ = 0;
|
|
|
|
#define OR_DEADCODE_DUMMY || socks_deadcode_dummy__
|
|
|
|
#else
|
|
|
|
#define OR_DEADCODE_DUMMY
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#define CHECK_REMAINING(nbytes, label) \
|
|
|
|
do { \
|
|
|
|
if (remaining < (nbytes) OR_DEADCODE_DUMMY) { \
|
|
|
|
goto label; \
|
|
|
|
} \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
domainname_t *
|
|
|
|
domainname_new(void)
|
|
|
|
{
|
|
|
|
domainname_t *val = trunnel_calloc(1, sizeof(domainname_t));
|
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
domainname_clear(domainname_t *obj)
|
|
|
|
{
|
|
|
|
(void) obj;
|
|
|
|
TRUNNEL_DYNARRAY_WIPE(&obj->name);
|
|
|
|
TRUNNEL_DYNARRAY_CLEAR(&obj->name);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
domainname_free(domainname_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
|
|
|
domainname_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(domainname_t));
|
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
domainname_get_len(const domainname_t *inp)
|
|
|
|
{
|
|
|
|
return inp->len;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
domainname_set_len(domainname_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
inp->len = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
size_t
|
|
|
|
domainname_getlen_name(const domainname_t *inp)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_LEN(&inp->name);
|
|
|
|
}
|
|
|
|
|
|
|
|
char
|
|
|
|
domainname_get_name(domainname_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_GET(&inp->name, idx);
|
|
|
|
}
|
|
|
|
|
|
|
|
char
|
|
|
|
domainname_getconst_name(const domainname_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return domainname_get_name((domainname_t*)inp, idx);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
domainname_set_name(domainname_t *inp, size_t idx, char elt)
|
|
|
|
{
|
|
|
|
TRUNNEL_DYNARRAY_SET(&inp->name, idx, elt);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
domainname_add_name(domainname_t *inp, char elt)
|
|
|
|
{
|
|
|
|
#if SIZE_MAX >= UINT8_MAX
|
|
|
|
if (inp->name.n_ == UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
TRUNNEL_DYNARRAY_ADD(char, &inp->name, elt, {});
|
|
|
|
return 0;
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *
|
|
|
|
domainname_getarray_name(domainname_t *inp)
|
|
|
|
{
|
|
|
|
return inp->name.elts_;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
domainname_getconstarray_name(const domainname_t *inp)
|
|
|
|
{
|
|
|
|
return (const char *)domainname_getarray_name((domainname_t*)inp);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
domainname_setlen_name(domainname_t *inp, size_t newlen)
|
|
|
|
{
|
|
|
|
#if UINT8_MAX < SIZE_MAX
|
|
|
|
if (newlen > UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
return trunnel_string_setlen(&inp->name, newlen,
|
|
|
|
&inp->trunnel_error_code_);
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
domainname_getstr_name(domainname_t *inp)
|
|
|
|
{
|
|
|
|
return trunnel_string_getstr(&inp->name);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
domainname_setstr0_name(domainname_t *inp, const char *val, size_t len)
|
|
|
|
{
|
|
|
|
#if UINT8_MAX < SIZE_MAX
|
|
|
|
if (len > UINT8_MAX) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return trunnel_string_setstr0(&inp->name, val, len, &inp->trunnel_error_code_);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
domainname_setstr_name(domainname_t *inp, const char *val)
|
|
|
|
{
|
|
|
|
return domainname_setstr0_name(inp, val, strlen(val));
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
domainname_check(const domainname_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
|
|
|
if (TRUNNEL_DYNARRAY_LEN(&obj->name) != obj->len)
|
|
|
|
return "Length mismatch for name";
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
domainname_encoded_len(const domainname_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
|
|
|
if (NULL != domainname_check(obj))
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
|
|
/* Length of u8 len */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of char name[len] */
|
|
|
|
result += TRUNNEL_DYNARRAY_LEN(&obj->name);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
domainname_clear_errors(domainname_t *obj)
|
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
|
|
|
domainname_encode(uint8_t *output, const size_t avail, const domainname_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
const ssize_t encoded_len = domainname_encoded_len(obj);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (NULL != (msg = domainname_check(obj)))
|
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Encode u8 len */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->len));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode char name[len] */
|
|
|
|
{
|
|
|
|
size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->name);
|
|
|
|
trunnel_assert(obj->len == elt_len);
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < elt_len)
|
|
|
|
goto truncated;
|
|
|
|
if (elt_len)
|
|
|
|
memcpy(ptr, obj->name.elts_, elt_len);
|
|
|
|
written += elt_len; ptr += elt_len;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** As domainname_parse(), but do not allocate the output object.
|
|
|
|
*/
|
|
|
|
static ssize_t
|
|
|
|
domainname_parse_into(domainname_t *obj, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
|
|
|
/* Parse u8 len */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->len = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Parse char name[len] */
|
|
|
|
CHECK_REMAINING(obj->len, truncated);
|
|
|
|
if (domainname_setstr0_name(obj, (const char*)ptr, obj->len))
|
|
|
|
goto fail;
|
|
|
|
ptr += obj->len; remaining -= obj->len;
|
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
domainname_parse(domainname_t **output, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
ssize_t result;
|
|
|
|
*output = domainname_new();
|
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
|
|
|
result = domainname_parse_into(*output, input, len_in);
|
|
|
|
if (result < 0) {
|
|
|
|
domainname_free(*output);
|
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
socks4_client_request_t *
|
|
|
|
socks4_client_request_new(void)
|
|
|
|
{
|
|
|
|
socks4_client_request_t *val = trunnel_calloc(1, sizeof(socks4_client_request_t));
|
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
|
|
|
val->version = 4;
|
|
|
|
val->command = CMD_BIND;
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
socks4_client_request_clear(socks4_client_request_t *obj)
|
|
|
|
{
|
|
|
|
(void) obj;
|
|
|
|
trunnel_wipestr(obj->username);
|
|
|
|
trunnel_free(obj->username);
|
|
|
|
trunnel_wipestr(obj->socks4a_addr_hostname);
|
|
|
|
trunnel_free(obj->socks4a_addr_hostname);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
socks4_client_request_free(socks4_client_request_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
|
|
|
socks4_client_request_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks4_client_request_t));
|
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
socks4_client_request_get_version(const socks4_client_request_t *inp)
|
|
|
|
{
|
|
|
|
return inp->version;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_client_request_set_version(socks4_client_request_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
if (! ((val == 4))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks4_client_request_get_command(const socks4_client_request_t *inp)
|
|
|
|
{
|
|
|
|
return inp->command;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_client_request_set_command(socks4_client_request_t *inp, uint8_t val)
|
|
|
|
{
|
2018-05-15 11:49:07 +02:00
|
|
|
if (! ((val == CMD_BIND || val == CMD_CONNECT || val == CMD_RESOLVE || val == CMD_RESOLVE_PTR))) {
|
2018-05-13 14:13:15 +02:00
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->command = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint16_t
|
|
|
|
socks4_client_request_get_port(const socks4_client_request_t *inp)
|
|
|
|
{
|
|
|
|
return inp->port;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_client_request_set_port(socks4_client_request_t *inp, uint16_t val)
|
|
|
|
{
|
|
|
|
inp->port = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint32_t
|
|
|
|
socks4_client_request_get_addr(const socks4_client_request_t *inp)
|
|
|
|
{
|
|
|
|
return inp->addr;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_client_request_set_addr(socks4_client_request_t *inp, uint32_t val)
|
|
|
|
{
|
|
|
|
inp->addr = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks4_client_request_get_username(const socks4_client_request_t *inp)
|
|
|
|
{
|
|
|
|
return inp->username;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_client_request_set_username(socks4_client_request_t *inp, const char *val)
|
|
|
|
{
|
|
|
|
trunnel_free(inp->username);
|
|
|
|
if (NULL == (inp->username = trunnel_strdup(val))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks4_client_request_get_socks4a_addr_hostname(const socks4_client_request_t *inp)
|
|
|
|
{
|
|
|
|
return inp->socks4a_addr_hostname;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_client_request_set_socks4a_addr_hostname(socks4_client_request_t *inp, const char *val)
|
|
|
|
{
|
|
|
|
trunnel_free(inp->socks4a_addr_hostname);
|
|
|
|
if (NULL == (inp->socks4a_addr_hostname = trunnel_strdup(val))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks4_client_request_check(const socks4_client_request_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
|
|
|
if (! (obj->version == 4))
|
|
|
|
return "Integer out of bounds";
|
2018-05-15 11:49:07 +02:00
|
|
|
if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR))
|
2018-05-13 14:13:15 +02:00
|
|
|
return "Integer out of bounds";
|
|
|
|
if (NULL == obj->username)
|
|
|
|
return "Missing username";
|
|
|
|
switch (obj->addr) {
|
|
|
|
|
|
|
|
case 1:
|
|
|
|
case 2:
|
|
|
|
case 3:
|
|
|
|
case 4:
|
|
|
|
case 5:
|
|
|
|
case 6:
|
|
|
|
case 7:
|
|
|
|
case 8:
|
|
|
|
case 9:
|
|
|
|
case 10:
|
|
|
|
case 11:
|
|
|
|
case 12:
|
|
|
|
case 13:
|
|
|
|
case 14:
|
|
|
|
case 15:
|
|
|
|
case 16:
|
|
|
|
case 17:
|
|
|
|
case 18:
|
|
|
|
case 19:
|
|
|
|
case 20:
|
|
|
|
case 21:
|
|
|
|
case 22:
|
|
|
|
case 23:
|
|
|
|
case 24:
|
|
|
|
case 25:
|
|
|
|
case 26:
|
|
|
|
case 27:
|
|
|
|
case 28:
|
|
|
|
case 29:
|
|
|
|
case 30:
|
|
|
|
case 31:
|
|
|
|
case 32:
|
|
|
|
case 33:
|
|
|
|
case 34:
|
|
|
|
case 35:
|
|
|
|
case 36:
|
|
|
|
case 37:
|
|
|
|
case 38:
|
|
|
|
case 39:
|
|
|
|
case 40:
|
|
|
|
case 41:
|
|
|
|
case 42:
|
|
|
|
case 43:
|
|
|
|
case 44:
|
|
|
|
case 45:
|
|
|
|
case 46:
|
|
|
|
case 47:
|
|
|
|
case 48:
|
|
|
|
case 49:
|
|
|
|
case 50:
|
|
|
|
case 51:
|
|
|
|
case 52:
|
|
|
|
case 53:
|
|
|
|
case 54:
|
|
|
|
case 55:
|
|
|
|
case 56:
|
|
|
|
case 57:
|
|
|
|
case 58:
|
|
|
|
case 59:
|
|
|
|
case 60:
|
|
|
|
case 61:
|
|
|
|
case 62:
|
|
|
|
case 63:
|
|
|
|
case 64:
|
|
|
|
case 65:
|
|
|
|
case 66:
|
|
|
|
case 67:
|
|
|
|
case 68:
|
|
|
|
case 69:
|
|
|
|
case 70:
|
|
|
|
case 71:
|
|
|
|
case 72:
|
|
|
|
case 73:
|
|
|
|
case 74:
|
|
|
|
case 75:
|
|
|
|
case 76:
|
|
|
|
case 77:
|
|
|
|
case 78:
|
|
|
|
case 79:
|
|
|
|
case 80:
|
|
|
|
case 81:
|
|
|
|
case 82:
|
|
|
|
case 83:
|
|
|
|
case 84:
|
|
|
|
case 85:
|
|
|
|
case 86:
|
|
|
|
case 87:
|
|
|
|
case 88:
|
|
|
|
case 89:
|
|
|
|
case 90:
|
|
|
|
case 91:
|
|
|
|
case 92:
|
|
|
|
case 93:
|
|
|
|
case 94:
|
|
|
|
case 95:
|
|
|
|
case 96:
|
|
|
|
case 97:
|
|
|
|
case 98:
|
|
|
|
case 99:
|
|
|
|
case 100:
|
|
|
|
case 101:
|
|
|
|
case 102:
|
|
|
|
case 103:
|
|
|
|
case 104:
|
|
|
|
case 105:
|
|
|
|
case 106:
|
|
|
|
case 107:
|
|
|
|
case 108:
|
|
|
|
case 109:
|
|
|
|
case 110:
|
|
|
|
case 111:
|
|
|
|
case 112:
|
|
|
|
case 113:
|
|
|
|
case 114:
|
|
|
|
case 115:
|
|
|
|
case 116:
|
|
|
|
case 117:
|
|
|
|
case 118:
|
|
|
|
case 119:
|
|
|
|
case 120:
|
|
|
|
case 121:
|
|
|
|
case 122:
|
|
|
|
case 123:
|
|
|
|
case 124:
|
|
|
|
case 125:
|
|
|
|
case 126:
|
|
|
|
case 127:
|
|
|
|
case 128:
|
|
|
|
case 129:
|
|
|
|
case 130:
|
|
|
|
case 131:
|
|
|
|
case 132:
|
|
|
|
case 133:
|
|
|
|
case 134:
|
|
|
|
case 135:
|
|
|
|
case 136:
|
|
|
|
case 137:
|
|
|
|
case 138:
|
|
|
|
case 139:
|
|
|
|
case 140:
|
|
|
|
case 141:
|
|
|
|
case 142:
|
|
|
|
case 143:
|
|
|
|
case 144:
|
|
|
|
case 145:
|
|
|
|
case 146:
|
|
|
|
case 147:
|
|
|
|
case 148:
|
|
|
|
case 149:
|
|
|
|
case 150:
|
|
|
|
case 151:
|
|
|
|
case 152:
|
|
|
|
case 153:
|
|
|
|
case 154:
|
|
|
|
case 155:
|
|
|
|
case 156:
|
|
|
|
case 157:
|
|
|
|
case 158:
|
|
|
|
case 159:
|
|
|
|
case 160:
|
|
|
|
case 161:
|
|
|
|
case 162:
|
|
|
|
case 163:
|
|
|
|
case 164:
|
|
|
|
case 165:
|
|
|
|
case 166:
|
|
|
|
case 167:
|
|
|
|
case 168:
|
|
|
|
case 169:
|
|
|
|
case 170:
|
|
|
|
case 171:
|
|
|
|
case 172:
|
|
|
|
case 173:
|
|
|
|
case 174:
|
|
|
|
case 175:
|
|
|
|
case 176:
|
|
|
|
case 177:
|
|
|
|
case 178:
|
|
|
|
case 179:
|
|
|
|
case 180:
|
|
|
|
case 181:
|
|
|
|
case 182:
|
|
|
|
case 183:
|
|
|
|
case 184:
|
|
|
|
case 185:
|
|
|
|
case 186:
|
|
|
|
case 187:
|
|
|
|
case 188:
|
|
|
|
case 189:
|
|
|
|
case 190:
|
|
|
|
case 191:
|
|
|
|
case 192:
|
|
|
|
case 193:
|
|
|
|
case 194:
|
|
|
|
case 195:
|
|
|
|
case 196:
|
|
|
|
case 197:
|
|
|
|
case 198:
|
|
|
|
case 199:
|
|
|
|
case 200:
|
|
|
|
case 201:
|
|
|
|
case 202:
|
|
|
|
case 203:
|
|
|
|
case 204:
|
|
|
|
case 205:
|
|
|
|
case 206:
|
|
|
|
case 207:
|
|
|
|
case 208:
|
|
|
|
case 209:
|
|
|
|
case 210:
|
|
|
|
case 211:
|
|
|
|
case 212:
|
|
|
|
case 213:
|
|
|
|
case 214:
|
|
|
|
case 215:
|
|
|
|
case 216:
|
|
|
|
case 217:
|
|
|
|
case 218:
|
|
|
|
case 219:
|
|
|
|
case 220:
|
|
|
|
case 221:
|
|
|
|
case 222:
|
|
|
|
case 223:
|
|
|
|
case 224:
|
|
|
|
case 225:
|
|
|
|
case 226:
|
|
|
|
case 227:
|
|
|
|
case 228:
|
|
|
|
case 229:
|
|
|
|
case 230:
|
|
|
|
case 231:
|
|
|
|
case 232:
|
|
|
|
case 233:
|
|
|
|
case 234:
|
|
|
|
case 235:
|
|
|
|
case 236:
|
|
|
|
case 237:
|
|
|
|
case 238:
|
|
|
|
case 239:
|
|
|
|
case 240:
|
|
|
|
case 241:
|
|
|
|
case 242:
|
|
|
|
case 243:
|
|
|
|
case 244:
|
|
|
|
case 245:
|
|
|
|
case 246:
|
|
|
|
case 247:
|
|
|
|
case 248:
|
|
|
|
case 249:
|
|
|
|
case 250:
|
|
|
|
case 251:
|
|
|
|
case 252:
|
|
|
|
case 253:
|
|
|
|
case 254:
|
|
|
|
case 255:
|
|
|
|
if (NULL == obj->socks4a_addr_hostname)
|
|
|
|
return "Missing socks4a_addr_hostname";
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks4_client_request_encoded_len(const socks4_client_request_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
|
|
|
if (NULL != socks4_client_request_check(obj))
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
|
|
/* Length of u8 version IN [4] */
|
|
|
|
result += 1;
|
|
|
|
|
2018-05-15 11:49:07 +02:00
|
|
|
/* Length of u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR] */
|
2018-05-13 14:13:15 +02:00
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u16 port */
|
|
|
|
result += 2;
|
|
|
|
|
|
|
|
/* Length of u32 addr */
|
|
|
|
result += 4;
|
|
|
|
|
|
|
|
/* Length of nulterm username */
|
|
|
|
result += strlen(obj->username) + 1;
|
|
|
|
switch (obj->addr) {
|
|
|
|
|
|
|
|
case 1:
|
|
|
|
case 2:
|
|
|
|
case 3:
|
|
|
|
case 4:
|
|
|
|
case 5:
|
|
|
|
case 6:
|
|
|
|
case 7:
|
|
|
|
case 8:
|
|
|
|
case 9:
|
|
|
|
case 10:
|
|
|
|
case 11:
|
|
|
|
case 12:
|
|
|
|
case 13:
|
|
|
|
case 14:
|
|
|
|
case 15:
|
|
|
|
case 16:
|
|
|
|
case 17:
|
|
|
|
case 18:
|
|
|
|
case 19:
|
|
|
|
case 20:
|
|
|
|
case 21:
|
|
|
|
case 22:
|
|
|
|
case 23:
|
|
|
|
case 24:
|
|
|
|
case 25:
|
|
|
|
case 26:
|
|
|
|
case 27:
|
|
|
|
case 28:
|
|
|
|
case 29:
|
|
|
|
case 30:
|
|
|
|
case 31:
|
|
|
|
case 32:
|
|
|
|
case 33:
|
|
|
|
case 34:
|
|
|
|
case 35:
|
|
|
|
case 36:
|
|
|
|
case 37:
|
|
|
|
case 38:
|
|
|
|
case 39:
|
|
|
|
case 40:
|
|
|
|
case 41:
|
|
|
|
case 42:
|
|
|
|
case 43:
|
|
|
|
case 44:
|
|
|
|
case 45:
|
|
|
|
case 46:
|
|
|
|
case 47:
|
|
|
|
case 48:
|
|
|
|
case 49:
|
|
|
|
case 50:
|
|
|
|
case 51:
|
|
|
|
case 52:
|
|
|
|
case 53:
|
|
|
|
case 54:
|
|
|
|
case 55:
|
|
|
|
case 56:
|
|
|
|
case 57:
|
|
|
|
case 58:
|
|
|
|
case 59:
|
|
|
|
case 60:
|
|
|
|
case 61:
|
|
|
|
case 62:
|
|
|
|
case 63:
|
|
|
|
case 64:
|
|
|
|
case 65:
|
|
|
|
case 66:
|
|
|
|
case 67:
|
|
|
|
case 68:
|
|
|
|
case 69:
|
|
|
|
case 70:
|
|
|
|
case 71:
|
|
|
|
case 72:
|
|
|
|
case 73:
|
|
|
|
case 74:
|
|
|
|
case 75:
|
|
|
|
case 76:
|
|
|
|
case 77:
|
|
|
|
case 78:
|
|
|
|
case 79:
|
|
|
|
case 80:
|
|
|
|
case 81:
|
|
|
|
case 82:
|
|
|
|
case 83:
|
|
|
|
case 84:
|
|
|
|
case 85:
|
|
|
|
case 86:
|
|
|
|
case 87:
|
|
|
|
case 88:
|
|
|
|
case 89:
|
|
|
|
case 90:
|
|
|
|
case 91:
|
|
|
|
case 92:
|
|
|
|
case 93:
|
|
|
|
case 94:
|
|
|
|
case 95:
|
|
|
|
case 96:
|
|
|
|
case 97:
|
|
|
|
case 98:
|
|
|
|
case 99:
|
|
|
|
case 100:
|
|
|
|
case 101:
|
|
|
|
case 102:
|
|
|
|
case 103:
|
|
|
|
case 104:
|
|
|
|
case 105:
|
|
|
|
case 106:
|
|
|
|
case 107:
|
|
|
|
case 108:
|
|
|
|
case 109:
|
|
|
|
case 110:
|
|
|
|
case 111:
|
|
|
|
case 112:
|
|
|
|
case 113:
|
|
|
|
case 114:
|
|
|
|
case 115:
|
|
|
|
case 116:
|
|
|
|
case 117:
|
|
|
|
case 118:
|
|
|
|
case 119:
|
|
|
|
case 120:
|
|
|
|
case 121:
|
|
|
|
case 122:
|
|
|
|
case 123:
|
|
|
|
case 124:
|
|
|
|
case 125:
|
|
|
|
case 126:
|
|
|
|
case 127:
|
|
|
|
case 128:
|
|
|
|
case 129:
|
|
|
|
case 130:
|
|
|
|
case 131:
|
|
|
|
case 132:
|
|
|
|
case 133:
|
|
|
|
case 134:
|
|
|
|
case 135:
|
|
|
|
case 136:
|
|
|
|
case 137:
|
|
|
|
case 138:
|
|
|
|
case 139:
|
|
|
|
case 140:
|
|
|
|
case 141:
|
|
|
|
case 142:
|
|
|
|
case 143:
|
|
|
|
case 144:
|
|
|
|
case 145:
|
|
|
|
case 146:
|
|
|
|
case 147:
|
|
|
|
case 148:
|
|
|
|
case 149:
|
|
|
|
case 150:
|
|
|
|
case 151:
|
|
|
|
case 152:
|
|
|
|
case 153:
|
|
|
|
case 154:
|
|
|
|
case 155:
|
|
|
|
case 156:
|
|
|
|
case 157:
|
|
|
|
case 158:
|
|
|
|
case 159:
|
|
|
|
case 160:
|
|
|
|
case 161:
|
|
|
|
case 162:
|
|
|
|
case 163:
|
|
|
|
case 164:
|
|
|
|
case 165:
|
|
|
|
case 166:
|
|
|
|
case 167:
|
|
|
|
case 168:
|
|
|
|
case 169:
|
|
|
|
case 170:
|
|
|
|
case 171:
|
|
|
|
case 172:
|
|
|
|
case 173:
|
|
|
|
case 174:
|
|
|
|
case 175:
|
|
|
|
case 176:
|
|
|
|
case 177:
|
|
|
|
case 178:
|
|
|
|
case 179:
|
|
|
|
case 180:
|
|
|
|
case 181:
|
|
|
|
case 182:
|
|
|
|
case 183:
|
|
|
|
case 184:
|
|
|
|
case 185:
|
|
|
|
case 186:
|
|
|
|
case 187:
|
|
|
|
case 188:
|
|
|
|
case 189:
|
|
|
|
case 190:
|
|
|
|
case 191:
|
|
|
|
case 192:
|
|
|
|
case 193:
|
|
|
|
case 194:
|
|
|
|
case 195:
|
|
|
|
case 196:
|
|
|
|
case 197:
|
|
|
|
case 198:
|
|
|
|
case 199:
|
|
|
|
case 200:
|
|
|
|
case 201:
|
|
|
|
case 202:
|
|
|
|
case 203:
|
|
|
|
case 204:
|
|
|
|
case 205:
|
|
|
|
case 206:
|
|
|
|
case 207:
|
|
|
|
case 208:
|
|
|
|
case 209:
|
|
|
|
case 210:
|
|
|
|
case 211:
|
|
|
|
case 212:
|
|
|
|
case 213:
|
|
|
|
case 214:
|
|
|
|
case 215:
|
|
|
|
case 216:
|
|
|
|
case 217:
|
|
|
|
case 218:
|
|
|
|
case 219:
|
|
|
|
case 220:
|
|
|
|
case 221:
|
|
|
|
case 222:
|
|
|
|
case 223:
|
|
|
|
case 224:
|
|
|
|
case 225:
|
|
|
|
case 226:
|
|
|
|
case 227:
|
|
|
|
case 228:
|
|
|
|
case 229:
|
|
|
|
case 230:
|
|
|
|
case 231:
|
|
|
|
case 232:
|
|
|
|
case 233:
|
|
|
|
case 234:
|
|
|
|
case 235:
|
|
|
|
case 236:
|
|
|
|
case 237:
|
|
|
|
case 238:
|
|
|
|
case 239:
|
|
|
|
case 240:
|
|
|
|
case 241:
|
|
|
|
case 242:
|
|
|
|
case 243:
|
|
|
|
case 244:
|
|
|
|
case 245:
|
|
|
|
case 246:
|
|
|
|
case 247:
|
|
|
|
case 248:
|
|
|
|
case 249:
|
|
|
|
case 250:
|
|
|
|
case 251:
|
|
|
|
case 252:
|
|
|
|
case 253:
|
|
|
|
case 254:
|
|
|
|
case 255:
|
|
|
|
|
|
|
|
/* Length of nulterm socks4a_addr_hostname */
|
|
|
|
result += strlen(obj->socks4a_addr_hostname) + 1;
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_client_request_clear_errors(socks4_client_request_t *obj)
|
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
|
|
|
socks4_client_request_encode(uint8_t *output, const size_t avail, const socks4_client_request_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
const ssize_t encoded_len = socks4_client_request_encoded_len(obj);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (NULL != (msg = socks4_client_request_check(obj)))
|
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Encode u8 version IN [4] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
2018-05-15 11:49:07 +02:00
|
|
|
/* Encode u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR] */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->command));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u16 port */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 2)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint16(ptr, trunnel_htons(obj->port));
|
|
|
|
written += 2; ptr += 2;
|
|
|
|
|
|
|
|
/* Encode u32 addr */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 4)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint32(ptr, trunnel_htonl(obj->addr));
|
|
|
|
written += 4; ptr += 4;
|
|
|
|
|
|
|
|
/* Encode nulterm username */
|
|
|
|
{
|
|
|
|
size_t len = strlen(obj->username);
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < len + 1)
|
|
|
|
goto truncated;
|
|
|
|
memcpy(ptr, obj->username, len + 1);
|
|
|
|
ptr += len + 1; written += len + 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Encode union socks4a_addr[addr] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
switch (obj->addr) {
|
|
|
|
|
|
|
|
case 1:
|
|
|
|
case 2:
|
|
|
|
case 3:
|
|
|
|
case 4:
|
|
|
|
case 5:
|
|
|
|
case 6:
|
|
|
|
case 7:
|
|
|
|
case 8:
|
|
|
|
case 9:
|
|
|
|
case 10:
|
|
|
|
case 11:
|
|
|
|
case 12:
|
|
|
|
case 13:
|
|
|
|
case 14:
|
|
|
|
case 15:
|
|
|
|
case 16:
|
|
|
|
case 17:
|
|
|
|
case 18:
|
|
|
|
case 19:
|
|
|
|
case 20:
|
|
|
|
case 21:
|
|
|
|
case 22:
|
|
|
|
case 23:
|
|
|
|
case 24:
|
|
|
|
case 25:
|
|
|
|
case 26:
|
|
|
|
case 27:
|
|
|
|
case 28:
|
|
|
|
case 29:
|
|
|
|
case 30:
|
|
|
|
case 31:
|
|
|
|
case 32:
|
|
|
|
case 33:
|
|
|
|
case 34:
|
|
|
|
case 35:
|
|
|
|
case 36:
|
|
|
|
case 37:
|
|
|
|
case 38:
|
|
|
|
case 39:
|
|
|
|
case 40:
|
|
|
|
case 41:
|
|
|
|
case 42:
|
|
|
|
case 43:
|
|
|
|
case 44:
|
|
|
|
case 45:
|
|
|
|
case 46:
|
|
|
|
case 47:
|
|
|
|
case 48:
|
|
|
|
case 49:
|
|
|
|
case 50:
|
|
|
|
case 51:
|
|
|
|
case 52:
|
|
|
|
case 53:
|
|
|
|
case 54:
|
|
|
|
case 55:
|
|
|
|
case 56:
|
|
|
|
case 57:
|
|
|
|
case 58:
|
|
|
|
case 59:
|
|
|
|
case 60:
|
|
|
|
case 61:
|
|
|
|
case 62:
|
|
|
|
case 63:
|
|
|
|
case 64:
|
|
|
|
case 65:
|
|
|
|
case 66:
|
|
|
|
case 67:
|
|
|
|
case 68:
|
|
|
|
case 69:
|
|
|
|
case 70:
|
|
|
|
case 71:
|
|
|
|
case 72:
|
|
|
|
case 73:
|
|
|
|
case 74:
|
|
|
|
case 75:
|
|
|
|
case 76:
|
|
|
|
case 77:
|
|
|
|
case 78:
|
|
|
|
case 79:
|
|
|
|
case 80:
|
|
|
|
case 81:
|
|
|
|
case 82:
|
|
|
|
case 83:
|
|
|
|
case 84:
|
|
|
|
case 85:
|
|
|
|
case 86:
|
|
|
|
case 87:
|
|
|
|
case 88:
|
|
|
|
case 89:
|
|
|
|
case 90:
|
|
|
|
case 91:
|
|
|
|
case 92:
|
|
|
|
case 93:
|
|
|
|
case 94:
|
|
|
|
case 95:
|
|
|
|
case 96:
|
|
|
|
case 97:
|
|
|
|
case 98:
|
|
|
|
case 99:
|
|
|
|
case 100:
|
|
|
|
case 101:
|
|
|
|
case 102:
|
|
|
|
case 103:
|
|
|
|
case 104:
|
|
|
|
case 105:
|
|
|
|
case 106:
|
|
|
|
case 107:
|
|
|
|
case 108:
|
|
|
|
case 109:
|
|
|
|
case 110:
|
|
|
|
case 111:
|
|
|
|
case 112:
|
|
|
|
case 113:
|
|
|
|
case 114:
|
|
|
|
case 115:
|
|
|
|
case 116:
|
|
|
|
case 117:
|
|
|
|
case 118:
|
|
|
|
case 119:
|
|
|
|
case 120:
|
|
|
|
case 121:
|
|
|
|
case 122:
|
|
|
|
case 123:
|
|
|
|
case 124:
|
|
|
|
case 125:
|
|
|
|
case 126:
|
|
|
|
case 127:
|
|
|
|
case 128:
|
|
|
|
case 129:
|
|
|
|
case 130:
|
|
|
|
case 131:
|
|
|
|
case 132:
|
|
|
|
case 133:
|
|
|
|
case 134:
|
|
|
|
case 135:
|
|
|
|
case 136:
|
|
|
|
case 137:
|
|
|
|
case 138:
|
|
|
|
case 139:
|
|
|
|
case 140:
|
|
|
|
case 141:
|
|
|
|
case 142:
|
|
|
|
case 143:
|
|
|
|
case 144:
|
|
|
|
case 145:
|
|
|
|
case 146:
|
|
|
|
case 147:
|
|
|
|
case 148:
|
|
|
|
case 149:
|
|
|
|
case 150:
|
|
|
|
case 151:
|
|
|
|
case 152:
|
|
|
|
case 153:
|
|
|
|
case 154:
|
|
|
|
case 155:
|
|
|
|
case 156:
|
|
|
|
case 157:
|
|
|
|
case 158:
|
|
|
|
case 159:
|
|
|
|
case 160:
|
|
|
|
case 161:
|
|
|
|
case 162:
|
|
|
|
case 163:
|
|
|
|
case 164:
|
|
|
|
case 165:
|
|
|
|
case 166:
|
|
|
|
case 167:
|
|
|
|
case 168:
|
|
|
|
case 169:
|
|
|
|
case 170:
|
|
|
|
case 171:
|
|
|
|
case 172:
|
|
|
|
case 173:
|
|
|
|
case 174:
|
|
|
|
case 175:
|
|
|
|
case 176:
|
|
|
|
case 177:
|
|
|
|
case 178:
|
|
|
|
case 179:
|
|
|
|
case 180:
|
|
|
|
case 181:
|
|
|
|
case 182:
|
|
|
|
case 183:
|
|
|
|
case 184:
|
|
|
|
case 185:
|
|
|
|
case 186:
|
|
|
|
case 187:
|
|
|
|
case 188:
|
|
|
|
case 189:
|
|
|
|
case 190:
|
|
|
|
case 191:
|
|
|
|
case 192:
|
|
|
|
case 193:
|
|
|
|
case 194:
|
|
|
|
case 195:
|
|
|
|
case 196:
|
|
|
|
case 197:
|
|
|
|
case 198:
|
|
|
|
case 199:
|
|
|
|
case 200:
|
|
|
|
case 201:
|
|
|
|
case 202:
|
|
|
|
case 203:
|
|
|
|
case 204:
|
|
|
|
case 205:
|
|
|
|
case 206:
|
|
|
|
case 207:
|
|
|
|
case 208:
|
|
|
|
case 209:
|
|
|
|
case 210:
|
|
|
|
case 211:
|
|
|
|
case 212:
|
|
|
|
case 213:
|
|
|
|
case 214:
|
|
|
|
case 215:
|
|
|
|
case 216:
|
|
|
|
case 217:
|
|
|
|
case 218:
|
|
|
|
case 219:
|
|
|
|
case 220:
|
|
|
|
case 221:
|
|
|
|
case 222:
|
|
|
|
case 223:
|
|
|
|
case 224:
|
|
|
|
case 225:
|
|
|
|
case 226:
|
|
|
|
case 227:
|
|
|
|
case 228:
|
|
|
|
case 229:
|
|
|
|
case 230:
|
|
|
|
case 231:
|
|
|
|
case 232:
|
|
|
|
case 233:
|
|
|
|
case 234:
|
|
|
|
case 235:
|
|
|
|
case 236:
|
|
|
|
case 237:
|
|
|
|
case 238:
|
|
|
|
case 239:
|
|
|
|
case 240:
|
|
|
|
case 241:
|
|
|
|
case 242:
|
|
|
|
case 243:
|
|
|
|
case 244:
|
|
|
|
case 245:
|
|
|
|
case 246:
|
|
|
|
case 247:
|
|
|
|
case 248:
|
|
|
|
case 249:
|
|
|
|
case 250:
|
|
|
|
case 251:
|
|
|
|
case 252:
|
|
|
|
case 253:
|
|
|
|
case 254:
|
|
|
|
case 255:
|
|
|
|
|
|
|
|
/* Encode nulterm socks4a_addr_hostname */
|
|
|
|
{
|
|
|
|
size_t len = strlen(obj->socks4a_addr_hostname);
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < len + 1)
|
|
|
|
goto truncated;
|
|
|
|
memcpy(ptr, obj->socks4a_addr_hostname, len + 1);
|
|
|
|
ptr += len + 1; written += len + 1;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** As socks4_client_request_parse(), but do not allocate the output
|
|
|
|
* object.
|
|
|
|
*/
|
|
|
|
static ssize_t
|
|
|
|
socks4_client_request_parse_into(socks4_client_request_t *obj, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
|
|
|
/* Parse u8 version IN [4] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->version == 4))
|
|
|
|
goto fail;
|
|
|
|
|
2018-05-15 11:49:07 +02:00
|
|
|
/* Parse u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR] */
|
2018-05-13 14:13:15 +02:00
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->command = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
2018-05-15 11:49:07 +02:00
|
|
|
if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR))
|
2018-05-13 14:13:15 +02:00
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* Parse u16 port */
|
|
|
|
CHECK_REMAINING(2, truncated);
|
|
|
|
obj->port = trunnel_ntohs(trunnel_get_uint16(ptr));
|
|
|
|
remaining -= 2; ptr += 2;
|
|
|
|
|
|
|
|
/* Parse u32 addr */
|
|
|
|
CHECK_REMAINING(4, truncated);
|
|
|
|
obj->addr = trunnel_ntohl(trunnel_get_uint32(ptr));
|
|
|
|
remaining -= 4; ptr += 4;
|
|
|
|
|
|
|
|
/* Parse nulterm username */
|
|
|
|
{
|
|
|
|
uint8_t *eos = (uint8_t*)memchr(ptr, 0, remaining);
|
|
|
|
size_t memlen;
|
|
|
|
if (eos == NULL)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_assert(eos >= ptr);
|
|
|
|
trunnel_assert((size_t)(eos - ptr) < SIZE_MAX - 1);
|
|
|
|
memlen = ((size_t)(eos - ptr)) + 1;
|
|
|
|
if (!(obj->username = trunnel_malloc(memlen)))
|
|
|
|
goto fail;
|
|
|
|
memcpy(obj->username, ptr, memlen);
|
|
|
|
remaining -= memlen; ptr += memlen;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Parse union socks4a_addr[addr] */
|
|
|
|
switch (obj->addr) {
|
|
|
|
|
|
|
|
case 1:
|
|
|
|
case 2:
|
|
|
|
case 3:
|
|
|
|
case 4:
|
|
|
|
case 5:
|
|
|
|
case 6:
|
|
|
|
case 7:
|
|
|
|
case 8:
|
|
|
|
case 9:
|
|
|
|
case 10:
|
|
|
|
case 11:
|
|
|
|
case 12:
|
|
|
|
case 13:
|
|
|
|
case 14:
|
|
|
|
case 15:
|
|
|
|
case 16:
|
|
|
|
case 17:
|
|
|
|
case 18:
|
|
|
|
case 19:
|
|
|
|
case 20:
|
|
|
|
case 21:
|
|
|
|
case 22:
|
|
|
|
case 23:
|
|
|
|
case 24:
|
|
|
|
case 25:
|
|
|
|
case 26:
|
|
|
|
case 27:
|
|
|
|
case 28:
|
|
|
|
case 29:
|
|
|
|
case 30:
|
|
|
|
case 31:
|
|
|
|
case 32:
|
|
|
|
case 33:
|
|
|
|
case 34:
|
|
|
|
case 35:
|
|
|
|
case 36:
|
|
|
|
case 37:
|
|
|
|
case 38:
|
|
|
|
case 39:
|
|
|
|
case 40:
|
|
|
|
case 41:
|
|
|
|
case 42:
|
|
|
|
case 43:
|
|
|
|
case 44:
|
|
|
|
case 45:
|
|
|
|
case 46:
|
|
|
|
case 47:
|
|
|
|
case 48:
|
|
|
|
case 49:
|
|
|
|
case 50:
|
|
|
|
case 51:
|
|
|
|
case 52:
|
|
|
|
case 53:
|
|
|
|
case 54:
|
|
|
|
case 55:
|
|
|
|
case 56:
|
|
|
|
case 57:
|
|
|
|
case 58:
|
|
|
|
case 59:
|
|
|
|
case 60:
|
|
|
|
case 61:
|
|
|
|
case 62:
|
|
|
|
case 63:
|
|
|
|
case 64:
|
|
|
|
case 65:
|
|
|
|
case 66:
|
|
|
|
case 67:
|
|
|
|
case 68:
|
|
|
|
case 69:
|
|
|
|
case 70:
|
|
|
|
case 71:
|
|
|
|
case 72:
|
|
|
|
case 73:
|
|
|
|
case 74:
|
|
|
|
case 75:
|
|
|
|
case 76:
|
|
|
|
case 77:
|
|
|
|
case 78:
|
|
|
|
case 79:
|
|
|
|
case 80:
|
|
|
|
case 81:
|
|
|
|
case 82:
|
|
|
|
case 83:
|
|
|
|
case 84:
|
|
|
|
case 85:
|
|
|
|
case 86:
|
|
|
|
case 87:
|
|
|
|
case 88:
|
|
|
|
case 89:
|
|
|
|
case 90:
|
|
|
|
case 91:
|
|
|
|
case 92:
|
|
|
|
case 93:
|
|
|
|
case 94:
|
|
|
|
case 95:
|
|
|
|
case 96:
|
|
|
|
case 97:
|
|
|
|
case 98:
|
|
|
|
case 99:
|
|
|
|
case 100:
|
|
|
|
case 101:
|
|
|
|
case 102:
|
|
|
|
case 103:
|
|
|
|
case 104:
|
|
|
|
case 105:
|
|
|
|
case 106:
|
|
|
|
case 107:
|
|
|
|
case 108:
|
|
|
|
case 109:
|
|
|
|
case 110:
|
|
|
|
case 111:
|
|
|
|
case 112:
|
|
|
|
case 113:
|
|
|
|
case 114:
|
|
|
|
case 115:
|
|
|
|
case 116:
|
|
|
|
case 117:
|
|
|
|
case 118:
|
|
|
|
case 119:
|
|
|
|
case 120:
|
|
|
|
case 121:
|
|
|
|
case 122:
|
|
|
|
case 123:
|
|
|
|
case 124:
|
|
|
|
case 125:
|
|
|
|
case 126:
|
|
|
|
case 127:
|
|
|
|
case 128:
|
|
|
|
case 129:
|
|
|
|
case 130:
|
|
|
|
case 131:
|
|
|
|
case 132:
|
|
|
|
case 133:
|
|
|
|
case 134:
|
|
|
|
case 135:
|
|
|
|
case 136:
|
|
|
|
case 137:
|
|
|
|
case 138:
|
|
|
|
case 139:
|
|
|
|
case 140:
|
|
|
|
case 141:
|
|
|
|
case 142:
|
|
|
|
case 143:
|
|
|
|
case 144:
|
|
|
|
case 145:
|
|
|
|
case 146:
|
|
|
|
case 147:
|
|
|
|
case 148:
|
|
|
|
case 149:
|
|
|
|
case 150:
|
|
|
|
case 151:
|
|
|
|
case 152:
|
|
|
|
case 153:
|
|
|
|
case 154:
|
|
|
|
case 155:
|
|
|
|
case 156:
|
|
|
|
case 157:
|
|
|
|
case 158:
|
|
|
|
case 159:
|
|
|
|
case 160:
|
|
|
|
case 161:
|
|
|
|
case 162:
|
|
|
|
case 163:
|
|
|
|
case 164:
|
|
|
|
case 165:
|
|
|
|
case 166:
|
|
|
|
case 167:
|
|
|
|
case 168:
|
|
|
|
case 169:
|
|
|
|
case 170:
|
|
|
|
case 171:
|
|
|
|
case 172:
|
|
|
|
case 173:
|
|
|
|
case 174:
|
|
|
|
case 175:
|
|
|
|
case 176:
|
|
|
|
case 177:
|
|
|
|
case 178:
|
|
|
|
case 179:
|
|
|
|
case 180:
|
|
|
|
case 181:
|
|
|
|
case 182:
|
|
|
|
case 183:
|
|
|
|
case 184:
|
|
|
|
case 185:
|
|
|
|
case 186:
|
|
|
|
case 187:
|
|
|
|
case 188:
|
|
|
|
case 189:
|
|
|
|
case 190:
|
|
|
|
case 191:
|
|
|
|
case 192:
|
|
|
|
case 193:
|
|
|
|
case 194:
|
|
|
|
case 195:
|
|
|
|
case 196:
|
|
|
|
case 197:
|
|
|
|
case 198:
|
|
|
|
case 199:
|
|
|
|
case 200:
|
|
|
|
case 201:
|
|
|
|
case 202:
|
|
|
|
case 203:
|
|
|
|
case 204:
|
|
|
|
case 205:
|
|
|
|
case 206:
|
|
|
|
case 207:
|
|
|
|
case 208:
|
|
|
|
case 209:
|
|
|
|
case 210:
|
|
|
|
case 211:
|
|
|
|
case 212:
|
|
|
|
case 213:
|
|
|
|
case 214:
|
|
|
|
case 215:
|
|
|
|
case 216:
|
|
|
|
case 217:
|
|
|
|
case 218:
|
|
|
|
case 219:
|
|
|
|
case 220:
|
|
|
|
case 221:
|
|
|
|
case 222:
|
|
|
|
case 223:
|
|
|
|
case 224:
|
|
|
|
case 225:
|
|
|
|
case 226:
|
|
|
|
case 227:
|
|
|
|
case 228:
|
|
|
|
case 229:
|
|
|
|
case 230:
|
|
|
|
case 231:
|
|
|
|
case 232:
|
|
|
|
case 233:
|
|
|
|
case 234:
|
|
|
|
case 235:
|
|
|
|
case 236:
|
|
|
|
case 237:
|
|
|
|
case 238:
|
|
|
|
case 239:
|
|
|
|
case 240:
|
|
|
|
case 241:
|
|
|
|
case 242:
|
|
|
|
case 243:
|
|
|
|
case 244:
|
|
|
|
case 245:
|
|
|
|
case 246:
|
|
|
|
case 247:
|
|
|
|
case 248:
|
|
|
|
case 249:
|
|
|
|
case 250:
|
|
|
|
case 251:
|
|
|
|
case 252:
|
|
|
|
case 253:
|
|
|
|
case 254:
|
|
|
|
case 255:
|
|
|
|
|
|
|
|
/* Parse nulterm socks4a_addr_hostname */
|
|
|
|
{
|
|
|
|
uint8_t *eos = (uint8_t*)memchr(ptr, 0, remaining);
|
|
|
|
size_t memlen;
|
|
|
|
if (eos == NULL)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_assert(eos >= ptr);
|
|
|
|
trunnel_assert((size_t)(eos - ptr) < SIZE_MAX - 1);
|
|
|
|
memlen = ((size_t)(eos - ptr)) + 1;
|
|
|
|
if (!(obj->socks4a_addr_hostname = trunnel_malloc(memlen)))
|
|
|
|
goto fail;
|
|
|
|
memcpy(obj->socks4a_addr_hostname, ptr, memlen);
|
|
|
|
remaining -= memlen; ptr += memlen;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks4_client_request_parse(socks4_client_request_t **output, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
ssize_t result;
|
|
|
|
*output = socks4_client_request_new();
|
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
|
|
|
result = socks4_client_request_parse_into(*output, input, len_in);
|
|
|
|
if (result < 0) {
|
|
|
|
socks4_client_request_free(*output);
|
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
socks4_server_reply_t *
|
|
|
|
socks4_server_reply_new(void)
|
|
|
|
{
|
|
|
|
socks4_server_reply_t *val = trunnel_calloc(1, sizeof(socks4_server_reply_t));
|
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
socks4_server_reply_clear(socks4_server_reply_t *obj)
|
|
|
|
{
|
|
|
|
(void) obj;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
socks4_server_reply_free(socks4_server_reply_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
|
|
|
socks4_server_reply_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks4_server_reply_t));
|
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
socks4_server_reply_get_version(const socks4_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return inp->version;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_server_reply_set_version(socks4_server_reply_t *inp, uint8_t val)
|
|
|
|
{
|
2018-09-12 10:18:43 +02:00
|
|
|
if (! ((val == 0 || val == 4))) {
|
2018-05-13 14:13:15 +02:00
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks4_server_reply_get_status(const socks4_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return inp->status;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_server_reply_set_status(socks4_server_reply_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
inp->status = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint16_t
|
|
|
|
socks4_server_reply_get_port(const socks4_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return inp->port;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_server_reply_set_port(socks4_server_reply_t *inp, uint16_t val)
|
|
|
|
{
|
|
|
|
inp->port = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint32_t
|
|
|
|
socks4_server_reply_get_addr(const socks4_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return inp->addr;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_server_reply_set_addr(socks4_server_reply_t *inp, uint32_t val)
|
|
|
|
{
|
|
|
|
inp->addr = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks4_server_reply_check(const socks4_server_reply_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
2018-09-12 10:18:43 +02:00
|
|
|
if (! (obj->version == 0 || obj->version == 4))
|
2018-05-13 14:13:15 +02:00
|
|
|
return "Integer out of bounds";
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks4_server_reply_encoded_len(const socks4_server_reply_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
|
|
|
if (NULL != socks4_server_reply_check(obj))
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
2018-09-12 10:18:43 +02:00
|
|
|
/* Length of u8 version IN [0, 4] */
|
2018-05-13 14:13:15 +02:00
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u8 status */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u16 port */
|
|
|
|
result += 2;
|
|
|
|
|
|
|
|
/* Length of u32 addr */
|
|
|
|
result += 4;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks4_server_reply_clear_errors(socks4_server_reply_t *obj)
|
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
|
|
|
socks4_server_reply_encode(uint8_t *output, const size_t avail, const socks4_server_reply_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
const ssize_t encoded_len = socks4_server_reply_encoded_len(obj);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (NULL != (msg = socks4_server_reply_check(obj)))
|
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
2018-09-12 10:18:43 +02:00
|
|
|
/* Encode u8 version IN [0, 4] */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u8 status */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->status));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u16 port */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 2)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint16(ptr, trunnel_htons(obj->port));
|
|
|
|
written += 2; ptr += 2;
|
|
|
|
|
|
|
|
/* Encode u32 addr */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 4)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint32(ptr, trunnel_htonl(obj->addr));
|
|
|
|
written += 4; ptr += 4;
|
|
|
|
|
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** As socks4_server_reply_parse(), but do not allocate the output
|
|
|
|
* object.
|
|
|
|
*/
|
|
|
|
static ssize_t
|
|
|
|
socks4_server_reply_parse_into(socks4_server_reply_t *obj, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
2018-09-12 10:18:43 +02:00
|
|
|
/* Parse u8 version IN [0, 4] */
|
2018-05-13 14:13:15 +02:00
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
2018-09-12 10:18:43 +02:00
|
|
|
if (! (obj->version == 0 || obj->version == 4))
|
2018-05-13 14:13:15 +02:00
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* Parse u8 status */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->status = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Parse u16 port */
|
|
|
|
CHECK_REMAINING(2, truncated);
|
|
|
|
obj->port = trunnel_ntohs(trunnel_get_uint16(ptr));
|
|
|
|
remaining -= 2; ptr += 2;
|
|
|
|
|
|
|
|
/* Parse u32 addr */
|
|
|
|
CHECK_REMAINING(4, truncated);
|
|
|
|
obj->addr = trunnel_ntohl(trunnel_get_uint32(ptr));
|
|
|
|
remaining -= 4; ptr += 4;
|
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks4_server_reply_parse(socks4_server_reply_t **output, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
ssize_t result;
|
|
|
|
*output = socks4_server_reply_new();
|
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
|
|
|
result = socks4_server_reply_parse_into(*output, input, len_in);
|
|
|
|
if (result < 0) {
|
|
|
|
socks4_server_reply_free(*output);
|
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
socks5_client_userpass_auth_t *
|
|
|
|
socks5_client_userpass_auth_new(void)
|
|
|
|
{
|
|
|
|
socks5_client_userpass_auth_t *val = trunnel_calloc(1, sizeof(socks5_client_userpass_auth_t));
|
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
|
|
|
val->version = 1;
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
socks5_client_userpass_auth_clear(socks5_client_userpass_auth_t *obj)
|
|
|
|
{
|
|
|
|
(void) obj;
|
|
|
|
TRUNNEL_DYNARRAY_WIPE(&obj->username);
|
|
|
|
TRUNNEL_DYNARRAY_CLEAR(&obj->username);
|
|
|
|
TRUNNEL_DYNARRAY_WIPE(&obj->passwd);
|
|
|
|
TRUNNEL_DYNARRAY_CLEAR(&obj->passwd);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
socks5_client_userpass_auth_free(socks5_client_userpass_auth_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
|
|
|
socks5_client_userpass_auth_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks5_client_userpass_auth_t));
|
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
socks5_client_userpass_auth_get_version(const socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return inp->version;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_set_version(socks5_client_userpass_auth_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
if (! ((val == 1))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks5_client_userpass_auth_get_username_len(const socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return inp->username_len;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_set_username_len(socks5_client_userpass_auth_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
inp->username_len = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
size_t
|
|
|
|
socks5_client_userpass_auth_getlen_username(const socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_LEN(&inp->username);
|
|
|
|
}
|
|
|
|
|
|
|
|
char
|
|
|
|
socks5_client_userpass_auth_get_username(socks5_client_userpass_auth_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_GET(&inp->username, idx);
|
|
|
|
}
|
|
|
|
|
|
|
|
char
|
|
|
|
socks5_client_userpass_auth_getconst_username(const socks5_client_userpass_auth_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return socks5_client_userpass_auth_get_username((socks5_client_userpass_auth_t*)inp, idx);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_set_username(socks5_client_userpass_auth_t *inp, size_t idx, char elt)
|
|
|
|
{
|
|
|
|
TRUNNEL_DYNARRAY_SET(&inp->username, idx, elt);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_add_username(socks5_client_userpass_auth_t *inp, char elt)
|
|
|
|
{
|
|
|
|
#if SIZE_MAX >= UINT8_MAX
|
|
|
|
if (inp->username.n_ == UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
TRUNNEL_DYNARRAY_ADD(char, &inp->username, elt, {});
|
|
|
|
return 0;
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *
|
|
|
|
socks5_client_userpass_auth_getarray_username(socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return inp->username.elts_;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks5_client_userpass_auth_getconstarray_username(const socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return (const char *)socks5_client_userpass_auth_getarray_username((socks5_client_userpass_auth_t*)inp);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_setlen_username(socks5_client_userpass_auth_t *inp, size_t newlen)
|
|
|
|
{
|
|
|
|
#if UINT8_MAX < SIZE_MAX
|
|
|
|
if (newlen > UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
return trunnel_string_setlen(&inp->username, newlen,
|
|
|
|
&inp->trunnel_error_code_);
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks5_client_userpass_auth_getstr_username(socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return trunnel_string_getstr(&inp->username);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_setstr0_username(socks5_client_userpass_auth_t *inp, const char *val, size_t len)
|
|
|
|
{
|
|
|
|
#if UINT8_MAX < SIZE_MAX
|
|
|
|
if (len > UINT8_MAX) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return trunnel_string_setstr0(&inp->username, val, len, &inp->trunnel_error_code_);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_setstr_username(socks5_client_userpass_auth_t *inp, const char *val)
|
|
|
|
{
|
|
|
|
return socks5_client_userpass_auth_setstr0_username(inp, val, strlen(val));
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks5_client_userpass_auth_get_passwd_len(const socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return inp->passwd_len;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_set_passwd_len(socks5_client_userpass_auth_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
inp->passwd_len = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
size_t
|
|
|
|
socks5_client_userpass_auth_getlen_passwd(const socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_LEN(&inp->passwd);
|
|
|
|
}
|
|
|
|
|
|
|
|
char
|
|
|
|
socks5_client_userpass_auth_get_passwd(socks5_client_userpass_auth_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_GET(&inp->passwd, idx);
|
|
|
|
}
|
|
|
|
|
|
|
|
char
|
|
|
|
socks5_client_userpass_auth_getconst_passwd(const socks5_client_userpass_auth_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return socks5_client_userpass_auth_get_passwd((socks5_client_userpass_auth_t*)inp, idx);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_set_passwd(socks5_client_userpass_auth_t *inp, size_t idx, char elt)
|
|
|
|
{
|
|
|
|
TRUNNEL_DYNARRAY_SET(&inp->passwd, idx, elt);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_add_passwd(socks5_client_userpass_auth_t *inp, char elt)
|
|
|
|
{
|
|
|
|
#if SIZE_MAX >= UINT8_MAX
|
|
|
|
if (inp->passwd.n_ == UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
TRUNNEL_DYNARRAY_ADD(char, &inp->passwd, elt, {});
|
|
|
|
return 0;
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *
|
|
|
|
socks5_client_userpass_auth_getarray_passwd(socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return inp->passwd.elts_;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks5_client_userpass_auth_getconstarray_passwd(const socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return (const char *)socks5_client_userpass_auth_getarray_passwd((socks5_client_userpass_auth_t*)inp);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_setlen_passwd(socks5_client_userpass_auth_t *inp, size_t newlen)
|
|
|
|
{
|
|
|
|
#if UINT8_MAX < SIZE_MAX
|
|
|
|
if (newlen > UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
return trunnel_string_setlen(&inp->passwd, newlen,
|
|
|
|
&inp->trunnel_error_code_);
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks5_client_userpass_auth_getstr_passwd(socks5_client_userpass_auth_t *inp)
|
|
|
|
{
|
|
|
|
return trunnel_string_getstr(&inp->passwd);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_setstr0_passwd(socks5_client_userpass_auth_t *inp, const char *val, size_t len)
|
|
|
|
{
|
|
|
|
#if UINT8_MAX < SIZE_MAX
|
|
|
|
if (len > UINT8_MAX) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return trunnel_string_setstr0(&inp->passwd, val, len, &inp->trunnel_error_code_);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_setstr_passwd(socks5_client_userpass_auth_t *inp, const char *val)
|
|
|
|
{
|
|
|
|
return socks5_client_userpass_auth_setstr0_passwd(inp, val, strlen(val));
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks5_client_userpass_auth_check(const socks5_client_userpass_auth_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
|
|
|
if (! (obj->version == 1))
|
|
|
|
return "Integer out of bounds";
|
|
|
|
if (TRUNNEL_DYNARRAY_LEN(&obj->username) != obj->username_len)
|
|
|
|
return "Length mismatch for username";
|
|
|
|
if (TRUNNEL_DYNARRAY_LEN(&obj->passwd) != obj->passwd_len)
|
|
|
|
return "Length mismatch for passwd";
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks5_client_userpass_auth_encoded_len(const socks5_client_userpass_auth_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
|
|
|
if (NULL != socks5_client_userpass_auth_check(obj))
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
|
|
/* Length of u8 version IN [1] */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u8 username_len */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of char username[username_len] */
|
|
|
|
result += TRUNNEL_DYNARRAY_LEN(&obj->username);
|
|
|
|
|
|
|
|
/* Length of u8 passwd_len */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of char passwd[passwd_len] */
|
|
|
|
result += TRUNNEL_DYNARRAY_LEN(&obj->passwd);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_userpass_auth_clear_errors(socks5_client_userpass_auth_t *obj)
|
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
|
|
|
socks5_client_userpass_auth_encode(uint8_t *output, const size_t avail, const socks5_client_userpass_auth_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
const ssize_t encoded_len = socks5_client_userpass_auth_encoded_len(obj);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (NULL != (msg = socks5_client_userpass_auth_check(obj)))
|
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Encode u8 version IN [1] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u8 username_len */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->username_len));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode char username[username_len] */
|
|
|
|
{
|
|
|
|
size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->username);
|
|
|
|
trunnel_assert(obj->username_len == elt_len);
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < elt_len)
|
|
|
|
goto truncated;
|
|
|
|
if (elt_len)
|
|
|
|
memcpy(ptr, obj->username.elts_, elt_len);
|
|
|
|
written += elt_len; ptr += elt_len;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Encode u8 passwd_len */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->passwd_len));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode char passwd[passwd_len] */
|
|
|
|
{
|
|
|
|
size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->passwd);
|
|
|
|
trunnel_assert(obj->passwd_len == elt_len);
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < elt_len)
|
|
|
|
goto truncated;
|
|
|
|
if (elt_len)
|
|
|
|
memcpy(ptr, obj->passwd.elts_, elt_len);
|
|
|
|
written += elt_len; ptr += elt_len;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** As socks5_client_userpass_auth_parse(), but do not allocate the
|
|
|
|
* output object.
|
|
|
|
*/
|
|
|
|
static ssize_t
|
|
|
|
socks5_client_userpass_auth_parse_into(socks5_client_userpass_auth_t *obj, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
|
|
|
/* Parse u8 version IN [1] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->version == 1))
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* Parse u8 username_len */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->username_len = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Parse char username[username_len] */
|
|
|
|
CHECK_REMAINING(obj->username_len, truncated);
|
|
|
|
if (socks5_client_userpass_auth_setstr0_username(obj, (const char*)ptr, obj->username_len))
|
|
|
|
goto fail;
|
|
|
|
ptr += obj->username_len; remaining -= obj->username_len;
|
|
|
|
|
|
|
|
/* Parse u8 passwd_len */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->passwd_len = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Parse char passwd[passwd_len] */
|
|
|
|
CHECK_REMAINING(obj->passwd_len, truncated);
|
|
|
|
if (socks5_client_userpass_auth_setstr0_passwd(obj, (const char*)ptr, obj->passwd_len))
|
|
|
|
goto fail;
|
|
|
|
ptr += obj->passwd_len; remaining -= obj->passwd_len;
|
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks5_client_userpass_auth_parse(socks5_client_userpass_auth_t **output, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
ssize_t result;
|
|
|
|
*output = socks5_client_userpass_auth_new();
|
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
|
|
|
result = socks5_client_userpass_auth_parse_into(*output, input, len_in);
|
|
|
|
if (result < 0) {
|
|
|
|
socks5_client_userpass_auth_free(*output);
|
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
socks5_client_version_t *
|
|
|
|
socks5_client_version_new(void)
|
|
|
|
{
|
|
|
|
socks5_client_version_t *val = trunnel_calloc(1, sizeof(socks5_client_version_t));
|
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
|
|
|
val->version = 5;
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
socks5_client_version_clear(socks5_client_version_t *obj)
|
|
|
|
{
|
|
|
|
(void) obj;
|
|
|
|
TRUNNEL_DYNARRAY_WIPE(&obj->methods);
|
|
|
|
TRUNNEL_DYNARRAY_CLEAR(&obj->methods);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
socks5_client_version_free(socks5_client_version_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
|
|
|
socks5_client_version_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks5_client_version_t));
|
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
socks5_client_version_get_version(const socks5_client_version_t *inp)
|
|
|
|
{
|
|
|
|
return inp->version;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_version_set_version(socks5_client_version_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
if (! ((val == 5))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks5_client_version_get_n_methods(const socks5_client_version_t *inp)
|
|
|
|
{
|
|
|
|
return inp->n_methods;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_version_set_n_methods(socks5_client_version_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
inp->n_methods = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
size_t
|
|
|
|
socks5_client_version_getlen_methods(const socks5_client_version_t *inp)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_LEN(&inp->methods);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
socks5_client_version_get_methods(socks5_client_version_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return TRUNNEL_DYNARRAY_GET(&inp->methods, idx);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
socks5_client_version_getconst_methods(const socks5_client_version_t *inp, size_t idx)
|
|
|
|
{
|
|
|
|
return socks5_client_version_get_methods((socks5_client_version_t*)inp, idx);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_version_set_methods(socks5_client_version_t *inp, size_t idx, uint8_t elt)
|
|
|
|
{
|
|
|
|
TRUNNEL_DYNARRAY_SET(&inp->methods, idx, elt);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_version_add_methods(socks5_client_version_t *inp, uint8_t elt)
|
|
|
|
{
|
|
|
|
#if SIZE_MAX >= UINT8_MAX
|
|
|
|
if (inp->methods.n_ == UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
TRUNNEL_DYNARRAY_ADD(uint8_t, &inp->methods, elt, {});
|
|
|
|
return 0;
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t *
|
|
|
|
socks5_client_version_getarray_methods(socks5_client_version_t *inp)
|
|
|
|
{
|
|
|
|
return inp->methods.elts_;
|
|
|
|
}
|
|
|
|
const uint8_t *
|
|
|
|
socks5_client_version_getconstarray_methods(const socks5_client_version_t *inp)
|
|
|
|
{
|
|
|
|
return (const uint8_t *)socks5_client_version_getarray_methods((socks5_client_version_t*)inp);
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_version_setlen_methods(socks5_client_version_t *inp, size_t newlen)
|
|
|
|
{
|
|
|
|
uint8_t *newptr;
|
|
|
|
#if UINT8_MAX < SIZE_MAX
|
|
|
|
if (newlen > UINT8_MAX)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
#endif
|
|
|
|
newptr = trunnel_dynarray_setlen(&inp->methods.allocated_,
|
|
|
|
&inp->methods.n_, inp->methods.elts_, newlen,
|
|
|
|
sizeof(inp->methods.elts_[0]), (trunnel_free_fn_t) NULL,
|
|
|
|
&inp->trunnel_error_code_);
|
|
|
|
if (newlen != 0 && newptr == NULL)
|
|
|
|
goto trunnel_alloc_failed;
|
|
|
|
inp->methods.elts_ = newptr;
|
|
|
|
return 0;
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks5_client_version_check(const socks5_client_version_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
|
|
|
if (! (obj->version == 5))
|
|
|
|
return "Integer out of bounds";
|
|
|
|
if (TRUNNEL_DYNARRAY_LEN(&obj->methods) != obj->n_methods)
|
|
|
|
return "Length mismatch for methods";
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks5_client_version_encoded_len(const socks5_client_version_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
|
|
|
if (NULL != socks5_client_version_check(obj))
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
|
|
/* Length of u8 version IN [5] */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u8 n_methods */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u8 methods[n_methods] */
|
|
|
|
result += TRUNNEL_DYNARRAY_LEN(&obj->methods);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_version_clear_errors(socks5_client_version_t *obj)
|
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
|
|
|
socks5_client_version_encode(uint8_t *output, const size_t avail, const socks5_client_version_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
const ssize_t encoded_len = socks5_client_version_encoded_len(obj);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (NULL != (msg = socks5_client_version_check(obj)))
|
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Encode u8 version IN [5] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u8 n_methods */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->n_methods));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u8 methods[n_methods] */
|
|
|
|
{
|
|
|
|
size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->methods);
|
|
|
|
trunnel_assert(obj->n_methods == elt_len);
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < elt_len)
|
|
|
|
goto truncated;
|
|
|
|
if (elt_len)
|
|
|
|
memcpy(ptr, obj->methods.elts_, elt_len);
|
|
|
|
written += elt_len; ptr += elt_len;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** As socks5_client_version_parse(), but do not allocate the output
|
|
|
|
* object.
|
|
|
|
*/
|
|
|
|
static ssize_t
|
|
|
|
socks5_client_version_parse_into(socks5_client_version_t *obj, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
|
|
|
/* Parse u8 version IN [5] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->version == 5))
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* Parse u8 n_methods */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->n_methods = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Parse u8 methods[n_methods] */
|
|
|
|
CHECK_REMAINING(obj->n_methods, truncated);
|
|
|
|
TRUNNEL_DYNARRAY_EXPAND(uint8_t, &obj->methods, obj->n_methods, {});
|
|
|
|
obj->methods.n_ = obj->n_methods;
|
|
|
|
if (obj->n_methods)
|
|
|
|
memcpy(obj->methods.elts_, ptr, obj->n_methods);
|
|
|
|
ptr += obj->n_methods; remaining -= obj->n_methods;
|
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
trunnel_alloc_failed:
|
|
|
|
return -1;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks5_client_version_parse(socks5_client_version_t **output, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
ssize_t result;
|
|
|
|
*output = socks5_client_version_new();
|
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
|
|
|
result = socks5_client_version_parse_into(*output, input, len_in);
|
|
|
|
if (result < 0) {
|
|
|
|
socks5_client_version_free(*output);
|
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
socks5_server_method_t *
|
|
|
|
socks5_server_method_new(void)
|
|
|
|
{
|
|
|
|
socks5_server_method_t *val = trunnel_calloc(1, sizeof(socks5_server_method_t));
|
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
|
|
|
val->version = 5;
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
socks5_server_method_clear(socks5_server_method_t *obj)
|
|
|
|
{
|
|
|
|
(void) obj;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
socks5_server_method_free(socks5_server_method_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
|
|
|
socks5_server_method_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks5_server_method_t));
|
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
|
|
|
socks5_server_method_get_version(const socks5_server_method_t *inp)
|
|
|
|
{
|
|
|
|
return inp->version;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_server_method_set_version(socks5_server_method_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
if (! ((val == 5))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks5_server_method_get_method(const socks5_server_method_t *inp)
|
|
|
|
{
|
|
|
|
return inp->method;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_server_method_set_method(socks5_server_method_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
inp->method = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
|
|
|
socks5_server_method_check(const socks5_server_method_t *obj)
|
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
|
|
|
if (! (obj->version == 5))
|
|
|
|
return "Integer out of bounds";
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks5_server_method_encoded_len(const socks5_server_method_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
|
|
|
if (NULL != socks5_server_method_check(obj))
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
|
|
/* Length of u8 version IN [5] */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u8 method */
|
|
|
|
result += 1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_server_method_clear_errors(socks5_server_method_t *obj)
|
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
|
|
|
socks5_server_method_encode(uint8_t *output, const size_t avail, const socks5_server_method_t *obj)
|
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
const ssize_t encoded_len = socks5_server_method_encoded_len(obj);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (NULL != (msg = socks5_server_method_check(obj)))
|
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Encode u8 version IN [5] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u8 method */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->method));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** As socks5_server_method_parse(), but do not allocate the output
|
|
|
|
* object.
|
|
|
|
*/
|
|
|
|
static ssize_t
|
|
|
|
socks5_server_method_parse_into(socks5_server_method_t *obj, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
|
|
|
/* Parse u8 version IN [5] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->version == 5))
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* Parse u8 method */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->method = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
socks5_server_method_parse(socks5_server_method_t **output, const uint8_t *input, const size_t len_in)
|
|
|
|
{
|
|
|
|
ssize_t result;
|
|
|
|
*output = socks5_server_method_new();
|
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
|
|
|
result = socks5_server_method_parse_into(*output, input, len_in);
|
|
|
|
if (result < 0) {
|
|
|
|
socks5_server_method_free(*output);
|
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_t *
|
|
|
|
socks5_server_userpass_auth_new(void)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_t *val = trunnel_calloc(1, sizeof(socks5_server_userpass_auth_t));
|
2018-05-13 14:13:15 +02:00
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
|
|
|
val->version = 1;
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_clear(socks5_server_userpass_auth_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
(void) obj;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_free(socks5_server_userpass_auth_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks5_server_userpass_auth_t));
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_get_version(const socks5_server_userpass_auth_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
return inp->version;
|
|
|
|
}
|
|
|
|
int
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_set_version(socks5_server_userpass_auth_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
if (! ((val == 1))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_get_status(const socks5_server_userpass_auth_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
return inp->status;
|
|
|
|
}
|
|
|
|
int
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_set_status(socks5_server_userpass_auth_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
inp->status = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_check(const socks5_server_userpass_auth_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
|
|
|
if (! (obj->version == 1))
|
|
|
|
return "Integer out of bounds";
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_encoded_len(const socks5_server_userpass_auth_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
2018-05-16 14:47:46 +02:00
|
|
|
if (NULL != socks5_server_userpass_auth_check(obj))
|
2018-05-13 14:13:15 +02:00
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
|
|
/* Length of u8 version IN [1] */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u8 status */
|
|
|
|
result += 1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_clear_errors(socks5_server_userpass_auth_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_encode(uint8_t *output, const size_t avail, const socks5_server_userpass_auth_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
2018-05-16 14:47:46 +02:00
|
|
|
const ssize_t encoded_len = socks5_server_userpass_auth_encoded_len(obj);
|
2018-05-13 14:13:15 +02:00
|
|
|
#endif
|
|
|
|
|
2018-05-16 14:47:46 +02:00
|
|
|
if (NULL != (msg = socks5_server_userpass_auth_check(obj)))
|
2018-05-13 14:13:15 +02:00
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* Encode u8 version IN [1] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u8 status */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->status));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2018-05-16 14:47:46 +02:00
|
|
|
/** As socks5_server_userpass_auth_parse(), but do not allocate the
|
2018-05-13 14:13:15 +02:00
|
|
|
* output object.
|
|
|
|
*/
|
|
|
|
static ssize_t
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_parse_into(socks5_server_userpass_auth_t *obj, const uint8_t *input, const size_t len_in)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
|
|
|
/* Parse u8 version IN [1] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->version == 1))
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* Parse u8 status */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->status = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_parse(socks5_server_userpass_auth_t **output, const uint8_t *input, const size_t len_in)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result;
|
2018-05-16 14:47:46 +02:00
|
|
|
*output = socks5_server_userpass_auth_new();
|
2018-05-13 14:13:15 +02:00
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
2018-05-16 14:47:46 +02:00
|
|
|
result = socks5_server_userpass_auth_parse_into(*output, input, len_in);
|
2018-05-13 14:13:15 +02:00
|
|
|
if (result < 0) {
|
2018-05-16 14:47:46 +02:00
|
|
|
socks5_server_userpass_auth_free(*output);
|
2018-05-13 14:13:15 +02:00
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_t *
|
|
|
|
socks5_client_request_new(void)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_t *val = trunnel_calloc(1, sizeof(socks5_client_request_t));
|
2018-05-13 14:13:15 +02:00
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
2018-06-26 18:31:26 +02:00
|
|
|
val->version = 5;
|
|
|
|
val->command = CMD_BIND;
|
2018-05-13 14:13:15 +02:00
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_clear(socks5_client_request_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
(void) obj;
|
2018-06-26 18:31:26 +02:00
|
|
|
domainname_free(obj->dest_addr_domainname);
|
|
|
|
obj->dest_addr_domainname = NULL;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_free(socks5_client_request_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks5_client_request_t));
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_client_request_get_version(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->version;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set_version(socks5_client_request_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! ((val == 5))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_client_request_get_command(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->command;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set_command(socks5_client_request_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! ((val == CMD_BIND || val == CMD_CONNECT || val == CMD_RESOLVE || val == CMD_RESOLVE_PTR || val == CMD_UDP_ASSOCIATE))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->command = val;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_client_request_get_reserved(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->reserved;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set_reserved(socks5_client_request_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! ((val == 0))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->reserved = val;
|
|
|
|
return 0;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_client_request_get_atype(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->atype;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set_atype(socks5_client_request_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->atype = val;
|
|
|
|
return 0;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
uint32_t
|
|
|
|
socks5_client_request_get_dest_addr_ipv4(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->dest_addr_ipv4;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set_dest_addr_ipv4(socks5_client_request_t *inp, uint32_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->dest_addr_ipv4 = val;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
size_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_getlen_dest_addr_ipv6(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
(void)inp; return 16;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_client_request_get_dest_addr_ipv6(socks5_client_request_t *inp, size_t idx)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
trunnel_assert(idx < 16);
|
|
|
|
return inp->dest_addr_ipv6[idx];
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_client_request_getconst_dest_addr_ipv6(const socks5_client_request_t *inp, size_t idx)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return socks5_client_request_get_dest_addr_ipv6((socks5_client_request_t*)inp, idx);
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set_dest_addr_ipv6(socks5_client_request_t *inp, size_t idx, uint8_t elt)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
trunnel_assert(idx < 16);
|
|
|
|
inp->dest_addr_ipv6[idx] = elt;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t *
|
|
|
|
socks5_client_request_getarray_dest_addr_ipv6(socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->dest_addr_ipv6;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
const uint8_t *
|
|
|
|
socks5_client_request_getconstarray_dest_addr_ipv6(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return (const uint8_t *)socks5_client_request_getarray_dest_addr_ipv6((socks5_client_request_t*)inp);
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
struct domainname_st *
|
|
|
|
socks5_client_request_get_dest_addr_domainname(socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->dest_addr_domainname;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
const struct domainname_st *
|
|
|
|
socks5_client_request_getconst_dest_addr_domainname(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return socks5_client_request_get_dest_addr_domainname((socks5_client_request_t*) inp);
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set_dest_addr_domainname(socks5_client_request_t *inp, struct domainname_st *val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
if (inp->dest_addr_domainname && inp->dest_addr_domainname != val)
|
|
|
|
domainname_free(inp->dest_addr_domainname);
|
|
|
|
return socks5_client_request_set0_dest_addr_domainname(inp, val);
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_set0_dest_addr_domainname(socks5_client_request_t *inp, struct domainname_st *val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->dest_addr_domainname = val;
|
|
|
|
return 0;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
uint16_t
|
|
|
|
socks5_client_request_get_dest_port(const socks5_client_request_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->dest_port;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_client_request_set_dest_port(socks5_client_request_t *inp, uint16_t val)
|
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->dest_port = val;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_check(const socks5_client_request_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! (obj->version == 5))
|
2018-05-13 14:13:15 +02:00
|
|
|
return "Integer out of bounds";
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR || obj->command == CMD_UDP_ASSOCIATE))
|
|
|
|
return "Integer out of bounds";
|
|
|
|
if (! (obj->reserved == 0))
|
|
|
|
return "Integer out of bounds";
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_DOMAINNAME:
|
|
|
|
{
|
|
|
|
const char *msg;
|
|
|
|
if (NULL != (msg = domainname_check(obj->dest_addr_domainname)))
|
|
|
|
return msg;
|
|
|
|
}
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
default:
|
|
|
|
return "Bad tag for union";
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_encoded_len(const socks5_client_request_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
if (NULL != socks5_client_request_check(obj))
|
2018-05-13 14:13:15 +02:00
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Length of u8 version IN [5] */
|
2018-05-13 14:13:15 +02:00
|
|
|
result += 1;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Length of u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR, CMD_UDP_ASSOCIATE] */
|
|
|
|
result += 1;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Length of u8 reserved IN [0] */
|
|
|
|
result += 1;
|
|
|
|
|
|
|
|
/* Length of u8 atype */
|
|
|
|
result += 1;
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
|
|
|
|
/* Length of u32 dest_addr_ipv4 */
|
|
|
|
result += 4;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
|
|
|
|
/* Length of u8 dest_addr_ipv6[16] */
|
|
|
|
result += 16;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_DOMAINNAME:
|
|
|
|
|
|
|
|
/* Length of struct domainname dest_addr_domainname */
|
|
|
|
result += domainname_encoded_len(obj->dest_addr_domainname);
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
default:
|
|
|
|
trunnel_assert(0);
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
|
|
|
|
/* Length of u16 dest_port */
|
|
|
|
result += 2;
|
2018-05-13 14:13:15 +02:00
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_clear_errors(socks5_client_request_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_encode(uint8_t *output, const size_t avail, const socks5_client_request_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
2018-06-26 18:31:26 +02:00
|
|
|
const ssize_t encoded_len = socks5_client_request_encoded_len(obj);
|
2018-05-13 14:13:15 +02:00
|
|
|
#endif
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
if (NULL != (msg = socks5_client_request_check(obj)))
|
2018-05-13 14:13:15 +02:00
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u8 version IN [5] */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR, CMD_UDP_ASSOCIATE] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->command));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode u8 reserved IN [0] */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
2018-06-26 18:31:26 +02:00
|
|
|
if (avail - written < 1)
|
2018-05-13 14:13:15 +02:00
|
|
|
goto truncated;
|
2018-06-26 18:31:26 +02:00
|
|
|
trunnel_set_uint8(ptr, (obj->reserved));
|
|
|
|
written += 1; ptr += 1;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u8 atype */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->atype));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode union dest_addr[atype] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
|
|
|
|
/* Encode u32 dest_addr_ipv4 */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 4)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint32(ptr, trunnel_htonl(obj->dest_addr_ipv4));
|
|
|
|
written += 4; ptr += 4;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
|
|
|
|
/* Encode u8 dest_addr_ipv6[16] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 16)
|
|
|
|
goto truncated;
|
|
|
|
memcpy(ptr, obj->dest_addr_ipv6, 16);
|
|
|
|
written += 16; ptr += 16;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_DOMAINNAME:
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode struct domainname dest_addr_domainname */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
2018-06-26 18:31:26 +02:00
|
|
|
result = domainname_encode(ptr, avail - written, obj->dest_addr_domainname);
|
2018-05-13 14:13:15 +02:00
|
|
|
if (result < 0)
|
|
|
|
goto fail; /* XXXXXXX !*/
|
|
|
|
written += result; ptr += result;
|
2018-06-26 18:31:26 +02:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
trunnel_assert(0);
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u16 dest_port */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 2)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint16(ptr, trunnel_htons(obj->dest_port));
|
|
|
|
written += 2; ptr += 2;
|
|
|
|
|
2018-05-13 14:13:15 +02:00
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/** As socks5_client_request_parse(), but do not allocate the output
|
|
|
|
* object.
|
2018-05-13 14:13:15 +02:00
|
|
|
*/
|
|
|
|
static ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_parse_into(socks5_client_request_t *obj, const uint8_t *input, const size_t len_in)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Parse u8 version IN [5] */
|
2018-05-13 14:13:15 +02:00
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! (obj->version == 5))
|
2018-05-13 14:13:15 +02:00
|
|
|
goto fail;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Parse u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR, CMD_UDP_ASSOCIATE] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->command = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR || obj->command == CMD_UDP_ASSOCIATE))
|
|
|
|
goto fail;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Parse u8 reserved IN [0] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->reserved = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->reserved == 0))
|
|
|
|
goto fail;
|
|
|
|
|
|
|
|
/* Parse u8 atype */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->atype = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Parse union dest_addr[atype] */
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
|
|
|
|
/* Parse u32 dest_addr_ipv4 */
|
|
|
|
CHECK_REMAINING(4, truncated);
|
|
|
|
obj->dest_addr_ipv4 = trunnel_ntohl(trunnel_get_uint32(ptr));
|
|
|
|
remaining -= 4; ptr += 4;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
|
|
|
|
/* Parse u8 dest_addr_ipv6[16] */
|
|
|
|
CHECK_REMAINING(16, truncated);
|
|
|
|
memcpy(obj->dest_addr_ipv6, ptr, 16);
|
|
|
|
remaining -= 16; ptr += 16;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_DOMAINNAME:
|
|
|
|
|
|
|
|
/* Parse struct domainname dest_addr_domainname */
|
|
|
|
result = domainname_parse(&obj->dest_addr_domainname, ptr, remaining);
|
2018-05-13 14:13:15 +02:00
|
|
|
if (result < 0)
|
|
|
|
goto relay_fail;
|
|
|
|
trunnel_assert((size_t)result <= remaining);
|
|
|
|
remaining -= result; ptr += result;
|
2018-06-26 18:31:26 +02:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
goto fail;
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
|
|
|
|
/* Parse u16 dest_port */
|
|
|
|
CHECK_REMAINING(2, truncated);
|
|
|
|
obj->dest_port = trunnel_ntohs(trunnel_get_uint16(ptr));
|
|
|
|
remaining -= 2; ptr += 2;
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
relay_fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_parse(socks5_client_request_t **output, const uint8_t *input, const size_t len_in)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result;
|
2018-06-26 18:31:26 +02:00
|
|
|
*output = socks5_client_request_new();
|
2018-05-13 14:13:15 +02:00
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
2018-06-26 18:31:26 +02:00
|
|
|
result = socks5_client_request_parse_into(*output, input, len_in);
|
2018-05-13 14:13:15 +02:00
|
|
|
if (result < 0) {
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_client_request_free(*output);
|
2018-05-13 14:13:15 +02:00
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_t *
|
|
|
|
socks5_server_reply_new(void)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_t *val = trunnel_calloc(1, sizeof(socks5_server_reply_t));
|
2018-05-13 14:13:15 +02:00
|
|
|
if (NULL == val)
|
|
|
|
return NULL;
|
2018-06-26 18:31:26 +02:00
|
|
|
val->version = 5;
|
2018-05-13 14:13:15 +02:00
|
|
|
return val;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** Release all storage held inside 'obj', but do not free 'obj'.
|
|
|
|
*/
|
|
|
|
static void
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_clear(socks5_server_reply_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
(void) obj;
|
2018-06-26 18:31:26 +02:00
|
|
|
domainname_free(obj->bind_addr_domainname);
|
|
|
|
obj->bind_addr_domainname = NULL;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_free(socks5_server_reply_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return;
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_clear(obj);
|
|
|
|
trunnel_memwipe(obj, sizeof(socks5_server_reply_t));
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_free_(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_get_version(const socks5_server_reply_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
return inp->version;
|
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_set_version(socks5_server_reply_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! ((val == 5))) {
|
2018-05-13 14:13:15 +02:00
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->version = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_get_reply(const socks5_server_reply_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->reply;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_set_reply(socks5_server_reply_t *inp, uint8_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->reply = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks5_server_reply_get_reserved(const socks5_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return inp->reserved;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_server_reply_set_reserved(socks5_server_reply_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
if (! ((val == 0))) {
|
|
|
|
TRUNNEL_SET_ERROR_CODE(inp);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
inp->reserved = val;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
uint8_t
|
|
|
|
socks5_server_reply_get_atype(const socks5_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return inp->atype;
|
|
|
|
}
|
|
|
|
int
|
|
|
|
socks5_server_reply_set_atype(socks5_server_reply_t *inp, uint8_t val)
|
|
|
|
{
|
|
|
|
inp->atype = val;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
uint32_t
|
|
|
|
socks5_server_reply_get_bind_addr_ipv4(const socks5_server_reply_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->bind_addr_ipv4;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_set_bind_addr_ipv4(socks5_server_reply_t *inp, uint32_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->bind_addr_ipv4 = val;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
size_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_getlen_bind_addr_ipv6(const socks5_server_reply_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
(void)inp; return 16;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_server_reply_get_bind_addr_ipv6(socks5_server_reply_t *inp, size_t idx)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
trunnel_assert(idx < 16);
|
|
|
|
return inp->bind_addr_ipv6[idx];
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
uint8_t
|
|
|
|
socks5_server_reply_getconst_bind_addr_ipv6(const socks5_server_reply_t *inp, size_t idx)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return socks5_server_reply_get_bind_addr_ipv6((socks5_server_reply_t*)inp, idx);
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_set_bind_addr_ipv6(socks5_server_reply_t *inp, size_t idx, uint8_t elt)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
trunnel_assert(idx < 16);
|
|
|
|
inp->bind_addr_ipv6[idx] = elt;
|
|
|
|
return 0;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
|
|
|
|
uint8_t *
|
|
|
|
socks5_server_reply_getarray_bind_addr_ipv6(socks5_server_reply_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->bind_addr_ipv6;
|
|
|
|
}
|
|
|
|
const uint8_t *
|
|
|
|
socks5_server_reply_getconstarray_bind_addr_ipv6(const socks5_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return (const uint8_t *)socks5_server_reply_getarray_bind_addr_ipv6((socks5_server_reply_t*)inp);
|
|
|
|
}
|
|
|
|
struct domainname_st *
|
|
|
|
socks5_server_reply_get_bind_addr_domainname(socks5_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return inp->bind_addr_domainname;
|
|
|
|
}
|
|
|
|
const struct domainname_st *
|
|
|
|
socks5_server_reply_getconst_bind_addr_domainname(const socks5_server_reply_t *inp)
|
|
|
|
{
|
|
|
|
return socks5_server_reply_get_bind_addr_domainname((socks5_server_reply_t*) inp);
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_set_bind_addr_domainname(socks5_server_reply_t *inp, struct domainname_st *val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
if (inp->bind_addr_domainname && inp->bind_addr_domainname != val)
|
|
|
|
domainname_free(inp->bind_addr_domainname);
|
|
|
|
return socks5_server_reply_set0_bind_addr_domainname(inp, val);
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
int
|
|
|
|
socks5_server_reply_set0_bind_addr_domainname(socks5_server_reply_t *inp, struct domainname_st *val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->bind_addr_domainname = val;
|
|
|
|
return 0;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
uint16_t
|
|
|
|
socks5_server_reply_get_bind_port(const socks5_server_reply_t *inp)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
return inp->bind_port;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_set_bind_port(socks5_server_reply_t *inp, uint16_t val)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
2018-06-26 18:31:26 +02:00
|
|
|
inp->bind_port = val;
|
2018-05-13 14:13:15 +02:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
const char *
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_check(const socks5_server_reply_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
if (obj == NULL)
|
|
|
|
return "Object was NULL";
|
|
|
|
if (obj->trunnel_error_code_)
|
|
|
|
return "A set function failed on this object";
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! (obj->version == 5))
|
2018-05-13 14:13:15 +02:00
|
|
|
return "Integer out of bounds";
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! (obj->reserved == 0))
|
|
|
|
return "Integer out of bounds";
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_DOMAINNAME:
|
|
|
|
{
|
|
|
|
const char *msg;
|
|
|
|
if (NULL != (msg = domainname_check(obj->bind_addr_domainname)))
|
|
|
|
return msg;
|
|
|
|
}
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
default:
|
|
|
|
return "Bad tag for union";
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_encoded_len(const socks5_server_reply_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
if (NULL != socks5_server_reply_check(obj))
|
2018-05-13 14:13:15 +02:00
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Length of u8 version IN [5] */
|
2018-05-13 14:13:15 +02:00
|
|
|
result += 1;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Length of u8 reply */
|
2018-05-13 14:13:15 +02:00
|
|
|
result += 1;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Length of u8 reserved IN [0] */
|
|
|
|
result += 1;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Length of u8 atype */
|
|
|
|
result += 1;
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
|
|
|
|
/* Length of u32 bind_addr_ipv4 */
|
|
|
|
result += 4;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
|
|
|
|
/* Length of u8 bind_addr_ipv6[16] */
|
|
|
|
result += 16;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_DOMAINNAME:
|
|
|
|
|
|
|
|
/* Length of struct domainname bind_addr_domainname */
|
|
|
|
result += domainname_encoded_len(obj->bind_addr_domainname);
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
default:
|
|
|
|
trunnel_assert(0);
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
|
|
|
|
/* Length of u16 bind_port */
|
|
|
|
result += 2;
|
2018-05-13 14:13:15 +02:00
|
|
|
return result;
|
|
|
|
}
|
|
|
|
int
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_clear_errors(socks5_server_reply_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
int r = obj->trunnel_error_code_;
|
|
|
|
obj->trunnel_error_code_ = 0;
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_encode(uint8_t *output, const size_t avail, const socks5_server_reply_t *obj)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result = 0;
|
|
|
|
size_t written = 0;
|
|
|
|
uint8_t *ptr = output;
|
|
|
|
const char *msg;
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
2018-06-26 18:31:26 +02:00
|
|
|
const ssize_t encoded_len = socks5_server_reply_encoded_len(obj);
|
2018-05-13 14:13:15 +02:00
|
|
|
#endif
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
if (NULL != (msg = socks5_server_reply_check(obj)))
|
2018-05-13 14:13:15 +02:00
|
|
|
goto check_failed;
|
|
|
|
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
#endif
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u8 version IN [5] */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->version));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u8 reply */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
2018-06-26 18:31:26 +02:00
|
|
|
trunnel_set_uint8(ptr, (obj->reply));
|
2018-05-13 14:13:15 +02:00
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u8 reserved IN [0] */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
2018-06-26 18:31:26 +02:00
|
|
|
if (avail - written < 1)
|
2018-05-13 14:13:15 +02:00
|
|
|
goto truncated;
|
2018-06-26 18:31:26 +02:00
|
|
|
trunnel_set_uint8(ptr, (obj->reserved));
|
|
|
|
written += 1; ptr += 1;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u8 atype */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 1)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint8(ptr, (obj->atype));
|
|
|
|
written += 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Encode union bind_addr[atype] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
|
|
|
|
/* Encode u32 bind_addr_ipv4 */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 4)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint32(ptr, trunnel_htonl(obj->bind_addr_ipv4));
|
|
|
|
written += 4; ptr += 4;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
|
|
|
|
/* Encode u8 bind_addr_ipv6[16] */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 16)
|
|
|
|
goto truncated;
|
|
|
|
memcpy(ptr, obj->bind_addr_ipv6, 16);
|
|
|
|
written += 16; ptr += 16;
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
case ATYPE_DOMAINNAME:
|
|
|
|
|
|
|
|
/* Encode struct domainname bind_addr_domainname */
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(written <= avail);
|
2018-06-26 18:31:26 +02:00
|
|
|
result = domainname_encode(ptr, avail - written, obj->bind_addr_domainname);
|
2018-05-13 14:13:15 +02:00
|
|
|
if (result < 0)
|
|
|
|
goto fail; /* XXXXXXX !*/
|
|
|
|
written += result; ptr += result;
|
2018-06-26 18:31:26 +02:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
trunnel_assert(0);
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Encode u16 bind_port */
|
|
|
|
trunnel_assert(written <= avail);
|
|
|
|
if (avail - written < 2)
|
|
|
|
goto truncated;
|
|
|
|
trunnel_set_uint16(ptr, trunnel_htons(obj->bind_port));
|
|
|
|
written += 2; ptr += 2;
|
|
|
|
|
2018-05-13 14:13:15 +02:00
|
|
|
|
|
|
|
trunnel_assert(ptr == output + written);
|
|
|
|
#ifdef TRUNNEL_CHECK_ENCODED_LEN
|
|
|
|
{
|
|
|
|
trunnel_assert(encoded_len >= 0);
|
|
|
|
trunnel_assert((size_t)encoded_len == written);
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return written;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
result = -2;
|
|
|
|
goto fail;
|
|
|
|
check_failed:
|
|
|
|
(void)msg;
|
|
|
|
result = -1;
|
|
|
|
goto fail;
|
|
|
|
fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/** As socks5_server_reply_parse(), but do not allocate the output
|
|
|
|
* object.
|
2018-05-13 14:13:15 +02:00
|
|
|
*/
|
|
|
|
static ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_parse_into(socks5_server_reply_t *obj, const uint8_t *input, const size_t len_in)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
const uint8_t *ptr = input;
|
|
|
|
size_t remaining = len_in;
|
|
|
|
ssize_t result = 0;
|
|
|
|
(void)result;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Parse u8 version IN [5] */
|
2018-05-13 14:13:15 +02:00
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->version = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
2018-06-26 18:31:26 +02:00
|
|
|
if (! (obj->version == 5))
|
2018-05-13 14:13:15 +02:00
|
|
|
goto fail;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Parse u8 reply */
|
2018-05-13 14:13:15 +02:00
|
|
|
CHECK_REMAINING(1, truncated);
|
2018-06-26 18:31:26 +02:00
|
|
|
obj->reply = (trunnel_get_uint8(ptr));
|
2018-05-13 14:13:15 +02:00
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Parse u8 reserved IN [0] */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->reserved = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
if (! (obj->reserved == 0))
|
|
|
|
goto fail;
|
2018-05-13 14:13:15 +02:00
|
|
|
|
2018-06-26 18:31:26 +02:00
|
|
|
/* Parse u8 atype */
|
|
|
|
CHECK_REMAINING(1, truncated);
|
|
|
|
obj->atype = (trunnel_get_uint8(ptr));
|
|
|
|
remaining -= 1; ptr += 1;
|
|
|
|
|
|
|
|
/* Parse union bind_addr[atype] */
|
|
|
|
switch (obj->atype) {
|
|
|
|
|
|
|
|
case ATYPE_IPV4:
|
|
|
|
|
|
|
|
/* Parse u32 bind_addr_ipv4 */
|
|
|
|
CHECK_REMAINING(4, truncated);
|
|
|
|
obj->bind_addr_ipv4 = trunnel_ntohl(trunnel_get_uint32(ptr));
|
|
|
|
remaining -= 4; ptr += 4;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_IPV6:
|
|
|
|
|
|
|
|
/* Parse u8 bind_addr_ipv6[16] */
|
|
|
|
CHECK_REMAINING(16, truncated);
|
|
|
|
memcpy(obj->bind_addr_ipv6, ptr, 16);
|
|
|
|
remaining -= 16; ptr += 16;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ATYPE_DOMAINNAME:
|
|
|
|
|
|
|
|
/* Parse struct domainname bind_addr_domainname */
|
|
|
|
result = domainname_parse(&obj->bind_addr_domainname, ptr, remaining);
|
2018-05-13 14:13:15 +02:00
|
|
|
if (result < 0)
|
|
|
|
goto relay_fail;
|
|
|
|
trunnel_assert((size_t)result <= remaining);
|
|
|
|
remaining -= result; ptr += result;
|
2018-06-26 18:31:26 +02:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
goto fail;
|
|
|
|
break;
|
2018-05-13 14:13:15 +02:00
|
|
|
}
|
2018-06-26 18:31:26 +02:00
|
|
|
|
|
|
|
/* Parse u16 bind_port */
|
|
|
|
CHECK_REMAINING(2, truncated);
|
|
|
|
obj->bind_port = trunnel_ntohs(trunnel_get_uint16(ptr));
|
|
|
|
remaining -= 2; ptr += 2;
|
2018-05-13 14:13:15 +02:00
|
|
|
trunnel_assert(ptr + remaining == input + len_in);
|
|
|
|
return len_in - remaining;
|
|
|
|
|
|
|
|
truncated:
|
|
|
|
return -2;
|
|
|
|
relay_fail:
|
|
|
|
trunnel_assert(result < 0);
|
|
|
|
return result;
|
|
|
|
fail:
|
|
|
|
result = -1;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_parse(socks5_server_reply_t **output, const uint8_t *input, const size_t len_in)
|
2018-05-13 14:13:15 +02:00
|
|
|
{
|
|
|
|
ssize_t result;
|
2018-06-26 18:31:26 +02:00
|
|
|
*output = socks5_server_reply_new();
|
2018-05-13 14:13:15 +02:00
|
|
|
if (NULL == *output)
|
|
|
|
return -1;
|
2018-06-26 18:31:26 +02:00
|
|
|
result = socks5_server_reply_parse_into(*output, input, len_in);
|
2018-05-13 14:13:15 +02:00
|
|
|
if (result < 0) {
|
2018-06-26 18:31:26 +02:00
|
|
|
socks5_server_reply_free(*output);
|
2018-05-13 14:13:15 +02:00
|
|
|
*output = NULL;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|