Merge remote-tracking branch 'origin/topic/timw/2136-iosource-shutdown'

* origin/topic/timw/2136-iosource-shutdown:
  GH-2136: Clean up DNS_Mgr before shutting down

CHANGES

@@ -1,3 +1,7 @@
+5.0.0-dev.540 | 2022-06-01 11:08:42 -0700
+
+  * GH-2136: Clean up DNS_Mgr before shutting down (Tim Wojtulewicz, Corelight)
+
 5.0.0-dev.538 | 2022-06-01 09:20:21 -0700
 
   * GH-2101: Update cmake submodule to fix c-ares include path problem (Tim Wojtulewicz, Corelight)

VERSION

@@ -1 +1 @@
-5.0.0-dev.538
+5.0.0-dev.540

src/DNS_Mgr.cc

@@ -560,6 +560,12 @@ DNS_Mgr::~DNS_Mgr()
 	ares_library_cleanup();
 	}
 
+void DNS_Mgr::Done()
+	{
+	shutting_down = true;
+	Flush();
+	}
+
 void DNS_Mgr::RegisterSocket(int fd, bool read, bool write)
 	{
 	if ( read && socket_fds.count(fd) == 0 )
@@ -713,6 +719,9 @@ static void resolve_lookup_cb(DNS_Mgr::LookupCallback* callback, const std::stri
 
 ValPtr DNS_Mgr::Lookup(const std::string& name, int request_type)
 	{
+	if ( shutting_down )
+		return nullptr;
+
 	if ( request_type == T_A || request_type == T_AAAA )
 		return LookupHost(name);
 
@@ -761,6 +770,9 @@ ValPtr DNS_Mgr::Lookup(const std::string& name, int request_type)
 
 TableValPtr DNS_Mgr::LookupHost(const std::string& name)
 	{
+	if ( shutting_down )
+		return nullptr;
+
 	if ( mode == DNS_FAKE )
 		return fake_name_lookup_result(name);
 
@@ -809,6 +821,9 @@ TableValPtr DNS_Mgr::LookupHost(const std::string& name)
 
 StringValPtr DNS_Mgr::LookupAddr(const IPAddr& addr)
 	{
+	if ( shutting_down )
+		return nullptr;
+
 	if ( mode == DNS_FAKE )
 		return make_intrusive<StringVal>(fake_addr_lookup_result(addr));
 
@@ -853,6 +868,9 @@ StringValPtr DNS_Mgr::LookupAddr(const IPAddr& addr)
 
 void DNS_Mgr::LookupHost(const std::string& name, LookupCallback* callback)
 	{
+	if ( shutting_down )
+		return;
+
 	if ( mode == DNS_FAKE )
 		{
 		resolve_lookup_cb(callback, fake_name_lookup_result(name));
@@ -893,6 +911,9 @@ void DNS_Mgr::LookupHost(const std::string& name, LookupCallback* callback)
 
 void DNS_Mgr::LookupAddr(const IPAddr& addr, LookupCallback* callback)
 	{
+	if ( shutting_down )
+		return;
+
 	if ( mode == DNS_FAKE )
 		{
 		resolve_lookup_cb(callback, fake_addr_lookup_result(addr));
@@ -932,6 +953,9 @@ void DNS_Mgr::LookupAddr(const IPAddr& addr, LookupCallback* callback)
 
 void DNS_Mgr::Lookup(const std::string& name, int request_type, LookupCallback* callback)
 	{
+	if ( shutting_down )
+		return;
+
 	if ( mode == DNS_FAKE )
 		{
 		resolve_lookup_cb(callback, fake_lookup_result(name, request_type));

src/DNS_Mgr.h

@@ -97,6 +97,11 @@ public:
 	explicit DNS_Mgr(DNS_MgrMode mode);
 	~DNS_Mgr() override;
 
+	/**
+	 * Finalizes the source when it's being closed.
+	 */
+	void Done() override;
+
 	/**
 	 * Finalizes the manager initialization. This should be called only after all
 	 * of the scripts have been parsed at startup.
@@ -337,6 +342,8 @@ protected:
 
 	std::set<int> socket_fds;
 	std::set<int> write_socket_fds;
+
+	bool shutting_down = false;
 	};
 
 extern DNS_Mgr* dns_mgr;

src/iosource/Manager.cc

@@ -61,11 +61,12 @@ Manager::~Manager()
 	delete wakeup;
 	wakeup = nullptr;
 
-	for ( SourceList::iterator i = sources.begin(); i != sources.end(); ++i )
+	// Make sure all of the sources are done before we try to delete any of them.
-		{
+	for ( auto& src : sources )
-		auto src = *i;
 		src->src->Done();
 
+	for ( auto& src : sources )
+		{
 		if ( src->manage_lifetime )
 			delete src->src;