Crypto/ideaalg.py at master · AleKit/Crypto · 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
#International Data Encryption Algorithm
#necesita operaciones.py

#funcion para generar las subclaves a partir de la clave original
def generadorclave(clave):
    z = []
    while len(z) < 52:
        z.append(clave[:16])
        z.append(clave[16:32])
        z.append(clave[32:48])
        z.append(clave[48:64])
        z.append(clave[64:80])
        z.append(clave[80:96])
        z.append(clave[96:112])
        z.append(clave[112:128])

        clave = clave[25:] + clave[:25]

    while len(z) > 52:
        z.pop()
    return z

#funcion para generar las subclaves para descifrar
def generadordecr(z):
    w = []
    j = 48
    w.append(inversomodulo(z[j]))
    w.append(opuestomodulo(z[j+1]))
    w.append(opuestomodulo(z[j+2]))
    w.append(inversomodulo(z[j+3]))
    while len(w) < 52:
        w.append(z[j-2])
        w.append(z[j-1])
        #print(w)
        j -= 6
        w.append(inversomodulo(z[j]))
        #print j
        w.append(opuestomodulo(z[j+2]))
        w.append(opuestomodulo(z[j+1]))
        w.append(inversomodulo(z[j+3]))
    aux1 = w[-2]
    aux2 = w[-3]
    w[-2] = aux2
    w[-3] = aux1
    return w

#funcion que aplica el algoritmo, para descifrar es necesario encriptar = 0
def ideaalg(clave,texto,encriptar = 1):
    z = generadorclave(clave)
    if encriptar == 0:
        z = generadordecr(z)
    x1 = texto[:16]
    x2 = texto[16:32]
    x3 = texto[32:48]
    x4 = texto[48:64]
    j = 0
    for i in xrange(8):
        g1 = multimodulo(x1,z[j])
        j += 1
        g2 = sumamodulo(x2,z[j])
        j += 1
        g3 = sumamodulo(x3,z[j])
        j += 1
        g4 = multimodulo(x4,z[j])
        j += 1
        g5 = xor(g1,g3)
        g6 = xor(g2,g4)

        g7 = multimodulo(g5,z[j])
        j += 1
        g8 = sumamodulo(g6,g7)
        g9 = multimodulo(g8,z[j])
        j += 1
        g10 = sumamodulo(g7,g9)
        g11 = xor(g1,g9)
        g12 = xor(g3,g9)
        g13 = xor(g2,g10)
        g14 = xor(g4,g10)

        x1 = g11
        x4 = g14
        if i < 7:
            x2 = g12
            x3 = g13
        elif i == 7:
            x2 = g13
            x3 = g12

    g1 = multimodulo(x1,z[j])
    j += 1
    g2 = sumamodulo(x2,z[j])
    j += 1
    g3 = sumamodulo(x3,z[j])
    j += 1
    g4 = multimodulo(x4,z[j])
    print j
    return g1 + g2 + g3 + g4

#ejemplo de uso (msg == desenc)
clave = '00110001001100100011001100110100001101010011011000110111001110000011100100110000001100010011001000110011001101000011010100110110'
msg = '0110111101110010011010010110011101101110011011010111001101100111'
res = ideaalg(clave, msg)
expected = '0100001111011010000010010100111111000110001101111001110100010110'
desenc = ideaalg(clave, ideaalg(clave, msg), 0)