summaryrefslogtreecommitdiff
path: root/src/plugins/opensubtitles/hash.py
blob: f28341658d4847dc006c28d957c6ec7b55273c0f (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
import struct
import os
from gi.repository import Gio

SIZE_ERROR = -1
SEEK_ERROR = -2

def hash_file (name):
    """ FIXME Need to handle exceptions !! """

    longlongformat = 'q' # long long
    bytesize = struct.calcsize (longlongformat)

    file_to_hash = Gio.File.new_for_uri (name)

    file_info = file_to_hash.query_info ('standard::size', 0, None)
    filesize = file_info.get_attribute_uint64 ('standard::size')

    file_hash = filesize

    if filesize < 65536 * 2:
        return SIZE_ERROR, 0

    file_handle = open (file_to_hash.get_path (), "rb")

    for _ in range(int(65536 / bytesize)):
        buf = file_handle.read (bytesize)
        (l_value,) = struct.unpack (longlongformat, buf)
        file_hash += l_value
        file_hash = file_hash & 0xFFFFFFFFFFFFFFFF #to remain as 64bit number

    file_handle.seek (max (0, filesize - 65536), os.SEEK_SET)

    if file_handle.tell() != max (0, filesize - 65536):
        return SEEK_ERROR, 0

    for _ in range(int(65536/bytesize)):
        buf = file_handle.read (bytesize)
        (l_value,) = struct.unpack (longlongformat, buf)
        file_hash += l_value
        file_hash = file_hash & 0xFFFFFFFFFFFFFFFF

    file_handle.close ()
    returnedhash = "%016x" % file_hash

    return returnedhash, filesize