python - Python中的Netcat实现

Question

我找到了这个并将其用作我的基础，但它并不能直接使用。我的目标也是将其视为一个包而不是命令行实用程序，因此我的代码更改将反映这一点。

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

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

Answer 1

如果你只使用它会起作用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 len(data) == 0:
            break
        print("Received:", repr(data))
    print("Connection closed.")
    s.close()

我添加了shutdown呼叫，因为您的设备可能正在等待您说您已完成发送数据。（这会有点奇怪，但这是可能的。）

Answer 2

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

它可用于将“内容”发送到“端口”上的“主机”（所有这些都可以作为字符串输入）。

问候

Answer 3

如果您不介意完全废弃该代码，您可能想看看scapy——它基本上是 python 中数据包工具的瑞士军刀。查看交互式教程，看看它是否符合您的需求。

如果你想要比数据包更高级别的东西，那么在python中网络的首选库是……不幸的是，学习曲线有点陡峭。