mirror of
https://github.com/warp-tech/warpgate.git
synced 2024-11-15 04:21:59 +08:00
94 lines
2.3 KiB
Python
94 lines
2.3 KiB
Python
import logging
|
|
import os
|
|
import requests
|
|
import socket
|
|
import subprocess
|
|
import threading
|
|
import time
|
|
|
|
|
|
last_port = 1234
|
|
|
|
mysql_client_ssl_opt = '--ssl'
|
|
mysql_client_opts = []
|
|
if 'GITHUB_ACTION' in os.environ:
|
|
# Github uses MySQL instead of MariaDB
|
|
mysql_client_ssl_opt = '--ssl-mode=REQUIRED'
|
|
mysql_client_opts = ['--enable-cleartext-plugin']
|
|
|
|
|
|
def alloc_port():
|
|
global last_port
|
|
last_port += 1
|
|
return last_port
|
|
|
|
|
|
def wait_port(port, recv=True):
|
|
logging.debug(f'Waiting for port {port}')
|
|
|
|
data = b''
|
|
|
|
def wait():
|
|
nonlocal data
|
|
while True:
|
|
try:
|
|
s = socket.create_connection(('localhost', port), timeout=5)
|
|
if recv:
|
|
data = s.recv(100)
|
|
else:
|
|
data = b''
|
|
s.close()
|
|
logging.debug(f'Port {port} is up')
|
|
return data
|
|
except socket.error:
|
|
time.sleep(0.1)
|
|
continue
|
|
|
|
t = threading.Thread(target=wait, daemon=True)
|
|
t.start()
|
|
t.join(timeout=5)
|
|
if t.is_alive():
|
|
raise Exception(f'Port {port} is not up')
|
|
return data
|
|
|
|
|
|
def wait_mysql_port(port):
|
|
logging.debug(f'Waiting for MySQL port {port}')
|
|
|
|
def wait():
|
|
while True:
|
|
try:
|
|
subprocess.check_call(f'mysql --user=root --password=123 --host=127.0.0.1 --port={port} --execute="show schemas;"', shell=True)
|
|
logging.debug(f'Port {port} is up')
|
|
break
|
|
except subprocess.CalledProcessError:
|
|
time.sleep(1)
|
|
continue
|
|
|
|
t = threading.Thread(target=wait, daemon=True)
|
|
t.start()
|
|
t.join(timeout=60)
|
|
if t.is_alive():
|
|
raise Exception(f'Port {port} is not up')
|
|
|
|
|
|
def create_ticket(url, username, target_name):
|
|
session = requests.Session()
|
|
session.verify = False
|
|
response = session.post(
|
|
f'{url}/@warpgate/api/auth/login',
|
|
json={
|
|
'username': 'admin',
|
|
'password': '123',
|
|
},
|
|
)
|
|
assert response.status_code // 100 == 2
|
|
response = session.post(
|
|
f'{url}/@warpgate/admin/api/tickets',
|
|
json={
|
|
'username': username,
|
|
'target_name': target_name,
|
|
},
|
|
)
|
|
assert response.status_code == 201
|
|
return response.json()['secret']
|