string - bufio 扫描仪和处理新线路

Question

我有 2 个进程通过 TCP 套接字进行通信。 A 方向 B 方发送一个字符串，该字符串有时使用标准加密/密码包进行加密。结果字符串可能包含一个换行符，但 B 端的 bufio 扫描器将其解释为请求的结束。我希望 B 方继续接受行，附加它们并等待已知的命令结束字符，然后再进一步处理它。 B 方将向 A 方返回响应，因此连接保持打开状态，因此不能使用关闭连接事件作为命令分隔符。

单行命令一切正常，但加密输出中的这些新行字符会导致问题(大约 10% 的时间)。

A 面将以以下格式发送(第三个是我正在尝试正确处理的问题字符串的合法示例):

调用命令()

callCommand("一","二","三")

callCommand("string","encrypted-data-to-follow","[7b��Cr��l��G����bH�@x�������������(z�$� a��0���5Y7+��U�QT�ΐl�K�(�n�U��J����QK�BX�+�l\8H��-g�y..�1� f��I�C�Ȓ㳿���o�xz�8?��c�e���Tb��?4�hD W���
�<����Е�gc�������N�V���ۓP8 �����O3")

我们可以相当可靠地说命令结束键是一个右括号“)”和一个换行符。

A方发送到B方的功能:

func writer(text string) string {
    conn, err := net.Dial("tcp", TCPdest)
    t := time.Now()
    if err != nil {
        if _, t := err.(*net.OpError); t {
            fmt.Println("Some problem connecting.\r\n")
        } else {
            fmt.Println("Unknown error: " + err.Error()+"\r\n")
        }
    } else {
        conn.SetWriteDeadline(time.Now().Add(1 * time.Second))
        _, err = conn.Write([]byte(text+"\r\n"))
        if err != nil {
            fmt.Println("Error writing to stream.\r\n")
        } else {
            timeNow := time.Now()           
            if timeNow.Sub(t.Add(time.Duration(5*time.Second))).Seconds() > 5 {
                return "timeout"
            }
            scanner := bufio.NewScanner(conn)
            for {
                ok := scanner.Scan()
                if !ok {
                    break
                }
                if strings.HasPrefix(scanner.Text(), "callCommand(") && strings.HasSuffix(scanner.Text(), ")") {
                    conn.Close()
                    return scanner.Text()
                }
            }
        }
    }
    return "unspecified error"
}

B 方对传入连接的处理:

src := "192.168.68.100:9000"
listener, _ := net.Listen("tcp", src)

defer listener.Close()

for {
    conn, err := listener.Accept()
    if err != nil {
        fmt.Println("Some connection error: %s\r\n", err)
    }
    go handleConnection(conn)
}   

func handleConnection(conn net.Conn) {
    remoteAddr := conn.RemoteAddr().String()
    fmt.Println("Client connected from " + remoteAddr + "\r\n")

    scanner := bufio.NewScanner(conn)
    wholeString := ""
    for {
        ok := scanner.Scan()

        if !ok {
            break
        }

        //Trying to find the index of a new-line character, to help me understand how it's being processed
        fmt.Println(strings.Index(scanner.Text(), "\n"))
        fmt.Println(strings.Index(wholeString, "\n"))

        //for the first line received, add it to wholeString
        if len(wholeString) == 0 {
            wholeString = scanner.Text()
        }

        re := regexp.MustCompile(`[a-zA-Z]+\(.*\)\r?\n?`)

        if re.Match([]byte(wholeString)) {
            fmt.Println("Matched command format")
            handleRequest(wholeString, conn)
        } else if len(wholeString) > 0 && !re.Match([]byte(wholeString)) {
            //Since we didn't match regex, we can assume there's a new-line mid string, so append to wholeString
            wholeString += "\n"+scanner.Text()
        }

    }
    conn.Close()
    fmt.Println("Client at " + remoteAddr + " disconnected.\r\n")
}

func handleRequest(request string, conn net.Conn) {
    fmt.Println("Received: "+request)
}

我不确定 B 面的这种方法是否正确，但在上面包含了我的代码。我见过一些实现，但很多似乎依赖于关闭连接来开始处理请求，这不适合我的场景。

任何指针表示赞赏，谢谢。

Answer 1

您的通信“协议(protocol)”(一行是一条消息，不完全是协议(protocol))显然无法处理二进制数据。如果您想在您的协议(protocol)中发送文本数据，您可以将二进制数据转换为文本，例如使用 Base64 编码。您还需要一些语义来指示某些文本是从二进制转换而来的。

或者您可以更改您的协议(protocol)以本地处理二进制数据。您可以在后面加上二进制数据的长度，这样您就知道必须将此数据作为二进制读取，而不是将换行符解释为消息的结尾。

有许多协议(protocol)做得很好，也许你不需要想出你的自定义协议(protocol)。如果你想发送文本消息，HTTP 使用非常简单，你可以将数据格式化为 JSON，使用 Base64 将二进制数据转换为文本:

{
    "command": "string",
    "args": [
        "binaryDataAsBase64"
    ]
}