支持ftp命令的远程cmd

#include "stdafx.h" #include "CSever.h" CSever::~CSever() { } CSever* CSever::serverptr; CSever::CSever(int PORT, bool BroadcastPublically) { WSAData wsaData; WORD DllVersion = MAKEWORD(2, 2); if (WSAStartup(DllVersion, &wsaData) != 0) { MessageBoxA(NULL, "WinSock startup failed", "Error", MB_OK | MB_ICONERROR); exit(1); } if (BroadcastPublically) addr.sin_addr.s_addr = htonl(INADDR_ANY); else addr.sin_addr.s_addr = inet_addr("127.0.0.1"); addr.sin_port = htons(PORT); addr.sin_family = AF_INET; sListen = socket(AF_INET, SOCK_STREAM, NULL); if (bind(sListen, (SOCKADDR*)&addr, sizeof(addr)) == SOCKET_ERROR) { std::string ErrorMsg = "Failed to bind the address to our listening socket. Winsock Error:" + std::to_string(WSAGetLastError()); MessageBoxA(NULL, ErrorMsg.c_str(), "Error", MB_OK | MB_ICONERROR); exit(1); } if (listen(sListen, SOMAXCONN) == SOCKET_ERROR) { std::string ErrorMsg = "Failed to listen on listening socket. Winsock Error:" + std::to_string(WSAGetLastError()); MessageBoxA(NULL, ErrorMsg.c_str(), "Error", MB_OK | MB_ICONERROR); exit(1); } serverptr = this; CreateThread(NULL, NULL, (LPTHREAD_START_ROUTINE)PacketSenderThread, NULL, NULL, NULL); } void CSever::ListenForNewConnection() { CreateThread(NULL, NULL, (LPTHREAD_START_ROUTINE)ListenerThread, NULL, NULL, NULL); } void CSever::HandleInput() { std::string userinput; int inputInt; currentSessionID = -1; while (true) { std::getline(std::cin, userinput); if (currentSessionID == -1) { if (CData::processParameter(userinput, "connect")) { inputInt = atoi(userinput.c_str()); int tempInt = connections.size() - 1; if (inputInt > tempInt) CData::outputMsg("Session doesn't exist.", 2); else { currentSessionID = inputInt; CData::outputMsg("Connected to Session " + std::to_string(currentSessionID), 1); } inputInt = 0; userinput.empty(); } else CData::outputMsg("Please connect to a session with 'connect'", 2); } else { if (userinput == "exitSession") { CData::outputMsg("Exited Session " + std::to_string(currentSessionID), 1); currentSessionID = -1; } else if (userinput.find("remoteControl") != std::string::npos) { CData::cmdMode = !CData::cmdMode; SendString(currentSessionID, userinput, PacketType::Instruction); } else if (CData::cmdMode) { SendString(currentSessionID, userinput, PacketType::CMDCommand); } else { SendString(currentSessionID, userinput, PacketType::Instruction); } } } } bool CSever::ProcessPacket(int ID, PacketType _packettype) { switch (_packettype) { case PacketType::Instruction: { std::string message; if (!GetString(ID, message)) return false; CData::outputMsg("ID [" + std::to_string(ID) + "]: " + message, 1); break; } case PacketType::CMDCommand: { std::string message; if (!GetString(ID, message)) return false; CData::outputMsg(message, 3); break; } case PacketType::Warning: { std::string message; if (!GetString(ID, message)) CData::outputMsg("ID [" + std::to_string(ID) + "]: " + message, 2); break; } default: { std::cout << "Unrecognized packet: " << (int32_t)_packettype << std::endl; break; } } return true; } void CSever::ClientHandlerThread(int ID) { PacketType packettype; while (true) { if (!serverptr->GetPacketType(ID, packettype)) break; if (!serverptr->ProcessPacket(ID, packettype)) break; } std::cout << "Lost connection to client ID: " << ID << std::endl; serverptr->DisconnectClient(ID); return; } void CSever::PacketSenderThread() { while (true) { for (size_t i = 0; i < serverptr->connections.size(); i++) { if (serverptr->connections[i]->pm.HasPendingPackets()) { CPacket p = serverptr->connections[i]->pm.Retrieve(); if (!serverptr->sendall(i, p.buffer, p.size)) { std::cout << "Failed to send packet to ID: " << i << std::endl; } delete p.buffer; } } Sleep(5); } } void CSever::ListenerThread() { while (true) { SOCKET newConnectionSocket = accept(serverptr->sListen, (SOCKADDR*)&serverptr->addr, &serverptr->addrlen); //Accept a new connection if (newConnectionSocket == 0) //If accepting the client connection failed { std::cout << "Failed to accept the client's connection." << std::endl; } else //If client connection properly accepted { std::lock_guard<std::mutex> lock(serverptr->connectionMgr_mutex); int NewConnectionID = serverptr->connections.size(); if (serverptr->UnusedConnections > 0) { for (size_t i = 0; i < serverptr->connections.size(); i++) { if (serverptr->connections[i]->ActiveConnection == false) { serverptr->connections[i]->socket = newConnectionSocket; serverptr->connections[i]->ActiveConnection = true; NewConnectionID = i; serverptr->UnusedConnections -= 1; break; } } } else //add new connection to the socket { std::shared_ptr<Connection> newConnection(new Connection(newConnectionSocket)); serverptr->connections.push_back(newConnection); } std::cout << "Client Connected! ID:" << NewConnectionID << " | IP: " << inet_ntoa(serverptr->addr.sin_addr) << std::endl; CreateThread(NULL, NULL, (LPTHREAD_START_ROUTINE)ClientHandlerThread, (LPVOID)(NewConnectionID), NULL, NULL); } } } void CSever::DisconnectClient(int ID) { currentSessionID = -1; std::lock_guard<std::mutex> lock(connectionMgr_mutex); if (connections[ID]->ActiveConnection == false) { return; } connections[ID]->pm.Clear(); connections[ID]->ActiveConnection = false; closesocket(connections[ID]->socket); if (ID == (connections.size() - 1)) { connections.pop_back(); for (size_t i = connections.size() - 1; i >= 0 && connections.size() > 0; i--) { if (connections[i]->ActiveConnection) break; connections.pop_back(); UnusedConnections -= 1; } } else { UnusedConnections += 1; } } bool CSever::recvall(int ID, char * data, int totalbytes) { int bytesreceived = 0; while (bytesreceived < totalbytes) { int RetnCheck = recv(connections[ID]->socket, data, totalbytes - bytesreceived, NULL); if (RetnCheck == SOCKET_ERROR) return false; bytesreceived += RetnCheck; } return true; } bool CSever::sendall(int ID, char * data, int totalbytes) { int bytessent = 0; while (bytessent < totalbytes) { int RetnCheck = send(connections[ID]->socket, data + bytessent, totalbytes - bytessent, NULL); if (RetnCheck == SOCKET_ERROR) return false; bytessent += RetnCheck; } return true; } bool CSever::Sendint32_t(int ID, int32_t _int32_t) { _int32_t = htonl(_int32_t); if (!sendall(ID, (char*)&_int32_t, sizeof(int32_t))) //Try to send long (4 byte int) return false; return true; } bool CSever::Getint32_t(int ID, int32_t & _int32_t) { if (!recvall(ID, (char*)&_int32_t, sizeof(int32_t))) return false; _int32_t = ntohl(_int32_t); return true; } bool CSever::SendPacketType(int ID, PacketType _packettype) { if (!Sendint32_t(ID, (int32_t)_packettype)) return false; return true; } //Try to receive packet type bool CSever::GetPacketType(int ID, PacketType & _packettype) { int packettype; if (!Getint32_t(ID, packettype)) return false; _packettype = (PacketType)packettype; return true; } void CSever::SendString(int ID, std::string & _string, PacketType _packettype) { PS::Message message(_string); if (ID == -2) { for (int i = 0; i < connections.size(); i++) { connections[i]->pm.Append(message.toPacket(_packettype)); } } else connections[ID]->pm.Append(message.toPacket(_packettype)); } bool CSever::GetStrin