vrr/ntopng
2
0
Fork 0
mirror of https://github.com/ntop/ntopng.git synced 2026-05-06 12:15:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Initial work to check available space in socket buffers to send pcaps

Browse source
This commit is contained in:
Simone Mainardi 2018-08-14 19:55:10 +02:00
parent 4390308eda
commit 6f0104045a
3 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/HTTPserver.cpp
View file

@ -110,6 +110,30 @@ static void generate_session_id(char *buf, const char *random, const char *user)
/* ****************************************** */
bool HTTPserver::sender_queue_free_space(const struct mg_connection *conn, u_int32_t space_needed) {
int free_space = 0, send_buf, unsent_buf;
if(conn) {
#if defined(linux) && defined(SIOCOUTQ)
socklen_t optlen = sizeof(send_buf);
if(!getsockopt(conn->client.sock, SOL_SOCKET, SO_SNDBUF, &send_buf, (socklen_t*)&optlen)
&& !ioctl(conn->client.sock, SIOCOUTQ, &unsent_buf)) {
if(send_buf > unsent_buf)
free_space = send_buf - unsent_buf;
ntop->getTrace()->traceEvent(TRACE_NORMAL, "SO_SNDBUF: %d SIOCOUTQ: %d (free space: %d)", send_buf, unsent_buf, free_space);
}
#endif
}
return((int)space_needed < 2 * free_space ? true : false);
}
/* ****************************************** */
bool HTTPserver::authorized_localhost_user_login(const struct mg_connection *conn) {
if(ntop->getPrefs()->is_localhost_users_login_disabled()
&& (conn->request_info.remote_ip == 0x7F000001 /* 127.0.0.1 */))