# !/usr/bin/env python # -*- coding: utf-8 -*- import os import re Path="/Users/ascotbe/code/Kernelhub" window_success_list=[] window_failure_list=[] def WindowsConfig(): #中文配置的开头 chinese_begin="""let config = {\n title: 'Windows',\n home: 'Info.md',\n repo: 'Ascotbe/Kernelhub/Windows',\n nav: [\n {\n title: '简介', path: '/'\n },\n {\n title: '目录导航',type: 'dropdown', items: [\n {\n title: "有利用脚本(测试成功)",path: '/TestSuccess'\n },""" #所有配置的结尾 config_end="""\n ]\n },\n ],\n tocVisibleDepth: 10,\n plugins: []\n};""" #下表是黑名单，垃圾文件 reject_list=['.DS_Store', '.git','.idea', '.gitignore','LICENSE', 'Patch', 'README.CN.md', 'README.md', 'TestFailure', 'docs', 'DocumentGeneration.py','StencilPlate'] #英文配置的开头 english_begin="""let config = {\n title: 'Windows',\n home: 'Info.md',\n repo: 'Ascotbe/Kernelhub/Windows',\n nav: [\n {\n title: 'Introduction', path: '/'\n },\n {\n title: 'DirectoryNavigation',type: 'dropdown', items: [\n {\n title: "exploit script (successful test)",path: '/TestSuccess'\n },""" #处理成功的编号 for i in os.listdir(Path+"/Windows"): if i not in reject_list: window_success_list.append(i) #循环成功的编号和开头配置进行拼接 for x in sorted(window_success_list): chinese_begin+="""\n {\n path: '/CN/"""+x+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/"""+x+"""/README.md'\n },""" english_begin+="""\n {\n path: '/EN/"""+x+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/"""+x+"""/README_EN.md'\n },""" #插入中间内容，未测试成功内容 chinese_begin+="""\n {\n title: '有利用脚本(测试未成功)', path: '/TestFailure'\n },""" english_begin+="""\n {\n title: 'exploit script (test failure)', path: '/TestFailure'\n },""" for i in os.listdir(Path+"/Windows/TestFailure"): if i not in reject_list: window_failure_list.append(i) #写入windows测试失败漏洞 for xx in sorted(window_failure_list): chinese_begin+="""\n {\n path: '/CN/"""+xx+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/TestFailure/"""+xx+"""/README.md'\n },""" english_begin+="""\n {\n path: '/EN/"""+xx+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/TestFailure/"""+xx+"""/README_EN.md'\n },""" chinese_config_file=open(Path+"/docs/Windows/Docs/config.js","w+") chinese_config_file.write(chinese_begin+config_end) chinese_config_file.close() english_config_file=open(Path+"/docs/Windows/EnglishDocs/config.js","w+") english_config_file.write(english_begin+config_end) english_config_file.close() def WindowsMarkdown(): """ 从Windows目录中提取README内容中的《编号列表》内容进行拆分，判断那些是测试过的那些是没测试过的 把他们分成两个文件存放，然后再区分下语言 """ common_vulnerabilities_and_exposures_list=[] chinese_success_list=[] chinese_failure_list=[] english_success_list=[] english_failure_list=[] #打开Windows目录下面的文件读取内容 file=open(Path+"/Windows/README.md","r").read() #提取《编号列表》的内容 date = re.findall(r'(?<=> Numbered list)([\w\W]+)(?=### Required environment)', file)[0].split("\n") #进行处理拼接 for i in date: if i != "": common_vulnerabilities_and_exposures_list.append(i) for x in common_vulnerabilities_and_exposures_list[2:]: date = re.findall(r'(?<=\[)([\w\W]+)(?=\])', x)[0] if date in window_success_list: chinese_success_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/Docs/#/CN/"+date, x)) english_success_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/EnglishDocs/#/EN/"+date, x)) if date in window_failure_list: chinese_failure_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/Docs/#/CN/"+date, x)) english_failure_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/EnglishDocs/#/EN/"+date, x)) chinese_success_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n" chinese_failure_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n" english_success_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n" english_failure_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n" for x1 in chinese_success_list: chinese_success_date+=x1+"\n" for x2 in chinese_failure_list: chinese_failure_date+=x2+"\n" for x3 in english_success_list: english_success_date+=x3+"\n" for x4 in english_failure_list: english_failure_date+=x4+"\n" #写入文件中 chinese_success_file=open(Path+"/docs/Windows/Docs/TestSuccess.md","w+") chinese_success_file.write(chinese_success_date) chinese_success_file.close() chinese_failure_file=open(Path+"/docs/Windows/Docs/TestFailure.md","w+") chinese_failure_file.write(chinese_failure_date) chinese_failure_file.close() english_success_file=open(Path+"/docs/Windows/EnglishDocs/TestSuccess.md","w+") english_success_file.write(english_success_date) english_success_file.close() english_failure_file=open(Path+"/docs/Windows/EnglishDocs/TestFailure.md","w+") english_failure_file.write(english_failure_date) english_failure_file.close() """ 处理Windows目录下面readme文件，没有测试成功的编号进行更新 """ read_readme_english_file_data = open(Path + "/Windows/README.md", "r").read() read_readme_chinese_file_data = open(Path + "/Windows/README.CN.md", "r").read() header="\n\n| SecurityBulletin | | | |\n| ---------------- |---------------- | ---------------- | ---------------- |\n" common_vulnerabilities_and_exposures="" count=0 for i in sorted(window_failure_list): count+=1 common_vulnerabilities_and_exposures += "| " + str(i) if count>4: header+=common_vulnerabilities_and_exposures+"|\n" common_vulnerabilities_and_exposures="" count=0 if count!=0: header+=str(common_vulnerabilities_and_exposures)+" |\n" #对内容进行替换 tmp_english = re.sub(r'(?<=and welcome to submit PR)([\w\W]+)(?=### Disclaimer)', header+"\n\n",read_readme_english_file_data) tmp_chinese = re.sub(r'(?<=欢迎提交PR)([\w\W]+)(?=### 免责声明)', header+"\n\n",read_readme_chinese_file_data) open(Path + "/Windows/README.md", "w+").write(tmp_english) open(Path + "/Windows/README.CN.md", "w+").write(tmp_chinese) if __name__ == '__main__': WindowsConfig() WindowsMarkdown()