feat: update subprocess

Change-Id: Ib095986ad8686baa2c9134797baa7ed07900cdfb

Author: windard

code/subprocess_demo.py

@@ -1,22 +1,39 @@
 # coding=utf-8
-
+import os
+import six
 import subprocess
 
+if six.PY2:
+    input = raw_input
+cwd = os.getcwd()
+
+
 def run_command(command):
+    global cwd
     command = command.rstrip()
+    commands = command.split()
+    if commands and commands[0] == "cd":
+        cwd = (len(commands) > 1 and os.path.join(cwd, commands[1])) or os.path.expanduser('~')
+        return b""
+
     try:
-        output = subprocess.check_output(command, stderr=subprocess.STDOUT, shell=True)
+        process = subprocess.Popen(command,stdout=subprocess.PIPE,stderr=subprocess.PIPE, shell=True, cwd=cwd)
+        stdout, stderr = process.communicate()
+        retcode = process.poll()
+        if retcode:
+            return stderr
+        return stdout
     except:
         output = 'Failed to execute command.\r\n'
 
     return output
 
 
 if __name__ == '__main__':
-	while 1:
-		command = raw_input("$ ")
-		if command == "exit" or command == "quit":
-			break
-		result = run_command(command)
-		
-		print result,
+    while 1:
+        command = input("$ ")
+        if command == "exit" or command == "quit":
+            break
+        result = run_command(command)
+        
+        print(result.decode("utf-8"))

code/subprocess_shell.py

@@ -0,0 +1,109 @@
+# -*- coding: utf-8 -*-
+
+import sys
+import socket
+import argparse
+import threading
+import subprocess
+
+
+class TargetServer(object):
+
+    def __init__(self, port):
+        self.port = port
+        self.host = socket.gethostname()
+        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+        self.server.bind(("0.0.0.0", int(self.port)))
+        self.server.listen(10)
+
+    def run(self):
+        while 1:
+            client_socket, client_addr = self.server.accept()
+            client_thread = threading.Thread(target=self.client_handler,
+                                             args=(client_socket,))
+            client_thread.start()
+
+    def client_handler(self, client_socket):
+        client_socket.sendall("<@ %s $ >" % self.host)
+        while 1:
+            try:
+                cmd_buffer = client_socket.recv(1024)
+                response = self.run_command(cmd_buffer)
+                if len(response) == 0:
+                    response = "[Successful!]\n"
+                client_socket.sendall(response)
+            except Exception as e:
+                # print e
+                break
+
+    def run_command(self, command):
+        command = command.strip()
+        try:
+            output = subprocess.check_output(command, stderr=subprocess.STDOUT,
+                                             shell=True)
+        except:
+            output = '[*]Failed to execute command ! \n'
+
+        return output
+
+
+class Client(object):
+
+    def __init__(self, host, port):
+        self.host = host
+        self.port = port
+        self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.client.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+
+    def run(self):
+        try:
+            self.client.connect((self.host, int(self.port)))
+            header = self.client.recv(4096)
+            command = raw_input(header)
+            if command == "exit" or command == "quit":
+                self.clien.close()
+                sys.exit(0)
+            self.client.sendall(command)
+            while 1:
+                recv_len = 1
+                response = ""
+
+                while recv_len:
+                    data = self.client.recv(4096)
+                    recv_len = len(data)
+                    response += data
+                    if recv_len < 4096:
+                        break
+
+                print(response)
+
+                command = raw_input(header)
+                if command == "exit" or command == "quit":
+                    self.client.close()
+                    break
+                self.client.sendall(command)
+
+        except:
+            print("[*] Exception Failed ! \n")
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description="NetCat Shell")
+    parser.add_argument("-s", "--server", help="Target Server",
+                        action="store_true")
+    parser.add_argument("-c", "--client", help="Client", action="store_true")
+    parser.add_argument("--host", help="target host IP", action="store",
+                        default="127.0.0.1")
+    parser.add_argument("-p", "--port", help="target host port", action="store",
+                        type=int)
+    args = parser.parse_args()
+    port = args.port
+    if args.server:
+        s = TargetServer(port)
+        s.run()
+    if args.client:
+        host = args.host
+        c = Client(host, port)
+        c.run()
+