从C到C++（七）本文已参与 ⌈新人创作礼⌋ 活动，一起开启掘金创作之路。八、socket封装 C对socket的封装

本文已参与 ⌈新人创作礼⌋ 活动，一起开启掘金创作之路。

八、socket封装

C对socket的封装：

//客户端
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
 
// TCP客户端连服务端的函数，serverip-服务端ip，port通信端口
int connecttoserver(const char *serverip,const int port);
 
int main()
{
  // 向服务器发起连接请求
  int sockfd=connecttoserver("118.89.50.198",5051);
  if (sockfd<=0) { printf("连接服务器失败，程序退出。\n"); return -1; }
 
  char strbuffer[1024];
 
  // 与服务端通信，发送一个报文后等待回复，然后再发下一个报文。
  for (int ii=0;ii<10;ii++)
  {
    memset(strbuffer,0,sizeof(strbuffer));
    sprintf(strbuffer,"这是第%d个超级女生，编号%03d。",ii+1,ii+1);
    if (send(sockfd,strbuffer,strlen(strbuffer),0)<=0) break;
    printf("发送：%s\n",strbuffer);
 
    memset(strbuffer,0,sizeof(strbuffer));
    if (recv(sockfd,strbuffer,sizeof(strbuffer),0)<=0) break;
    printf("接收：%s\n",strbuffer);
  }
 
  close(sockfd);
}
 
// TCP客户端连服务端的函数，serverip-服务端ip，port通信端口
// 返回值：成功返回已连接socket，失败返回-1。
int connecttoserver(const char *serverip,const int port)
{
  int sockfd = socket(AF_INET,SOCK_STREAM,0); // 创建客户端的socket
 
  struct hostent* h; // ip地址信息的数据结构
  if ( (h = gethostbyname(serverip)) == 0 )
  { perror("gethostbyname"); close(sockfd); return -1; }
 
  // 把服务器的地址和端口转换为数据结构
  struct sockaddr_in servaddr;
  memset(&servaddr,0,sizeof(servaddr));
  servaddr.sin_family = AF_INET;
  servaddr.sin_port = htons(port);
  memcpy(&servaddr.sin_addr,h->h_addr,h->h_length);
 
  // 向服务器发起连接请求
  if (connect(sockfd, (struct sockaddr *)&servaddr,sizeof(servaddr)) != 0)
  { perror("connect"); close(sockfd); return -1; }
 
  return sockfd;
}

//服务端
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
 
// 初始化服务端的socket，port为通信端口
// 返回值：成功返回初始化的socket，失败返回-1。
int initserver(int port);
 
int main()
{
  int listenfd=initserver(5051);  // 服务端用于监听的socket
  if (listenfd<=0) { printf("服务端初始化失败，程序退出。\n"); return -1; }
 
  // 接受客户端的连接。
  int  clientfd;                  // 客户端的socket
  if ( (clientfd=accept(listenfd,0,0)) <= 0) { printf("服务端accept失败，程序退出。\n"); return -1; }
 
  printf("客户端已连接。\n");
 
  // 与客户端通信，接收客户端发过来的报文后，回复ok。
  char strbuffer[1024];
 
  while (1)
  {
    memset(strbuffer,0,sizeof(strbuffer));
    if (recv(clientfd,strbuffer,sizeof(strbuffer),0)<=0) break;
    printf("接收：%s\n",strbuffer);
 
    strcpy(strbuffer,"ok");
    if (send(clientfd,strbuffer,strlen(strbuffer),0)<=0) break;
    printf("发送：%s\n",strbuffer);
  }
 
  printf("客户端已断开连接。\n");
 
  close(clientfd); close(listenfd); // 关闭socket
}
 
// 初始化服务端的socket，port为通信端口
// 返回值：成功返回初始化的socket，失败返回-1。
int initserver(int port)
{
  int listenfd = socket(AF_INET,SOCK_STREAM,0);  // 创建服务端的socket
 
  // 把服务端用于通信的地址和端口绑定到socket上
  struct sockaddr_in servaddr;    // 服务端地址信息的数据结构
  memset(&servaddr,0,sizeof(servaddr));
  servaddr.sin_family = AF_INET;  // 协议族，在socket编程中只能是AF_INET
  servaddr.sin_addr.s_addr = htonl(INADDR_ANY);  // 本主机的任意ip地址
  servaddr.sin_port = htons(port);  // 绑定通信端口
  if (bind(listenfd,(struct sockaddr *)&servaddr,sizeof(servaddr)) != 0 )
  { perror("bind"); close(listenfd); return -1; }
 
  // 把socket设置为监听模式
  if (listen(listenfd,5) != 0 ) { perror("listen"); close(listenfd); return -1; }
 
  return listenfd;
}