套接字编程 - 如何让服务器只存储数字

Question

我正在编写一个代码以在服务器端执行套接字编程。 现在，客户端基本上以字符串的形式发送数据（通过TCP）。我的服务器代码试图执行的操作是读取字符串并仅将数字存储在数组中，以便可以对获得的这些数字执行其他函数。但我完全不知道如何实现这个功能。

/* tcpserver.c */ 
#include <sys/types.h> 
#include <sys/socket.h> 
#include <netinet/in.h> 
#include <arpa/inet.h> 
#include <stdio.h> 
#include <stdlib.h> 
#include <unistd.h> 
#include <errno.h> 
#include <string.h> 


int main() 
{ 
    int sock, connected, bytes_recieved , true = 1; 
    int n[3],i,arr[3][5]; 
    char send_data [1024] , recv_data[1024];  

    struct sockaddr_in server_addr,client_addr;  
    int sin_size; 

    if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) { 
     perror("Socket"); 
     exit(1); 
    } 

    if (setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,&true,sizeof(int)) == -1) { 
     perror("Setsockopt"); 
     exit(1); 
    } 

    server_addr.sin_family = AF_INET;   
    server_addr.sin_port = htons(5001);  
    server_addr.sin_addr.s_addr = INADDR_ANY; 
    bzero(&(server_addr.sin_zero),8); 

    if (bind(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1) { 
     perror("Unable to bind"); 
     exit(1); 
    } 

    if (listen(sock, 5) == -1) { 
     perror("Listen"); 
     exit(1); 
    } 

    printf("\nTCPServer Waiting for client on port 5000"); 
    fflush(stdout); 

    while(1) { 
     sin_size = sizeof(struct sockaddr_in); 
     connected = accept(sock, (struct sockaddr *)&client_addr,&sin_size); 
     printf("\n I got a connection from (%s , %d)", 
       inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 
     while (1) { 
      printf("\n SEND (q or Q to quit) : "); 
      gets(send_data); 

      if (strcmp(send_data , "q") == 0 || strcmp(send_data , "Q") == 0) { 
       send(connected, send_data,strlen(send_data), 0); 
       close(connected); 
       break; 
      } 
      else { 
       send(connected, send_data,strlen(send_data), 0); 
      } 

      for(i=0;i<3;i++) { 
       bytes_recieved= recv(connected,recv_data,1024,0); 
       recv_data[bytes_recieved] = '\0'; 

       n[i]=atoi(recv_data);  
      } 

      if (strcmp(recv_data , "q") == 0 || strcmp(recv_data , "Q") == 0) { 
       close(connected); 
       break; 
      } 
      else { 
       for(i=0;i<3;i++) { 
        printf("\n RECIEVED DATA = %d \n",n[i]); 
        fflush(stdout); 
       } 
      } 
     } /* end inner while */ 

     close(sock); 
     return 0; 
    } /* end outer while */ 
} /* end main */ 

Answer 1

您没有真正说出问题所在，也没有显示客户端代码。但我怀疑这是你的服务器循环中的问题。首先，如果recv（）返回1024，你将在recv_data [1024]上写'\ 0'。该数组的有效索引范围是0..1023。你可以通过在这个函数的顶部声明recv_data长度为1025字节而不是1024来轻松解决这个问题。

但是，真正的问题是无论客户端实际发送的字节数是多少，recv（）都不能期望返回任何预期的长度。 （或者别人会告诉你：“TCP是STREAM协议”）

这可能是你想要的recv（）循环。

for (i = 0; i < 3; i++) 
{ 
    j=0; // j is declared as 32-bit int above 
    success = ReadIntegerFromSocketStream(connected, &j); 
    if (success == 0) 
    { 
     close(connected); 
     connected = -1; 
     break; 
    } 
    n[i] = j; 
} 

int ReadIntegerFromSocketStream(int sock, int* result) 
{ 
    int ret; 
    int count= 0; 

    char COMPILE_TIME_ASSERT[ (sizeof(int)==4) ? 1 : -1 ]; 

    unsigned char data[4]; 

    while (bytes_received < 4) 
    { 
     ret = recv(sock, data+count, 4-count, 0); 
     if (ret <= 0) 
     { 
      /* socket got closed, abort */ 
      return 0; 
     } 
     count += ret; 
    } 

    memcpy(&result, data, 4); 

    // not shown: calling result = ntohl(result); 
    // I'll let you look this function up 

    return 1; 
}