// ---------------< read the contents of the message body, if it is a normal message >---------- HttpMessage ClientHandler::readBody(HttpMessage msg, Socket& socket) { size_t numBytes = 0; size_t pos = msg.findAttribute("content-length"); HttpMessage message = msg; if (pos < msg.attributes().size()) { numBytes = Converter<size_t>::toValue(msg.attributes()[pos].second); Socket::byte* buffer = new Socket::byte[numBytes + 1]; socket.recv(numBytes, buffer); buffer[numBytes] = '\0'; std::string msgBody(buffer); message.addBody(msgBody); delete[] buffer; } return message; }
HttpMessage MsgClient::readMessage(Socket& socket) { Show::write("Client readMessage \n"); //connectionClosed_ = false; HttpMessage msg; //read attribute while (true) { std::string attribString = socket.recvString('\n'); if (attribString.size() > 1) { HttpMessage::Attribute attrib = HttpMessage::parseAttribute(attribString); msg.addAttribute(attrib); } else { break; } } if (msg.attributes().size() == 0) { return msg; } if (msg.attributes()[0].first == "POST") { std::string filePath = msg.findValue("filePath"); if (filePath != "") { Show::write("Test for get FolderList"); size_t numBytes = 0; size_t pos = msg.findAttribute("content-length"); if (pos < msg.attributes().size()) { numBytes = Converter<size_t>::toValue(msg.attributes()[pos].second); Socket::byte* buffer = new Socket::byte[numBytes + 1]; socket.recv(numBytes, buffer); buffer[numBytes] = '\0'; std::string msgBody(buffer); msg.addBody(msgBody); Show::write("getFolderList:" + msgBody + "\n"); } } } return msg; }
// ------------------------------------ Starts Receiver for the this component --------------------------- void Server::startReceiverThread() { BlockingQueue<HttpMessage> msgQ; try { SocketSystem ss; SocketListener sl(getServerPort(), Socket::IP6); Receiver cp(msgQ,"Server"); sl.start(cp); /* * Since this is a server the loop below never terminates. * We could easily change that by sending a distinguished * message for shutdown. */ while (true) { HttpMessage msg = msgQ.deQ(); std::string cPort = msg.findValue("From-Port"); if (msg.attributes()[0].first == "GET") { // Checks if its an file extraction request std::string filename = msg.findValue("file"); Show::write("\n\n Client @: "+cPort+ " => requested file :: " + filename); sendServerFiles(msg); } else Show::write("\n\n Message from client @:"+cPort +" => " + msg.bodyString()); } } catch (std::exception& exc) { Show::write("\n Exeception caught: "); std::string exMsg = "\n " + std::string(exc.what()) + "\n\n"; Show::write(exMsg); } }
//----< this defines processing to frame messages >------------------ HttpMessage ClientHandler::readMessage(Socket& socket) { connectionClosed_ = false; HttpMessage msg; while (true) { // read message attributes std::string attribString = socket.recvString('\n'); if (attribString.size() > 1) { HttpMessage::Attribute attrib = HttpMessage::parseAttribute(attribString); msg.addAttribute(attrib); } else break; } if (msg.attributes().size() == 0) { // If client is done, connection breaks connectionClosed_ = true; // and recvString returns empty string return msg; } if (msg.attributes()[0].first == "POST") // read body if POST { if (msg.attributes()[0].second == "Message") msg = readBody(msg, socket); // case 0 - normal message else if (msg.attributes()[0].second == "File") { // case 1 - client sending file to server saveFileServer(msg, socket); msg = constructMessage(msg); } else if (msg.attributes()[0].second == "closePackage") msg = closePackage(msg); // case 7 - close a package else if (msg.attributes()[0].second == "returnFile") { // case 4 - server sends files to client saveFileClient(msg, socket); msg = constructMessage(msg); } } else if (msg.attributes()[0].first == "GET") { // read message if GET msg = readBody(msg, socket); if (msg.attributes()[0].second == "getFileList") msg = getFileList(msg); // case 5 - request list of packages else if (msg.attributes()[0].second == "getFileNameList") msg = getFileNameList(msg); // case 6 - request list of files in a packages else if (msg.attributes()[0].second == "File" || msg.attributes()[0].second == "FileWithDeps") msg = getFiles(msg, msg.attributes()[0].second); } else { msg.removeAttribute("content-length"); std::string bodyString = "<msg>Error message</msg>"; std::string sizeString = Converter<size_t>::toString(bodyString.size()); msg.addAttribute(HttpMessage::Attribute("content-length", sizeString)); msg.addBody(bodyString); } return msg; }
//----< receiver functionality is defined by this function >--------- void ClientHandler::operator()(Socket socket) { std::function<void()> threadProc = [&]() { while (true) { HttpMessage msg = readMessage(socket); // read a message from socket if (msg.attributes().size() == 0 || msg.bodyString() == "quit") break; if (msg.findValue("type") == "returnList") { sendMessage(msg); continue; } else if (msg.findValue("type") == "returnFiles") { Sender sndr(msg.findValue("fromAddr")); sndr.start(); std::vector<std::string> files = splitString(msg.bodyString()); for (std::string file : files) { std::string fileSpec = searchFile(file); FileSystem::FileInfo fi(fileSpec); FileSystem::File fileIn(fileSpec); fileIn.open(FileSystem::File::in, FileSystem::File::binary); if (!fileIn.isGood()) { std::cout << "\n could not open file " << file; continue; } size_t fileSize = fi.size(); std::string sizeString = Converter<size_t>::toString(fileSize); HttpMessage message = sndr.makeMessage(4, "", msg.findValue("toAddr")); message.addAttribute(HttpMessage::Attribute("file", fileSpec)); message.addAttribute(HttpMessage::Attribute("content-length", sizeString)); sndr.postMessage(message); } sndr.postMessage(sndr.makeMessage(0, "closeServer", msg.findValue("toAddr"))); sndr.postMessage(sndr.makeMessage(0, "quit", msg.findValue("toAddr"))); sndr.wait(); continue; } else if (msg.attributes()[0].second == "error") { sendMessage(msg); continue; } msgQ_.enQ(msg); } }; std::thread receiveThread(threadProc); receiveThread.join(); }