ReachableCEO
54cc5f7308
- Implements Apache APISIX packaging for Cloudron platform. - Includes Dockerfile, CloudronManifest.json, and start.sh. - Configured to use Cloudron's etcd addon. 🤖 Generated with Gemini CLI Co-Authored-By: Gemini <noreply@google.com>
101 lines
3.0 KiB
Python
Executable File
101 lines
3.0 KiB
Python
Executable File
#! /usr/bin/env python
|
|
|
|
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
# contributor license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright ownership.
|
|
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
# (the "License"); you may not use this file except in compliance with
|
|
# the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
# This file provides a fuzzing test with different upstreams
|
|
import http.client
|
|
import json
|
|
import random
|
|
import threading
|
|
from public import check_leak, run_test, connect_admin
|
|
import yaml
|
|
|
|
REQ_PER_THREAD = 50
|
|
THREADS_NUM = 4
|
|
TOTOL_ROUTES = 10
|
|
|
|
def get_admin_key_from_yaml(yaml_file_path):
|
|
with open(yaml_file_path, 'r') as file:
|
|
yaml_data = yaml.safe_load(file)
|
|
try:
|
|
admin_key = yaml_data['deployment']['admin']['admin_key'][0]['key']
|
|
return admin_key
|
|
except KeyError:
|
|
return None
|
|
def create_route():
|
|
key = get_admin_key_from_yaml('conf/config.yaml')
|
|
if key is None:
|
|
print("Key not found in the YAML file.")
|
|
return
|
|
for i in range(TOTOL_ROUTES):
|
|
conn = connect_admin()
|
|
scheme = "http" if i % 2 == 0 else "https"
|
|
port = ":6666" if i % 2 == 0 else ":6667"
|
|
suffix = str(i + 1)
|
|
i = str(i)
|
|
conf = json.dumps({
|
|
"uri": "/*",
|
|
"host": "test" + i + ".com",
|
|
"plugins": {
|
|
},
|
|
"upstream": {
|
|
"scheme": scheme,
|
|
"nodes": {
|
|
"127.0.0." + suffix + port: 1
|
|
},
|
|
"type": "roundrobin"
|
|
},
|
|
})
|
|
|
|
conn.request("PUT", "/apisix/admin/routes/" + i, conf,
|
|
headers={
|
|
"X-API-KEY":key,
|
|
})
|
|
response = conn.getresponse()
|
|
assert response.status <= 300, response.read()
|
|
|
|
def req():
|
|
route_id = random.randrange(TOTOL_ROUTES)
|
|
conn = http.client.HTTPConnection("127.0.0.1", port=9080)
|
|
conn.request("GET", "/server_addr",
|
|
headers={
|
|
"Host":"test" + str(route_id) + ".com",
|
|
})
|
|
response = conn.getresponse()
|
|
assert response.status == 200, response.read()
|
|
ip = response.read().rstrip().decode()
|
|
suffix = str(route_id + 1)
|
|
assert "127.0.0." + suffix == ip, f"expect: 127.0.0.{suffix}, actual: {ip}"
|
|
|
|
def run_in_thread():
|
|
for i in range(REQ_PER_THREAD):
|
|
req()
|
|
|
|
@check_leak
|
|
def run():
|
|
th = [threading.Thread(target=run_in_thread) for i in range(THREADS_NUM)]
|
|
for t in th:
|
|
t.start()
|
|
for t in th:
|
|
t.join()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_test(create_route, run)
|
|
|