2012-03-11 47 views
0

我想建立一个服务器,可以为许多客户端服务。好奇的事情发生在Linux套接字编程

服务器只做一个简单的工作:从客户端获取输入字符串,然后将每个字母改为大写。

但问题是,当我尝试关闭一个客户端时,例如,键入“Ctrl-C”,那么我的操作系统会突然关闭。

我使用Ubuntu 10.10和CentOS来测试我的程序,但总是出现同样的问题。 这里是我的源代码:

/* client.c */ 
#include <stdio.h> 
#include <signal.h> 
#include <stdlib.h> 
#include <string.h> 
#include <unistd.h> 
#include <sys/socket.h> 
#include <netinet/in.h> 
#define MAXLINE 80 
#define SERV_PORT 8000 

int main (void) 
{ 
    struct sockaddr_in servaddr, cliaddr; 
    char buf[MAXLINE]; 
    int sockfd, n; 

    sockfd = socket(AF_INET, SOCK_STREAM, 0); 

    bzero(&servaddr, sizeof(servaddr)); 
    servaddr.sin_family = AF_INET; 
    inet_pton(AF_INET, "127.0.0.1", &servaddr.sin_addr); 
    servaddr.sin_port = htons(SERV_PORT); 

    connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr)); 

    while (fgets(buf, MAXLINE, stdin) != NULL) { 
     write(sockfd, buf, strlen(buf)); 
     n = read(sockfd, buf, MAXLINE); 
     if(n == 0) printf("Connect closed\n"); 
     else write(STDOUT_FILENO, buf, n); 
    } 
    close(sockfd); 
    return 0; 
} 

/* server */ 
#include <stdio.h> 
#include <stdlib.h> 
#include <string.h> 
#include <unistd.h> 
#include <sys/socket.h> 
#include <netinet/in.h> 

#define MAXLINE 80 
#define SERV_PORT 8000 

void sigchld_func (int signo) { 
    wait(NULL); 
} 

int main (void) 
{ 
    struct sockaddr_in servaddr, cliaddr; 
    char buf[MAXLINE]; 
    char str[INET_ADDRSTRLEN]; 
    int  listenfd, connfd; 
    socklen_t cliaddr_len; 
    int  n, i; 
    pid_t pid; 

    signal(SIGCHLD, sigchld_func); 

    listenfd = socket(AF_INET, SOCK_STREAM, 0); 

    bzero(&servaddr, sizeof(servaddr)); 
    servaddr.sin_family = AF_INET; 
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY); 
    servaddr.sin_port = htons(SERV_PORT); 

    bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)); 

    listen(listenfd, 20); 

    printf("Accepting connections...\n"); 

    while (1) { 
     cliaddr_len = sizeof(cliaddr); 
     connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddr_len); 
     if ((pid = fork()) < 0) { 
      perror("fork error!"); 
      exit(1); 
     } else if (pid > 0) { 
      close(connfd); 
     } else { 
      close(listenfd); 
      while (1) { 
       n = read(connfd, buf, MAXLINE); 
       if (n <= 0) { 
        printf("Connection closed\n"); 
        break; 
       } 
       inet_ntop(AF_INET, &cliaddr.sin_addr, str, sizeof(str)); 
       printf("Received from %s at port %d\n", str, ntohs(cliaddr.sin_port)); 
       for (i = 0; i < n; i++) 
        buf[i] = toupper(buf[i]); 
       write(connfd, buf, n); 
      } 
      close(connfd); 
     } 
    } 
    return 0; 
} 
+5

呃,*** OS ***会关闭吗?它是否给你任何消息? – 2012-03-11 11:14:57

+0

如果没有正确的缩进,代码将比读取更困难。 – 2012-03-11 11:18:52

+0

对不起..操作系统暂停,我什么也没做,只是关闭我的电脑 – KUN 2012-03-11 11:19:42

回答

6

你的服务器代码导致叉炸弹,而你没有看到它正确的,因为你不检查accept返回值。

根本原因是,在读取/写入客户端的孩子中,连接关闭后,您不会调用exit或返回主体。

所以孩子进程停留在顶层while(1)循环,试图acceptlistenfd(但你关闭了那个,这是很好)。 accept失败,你不管。分叉的紧环会减慢你的计算机,调度程序不能正确处理它们(除非你有适当的反制措施)。

因此,在close(connfd)之后退出程序,并在代码中添加更多错误检查。

+0

谢谢。你的回答非常好 – KUN 2012-03-11 11:57:23