Python中的Netcat实现

时间:2009-12-15 17:01:42

标签: python netcat

我找到了this并且正在使用它作为我的基础,但它没有开箱即用。我的目标也是将其视为包而不是命令行实用程序,因此我的代码更改将反映出来。

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

我的代码读取了第一行,但后来我收到一条错误消息,指出该连接暂时不可用,并且没有获得第二行。如果我将其更改为阻止,它只会阻塞而永远不会返回。有什么想法吗?

3 个答案:

答案 0 :(得分:45)

如果您只使用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电话,因为您的设备可能正在等待您说您已完成发送数据。 (这有点奇怪,但它有可能。)

答案 1 :(得分:10)

以下是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()

它可用于发送内容"到主持人" on" port" (所有这些都可以作为刺痛输入)。

此致

答案 2 :(得分:3)

如果您不介意完全废弃该代码,您可能希望看一下scapy - 它基本上是python中包工具的瑞士军刀。看看interactive tutorial,看它是否符合您的需求。

如果你想要一些比数据包更高级别的东西twisted是python中用于联网的首选库......不幸的是,学习曲线有点陡峭。