Fix request arguments not properly handled (#661)

tindy2013 · Nov 9, 2023 · c530bc5 · c530bc5
1 parent 833dee1
commit c530bc5
Show file tree

Hide file tree

Showing 12 changed files with 177 additions and 543 deletions.
diff --git a/src/handler/interfaces.cpp b/src/handler/interfaces.cpp
diff --git a/src/script/cron.cpp b/src/script/cron.cpp
@@ -73,7 +73,7 @@ void refresh_schedule()
 
 std::string list_cron_schedule(RESPONSE_CALLBACK_ARGS)
 {
-    std::string &argument = request.argument;
+    auto &argument = request.argument;
     std::string token = getUrlArg(argument, "token");
     rapidjson::StringBuffer sb;
     rapidjson::Writer<rapidjson::StringBuffer> writer(sb);

diff --git a/src/script/script_quickjs.cpp b/src/script/script_quickjs.cpp
@@ -377,6 +377,11 @@ static qjs_fetch_Response qjs_fetch(qjs_fetch_Request request)
     return response;
 }
 
+static std::string qjs_getUrlArg(const std::string &url, const std::string &request)
+{
+    return getUrlArg(url, request);
+}
+
 std::string getGeoIP(const std::string &address, const std::string &proxy)
 {
     return fetchFile("https://api.ip.sb/geoip/" + address, parseProxy(proxy), global.cacheConfig);
@@ -506,7 +511,7 @@ int script_context_init(qjs::Context &context)
             .add<&currentTime>("time")
             .add<&sleepMs>("sleep")
             .add<&ShowMsgbox>("msgbox")
-            .add<&getUrlArg>("getUrlArg")
+            .add<&qjs_getUrlArg>("getUrlArg")
             .add<&fileGet>("fileGet")
             .add<&fileWrite>("fileWrite");
         context.eval(R"(