|
@@ -1,3 +1,6 @@
|
|
|
|
+#include <algorithm>
|
|
|
|
+#include <set>
|
|
|
|
+
|
|
extern "C"
|
|
extern "C"
|
|
{
|
|
{
|
|
#include <lua.h>
|
|
#include <lua.h>
|
|
@@ -7,6 +10,8 @@ extern "C"
|
|
#include "../common/HTTPS.h"
|
|
#include "../common/HTTPS.h"
|
|
#include "../common/config.h"
|
|
#include "../common/config.h"
|
|
|
|
|
|
|
|
+static std::set<std::string> validMethod = {"GET", "HEAD", "POST", "PUT", "DELETE", "PATCH"};
|
|
|
|
+
|
|
static std::string w_checkstring(lua_State *L, int idx)
|
|
static std::string w_checkstring(lua_State *L, int idx)
|
|
{
|
|
{
|
|
size_t len;
|
|
size_t len;
|
|
@@ -34,20 +39,18 @@ static void w_readheaders(lua_State *L, int idx, HTTPSClient::header_map &header
|
|
lua_pop(L, 1);
|
|
lua_pop(L, 1);
|
|
}
|
|
}
|
|
|
|
|
|
-static HTTPSClient::Request::Method w_optmethod(lua_State *L, int idx, HTTPSClient::Request::Method defaultMethod)
|
|
|
|
|
|
+static std::string w_optmethod(lua_State *L, int idx, const std::string &defaultMethod)
|
|
{
|
|
{
|
|
if (lua_isnoneornil(L, idx))
|
|
if (lua_isnoneornil(L, idx))
|
|
return defaultMethod;
|
|
return defaultMethod;
|
|
|
|
|
|
- auto str = w_checkstring(L, idx);
|
|
|
|
- if (str == "get")
|
|
|
|
- return HTTPSClient::Request::GET;
|
|
|
|
- else if (str == "post")
|
|
|
|
- return HTTPSClient::Request::POST;
|
|
|
|
- else
|
|
|
|
- luaL_argerror(L, idx, "expected one of \"get\" or \"set\"");
|
|
|
|
|
|
+ std::string str = w_checkstring(L, idx);
|
|
|
|
+ std::transform(str.begin(), str.end(), str.begin(), [](unsigned char c) { return toupper(c); });
|
|
|
|
+
|
|
|
|
+ if (validMethod.find(str) == validMethod.end())
|
|
|
|
+ luaL_argerror(L, idx, "expected one of \"get\", \"head\", \"post\", \"put\", \"delete\", or \"patch\"");
|
|
|
|
|
|
- return defaultMethod;
|
|
|
|
|
|
+ return str;
|
|
}
|
|
}
|
|
|
|
|
|
static int w_request(lua_State *L)
|
|
static int w_request(lua_State *L)
|
|
@@ -61,13 +64,13 @@ static int w_request(lua_State *L)
|
|
{
|
|
{
|
|
advanced = true;
|
|
advanced = true;
|
|
|
|
|
|
- HTTPSClient::Request::Method defaultMethod = HTTPSClient::Request::GET;
|
|
|
|
|
|
+ std::string defaultMethod = "GET";
|
|
|
|
|
|
lua_getfield(L, 2, "data");
|
|
lua_getfield(L, 2, "data");
|
|
if (!lua_isnoneornil(L, -1))
|
|
if (!lua_isnoneornil(L, -1))
|
|
{
|
|
{
|
|
req.postdata = w_checkstring(L, -1);
|
|
req.postdata = w_checkstring(L, -1);
|
|
- defaultMethod = HTTPSClient::Request::POST;
|
|
|
|
|
|
+ defaultMethod = "POST";
|
|
}
|
|
}
|
|
lua_pop(L, 1);
|
|
lua_pop(L, 1);
|
|
|
|
|