123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839 |
- /**
- * Copyright (c) 2019-2020 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 "include.hpp"
- #include "rtc.h"
- #include "datachannel.hpp"
- #include "log.hpp"
- #include "peerconnection.hpp"
- #if RTC_ENABLE_WEBSOCKET
- #include "websocket.hpp"
- #endif
- #include "plog/Formatters/FuncMessageFormatter.h"
- #include <chrono>
- #include <exception>
- #include <mutex>
- #include <type_traits>
- #include <unordered_map>
- #include <utility>
- #ifdef _WIN32
- #include <codecvt>
- #include <locale>
- #endif
- using namespace rtc;
- using std::shared_ptr;
- using std::string;
- using std::chrono::milliseconds;
- namespace {
- std::unordered_map<int, shared_ptr<PeerConnection>> peerConnectionMap;
- std::unordered_map<int, shared_ptr<DataChannel>> dataChannelMap;
- std::unordered_map<int, shared_ptr<Track>> trackMap;
- #if RTC_ENABLE_WEBSOCKET
- std::unordered_map<int, shared_ptr<WebSocket>> webSocketMap;
- #endif
- std::unordered_map<int, void *> userPointerMap;
- std::mutex mutex;
- int lastId = 0;
- std::optional<void *> getUserPointer(int id) {
- std::lock_guard lock(mutex);
- auto it = userPointerMap.find(id);
- return it != userPointerMap.end() ? std::make_optional(it->second) : nullopt;
- }
- void setUserPointer(int i, void *ptr) {
- std::lock_guard lock(mutex);
- userPointerMap[i] = ptr;
- }
- shared_ptr<PeerConnection> getPeerConnection(int id) {
- std::lock_guard lock(mutex);
- if (auto it = peerConnectionMap.find(id); it != peerConnectionMap.end())
- return it->second;
- else
- throw std::invalid_argument("PeerConnection ID does not exist");
- }
- shared_ptr<DataChannel> getDataChannel(int id) {
- std::lock_guard lock(mutex);
- if (auto it = dataChannelMap.find(id); it != dataChannelMap.end())
- return it->second;
- else
- throw std::invalid_argument("DataChannel ID does not exist");
- }
- shared_ptr<Track> getTrack(int id) {
- std::lock_guard lock(mutex);
- if (auto it = trackMap.find(id); it != trackMap.end())
- return it->second;
- else
- throw std::invalid_argument("Track ID does not exist");
- }
- int emplacePeerConnection(shared_ptr<PeerConnection> ptr) {
- std::lock_guard lock(mutex);
- int pc = ++lastId;
- peerConnectionMap.emplace(std::make_pair(pc, ptr));
- userPointerMap.emplace(std::make_pair(pc, nullptr));
- return pc;
- }
- int emplaceDataChannel(shared_ptr<DataChannel> ptr) {
- std::lock_guard lock(mutex);
- int dc = ++lastId;
- dataChannelMap.emplace(std::make_pair(dc, ptr));
- userPointerMap.emplace(std::make_pair(dc, nullptr));
- return dc;
- }
- int emplaceTrack(shared_ptr<Track> ptr) {
- std::lock_guard lock(mutex);
- int tr = ++lastId;
- trackMap.emplace(std::make_pair(tr, ptr));
- userPointerMap.emplace(std::make_pair(tr, nullptr));
- return tr;
- }
- void erasePeerConnection(int pc) {
- std::lock_guard lock(mutex);
- if (peerConnectionMap.erase(pc) == 0)
- throw std::invalid_argument("PeerConnection ID does not exist");
- userPointerMap.erase(pc);
- }
- void eraseDataChannel(int dc) {
- std::lock_guard lock(mutex);
- if (dataChannelMap.erase(dc) == 0)
- throw std::invalid_argument("DataChannel ID does not exist");
- userPointerMap.erase(dc);
- }
- void eraseTrack(int tr) {
- std::lock_guard lock(mutex);
- if (trackMap.erase(tr) == 0)
- throw std::invalid_argument("Track ID does not exist");
- userPointerMap.erase(tr);
- }
- #if RTC_ENABLE_WEBSOCKET
- shared_ptr<WebSocket> getWebSocket(int id) {
- std::lock_guard lock(mutex);
- if (auto it = webSocketMap.find(id); it != webSocketMap.end())
- return it->second;
- else
- throw std::invalid_argument("WebSocket ID does not exist");
- }
- int emplaceWebSocket(shared_ptr<WebSocket> ptr) {
- std::lock_guard lock(mutex);
- int ws = ++lastId;
- webSocketMap.emplace(std::make_pair(ws, ptr));
- userPointerMap.emplace(std::make_pair(ws, nullptr));
- return ws;
- }
- void eraseWebSocket(int ws) {
- std::lock_guard lock(mutex);
- if (webSocketMap.erase(ws) == 0)
- throw std::invalid_argument("WebSocket ID does not exist");
- userPointerMap.erase(ws);
- }
- #endif
- shared_ptr<Channel> getChannel(int id) {
- std::lock_guard lock(mutex);
- if (auto it = dataChannelMap.find(id); it != dataChannelMap.end())
- return it->second;
- if (auto it = trackMap.find(id); it != trackMap.end())
- return it->second;
- #if RTC_ENABLE_WEBSOCKET
- if (auto it = webSocketMap.find(id); it != webSocketMap.end())
- return it->second;
- #endif
- throw std::invalid_argument("DataChannel, Track, or WebSocket ID does not exist");
- }
- template <typename F> int wrap(F func) {
- try {
- return int(func());
- } catch (const std::invalid_argument &e) {
- PLOG_ERROR << e.what();
- return RTC_ERR_INVALID;
- } catch (const std::exception &e) {
- PLOG_ERROR << e.what();
- return RTC_ERR_FAILURE;
- }
- }
- #define WRAP(statement) \
- wrap([&]() { \
- statement; \
- return RTC_ERR_SUCCESS; \
- })
- int copyAndReturn(string s, char *buffer, int size) {
- if (!buffer)
- return int(s.size() + 1);
- if (size < int(s.size()))
- return RTC_ERR_TOO_SMALL;
- std::copy(s.begin(), s.end(), buffer);
- buffer[s.size()] = '\0';
- return int(s.size() + 1);
- }
- int copyAndReturn(binary b, char *buffer, int size) {
- if (!buffer)
- return int(b.size());
- if (size < int(b.size()))
- return RTC_ERR_TOO_SMALL;
- auto data = reinterpret_cast<const char *>(b.data());
- std::copy(data, data + b.size(), buffer);
- buffer[b.size()] = '\0';
- return int(b.size());
- }
- class plogAppender : public plog::IAppender {
- public:
- plogAppender(rtcLogCallbackFunc cb = nullptr) { setCallback(cb); }
- plogAppender(plogAppender &&appender) : callback(nullptr) {
- std::lock_guard lock(appender.callbackMutex);
- std::swap(appender.callback, callback);
- }
- void setCallback(rtcLogCallbackFunc cb) {
- std::lock_guard lock(callbackMutex);
- callback = cb;
- }
- void write(const plog::Record &record) override {
- plog::Severity severity = record.getSeverity();
- auto formatted = plog::FuncMessageFormatter::format(record);
- formatted.pop_back(); // remove newline
- #ifdef _WIN32
- using convert_type = std::codecvt_utf8<wchar_t>;
- std::wstring_convert<convert_type, wchar_t> converter;
- std::string str = converter.to_bytes(formatted);
- #else
- std::string str = formatted;
- #endif
- std::lock_guard lock(callbackMutex);
- if (callback)
- callback(static_cast<rtcLogLevel>(record.getSeverity()), str.c_str());
- else
- std::cout << plog::severityToString(severity) << " " << str << std::endl;
- }
- private:
- rtcLogCallbackFunc callback;
- std::mutex callbackMutex;
- };
- } // namespace
- void rtcInitLogger(rtcLogLevel level, rtcLogCallbackFunc cb) {
- static std::optional<plogAppender> appender;
- const auto severity = static_cast<plog::Severity>(level);
- std::lock_guard lock(mutex);
- if (appender) {
- appender->setCallback(cb);
- InitLogger(severity, nullptr); // change the severity
- } else if (cb) {
- appender.emplace(plogAppender(cb));
- InitLogger(severity, &appender.value());
- } else {
- InitLogger(severity, nullptr); // log to stdout
- }
- }
- void rtcSetUserPointer(int i, void *ptr) { setUserPointer(i, ptr); }
- int rtcCreatePeerConnection(const rtcConfiguration *config) {
- return WRAP({
- Configuration c;
- for (int i = 0; i < config->iceServersCount; ++i)
- c.iceServers.emplace_back(string(config->iceServers[i]));
- if (config->portRangeBegin || config->portRangeEnd) {
- c.portRangeBegin = config->portRangeBegin;
- c.portRangeEnd = config->portRangeEnd;
- }
- return emplacePeerConnection(std::make_shared<PeerConnection>(c));
- });
- }
- int rtcDeletePeerConnection(int pc) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- peerConnection->onDataChannel(nullptr);
- peerConnection->onLocalDescription(nullptr);
- peerConnection->onLocalCandidate(nullptr);
- peerConnection->onStateChange(nullptr);
- peerConnection->onGatheringStateChange(nullptr);
- erasePeerConnection(pc);
- });
- }
- int rtcAddDataChannel(int pc, const char *label) {
- return rtcAddDataChannelEx(pc, label, nullptr);
- }
- int rtcAddDataChannelEx(int pc, const char *label, const rtcDataChannelInit *init) {
- return WRAP({
- DataChannelInit dci = {};
- if (init) {
- auto *reliability = &init->reliability;
- dci.reliability.unordered = reliability->unordered;
- if (reliability->unreliable) {
- if (reliability->maxPacketLifeTime > 0) {
- dci.reliability.type = Reliability::Type::Timed;
- dci.reliability.rexmit = milliseconds(reliability->maxPacketLifeTime);
- } else {
- dci.reliability.type = Reliability::Type::Rexmit;
- dci.reliability.rexmit = int(reliability->maxRetransmits);
- }
- } else {
- dci.reliability.type = Reliability::Type::Reliable;
- }
- dci.negotiated = init->negotiated;
- dci.id = init->manualStream ? std::make_optional(init->stream) : nullopt;
- dci.protocol = init->protocol ? init->protocol : "";
- }
- auto peerConnection = getPeerConnection(pc);
- int dc = emplaceDataChannel(
- peerConnection->addDataChannel(string(label ? label : ""), std::move(dci)));
- if (auto ptr = getUserPointer(pc))
- rtcSetUserPointer(dc, *ptr);
- return dc;
- });
- }
- int rtcCreateDataChannel(int pc, const char *label) {
- return rtcCreateDataChannelEx(pc, label, nullptr);
- }
- int rtcCreateDataChannelEx(int pc, const char *label, const rtcDataChannelInit *init) {
- int dc = rtcAddDataChannelEx(pc, label, init);
- rtcSetLocalDescription(pc, NULL);
- return dc;
- }
- int rtcDeleteDataChannel(int dc) {
- return WRAP({
- auto dataChannel = getDataChannel(dc);
- dataChannel->onOpen(nullptr);
- dataChannel->onClosed(nullptr);
- dataChannel->onError(nullptr);
- dataChannel->onMessage(nullptr);
- dataChannel->onBufferedAmountLow(nullptr);
- dataChannel->onAvailable(nullptr);
- eraseDataChannel(dc);
- });
- }
- int rtcAddTrack(int pc, const char *mediaDescriptionSdp) {
- return WRAP({
- if (!mediaDescriptionSdp)
- throw std::invalid_argument("Unexpected null pointer for track media description");
- auto peerConnection = getPeerConnection(pc);
- Description::Media media{string(mediaDescriptionSdp)};
- int tr = emplaceTrack(peerConnection->addTrack(std::move(media)));
- if (auto ptr = getUserPointer(pc))
- rtcSetUserPointer(tr, *ptr);
- return tr;
- });
- }
- int rtcDeleteTrack(int tr) {
- return WRAP({
- auto track = getTrack(tr);
- track->onOpen(nullptr);
- track->onClosed(nullptr);
- track->onError(nullptr);
- track->onMessage(nullptr);
- track->onBufferedAmountLow(nullptr);
- track->onAvailable(nullptr);
- eraseTrack(tr);
- });
- }
- int rtcGetTrackDescription(int tr, char *buffer, int size) {
- return WRAP({
- auto track = getTrack(tr);
- return copyAndReturn(track->description(), buffer, size);
- });
- }
- #if RTC_ENABLE_WEBSOCKET
- int rtcCreateWebSocket(const char *url) {
- return WRAP({
- auto ws = std::make_shared<WebSocket>();
- ws->open(url);
- return emplaceWebSocket(ws);
- });
- }
- int rtcCreateWebSocketEx(const char *url, const rtcWsConfiguration *config) {
- return WRAP({
- WebSocket::Configuration c;
- c.disableTlsVerification = config->disableTlsVerification;
- auto ws = std::make_shared<WebSocket>(c);
- ws->open(url);
- return emplaceWebSocket(ws);
- });
- }
- int rtcDeleteWebsocket(int ws) {
- return WRAP({
- auto webSocket = getWebSocket(ws);
- webSocket->onOpen(nullptr);
- webSocket->onClosed(nullptr);
- webSocket->onError(nullptr);
- webSocket->onMessage(nullptr);
- webSocket->onBufferedAmountLow(nullptr);
- webSocket->onAvailable(nullptr);
- eraseWebSocket(ws);
- });
- }
- #endif
- int rtcSetLocalDescriptionCallback(int pc, rtcDescriptionCallbackFunc cb) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (cb)
- peerConnection->onLocalDescription([pc, cb](Description desc) {
- if (auto ptr = getUserPointer(pc))
- cb(pc, string(desc).c_str(), desc.typeString().c_str(), *ptr);
- });
- else
- peerConnection->onLocalDescription(nullptr);
- });
- }
- int rtcSetLocalCandidateCallback(int pc, rtcCandidateCallbackFunc cb) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (cb)
- peerConnection->onLocalCandidate([pc, cb](Candidate cand) {
- if (auto ptr = getUserPointer(pc))
- cb(pc, cand.candidate().c_str(), cand.mid().c_str(), *ptr);
- });
- else
- peerConnection->onLocalCandidate(nullptr);
- });
- }
- int rtcSetStateChangeCallback(int pc, rtcStateChangeCallbackFunc cb) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (cb)
- peerConnection->onStateChange([pc, cb](PeerConnection::State state) {
- if (auto ptr = getUserPointer(pc))
- cb(pc, static_cast<rtcState>(state), *ptr);
- });
- else
- peerConnection->onStateChange(nullptr);
- });
- }
- int rtcSetGatheringStateChangeCallback(int pc, rtcGatheringStateCallbackFunc cb) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (cb)
- peerConnection->onGatheringStateChange([pc, cb](PeerConnection::GatheringState state) {
- if (auto ptr = getUserPointer(pc))
- cb(pc, static_cast<rtcGatheringState>(state), *ptr);
- });
- else
- peerConnection->onGatheringStateChange(nullptr);
- });
- }
- int rtcSetSignalingStateChangeCallback(int pc, rtcSignalingStateCallbackFunc cb) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (cb)
- peerConnection->onSignalingStateChange([pc, cb](PeerConnection::SignalingState state) {
- if (auto ptr = getUserPointer(pc))
- cb(pc, static_cast<rtcSignalingState>(state), *ptr);
- });
- else
- peerConnection->onGatheringStateChange(nullptr);
- });
- }
- int rtcSetDataChannelCallback(int pc, rtcDataChannelCallbackFunc cb) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (cb)
- peerConnection->onDataChannel([pc, cb](std::shared_ptr<DataChannel> dataChannel) {
- int dc = emplaceDataChannel(dataChannel);
- if (auto ptr = getUserPointer(pc)) {
- rtcSetUserPointer(dc, *ptr);
- cb(pc, dc, *ptr);
- }
- });
- else
- peerConnection->onDataChannel(nullptr);
- });
- }
- int rtcSetTrackCallback(int pc, rtcTrackCallbackFunc cb) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (cb)
- peerConnection->onTrack([pc, cb](std::shared_ptr<Track> track) {
- int tr = emplaceTrack(track);
- if (auto ptr = getUserPointer(pc)) {
- rtcSetUserPointer(tr, *ptr);
- cb(pc, tr, *ptr);
- }
- });
- else
- peerConnection->onTrack(nullptr);
- });
- }
- int rtcSetLocalDescription(int pc, const char *type) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- peerConnection->setLocalDescription(type ? Description::stringToType(type)
- : Description::Type::Unspec);
- });
- }
- int rtcSetRemoteDescription(int pc, const char *sdp, const char *type) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (!sdp)
- throw std::invalid_argument("Unexpected null pointer for remote description");
- peerConnection->setRemoteDescription({string(sdp), type ? string(type) : ""});
- });
- }
- int rtcAddRemoteCandidate(int pc, const char *cand, const char *mid) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (!cand)
- throw std::invalid_argument("Unexpected null pointer for remote candidate");
- peerConnection->addRemoteCandidate({string(cand), mid ? string(mid) : ""});
- });
- }
- int rtcGetLocalDescription(int pc, char *buffer, int size) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (auto desc = peerConnection->localDescription())
- return copyAndReturn(string(*desc), buffer, size);
- else
- return RTC_ERR_NOT_AVAIL;
- });
- }
- int rtcGetRemoteDescription(int pc, char *buffer, int size) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (auto desc = peerConnection->remoteDescription())
- return copyAndReturn(string(*desc), buffer, size);
- else
- return RTC_ERR_NOT_AVAIL;
- });
- }
- int rtcGetLocalAddress(int pc, char *buffer, int size) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (auto addr = peerConnection->localAddress())
- return copyAndReturn(std::move(*addr), buffer, size);
- else
- return RTC_ERR_NOT_AVAIL;
- });
- }
- int rtcGetRemoteAddress(int pc, char *buffer, int size) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- if (auto addr = peerConnection->remoteAddress())
- return copyAndReturn(std::move(*addr), buffer, size);
- else
- return RTC_ERR_NOT_AVAIL;
- });
- }
- int rtcGetSelectedCandidatePair(int pc, char *local, int localSize, char *remote, int remoteSize) {
- return WRAP({
- auto peerConnection = getPeerConnection(pc);
- Candidate localCand;
- Candidate remoteCand;
- if (!peerConnection->getSelectedCandidatePair(&localCand, &remoteCand))
- return RTC_ERR_NOT_AVAIL;
- int localRet = copyAndReturn(string(localCand), local, localSize);
- if (localRet < 0)
- return localRet;
- int remoteRet = copyAndReturn(string(remoteCand), remote, remoteSize);
- if (remoteRet < 0)
- return remoteRet;
- return std::max(localRet, remoteRet);
- });
- }
- int rtcGetDataChannelStream(int dc) {
- return WRAP({
- auto dataChannel = getDataChannel(dc);
- return int(dataChannel->id());
- });
- }
- int rtcGetDataChannelLabel(int dc, char *buffer, int size) {
- return WRAP({
- auto dataChannel = getDataChannel(dc);
- return copyAndReturn(dataChannel->label(), buffer, size);
- });
- }
- int rtcGetDataChannelProtocol(int dc, char *buffer, int size) {
- return WRAP({
- auto dataChannel = getDataChannel(dc);
- return copyAndReturn(dataChannel->protocol(), buffer, size);
- });
- }
- int rtcGetDataChannelReliability(int dc, rtcReliability *reliability) {
- return WRAP({
- auto dataChannel = getDataChannel(dc);
- if (!reliability)
- throw std::invalid_argument("Unexpected null pointer for reliability");
- Reliability dcr = dataChannel->reliability();
- std::memset(reliability, 0, sizeof(*reliability));
- reliability->unordered = dcr.unordered;
- if (dcr.type == Reliability::Type::Timed) {
- reliability->unreliable = true;
- reliability->maxPacketLifeTime = unsigned(std::get<milliseconds>(dcr.rexmit).count());
- } else if (dcr.type == Reliability::Type::Rexmit) {
- reliability->unreliable = true;
- reliability->maxRetransmits = unsigned(std::get<int>(dcr.rexmit));
- } else {
- reliability->unreliable = false;
- }
- return RTC_ERR_SUCCESS;
- });
- }
- int rtcSetOpenCallback(int id, rtcOpenCallbackFunc cb) {
- return WRAP({
- auto channel = getChannel(id);
- if (cb)
- channel->onOpen([id, cb]() {
- if (auto ptr = getUserPointer(id))
- cb(id, *ptr);
- });
- else
- channel->onOpen(nullptr);
- });
- }
- int rtcSetClosedCallback(int id, rtcClosedCallbackFunc cb) {
- return WRAP({
- auto channel = getChannel(id);
- if (cb)
- channel->onClosed([id, cb]() {
- if (auto ptr = getUserPointer(id))
- cb(id, *ptr);
- });
- else
- channel->onClosed(nullptr);
- });
- }
- int rtcSetErrorCallback(int id, rtcErrorCallbackFunc cb) {
- return WRAP({
- auto channel = getChannel(id);
- if (cb)
- channel->onError([id, cb](string error) {
- if (auto ptr = getUserPointer(id))
- cb(id, error.c_str(), *ptr);
- });
- else
- channel->onError(nullptr);
- });
- }
- int rtcSetMessageCallback(int id, rtcMessageCallbackFunc cb) {
- return WRAP({
- auto channel = getChannel(id);
- if (cb)
- channel->onMessage(
- [id, cb](binary b) {
- if (auto ptr = getUserPointer(id))
- cb(id, reinterpret_cast<const char *>(b.data()), int(b.size()), *ptr);
- },
- [id, cb](string s) {
- if (auto ptr = getUserPointer(id))
- cb(id, s.c_str(), -int(s.size() + 1), *ptr);
- });
- else
- channel->onMessage(nullptr);
- });
- }
- int rtcSendMessage(int id, const char *data, int size) {
- return WRAP({
- auto channel = getChannel(id);
- if (!data && size != 0)
- throw std::invalid_argument("Unexpected null pointer for data");
- if (size >= 0) {
- auto b = reinterpret_cast<const byte *>(data);
- channel->send(binary(b, b + size));
- return size;
- } else {
- string str(data);
- int len = int(str.size());
- channel->send(std::move(str));
- return len;
- }
- });
- }
- int rtcGetBufferedAmount(int id) {
- return WRAP({
- auto channel = getChannel(id);
- return int(channel->bufferedAmount());
- });
- }
- int rtcSetBufferedAmountLowThreshold(int id, int amount) {
- return WRAP({
- auto channel = getChannel(id);
- channel->setBufferedAmountLowThreshold(size_t(amount));
- });
- }
- int rtcSetBufferedAmountLowCallback(int id, rtcBufferedAmountLowCallbackFunc cb) {
- return WRAP({
- auto channel = getChannel(id);
- if (cb)
- channel->onBufferedAmountLow([id, cb]() {
- if (auto ptr = getUserPointer(id))
- cb(id, *ptr);
- });
- else
- channel->onBufferedAmountLow(nullptr);
- });
- }
- int rtcGetAvailableAmount(int id) {
- return WRAP({ return int(getChannel(id)->availableAmount()); });
- }
- int rtcSetAvailableCallback(int id, rtcAvailableCallbackFunc cb) {
- return WRAP({
- auto channel = getChannel(id);
- if (cb)
- channel->onAvailable([id, cb]() {
- if (auto ptr = getUserPointer(id))
- cb(id, *ptr);
- });
- else
- channel->onAvailable(nullptr);
- });
- }
- int rtcReceiveMessage(int id, char *buffer, int *size) {
- return WRAP({
- auto channel = getChannel(id);
- if (!size)
- throw std::invalid_argument("Unexpected null pointer for size");
- *size = std::abs(*size);
- auto message = channel->peek();
- if (!message)
- return RTC_ERR_NOT_AVAIL;
- return std::visit( //
- overloaded{
- [&](binary b) {
- int ret = copyAndReturn(std::move(b), buffer, *size);
- if (ret >= 0) {
- channel->receive(); // discard
- *size = ret;
- return RTC_ERR_SUCCESS;
- } else {
- *size = int(b.size());
- return ret;
- }
- },
- [&](string s) {
- int ret = copyAndReturn(std::move(s), buffer, *size);
- if (ret >= 0) {
- channel->receive(); // discard
- *size = -ret;
- return RTC_ERR_SUCCESS;
- } else {
- *size = -int(s.size() + 1);
- return ret;
- }
- },
- },
- *message);
- });
- }
- void rtcPreload() { rtc::Preload(); }
- void rtcCleanup() { rtc::Cleanup(); }
|