summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/subresource-integrity/tools/list_hashes.py
blob: 5e3830ab61d38bb8f28837f38cc992de02fb197b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
from os import path, listdir
from hashlib import sha512, sha384, sha256, md5
from base64 import b64encode
import re

DIR = path.normpath(path.join(__file__, "..", ".."))

'''
Yield each javascript and css file in the directory
'''
def js_and_css_files():
  for f in listdir(DIR):
    if path.isfile(f) and (f.endswith(".js") or f.endswith(".css")):
      yield f

'''
URL-safe base64 encode a binary digest and strip any padding.
'''
def format_digest(digest):
  return b64encode(digest)

'''
Generate an encoded sha512 URI.
'''
def sha512_uri(content):
  return "sha512-%s" % format_digest(sha512(content).digest())

'''
Generate an encoded sha384 URI.
'''
def sha384_uri(content):
  return "sha384-%s" % format_digest(sha384(content).digest())

'''
Generate an encoded sha256 URI.
'''
def sha256_uri(content):
  return "sha256-%s" % format_digest(sha256(content).digest())

'''
Generate an encoded md5 digest URI.
'''
def md5_uri(content):
  return "md5-%s" % format_digest(md5(content).digest())

def main():
  for file in js_and_css_files():
    print "Listing hash values for %s" % file
    with open(file, "r") as content_file:
      content = content_file.read()
      print "\tSHA512 integrity: %s" % sha512_uri(content)
      print "\tSHA384 integrity: %s" % sha384_uri(content)
      print "\tSHA256 integrity: %s" % sha256_uri(content)
      print "\tMD5 integrity:    %s" % md5_uri(content)

if __name__ == "__main__":
  main()