kiwix/kiwix-tools
1
0
Fork 0
mirror of https://github.com/kiwix/kiwix-tools.git synced 2025-09-27 14:21:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add ip connection limit option

Browse Source 
Adds --ipConnectionLimit/-L option to change MHD_OPTION_PER_IP_CONNECTION_LIMIT by calling server.setIpConnectionLimit()
This commit is contained in:
Nikhil Tanwar 2022-02-05 18:27:22 +05:30
parent b10f2ed1a1
commit 8f4758a6a9
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/server/kiwix-serve.cpp
View File

@ -74,6 +74,7 @@ void usage()
<< "\t-V, --version\t\tprint software version" << std::endl << "\t-V, --version\t\tprint software version" << std::endl
<< "\t-z, --nodatealiases\tcreate URL aliases for each content by removing the date" << std::endl << "\t-z, --nodatealiases\tcreate URL aliases for each content by removing the date" << std::endl
<< "\t-c, --customIndex\tadd path to custom index.html for welcome page" << std::endl << "\t-c, --customIndex\tadd path to custom index.html for welcome page" << std::endl
<< "\t-L, --ipConnectionLimit\tLimit on the number of (concurrent) connections made to the server from the same IP address (default: infinite, recommended: >= 6)" << std::endl
<< std::endl << std::endl
<< "Documentation:" << std::endl << "Documentation:" << std::endl
@ -202,6 +203,7 @@ int main(int argc, char** argv)
bool isVerboseFlag = false; bool isVerboseFlag = false;
bool monitorLibrary = false; bool monitorLibrary = false;
unsigned int PPID = 0; unsigned int PPID = 0;
int ipConnectionLimit = 0;
static struct option long_options[] static struct option long_options[]
= {{"daemon", no_argument, 0, 'd'}, = {{"daemon", no_argument, 0, 'd'},
@ -220,13 +222,14 @@ int main(int argc, char** argv)
{"urlRootLocation", required_argument, 0, 'r'}, {"urlRootLocation", required_argument, 0, 'r'},
{"customIndex", required_argument, 0, 'c'}, {"customIndex", required_argument, 0, 'c'},
{"monitorLibrary", no_argument, 0, 'M'}, {"monitorLibrary", no_argument, 0, 'M'},
{"ipConnectionLimit", required_argument, 0, 'L'},
{0, 0, 0, 0}}; {0, 0, 0, 0}};
/* Argument parsing */ /* Argument parsing */
while (true) { while (true) {
int option_index = 0; int option_index = 0;
int c int c
= getopt_long(argc, argv, "hzmnbdvVla:p:f:t:r:i:c:M", long_options, &option_index); = getopt_long(argc, argv, "hzmnbdvVla:p:f:t:r:i:c:ML:", long_options, &option_index);
if (c != -1) { if (c != -1) {
switch (c) { switch (c) {
@ -278,6 +281,9 @@ int main(int argc, char** argv)
case 'M': case 'M':
monitorLibrary = true; monitorLibrary = true;
break; break;
case 'L':
ipConnectionLimit = atoi(optarg);
break;
} }
} else { } else {
if (optind < argc) { if (optind < argc) {
@ -364,6 +370,7 @@ int main(int argc, char** argv)
server.setTaskbar(!noSearchBarFlag, !noLibraryButtonFlag); server.setTaskbar(!noSearchBarFlag, !noLibraryButtonFlag);
server.setBlockExternalLinks(blockExternalLinks); server.setBlockExternalLinks(blockExternalLinks);
server.setIndexTemplateString(indexTemplateString); server.setIndexTemplateString(indexTemplateString);
server.setIpConnectionLimit(ipConnectionLimit);
if (! server.start()) { if (! server.start()) {
exit(1); exit(1);