added inline tests

This commit is contained in:
amammad
2023-12-07 14:04:33 +01:00
parent 6ebdae3bab
commit 5795c72a99
5 changed files with 149 additions and 88 deletions

View File

@@ -7,70 +7,75 @@ app = FastAPI()
@app.post("/bomb")
async def bomb(file_path):
zipfile.ZipFile(file_path, "r").extract("file1")
zipfile.ZipFile(file_path, "r").extractall()
zipfile.ZipFile(file_path, "r").extract("file1") # $ result=BAD
zipfile.ZipFile(file_path, "r").extractall() # $ result=BAD
with zipfile.ZipFile(file_path) as myzip:
with myzip.open('ZZ') as myfile:
with myzip.open('ZZ') as myfile: # $ result=BAD
a = myfile.readline()
with zipfile.ZipFile(file_path) as myzip:
with myzip.open('ZZ', mode="w") as myfile:
with myzip.open('ZZ', mode="w") as myfile: # $result=OK
myfile.write(b"tmpppp")
zipfile.ZipFile(file_path).read("aFileNameInTheZipFile")
zipfile.ZipFile(file_path).read("aFileNameInTheZipFile") # $ result=BAD
tarfile.open(file_path).extractfile("file1.txt")
tarfile.TarFile.open(file_path).extract("somefile")
tarfile.TarFile.xzopen(file_path).extract("somefile")
tarfile.TarFile.gzopen(file_path).extractall()
tarfile.TarFile.open(file_path).extractfile("file1.txt")
tarfile.open(file_path).extractfile("file1.txt") # $ result=BAD
tarfile.TarFile.open(file_path).extract("somefile") # $ result=BAD
tarfile.TarFile.xzopen(file_path).extract("somefile") # $ result=BAD
tarfile.TarFile.gzopen(file_path).extractall() # $ result=BAD
tarfile.TarFile.open(file_path).extractfile("file1.txt") # $ result=BAD
tarfile.open(file_path, mode="w")
tarfile.TarFile.gzopen(file_path, mode="w")
tarfile.TarFile.open(file_path, mode="r:")
tarfile.open(file_path, mode="w") # $result=OK
tarfile.TarFile.gzopen(file_path, mode="w") # $result=OK
tarfile.TarFile.open(file_path, mode="r:") # $ result=BAD
import shutil
shutil.unpack_archive(file_path)
shutil.unpack_archive(file_path) # $ result=BAD
import lzma
lzma.open(file_path)
lzma.LZMAFile(file_path).read()
lzma.open(file_path) # $ result=BAD
lzma.LZMAFile(file_path).read() # $ result=BAD
import bz2
bz2.open(file_path)
bz2.BZ2File(file_path).read()
bz2.open(file_path) # $ result=BAD
bz2.BZ2File(file_path).read() # $ result=BAD
import gzip
gzip.open(file_path)
gzip.GzipFile(file_path)
gzip.open(file_path) # $ result=BAD
gzip.GzipFile(file_path) # $ result=BAD
import pandas
pandas.read_csv(filepath_or_buffer=file_path)
pandas.read_csv(filepath_or_buffer=file_path) # $ result=BAD
pandas.read_table(file_path, compression='gzip')
pandas.read_xml(file_path, compression='gzip')
pandas.read_table(file_path, compression='gzip') # $ result=BAD
pandas.read_xml(file_path, compression='gzip') # $ result=BAD
pandas.read_csv(filepath_or_buffer=file_path, compression='gzip')
pandas.read_json(file_path, compression='gzip')
pandas.read_sas(file_path, compression='gzip')
pandas.read_stata(filepath_or_buffer=file_path, compression='gzip')
pandas.read_table(file_path, compression='gzip')
pandas.read_xml(path_or_buffer=file_path, compression='gzip')
pandas.read_csv(filepath_or_buffer=file_path,
compression='gzip') # $ result=BAD
pandas.read_json(file_path, compression='gzip') # $ result=BAD
pandas.read_sas(file_path, compression='gzip') # $ result=BAD
pandas.read_stata(filepath_or_buffer=file_path,
compression='gzip') # $ result=BAD
pandas.read_table(file_path, compression='gzip') # $ result=BAD
pandas.read_xml(path_or_buffer=file_path,
compression='gzip') # $ result=BAD
# no compression no DOS
pandas.read_table(file_path, compression='tar')
pandas.read_xml(file_path, compression='tar')
pandas.read_table(file_path, compression='tar') # $result=OK
pandas.read_xml(file_path, compression='tar') # $result=OK
pandas.read_csv(filepath_or_buffer=file_path, compression='tar')
pandas.read_json(file_path, compression='tar')
pandas.read_sas(file_path, compression='tar')
pandas.read_stata(filepath_or_buffer=file_path, compression='tar')
pandas.read_table(file_path, compression='tar')
pandas.read_xml(path_or_buffer=file_path, compression='tar')
pandas.read_csv(filepath_or_buffer=file_path,
compression='tar') # $result=OK
pandas.read_json(file_path, compression='tar') # $result=OK
pandas.read_sas(file_path, compression='tar') # $result=OK
pandas.read_stata(filepath_or_buffer=file_path,
compression='tar') # $result=OK
pandas.read_table(file_path, compression='tar') # $result=OK
pandas.read_xml(path_or_buffer=file_path, compression='tar') # $result=OK
return {"message": "bomb"}