#!/usr/bin/python
from pygenere import *
import socket
import string
import time

s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
connect=s.connect(("54.209.5.48",12345))

data = s.recv(1024)
print data
data = s.recv(1024)
time.sleep(.2)
print data

cipher = data[data.index("psifer text:") + 13:-1]

caesar = ""
for shift in xrange(26):
	alphabet = string.ascii_lowercase
	shifted_alphabet = alphabet[shift:] + alphabet[:shift]
	table = string.maketrans(alphabet, shifted_alphabet)
	caesar += cipher.translate(table)

plain = caesar[caesar.index("stage is ") + 9:caesar.index("ui", caesar.index("stage is") + 9)]
print plain

s.send(plain + "\n")

data = s.recv(1024)
time.sleep(.2)
print "\n" + data

cipher = data[data.index("psifer text:") + 13:-1]

for i in xrange(1,50):
	box = ""
	for j in xrange(i):
		for k in xrange(j,len(cipher),i):
			box += cipher[k]
	if "I hope" in box:
		print box + "\n"
		plain = box[box.index("is \"") + 4:box.index("\"", box.index("is \"") + 4)]
		break

print plain + "\n"
s.send(plain + "\n")

data = s.recv(4096)
time.sleep(.2)
print "\n" + data

cipher = data[data.index("psifer text:") + 13:-1]

vig = VigCrack(cipher).crack_message().replace(" ","")
plain = vig[vig.index("RIGHTHERE") + 9:vig.index("OKNOW", vig.index("RIGHTHERE") + 9)]

print plain + "\n"
s.send(plain + "\n")

data = s.recv(4096)
time.sleep(.2)
print "\n" + data

s.close()