2009-12-15 44 views
23

我找到了this,我正在使用它作为我的基础,但它在开箱时并不能正常工作。我的目标也是将其视为一个包而不是命令行实用程序,所以我的代码更改将反映这一点。Netcat在Python中的实现

class Netcat: 
    def __init__(self, hostname, port): 
     self.hostname = hostname 
     self.port = port 
    def send(self, content): 
    self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    self.socket.connect((self.hostname, self.port)) 
    self.socket.setblocking(0) 
    result = ''; 
    read_ready, write_ready, in_error = select.select([self.socket], [], [self.socket], 5) 
    if(self.socket.sendall(content) != None): 
     return 
    while(1): 
     buffer = '' 
     try:     
      buffer = self.socket.recv(128) 
      while(buffer != ''): 
       result += buffer 
       try: 
        buffer = self.socket.recv(128) 
       except socket.error as err: 
        print (err, type(err)) 
        buffer = '' 
      if(buffer == ''): 
       break 
     except socket.error as err: 
      print (err, type(err)) 
     if(buffer == ''): 
      break 
    return result 

当我发送一个基本的命令到我的设备,它返回以下内容。

50PMA-019 Connection Open 
Atten #1 = 63dB 

我的代码读取第一行,但后来我得到一个错误说连接暂时不可用,并没有得到第二行。如果我将其更改为阻止,它会阻止并永不返回。有什么想法吗?

回答

40

如果您只是使用nc,它会工作吗?

我想你应该尝试一些简单的一点:

import socket 

def netcat(hostname, port, content): 
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    s.connect((hostname, port)) 
    s.sendall(content) 
    s.shutdown(socket.SHUT_WR) 
    while 1: 
     data = s.recv(1024) 
     if data == "": 
      break 
     print "Received:", repr(data) 
    print "Connection closed." 
    s.close() 

我加入了shutdown调用,因为也许你的设备正等着你说你完成发送数据。 (这将是一个有点怪异,但它是可能的。)

+2

将关机行添加到我原来的代码使它工作,我会借用你的代码来修剪我的代码。 – unholysampler 2009-12-15 18:36:50

+0

我有问题,我的mac os不会发送数据包,直到我关闭连接。有没有办法解决,因为我不能再次建立连接。 – 2016-03-07 17:25:50

+0

鲍勃,你应该考虑点击提问并询问这个,如果你还没有。 – 2016-03-11 23:17:44

3

如果你不介意完全废除该代码,你可能会喜欢看scapy--它基本上就是Python中包工具的瑞士军刀。看看interactive tutorial看看它是否适合您的需求。

如果你想要比数据包更高级别的东西twisted是python网络的前往图书馆...不幸的是学习曲线是一个有点陡峭。

+1

我没什么特别的。我只需要一个简单的界面让我发送几个字符并接收设备的响应。我会检查你给我的链接。 – unholysampler 2009-12-15 18:08:59

7

下面是关于python3工作实施:

import socket 

def netcat(host, port, content): 
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    s.connect((host, int(port))) 
    s.sendall(content.encode()) 
    s.shutdown(socket.SHUT_WR) 
    while True: 
     data = s.recv(4096) 
     if not data: 
      break 
     print(repr(data)) 
    s.close() 

它可以用于“内容”发送到“主机”端口” “(这一切都可能被视为刺痛)。

Regards

+0

我发现我在这里也需要在shutdown之前添加'time.sleep(0.5)'。 – tripleee 2016-01-29 06:43:46