52 lines
1.1 KiB
C++
52 lines
1.1 KiB
C++
|
/*
|
||
|
|
* UdpServer_test.cpp
|
||
|
|
*
|
||
|
|
* @build make evpp
|
||
|
|
* @server bin/UdpServer_test 1234
|
||
|
|
* @client bin/UdpClient_test 1234
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
#include "UdpServer.h"
|
||
|
|
|
||
|
|
using namespace hv;
|
||
|
|
|
||
|
|
int main(int argc, char* argv[]) {
|
||
|
|
if (argc < 2) {
|
||
|
|
printf("Usage: %s port\n", argv[0]);
|
||
|
|
return -10;
|
||
|
|
}
|
||
|
|
int port = atoi(argv[1]);
|
||
|
|
|
||
|
|
UdpServer srv;
|
||
|
|
int bindfd = srv.createsocket(port);
|
||
|
|
if (bindfd < 0) {
|
||
|
|
return -20;
|
||
|
|
}
|
||
|
|
printf("server bind on port %d, bindfd=%d ...\n", port, bindfd);
|
||
|
|
srv.onMessage = [](const SocketChannelPtr& channel, Buffer* buf) {
|
||
|
|
// echo
|
||
|
|
printf("< %.*s\n", (int)buf->size(), (char*)buf->data());
|
||
|
|
channel->write(buf);
|
||
|
|
};
|
||
|
|
srv.start();
|
||
|
|
|
||
|
|
std::string str;
|
||
|
|
while (std::getline(std::cin, str)) {
|
||
|
|
if (str == "close") {
|
||
|
|
srv.closesocket();
|
||
|
|
} else if (str == "start") {
|
||
|
|
srv.start();
|
||
|
|
} else if (str == "stop") {
|
||
|
|
srv.stop();
|
||
|
|
break;
|
||
|
|
} else {
|
||
|
|
srv.sendto(str);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|