Crypto/client.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
# -*- coding: utf-8 -*-
"""
Created on Wed Feb 15 15:45:10 2017

@author: adrian
"""

#script para enviar y recibir mensajes, correspondiente al cliente o Bob

from multiprocessing.connection import Client
import cripto1
import random

address = ('localhost', 6000)
conn = Client(address, authkey='secret password')

#elegimos si solo nos interesa Diffie-Hellman o no
dh = input("Diffie-Hellman yes or no? yes = 1, no = 0: ")
if dh == 0:
    msg1 = conn.recv()
else:
    msg1 = '0'

#si hemos elegido RSA (en la parte del servidor) enviaremos este mensaje
if msg1[:3] == 'RSA':
    m_rsa = 123
    comma = msg1.index(',')
    encr_rsa = modular_pow(m_rsa,int(msg1[3:comma]),int(msg1[comma+1:]))
    conn.send('RSAe' + str(encr_rsa))

#si hemos elegido ElGamal (en la parte del servidor) enviaremos este mensaje
if msg1[:2] == 'EG':
    comma = msg1.index(',')
    semicolon = msg1.index(';')
    m_gamal = 128688
    b_gamal = 480
    y1_gamal = modular_pow(int(msg1[2:comma]),b_gamal,
                           int(msg1[comma+1:semicolon]))
    y2_gamal = (modular_pow(int(msg1[semicolon+1:]),b_gamal,
                           int(msg1[comma+1:semicolon])) * m_gamal
                           ) % int(msg1[comma+1:semicolon])
    conn.send('EGe' + str(y1_gamal) + ',' + str(y2_gamal))

#Diffie-Hellman
p = 2**128-159
#g = 127696581778553318510832226979062695964
g = 5
b = p
while b > p-1:
    bstr = numerogrande(int(round(random.random()*10)+10))
    b = int(bstr,2)
#B = g**b % p
print "b es"
print b
B = modular_pow(g,b,p)
print "B es"
print B
message = 'B' + str(B)

conn.send(message)

msg = conn.recv()
if msg[0] == 'A':
    message2 = msg[1:]

A = int(message2)
print "A es"
print A
#s = A**b % p
s = modular_pow(A,b,p)
print "s es"
print s      #clave secreta Diffie-Hellman

#ahora ciframos un texto con Vigenere y lo enviamos
cifrador = 'prueba'
conn.send('cif' + cifrador)

text = cripto1.vigenere('holaaliceestoyprobando', cifrador)
conn.send('vig' + text)

#ahora ciframos un texto con IDEA y lo enviamos
textoidea = '0110111101110010011010010110011101101110011011010111001101100111'
e = bin(s)
f = int(e,2)
claveidea = '{0:0{1}b}'.format(f,128)
cifradoidea = ideaalg(clave,textoidea)

conn.send('idea' + cifradoidea)

#cerramos la conexion con el servidor
conn.send('close')

# can also send arbitrary objects:
# conn.send(['a', 2.5, None, int, sum])

conn.close()