手搓TCP服务器实现基础IO
在有了UDP服务器的基础,搓一个TCP服务器就简单很多了,套接字、绑定,结构体协议家族都是套路,我们主要看二者的区别在代码上的体现。其中TCP是有连接的,也就是说当有客户端访问服务器想发送数据包时,需先和服务器进行连接,然后进行相关操作,而UDP的无连接则不需要,直接向服务器发送数据即可。建议先看一下UDP服务器的代码再看此文会更加清晰。、
套接字与UDP服务器代码https://blog.csdn.net/czt230610/article/details/149817538?fromshare=blogdetail&sharetype=blogdetail&sharerId=149817538&sharerefer=PC&sharesource=czt230610&sharefrom=from_link上面说过 socket的套路不变,但其通信方式我们需要改变一下,在UDP中是SOCK_DGRAM,而TCP是SOCK_STREAM。
除此之外,TCP需要时刻观察是否有客户端访问,就像饭店老板时刻等待顾客一样,因此我们需要给绑定的套接字加上监听功能。
第一个参数就是我们需要监听的套接字文件描述符,第二个参数是排队的容量(即如果有多个客户端访问最多支持排多少个,这里我们定义了宏为8)。
除此之外,我们也要进行“揽客”即告诉客户端我这里可以连接了。
第一个参数是我们负责“揽客”的套接字(最初绑定的),就相当于饭店外面揽客的人,他们并不在里面吃饭,但是会告诉别人让他们来吃饭,因此这个描述符不做IO工作,只是单纯的“揽客”。后两个是输出型参数,代表哪个客户端来访问了。
可是,他返回值也是一个文件描述符。这个和前面的功能一样吗?
这个返回的文件描述符,其实才是真正做IO工作的。
图中的sockfd才是我们真正做IO的,我们类内定义的是_listensockfd。
接下来就是IO工作,我们单独写一个函数来处理任务,由于TCP是面向字节流的,IO更像文件,因此我们用文件的读写接口。
void HandlerRequest(int sockfd){while(true){char inbuffer[4096];ssize_t n=::read(sockfd,inbuffer,sizeof(inbuffer)-1);if(n>0){inbuffer[n]=0;std::string echo_str="server echo";echo_str+=inbuffer;::write(sockfd,echo_str.c_str(),echo_str.size());}else if (n=0)//客户端退出{std::cout<<"client exit"<<std::endl;break; }else{break;}}::close(sockfd);}
这个函数能够基础的进行IO但是只支持一个客户端连接,先来看客户端的编写。
在进行套接字的创建后,我们遇到了新的问题,如何建立连接?因为TCP是有连接的,需要连接才能进行read write操作。因此我们介绍一个新接口。
即把创建的套接字与目标服务器的ip+port连接。(客户端不需要显示的bind)
剩下的IO工作和UDP就类似了,只是接口换一下。
以上即可满足单个客户端进行不断重连且能进行IO的服务器操作了。
完整代码:
服务器hpp
#pragma once#include <iostream>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <cstring>
#include <unistd.h>
#include <sys/wait.h>
#include <signal.h>
static const uint16_t gport=8080;#define BACKLOG 8 //排队容量
class TcpServer
{
public:TcpServer(int port=gport):_port(port),_isrunning(false){}void InitServer(){_listensockfd = socket(AF_INET, SOCK_STREAM, 0);if (_listensockfd < 0){std::cout << "socket falied" << std::endl;exit(1);}std::cout<<"socket success"<<std::endl;struct sockaddr_in local;memset(&local,0,sizeof(local));local.sin_family=AF_INET;local.sin_port=htons(_port);local.sin_addr.s_addr=INADDR_ANY;int n=::bind(_listensockfd,(struct sockaddr*)&local,sizeof(local));if(n<0){std::cout<<"bind error"<<std::endl;}std::cout<<"bind success"<<std::endl;//tcp随时等待被连接//需要将tcp的socket设置为监听状态n=::listen(_listensockfd,BACKLOG);if(n<0){std::cout<<"listen failed"<<std::endl;}std::cout<<"listen success"<<std::endl;}void HandlerRequest(int sockfd){while(true){char inbuffer[4096];ssize_t n=::read(sockfd,inbuffer,sizeof(inbuffer)-1);if(n>0){inbuffer[n]=0;std::string echo_str="server echo";echo_str+=inbuffer;::write(sockfd,echo_str.c_str(),echo_str.size());}else if (n=0)//客户端退出{std::cout<<"client exit"<<std::endl;break; }else{break;}}::close(sockfd);}void Start(){_isrunning =true;while(_isrunning){//获取新连接struct sockaddr_in peer;socklen_t peerlen;//获取客户端信息int sockfd=::accept(_listensockfd,(struct sockaddr*)&peer,&peerlen);if(sockfd<0){//揽客失败std::cout<<"accept failed"<<std::endl;continue;}//获取成功std::cout<<"success : "<<sockfd<<std::endl;//EchoHandlerRequest(sockfd);}}void Stop(){_isrunning=false;}~TcpServer(){}private:int _listensockfd;uint16_t _port;bool _isrunning;
};
服务端.cc
#include "TcpServer.hpp"
#include <memory>int main()
{std::unique_ptr<TcpServer> tsvr=std::make_unique<TcpServer>();tsvr->InitServer();tsvr->Start();return 0;
}
客户端.cc
#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <cstring>
#include <unistd.h>// ./client_tcp ip port
int main(int argc, char* argv[])
{if(argc!= 3){std::cout << "Usage:./client_tcp ip port" << std::endl;return 1;}std::string server_ip = argv[1];int server_port = std::stoi(argv[2]);int sockfd = socket(AF_INET, SOCK_STREAM, 0);if(sockfd < 0){std::cout << "Error in creating socket" << std::endl;return 2;}//服务器的socket已传入struct sockaddr_in server_addr;memset(&server_addr, 0, sizeof(server_addr));server_addr.sin_family = AF_INET;server_addr.sin_port = htons(server_port);server_addr.sin_addr.s_addr = inet_addr(server_ip.c_str());int n=::connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr));if(n < 0){std::cout << "Error in connecting to server" << std::endl;return 3;}std::string message;while(true){std::cout << "Enter message to send: ";std::getline(std::cin, message);int n=::write(sockfd, message.c_str(), message.size());char buffer[1024];if(n>0){int m=::read(sockfd, buffer, sizeof(buffer));if(m>0){buffer[m]=0;std::cout << "Received message: " << buffer << std::endl;}}}::close(sockfd);return 0;
}