def test_1(self):
logging.init_log(
StreamHandler(format=logging.FORMAT_SIMPLE),
FileHandler(format=logging.FORMAT_VERBOSE, level=logging.DEBUG))
logger = logging.getLogger(__name__)
print('123')
logger.debug('hello')
def __init__(self, file_path, flag_strict_mode=True):
self.file_path = file_path
self.current_path = os.path.dirname(file_path)
base_name = os.path.basename(file_path)
# 文件的扩展名
self.file_ext = os.path.splitext(base_name)[1]
# 文件类型
self.file_type = ''
self.output_path = os.path.join(self.current_path, 'output_%s' % base_name)
self.flag_strict_mode = flag_strict_mode
# 需要强调输出的结果内容
self.result_list = []
self.extract_file_md5_dict = {}
self.log_file_name = 'log.txt'
self.file_img_size = None
# 删除旧的数据
self.remove_dir(self.output_path)
# 创建输出路径
if not os.path.exists(self.output_path):
os.mkdir(self.output_path)
logging.init_log(ColorStreamHandler(logging.INFO, '%(message)s'),
FileHandler(filename=os.path.join(self.output_path, self.log_file_name),
format='%(message)s', level=logging.DEBUG))
def test_3(self):
logging.init_log(
StreamHandler(format=logging.FORMAT_SIMPLE),
TimedRotatingFileHandler(format=logging.FORMAT_VERBOSE,
level=logging.DEBUG))
logger = logging.getLogger(__name__)
logger.debug('hello')
"""
依赖 pngcheck、zsteg、stegdetect
"""
"""
自动检测文件可能的隐写，需要在Linux下使用 Python3 运行
一些依赖还需要手动安装
TODO 文件中可见字符的处理，对于 \00 这种分隔开的字符，需要能够分离
有些RAR等压缩包的文件头被修改，例如改成raR!，可以搜索strings里面的文本是否有符合的特征字符串，做个提醒
FFD9 后的文件内容显示出来
"""
logging.init_log(ColorStreamHandler(logging.INFO, '%(message)s'),
FileHandler(level=logging.INFO))
logger = logging.getLogger(__name__)
class WhatSteg(object):
def __init__(self, file_path, flag_strict_mode=True):
self.file_path = file_path
self.current_path = os.path.dirname(file_path)
base_name = os.path.basename(file_path)
# 文件的扩展名
self.file_ext = os.path.splitext(base_name)[1]
# 文件类型
self.file_type = ''
import tornado.options
import tornado.web
from mountains import logging
from mountains.logging import StreamHandler, RotatingFileHandler
import settings
from settings import get_path
from urls import url_handlers
if settings.DEBUG:
level = logging.DEBUG
else:
level = logging.INFO
logging.init_log(StreamHandler(level=level),
RotatingFileHandler(level=level, max_bytes=1024 * 1024 * 3, backup_count=1),
disable_existing_loggers=False)
logging.getLogger('tornado.curl_httpclient').disabled = True
logging.getLogger('tornado.general').disabled = True
logger = logging.getLogger(__name__)
logger.info('%s' % pycurl.version)
SQL_CREATE_TABLE_INTERCEPTOR = """
CREATE TABLE "interceptor" (
"uuid" TEXT NOT NULL,
"code" TEXT,
"create_date" TEXT,
"name" TEXT,
PRIMARY KEY ("uuid" ASC)
from optparse import OptionParser
from bs4 import BeautifulSoup
from future.moves.urllib.parse import urlunparse, urlparse, urljoin, quote, unquote
from mountains import logging
from mountains import text_type
from mountains.http import random_agent
from mountains.logging import ColorStreamHandler
from mountains.tornado import async_request
from tornado import gen, ioloop
from tornado.escape import native_str
from tornado.gen import Return
from tornado.httpclient import HTTPError
logging.init_log(
ColorStreamHandler(logging.INFO, '[%(asctime)s] %(message)s',
logging.DATE_FMT_SIMPLE))
logger = logging.getLogger(__name__)
parser = OptionParser()
parser.add_option("-u",
"--url",
dest="target_url",
type="string",
help="target url, e.g. http://127.0.0.1:8080/index.php")
parser.add_option("-w",
"--worker",
dest="worker_num",
type="int",
default=5,
import validators
parser = OptionParser()
parser.add_option('-l', '--host', help='local listen host', default='0.0.0.0')
parser.add_option('-p', '--port', help='local listen port', default=8083, type='int')
parser.add_option('-r', '--remote_url', help='remote url, e.g. http://192.168.85.131:8080', default=None)
parser.add_option('-d', '--display_url', help='display url', default=False,
action="store_true")
parser.add_option('-v', '--verbose', help='verbose', default=False, action="store_true")
log_format = '[%(asctime)s] %(levelname)s %(message)s'
(options, args) = parser.parse_args()
logging.init_log(StreamHandler(
logging.INFO, log_format, logging.DATE_FMT_SIMPLE),
disable_existing_loggers=not options.display_url)
logger = logging.getLogger(__name__)
ASYNC_HTTP_CONNECT_TIMEOUT = 60
ASYNC_HTTP_REQUEST_TIMEOUT = 60
ASYNC_HTTP_CLIENT_MAX_CLIENTS = 20
PARSED_REMOTE_URL = None
if platform.system().lower() != 'windows':
try:
# curl_httpclient is faster than simple_httpclient
AsyncHTTPClient.configure(
'tornado.curl_httpclient.CurlAsyncHTTPClient',
max_clients=ASYNC_HTTP_CLIENT_MAX_CLIENTS)
except ImportError:
from flask import Flask, request, jsonify, g
from future.moves.urllib.parse import urlparse
from mountains import logging
from mountains.logging import StreamHandler, RotatingFileHandler
from werkzeug.routing import BaseConverter
from config import *
from index import WizIndex
if not os.path.exists('logs'):
os.makedirs('logs')
logging.init_log(StreamHandler(level=logging.INFO,
format=logging.FORMAT_VERBOSE,
datefmt=logging.DATE_FMT_SIMPLE),
RotatingFileHandler(level=logging.INFO, filename='logs/app.txt'),
disable_existing_loggers=False)
class RegexConverter(BaseConverter):
def __init__(self, _map, *args):
super(RegexConverter, self).__init__(_map)
self.map = _map
self.regex = args[0]
parser = OptionParser()
parser.add_option("-p", "--port", dest="port", default=5000, type="int", help="port")
app = Flask(__name__)
# Created by restran on 2017/7/17
from __future__ import unicode_literals, absolute_import
import os
from flask import Flask, render_template
from mountains import logging
from mountains.logging import StreamHandler, RotatingFileHandler
import settings
from urls import init_url_rules
if settings.DEBUG:
level = logging.DEBUG
else:
level = logging.INFO
logging.init_log(StreamHandler(level=level),
RotatingFileHandler(level=level, max_bytes=1024 * 1024 * 3, backup_count=1),
disable_existing_loggers=True)
logger = logging.getLogger(__name__)
def get_path(target_path):
path = os.path.join(os.getcwd(), target_path) # development path
if not os.path.exists(path): # frozen executable path
path = os.path.join(os.path.dirname(os.path.abspath(__file__)), target_path)
return path
# 静态文件目录，要用这个格式配置，不然会找不到
app = Flask(__name__, static_folder=get_path('static'),
template_folder=get_path('templates'), )
app.config["SEND_FILE_MAX_AGE_DEFAULT"] = 1 # disable caching
