2012-05-16 55 views
2

我在服务器端为使用C的聊天客户端编写了下面的代码。但是,在while循环中,我需要检查客户端是否已退出程序,即与客户端的连接是否丢失。请告诉我如何编写这样的代码。如何检查服务器端是否建立了连接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> 
#include <time.h> 
int main() 
{ 
int sock, connected, bytes_recieved , true = 1, pid; 
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(3128);  
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 3128"); 
fflush(stdout); 
FILE *log; 
log = fopen("time.log", "a"); 
time_t t; 
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)); 
    fflush(stdout);   
    int pid=fork(); 
    while(1) 
    {  
     if(pid == 0) 
     { 
      usleep(1000000); 
      time(&t); 
      send(connected, ctime(&t),30, 0); 
     } 
     else 
     { 
      bytes_recieved = recv(connected,recv_data,1024,0); 
      recv_data[bytes_recieved] = '\0'; 
      fputs(recv_data, log); 
      fputs("Time: ", log); 
      time(&t); 
      fputs(ctime(&t), log); 
     } 
     fflush(log); 
    } 
    } 
fclose(log); 
close(sock); 
return 0; 
} 

请帮忙。

现在我修改这样的代码:

while(1) 
{  
    if(pid == 0) 
    { 
     usleep(1000000); 
     time(&t); 
     send(connected, ctime(&t),30, 0); 
    } 
    else 
    { 
     bytes_recieved = recv(connected,recv_data,1024,0); 
     recv_data[bytes_recieved] = '\0'; 
     fputs(recv_data, log); 
     fputs("Time: ", log); 
     time(&t); 
     fputs(ctime(&t), log); 
    } 
    fflush(stdout); 
    fflush(log); 
    fclose(log); 
    if(bytes_recieved == 0 || bytes_recieved == -1) 
    { 
     close(sock); goto connct;} 
    } 
} 
close(sock); 
return 0; 

在编译时,它会很好,但上运行,它提供了以下错误的输出:

*** glibc detected *** ./server: double free or corruption (!prev): 0x09936008 *** 
======= Backtrace: ========= 
/lib/i686/cmov/libc.so.6(+0x6b381)[0xb7659381] 
/lib/i686/cmov/libc.so.6(+0x6cbd8)[0xb765abd8] 
/lib/i686/cmov/libc.so.6(cfree+0x6d)[0xb765dcbd] 
/lib/i686/cmov/libc.so.6(fclose+0x14a)[0xb764982a] 
./server[0x8048c21] 
/lib/i686/cmov/libc.so.6(__libc_start_main+0xe6)[0xb7604ca6] 
./server[0x8048861] 
======= Memory map: ======== 
08048000-08049000 r-xp 00000000 08:01 133298  /home/harikrishnan/server/server 
08049000-0804a000 rw-p 00000000 08:01 133298  /home/harikrishnan/server/server 
09936000-09957000 rw-p 00000000 00:00 0   [heap] 
b7400000-b7421000 rw-p 00000000 00:00 0 
b7421000-b7500000 ---p 00000000 00:00 0 
b75bf000-b75dc000 r-xp 00000000 08:01 3653635 /lib/libgcc_s.so.1 
b75dc000-b75dd000 rw-p 0001c000 08:01 3653635 /lib/libgcc_s.so.1 
b75ed000-b75ee000 rw-p 00000000 00:00 0 
b75ee000-b772e000 r-xp 00000000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b772e000-b772f000 ---p 00140000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b772f000-b7731000 r--p 00140000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b7731000-b7732000 rw-p 00142000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b7732000-b7735000 rw-p 00000000 00:00 0 
b7744000-b7747000 rw-p 00000000 00:00 0 
b7747000-b7748000 r-xp 00000000 00:00 0   [vdso] 
b7748000-b7763000 r-xp 00000000 08:01 3654097 /lib/ld-2.11.3.so 
b7763000-b7764000 r--p 0001b000 08:01 3654097 /lib/ld-2.11.3.so 
b7764000-b7765000 rw-p 0001c000 08:01 3654097 /lib/ld-2.11.3.so 
bfd94000-bfda9000 rw-p 00000000 00:00 0   [stack] 
*** glibc detected *** ./server: double free or corruption (!prev): 0x09936008 *** 
======= Backtrace: ========= 
/lib/i686/cmov/libc.so.6(+0x6b381)[0xb7659381] 
/lib/i686/cmov/libc.so.6(+0x6cbd8)[0xb765abd8] 
/lib/i686/cmov/libc.so.6(cfree+0x6d)[0xb765dcbd] 
/lib/i686/cmov/libc.so.6(fclose+0x14a)[0xb764982a] 
./server[0x8048c21] 
/lib/i686/cmov/libc.so.6(__libc_start_main+0xe6)[0xb7604ca6] 
./server[0x8048861] 
======= Memory map: ======== 
08048000-08049000 r-xp 00000000 08:01 133298  /home/harikrishnan/server/server 
08049000-0804a000 rw-p 00000000 08:01 133298  /home/harikrishnan/server/server 
09936000-09957000 rw-p 00000000 00:00 0   [heap] 
b7400000-b7421000 rw-p 00000000 00:00 0 
b7421000-b7500000 ---p 00000000 00:00 0 
b75bf000-b75dc000 r-xp 00000000 08:01 3653635 /lib/libgcc_s.so.1 
b75dc000-b75dd000 rw-p 0001c000 08:01 3653635 /lib/libgcc_s.so.1 
b75ed000-b75ee000 rw-p 00000000 00:00 0 
b75ee000-b772e000 r-xp 00000000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b772e000-b772f000 ---p 00140000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b772f000-b7731000 r--p 00140000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b7731000-b7732000 rw-p 00142000 08:01 3670051 /lib/i686/cmov/libc-2.11.3.so 
b7732000-b7735000 rw-p 00000000 00:00 0 
b7744000-b7747000 rw-p 00000000 00:00 0 
b7747000-b7748000 r-xp 00000000 00:00 0   [vdso] 
b7748000-b7763000 r-xp 00000000 08:01 3654097 /lib/ld-2.11.3.so 
b7763000-b7764000 r--p 0001b000 08:01 3654097 /lib/ld-2.11.3.so 
b7764000-b7765000 rw-p 0001c000 08:01 3654097 /lib/ld-2.11.3.so 
bfd94000-bfda9000 rw-p 00000000 00:00 0   [stack] 
Aborted 

