Question

我想实现这个读/写 UNIX套接字的bash命令的功能：

echo „READ <a command>“ | nc -U /tmp/socket

此命令产生以下输出：

Click::ControlSocket/1.3
200 Read handler 'debug_handler.ping' OK
DATA 0

我目前的c / c ++实现看起来像这样：

myprogram.hh

class Myprogram {
public:
    Myprogram(int argc, char** argv);
    ~Myprogram() { };   
private:
    foo();
    int sockfd;
    int len;
    sockaddr_un address;
};

myprogram.cc

Myprogram::foo() {
    sockfd = socket(AF_UNIX, SOCK_STREAM, 0);
    address.sun_family = AF_UNIX;
    strcpy(address.sun_path, "/tmp/socket");
    len = sizeof(address);

    if(connect(sockfd, (sockaddr*)&address, len) == -1)
        return;

    std::string args = "READ <a command>";

    if (write(sockfd, args.c_str(), sizeof(args.c_str())) == -1){
        std::cout << "Error" << std::endl;
        return;
    }   

    int n;
    char ret_value[200];
    do {
        n = read(sockfd, ret_value, 200);
        std::cout << std::string(ret_value) << std::endl;
        std::cout << "n=" << std::to_string(n) << std::endl;
        *ret_value+= n;
    } while (n > 0);

    close(sockfd);
}

我只得到这个输出：

Click::ControlSocket/1.3
n=26

下一次read（）调用没有返回，程序等待。我已经尝试了很多来读取套接字，所有结果都相同，所以，如果有人能给我一个暗示我做错了什么的话会很棒。

Answer 1

std::string args = "READ <a command>";

服务器正在等待换行符（默认情况下echo添加一个换行符），所以：

std::string args = "READ <a command>\n";

另外

char ret_value[200];
do {
    n = read(sockfd, ret_value, 200);

应该更像

const size_t buflen = 200;
char ret_value[buflen+1];
do {
    n = read(sockfd, ret_value, buflen);
    if (n < 0)
      // error
    else
      ret_value[n] = 0;

和这一行

    *ret_value+= n;

似乎不清楚 - 你为什么会将ret_value的第一个字节递增n？

Answer 2

尝试更像这样的事情：

Myprogram::foo() {
    sockfd = socket(AF_UNIX, SOCK_STREAM, 0);
    if (sockfd == -1) {
        std::cout << "Error" << std::endl;
        return;
    }

    address.sun_family = AF_UNIX;
    strcpy(address.sun_path, "/tmp/socket");

    if (connect(sockfd, (sockaddr*)&address, sizeof(address)) == -1) {
        std::cout << "Error" << std::endl;
        close(sockfd);
        return;
    }

    std::string args = "READ <a command>\n";
    if (write(sockfd, args.c_str(), args.size())) == -1) {
        std::cout << "Error" << std::endl;
        close(sockfd);
        return;
    }   

    int n;
    char ret_value[200];
    do {
        n = read(sockfd, ret_value, sizeof(ret_value));
        if (n < 0) {
            std::cout << "Error" << std::endl;
            break;
        }

        if (n == 0) {
            break;
        }

        std::cout.write(ret_value, n) << std::endl;
        std::cout << "n=" << n << std::endl;
    } while (true);

    close(sockfd);
}

当然，如果你真的希望输出看起来像原版，而不是将自己的字符添加到cout（除非发生错误）：

int n;
char ret_value[200];
do {
    n = read(sockfd, ret_value, sizeof(ret_value));
    if (n < 0) {
        //std::cout << "Error" << std::endl;
        break;
    }

    if (n == 0) {
        break;
    }

    std::cout.write(ret_value, n);
} while (true);

如何以正确的方式读/写UNIX套接字？

2 个答案: