summaryrefslogtreecommitdiff
path: root/src/plugins/opensubtitles/hash.py
blob: f780519c59701bffaa5fb6d12f4a30fb32ad09a0 (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
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

    with open (file_to_hash.get_path (), "rb") as file_handle:
        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

    returnedhash = f"{file_hash:%016x}"

    return returnedhash, filesize