基于python3.8.0和windows11
其他版本3.7.0我在本机测试之后也能使用
数据库：mysql 8.0.28
连接方式是pymysql 不是QtSql.QSqlDatabase
安装依赖 pip install -r requirements.txt
所使用的样式需要下载
pip install qt_material样式表可以自定义和修改，在安装包的位置找到：
????\????\???\Lib\site-packages\qt_material里的material.css.template可以直接编辑所有样式，为了把字体调大，搜索font-size，修改数字为16：
{% if font_size %}
font-size: {{font_size|density(density_scale, density_interval=1)}}px;
{% else %}
font-size: {{14|density(density_scale, density_interval=1)}}px;
{% endif %}把标签页名字改小写，也就不应用大写
搜QtabBar，修改成为如下，就是把text-transform的uppercase修改为none
QTabBar{
text-transform: none;
font-weight: bold;
}移植项目需要先修改config.py里config.ini的绝对路径
self.path = r'???\???\CodeReview\src\config\config.ini'再到config.ini里修改如下的值，注意要在project_name这里加上\
[main_project]
project_name = 你项目文件夹所在位置\
[mysql]
db_password = 你的密码
...后期给加密了，所以无法修改...，破解版在test分支里
如果你想要用QtSql，那问题会像蚂蚁一样多！
如果你发现连接数据库连不上，是因为你没有mysql的dll文件
我使用的pyQt5.15.9，根据自己所配置的pyqt5版本下载驱动
所以我使用pymysql专门对接mysql和python
不要用它原生的，不然你会掉入问题黑洞......
codereview
user{
id: VARCHAR(50),
username: VARCHAR(100),
password: VARCHAR(200),
salt: BINARY(64),
aes_key: BINARY(32),
}含自增id
danger_func{
user_id: VARCHAR(50),
func_name: VARCHAR(100),
level: VARCHAR(50),
solution: VARCHAR(300),
id: auto_increment
}含自增id
公钥pubilc_key用于加密,私钥private_key存于user表用于解密
report有三种形式 : doc/docx pdf md
files{
user_id: VARCHAR(50),
report_type: VARCHAR(10),
report_path: VARCHAR(200),
log_path: VARCHAR(200),
id: auto_increment
}使用QsciScintilla
自定义文本编辑器动作和配置，只写了cpp和c的基本筛选器
python的基础api官方文档在这里,但是只是初级小白版
后续的一系列骚操作请去正规的Scintilla官网上看接口是怎么写的
光标跳转是我一步步实验出来硬写的，结合大爹chatgpt的分析~~(它的唯一作用就是帮我分析了参数含义)~~,再修改的.
硬算出范围，然后暴力，因为api比较底层
这里直接让后台开一个子进程，用线程加速，对于特殊操作做区分，重定到print操作里输出终端内容
其实子进程通过命令行是无法直接切换的，它至始至终都只能在你调用它的地方。
但是可以通过这个os.chdir(directory)指定当前工作目录为directory，这样就完美解决！
cls之后会让输出为空，但因为是模拟，所以直接强制把输出框内容清除就能达到效果
使用的chatgpt3.5主页的颜色作为终端配色
背景:#143113,字体:#FF8BFF && Consolas
通过语法树节点，得到函数声明、函数定义、函数调用、变量声明、变量赋值、运算符，返回....
需要安装windows64.exe的LLVM，并把???/???/bin/libclang.dll加入环境变量里。
之后即将封神！看看python库有多离谱，安装pip install clang安装的是clang调用的接口api
调用clang.index模块里的多个类:
from clang.cindex import Index, Config
# 真好啊 API!
# 配个环境
libclangPath = r"???\????\bin\libclang.dll"
Config.set_library_file(libclangPath)
# 写你想要分析的路径 c/cpp/hh/cc/h文件
file_path = r"test_no_headers.c"
index = Index.create()
your_father = index.parse(file_path)
AST_Root = your_father.cursor之前分词还看了半天的flex, 呵呵T^T, 这里直接?句话秒了
cursor_content = ""
for token in AST_Root.get_tokens():
cursor_content += token.spelling + '\n'
print(cursor_content)传入语法树的根，查看api接口提炼关键字和内容
api: ???/???/???/?????/Lib/site-packages/clang/cindex.py
def iterAST(cursor):
from clang.cindex import CursorKind
for cur in cursor.get_children():
if cur.kind == CursorKind.FUNCTION_DECL:
if cur:
# 函数声明
print("FUNCTION_DECL: {}".format(cur.spelling))
for cur_item in cur.get_children():
if cur_item.kind == CursorKind.CALL_EXPR:
# 函数调用
print("CALL_EXPR: {}".formate(cur_item.spelling))
elif cur.kind == CursorKind.VAR_DECL:
if cur:
# 变量声明
print("VAR_DECL: {}".format(cur.spelling))
elif cur.kind == CursorKind.FIELD_DECL:
if cur:
# 字段声明
print("FIELD_DECL: {}".format(cur.spelling))
elif cur.kind == CursorKind.TYPEDEF_DECL:
if cur:
# 类型定义
print("TYPEDEF_DECL: {}".format(cur.spelling))
else:
pass
# done??
# 递归调用
iterAST(cur)
# 入口
iterAST(AST_Root)对，没错，依旧是通过绝对位置硬算坐标TVT~~~
但是已经算过很多次了，所以问题不大
计算方法大致就是 : 从上面的语法树里获取到关键位置*position
position = (start_line, start_index, end_line, end_index)
positions_func = [position_0,position_1,position_2, ...,position_n]
传入高亮函数highlight_all_text(positions)里进行高亮
为什么要分初级还是中级?
初级就是只能在同一个文件里跳转声明\定义\调用
中级就是在不同文件里跳转声明\定义\调用,这里只限制头文件\源文件
高级就是在所有项目文件里面跳转,我暂时还不会...
因为如果你把函数的声明写到头文件里，再引用头文件，那么你需要跳转这个函数的时候就会在原来的位置找不到声明...
如果在文件里存在标准头文件，就会把里面的函数一起分析了，这就会分析上千行......
在文本编辑器里面显示源文件带头文件的内容，但是我实际上分析的时候，传入的内容是含注释掉头文件的源文件内容.
这样之后文本编辑器上的文本位置不会乱，而分析器也不会分析头文件，但是这样会生成临时文件，所以在分析结束之后把它删掉，看起来就是分析了源文件的内容.(๑•̀ㅂ•́)و✧)
对于分析的时候先查一轮源文件里自定义的头文件集合，再对头文件筛选一遍函数声明\定义.
接着对临时文件内容筛选一遍函数声明\定义\调用,在函数声明里查没有在源文件列表里出现的目标
再到头文件函数声明列表里查一遍，查到就输出其信息.
filename = r'???/????/???/test.cpp'
def get_headers(filename):
import re
with open(filename, 'r', encoding='utf-8') as file:
contents = file.readlines()
headers = []
# 正则表达式筛选所有 #include"???.h" / #include "???.h" 自定义头文件
pattern = r'#include\s*["]\s*(\w+\.h)\s*["]'
for item in contents:
match = re.search(pattern, item)
if match:
dependency = match.group(1)
headers.append(dependency)
return headers
# 对所有自定义引用的头文件分析
def headerAnalyzer(filename):
# 伪代码 FunctionDump是存储过滤结果的类
headers_path = get_headers(filename)
headers_analyzers_list = []
for item in headers_path:
analyzer_obj = FunctionDump(filename)
analyzer_obj.Launcher()
analyzer_obj.show_details()
headers_analyzers_list.append((item, analyzer_obj))
return headers_analyzers_list
# 除去头文件的源代码分析
def excludeHeaderAnalyzer(filename):
# 伪代码 FunctionDump是存储过滤结果的类
analyzer_obj = FunctionDump(filename)
analyzer_obj.Launcher()
analyzer_obj.show_details()
return analyzer_obj
def preProcessTools(filename, keyword):
# 伪代码 举声明函数查找例子
header_objs = headerAnalyzer(filename)
exclude_header_objs = excludeHeaderAnalyzer(filename)
# 先把所有名字存起来 之后筛选
exclude_declaration_name_list = [item.function_name for item in exclude_header_objs.function_declaration_list]
# 设置声明标志位
declared_flag = True
target_function_name = keyword
if target_function_name not in exclude_declaration_name_list:
declared_flag = False
# 根据标志位开始判断是哪个集合里的函数声明 是头文件里的就查头文件 否则查源文件
if not declared_flag:
for item in header_objs:
path, header_obj = item
for function_obj in header_obj.function_declaration_list:
if target_function_name == function_obj.function_name and function_obj.declared_contents is not None:
print(function_obj.declared_contents)
print(function_obj.declared_location)
elif declared_flag:
for item in exclude_header_objs.function_declaration_list:
if target_function_name == item.function_name and item.declared_contents is not None:
print(item.declared_contents)
print(item.declared_location)
# ... 基本照猫画虎 大致思路如上 看具体代码的时候能更清楚地明白我在说什么??0-<我自己写的时候都蒙圈好几次优化函数调用，加上三种类型判断
-
只有一个源文件
>> ???.c/???.cpp函数声明、定义、调用都在同一个源文件里
-
只有一个源文件和多个头文件
>> ???.c/???.cpp + (??.h,???.h,????.h)函数声明/定义在头文件,源文件可能含声明、定义、调用
-
多个源文件和多个头文件，但只有一个源文件含
main函数>> ???.cpp/????.cpp/main.cpp + (??.h,???.h,????.h)函数声明/定义在头文件,除
main函数以外的源文件可能包含声明、定义,含main的源文件可能包含声明、定义、调用
把所有源文件的内容拼接在一起，最后拼接main函数的内容...组合器问世！
# 组合器
class DefinitionCallExpressCombiner:
def __init__(self, file_path):
self.file_path = file_path
self.headers = []
self.main_sign = None
self.definition_contents = []
self.mix_contents = []
self.main_length = 0
self.offset_length = 0
def find_all_files(self, filepath):
directory, _ = os.path.split(filepath)
file_list = []
for root, _, files in os.walk(directory):
for file in files:
if file.endswith('.c') or file.endswith('.cpp'):
file_list.append(os.path.abspath(os.path.join(root, file)))
return file_list
def has_main_function(self, file_path):
with open(file_path, "r") as file:
content = file.read()
return "int main(" in content
def getDefinitionCodes(self):
source_files = self.find_all_files(self.file_path)
for file_path in source_files:
with open(file_path, "r") as file:
content = file.readlines()
if self.has_main_function(file_path):
if self.main_sign is None:
self.main_sign = file_path
else:
# print('main function is None.')
pass
else:
self.definition_contents += content
def Combiner(self):
self.getDefinitionCodes()
path, name = split(self.main_sign)
name = '.' + name
temp_path = os.path.join(path, name)
with open(self.main_sign, "r", encoding='utf-8') as main_file:
main_file_content = main_file.readlines()
self.main_length = len(main_file_content)
last_line = self.definition_contents[-1]
if last_line == '}\n':
pass
elif last_line == '}':
self.definition_contents[-1] = '}\n'
if main_file_content:
self.mix_contents = self.definition_contents + main_file_content
new_data = ["//" + line if line.startswith("#include") else line for line in self.mix_contents]
with open(temp_path, 'w', encoding='utf-8') as temp_obj:
temp_obj.writelines(new_data)
self.offset_length = len(new_data) - self.main_length
return temp_path调用这个函数计算真正的main函数声明、定义、调用函数的位置坐标，覆盖临时文件的数据，返回数据
def multiCallExpressCombiner(self, filepath):
combiner = DefinitionCallExpressCombiner(filepath)
temp_filepath = combiner.Combiner()
call_analyzer = FunctionDump(temp_filepath)
call_analyzer.analyseLauncher()
os.remove(temp_filepath)
offset = combiner.offset_length
function_declaration_list = []
function_definition_list = []
function_call_express_list = []
for item in call_analyzer.function_declaration_list:
if item.declared_location[0] > offset:
start_line, start_index, end_line, end_index = item.declared_location
item.declared_location = (start_line - offset, start_index, end_line - offset, end_index)
function_declaration_list.append(item)
else:
continue
for item in call_analyzer.function_definition_list:
if item.definition_location[0] > offset:
start_line, start_index, end_line, end_index = item.definition_location
item.definition_location = (start_line - offset, start_index, end_line - offset, end_index)
function_definition_list.append(item)
else:
continue
for item in call_analyzer.function_callexpress_list:
if item.call_express_location[0] > offset:
start_line, start_index, end_line, end_index = item.call_express_location
item.call_express_location = (start_line - offset, start_index, end_line - offset, end_index)
function_call_express_list.append(item)
else:
continue
# 覆盖原文
call_analyzer.function_declaration_list = function_declaration_list
call_analyzer.function_definition_list = function_definition_list
call_analyzer.function_callexpress_list = function_call_express_list
return call_analyzer登录之后的用户进主页面 进行代码审计之后生成的报告文件
生成的报告是即时审计结果 数据库里存储的是用户自己添加的危险函数数据
饼子图也是即时审计结果
请按照格式添加窗口, 否则全是bug
只会导出项目所使用的依赖 而不是所有依赖
pip install pipreq
关闭代理之后再在当前项目文件夹操作, 不然爆代理错误.
要用utf-8否则爆gbk错误,因为我的主机是win11.
pipreqs ./ --encoding=utf8
clang.exe 请自己去分支test里下载clang.7z.001并解压,然后放在compile目录下
如果遇到compile出现问题，因为不是我负责的部分
我就直说了:
它的最大缺陷就是只能处理有依赖的.c/cpp,.h文件编译问题
它只能把有依赖关系的文件放在test\init_data\test_data\test_c里，因为它强制查含.c的文件，没有更新的分析，这里后续是可以修改的
test.h里有函数int add(int a,int b)声明
#include<stdio.h>
#include<stdlib.h>
int add(int a, int b);test.cpp里对其进行引用#include "test.h",并调用add(12,15)
#include "test.h"
int sum;
int main()
{
sum = add(12, 15);
printf("sum = %d",sum);
return 0;
}test2.cpp里实现了add方法
#include "test.h"
int add(int a,int b)
{
return a+b;
}放在一个文件夹下用clang.exe编译，会自动查找test.h只要在同一文件夹下就可以
clang.exe -o test.exe test.cpp test2.cpp 当你不想在同一文件夹下生成exe文件，请指定固定的绝对路径
在另一分支test里,暂不描述内容