import os import sys import toml import time import re import tempfile import pycurl from bs4 import BeautifulSoup from urllib.parse import urlencode from io import BytesIO DOMAIN = os.environ["DOMAIN"] SUBDOMAIN = os.environ["SUBDOMAIN"] USER = os.environ["USER"] PASSWORD = os.environ["PASSWORD"] LXC_IP = os.environ["LXC_IP"] BASE_URL = os.environ["BASE_URL"].rstrip("/") DEFAULTS = { "base_url": BASE_URL, "path": "/", "logged_on_sso": False, "expect_title": None, "expect_content": None, "expect_title": None, "expect_effective_url": None, } # Example of expected conf: # ============================================== # #home.path = "/" # home.expect_title = "Login - Nextcloud" # # #dash.path = "/" # dash.logged_on_sso = true # dash.expect_title = "Tableau de bord - Nextcloud" # # admin.path = "/settings/admin" # admin.logged_on_sso = true # admin.expect_title = "Paramètres d'administration - Nextcloud" # # asset.path = "/core/img/logo/logo.svg" # # file.path = "/remote.php/dav/files/__USER__/Readme.md" # file.logged_on_sso = true # file.expect_content = "# Welcome to Nextcloud!" # # caldav.base_url = "https://yolo.test" # caldav.path = "/.well-known/caldav" # caldav.logged_on_sso = true # caldav.expect_content = "This is the WebDAV interface." # ============================================== def curl(base_url, path, method="GET", use_cookies=None, save_cookies=None, post=None): domain = base_url.replace("https://", "").replace("http://", "").split("/")[0] c = pycurl.Curl() # curl c.setopt(c.URL, f"{base_url}{path}") # https://domain.tld/foo/bar c.setopt(c.FOLLOWLOCATION, True) # --location c.setopt(c.SSL_VERIFYPEER, False) # --insecure c.setopt(c.RESOLVE, [f"{DOMAIN}:80:{LXC_IP}", f"{DOMAIN}:443:{LXC_IP}", f"{SUBDOMAIN}:80:{LXC_IP}", f"{SUBDOMAIN}:443:{LXC_IP}"]) # --resolve c.setopt(c.HTTPHEADER, [f"Host: {domain}", "X-Requested-With: libcurl"]) # --header if use_cookies: c.setopt(c.COOKIEFILE, use_cookies) if save_cookies: c.setopt(c.COOKIEJAR, save_cookies) if post: c.setopt(c.POSTFIELDS, urlencode(post)) buffer = BytesIO() c.setopt(c.WRITEDATA, buffer) c.perform() effective_url = c.getinfo(c.EFFECTIVE_URL) return_code = c.getinfo(c.RESPONSE_CODE) try: return_content = buffer.getvalue().decode() except UnicodeDecodeError: return_content = "(Binary content?)" c.close() return (return_code, return_content, effective_url) def test(base_url, path, post=None, logged_on_sso=False, expect_return_code=200, expect_content=None, expect_title=None, expect_effective_url=None): if logged_on_sso: cookies = tempfile.NamedTemporaryFile().name domain = base_url.replace("https://", "").replace("http://", "").split("/")[0] code, content, _ = curl(f"https://{domain}/yunohost/portalapi", "/login", save_cookies=cookies, post={"credentials": f"{USER}:{PASSWORD}"}) assert code == 200 and content == "Logged in", f"Failed to log in: got code {code} and content: {content}" else: cookies = None code = None retried = 0 while code is None or code in {502, 503, 504}: time.sleep(retried * 5) code, content, effective_url = curl(base_url, path, post=post, use_cookies=cookies) retried += 1 if retried > 3: break try: title = BeautifulSoup(content, features="lxml").find("title").string title = title.strip().replace("\u2013", "-") except Exception: title = "" content = BeautifulSoup(content, features="lxml").find("body").get_text().strip() content = re.sub(r"[\t\n\s]{3,}", "\n\n", content) errors = [] if expect_effective_url is None and "/yunohost/sso" in effective_url: errors.append(f"The request was redirected to yunohost's portal ({effective_url})") if expect_effective_url and expect_effective_url != effective_url: errors.append(f"Ended up on URL '{effective_url}', but was expecting '{expect_effective_url}'") if expect_return_code and code != expect_return_code: errors.append(f"Got return code {code}, but was expecting {expect_return_code}") if expect_title is None and "Welcome to nginx" in title: errors.append("The request ended up on the default nginx page?") if expect_title and not re.search(expect_title, title): errors.append(f"Got title '{title}', but was expecting something containing '{expect_title}'") if expect_content and not re.search(expect_content, content): errors.append(f"Did not find pattern '{expect_content}' in the page content: '{content[:50]}' (on URL {effective_url})") return { "url": f"{base_url}{path}", "effective_url": effective_url, "code": code, "title": title, "content": content, "errors": errors, } def run(tests): results = {} for name, params in tests.items(): full_params = DEFAULTS.copy() full_params.update(params) for key, value in full_params.items(): if isinstance(value, str): full_params[key] = value.replace("__USER__", USER) results[name] = test(**full_params) display_result(results[name]) if full_params["path"] == "/": full_params["path"] = "" results[name + "_noslash"] = test(**full_params) display_result(results[name]) return results def display_result(result): print(f"URL : {result['url']}") print(f"Real URL: {result['effective_url']}") print(f"Code : {result['code']}") print(f"Title : {result['title'].strip()}") print(f"Content extract:\n{result['content'][:250].strip()}") if result["errors"]: print("Errors :\n -" + "\n -".join(result['errors'])) print("\033[1m\033[91mFAIL\033[0m") else: print("\033[1m\033[92mOK\033[0m") print("========") def main(): tests = sys.stdin.read() if not tests: tests = "home.path = '/'" tests = toml.loads(tests) results = run(tests) # If there was at least one error 50x if any(str(r['code']).startswith("5") for r in results.values()): sys.exit(5) elif any(r["errors"] for r in results.values()): sys.exit(1) else: sys.exit(0) main()