tcp_server.hpp
#pragma once
#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
//netstat -nltp只查看listen状态的tcp服务的所有进程并把可以显示成数字的都显示成数字
using namespace std;
class Tcpserver
{
public:
Tcpserver(uint16_t port) : _port(port), _isrunning(false)
{
}
void Init()
{
// 1.创建套接字 底层创建tcp的结构体,管理缓冲区
_listensock = socket(AF_INET, SOCK_STREAM, 0); // 创建套接字时候底层发生了什么。为啥要创建
if (_listensock < 0)
{
cout << "set socket error" << endl;
exit(1);
}
// 填充本地网络信息,并且bind
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;
// 将写的本地网络信息设置到内核缓冲区
if (bind(_listensock, (struct sockaddr *)&local, sizeof(local)) != 0)
{
cout << "bind error " << endl;
exit(1);
}
// 先建立链接,client建立连接,server等待连接,三次握手
if (listen(_listensock, 5) != 0) // 第二个参数是全连接队列,忘了,一会再看看
{
cout << "listen error" << endl;
exit(1);
}
}
void service(int sockfd)
{
char buffer[1024];
while (true)
{
//写的时候底层发生了啥
ssize_t n = read(sockfd, buffer, sizeof(buffer) - 1);
if (n > 0)
{
buffer[n] = 0; // 结尾加上\0
cout << "client say" << buffer << endl;
//把消息写过去
string echo_string="server say#";
echo_string+=buffer;
write(sockfd,echo_string.c_str(),echo_string.size());
}
else if (n == 0)
{
// 0表示读到了文件结尾,对端关闭链接
cout << "client quit" << endl;
break;
}
else
{
cout << "client read error" << endl;
break;
}
}
}
void Start()
{
_isrunning = true;
while (_isrunning)
{
cout<<"qqqqqqqqqqqqqqqqqqqqqqqw";
struct sockaddr_in peer;
socklen_t len = sizeof(peer);
// 获取连接
cout<<"wwwwwwwwwwwwwwwwwwwwww";
int sockfd = accept(_listensock, (struct sockaddr *)&peer, &len); // 获取client的ip和port
// 返回一个新的文件描述符,饭店里提供服务的李四
cout<<"get a newsockfd"<<sockfd<<endl;
if (sockfd < 0)
continue;
cout << "accept success" << endl;
// 提供服务
service(sockfd); // 这个文件描述符干啥用的
close(sockfd);
}
}
~Tcpserver()
{
}
private:
uint16_t _port;
int _listensock; // 拉客用的套接字,张三
bool _isrunning;
};
tcp_client.cc
#include <iostream>
#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
using namespace std;
int main(int argc,char*argv[])
{
if(argc!=3)
{
cout<<"./tcp_server server_ip server_port"<<endl;
return 1;
}
string serverip=argv[1];
uint16_t serverport=stoi(argv[2]);
//创建套接字 //底层发生了啥???
int sockfd=socket(AF_INET,SOCK_STREAM,0);
if(sockfd<0)
{
cout<<"sock error"<<endl;
return 1;
}
//bind,不需要显示的绑定,显示bind就把ip,port邦死了,不能改了
//三次握手的时候就会绑定
struct sockaddr_in server;
memset(&server,0,sizeof(server));
//刚才这里没写完全,为啥没提示
server.sin_family=AF_INET;
server.sin_port=htons(serverport);
inet_pton(AF_INET,serverip.c_str(),&server.sin_addr);//p:proccess本地
int n=connect(sockfd,(struct sockaddr*)&server,sizeof(server));
if(n<0)
{
cout<<"connect error"<<endl;
return 1;
}
while(true)
{
string buffer;
cout<<"plase inter:"<<endl;
getline(cin,buffer);
ssize_t n=write(sockfd,buffer.c_str(),buffer.size());
if(n>0)
{
//发出去,再读回来
char getbuffer[1024];
ssize_t m=read(sockfd,getbuffer,sizeof(getbuffer)-1);
if(m>0)
{
getbuffer[m]=0;
cout<<"get amessage:"<<buffer<<endl;
}
else
break;
}
else
break;
}
close(sockfd);
return 0;
}
main.cc
#include"Tcpserver.hpp"
#include<memory>//智能指针
using namespace std;
int main(int argc,char*args[])
{
if(argc!=2)
{
cout<<"putin error:Tcp_server port"<<endl;
return 1;
}
uint16_t port=stoi(args[1]);//转端口号
std::unique_ptr<Tcpserver> tsvr=make_unique<Tcpserver>(port);
tsvr->Init();
tsvr->Start();
return 0;
}