SecureUpdateDynamicHash/partialhash.py at main · bboymega/SecureUpdateDynamicHash · GitHub

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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import random
import xxhash
import cityhash
from Crypto.Hash import SHA256
import spookyhash
import pickle
class partialhash:
    def getpartial(data, initial, terminal):
        return data[initial:terminal]

    def generatepartiallabel(n, max_partial_size, data_size):
        label = []
        label.append(random.randint(1, max_partial_size))
        for i in range(1,n-1):
            label.append(random.randint(1,max_partial_size)+label[i-1])
        label.append(data_size)
        return label

    def generatepartialdata(data, partial_label):
        label_size = len(partial_label)
        partial_data = []
        partial_data.append(partialhash.getpartial(data,0,partial_label[0]))
        for i in range(1,label_size):
            partial_data.append(partialhash.getpartial(data,partial_label[i-1],partial_label[i]))
        return partial_data

    def generateinstructiontag(n):
        instruction_tag = []
        for i in range(0,n):
            instruction_tag.append(random.randint(0,2))
        return instruction_tag

    def generatepartialhash(instruction_tag, partial_data):
        partial_hash = []
        partial_array_size = len(instruction_tag)
        for i in range (0,partial_array_size):
            if instruction_tag[i] == 0:
                city64_hash = cityhash.CityHash64(partial_data[i])
                partial_hash.append(str(city64_hash))
            else:
                if instruction_tag[i] == 1:
                    xxh64_hash = xxhash.xxh64(partial_data[i])
                    partial_hash.append(xxh64_hash.hexdigest())
                else:
                    if instruction_tag[i] == 2:
                        spooky64_hash = spookyhash.hash64(partial_data[i])
                        partial_hash.append(str(spooky64_hash))
        return partial_hash

    def generatefinalhashquick(n, data, max_partial_size, data_size):
        param = []
        label = []
        instruction_tag = []
        partial_hash = []
        finalhashstr = ''

        for i in range(0, n):
            if i == 0:
                label.append(random.randint(1, max_partial_size))
                partial_data = bytearray(partialhash.getpartial(data, 0, label[0]))
            else:
                label.append(random.randint(1, max_partial_size) + label[i - 1])
                partial_data = bytearray(partialhash.getpartial(data, label[i - 1], label[i]))
            if i == n-1:
                label.append(data_size)
            instruction_tag.append(random.randint(0, 2))
            if instruction_tag[i] == 0:
                city64_hash = cityhash.CityHash64(partial_data)
                partial_hash.append(str(city64_hash))
            else:
                if instruction_tag[i] == 1:
                    xxh64_hash = xxhash.xxh64(partial_data)
                    partial_hash.append(xxh64_hash.hexdigest())
                else:
                    if instruction_tag[i] == 2:
                        spooky64_hash = spookyhash.hash64(partial_data)
                        partial_hash.append(str(spooky64_hash))
            finalhashstr = finalhashstr + partial_hash[i]

        sha256_hash = SHA256.new(str.encode(finalhashstr))
        finalhash = sha256_hash.hexdigest()
        param.append(pickle.dumps(label))
        param.append(pickle.dumps(instruction_tag))
        param.append(finalhash.encode("utf-8"))
        return param

    def generatefinalhashquickserver(data, partial_label, instruction_tag):
        n = len(partial_label)
        partial_data = []
        partial_hash = []
        finalhashstr = ''
        for i in range(0, n):
            if i == 0:
                partial_data = bytearray(partialhash.getpartial(data, 0, partial_label[0]))
            else:
                partial_data = bytearray((partialhash.getpartial(data, partial_label[i - 1], partial_label[i])))
            if i < n-1:
                if instruction_tag[i] == 0:
                    city64_hash = cityhash.CityHash64(partial_data)
                    partial_hash.append(str(city64_hash))
                else:
                    if instruction_tag[i] == 1:
                        xxh64_hash = xxhash.xxh64(partial_data)
                        partial_hash.append(xxh64_hash.hexdigest())
                    else:
                        if instruction_tag[i] == 2:
                            spooky64_hash = spookyhash.hash64(partial_data)
                            partial_hash.append(str(spooky64_hash))
                finalhashstr = finalhashstr + partial_hash[i]

        sha256_hash = SHA256.new(str.encode(finalhashstr))
        finalhash = sha256_hash.hexdigest()
        return finalhash

    def generatefinalhash(partial_hash):
        finalhashstr = ''
        for i in partial_hash:
            finalhashstr = finalhashstr + i
        sha256_hash = SHA256.new(str.encode(finalhashstr))
        finalhash = sha256_hash.hexdigest()
        return finalhash