python socket

 1 #!/usr/bin/env python
 2 # coding: utf-8
 3 
 4 import random
 5 import time
 6 import SocketServer
 7 
 8 from congratulation import FLAG
 9 
10 
11 banner = r‘‘‘
12   ___|      _)         ___|
13  |      _ \  | __ \   |      _` | __ `__ \   _ 14  |     (   | | |   |  |   | (   | |   |   |  __/
15 \____|\___/ _|_|  _| \____|\__,_|_|  _|  _|\___|
16 ‘‘‘
17 
18 
19 class HandleClient(SocketServer.StreamRequestHandler):
20     timeout = 3
21 
22     def handle(self):
23         req = self.request
24         req.sendall(banner + ‘\n\n‘)
25 
26         iround = 0
27         allround = 1000
28         start = int(time.time())
29         seed = int(time.time())
30         random.seed(seed)
31         t = time.localtime(seed)
32         req.sendall(‘Server time: [%s]\n\n‘ % time.asctime(time.localtime(seed)))
33         req.sendall(‘Welcome to CCTF final battle.\n\nI think you are good at programing, so let\‘s play coin game.\n‘)
34         req.sendall(‘Rules: just guess positive(1) or negative(0) of coin status in each round.\n\n‘)
35         req.sendall(‘I will print flag to you, if you guess all status in 1000 rounds. :)\n\n‘)
36 
37         while iround < allround:
38             """
39             if int(time.time()) - start > 30:
40                 req.sendall("Time Out...")
41                 req.close()
42                 return
43             """
44             answer = random.choice([‘0‘, ‘1‘])
45             req.sendall(‘::::: Round %d :::::\nguess:‘ % (iround + 1))
46 
47             try:
48                 t = req.recv(8).strip()
49             except Exception:
50                 req.close()
51                 return
52 
53             if t == answer:
54                 req.sendall(‘Nice job, go on.\n‘)
55                 iround += 1
56             else:
57                 req.sendall("Wrong :(\n")
58                 req.close()
59                 return
60 
61         req.sendall("You hit it! Flag is %s" % FLAG)
62         print ‘Got flag from‘, self.client_address
63 
64 
65 class ThreadedServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
66     pass
67 
68 
69 if __name__ == ‘__main__‘:
70     host, port = "", int(55554)
71     server = ThreadedServer((host, port), HandleClient)
72     server.allow_reuse_address = True
73     server.serve_forever()

 1 import socket, time, re, random
 2 HOST = ‘‘
 3 PORT = 55554
 4 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 5 s.connect((HOST, PORT))
 6 time.sleep(0.5)
 7 f = s.recv(512)
 8 print f
 9 t = re.findall(re.compile(r‘time:\s\[(.*?)\]‘), f)[0]
10 u = int(time.mktime(time.strptime(t, ‘%a %b %d %H:%M:%S %Y‘)))
11 random.seed(u)
12 for i in range(1000):
13     answer = random.choice([‘0‘, ‘1‘])
14     s.send(answer)
15     time.sleep(0.00025)
16     print s.recv(64)