mirror of
https://gitee.com/iresty/apisix.git
synced 2024-12-15 09:21:26 +08:00
77 lines
2.7 KiB
Python
77 lines
2.7 KiB
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.
|
|
#
|
|
|
|
import subprocess
|
|
import os
|
|
from pathlib import Path
|
|
import psutil
|
|
from boofuzz import FuzzLoggerText, Session, TCPSocketConnection, Target
|
|
|
|
def cur_dir():
|
|
return os.path.split(os.path.realpath(__file__))[0]
|
|
|
|
def apisix_pwd():
|
|
return os.environ.get("APISIX_FUZZING_PWD") or \
|
|
(str(Path.home()) + "/work/apisix/apisix")
|
|
|
|
def check_log():
|
|
boofuzz_log = cur_dir() + "/test.log"
|
|
apisix_errorlog = apisix_pwd() + "/logs/error.log"
|
|
apisix_accesslog = apisix_pwd() + "/logs/access.log"
|
|
|
|
cmds = ['cat %s | grep -a "error" | grep -v "invalid request body"'%apisix_errorlog, 'cat %s | grep -a " 500 "'%apisix_accesslog]
|
|
if os.path.exists(boofuzz_log):
|
|
cmds.append('cat %s | grep -a "fail"'%boofuzz_log)
|
|
for cmd in cmds:
|
|
r = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
|
|
err = r.stdout.read().strip()
|
|
print("Error in log: ", err)
|
|
assert err == b""
|
|
|
|
def check_process():
|
|
with open(apisix_pwd() + "/logs/nginx.pid") as f:
|
|
pid = int(f.read().strip())
|
|
parent = psutil.Process(pid)
|
|
children = parent.children(recursive=True)
|
|
process = {p.pid for p in children if "cache loader process" not in p.cmdline()[0]}
|
|
process.add(parent.pid)
|
|
return process
|
|
|
|
def initfuzz():
|
|
fw = open(cur_dir() + "/test.log",'w')
|
|
fuzz_loggers = [FuzzLoggerText(file_handle=fw)]
|
|
session = Session(
|
|
target=Target(
|
|
connection=TCPSocketConnection("127.0.0.1", 9080, send_timeout=5.0, recv_timeout=5.0, server=False)
|
|
),
|
|
fuzz_loggers=fuzz_loggers,
|
|
keep_web_open=False,
|
|
)
|
|
return session
|
|
|
|
def run_test(create_route, run):
|
|
# before test
|
|
create_route()
|
|
r1 = check_process()
|
|
run()
|
|
# after test
|
|
check_log()
|
|
r2 = check_process()
|
|
if r2 != r1:
|
|
print("before test, nginx's process list:%s,\nafter test, nginx's process list:%s"%(r1,r2))
|
|
raise AssertionError
|