任何想法的实际问题是什么?

+0

您需要[select()](http://linux.die.net/man/2/select)等待输入,关闭事件等。 – Rup

+0

您在每次迭代时都记录(记录)。只有第一个可以成功。 – wildplasser

回答

0

如果发送返回SOCKET_ERROR这意味着关闭套接字或连接丢失

所以只是做

if (send(connected, ctime(&t),30, 0) == SOCKET_ERROR) 
{ 
    // blablabla 
} 
+0

'server.c:80:error:'SOCKET_ERROR'未声明(首次在此函数中使用)'这是编译上述代码时出现的错误。 – binish

+1

@JohnSmith:SOCKET_ERROR没有在linux上定义,所以检查应该是'if(send()== -1)',然后评估错误的'errno'。 – dwalter

+0

对不起,我看到了标签...>< –

3

recv()将尽快在客户端关闭了连接返回0。如果在尝试从套接字获取数据时出现任何错误,将返回-1

参见man page了解更多信息。

所以更改您的代码

int conn_closed = 0; 

    bytes_recieved = recv(connected,recv_data,1024,0); 
    switch (bytes_recieved) { 
    case 0: 
     /* connection closed by client */ 
     close(connected); 
     conn_closed = 1; 
     break; 
    case -1: 
     /* error on socket */ 
     /* TODO: insert error handling code */ 
     if (errno == EAGAIN || errno == EINTR) { 
      /* safe to retry */ 
      conn_closed = 0; 
     } else { 
      close(connected) 
      conn_closed = 1; 
     } 
     break; 
    default: 
     recv_data[bytes_recieved] = '\0'; 
     fputs(recv_data, log); 
     fputs("Time: ", log); 
     time(&t); 
     fputs(ctime(&t), log); 
     conn_closed = 0; 
     break; 
    } 
    /* break out of the loop if conn_closed */ 
    if (conn_closed) 
     break; 

,不要忘记检查的send()的返回值,看是否在发送数据时发生错误。

更新:增加了错误处理代码EAGAINEINTR

月2日更新:

为了完整起见,这里是你应该添加发送

if (send(connected, ctime(&t),30, 0) == -1) 
{ 
    if (errno == EAGAIN || errno == EINTR) { 
     /* retry to send */ 
    } 
    close(connected); 
    break; 
} 
+3

我不认为这个开关真的需要。如果上面的代码将正常工作,那么将会如下: int conn_closed; bytes_recieved = recv(connected,recv_data,1024,0); 如果(bytes_recieved == 0 || bytes_recieved == -1){ // 声明 } –

+0

顺便说一句,你可能还需要关闭出错的插座,不只是断开 – Hasturkun

+0

你应该处理的'的返回值-1',并检查“errno”,因为它可能是您刚刚被信号中断的情况,并且可以安全地重试。这就是我选择switch()的原因。 – dwalter

0

一个小小的建议你的情况,而不是给一个“如果”条件检查连接的代码,在'while'中做同样的事情,而不是'while(1)',给出条件。

相关问题