add PIClientServer::Config, common configuration type for both sides, stream packer and encryption settings
This commit is contained in:
@@ -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();
|
||||||
|
|||||||
@@ -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) {
|
||||||
|
|||||||
50
libs/client_server/piclientserver_config.cpp
Normal file
50
libs/client_server/piclientserver_config.cpp
Normal 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);
|
||||||
|
}
|
||||||
@@ -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";
|
||||||
|
|||||||
@@ -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();
|
||||||
|
|||||||
60
libs/main/client_server/piclientserver_config.h
Normal file
60
libs/main/client_server/piclientserver_config.h
Normal 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
|
||||||
@@ -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;
|
||||||
|
|
||||||
|
|||||||
12
main.cpp
12
main.cpp
@@ -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;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user