mirror of https://github.com/proot-me/proot.git
78 lines
1.6 KiB
Python
78 lines
1.6 KiB
Python
from __future__ import print_function
|
|
|
|
import socket
|
|
import sys
|
|
import os
|
|
import time
|
|
|
|
HOST = 'localhost'
|
|
PORT = 5432
|
|
|
|
pid = os.fork()
|
|
|
|
# Server
|
|
if pid:
|
|
|
|
# Create syscall
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
if len(sys.argv) > 1:
|
|
time.sleep(int(sys.argv[1]))
|
|
|
|
# Bind syscall
|
|
print("Server bind")
|
|
sock.bind((HOST, PORT))
|
|
|
|
# Listen syscall
|
|
print("Server listen")
|
|
sock.listen(1)
|
|
|
|
try:
|
|
# Accept syscall
|
|
print("Server accept")
|
|
conn, addr = sock.accept()
|
|
|
|
while True:
|
|
data = conn.recv(1024)
|
|
if not data:
|
|
break
|
|
else:
|
|
#if len(sys.argv) > 4:
|
|
# with open("fakeoutput" + sys.argv[4] + ".txt", "a") as testfile:
|
|
# testfile.write(data)
|
|
# os.remove("fakeoutput" + sys.argv[4] + ".txt")
|
|
print("Server data received : " + data.decode())
|
|
break
|
|
finally:
|
|
# Close syscall
|
|
sock.close()
|
|
|
|
# Client
|
|
else:
|
|
|
|
# Socket syscall
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
if len(sys.argv) > 2:
|
|
time.sleep(int(sys.argv[2]))
|
|
|
|
try:
|
|
# Connect syscall
|
|
print("Client connect")
|
|
sock.connect((HOST, PORT))
|
|
except socket.error as msg:
|
|
print(msg, file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
if len(sys.argv) > 3:
|
|
time.sleep(int(sys.argv[3]))
|
|
|
|
try:
|
|
# send Syscall
|
|
if len(sys.argv) > 4:
|
|
sock.send(("test " + sys.argv[4]).encode())
|
|
else:
|
|
sock.send(b"test")
|
|
finally:
|
|
sock.close()
|