//// //// Created by Anonymous275 on 3/3/2020. //// #define ENET_IMPLEMENTATION #include "include/enet.h" #include #include #include #include #include #include #include #define DEFAULT_BUFLEN 64000 #define DEFAULT_PORT "4445" typedef struct { ENetHost *host; ENetPeer *peer; } Client; std::queue RUDPData; std::queue RUDPToSend; bool Terminate = false; void CoreNetworkThread(); #pragma clang diagnostic pop #pragma clang diagnostic pop void HandleEvent(ENetEvent event,Client client){ switch (event.type){ case ENET_EVENT_TYPE_CONNECT: printf("Client Connected port : %u.\n",event.peer->address.port); //Name Of the Server event.peer->data = (void *)"Connected Server"; break; case ENET_EVENT_TYPE_RECEIVE: printf("Received: %s\n",event.packet->data); RUDPData.push(reinterpret_cast(event.packet->data)); enet_packet_destroy (event.packet); break; case ENET_EVENT_TYPE_DISCONNECT: printf ("%s disconnected.\n", (char *)event.peer->data); // Reset the peer's client information. event.peer->data = NULL; break; case ENET_EVENT_TYPE_DISCONNECT_TIMEOUT: printf ("%s timeout.\n", (char *)event.peer->data); event.peer->data = NULL; break; case ENET_EVENT_TYPE_NONE: break; } } void RUDPClientThread(const std::string& IP, int Port){ if (enet_initialize() != 0) { std::cout << "An error occurred while initializing ENet.\n"; } Client client; ENetAddress address = {0}; address.host = ENET_HOST_ANY; address.port = Port; std::cout << "starting client...\n"; enet_address_set_host(&address, IP.c_str()); client.host = enet_host_create(nullptr, 1, 2, 0, 0); client.peer = enet_host_connect(client.host, &address, 2, 0); if (client.peer == nullptr) { std::cout << "could not connect\n"; } do { ENetEvent event; enet_host_service(client.host, &event, 0); HandleEvent(event,client); //Handles the Events while (!RUDPToSend.empty()){ ENetPacket* packet = enet_packet_create (RUDPToSend.front().c_str(), RUDPToSend.front().size()+1, ENET_PACKET_FLAG_RELIABLE); //Create A reliable packet using the data enet_peer_send(client.peer, 0, packet); std::cout << "sending : " << RUDPToSend.front() << std::endl; RUDPToSend.pop(); } Sleep(10); } while (!Terminate); } #pragma clang diagnostic push #pragma clang diagnostic ignored "-Wmissing-noreturn" void TCPServerThread(const std::string& IP, int Port){ Terminate = false; WSADATA wsaData; int iResult; SOCKET ListenSocket = INVALID_SOCKET; SOCKET ClientSocket = INVALID_SOCKET; struct addrinfo *result = NULL; struct addrinfo hints; int iSendResult; char recvbuf[DEFAULT_BUFLEN]; int recvbuflen = DEFAULT_BUFLEN; // Initialize Winsock iResult = WSAStartup(MAKEWORD(2,2), &wsaData); if (iResult != 0) { std::cout <<"WSAStartup failed with error: " << iResult << std::endl; } ZeroMemory(&hints, sizeof(hints)); hints.ai_family = AF_INET; hints.ai_socktype = SOCK_STREAM; hints.ai_protocol = IPPROTO_TCP; hints.ai_flags = AI_PASSIVE; // Resolve the server address and port iResult = getaddrinfo(NULL, DEFAULT_PORT, &hints, &result); if ( iResult != 0 ) { std::cout << "getaddrinfo failed with error: " << iResult << std::endl; WSACleanup(); } // Create a socket for connecting to server ListenSocket = socket(result->ai_family, result->ai_socktype, result->ai_protocol); if (ListenSocket == INVALID_SOCKET) { std::cout << "socket failed with error: " << WSAGetLastError() << std::endl; freeaddrinfo(result); WSACleanup(); } // Setup the TCP listening socket iResult = bind(ListenSocket, result->ai_addr, (int)result->ai_addrlen); if (iResult == SOCKET_ERROR) { std::cout << "bind failed with error: " << WSAGetLastError() << std::endl; freeaddrinfo(result); closesocket(ListenSocket); WSACleanup(); } freeaddrinfo(result); iResult = listen(ListenSocket, SOMAXCONN); if (iResult == SOCKET_ERROR) { std::cout << "listen failed with error: " << WSAGetLastError() << std::endl; closesocket(ListenSocket); WSACleanup(); } ClientSocket = accept(ListenSocket, NULL, NULL); if (ClientSocket == INVALID_SOCKET) { std::cout << "accept failed with error: " << WSAGetLastError() << std::endl; closesocket(ListenSocket); WSACleanup(); } closesocket(ListenSocket); std::thread t1(RUDPClientThread,IP,Port); t1.detach(); do { while(!RUDPData.empty()){ iSendResult = send( ClientSocket, RUDPData.front().c_str(), int(RUDPData.front().length())+1, 0); if (iSendResult == SOCKET_ERROR) { std::cout << "send failed with error: " << WSAGetLastError() << std::endl; closesocket(ClientSocket); WSACleanup(); }else{ RUDPData.pop(); std::cout << "Bytes sent: " << iSendResult << std::endl; } } iResult = recv(ClientSocket, recvbuf, recvbuflen, 0); if (iResult > 0) { std::string buff = recvbuf; buff.resize(iResult); RUDPToSend.push(buff); std::cout << "size : " << buff.size() << std::endl; std::cout << "Data : " << buff.c_str() << std::endl; } else if (iResult == 0) { std::cout << "Connection closing...\n"; } else { std::cout << "recv failed with error: " << WSAGetLastError() << std::endl; closesocket(ClientSocket); WSACleanup(); } } while (iResult > 0); iResult = shutdown(ClientSocket, SD_SEND); if (iResult == SOCKET_ERROR) { std::cout << "shutdown failed with error: " << WSAGetLastError() << std::endl; closesocket(ClientSocket); WSACleanup(); } closesocket(ClientSocket); WSACleanup(); Terminate = true; } void ProxyStart(){ std::thread t1(CoreNetworkThread); std::cout << "Core Network Started!\n"; t1.join(); } void ProxyThread(const std::string& IP, int port){ Terminate = false; std::thread t1(TCPServerThread,IP,port); t1.detach(); }