PIEthernet fix tcp-server close (properly delete all clients)

PIEthernet::stopThreadedListen() method
decompose client to 2 implementations - server-side and client-side

libs/client_server/piclientserver_client.cpp

@@ -0,0 +1,50 @@
+/*
+    PIP - Platform Independent Primitives
+	Ivan Pelipenko peri4ko@yandex.ru
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    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 Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "piclientserver_client.h"
+
+#include "piethernet.h"
+
+
+void PIClientServer::ServerClient::createForServer(PIEthernet * tcp_) {
+	tcp = tcp_;
+	tcp->setParameter(PIEthernet::KeepConnection, false);
+	init();
+}
+
+
+PIClientServer::Client::Client() {
+	tcp = new PIEthernet(PIEthernet::TCP_Client);
+	tcp->setParameter(PIEthernet::KeepConnection, true);
+	own_tcp = true;
+	init();
+}
+
+
+PIClientServer::Client::~Client() {
+	stop();
+}
+
+
+void PIClientServer::Client::connect(PINetworkAddress addr) {
+	if (!tcp || !own_tcp) return;
+	stop();
+	tcp->connect(addr, true);
+	tcp->startThreadedRead();
+	piCout << "Connect to" << addr.toString();
+}

libs/client_server/piclientserver_client_base.cpp

@@ -16,41 +16,22 @@
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#include "piclient_server_client.h"
+#include "piclientserver_client_base.h"
 
 #include "piethernet.h"
 #include "piliterals_time.h"
-#include "pitime.h"
 
 
-PIClientServer::Client::Client() {}
+PIClientServer::ClientBase::ClientBase() {}
 
 
-PIClientServer::Client::~Client() {
+PIClientServer::ClientBase::~ClientBase() {
 	stop();
 	if (own_tcp) piDeleteSafety(tcp);
 }
 
 
-void PIClientServer::Client::createNew() {
-	if (tcp) return;
-	tcp = new PIEthernet(PIEthernet::TCP_Client);
-	tcp->setParameter(PIEthernet::KeepConnection, true);
-	own_tcp = true;
-	init();
-}
-
-
-void PIClientServer::Client::connect(PINetworkAddress addr) {
-	if (!tcp || !own_tcp) return;
-	stop();
-	tcp->connect(addr, true);
-	tcp->startThreadedRead();
-	piCout << "Connect to" << addr.toString();
-}
-
-
-void PIClientServer::Client::stop() {
+void PIClientServer::ClientBase::stop() {
 	if (!tcp) return;
 	can_write = false;
 	tcp->interrupt();
@@ -60,18 +41,18 @@ void PIClientServer::Client::stop() {
 }
 
 
-int PIClientServer::Client::write(const void * d, const size_t s) {
+int PIClientServer::ClientBase::write(const void * d, const size_t s) {
 	if (!tcp) return -1;
 	if (!can_write) return 0;
 	PIMutexLocker guard(write_mutex);
-	piCout << "... send ...";
+	// piCout << "... send ...";
 	stream.send(PIByteArray(d, s));
-	piCout << "... send ok";
+	// piCout << "... send ok";
 	return s;
 }
 
 
-void PIClientServer::Client::enableSymmetricEncryption(const PIByteArray & key) {
+void PIClientServer::ClientBase::enableSymmetricEncryption(const PIByteArray & key) {
 	if (key.isNotEmpty()) {
 		stream.setCryptEnabled(true);
 		stream.setCryptKey(key);
@@ -80,46 +61,35 @@ void PIClientServer::Client::enableSymmetricEncryption(const PIByteArray & key)
 }
 
 
-void PIClientServer::Client::createForServer(PIEthernet * tcp_) {
-	tcp = tcp_;
-	tcp->setParameter(PIEthernet::KeepConnection, false);
-	init();
-}
-
-
-void PIClientServer::Client::init() {
+void PIClientServer::ClientBase::init() {
 	if (!tcp) return;
 	CONNECTL(&stream, sendRequest, [this](const PIByteArray & ba) {
 		if (!can_write) return;
 		tcp->send(ba);
 		// piMSleep(1);
 	});
-	CONNECTL(&stream, packetReceiveEvent, [this](PIByteArray & ba) {
-		if (readed_func) readed_func(ba);
-		readed(ba);
-	});
+	CONNECTL(&stream, packetReceiveEvent, [this](PIByteArray & ba) { readed(ba); });
 	CONNECTL(tcp, threadedReadEvent, [this](const uchar * readed, ssize_t size) {
 		if (!can_write) return;
 		stream.received(readed, size);
 	});
 	CONNECTL(tcp, connected, [this]() {
 		can_write = true;
-		piCout << "Connected";
+		// piCout << "Connected";
 		connected();
 	});
 	CONNECTL(tcp, disconnected, [this](bool) {
 		can_write = false;
 		stream.clear();
-		piCout << "Disconnected";
+		// piCout << "Disconnected";
 		disconnected();
 	});
 }
 
 
-void PIClientServer::Client::destroy() {
+void PIClientServer::ClientBase::destroy() {
 	can_write = false;
 	write_mutex.lock();
 	piDeleteSafety(tcp);
-	aboutDelete();
-	piCout << "Destroyed";
+	// piCout << "Destroyed";
 }

libs/client_server/piclientserver_server.cpp

@@ -16,9 +16,9 @@
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#include "piclient_server_server.h"
+#include "piclientserver_server.h"
 
-#include "piclient_server_client.h"
+#include "piclientserver_client.h"
 #include "piethernet.h"
 #include "piliterals_time.h"
 
@@ -26,7 +26,7 @@
 PIClientServer::Server::Server() {
 	tcp_server     = new PIEthernet(PIEthernet::TCP_Server);
 	clean_thread   = new PIThread();
-	client_factory = [] { return new Client(); };
+	client_factory = [] { return new ServerClient(); };
 	CONNECTL(tcp_server, newConnection, [this](PIEthernet * c) {
 		PIMutexLocker guard(clients_mutex);
 		if (clients.size_s() >= max_clients) {
@@ -45,7 +45,7 @@ PIClientServer::Server::Server() {
 
 	clean_thread->start(
 		[this]() {
-			PIVector<Client *> to_delete;
+			PIVector<ServerClient *> to_delete;
 			clients_mutex.lock();
 			for (auto c: clients) {
 				const PIEthernet * eth = c->getTCP();
@@ -58,6 +58,7 @@ PIClientServer::Server::Server() {
 				clients.removeOne(c);
 			clients_mutex.unlock();
 			for (auto c: to_delete) {
+				c->aboutDelete();
 				c->destroy();
 				delete c;
 			}
@@ -71,6 +72,7 @@ PIClientServer::Server::~Server() {
 	piDeleteSafety(clean_thread);
 	stopServer();
 	for (auto c: clients) {
+		c->aboutDelete();
 		c->destroy();
 		delete c;
 	}
@@ -81,6 +83,7 @@ PIClientServer::Server::~Server() {
 void PIClientServer::Server::listen(PINetworkAddress addr) {
 	if (!tcp_server) return;
 	stopServer();
+	is_closing = false;
 	tcp_server->listen(addr, true);
 	// piCout << "Listen on" << addr.toString();
 }
@@ -91,6 +94,21 @@ void PIClientServer::Server::setMaxClients(int new_max_clients) {
 }
 
 
+int PIClientServer::Server::clientsCount() const {
+	PIMutexLocker guard(clients_mutex);
+	return clients.size_s();
+}
+
+
+void PIClientServer::Server::forEachClient(std::function<void(ServerClient *)> func) {
+	PIMutexLocker guard(clients_mutex);
+	for (auto * c: clients) {
+		func(c);
+		if (is_closing) break;
+	}
+}
+
+
 void PIClientServer::Server::enableSymmetricEncryption(const PIByteArray & key) {
 	crypt_key = key;
 }
@@ -98,14 +116,15 @@ void PIClientServer::Server::enableSymmetricEncryption(const PIByteArray & key)
 
 void PIClientServer::Server::stopServer() {
 	if (!tcp_server) return;
+	is_closing = true;
+	tcp_server->stopThreadedListen();
 	tcp_server->stopAndWait();
 }
 
 
-void PIClientServer::Server::newClient(Client * c) {
+void PIClientServer::Server::newClient(ServerClient * c) {
 	clients << c;
 	c->enableSymmetricEncryption(crypt_key);
-	c->readed_func = [this, c](PIByteArray ba) { readed(c, ba); };
 	c->tcp->startThreadedRead();
 	c->connected();
 	piCout << "New client";

libs/main/client_server/piclientserver_client.h

@@ -0,0 +1,69 @@
+/*! \file piclientserver_client.h
+ * \ingroup ClientServer
+ * \~\brief
+ * \~english
+ * \~russian
+ */
+/*
+    PIP - Platform Independent Primitives
+	Ivan Pelipenko peri4ko@yandex.ru
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    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 Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef piclientserver_client_H
+#define piclientserver_client_H
+
+#include "piclientserver_client_base.h"
+
+
+namespace PIClientServer {
+
+
+// ServerClient
+
+class PIP_CLIENT_SERVER_EXPORT ServerClient: public ClientBase {
+	friend class Server;
+	NO_COPY_CLASS(ServerClient);
+
+public:
+	ServerClient() {}
+
+protected:
+	virtual void aboutDelete() {}
+
+private:
+	void createForServer(PIEthernet * tcp_);
+};
+
+
+// Client
+
+class PIP_CLIENT_SERVER_EXPORT Client: public ClientBase {
+	NO_COPY_CLASS(Client);
+
+public:
+	Client();
+	~Client();
+
+	void connect(PINetworkAddress addr);
+
+protected:
+
+private:
+};
+
+} // namespace PIClientServer
+
+#endif

libs/main/client_server/piclientserver_client_base.h

@@ -1,4 +1,4 @@
-/*! \file piclient_server_client.h
+/*! \file piclientserver_client_base.h
  * \ingroup ClientServer
  * \~\brief
  * \~english
@@ -22,8 +22,8 @@
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#ifndef piclient_server_client_H
-#define piclient_server_client_H
+#ifndef piclientserver_client_base_H
+#define piclientserver_client_base_H
 
 #include "pinetworkaddress.h"
 #include "pip_client_server_export.h"
@@ -35,18 +35,14 @@ namespace PIClientServer {
 
 class Server;
 
-class PIP_CLIENT_SERVER_EXPORT Client: public PIObject {
+class PIP_CLIENT_SERVER_EXPORT ClientBase {
 	friend class Server;
-	NO_COPY_CLASS(Client);
-	PIOBJECT(Client);
+	NO_COPY_CLASS(ClientBase);
 
 public:
-	Client();
-	virtual ~Client();
+	ClientBase();
+	virtual ~ClientBase();
 
-	void createNew();
-
-	void connect(PINetworkAddress addr);
 	const PIEthernet * getTCP() const { return tcp; }
 
 	void stop();
@@ -60,19 +56,18 @@ protected:
 	virtual void readed(PIByteArray data) {}
 	virtual void connected() {}
 	virtual void disconnected() {}
-	virtual void aboutDelete() {}
 
-private:
-	void createForServer(PIEthernet * tcp_);
 	void init();
-	void destroy();
 
 	bool own_tcp               = false;
 	std::atomic_bool can_write = {true};
 	PIEthernet * tcp           = nullptr;
+
+private:
+	void destroy();
+
 	PIStreamPacker stream;
 	mutable PIMutex write_mutex;
-	std::function<void(PIByteArray data)> readed_func = nullptr;
 };
 
 } // namespace PIClientServer

libs/main/client_server/piclientserver_server.h

@@ -1,4 +1,4 @@
-/*! \file piclient_server_server.h
+/*! \file piclientserver_server.h
  * \ingroup ClientServer
  * \~\brief
  * \~english
@@ -22,8 +22,8 @@
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#ifndef piclient_server_server_H
-#define piclient_server_server_H
+#ifndef piclientserver_server_H
+#define piclientserver_server_H
 
 #include "pimutex.h"
 #include "pinetworkaddress.h"
@@ -34,7 +34,7 @@ class PIThread;
 
 namespace PIClientServer {
 
-class Client;
+class ServerClient;
 
 class PIP_CLIENT_SERVER_EXPORT Server {
 public:
@@ -46,24 +46,24 @@ public:
 
 	int getMaxClients() const { return max_clients; }
 	void setMaxClients(int new_max_clients);
+	int clientsCount() const;
+	void forEachClient(std::function<void(ServerClient *)> func);
 
-	void setClientFactory(std::function<Client *()> f) { client_factory = f; }
+	void setClientFactory(std::function<ServerClient *()> f) { client_factory = f; }
 
 	void enableSymmetricEncryption(const PIByteArray & key);
 
-protected:
-	virtual void readed(Client * c, PIByteArray data) {}
-
 private:
 	void stopServer();
-	void newClient(Client * c);
+	void newClient(ServerClient * c);
 
-	std::function<Client *()> client_factory;
-	PIEthernet * tcp_server = nullptr;
-	PIThread * clean_thread = nullptr;
+	std::function<ServerClient *()> client_factory;
+	std::atomic_bool is_closing = {false};
+	PIEthernet * tcp_server     = nullptr;
+	PIThread * clean_thread     = nullptr;
 	PIByteArray crypt_key;
-	PIVector<Client *> clients;
-	PIMutex clients_mutex;
+	PIVector<ServerClient *> clients;
+	mutable PIMutex clients_mutex;
 
 	int max_clients = 1000;
 };

libs/main/io_devices/piethernet.cpp

@@ -303,18 +303,12 @@ bool PIEthernet::closeDevice() {
 	// piCoutObj << "close";
 	bool ned   = connected_;
 	connected_ = connecting_ = false;
-	server_thread_.stop();
-	PRIVATE->event.interrupt();
-	if (server_thread_.isRunning()) {
-		if (!server_thread_.waitForFinish(1_s)) server_thread_.terminate();
-	}
-	PRIVATE->event.destroy();
-	if (sock_s == sock) sock_s = -1;
-	closeSocket(sock);
-	closeSocket(sock_s);
+	stopThreadedListen();
 	clients_mutex.lock();
-	piDeleteAllAndClear(clients_);
+	auto cl = clients_;
+	clients_.clear();
 	clients_mutex.unlock();
+	piDeleteAll(cl);
 	if (ned) {
 		// piCoutObj << "Disconnect on close";
 		disconnected(false);
@@ -528,16 +522,32 @@ bool PIEthernet::listen(const PINetworkAddress & addr, bool threaded) {
 	return listen(threaded);
 }
 
+
+void PIEthernet::stopThreadedListen() {
+	server_thread_.stop();
+	PRIVATE->event.interrupt();
+	if (server_thread_.isRunning()) {
+		if (!server_thread_.waitForFinish(1_s)) server_thread_.terminate();
+	}
+	PRIVATE->event.destroy();
+	if (sock_s == sock) sock_s = -1;
+	closeSocket(sock);
+	closeSocket(sock_s);
+}
+
+
 PIEthernet * PIEthernet::client(int index) {
 	PIMutexLocker locker(clients_mutex);
 	return clients_[index];
 }
 
+
 int PIEthernet::clientsCount() const {
 	PIMutexLocker locker(clients_mutex);
 	return clients_.size_s();
 }
 
+
 PIVector<PIEthernet *> PIEthernet::clients() const {
 	PIMutexLocker locker(clients_mutex);
 	return clients_;

libs/main/io_devices/piethernet.h

@@ -251,6 +251,8 @@ public:
 	//! Start listen for incoming TCP connections on address "addr". Use only for TCP_Server
 	bool listen(const PINetworkAddress & addr, bool threaded = false);
 
+	void stopThreadedListen();
+
 	PIEthernet * client(int index);
 	int clientsCount() const;
 	PIVector<PIEthernet *> clients() const;

libs/main/io_devices/piiodevice.h

@@ -243,6 +243,10 @@ public:
 	//! \~russian Возвращает запущен ли поток чтения
 	bool isThreadedRead() const;
 
+	//! \~english Returns if threaded read is stopping
+	//! \~russian Возвращает останавливается ли поток чтения
+	bool isThreadedReadStopping() const { return read_thread.isStopping(); }
+
 	//! \~english Start threaded read
 	//! \~russian Запускает потоковое чтение
 	void startThreadedRead();
@@ -565,8 +569,6 @@ protected:
 
 	static PIIODevice * newDeviceByPrefix(const char * prefix);
 
-	bool isThreadedReadStopping() const { return read_thread.isStopping(); }
-
 	DeviceMode mode_ = ReadOnly;
 	DeviceOptions options_;
 	ReadRetFunc func_read = nullptr;

main.cpp

@@ -1,6 +1,6 @@
 #include "pibytearray.h"
-#include "piclient_server_client.h"
-#include "piclient_server_server.h"
+#include "piclientserver_client.h"
+#include "piclientserver_server.h"
 #include "picodeparser.h"
 #include "piiostream.h"
 #include "pijson.h"
@@ -21,10 +21,32 @@ enum MyEvent {
 PIKbdListener kbd;
 
 
+class MyServerClient: public PIClientServer::ServerClient {
+protected:
+	void readed(PIByteArray data) override { piCout << "readed" << (data.size()); }
+	void aboutDelete() override { piCout << "aboutDelete"; }
+	void disconnected() override { piCout << "disconnected"; }
+	void connected() override {
+		piCout << "connected";
+		send_thread.start(
+			[this] {
+				// write((PIString::fromNumber(++counter)).toUTF8());
+				PIByteArray ba(64_KiB);
+				write(ba);
+			},
+			2_Hz);
+	}
+	PIThread send_thread;
+	int counter = 0;
+};
+
+
 class MyClient: public PIClientServer::Client {
 protected:
 	void readed(PIByteArray data) override { piCout << "readed" << (data.size()); }
+	void disconnected() override { piCout << "disconnected"; }
 	void connected() override {
+		piCout << "connected";
 		send_thread.start(
 			[this] {
 				// write((PIString::fromNumber(++counter)).toUTF8());
@@ -43,26 +65,43 @@ int main(int argc, char * argv[]) {
 
 
 	PIClientServer::Server * s = nullptr;
-	PIClientServer::Client * c = nullptr;
+	PIThread s_thread;
+	PIVector<PIClientServer::Client *> cv;
 
 	if (argc > 1) {
 		piCout << "Server";
 		s = new PIClientServer::Server();
-		s->setClientFactory([] { return new MyClient(); });
+		s->setClientFactory([] { return new MyServerClient(); });
 		s->enableSymmetricEncryption("1122334455667788"_hex);
 		s->listenAll(12345);
+		s_thread.start(
+			[s] {
+				piCout << "*** clients" << s->clientsCount();
+				int i = 0;
+				s->forEachClient([&i](PIClientServer::ServerClient * c) {
+					piCout << "client" << ++i << c;
+					c->write(PIByteArray(16_KiB));
+					piMSleep(200);
+				});
+			},
+			0.5_Hz);
 	} else {
 		piCout << "Client";
-		c = new MyClient();
-		c->createNew();
-		c->enableSymmetricEncryption("1122334455667788"_hex);
-		c->connect(PINetworkAddress::resolve("127.0.0.1", 12345));
+		piForTimes(5) {
+			piMSleep(50);
+			auto c = new MyClient();
+			c->enableSymmetricEncryption("1122334455667788"_hex);
+			c->connect(PINetworkAddress::resolve("127.0.0.1", 12345));
+			cv << c;
+		}
 	}
 
 	WAIT_FOR_EXIT;
 
+	s_thread.stopAndWait();
+
 	piDeleteSafety(s);
-	piDeleteSafety(c);
+	piDeleteAllAndClear(cv);
 
 	return 0;
 	/*PIPackedTCP * tcp_s =