/* */ #include "ExpatXmlRpcRequestProcessor.h" #include #include #include "XmlRpcRequestParserStateMachine.h" #include "util.h" #include "DlAbortEx.h" #include "message.h" namespace aria2 { namespace rpc { namespace { struct SessionData { XmlRpcRequestParserStateMachine* stm_; std::stack charactersStack_; SessionData(XmlRpcRequestParserStateMachine* stm):stm_(stm) {} }; } // namespace namespace { void mlStartElement(void* userData, const char* name, const char** attrs) { SessionData* sd = reinterpret_cast(userData); std::map attrmap; if(attrs) { const char** p = attrs; while(*p != 0) { std::string name = *p++; if(*p == 0) { break; } std::string value = util::strip(*p++); attrmap[name] = value; } } sd->stm_->beginElement(name, attrmap); if(sd->stm_->needsCharactersBuffering()) { sd->charactersStack_.push(std::string()); } } } // namespace namespace { void mlEndElement(void* userData, const char* name) { SessionData* sd = reinterpret_cast(userData); std::string characters; if(sd->stm_->needsCharactersBuffering()) { characters = util::strip(sd->charactersStack_.top()); sd->charactersStack_.pop(); } sd->stm_->endElement(name, characters); } } // namespace namespace { void mlCharacters(void* userData, const char* ch, int len) { SessionData* sd = reinterpret_cast(userData); if(sd->stm_->needsCharactersBuffering()) { sd->charactersStack_.top() += std::string(&ch[0], &ch[len]); } } } // namespace RpcRequest XmlRpcRequestProcessor::parseMemory(const std::string& xml) { stm_.reset(new XmlRpcRequestParserStateMachine()); SharedHandle sessionData(new SessionData(stm_.get())); XML_Parser parser = XML_ParserCreate(0); XML_SetUserData(parser, sessionData.get()); XML_SetElementHandler(parser, &mlStartElement, &mlEndElement); XML_SetCharacterDataHandler(parser, &mlCharacters); int r = XML_Parse(parser, xml.data(), xml.size(), 1); XML_ParserFree(parser); if(r == XML_STATUS_ERROR) { throw DL_ABORT_EX(MSG_CANNOT_PARSE_XML_RPC_REQUEST); } if(!asList(stm_->getCurrentFrameValue())) { throw DL_ABORT_EX("Bad XML-RPC parameter list"); } return RpcRequest(stm_->getMethodName(), static_pointer_cast(stm_->getCurrentFrameValue())); } } // namespace rpc } // namespace aria2