Все приведенное ниже относится к машинам Windows, использующим python 2.7
Здравствуйте,
В настоящее время я пытаюсь прослушивать сокет для отправки данных с помощью удаленной программы. Затем эти данные печатаются на экране, и запрашивается пользовательский ввод, который затем возвращается в удаленную программу. При тестировании я смог отправить удаленную программу мне меню программ командной строки (cmd, ipconfig, whoami, ftp), а затем моя программа вернется с номером в качестве выбора опции меню.
Удаленная программа получает мой ответ и отправляет вывод выбранной команды. ipconfig и whoami работают отлично, но cmd и ftp только возвращает вывод терминала один раз. (I.E. Я могу ввести одну команду в программу FTP и отправить эту слишком удаленную программу, прежде чем я никогда не услышу)
Часть моего кода, которая терпит неудачу, заключается в том, что
if ready[0]:
никогда не будет готов второй раз после первого разговора.
Я знаю, что удаленная программа работает правильно, так как я могу использовать netcat для работы вместо моего кода и неограниченно управлять терминалом cmd.
Как я могу правильно реализовать прослушиватель сокета python, который может учитывать этот тип соединения?
Моя "программа" в целом:
import socket, sys, struct, time, select
host = ''
port = 50000
connectionSevered=0
try:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
except socket.error:
print 'Failed to create socket'
sys.exit()
print '[+] Listening for connections on port '+str(port)+'.'
s.bind((host,port))
s.listen(5)
def recvall(the_socket,timeout=2):
global connectionSevered
data=''; # Data found by recv
total_data=[]; # Finally list of everything
s.setblocking(0) #make socket non blocking
begin=time.time() #beginning time
while 1:
ready = select.select([client], [], [], .2)
if time.time()-begin > timeout:
print 'Timeout reached'
#Leave loop, timer has reached its threshold
break
if ready[0]:
print 'In ready loop!'
try:
data = client.recv(4096) #attempt to fetch data
if data:
begin=time.time() #reset timeout timer
total_data.append(data)
data='';
except socket.error:
print '[+] Lost connection to client. Printing buffer...'
connectionSevered=1 # Let main loop know connection has errored
pass
time.sleep(1)
#join all parts to make final string
return ''.join(total_data)
client, address = s.accept()
print '[+] Client connected!'
while (connectionSevered==0): # While connection hasn't errored
print "connectionSevered="+str(connectionSevered) # DEBUG
recvall(s)
response = raw_input() #take user input
client.sendto(response) #send input
client.close(0)
Пожалуйста, дайте мне знать, если вам нужна дополнительная информация, любая помощь будет очень признательна, я очень новичок в этом и хочу учиться.