mirrors/haproxy
1
0
Fork 0
mirror of https://git.haproxy.org/git/haproxy.git/ synced 2025-12-16 15:11:01 +01:00
Code Issues Projects Releases Wiki Activity

DEV: tcploop: make it possible to change the target address of a connect()

Browse Source 
Sometimes it's more convenient to be able to specify where to connect on
the connect() statement, let's make it possible to pass it in argument to
the C command.
This commit is contained in:
Willy Tarreau 2022-06-07 11:46:57 +02:00
parent 7184ca23c6
commit 542bf0a7bb
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
dev/tcploop/tcploop.c
View File

@ -99,7 +99,7 @@ __attribute__((noreturn)) void usage(int code, const char *arg0)
"actions :\n" "actions :\n"
" A[<count>] : Accepts <count> incoming sockets and closes count-1\n" " A[<count>] : Accepts <count> incoming sockets and closes count-1\n"
" Note: fd=accept(fd)\n" " Note: fd=accept(fd)\n"
" C : Connects to ip:port\n" " C[[ip]:port] : Connects to ip:port or default ones if unspecified.\n"
" Note: fd=socket,connect(fd)\n" " Note: fd=socket,connect(fd)\n"
" D : Disconnect (connect to AF_UNSPEC)\n" " D : Disconnect (connect to AF_UNSPEC)\n"
" E[<size>] : Echo this amount of bytes. 0=infinite. unset=any amount.\n" " E[<size>] : Echo this amount of bytes. 0=infinite. unset=any amount.\n"
@ -236,7 +236,7 @@ void sig_handler(int sig)
/* converts str in the form [[<ipv4>|<ipv6>|<hostname>]:]port to struct sockaddr_storage. /* converts str in the form [[<ipv4>|<ipv6>|<hostname>]:]port to struct sockaddr_storage.
* Returns < 0 with err set in case of error. * Returns < 0 with err set in case of error.
*/ */
int addr_to_ss(char *str, struct sockaddr_storage *ss, struct err_msg *err) int addr_to_ss(const char *str, struct sockaddr_storage *ss, struct err_msg *err)
{ {
char *port_str; char *port_str;
int port; int port;
@ -418,8 +418,17 @@ int tcp_accept(int sock, const char *arg)
/* Try to establish a new connection to <sa>. Return the fd or -1 in case of error */ /* Try to establish a new connection to <sa>. Return the fd or -1 in case of error */
int tcp_connect(const struct sockaddr_storage *sa, const char *arg) int tcp_connect(const struct sockaddr_storage *sa, const char *arg)
{ {
struct sockaddr_storage conn_addr;
int sock; int sock;
if (arg[1]) {
struct err_msg err;
if (addr_to_ss(arg + 1, &conn_addr, &err) < 0)
die(1, "%s\n", err.msg);
sa = &conn_addr;
}
sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sock < 0) if (sock < 0)
return -1; return -1;