--mss, fix cache

This commit is contained in:
ruti 2024-03-28 20:42:43 +03:00
parent dbbdb005a1
commit a909f44b35
4 changed files with 67 additions and 31 deletions

15
main.c
View File

@ -95,6 +95,9 @@ const char help_text[] = {
" -e, --oob-data <f|:str> Set custom OOB data, filename or :string\n" " -e, --oob-data <f|:str> Set custom OOB data, filename or :string\n"
" -M, --mod-http <h,d,r> Modify HTTP: hcsmix,dcsmix,rmspace\n" " -M, --mod-http <h,d,r> Modify HTTP: hcsmix,dcsmix,rmspace\n"
" -r, --tlsrec <n[+s]> Make TLS record at position\n" " -r, --tlsrec <n[+s]> Make TLS record at position\n"
#ifdef __linux__
" -m, --mss <size> Set MSS for outgoing connections\n"
#endif
}; };
@ -138,6 +141,9 @@ const struct option options[] = {
{"mod-http", 1, 0, 'M'}, {"mod-http", 1, 0, 'M'},
{"tlsrec", 1, 0, 'r'}, {"tlsrec", 1, 0, 'r'},
{"def-ttl", 1, 0, 'g'}, {"def-ttl", 1, 0, 'g'},
#ifdef __linux__
{"mss", 1, 0, 'm'},
#endif
{"delay", 1, 0, 'w'}, // {"delay", 1, 0, 'w'}, //
{"not-wait-send", 0, 0, 'W'}, // {"not-wait-send", 0, 0, 'W'}, //
{0} {0}
@ -634,6 +640,15 @@ int main(int argc, char **argv)
} }
break; break;
case 'm':
val = strtol(optarg, &end, 0);
if (val < 88 || val > 32767 || *end)
invalid = 1;
else {
dp->mss = val;
}
break;
case 'g': case 'g':
val = strtol(optarg, &end, 0); val = strtol(optarg, &end, 0);
if (val <= 0 || val > 255 || *end) if (val <= 0 || val > 255 || *end)

View File

@ -49,6 +49,7 @@ struct desync_params {
int mod_http; int mod_http;
int tlsrec_n; int tlsrec_n;
struct part *tlsrec; struct part *tlsrec;
int mss;
int spos_n; int spos_n;
struct spos *spos; struct spos *spos;
}; };

77
proxy.c
View File

