Gymkhana.py

#!/usr/bin/python3
# _*_ mode:python; coding:utf-8; tab-width:4 _*_

import socket
import gymkhana_operations
import httplib2
import struct
import icmp_checksum
import time
import multiprocessing


class Gymkhana:

    def __init__(self):
        init = True

    def step0(self):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        address = ('atclab.esi.uclm.es', 2000)

        sock.connect(address)

        data = sock.recv(1600).decode()
        print(data)

        sock.close()
        return data.splitlines()[0]

    def step1(self, secret_connexion_number):
        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        server_address = ('', 50000)
        sock.bind(server_address)
        address = ('atclab.esi.uclm.es', 2000)

        msg = secret_connexion_number + " 50000"
        sock.sendto(msg.encode(), address)
        data = sock.recv(1600).decode()
        print(data)
        sock.close()
        return int(data.splitlines()[0])

    def step2(self, port):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        mathematical_server = ("atclab.esi.uclm.es", port)
        sock.connect(mathematical_server)

        while 1:
            data = gymkhana_operations.recv_all_data(sock, 0.3)
            print("\n", data)
            var = gymkhana_operations.split_ecuacion(data)
            if not gymkhana_operations.is_open_parenthesis(var[0]):
                break
            result = int(gymkhana_operations.crear_arbol(data))
            string = "({})".format(result)
            print("Result is {}\n".format(result))
            sock.sendall(string.encode())

        sock.close()
        return data.splitlines()[0]

    def step3(self, download_file_number):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        direccion = 'atclab.esi.uclm.es'
        sock.connect((direccion, 5000))
        sock.sendall("GET /{} HTTP/1.1\r\nHost: atclab.esi.uclm.es\r\n\n".format(download_file_number).encode())
        content = gymkhana_operations.recv_all_data(sock, 0.3)
        print(content)
        icmp_data = content.splitlines()[5]
        return icmp_data

    def step4(self, icmp_data):
        sock = socket.socket(socket.AF_INET, socket.SOCK_RAW,
                             socket.getprotobyname("icmp"))
        echo_icmp_type = 8
        echo_icmp_code = 0
        timestamp = time.strftime("%H:%M:%S")
        payload = "{} {}".format(timestamp, icmp_data).encode()
        icmp_packet = struct.pack("!BBHHH{}s".format(len(payload))
                                  , echo_icmp_type, echo_icmp_code, 0, 0, 1, payload)
        chk = icmp_checksum.cksum(icmp_packet)
        icmp_packet = struct.pack("!BBHHH{}s".format(len(payload))
                                  , echo_icmp_type, echo_icmp_code, chk, 0, 1, payload)
        sock.sendto(icmp_packet, ("atclab.esi.uclm.es", 2000))
        reply = sock.recv(1600)
        data = sock.recv(1600)
        (echo_icmp_type,
         echo_icmp_code,
         checksum,
         x,
         y,
         data) = struct.unpack("!bbHHH{}s".format(len(data[8:])), data)
        first_line = str(data.splitlines()[0]).replace("'", "")
        code = int(first_line.split(":")[2][2:])
        print(code)
        enunciado5step = data.splitlines()[1:]
        for i in enunciado5step:
            print(i.decode())
        return code

    def step5(self, connexion_number):

        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.connect(('atclab.esi.uclm.es', 9000))
        envio = str(connexion_number)+" 40000"
        sock.sendall(envio.encode())
        multiprocessing.Process(target=gymkhana_operations.receive_end, args=(sock,)).start()

    def proxy_server(self):
        lock = multiprocessing.Lock()
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.bind(("", 40000))
        sock.listen(3)
        try:
            while True:
                sock.settimeout(6)
                connection, client_address = sock.accept()
                multiprocessing.Process(target=self.listening_to_client, args=(connection, lock)).start()
        except socket.timeout:
            pass

        sock.close

    def listening_to_client(self, client, lock):
        while True:
            data = client.recv(1600).decode()
            if data != "":
                lock.acquire()
                print("\n---------- Enviando petición HTTP... ----------\n",
                      data, "---------- Petición enviada. ------------\n")
                lock.release()
                data_splited = data.splitlines()
                host = data_splited[2].replace("Host: ", "")
                response = gymkhana_operations.get_http_resources(host, data)
                client.sendall(response.encode())
            break

        client.close


g = Gymkhana()
secret_connexion_number = g.step0()
port = g.step1(secret_connexion_number)
download_file_number = g.step2(port)
icmp_data = g.step3(download_file_number)
code = g.step4(icmp_data)
g.step5(code)
p = multiprocessing.Process(target=g.proxy_server(), args=()).start()