Linux Socket编程的一些总结

最近写了一些Linux下网络编程的一些程序，做几点总结吧。

先给出客户端后服务器的一些Socket初始化的代码，以后可以直接拿来调用。

客户端Socket初始化代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <time.h>
#include <strings.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <errno.h>
#include <sys/time.h>
#define IP "192.168.1.111"
#define PORT 12346
int init_socket()
{
   struct sockaddr_in server_addr;
   int err;
   int client_socket;
   client_socket=socket(AF_INET,SOCK_STREAM,0);
   if(client_socket<0)
   {
      printf("socket erro\n");
      return ;
   }
   //设置服务器端的地址，端口等
  server_addr.sin_family=AF_INET;
  server_addr.sin_addr.s_addr=inet_addr(IP);//字符串类型转IP类型
  server_addr.sin_port=htons(PORT);
  bzero(&(server_addr.sin_zero),8);
  connect(client_socket,(struct sockaddr *)&server_addr,sizeof(struct sockaddr));//连接服务器
  return client_socket;
}

直接调用上面的函数返回一个套接字的文件描述符，利用它便可以和服务器进行通信。

下面给出服务器socket初始化的代码

#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <errno.h>
#include <openssl/des.h>
#include <sys/time.h>
#include <fcntl.h>
#define PORT 12346
#define BACKLOG 20
int init_server_socket()
{
   int server_socket,server_client_socket;
   struct sockaddr_in server_addr;

   int err;
   //创建套接字
   server_socket=socket(AF_INET,SOCK_STREAM,0);
   if(server_socket<0)
   {
      printf("socket error\n");
      return -1;
   }
   //设置服务器端的地址，端口等
   server_addr.sin_family = AF_INET;
   server_addr.sin_port = htons(PORT);
   server_addr.sin_addr.s_addr = htons(INADDR_ANY);
   bzero(&(server_addr.sin_zero), 8);
   //将创建的套接字绑定到服务器端
   err = bind(server_socket, (struct sockaddr *)&server_addr, sizeof(struct sockaddr));
   if(err<0)
   {
      printf("bind error\n");
      return -1;
   }
   //监听套接字
   err=listen(server_socket,BACKLOG);
   if(err<0)
   {
      printf("listen error\n");
      return -1;
   }
   return server_socket;
}

上面的初始化函数返回了一个套接字文件描述符，我们在服务器程序中，可以写一个循环，当有客户端连接请求的时候，服务器就建立一个线程和服务器进行通信。代码如下所示：

while(1)
    {
        int addrlen=sizeof(struct sockaddr);
        struct sockaddr_in client_addr;
        //如果调用成功，将返回一个新的套接字与客户端通信
        server_client_socket=accept(server_socket,(struct sockaddr*)&client_addr,&addrlen);
        //下面两行是将这个socket连接设置成非阻塞模式
        // int flags = fcntl(server_client_socket, F_GETFL, 0);
        // fcntl(server_client_socket, F_SETFL, flags | O_NONBLOCK);
        printf("%s has connected success\n",inet_ntoa(client_addr.sin_addr));
        //在这里写创建线程程序
        。。。。。。
        usleep(100);
    }

要点总结：

1、在服务器端程序要判断recv()函数的返回值，如果对方发送的数据比较的则需采用循环接收的方法，比如对方一次发了4096字节的数据，一次rrecv()可能接收不了，则需要多次接收，直到接收的总数等于4096字节，看下面的代码你就会明白

#define MAXBUFFER_SIZE 4096
int total_reieved =0;
char recvbuffer[MAXBUFFER_SIZE];
while(1)
       {
          if(MAXBUFFER_SIZE==total_reieved)
              break;
          int size=recv(server_client_socket,recvbuffer+total_reieved,MAXBUFFER_SIZE-total_reieved,0);
          total_reieved+=size;
      }

2、如果发送方的程序关闭了，在接收方就会不断recv()并返回0，则我们就要关闭这个连接，防止接收方陷入死循环

3、拔网线拔掉重新插上，socket可以自动恢复连接

4、默认的连接时阻塞模式的，如果要将套接字通信设置成非阻塞模式我们可以参考下面的代码（对套接字文件描述符设置），非阻塞模式下如果读不到数据则返回-1，网络断开也是返回-1

int flags = fcntl(server_client_socket, F_GETFL, 0);
fcntl(server_client_socket, F_SETFL, flags | O_NONBLOCK);

5、假如客户端程序一直发送数据，服务器程序一直接受数据，这时服务器程序关闭，则客户端程序也会关闭，这时因为服务器close一个连接时，若客户端接这发数据，系统会产生一个SIGPIPE信号，告诉进程连接已经断开了，而处理这个信号的默认动作是使进程退出，如果不想进程退出，可以把SIGPIPE设为SIG_IGN。调用函数signal(SIGPIPE,SIG_IGN);

大概就总结了这几点，欢迎批评和指正。