如何将cpp结构中的字符串转换为python结构

时间:2017-09-26 09:18:43

标签: python c++ struct

我写了一个tcp服务器和tcp客户端,我想从客户端发送一个结构到服务器,然后服务器发回结构。服务器是用cpp编写的,客户端是用python编写的。 然后我遇到了一个问题,我的结构有一个类型是字符串的成员,然后我找不到一个正确的格式字符串来描述python中的这个成员(https://docs.python.org/2/library/struct.html)。 所以我想知道如何在python中描述字符串属性?

下面是最小代码: ------服务器---------

#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <cstring>
#include <iostream>

struct Message
{
    int id;
    std::string text;
};

int main()  
{
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    struct sockaddr_in my_addr;  
    bzero(&my_addr, sizeof(my_addr));
    my_addr.sin_family = AF_INET;
    my_addr.sin_port = htons(11111);
    my_addr.sin_addr.s_addr = htonl(INADDR_ANY);

    unsigned retry_times = 0;
    while(retry_times < 3)
    {
        if(bind(sockfd, (struct sockaddr*)&my_addr, sizeof(my_addr)) == 0)
        {
            break;
        }
        sleep(1000);
        ++retry_times;
    }
    listen(sockfd, 1);

    struct sockaddr_in client_addr;              
    socklen_t addr_len = sizeof(client_addr);        
    int connfd = accept(sockfd, (struct sockaddr*)&client_addr, &addr_len);

    while(true)  
    {
        Message msg;
        unsigned int len = recv(connfd, &msg, sizeof(msg), 0);
        if(len == 0)
        {
            break;
        }
        std::cout << msg.id << "--" << msg.text << std::endl;
        send(connfd, &msg, sizeof(msg), 0);
    }

    close(connfd);
    close(sockfd);
}

----------- -----------客户

import socket
import struct

class TCPClient:
    def __init__(self):
        self.server_address = ('127.0.0.1', 11111)  
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.sock.connect_ex(self.server_address)

    def send(self, msg):
        self.sock.send(msg)

    def recv(self):
        return self.sock.recv(512)

    def close(self):
        self.sock.close()

if __name__ == '__main__':
    client = TCPClient()
    fmt = 'is'         #### what format string will be better? this fmt is useless 
    msg = bytearray(512)
    struct.pack_into(fmt, msg, 0, 1, "Hello")
    client.send(msg)
    data = client.recv()
    print struct.unpack_from(fmt, data)
    client.close()

1 个答案:

答案 0 :(得分:1)

没有Python格式字符串可以解决此问题。根本原因是您在网络上发送std::string的字节。但是std::string通常包含指向实际内容的指针(和大小),这意味着您根本不发送实际内容!

要解决此问题,您有两个主要选择 - 修改结构:

struct Message
{
    int32_t id; // use a fixed-size integer type
    char text [50]; // assumes you know the max length
};

或者更常规地,编写序列化例程:

void send_msg(int fd, const Message& msg)
{
    // XXX: you must check the return value of send()
    send(fd, &msg.id, sizeof(msg.id), 0);
    uint32_t text_size = msg.text.size();
    send(fd, &text_size, sizeof(text_size), 0);
    send(fd, &msg.text.data(), msg.text.size(), 0);
}

现在我们发送ID(4个字节,Python格式i),text_size(4个字节,无符号,Python格式I),然后是文本(text_size个字节, Python格式'{}s'.format(text_size))。您可以在接收端分两步解压缩它:首先解压缩iI以获取ID和text_size,然后解压缩'{}s'.format(text_size)以获取文本。

请注意,由于您使用的是TCP,因此可以进行部分写入和读取。因此,您需要处理发件人可能只发送一半消息的事实,而您的接收者可能只收到一半消息。您当前的逻辑不会处理此问题(但如果text_size小于500,则大部分时间似乎都可以正常工作。)