shaba/systemd
1
0
Fork 0
mirror of https://github.com/systemd/systemd.git synced 2025-03-28 02:50:16 +03:00
Code

socket-proxyd: Introduced dynamic connection limit via an option. (#4749)

Browse Source
This commit is contained in:
(GalaxyMaster) 2016-11-29 04:25:11 +11:00 committed by Lennart Poettering
parent a92cf7840f
commit dc3b8afb93
2 changed files with 30 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
man/systemd-socket-proxyd.xml
View File

@ -85,6 +85,13 @@
<variablelist>
<xi:include href="standard-options.xml" xpointer="help" />
<xi:include href="standard-options.xml" xpointer="version" />
<varlistentry>
<term><option>--max-connections=</option></term>
<term><option>-c</option></term>
<listitem><para>Sets the maximum number of simultaneous connections, defaults to 256.
If the limit of concurrent connections is reached further connections will be refused.</para></listitem>
</varlistentry>
</variablelist>
</refsect1>
<refsect1>

29
src/socket-proxy/socket-proxyd.c
View File

@ -39,10 +39,11 @@
#include "set.h"
#include "socket-util.h"
#include "string-util.h"
#include "parse-util.h"
#include "util.h"
#define BUFFER_SIZE (256 * 1024)
#define CONNECTIONS_MAX 256
static unsigned arg_connections_max = 256;
static const char *arg_remote_host = NULL;
@ -445,7 +446,7 @@ static int add_connection_socket(Context *context, int fd) {
assert(context);
assert(fd >= 0);
if (set_size(context->connections) > CONNECTIONS_MAX) {
if (set_size(context->connections) > arg_connections_max) {
log_warning("Hit connection limit, refusing connection.");
safe_close(fd);
return 0;
@ -563,6 +564,7 @@ static void help(void) {
printf("%1$s [HOST:PORT]\n"
"%1$s [SOCKET]\n\n"
"Bidirectionally proxy local sockets to another (possibly remote) socket.\n\n"
" -c --max-connections= Set the maximum number of connections to be accepted\n"
" -h --help Show this help\n"
" --version Show package version\n",
program_invocation_short_name);
@ -576,17 +578,18 @@ static int parse_argv(int argc, char *argv[]) {
};
static const struct option options[] = {
{ "help", no_argument, NULL, 'h' },
{ "version", no_argument, NULL, ARG_VERSION },
{ "connections-max", required_argument, NULL, 'c' },
{ "help", no_argument, NULL, 'h' },
{ "version", no_argument, NULL, ARG_VERSION },
{}
};
int c;
int c, r;
assert(argc >= 0);
assert(argv);
while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0)
while ((c = getopt_long(argc, argv, "c:h", options, NULL)) >= 0)
switch (c) {
@ -594,6 +597,20 @@ static int parse_argv(int argc, char *argv[]) {
help();
return 0;
case 'c':
r = safe_atou(optarg, &arg_connections_max);
if (r < 0) {
log_error("Failed to parse --connections-max= argument: %s", optarg);
return r;
}
if (arg_connections_max < 1) {
log_error("Connection limit is too low.");
return -EINVAL;
}
break;
case ARG_VERSION:
return version();