add PIClientServer::Config, common configuration type for both sides, stream packer and encryption settings

This commit is contained in:
2024-09-13 11:08:32 +03:00
parent 996b7ea403
commit 17b902ebcc
8 changed files with 125 additions and 25 deletions

View File

@@ -47,6 +47,7 @@ PIClientServer::Client::~Client() {
void PIClientServer::Client::connect(PINetworkAddress addr) { void PIClientServer::Client::connect(PINetworkAddress addr) {
if (!tcp || !own_tcp) return; if (!tcp || !own_tcp) return;
close(); close();
config.apply(this);
tcp->connect(addr, true); tcp->connect(addr, true);
tcp->startThreadedRead(); tcp->startThreadedRead();
piCout << "Connect to" << addr.toString(); piCout << "Connect to" << addr.toString();

View File

@@ -53,15 +53,6 @@ int PIClientServer::ClientBase::write(const void * d, const size_t s) {
} }
void PIClientServer::ClientBase::enableSymmetricEncryption(const PIByteArray & key) {
if (key.isNotEmpty()) {
stream.setCryptEnabled(true);
stream.setCryptKey(key);
} else
stream.setCryptEnabled(false);
}
void PIClientServer::ClientBase::init() { void PIClientServer::ClientBase::init() {
if (!tcp) return; if (!tcp) return;
CONNECTL(&stream, sendRequest, [this](const PIByteArray & ba) { CONNECTL(&stream, sendRequest, [this](const PIByteArray & ba) {

View File

@@ -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_config.h"
#include "piclientserver_client_base.h"
void PIClientServer::Config::setPacketSign(ushort sign) {
packet_sign = sign;
}
void PIClientServer::Config::setPacketSize(int bytes) {
packet_size = bytes;
}
void PIClientServer::Config::enableSymmetricEncryption(const PIByteArray & key) {
crypt_key = key;
}
void PIClientServer::Config::apply(ClientBase * client) {
auto & s(client->stream);
s.setPacketSign(packet_sign);
s.setMaxPacketSize(packet_size);
if (crypt_key.isNotEmpty()) {
s.setCryptEnabled(true);
s.setCryptKey(crypt_key);
} else
s.setCryptEnabled(false);
}

View File

@@ -110,11 +110,6 @@ void PIClientServer::Server::forEachClient(std::function<void(ServerClient *)> f
} }
void PIClientServer::Server::enableSymmetricEncryption(const PIByteArray & key) {
crypt_key = key;
}
void PIClientServer::Server::stopServer() { void PIClientServer::Server::stopServer() {
if (!tcp_server) return; if (!tcp_server) return;
is_closing = true; is_closing = true;
@@ -125,7 +120,7 @@ void PIClientServer::Server::stopServer() {
void PIClientServer::Server::newClient(ServerClient * c) { void PIClientServer::Server::newClient(ServerClient * c) {
clients << c; clients << c;
c->enableSymmetricEncryption(crypt_key); config.apply(c);
c->tcp->startThreadedRead(); c->tcp->startThreadedRead();
c->connected(); c->connected();
piCout << "New client"; piCout << "New client";

View File

@@ -25,7 +25,7 @@
#ifndef piclientserver_client_base_H #ifndef piclientserver_client_base_H
#define piclientserver_client_base_H #define piclientserver_client_base_H
#include "pinetworkaddress.h" #include "piclientserver_config.h"
#include "pip_client_server_export.h" #include "pip_client_server_export.h"
#include "pistreampacker.h" #include "pistreampacker.h"
@@ -36,6 +36,7 @@ namespace PIClientServer {
class Server; class Server;
class PIP_CLIENT_SERVER_EXPORT ClientBase { class PIP_CLIENT_SERVER_EXPORT ClientBase {
friend class Config;
friend class Server; friend class Server;
NO_COPY_CLASS(ClientBase); NO_COPY_CLASS(ClientBase);
@@ -50,7 +51,7 @@ public:
int write(const void * d, const size_t s); int write(const void * d, const size_t s);
int write(const PIByteArray & ba) { return write(ba.data(), ba.size()); } int write(const PIByteArray & ba) { return write(ba.data(), ba.size()); }
void enableSymmetricEncryption(const PIByteArray & key); Config & configuration() { return config; }
protected: protected:
virtual void readed(PIByteArray data) {} virtual void readed(PIByteArray data) {}
@@ -62,6 +63,7 @@ protected:
bool own_tcp = false; bool own_tcp = false;
std::atomic_bool can_write = {true}; std::atomic_bool can_write = {true};
PIEthernet * tcp = nullptr; PIEthernet * tcp = nullptr;
Config config;
private: private:
void destroy(); void destroy();

View File

@@ -0,0 +1,60 @@
/*! \file piclientserver_config.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_config_H
#define piclientserver_config_H
#include "pibytearray.h"
#include "pip_client_server_export.h"
namespace PIClientServer {
class Server;
class Client;
class ClientBase;
class PIP_CLIENT_SERVER_EXPORT Config {
friend class Server;
friend class Client;
public:
void setPacketSign(ushort sign);
void setPacketSize(int bytes);
void enableSymmetricEncryption(const PIByteArray & key);
protected:
void apply(ClientBase * client);
PIByteArray crypt_key;
ushort packet_sign = 0xAFBE;
int packet_size = 1400;
private:
};
} // namespace PIClientServer
#endif

View File

@@ -25,6 +25,7 @@
#ifndef piclientserver_server_H #ifndef piclientserver_server_H
#define piclientserver_server_H #define piclientserver_server_H
#include "piclientserver_config.h"
#include "pimutex.h" #include "pimutex.h"
#include "pinetworkaddress.h" #include "pinetworkaddress.h"
#include "pip_client_server_export.h" #include "pip_client_server_export.h"
@@ -55,7 +56,7 @@ public:
void setClientFactory(std::function<ServerClient *()> f) { client_factory = f; } void setClientFactory(std::function<ServerClient *()> f) { client_factory = f; }
void enableSymmetricEncryption(const PIByteArray & key); Config & configuration() { return config; }
private: private:
void stopServer(); void stopServer();
@@ -67,7 +68,7 @@ private:
PIEthernet * tcp_server = nullptr; PIEthernet * tcp_server = nullptr;
PIThread * clean_thread = nullptr; PIThread * clean_thread = nullptr;
PIThreadNotifier clean_notifier; PIThreadNotifier clean_notifier;
PIByteArray crypt_key; Config config;
PIVector<ServerClient *> clients; PIVector<ServerClient *> clients;
mutable PIMutex clients_mutex; mutable PIMutex clients_mutex;

View File

@@ -27,7 +27,7 @@ public:
~MyServerClient() { send_thread.stopAndWait(); } ~MyServerClient() { send_thread.stopAndWait(); }
protected: protected:
// void readed(PIByteArray data) override { piCout << "readed" << (data.size()); } void readed(PIByteArray data) override { piCout << "readed" << (data.size()); }
// void aboutDelete() override { piCout << "aboutDelete"; } // void aboutDelete() override { piCout << "aboutDelete"; }
// void disconnected() override { piCout << "disconnected"; } // void disconnected() override { piCout << "disconnected"; }
void connected() override { void connected() override {
@@ -50,7 +50,7 @@ public:
~MyClient() { send_thread.stopAndWait(); } ~MyClient() { send_thread.stopAndWait(); }
protected: protected:
// void readed(PIByteArray data) override { piCout << "readed" << (data.size()); } void readed(PIByteArray data) override { piCout << "readed" << (data.size()); }
void disconnected() override { piCout << "disconnected"; } void disconnected() override { piCout << "disconnected"; }
void connected() override { void connected() override {
piCout << "connected"; piCout << "connected";
@@ -101,7 +101,7 @@ int main(int argc, char * argv[]) {
piCout << "Server"; piCout << "Server";
s = new PIClientServer::Server(); s = new PIClientServer::Server();
s->setClientFactory([] { return new MyServerClient(); }); s->setClientFactory([] { return new MyServerClient(); });
s->enableSymmetricEncryption("1122334455667788"_hex); s->configuration().enableSymmetricEncryption("1122334455667788"_hex);
s->listenAll(12345); s->listenAll(12345);
s_thread->start( s_thread->start(
[s] { [s] {
@@ -110,17 +110,17 @@ int main(int argc, char * argv[]) {
s->forEachClient([&i](PIClientServer::ServerClient * c) { s->forEachClient([&i](PIClientServer::ServerClient * c) {
// piCout << "client" << ++i << c; // piCout << "client" << ++i << c;
c->write(PIByteArray(1_KiB)); c->write(PIByteArray(1_KiB));
c->close(); // c->close();
// piMSleep(200); // piMSleep(200);
}); });
}, },
1._Hz); 1._Hz);
} else { } else {
piCout << "Client"; piCout << "Client";
piForTimes(20) { piForTimes(2) {
piMSleep(25); piMSleep(25);
auto c = new MyClient(); auto c = new MyClient();
c->enableSymmetricEncryption("1122334455667788"_hex); c->configuration().enableSymmetricEncryption("1122334455667788"_hex);
c->connect(PINetworkAddress::resolve("127.0.0.1", 12345)); c->connect(PINetworkAddress::resolve("127.0.0.1", 12345));
cv << c; cv << c;
} }