检查文件头、签名和类型

Question

我正在制作一个文件类型检查程序（带有文件签名检查），但是文件签名太多，所以我无法将目标文件与文件签名进行比较。

是否有任何库检查文件签名，或者我怎样才能轻松地制作这个脚本？如何制作一个功能检查文件？

• http://forensic-proof.com/archives/300
• Python: Check if uploaded file is jpg

代码：

__author__ = 'jun'

# -*- coding: utf-8 -*-


import sys

if len(sys.argv) is 1:
    print(sys.stderr, '읽을 파일명을 입력해주세요')
    exit(1)

try:
    IN = open(sys.argv[1], 'rb')
except IOError:
    print(sys.stderr, '그런 파일이 없거나, 열기에러입니다.')
    exit(1)

offset = 0

def is_jpg(filename):
    data = IN.read(11)
    if data[:4] != '\xff\xd8\xff\xe0': return False
    if data[6:] != 'JFIF\0': return False
    return True


if is_jpg(IN) is True:
    print ('jpg file')

Answer 1

您要做的就是检测 mime 类型。

This library 可以为所欲为。