11557283197

Committed 28 Oct 2024 03:29PM UTC coverage: 64.653% (+0.04%) from 64.617%

Build # 11557283197

Build Type

Pull #14807

github

Committed by

web-flow

Commit Message

Merge cc5acce6f into dc2139af8

Pull Request Pull Request #14807: rec: drop packet ref in mtasker when it is no longer needed

Run Details

37121 of 88222 branches covered (42.08%)

Branch coverage included in aggregate %.

57 of 57 new or added lines in 2 files covered. (100.0%)

32 existing lines in 9 files now uncovered.

124827 of 162265 relevant lines covered (76.93%)

4511264.71 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

50.79

/pdns/recursordist/rec-tcpout.cc

/*
 * This file is part of PowerDNS or dnsdist.
 * Copyright -- PowerDNS.COM B.V. and its contributors
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * In addition, for the avoidance of any doubt, permission is granted to
 * link this program with OpenSSL and to (re)distribute the binaries
 * produced as the result of such linking.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include "rec-tcpout.hh"

// This line from /usr/include/openssl/ssl2.h: # define CERT char
// throws dnsrecords.hh off the rails.
#undef CERT

#include "syncres.hh"

timeval TCPOutConnectionManager::s_maxIdleTime;
size_t TCPOutConnectionManager::s_maxQueries;
size_t TCPOutConnectionManager::s_maxIdlePerAuth;
size_t TCPOutConnectionManager::s_maxIdlePerThread;

void TCPOutConnectionManager::cleanup(const struct timeval& now)
{
  if (s_maxIdleTime.tv_sec == 0 && s_maxIdleTime.tv_usec == 0) {
    // no maximum idle time
    return;
  }

  for (auto it = d_idle_connections.begin(); it != d_idle_connections.end();) {
    timeval idle = now - it->second.d_last_used;
    if (s_maxIdleTime < idle) {
      it = d_idle_connections.erase(it);
    }
    else {
      ++it;
    }
  }
}

void TCPOutConnectionManager::store(const struct timeval& now, const ComboAddress& ip, Connection&& connection)
{
  ++connection.d_numqueries;
  if (s_maxQueries > 0 && connection.d_numqueries >= s_maxQueries) {
    return;
  }

  if (d_idle_connections.size() >= s_maxIdlePerThread || d_idle_connections.count(ip) >= s_maxIdlePerAuth) {
    cleanup(now);
  }

  if (d_idle_connections.size() >= s_maxIdlePerThread) {
    return;
  }
  if (d_idle_connections.count(ip) >= s_maxIdlePerAuth) {
    return;
  }

  gettimeofday(&connection.d_last_used, nullptr);
  d_idle_connections.emplace(ip, std::move(connection));
}

TCPOutConnectionManager::Connection TCPOutConnectionManager::get(const ComboAddress& ip)
{
  if (d_idle_connections.count(ip) > 0) {
    auto h = d_idle_connections.extract(ip);
    return h.mapped();
  }
  return Connection{};
}

uint64_t getCurrentIdleTCPConnections()
{
  return broadcastAccFunction<uint64_t>([] { return t_tcp_manager.getSize(); });
}

1	/*
2	* This file is part of PowerDNS or dnsdist.
3	* Copyright -- PowerDNS.COM B.V. and its contributors
4	*
5	* This program is free software; you can redistribute it and/or modify
6	* it under the terms of version 2 of the GNU General Public License as
7	* published by the Free Software Foundation.
8	*
9	* In addition, for the avoidance of any doubt, permission is granted to
10	* link this program with OpenSSL and to (re)distribute the binaries
11	* produced as the result of such linking.
12	*
13	* This program is distributed in the hope that it will be useful,
14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16	* GNU General Public License for more details.
17	*
18	* You should have received a copy of the GNU General Public License
19	* along with this program; if not, write to the Free Software
20	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21	*/
22
23	#include "rec-tcpout.hh"
24
25	// This line from /usr/include/openssl/ssl2.h: # define CERT char
26	// throws dnsrecords.hh off the rails.
27	#undef CERT
28
29	#include "syncres.hh"
30
31	timeval TCPOutConnectionManager::s_maxIdleTime;
32	size_t TCPOutConnectionManager::s_maxQueries;
33	size_t TCPOutConnectionManager::s_maxIdlePerAuth;
34	size_t TCPOutConnectionManager::s_maxIdlePerThread;
35
36	void TCPOutConnectionManager::cleanup(const struct timeval& now)
37	{	702✔
38	if (s_maxIdleTime.tv_sec == 0 && s_maxIdleTime.tv_usec == 0) {	702!
39	// no maximum idle time
40	return;	×
41	}	×
42
43	for (auto it = d_idle_connections.begin(); it != d_idle_connections.end();) {	702!
UNCOV 44	timeval idle = now - it->second.d_last_used;	×
UNCOV 45	if (s_maxIdleTime < idle) {	×
UNCOV 46	it = d_idle_connections.erase(it);	×
UNCOV 47	}	×
UNCOV 48	else {	×
UNCOV 49	++it;	×
UNCOV 50	}	×
UNCOV 51	}	×
52	}	702✔
53
54	void TCPOutConnectionManager::store(const struct timeval& now, const ComboAddress& ip, Connection&& connection)
55	{	11✔
56	++connection.d_numqueries;	11✔
57	if (s_maxQueries > 0 && connection.d_numqueries >= s_maxQueries) {	11!
58	return;	×
59	}	×
60
61	if (d_idle_connections.size() >= s_maxIdlePerThread \|\| d_idle_connections.count(ip) >= s_maxIdlePerAuth) {	11!
62	cleanup(now);	×
63	}	×
64
65	if (d_idle_connections.size() >= s_maxIdlePerThread) {	11!
66	return;	×
67	}	×
68	if (d_idle_connections.count(ip) >= s_maxIdlePerAuth) {	11!
69	return;	×
70	}	×
71
72	gettimeofday(&connection.d_last_used, nullptr);	11✔
73	d_idle_connections.emplace(ip, std::move(connection));	11✔
74	}	11✔
75
76	TCPOutConnectionManager::Connection TCPOutConnectionManager::get(const ComboAddress& ip)
77	{	11✔
78	if (d_idle_connections.count(ip) > 0) {	11✔
79	auto h = d_idle_connections.extract(ip);	4✔
80	return h.mapped();	4✔
81	}	4✔
82	return Connection{};	7✔
83	}	11✔
84
85	uint64_t getCurrentIdleTCPConnections()
86	{	155✔
87	return broadcastAccFunction<uint64_t>([] { return t_tcp_manager.getSize(); });	854✔
88	}	155✔

PowerDNS / pdns / 11557283197

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous