// ---------------< construct a general message to be sent as acknowledgement >--------- HttpMessage ClientHandler::constructMessage(HttpMessage msg) { msg.removeAttribute("content-length"); std::string bodyString = "<file>" + msg.findValue("file") + "</file>"; std::string sizeString = Converter<size_t>::toString(bodyString.size()); msg.addAttribute(HttpMessage::Attribute("content-length", sizeString)); msg.addBody(bodyString); return msg; }
//----< 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; }
int main() { Utils::Title("Testing HttpMessage class", '='); HttpMessage msg; msg.addAttribute(HttpMessage::attribute("Command", "GetFiles")); msg.addAttribute(HttpMessage::attribute("ToAddr", "127.0.0.1:8080")); msg.addAttribute(HttpMessage::attribute("FromAddr", "127.0.0.1:8081")); msg.addAttribute(HttpMessage::attribute("Mode", "OneWay")); msg.addAttribute(HttpMessage::attribute("content_length", "10")); msg.addBody(std::vector<HttpMessage::byte> { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' }); std::cout << "\n" << Utils::addHeaderAndFooterLines(msg.toString()); Utils::title("testing headerString(), bodyString(), and toString()"); std::cout << "\nheader string:"; std::cout << "\n" << msg.headerString(); std::cout << "\nbody string:"; std::cout << "\n" << msg.bodyString(); std::cout << "\n\nmessage string:"; std::cout << "\n" << StringHelper::addHeaderAndFooterLines(msg.toString()); Utils::title("Testing removeAttribute"); putline(); msg.removeAttribute("content_length"); std::cout << msg.toString(); putline(); Utils::title("Testing addBody(const std::string&)"); std::string msgBody = "<msg>this is a message</msg>"; msg.addAttribute(HttpMessage::Attribute("content_length", Converter<size_t>::toString(msgBody.size()))); msg.addBody(msgBody); std::cout << "\n" << StringHelper::addHeaderAndFooterLines(msg.toString()); Utils::title("Testing parseAttribute(const std::string&)"); std::string test2 = "name:value"; std::cout << "\n input = \"" << test2 << "\""; Attribute attrib2 = HttpMessage::parseAttribute(test2); std::cout << "\n result is: " << HttpMessage::attribString(attrib2); test2 = " name : value "; std::cout << "\n input = \"" << test2 << "\""; attrib2 = HttpMessage::parseAttribute(test2); std::cout << "\n result is: " << HttpMessage::attribString(attrib2); Utils::title("Testing Message parsing"); MockSocket sock(msg); HttpMessage msg2; while (true) { std::string line = sock.recvString(); if (line.size() == 0) break; Attribute attrib = HttpMessage::parseAttribute(line); msg2.addAttribute(attrib); } Value val = msg2.findValue("content_length"); if (val.size() > 0) { size_t numBytes = Converter<size_t>::toValue(val); byte* pBuffer = new byte[numBytes]; sock.recv(numBytes, pBuffer); msg2.addBody(numBytes, pBuffer); } std::cout << "\n" << Utils::addHeaderAndFooterLines(msg2.toString()); std::cout << "\n\n"; }