項目完整版在：
一、socket模塊：套接字模塊
二、提供的功能
Socket模塊是對套接字操作封裝的一個模塊，主要實現的socket的各項操作。
socket 模塊：套接字的功能
創(chuàng)建套接字
綁定地址信息
開始監(jiān)聽
向服務器發(fā)起連接
獲取新連接
接受數據
發(fā)送數據
關閉套接字
創(chuàng)建一個監(jiān)聽鏈接
創(chuàng)建一個客戶端連接
設置套接字選項——開啟地址端口重用！
設置套接字阻塞屬性——設置為非阻塞！
三、實現思想
（一）功能
對socket套接字的操作進行封裝。文章來源地址http://www.zghlxwxcb.cn/news/detail-728986.html文章來源：http://www.zghlxwxcb.cn/news/detail-728986.html
（二）意義
對socket套接字的操作進行封裝。
（三）功能設計
- 創(chuàng)建套接字
- 綁定地址信息
- 開始監(jiān)聽
- 向服務器發(fā)起連接
- 獲取新連接
- 接受數據
- 發(fā)送數據
- 關閉套接字
- 創(chuàng)建一個監(jiān)聽鏈接
- 創(chuàng)建一個客戶端連接
四、代碼
#define MAX_LISTEN 1024
class Socket {
private:
int _sockfd;
public:
Socket() :_sockfd(-1) {}
Socket(int fd) : _sockfd(fd) {}
~Socket() {Close(); }
int fd() {return _sockfd;}
// 1.創(chuàng)建套接字
bool Create() {
//int socket (int domain,int type,int protocol);
_sockfd = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
if (_sockfd < 0) {
ERR_LOG("CREATE SOCKET FAILED !!");
return false;
}
return true;
}
// 2.綁定地址信息
bool Bind(const std::string &ip,uint16_t port) {
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = inet_addr(ip.c_str());
socklen_t len = sizeof(struct sockaddr_in);
// int bind(int sockfd,struct sockaddr * addr,socklen_t len);
int ret = bind(_sockfd,(struct sockaddr*)&addr,len);
if (ret < 0) {
ERR_LOG("BIND ADDRESS FAILED!!!!");
return false;
}
return true;
}
// 3.開始監(jiān)聽
bool Listen(int backlog = MAX_LISTEN) {
// int listen(int backlog)
int ret = listen(_sockfd,backlog);
if (ret < 0) {
ERR_LOG("SOCKET LISTEN FAILED!!");
return false;
}
return true;
}
// 4. 向服務器發(fā)起連接
bool Connect(const std:: string& ip,uint16_t port) {
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = inet_addr(ip.c_str());
socklen_t len = sizeof(struct sockaddr_in);
// int bind(int sockfd,struct sockaddr * addr,socklen_t len);
int ret = connect(_sockfd,(struct sockaddr*)&addr,len);
if (ret < 0) {
ERR_LOG("CONNECT ADDRESS FAILED!!!!");
return false;
}
return true;
}
// 5. 獲取新連接
int Accept() {
// int accept(int sockfd,struct sockaddr* addr,socklen_t *len) /
int newfd = accept(_sockfd,NULL,NULL);
if (newfd < 0) {
ERR_LOG("SOCKET ACCEPT FAILED!!!!");
return false;
}
return newfd;
}
ssize_t Recv(void *buf,size_t len,int flag = 0){
// 6.接收數據
//有符號長整型
//ssize_t Recv(int sockfd,void* buf,size_t len,int flag);
ssize_t ret = recv(_sockfd,buf,len,flag);
if (ret <= 0) {
// EAGAIN 當前socket的接收緩沖區(qū)沒有數據來，在非阻塞二點情況下才會有這個錯誤
// ENTER 當前socket的阻塞等待，被信號打斷了
if (errno == EAGAIN || errno == EINTR) {
return 0; // 沒收到數據
}
ERR_LOG("SOCKET RECV FAILED!!");
return -1; // 出錯
}
return ret;
}
ssize_t nonBlockRecv(void* buf,size_t len) {
return Recv(buf,len,MSG_DONTWAIT); // MSG_DONTWAIT 表示當前接受為非阻塞
}
// 7.發(fā)送數據
ssize_t Send(const void* buf,size_t len,int flag = 0) {
// ssize_t send(int sockfd,void *data,size_t len,int flag)
ssize_t ret = send(_sockfd,buf,len,flag);
if (ret < 0) {
ERR_LOG("SOCKET SEND FAILED!!");
return -1; // 出錯
}
return ret; // 實際發(fā)送數據長度??！
}
ssize_t nonBlockSend(void* buf,size_t len) {
return Send(buf,len,MSG_DONTWAIT); // MSG_DONTWAIT 表示當前接受為非阻塞
}
// 8.關閉套接字
void Close() {
if (_sockfd != -1) {
close(_sockfd);
_sockfd = -1;
}
}
// 9.創(chuàng)建一個服務端鏈接
bool createServer(uint16_t port, const std::string &ip = "0.0.0.0", bool block_flag = false) {
// 1.創(chuàng)建套接字 2. 綁定地址 3.開始監(jiān)聽 4.設置非阻塞 5.啟動地址重用
if (Create() == false) return false;
if (Bind(ip,port) == false) return false;
if (Listen() == false) return false;
if (block_flag) NonBlock();
ReuseAddress();
return true;
}
// 10.創(chuàng)建一個客戶端鏈接
bool createClient(uint16_t port, const std::string &ip) {
if (Create() == false) return false;
if (Connect(ip,port) == false) return false;
return true;
}
// 11. 設置套接字選項——開啟地址端口重用！
void ReuseAddress() {
// int setsockopt(int fd,int leve,int optname,void *val,int vallen)
int val = 1;
setsockopt(_sockfd, SOL_SOCKET, SO_REUSEADDR, (void*)&val, sizeof(int));
val = 1;
setsockopt(_sockfd, SOL_SOCKET, SO_REUSEPORT, (void*)&val, sizeof(int));
}
// 12. 設置套接字阻塞屬性——設置為非阻塞！
void NonBlock() {
int flag = fcntl(_sockfd, F_GETFL, 0);
fcntl(_sockfd, F_SETFL, flag | O_NONBLOCK);
}
};
五、測試
（一）tcp_cli.cc
#include "../source/server.hpp"
int main() {
Socket cli_sock;
cli_sock.createClient(8500,"127.0.0.1");
std::string str = "nihao";
cli_sock.Send(str.c_str(),str.size());
char buf[1024] = {0};
cli_sock.Recv(buf,1023);
DBG_LOG("%s",buf);
return 0;
}
（二）tcp_srv.cc
#include "../source/server.hpp"
int main() {
Socket lst_sock;
bool ret = lst_sock.createServer(8500);
while (1) {
int newfd = lst_sock.Accept();
if (newfd < 0) {
continue;
}
Socket cli_sock(newfd);
char buf[1024] = {0};
int ret = cli_sock.Recv(buf,1023);
if(ret < 0) {
cli_sock.Close();
}
cli_sock.Send(buf,ret);
cli_sock.Close();
}
lst_sock.Close();
return 0;
}
（三）makefile
all:client server
client:tcp_cli.cc
g++ -std=c++11 $^ -o $@
server:tcp_srv.cc
g++ -std=c++11 $^ -o $@
到了這里，關于1.5.C++項目：仿muduo庫實現并發(fā)服務器之socket模塊的設計的文章就介紹完了。如果您還想了解更多內容，請在右上角搜索TOY模板網以前的文章或繼續(xù)瀏覽下面的相關文章，希望大家以后多多支持TOY模板網！