@ -291,7 +291,7 @@ int s5_get_addr(char *buffer, ssize_t n,
int create_conn(struct poolhd *pool, int create_conn(struct poolhd *pool,
struct eval *val, struct sockaddr_ina *dst, int next) struct eval *val, struct sockaddr_ina *dst, int next, int mss)
{ {
struct sockaddr_ina addr = *dst; struct sockaddr_ina addr = *dst;
@ -332,7 +332,12 @@ int create_conn(struct poolhd *pool,
close(sfd); close(sfd);
return -1; return -1;
} }
#endif if (mss && setsockopt(sfd, IPPROTO_TCP,
TCP_MAXSEG, (char *)&mss, sizeof(mss))) {
uniperror("setsockopt TCP_MAXSEG");
close(sfd);
return -1;
}
#ifdef TCP_FASTOPEN_CONNECT #ifdef TCP_FASTOPEN_CONNECT
int yes = 1; int yes = 1;
if (params.tfo && setsockopt(sfd, IPPROTO_TCP, if (params.tfo && setsockopt(sfd, IPPROTO_TCP,
@ -342,6 +347,7 @@ int create_conn(struct poolhd *pool,
return -1; return -1;
} }
#endif #endif
#endif
int one = 1; int one = 1;
if (setsockopt(sfd, IPPROTO_TCP, if (setsockopt(sfd, IPPROTO_TCP,
TCP_NODELAY, (char *)&one, sizeof(one))) { TCP_NODELAY, (char *)&one, sizeof(one))) {
@ -499,14 +505,19 @@ int mode_add_get(struct sockaddr_ina *dst, int m)
char *data; char *data;
int len; int len;
time_t t; time_t t;
struct elem *val; struct elem *val;
struct sockaddr_ina val_addr = *dst;
if (dst->sa.sa_family == AF_INET) { if (val_addr.sa.sa_family == AF_INET6) {
data = (char *)(&dst->in.sin_addr); map_fix(&val_addr, 0);
len = sizeof(dst->in.sin_addr); }
} else { if (val_addr.sa.sa_family == AF_INET) {
data = (char *)(&dst->in6.sin6_addr); data = (char *)(&val_addr.in.sin_addr);
len = sizeof(dst->in6.sin6_addr); len = sizeof(val_addr.in.sin_addr);
}
else {
data = (char *)(&val_addr.in6.sin6_addr);
len = sizeof(val_addr.in6.sin6_addr);
} }
int i = mem_index(params.mempool, data, len); int i = mem_index(params.mempool, data, len);
if (m == 0 && i >= 0) { if (m == 0 && i >= 0) {
@ -547,6 +558,8 @@ static inline int on_request(struct poolhd *pool, struct eval *val,
if (n) uniperror("ss recv"); if (n) uniperror("ss recv");
return -1; return -1;
} }
int error = 0;
if (*buffer == S_VER5) { if (*buffer == S_VER5) {
if (val->flag != FLAG_S5) { if (val->flag != FLAG_S5) {
if (auth_socks5(val->fd, buffer, n)) { if (auth_socks5(val->fd, buffer, n)) {
@ -555,35 +568,35 @@ static inline int on_request(struct poolhd *pool, struct eval *val,
val->flag = FLAG_S5; val->flag = FLAG_S5;
return 0; return 0;
} }
int s5e = s5_get_addr(buffer, n, &dst); error = s5_get_addr(buffer, n, &dst);
if (!s5e &&
create_conn(pool, val, &dst, EV_CONNECT)) {
s5e = S_ER_GEN;
}
if (s5e) {
resp_s5_error(val->fd, s5e);
return -1;
}
} }
else if (*buffer == S_VER4) { else if (*buffer == S_VER4) {
val->flag = FLAG_S4; val->flag = FLAG_S4;
error = s4_get_addr(buffer, n, &dst);
int error = s4_get_addr(buffer, n, &dst);
if (!error) {
error = create_conn(pool, val, &dst, EV_CONNECT);
}
if (error) {
if (resp_error(val->fd, error, FLAG_S4) < 0)
uniperror("send");
return -1;
}
} }
else { else {
LOG(LOG_E, "ss: invalid version: 0x%x (%lu)\n", *buffer, n); LOG(LOG_E, "ss: invalid version: 0x%x (%lu)\n", *buffer, n);
return -1; return -1;
} }
int m = mode_add_get( if (error) {
(struct sockaddr_ina *)&val->pair->in6, -1); if (val->flag == FLAG_S4
&& resp_error(val->fd, error, FLAG_S4) < 0) {
uniperror("send");
}
else if (resp_s5_error(val->fd, error) < 0) {
uniperror("send");
}
return -1;
}
int m = mode_add_get(&dst, -1);
struct desync_params dp = params.dp[m < 0 ? 0 : m];
error = create_conn(pool, val, &dst, EV_CONNECT, dp.mss);
if (error) {
if (resp_error(val->fd, error, val->flag) < 0)
uniperror("send");
return -1;
}
if (m >= 0) { if (m >= 0) {
val->attempt = m; val->attempt = m;
} }
@ -607,8 +620,10 @@ int try_again(struct poolhd *pool, struct eval *val, char data)
(struct sockaddr_ina *)&val->in6, 0); (struct sockaddr_ina *)&val->in6, 0);
return -1; return -1;
} }
struct desync_params dp = params.dp[m];
if (create_conn(pool, client, if (create_conn(pool, client,
(struct sockaddr_ina *)&val->in6, EV_DESYNC)) { (struct sockaddr_ina *)&val->in6, EV_DESYNC, dp.mss)) {
return -1; return -1;
} }
val->pair = 0; val->pair = 0;

View File

@ -119,6 +119,11 @@ $ ./ciadpi --disorder 3 -A --tlsrec 1+s
Разделить ClientHello на отдельные записи по указанному смещению Разделить ClientHello на отдельные записи по указанному смещению
Можно указывать несколько раз Можно указывать несколько раз
-m, --mss <size>
Установить опцию MSS для TCP соединения
Можно использовать для того, чтобы вынудить сервер разбить свой ответ
При низких значениях сильно сказывается на скорости
------- -------
Сборка: Сборка:
Для сборки понадобится: Для сборки понадобится: