今天在写一个有关于套接字的代码,其中一部分是这样写的。
while(str_len = read(clnt_sock, message, BUF_SIZE) != 0){
write(clnt_sock, message, str_len);
// write(clnt_sock, "hello world\n", strlen("hello world\n");
printf("%s\n", message);
}
调试时发现了这样的问题。每次执行后,str_len的值都是1,而本来这个值应该是read()函数的返回值,并且这个返回值应该返回的是从客户端发送到服务器端的数据字节数,所以不可能每次都是1。
经检查,是括号顺序错误。正确写法应该是下面这样。
while ((str_len = read(clnt_sock, message, BUF_SIZE)) != 0){
write(clnt_sock, message, str_len);
// write(clnt_sock, "hello world\n", strlen("hello world\n"));
printf("%s\n", message);
}
这段代码的本意是,将read()函数的返回值赋值给str_len,同时检查该值的值,如果该值为0,则跳出循环;否则进入循环。
调试过程:使用gdb调试,发现str_len的值不是预期值。于是查看read()函数的文档,发现自己的代码理解与文档没有不同。所以开始怀疑是不是标准库有问题。然后使用gdb查看main()函数的反汇编。
根据反汇编内容可以看到,read()函数被调用后,首先检查read()函数的返回值(存放在%rax寄存器中),如果返回值不是0,则执行setne语句,设置%al寄存器(%rax寄存器的低8位)的值,然后将%al寄存器的值放到%eax寄存器中。而%eax寄存器即为%rax寄存器低32位。所以可以看到在这里,read()函数的返回值被丢弃,而这个值在后面调用write()函数需要被用到。所以可以推断出代码逻辑错误。
附上源代码
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <unistd.h>
#include <errno.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define BUF_SIZE (1024)
void error_handling(const char* message){
printf("%s\n", message);
printf("ErrNo: %d\n", errno);
printf("ErrMsg: %s\n", strerror(errno));
exit(1);
}
int main(int argc, char** argv){
int serv_sock, clnt_sock;
char message[BUF_SIZE];
ssize_t str_len, i;
sockaddr_in serv_adr, clnt_adr;
socklen_t clnt_adr_sz;
if (argc != 2){
printf("Usage : %s <port>\n", argv[0]);
exit(1);
}
serv_sock = socket(PF_INET, SOCK_STREAM, 0);
if (serv_sock == -1){
error_handling("socket() error");
}
memset(&serv_adr, 0, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr= htonl(INADDR_ANY);
serv_adr.sin_port = htons(atoi(argv[1]));
if (bind(serv_sock, (sockaddr*) &serv_adr, sizeof(serv_adr))==-1
){
error_handling("bind() error");
}
if (listen(serv_sock, 5) == -1){
error_handling("listen() error");
}
clnt_adr_sz = sizeof(clnt_adr);
for (i=0; i<5; i++){
clnt_sock = accept(serv_sock, (sockaddr*) &clnt_adr, &cl
nt_adr_sz);
if (clnt_sock == -1){
error_handling("accept() error");
} else {
printf("Connected client %d\n", i+1);
}
while((str_len = read(clnt_sock, message, BUF_SIZE) != 0
)){
write(clnt_sock, message, str_len);
// write(clnt_sock, "hello world\n", strlen("hello
world\n"));
printf("%s\n", message);
}
close(clnt_sock);
}
close(serv_sock);
return 0;
}
main: echo_server.cpp
g++ echo_server.cpp -Wall -o -g 2> error.txt