69 lines
2.3 KiB
C++
69 lines
2.3 KiB
C++
|
/*
|
||
|
|
* ws_broadcast_server.cpp
|
||
|
|
* Author: Benjamin Sergeant
|
||
|
|
* Copyright (c) 2018 Machine Zone, Inc. All rights reserved.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
#include <sstream>
|
||
|
|
#include <ixwebsocket/IXWebSocketServer.h>
|
||
|
|
|
||
|
|
namespace ix
|
||
|
|
{
|
||
|
|
int ws_echo_server_main(int port)
|
||
|
|
{
|
||
|
|
std::cout << "Listening on port " << port << std::endl;
|
||
|
|
|
||
|
|
ix::WebSocketServer server(port);
|
||
|
|
|
||
|
|
server.setOnConnectionCallback(
|
||
|
|
[&server](std::shared_ptr<ix::WebSocket> webSocket)
|
||
|
|
{
|
||
|
|
webSocket->setOnMessageCallback(
|
||
|
|
[webSocket, &server](ix::WebSocketMessageType messageType,
|
||
|
|
const std::string& str,
|
||
|
|
size_t wireSize,
|
||
|
|
const ix::WebSocketErrorInfo& error,
|
||
|
|
const ix::WebSocketOpenInfo& openInfo,
|
||
|
|
const ix::WebSocketCloseInfo& closeInfo)
|
||
|
|
{
|
||
|
|
if (messageType == ix::WebSocket_MessageType_Open)
|
||
|
|
{
|
||
|
|
std::cerr << "New connection" << std::endl;
|
||
|
|
std::cerr << "Uri: " << openInfo.uri << std::endl;
|
||
|
|
std::cerr << "Headers:" << std::endl;
|
||
|
|
for (auto it : openInfo.headers)
|
||
|
|
{
|
||
|
|
std::cerr << it.first << ": " << it.second << std::endl;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (messageType == ix::WebSocket_MessageType_Close)
|
||
|
|
{
|
||
|
|
std::cerr << "Closed connection" << std::endl;
|
||
|
|
}
|
||
|
|
else if (messageType == ix::WebSocket_MessageType_Message)
|
||
|
|
{
|
||
|
|
std::cerr << "Received "
|
||
|
|
<< wireSize << " bytes"
|
||
|
|
<< std::endl;
|
||
|
|
webSocket->send(str);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
);
|
||
|
|
}
|
||
|
|
);
|
||
|
|
|
||
|
|
auto res = server.listen();
|
||
|
|
if (!res.first)
|
||
|
|
{
|
||
|
|
std::cerr << res.second << std::endl;
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
server.start();
|
||
|
|
server.wait();
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
}
|