1 /* <!-- copyright */
2 /*
3 * aria2 - The high speed download utility
4 *
5 * Copyright (C) 2011 Tatsuhiro Tsujikawa
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (at your option) any later version.
11 *
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 *
21 * In addition, as a special exception, the copyright holders give
22 * permission to link the code of portions of this program with the
23 * OpenSSL library under certain conditions as described in each
24 * individual source file, and distribute linked combinations
25 * including the two.
26 * You must obey the GNU General Public License in all respects
27 * for all of the code used other than OpenSSL. If you modify
28 * file(s) with this exception, you may extend this exception to your
29 * version of the file(s), but you are not obligated to do so. If you
30 * do not wish to do so, delete this exception statement from your
31 * version. If you delete this exception statement from all source
32 * files in the program, then also delete it here.
33 */
34 /* copyright --> */
35 #include "rpc_helper.h"
36 #include "XmlParser.h"
37 #include "RpcRequest.h"
38 #include "XmlRpcRequestParserStateMachine.h"
39 #include "message.h"
40 #include "DlAbortEx.h"
41 #include "DownloadEngine.h"
42 #include "RpcMethod.h"
43 #include "RpcResponse.h"
44 #include "RpcMethodFactory.h"
45 #include "LogFactory.h"
46 #include "fmt.h"
47
48 namespace aria2 {
49
50 namespace rpc {
51
52 #ifdef ENABLE_XML_RPC
xmlParseMemory(const char * xml,size_t size)53 RpcRequest xmlParseMemory(const char* xml, size_t size)
54 {
55 XmlRpcRequestParserStateMachine psm;
56 if (xml::XmlParser(&psm).parseFinal(xml, size) < 0) {
57 throw DL_ABORT_EX(MSG_CANNOT_PARSE_XML_RPC_REQUEST);
58 }
59 std::unique_ptr<List> params;
60 if (downcast<List>(psm.getCurrentFrameValue())) {
61 params.reset(static_cast<List*>(psm.popCurrentFrameValue().release()));
62 }
63 else {
64 params = List::g();
65 }
66 return {psm.getMethodName(), std::move(params)};
67 }
68 #endif // ENABLE_XML_RPC
69
createJsonRpcErrorResponse(int code,const std::string & msg,std::unique_ptr<ValueBase> id)70 RpcResponse createJsonRpcErrorResponse(int code, const std::string& msg,
71 std::unique_ptr<ValueBase> id)
72 {
73 auto params = Dict::g();
74 params->put("code", Integer::g(code));
75 params->put("message", msg);
76 return rpc::RpcResponse{code, rpc::RpcResponse::AUTHORIZED, std::move(params),
77 std::move(id)};
78 }
79
processJsonRpcRequest(Dict * jsondict,DownloadEngine * e)80 RpcResponse processJsonRpcRequest(Dict* jsondict, DownloadEngine* e)
81 {
82 auto id = jsondict->popValue("id");
83 if (!id) {
84 return createJsonRpcErrorResponse(-32600, "Invalid Request.", Null::g());
85 }
86 const String* methodName = downcast<String>(jsondict->get("method"));
87 if (!methodName) {
88 return createJsonRpcErrorResponse(-32600, "Invalid Request.",
89 std::move(id));
90 }
91 std::unique_ptr<List> params;
92 auto tempParams = jsondict->popValue("params");
93 if (downcast<List>(tempParams)) {
94 params.reset(static_cast<List*>(tempParams.release()));
95 }
96 else if (!tempParams) {
97 params = List::g();
98 }
99 else {
100 // TODO No support for Named params
101 return createJsonRpcErrorResponse(-32602, "Invalid params.", std::move(id));
102 }
103 A2_LOG_INFO(fmt("Executing RPC method %s", methodName->s().c_str()));
104 RpcRequest req = {methodName->s(), std::move(params), std::move(id), true};
105 return getMethod(methodName->s())->execute(std::move(req), e);
106 }
107
108 } // namespace rpc
109
110 } // namespace aria2
111