You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
218 lines
5.8 KiB
218 lines
5.8 KiB
#include <iostream>
|
|
#include <event2/event.h>
|
|
#include <event2/thread.h>
|
|
#include <event2/listener.h>
|
|
#include <event2/bufferevent.h>
|
|
|
|
|
|
#ifdef _WIN32
|
|
#include <string>
|
|
#else
|
|
#include <signal.h>
|
|
#include <string.h>
|
|
#endif // !_WIN32
|
|
|
|
using namespace std;
|
|
|
|
static string recvstr = "";
|
|
static int recvCount = 0;
|
|
static int sendCount = 0;
|
|
|
|
void read_cb(bufferevent * bev, void *arg) {
|
|
cout << "[R]" << endl;
|
|
|
|
char data[1024] = { 0 };
|
|
int len = bufferevent_read(bev, data, sizeof(data) - 1);
|
|
if (len <= 0)return;
|
|
recvstr += data;
|
|
recvCount += len;
|
|
|
|
/* if(strstr(data,"quit")!= NULL){
|
|
cout << "quit" << endl;
|
|
bufferevent_free(bev); // BEV_OPT_CLOSE_ON_FREE 设置了这个才会关闭
|
|
} */
|
|
//发送数据 写入到输出缓冲
|
|
bufferevent_write(bev, "ok", 3);
|
|
}
|
|
|
|
void write_cb(bufferevent * bev, void *arg) {
|
|
cout << "[W]" << endl;
|
|
}
|
|
|
|
//错误 超时 连接断开会进入
|
|
void event_cb(struct bufferevent *bev, short what, void *ctx) {
|
|
cout << "[CB]" << endl;
|
|
|
|
//读取超时事件发生后，数据读取停止
|
|
if (what & BEV_EVENT_TIMEOUT && what & BEV_EVENT_READING) {
|
|
char data[1024] = { 0 };
|
|
int len = bufferevent_read(bev, data, sizeof(data) - 1);
|
|
if (len > 0) {
|
|
recvCount += len;
|
|
recvstr += data;
|
|
}
|
|
cout << "read timeout " << endl;
|
|
|
|
//bufferevent_enable(bev,EV_READ); //可以重新设置
|
|
bufferevent_free(bev);
|
|
cout << recvstr << endl;
|
|
cout << "recvCOunt" << recvCount << " sendCount" << sendCount << endl;
|
|
|
|
}
|
|
else if (what & BEV_EVENT_ERROR) {
|
|
bufferevent_free(bev);
|
|
}
|
|
else {
|
|
cout << "other " << endl;
|
|
}
|
|
}
|
|
|
|
void listen_cb(struct evconnlistener *ev, evutil_socket_t s,
|
|
struct sockaddr *insin, int socklen, void *arg)
|
|
{
|
|
cout << "listen_cb" << endl;
|
|
|
|
struct sockaddr_in *sin = (struct sockaddr_in *)&insin;
|
|
int port = ntohs(sin->sin_port);
|
|
|
|
char ip[16] = { 0 };
|
|
evutil_inet_ntop(AF_INET, &sin->sin_addr, ip, sizeof(ip) - 1);
|
|
cout << "client ip is " << ip << "port is " << port << endl;
|
|
|
|
|
|
event_base* base = (event_base*)arg;
|
|
//创建bufferevent上下文 BEV_OPT_CLOSE_ON_FREE清理bufferevent时关闭socket
|
|
bufferevent * bev = bufferevent_socket_new(base, s, BEV_OPT_CLOSE_ON_FREE);
|
|
bufferevent_enable(bev, EV_READ | EV_WRITE);
|
|
|
|
//设置水位
|
|
//读取水位
|
|
bufferevent_setwatermark(bev, EV_READ,
|
|
10, //低水位0就是无限制 默认0 有10个数据才会回调函数被调用
|
|
10); //高水位0就是无限制 默认0 超过10的数据会拆分成多个10调用多次
|
|
|
|
bufferevent_setwatermark(bev, EV_WRITE,
|
|
5, // 低水位0就是无限制 默认0 缓冲数据低于5 写入回调内调用
|
|
0); //暂时无效
|
|
|
|
timeval t1 = { 0,50000 };
|
|
bufferevent_set_timeouts(bev, &t1, NULL);
|
|
bufferevent_setcb(bev, read_cb, write_cb, event_cb, base);
|
|
}
|
|
|
|
void clinet_read_cb(bufferevent * bev, void *arg) {
|
|
cout << "[clinet R]" << endl;
|
|
char data[1024] = { 0 };
|
|
int len = bufferevent_read(bev, data, sizeof(data) - 1);
|
|
if (len <= 0)return;
|
|
cout << "[" << data << "]" << endl;
|
|
/* if(strstr(data,"quit")!= NULL){
|
|
cout << "quit" << endl;
|
|
bufferevent_free(bev); // BEV_OPT_CLOSE_ON_FREE 设置了这个才会关闭
|
|
} */
|
|
//发送数据 写入到输出缓冲
|
|
//bufferevent_write(bev,"ok",3);
|
|
}
|
|
|
|
void clinet_write_cb(bufferevent * bev, void *arg) {
|
|
cout << "[clinet W]" << endl;
|
|
//激活write
|
|
FILE *fp = (FILE *)arg;
|
|
char data[1024] = { 0 };
|
|
int len = fread(data, 1, sizeof(data) - 1, fp);
|
|
if (len <= 0) {
|
|
// 读到文件结尾或者文件出错
|
|
fclose(fp);
|
|
//立刻清理 可能会造成缓冲数据没有发送结束 ！！ 不能在这里做
|
|
//bufferevent_free(bev);
|
|
bufferevent_disable(bev, EV_WRITE);
|
|
return;
|
|
}
|
|
sendCount += len;
|
|
bufferevent_write(bev, data, len);
|
|
|
|
}
|
|
|
|
//错误 超时 连接断开会进入
|
|
void clinet_event_cb(struct bufferevent *bev, short what, void *ctx) {
|
|
cout << "[ clinet CB]" << endl;
|
|
|
|
//读取超时事件发生后，数据读取停止
|
|
if (what & BEV_EVENT_TIMEOUT && what & BEV_EVENT_READING) {
|
|
cout << "read timeout " << endl;
|
|
|
|
//bufferevent_enable(bev,EV_READ); //可以重新设置
|
|
bufferevent_free(bev);
|
|
return;
|
|
|
|
}
|
|
else if (what & BEV_EVENT_ERROR) {
|
|
bufferevent_free(bev);
|
|
return;
|
|
}
|
|
|
|
//服务端的关闭事件
|
|
if (what & BEV_EVENT_EOF) {
|
|
cout << "BEV_EVENT_EOF" << endl;
|
|
bufferevent_free(bev);
|
|
}
|
|
|
|
if (what & BEV_EVENT_CONNECTED) {
|
|
cout << "BEV_EVENT_CONNECTED " << endl;
|
|
bufferevent_trigger(bev, EV_WRITE, 0); //触发写事件
|
|
}
|
|
}
|
|
|
|
int main(int agrc, char **agrv)
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
//初始化socket库
|
|
WSADATA wsa;
|
|
WSAStartup(MAKEWORD(2, 2), &wsa);
|
|
#else
|
|
if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
|
|
{ //忽略管道信号，发送数据给已关闭的socket，会飞掉！
|
|
return 1;
|
|
}
|
|
#endif
|
|
|
|
//创建网络服务器
|
|
event_base *base = event_base_new();
|
|
sockaddr_in sin;
|
|
memset(&sin, 0, sizeof(sin));
|
|
sin.sin_family = AF_INET;
|
|
sin.sin_port = htons(5001);
|
|
|
|
evconnlistener_new_bind(base,
|
|
listen_cb,
|
|
base, LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE,
|
|
10,
|
|
(sockaddr *)&sin,
|
|
sizeof(sin));
|
|
|
|
{ //防止变量名重定义
|
|
//调用客户端代码
|
|
// -1 内部创建socket
|
|
bufferevent * bev = bufferevent_socket_new(base, -1, BEV_OPT_CLOSE_ON_FREE);
|
|
sockaddr_in sin;
|
|
memset(&sin, 0, sizeof(sin));
|
|
sin.sin_family = AF_INET;
|
|
sin.sin_port = htons(5001);
|
|
evutil_inet_pton(AF_INET, "127.0.0.1", &sin.sin_addr.s_addr);
|
|
|
|
FILE *fp = fopen("test_buffer_client.cpp", "rb");
|
|
//设置回调函数
|
|
bufferevent_setcb(bev, clinet_read_cb, clinet_write_cb, clinet_event_cb, fp);
|
|
bufferevent_enable(bev, EV_READ | EV_WRITE);
|
|
int ret = bufferevent_socket_connect(bev, (sockaddr*)&sin, sizeof(sin));
|
|
if (ret == 0) {
|
|
cout << "connected" << endl;
|
|
}
|
|
}
|
|
//进入事件主循环
|
|
event_base_dispatch(base);
|
|
event_base_free(base);
|
|
|
|
return 0;
|
|
}
|