diff --git a/libs/cloud/picloudtcp.cpp b/libs/cloud/picloudtcp.cpp index 80786a1d..5cf2f700 100644 --- a/libs/cloud/picloudtcp.cpp +++ b/libs/cloud/picloudtcp.cpp @@ -19,10 +19,24 @@ #include "picloudtcp.h" #include "picrypt.h" +#include "pichunkstream.h" + const char hash_def_key[] = "_picrypt_"; +PIByteArray & operator <<(PIByteArray & s, const PICloud::Header & v) { + s << v.version << v.type << v.sname; + return s; +} + + +PIByteArray & operator >>(PIByteArray & s, PICloud::Header & v) { + s >> v.version >> v.type >> v.sname; + return s; +} + + PICloudTCP::PICloudTCP() { } diff --git a/libs/main/cloud/picloudtcp.h b/libs/main/cloud/picloudtcp.h index 5b7f1309..5a22b0ab 100644 --- a/libs/main/cloud/picloudtcp.h +++ b/libs/main/cloud/picloudtcp.h @@ -26,6 +26,30 @@ #include "pip_cloud_export.h" #include "pistring.h" +namespace PICloud { + +enum Version { + Version_1 = 1, +}; + +enum HeaderType { + Server = 1, + Client = 2, +}; + +struct PIP_CLOUD_EXPORT Header { + Header() { + version = Version_1; + } + uchar version; // PICloud::Version + uchar type; // PICloud::HeaderType + PIString sname; // server name +}; +} + +PIP_CLOUD_EXPORT PIByteArray & operator <<(PIByteArray & s, const PICloud::Header & v); +PIP_CLOUD_EXPORT PIByteArray & operator >>(PIByteArray & s, PICloud::Header & v); + class PIP_CLOUD_EXPORT PICloudTCP { public: diff --git a/main.cpp b/main.cpp index 193896f4..3764b12d 100644 --- a/main.cpp +++ b/main.cpp @@ -2,17 +2,7 @@ int main() { PICloudServer s("127.0.0.1:10101"); -// for (int i=0; i<3; ++i) { -// s.open(); -// piCout << "opened"; -// piSleep(10); -// s.close(); -// piCout << "closed"; -// piSleep(1); -// } -// s.open(); s.startThreadedRead(); - piSleep(2); -// s.stopThreadedRead(); + piSleep(10); return 0; } diff --git a/utils/cloud_dispatcher/cloudserver.cpp b/utils/cloud_dispatcher/cloudserver.cpp index 3a7afcba..01ff10d1 100644 --- a/utils/cloud_dispatcher/cloudserver.cpp +++ b/utils/cloud_dispatcher/cloudserver.cpp @@ -1,5 +1,24 @@ #include "cloudserver.h" -CloudServer::CloudServer(DispatcherClient * client) { +CloudServer::CloudServer(DispatcherClient * c) : server(c) { } + + +CloudServer::~CloudServer() { + for (auto c :clients) { + c->close(); + } +} + + +void CloudServer::addClient(DispatcherClient * c) { + clients << c; +} + +void CloudServer::printStatus() { + piCout << " " << "Clients for" << server->address() << ":"; + for (auto c: clients) { + piCout << " " << c->address(); + } +} diff --git a/utils/cloud_dispatcher/cloudserver.h b/utils/cloud_dispatcher/cloudserver.h index b3b6f3ff..be44352a 100644 --- a/utils/cloud_dispatcher/cloudserver.h +++ b/utils/cloud_dispatcher/cloudserver.h @@ -3,10 +3,18 @@ #include "dispatcherclient.h" -class CloudServer -{ + +class CloudServer : public PIObject { + PIOBJECT(CloudServer) public: - CloudServer(DispatcherClient * client); + CloudServer(DispatcherClient * c); + ~CloudServer(); + void addClient(DispatcherClient * c); + EVENT_HANDLER0(void, printStatus); + +private: + DispatcherClient * server; + PIVector clients; }; #endif // CLOUDSERVER_H diff --git a/utils/cloud_dispatcher/dispatcherclient.cpp b/utils/cloud_dispatcher/dispatcherclient.cpp index 7991d353..b9cdfd64 100644 --- a/utils/cloud_dispatcher/dispatcherclient.cpp +++ b/utils/cloud_dispatcher/dispatcherclient.cpp @@ -1,8 +1,8 @@ #include "dispatcherclient.h" +#include "picloudtcp.h" -DispatcherClient::DispatcherClient(PIEthernet * eth_) { - eth = eth_; +DispatcherClient::DispatcherClient(PIEthernet * eth_) : eth(eth_), authorised(false) { CONNECTU(&disconnect_tm, tickEvent, eth, close); eth->startThreadedRead(); CONNECTU(eth, threadedReadEvent, this, readed); @@ -21,6 +21,10 @@ PIString DispatcherClient::address() { return eth->path(); } +void DispatcherClient::close() { + eth->close(); +} + void DispatcherClient::disconnected(bool withError) { piCoutObj << "client disconnected" << eth->sendAddress(); @@ -30,7 +34,18 @@ void DispatcherClient::disconnected(bool withError) { void DispatcherClient::readed(uchar *data, int size) { PIByteArray ba(data, size); - piCoutObj << "readed" << ba.toHex(); - eth->write(ba); + if (authorised) { + dataReaded(ba); + } else { + if (ba.size() < 4) return; + PICloud::Header hdr; + ba >> hdr; + if ((PICloud::HeaderType)hdr.type == PICloud::Server) { + registerServer(hdr.sname, this); + } + if ((PICloud::HeaderType)hdr.type == PICloud::Client) { + registerClient(hdr.sname, this); + } + } } diff --git a/utils/cloud_dispatcher/dispatcherclient.h b/utils/cloud_dispatcher/dispatcherclient.h index 00dbbfaf..99f7cc62 100644 --- a/utils/cloud_dispatcher/dispatcherclient.h +++ b/utils/cloud_dispatcher/dispatcherclient.h @@ -11,7 +11,10 @@ public: ~DispatcherClient(); EVENT1(disconnectEvent, DispatcherClient *, client) EVENT2(registerServer, PIString, sname, DispatcherClient *, client) + EVENT2(registerClient, PIString, sname, DispatcherClient *, client) PIString address(); + void close(); + EVENT1(dataReaded, PIByteArray, ba) private: EVENT_HANDLER2(void, readed, uchar * , data, int, size); @@ -19,6 +22,7 @@ private: PIEthernet * eth; PITimer disconnect_tm; + bool authorised; }; diff --git a/utils/cloud_dispatcher/dispatcherserver.cpp b/utils/cloud_dispatcher/dispatcherserver.cpp index 99bdd045..4b5a50cf 100644 --- a/utils/cloud_dispatcher/dispatcherserver.cpp +++ b/utils/cloud_dispatcher/dispatcherserver.cpp @@ -28,6 +28,7 @@ void DispatcherServer::printStatus() { auto it = c_servers.makeIterator(); while(it.next()){ piCout << " " << it.key(); + it.value()->printStatus(); } map_mutex.unlock(); } @@ -50,6 +51,13 @@ void DispatcherServer::newConnection(PIEthernet *cl) { c_servers.insert(sname, new CloudServer(c)); map_mutex.unlock(); }); + CONNECTL(client, registerClient, [this](PIString sname, DispatcherClient * c){ + map_mutex.lock(); + if (c_servers.contains(sname)) { + c_servers[sname]->addClient(c); + } + map_mutex.unlock(); + }); piCoutObj << "add client" << client; map_mutex.lock(); clients.push_back(client); diff --git a/utils/cloud_dispatcher/dispatcherserver.h b/utils/cloud_dispatcher/dispatcherserver.h index a7205686..7fba7ef4 100644 --- a/utils/cloud_dispatcher/dispatcherserver.h +++ b/utils/cloud_dispatcher/dispatcherserver.h @@ -1,7 +1,6 @@ #ifndef DISPATCHERSERVER_H #define DISPATCHERSERVER_H -#include "dispatcherclient.h" #include "cloudserver.h"