greydgl--pentestgpt
Gelei Deng
29 行
692 B
Python
29 行
692 B
Python
# main.py
|
|
|
|
from pentestgpt.modules.commands import CommandParser
|
|
from pentestgpt.modules.core import CoreEngine
|
|
from pentestgpt.modules.state import State
|
|
|
|
|
|
|
|
def main():
|
|
state = State()
|
|
core_engine = CoreEngine(state)
|
|
command_parser = CommandParser(state, core_engine)
|
|
|
|
print("Welcome to PentestGPT")
|
|
print("Type 'help' to see available commands.")
|
|
|
|
while True:
|
|
try:
|
|
input_line = input("pentestgpt> ")
|
|
command_parser.parse(input_line)
|
|
except KeyboardInterrupt:
|
|
print("\nUse 'exit' command to quit.")
|
|
except EOFError:
|
|
print("\nUse 'exit' command to quit.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|