def dark_start(target):
from dark_core.database.mysqlManger import sqlMg
from hiddenDetect import hiddenlink_obj
from dark_core.output.console import consoleLog
from dark_core.output.textFile import fileLog
from dark_core.output.logging import logger
from dark_core.profile.profile import pf
from dark_core.settings.settings import settings
from dark_core.parser.urlParser import url_object
from datetime import datetime
# 设置日志模块
if pf.getLogType() == 'True':
file_path = settings.get('LOG_FILE_PATH')
datetimestrf = datetime.now().strftime('%Y-%m-%d')
domain = url_object(target).getRootDomain # 获取当前页面的根域名
file_name = file_path + domain + '_' + datetimestrf + '.log' # 检测文件名按域名_时间.log的形式加载
fileLog.set_file_name(file_name) # 设置日志文件名
logger.setOutputPlugin(fileLog)
else:
logger.setOutputPlugin(consoleLog)
# 执行检测
hidden = hiddenlink_obj(target)
hidden.init()
hidden.run()
hidden.finsh()
# 关闭相关数据库的连接和日志打印模块
sqlMg.dispose()
logger.endLogging()
def dark_start(target):
from dark_core.database.mysqlManger import sqlMg
from hiddenDetect import hiddenlink_obj
from dark_core.output.console import consoleLog
from dark_core.output.textFile import fileLog
from dark_core.output.logging import logger
from dark_core.profile.profile import pf
from dark_core.settings.settings import settings
from dark_core.parser.urlParser import url_object
from datetime import datetime
# 设置日志模块
if pf.getLogType() == "True":
file_path = settings.get("LOG_FILE_PATH")
datetimestrf = datetime.now().strftime("%Y-%m-%d")
domain = url_object(target).getRootDomain # 获取当前页面的根域名
file_name = file_path + domain + "_" + datetimestrf + ".log" # 检测文件名按域名_时间.log的形式加载
fileLog.set_file_name(file_name) # 设置日志文件名
logger.setOutputPlugin(fileLog)
else:
logger.setOutputPlugin(consoleLog)
# 执行检测
hidden = hiddenlink_obj(target)
hidden.init()
hidden.run()
hidden.finsh()
# 关闭相关数据库的连接和日志打印模块
sqlMg.dispose()
logger.endLogging()
'''
描述： 获取html中的class对应的style
'''
for script in self.styles:
try:
match = re.search(r'%s\x20[^\{]*\{([^\}]+)\}' % className, script, re.IGNORECASE | re.DOTALL)
if match is not None:
return match.group(1)
except Exception, msg:
logger.error(msg)
return None
if __name__ == '__main__':
from dark_core.output.console import consoleLog
logger.setOutputPlugin(consoleLog)
url = "http://www.dtzwdt.gov.cn/"
htObj = html_object(url)
print htObj.get_detect_html
# htObj.set_paresr_doc()
# aList = htObj.get_a_tag_link_list()
'''
styleList = htObj.get_all_style_content() # 获取所有的非style属性下的样式
for style in styleList:
print style
print '____________________________'
scriptList = htObj.get_all_script_content() # 获取当前页面中的所有javascript
except DarkException, msg:
logger.error(msg)
else:
try:
store_url_hidden_report_in_monitor_statistic(id, ref_id, threat_name, threat_level, threat_sum ,stat_time, report_part_path)
except DarkException, msg:
logger.error(msg)
def finsh(self):
logger.info('Detect modules finished, now will be quit...')
logger.info('Detect result: find %d url may have evil function!' % len(self.resultHiddenlink))
# 关闭日志模块
if __name__ == '__main__':
from dark_core.output.console import consoleLog
from dark_core.output.textFile import fileLog
# 设置日志模块
if pf.getLogType() == 'True':
logger.setOutputPlugin(fileLog)
else:
logger.setOutputPlugin(consoleLog)
url = 'http://www.kingboxs.com'
hidden = hiddenlink_obj(url)
hidden.init()
hidden.run()
hidden.finsh()
logger.endLogging()
