Socket编程——TCP
文章目录
-
- 一、TCP编程基础
- TCP socket API
- socket
- 二、EchoServer
- 对于服务端
- 对于客户端
- server.hpp
- Server.cc
- Client.cc
- CommandExec.hpp
一、TCP编程基础
TCP协议和UDP协议都属于网络通信协议，TCP协议是面向字节流的，UDP协议是面向数据报，这个特点后面会详谈的。对于现在来说UDP和TCP的区别为：
UDP协议不需要连接，即报文一来就立刻进行转发
TCP协议则需要建立连接，只有双方准备好才能进行报文的转发
TCP socket API
socket
NAME
socket - create an endpoint for communication
SYNOPSIS
#include /* See NOTES */
#include
- socket()打开一个网络通讯端口,如果成功的话,就像 open()一样返回一个文件描述符;
- 应用程序可以像读写文件一样用 read/write 在网络上收发数据;
- 如果 socket()调用出错则返回-1;
- 对于 IPv4, family 参数指定为 AF_INET;
- 对于 TCP 协议,type 参数指定为 SOCK_STREAM, 表示面向流的传输协议
- protocol 参数的介绍从略,指定为 0 即可。
bind
NAME
bind - bind a name to a socket
SYNOPSIS
#include /* See NOTES */
#include
- 服务器程序所监听的网络地址和端口号通常是固定不变的,客户端程序得知服务器程序的地址和端口号后就可以向服务器发起连接; 服务器需要调用 bind 绑定一个固定的网络地址和端口号;
- bind()成功返回 0,失败返回-1。
- bind()的作用是将参数 sockfd 和 myaddr 绑定在一起, 使 sockfd 这个用于网络通讯的文件描述符监听 myaddr 所描述的地址和端口号;
- 前面讲过,struct sockaddr *是一个通用指针类型,myaddr 参数实际上可以接受多种协议的 sockaddr 结构体,而它们的长度各不相同,所以需要第三个参数 addrlen指定结构体的长度;
listen
NAME
listen - listen for connections on a socket
SYNOPSIS
#include /* See NOTES */
#include
- listen()声明 sockfd 处于监听状态, 并且最多允许有 backlog 个客户端处于连接等待状态, 如果接收到更多的连接请求就忽略, 这里设置不会太大(一般是 5);
- listen()成功返回 0,失败返回-1
accept
NAME
accept, accept4 - accept a connection on a socket
SYNOPSIS
#include /* See NOTES */
#include
- 三次握手完成后, 服务器调用 accept()接受连接;
- 如果服务器调用 accept()时还没有客户端的连接请求,就阻塞等待直到有客户端连接上来;
- addr 是一个传出参数,accept()返回时传出客户端的地址和端口号;
- 如果给 addr 参数传 NULL,表示不关心客户端的地址;
- addrlen 参数是一个传入传出参数(value-result argument), 传入的是调用者提供的, 缓冲区 addr 的长度以避免缓冲区溢出问题, 传出的是客户端地址结构体的实际长度(有可能没有占满调用者提供的缓冲区);
connect
NAME
connect - initiate a connection on a socket
SYNOPSIS
#include /* See NOTES */
#include
- 客户端需要调用 connect()连接服务器;
- connect 和 bind 的参数形式一致, 区别在于 bind 的参数是自己的地址, 而connect 的参数是对方的地址;
- connect()成功返回 0,出错返回-1;
对于accept和listen的理解：
listen可以理解为在景区餐馆旁边拉客的人，他的任务只是把你拉进餐馆，对于后续的服务并不做关心，在你进入餐馆后，继续进行拉客
accept可以理解为在你进入这个餐馆后进行招待的人，也就是由他来对你进行服务
二、EchoServer
目标：
对于服务端
建立套接字
绑定套接字
监听等待
获取连接
提供服务
对于客户端
- 获取服务端IP和端口号
- 建立套接字
- 建立连接
- 发送消息
server.hpp
#pragma once
#include ::getInstance()->Equeue([this, sockfd]()
// { this->HandlerRequest(sockfd); });
}
}
void Stop()
{
_isrunning = false;
}
~TcpServer()
{
}
private:
uint16_t _port;
int _listensockfd;
bool _isrunning;
handler_t _handler;
};
Server.cc
#include "TcpServer.hpp"
#include "CommandExec.hpp"
#include
using namespace LogModule;
int main()
{
ENABLE_CONSOLE_LOG();
Command cmd;
std::unique_ptr tsvr = std::make_unique([&cmd](std::string cmdstr){
return cmd.Execute(cmdstr);
});
tsvr->InitServer();
tsvr->Start();
return 0;
}
Client.cc
#include
#include
#include
#include
#include
#include
#include
#include
// ./client_tcp server_ip server_port
int main(int argc, char *argv[])
{
if (argc != 3)
{
std::cout << "Usage:./client_tcp server_ip server_port" << std::endl;
return 1;
}
std::string server_ip = argv[1]; // "192.168.1.1"
int server_port = std::stoi(argv[2]);
int sockfd = ::socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0)
{
std::cout << "create socket failed" << std::endl;
return 2;
}
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());
// client 不需要显示的进行bind, tcp是面向连接的, connect 底层会自动进行bind
int n = ::connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr));
if(n < 0)
{
std::cout << "connect failed" << std::endl;
return 3;
}
// echo client
std::string message;
while(true)
{
char inbuffer[1024];
std::cout << "input message: ";
std::getline(std::cin, message);
n = ::write(sockfd, message.c_str(), message.size());
if(n > 0)
{
int m = ::read(sockfd, inbuffer, sizeof(inbuffer));
if(m > 0)
{
inbuffer[m] = 0;
std::cout << inbuffer << std::endl;
}
else
break;
}
else
break;
}
::close(sockfd);
return 0;
}
CommandExec.hpp
#pragma once
#include
#include
#include
#include
const int line_size = 1024;
class Command
{
public:
Command()
{
_white_list.insert("ls");
_white_list.insert("pwd");
_white_list.insert("ls -l");
_white_list.insert("ll");
_white_list.insert("touch");
_white_list.insert("who");
_white_list.insert("whoami");
}
bool SafeCheck(const std::string &cmdstr)
{
auto iter = _white_list.find(cmdstr);
return iter == _white_list.end() ? false : true;
}
// 给你一个命令字符串"ls -l", 让你执行，执行完，把结果返回
std::string Execute(std::string cmdstr)
{
// 1. pipe
// 2. fork + dup2(pipe[1], 1) + exec* , 执行结果给父进程，pipe[0]
// 3. return
// FILE *popen(const char *command, const char *type);
// int pclose(FILE * stream);
if (!SafeCheck(cmdstr))
{
return std::string(cmdstr + " 不支持");
}
FILE *fp = ::popen(cmdstr.c_str(), "r");
if (nullptr == fp)
{
return std::string("Failed");
}
char buffer[line_size];
std::string result;
while (true)
{
char *ret = ::fgets(buffer, sizeof(buffer), fp);
if (!ret)
break;
result += ret;
}
pclose(fp);
return result.empty() ? std::string("Done") : result;
}
private:
std::set _white_list;
};