mirror of
https://github.com/b3nguang/Ollama-Scan.git
synced 2025-06-21 18:30:19 +00:00
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
@ Author: b3nguang
|
|||
|
|
@ Date: 2025-02-12 01:30:26
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
@ Author: b3nguang
|
|||
|
|
@ Date: 2025-02-12 01:30:26
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
|
|||
|
|
from inc import output, proxycheck
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_parser():
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
usage="python3 main.py",
|
|||
|
|
description="Ollama-Scan: 针对Ollama的TCP Socket开源渗透框架",
|
|||
|
|
)
|
|||
|
|
p = parser.add_argument_group("Ollama-Scan 的参数")
|
|||
|
|
p.add_argument("-u", "--url", type=str, help="对单一URL进行Ollama端点探测")
|
|||
|
|
p.add_argument("-uf", "--urlfile", type=str, help="读取目标TXT进行Ollama端点探测")
|
|||
|
|
p.add_argument("-d", "--dump", type=str, help="下载指定Ollama端点的容器日志（可提取敏感信息）")
|
|||
|
|
p.add_argument("-p", "--proxy", type=str, default="", help="使用HTTP代理")
|
|||
|
|
p.add_argument("-z", "--zoomeye", type=str, default="", help="使用ZoomEye导出Ollama框架资产")
|
|||
|
|
p.add_argument("-f", "--fofa", type=str, default="", help="使用Fofa导出Ollama框架资产")
|
|||
|
|
p.add_argument("-y", "--hunter", type=str, default="", help="使用Hunter导出Ollama框架资产")
|
|||
|
|
p.add_argument("-t", "--newheader", type=str, help="从TXT文件中导入自定义HTTP头部")
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
return args
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
output.logo()
|
|||
|
|
args = get_parser()
|
|||
|
|
proxycheck.Ollama_Scan_Proxy(args)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|