diff --git a/core b/core index 1d5021d..c708d1c 160000 --- a/core +++ b/core @@ -1 +1 @@ -Subproject commit 1d5021dc60f94aa68efa683c104c2e9d223927d3 +Subproject commit c708d1c8d15c6e7f4b50a7c6525225860e431014 diff --git a/libs/libSocks5 b/libs/libSocks5 index 2b47c62..8bf0038 160000 --- a/libs/libSocks5 +++ b/libs/libSocks5 @@ -1 +1 @@ -Subproject commit 2b47c622d4cebd5e566ab70d060801bf3882ab30 +Subproject commit 8bf003896275a2994c58f6519c8a44ff944fac26 diff --git a/teamServer/teamServer/TeamServer.cpp b/teamServer/teamServer/TeamServer.cpp index 705c698..cf3c764 100644 --- a/teamServer/teamServer/TeamServer.cpp +++ b/teamServer/teamServer/TeamServer.cpp @@ -238,6 +238,7 @@ grpc::Status TeamServer::AddListener(grpc::ServerContext* context, const teamser int localPort = listenerToCreate->port(); string localHost = listenerToCreate->ip(); std::shared_ptr listenerTcp = make_shared(localHost, localPort); + listenerTcp->setIsPrimary(); m_listeners.push_back(std::move(listenerTcp)); m_logger->info("AddListener Tcp {0}:{1}", localHost, std::to_string(localPort)); @@ -249,6 +250,7 @@ grpc::Status TeamServer::AddListener(grpc::ServerContext* context, const teamser int localPort = listenerToCreate->port(); string localHost = listenerToCreate->ip(); std::shared_ptr listenerHttp = make_shared(localHost, localPort, configHttp, false); + listenerHttp->setIsPrimary(); m_listeners.push_back(std::move(listenerHttp)); m_logger->info("AddListener Http {0}:{1}", localHost, std::to_string(localPort)); @@ -260,6 +262,7 @@ grpc::Status TeamServer::AddListener(grpc::ServerContext* context, const teamser int localPort = listenerToCreate->port(); string localHost = listenerToCreate->ip(); std::shared_ptr listenerHttps = make_shared(localHost, localPort, configHttps, true); + listenerHttps->setIsPrimary(); m_listeners.push_back(std::move(listenerHttps)); m_logger->info("AddListener Https {0}:{1}", localHost, std::to_string(localPort)); @@ -269,6 +272,7 @@ grpc::Status TeamServer::AddListener(grpc::ServerContext* context, const teamser std::string token = listenerToCreate->token(); std::string project = listenerToCreate->project(); std::shared_ptr listenerGithub = make_shared(project, token); + listenerGithub->setIsPrimary(); m_listeners.push_back(std::move(listenerGithub)); m_logger->info("AddListener Github {0}:{1}", project, token); @@ -278,6 +282,7 @@ grpc::Status TeamServer::AddListener(grpc::ServerContext* context, const teamser std::string domain = listenerToCreate->domain(); int port = listenerToCreate->port(); std::shared_ptr listenerDns = make_shared(domain, port); + listenerDns->setIsPrimary(); m_listeners.push_back(std::move(listenerDns)); m_logger->info("AddListener Dns {0}:{1}", domain, std::to_string(port)); @@ -517,6 +522,8 @@ void TeamServer::runSocksServer(int port, const std::string& listenerHash, const socksListener = m_listeners[i]; } + m_logger->info("SocksListener ok"); + std::shared_ptr session; for(int kk=0; kkgetNumberOfSession(); kk++) { @@ -524,6 +531,8 @@ void TeamServer::runSocksServer(int port, const std::string& listenerHash, const session = socksListener->getSessionPtr(kk); } + m_logger->info("Session ok"); + // Start socks beacon side C2Message c2MessageStartSocks; c2MessageStartSocks.set_instruction(Socks5Cmd); @@ -754,7 +763,7 @@ grpc::Status TeamServer::SendCmdToSession(grpc::ServerContext* context, const te // start the server and launch a thread to handle socks messages int port = std::atoi(c2Message.data().c_str()); - std::thread t(&TeamServer::runSocksServer, this, port, listenerHash, beaconHash); + std::thread t(&TeamServer::runSocksServer, this, port, m_listeners[i]->getListenerHash(), beaconHash); t.detach(); }