mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-23 01:56:36 +03:00
30 lines
884 B
Python
30 lines
884 B
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2021 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'LICENSE' for copying permission
|
|
"""
|
|
|
|
import cProfile
|
|
import os
|
|
|
|
from lib.core.data import logger
|
|
from lib.core.data import paths
|
|
|
|
def profile(profileOutputFile=None):
|
|
"""
|
|
This will run the program and present profiling data in a nice looking graph
|
|
"""
|
|
|
|
if profileOutputFile is None:
|
|
profileOutputFile = os.path.join(paths.SQLMAP_OUTPUT_PATH, "sqlmap_profile.raw")
|
|
|
|
if os.path.exists(profileOutputFile):
|
|
os.remove(profileOutputFile)
|
|
|
|
# Start sqlmap main function and generate a raw profile file
|
|
cProfile.run("start()", profileOutputFile)
|
|
|
|
infoMsg = "execution profiled and stored into file '%s' (e.g. 'gprof2dot -f pstats %s | dot -Tpng -o /tmp/sqlmap_profile.png')" % (profileOutputFile, profileOutputFile)
|
|
logger.info(infoMsg)
|