下面提到的所有内容都是在使用 python 2.7 的 Windows 机器上
你好,
我目前正在尝试在套接字上侦听远程程序发送的数据。然后将该数据打印到屏幕上,并请求用户输入,然后将其返回到远程程序。在测试中,我能够让远程程序向我发送一个命令行程序菜单(cmd、ipconfig、whoami、ftp),然后我的程序返回一个数字作为菜单选项的选择。
远程程序接收我的响应并发送所选命令的输出。ipconfig 和 whoami 工作完美,但 cmd 和 ftp 只返回一次终端的输出。(即我可以在 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)
如果您需要更多信息,请告诉我,任何帮助将不胜感激,我对此很陌生并且渴望学习。