GitLab/ocserv
1
0
Fork 0
mirror of https://gitlab.com/openconnect/ocserv.git synced 2026-03-16 06:49:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow a certain number of requests to the HTTP server

Browse Source
This commit is contained in:
Nikos Mavrogiannopoulos
2013-01-29 20:24:17 +01:00
parent b24a3bf74b
commit 23fe42f5d4
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/vpn.c
View File

@@ -44,6 +44,9 @@
#include <http-parser/http_parser.h> #include <http-parser/http_parser.h>
/* HTTP requests prior to disconnection */
#define MAX_HTTP_REQUESTS 8
static int connect_handler(worker_st *server); static int connect_handler(worker_st *server);
typedef int (*url_handler_fn)(worker_st*); typedef int (*url_handler_fn)(worker_st*);
@@ -180,7 +183,6 @@ char* tmp = malloc(length+1);
return 0; return 0;
} }
void vpn_server(struct cfg_st *config, struct tls_st *creds, void vpn_server(struct cfg_st *config, struct tls_st *creds,
struct sockaddr_storage* r_addr, socklen_t r_addr_len, struct sockaddr_storage* r_addr, socklen_t r_addr_len,
int cmd_fd, int fd) int cmd_fd, int fd)
@@ -195,6 +197,7 @@ void vpn_server(struct cfg_st *config, struct tls_st *creds,
worker_st _server; worker_st _server;
worker_st *server; worker_st *server;
url_handler_fn fn; url_handler_fn fn;
int requests_left = MAX_HTTP_REQUESTS;
memset(&_server, 0, sizeof(_server)); memset(&_server, 0, sizeof(_server));
server = &_server; server = &_server;
@@ -242,6 +245,11 @@ void vpn_server(struct cfg_st *config, struct tls_st *creds,
server->tun_fd = -1; server->tun_fd = -1;
restart: restart:
if (requests_left-- <= 0) {
oclog(server, LOG_INFO, "Maximum number of HTTP requests reached.");
exit(1);
}
http_parser_init(&parser, HTTP_REQUEST); http_parser_init(&parser, HTTP_REQUEST);
memset(&req, 0, sizeof(req)); memset(&req, 0, sizeof(req));
parser.data = &req; parser.data = &req;