88436e85ef46a548f161dfa7dfd65a8e85878137
[stack/code/idfimage.git] / idfimage.py
1 #!/usr/bin/env python
2
3 #Identify false image files, for example a file that has a gif header and a php body
4 #author: stack@inventati.org
5 #copyleft: 2013-01-29
6
7 import os
8 import sys
9 from PIL import Image
10 import imghdr
11
12 #Log wrong image files found
13 def log(path, fd):
14     fd.write(path+"\n") 
15
16 #Echo a message if verbose is enabled
17 def verbose(message):
18     if debug is not None:
19         print(message)
20
21 #Check if a given file is an image, trying not to be fooled
22 def checkfimage(fpath):
23     #imghdr is fooled with false images, so use it to check if file would be a valid image file
24     if imghdr.what(fpath):        
25         try: 
26             #PIL Image is not fooled
27             Image.open(fpath)
28         except:
29             return False
30     return True
31
32 #Generate the file paths to traverse, or a single path if a file name was given
33 def getfiles(path):
34     if os.path.isdir(sys.argv[1]):
35         for root, dirs, files in os.walk(sys.argv[1]):
36             for name in files:
37                 yield os.path.join(root, name)
38     else:
39         yield path
40
41 if __name__ == "__main__":
42     if len(sys.argv) < 3:
43         sys.exit('Usage: %s path logfile [-v]' % sys.argv[0])
44
45     if not os.path.exists(sys.argv[1]):
46         sys.exit('ERROR: path %s was not found!' % sys.argv[1])
47
48     try:
49         fdlog = open(sys.argv[2],"w")
50     except:
51         sys.exit('ERROR: unable to open logfile' % sys.argv[2])
52
53     debug = 1 if len(sys.argv) == 4 and sys.argv[3] == "-v" else None
54
55     for fpath in getfiles(sys.argv[1]):
56         verbose("Checking: " + fpath)
57         if not checkfimage(fpath):
58             log(fpath, fdlog)
59             verbose("ERR not an image: " + fpath)
60
61     fdlog.close()