feat: download mcaptcha cli and setup mcaptcha server for testing
This commit is contained in:
parent
32a43b5445
commit
b6732958c9
2 changed files with 148 additions and 0 deletions
114
scripts/create_captcha.py
Normal file
114
scripts/create_captcha.py
Normal file
|
@ -0,0 +1,114 @@
|
||||||
|
# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
|
||||||
|
#
|
||||||
|
# SPDX-License-Identifier: Apache-2.0
|
||||||
|
# SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
|
import sys
|
||||||
|
from urllib.parse import urlparse, urlunparse, urlencode, parse_qsl
|
||||||
|
from requests import Session
|
||||||
|
|
||||||
|
|
||||||
|
def clean_url(url):
|
||||||
|
parsed = urlparse(url)
|
||||||
|
return urlunparse((parsed.scheme, parsed.netloc, "", "", "", ""))
|
||||||
|
|
||||||
|
|
||||||
|
def set_path(url: str, path: str) -> str:
|
||||||
|
parsed = urlparse(url)
|
||||||
|
return urlunparse((parsed.scheme, parsed.netloc, path, "", "", ""))
|
||||||
|
|
||||||
|
|
||||||
|
def help():
|
||||||
|
print("Usage: create_captcha.py <option>")
|
||||||
|
print("OPTIONS:\ncreate_captcha <mcaptcha_instance_url> <username> <password>")
|
||||||
|
print("get_secret <mcaptcha_instance_url> <username> <password>")
|
||||||
|
print("widget_url <mcaptcha_instance_url> <sitekey>")
|
||||||
|
|
||||||
|
|
||||||
|
c = Session()
|
||||||
|
if len(sys.argv) < 1:
|
||||||
|
help()
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
option = sys.argv[1]
|
||||||
|
|
||||||
|
|
||||||
|
def login():
|
||||||
|
if len(sys.argv) < 4:
|
||||||
|
help()
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
instance_url = clean_url(sys.argv[2])
|
||||||
|
username = sys.argv[3]
|
||||||
|
password = sys.argv[4]
|
||||||
|
|
||||||
|
url = set_path(instance_url, "/api/v1/signin")
|
||||||
|
payload = {"login": username, password: password}
|
||||||
|
resp = c.post(url, json=payload)
|
||||||
|
assert resp.status_code == 200
|
||||||
|
|
||||||
|
|
||||||
|
def create_captcha():
|
||||||
|
instance_url = clean_url(sys.argv[2])
|
||||||
|
username = sys.argv[3]
|
||||||
|
password = sys.argv[4]
|
||||||
|
|
||||||
|
levels = [
|
||||||
|
{"difficulty_factor": 50, "visitor_threshold": 50},
|
||||||
|
{"difficulty_factor": 500, "visitor_threshold": 500},
|
||||||
|
]
|
||||||
|
payload = {
|
||||||
|
"levels": levels,
|
||||||
|
"duration": 30,
|
||||||
|
"description": "create_captcha_test_script",
|
||||||
|
"publish_benchmarks": False,
|
||||||
|
}
|
||||||
|
url = set_path(instance_url, "/api/v1/mcaptcha/create")
|
||||||
|
|
||||||
|
resp = c.post(url, json=payload, cookies=c.cookies.get_dict())
|
||||||
|
assert resp.status_code == 200
|
||||||
|
return resp.json()["key"]
|
||||||
|
|
||||||
|
|
||||||
|
def get_secret():
|
||||||
|
if len(sys.argv) < 3:
|
||||||
|
help()
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
instance_url = clean_url(sys.argv[2])
|
||||||
|
username = sys.argv[3]
|
||||||
|
password = sys.argv[4]
|
||||||
|
|
||||||
|
instance_url = clean_url(sys.argv[2])
|
||||||
|
sitekey = sys.argv[3]
|
||||||
|
|
||||||
|
url = set_path(instance_url, "/api/v1/account/secret/get")
|
||||||
|
resp = c.get(url, cookies=c.cookies.get_dict())
|
||||||
|
assert resp.status_code == 200
|
||||||
|
return resp.json()["secret"]
|
||||||
|
|
||||||
|
|
||||||
|
def widget_url():
|
||||||
|
if len(sys.argv) < 4:
|
||||||
|
help()
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
instance_url = set_path(clean_url(sys.argv[2]), "/widget")
|
||||||
|
sitekey = sys.argv[3]
|
||||||
|
query = urlencode({"sitekey": sitekey})
|
||||||
|
url = f"{instance_url}?{query}"
|
||||||
|
|
||||||
|
return url
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
if option == "create_captcha":
|
||||||
|
login()
|
||||||
|
print(create_captcha())
|
||||||
|
elif option == "get_secret":
|
||||||
|
login()
|
||||||
|
print(get_secret())
|
||||||
|
elif option == "widget_url":
|
||||||
|
print(widget_url())
|
||||||
|
else:
|
||||||
|
help()
|
34
scripts/download-cli.sh
Executable file
34
scripts/download-cli.sh
Executable file
|
@ -0,0 +1,34 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
|
||||||
|
#
|
||||||
|
# SPDX-License-Identifier: Apache-2.0
|
||||||
|
# SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
|
readonly PROJECT_ROOT=$(pwd)
|
||||||
|
readonly SOURCE=https://dl.mcaptcha.org/mcaptcha/cli/0.3.0/x86_64-unknown-linux-gnu.tar.gz
|
||||||
|
readonly BIN_PATH=tmp/cli/mcaptcha-cli
|
||||||
|
readonly TARBALL=mcaptcha-cli.tar.gz
|
||||||
|
|
||||||
|
|
||||||
|
download() {
|
||||||
|
if [ ! -e $BIN_PATH ];
|
||||||
|
then
|
||||||
|
mkdir -p tmp/cli
|
||||||
|
cd $BIN_PATH
|
||||||
|
wget $SOURCE --quiet --output-document=$TARBALL $SOURCE
|
||||||
|
tar -xvzf $TARBALL > /dev/null
|
||||||
|
rm $TARBALL
|
||||||
|
echo "[*] Downloaded mCaptcha CLI"
|
||||||
|
mv build/x86_64-unknown-linux-gnu/mcaptcha-cli .
|
||||||
|
rm -rf build/
|
||||||
|
cd $PROJECT_ROOT
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 1: widget URI
|
||||||
|
gen_pow() {
|
||||||
|
$BIN_PATH widget-url $2 | cut -d ':' -f 2 | tr -d ' '
|
||||||
|
}
|
||||||
|
|
||||||
|
$1 $@
|
Loading…
Reference in a new issue