C/C ?套接字通信類的封裝

在掌握了基于 TCP 的套接字通信流程之后，為了方便使用，提高編碼效率，可以對通信操作進(jìn)行封裝，本著有淺入深的原則，先基于 C 語言進(jìn)行面向過程的函數(shù)封裝，然后再基于 C 進(jìn)行面向?qū)ο蟮念惙庋b。

1. 基于 C 語言的封裝

基于 TCP 的套接字通信分為兩部分：服務(wù)器端通信和客戶端通信。我們只要掌握了通信流程，封裝出對應(yīng)的功能函數(shù)也就不在話下了，先來回顧一下通信流程：

服務(wù)器端

• 創(chuàng)建用于監(jiān)聽的套接字
• 將用于監(jiān)聽的套接字和本地的 IP 以及端口進(jìn)行綁定
• 啟動(dòng)監(jiān)聽
• 等待并接受新的客戶端連接，連接建立得到用于通信的套接字和客戶端的 IP、端口信息
• 使用得到的通信的套接字和客戶端通信（接收和發(fā)送數(shù)據(jù)）
• 通信結(jié)束，關(guān)閉套接字（監(jiān)聽 通信）

客戶端

• 創(chuàng)建用于通信的套接字
• 使用服務(wù)器端綁定的 IP 和端口連接服務(wù)器
• 使用通信的套接字和服務(wù)器通信（發(fā)送和接收數(shù)據(jù)）
• 通信結(jié)束，關(guān)閉套接字（通信）

1.1 函數(shù)聲明

通過通信流程可以看出服務(wù)器和客戶端有些操作步驟是相同的，因此封裝的功能函數(shù)是可以共用的，相關(guān)的通信函數(shù)聲明如下：

///////////////////////////////////////////////////?
////////////////////?服務(wù)器?///////////////////////
///////////////////////////////////////////////////
int?bindSocket(int?lfd,?unsigned?short?port);
int?setListen(int?lfd);
int?acceptConn(int?lfd,?struct?sockaddr_in?*addr);

///////////////////////////////////////////////////?
////////////////////?客戶端?///////////////////////
///////////////////////////////////////////////////
int?connectToHost(int?fd,?const?char*?ip,?unsigned?short?port);

///////////////////////////////////////////////////?
/////////////////////?共用?////////////////////////
///////////////////////////////////////////////////
int?createSocket();
int?sendMsg(int?fd,?const?char*?msg);
int?recvMsg(int?fd,?char*?msg,?int?size);
int?closeSocket(int?fd);
int?readn(int?fd,?char*?buf,?int?size);
int?writen(int?fd,?const?char*?msg,?int?size);
關(guān)于函數(shù) readn() 和 writen() 的作用請參考TCP數(shù)據(jù)粘包的處理

1.2 函數(shù)定義

//?創(chuàng)建監(jiān)套接字
int?createSocket()
{
????int?fd?=?socket(AF_INET,?SOCK_STREAM,?0);
????if(fd?==?-1)
????{
????????perror("socket");
????????return?-1;
????}
????printf("套接字創(chuàng)建成功,?fd=%d\n",?fd);
????return?fd;
}

//?綁定本地的IP和端口
int?bindSocket(int?lfd,?unsigned?short?port)
{
????struct?sockaddr_in?saddr;
????saddr.sin_family?=?AF_INET;
????saddr.sin_port?=?htons(port);
????saddr.sin_addr.s_addr?=?INADDR_ANY;??//?0?=?0.0.0.0
????int?ret?=?bind(lfd,?(struct?sockaddr*)