mirror of
https://github.com/mii443/libdatachannel.git
synced 2025-08-22 15:15:28 +00:00
80 lines
2.1 KiB
C++
80 lines
2.1 KiB
C++
/**
|
|
* Copyright (c) 2019 Paul-Louis Ageneau
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "rtc/rtc.hpp"
|
|
|
|
#if RTC_ENABLE_WEBSOCKET
|
|
|
|
#include <atomic>
|
|
#include <chrono>
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <thread>
|
|
|
|
using namespace rtc;
|
|
using namespace std;
|
|
|
|
template <class T> weak_ptr<T> make_weak_ptr(shared_ptr<T> ptr) { return ptr; }
|
|
|
|
void test_websocket() {
|
|
InitLogger(LogLevel::Debug);
|
|
|
|
const string myMessage = "Hello world from libdatachannel";
|
|
|
|
WebSocket::Configuration config;
|
|
config.disableTlsVerification = true;
|
|
WebSocket ws(std::move(config));
|
|
|
|
ws.onOpen([&ws, &myMessage]() {
|
|
cout << "WebSocket: Open" << endl;
|
|
ws.send(myMessage);
|
|
});
|
|
|
|
ws.onClosed([]() { cout << "WebSocket: Closed" << endl; });
|
|
|
|
std::atomic<bool> received = false;
|
|
ws.onMessage([&received, &myMessage](variant<binary, string> message) {
|
|
if (holds_alternative<string>(message)) {
|
|
string str = std::move(get<string>(message));
|
|
if ((received = (str == myMessage)))
|
|
cout << "WebSocket: Received expected message" << endl;
|
|
else
|
|
cout << "WebSocket: Received UNEXPECTED message" << endl;
|
|
}
|
|
});
|
|
|
|
ws.open("wss://echo.websocket.org:443/");
|
|
|
|
int attempts = 10;
|
|
while ((!ws.isOpen() || !received) && attempts--)
|
|
this_thread::sleep_for(1s);
|
|
|
|
if (!ws.isOpen())
|
|
throw runtime_error("WebSocket is not open");
|
|
|
|
if (!received)
|
|
throw runtime_error("Expected message not received");
|
|
|
|
ws.close();
|
|
this_thread::sleep_for(1s);
|
|
|
|
cout << "Success" << endl;
|
|
}
|
|
|
|
#endif
|