| /* |
| * Copyright (C) 2004-2021 Savoir-faire Linux Inc. |
| * |
| * Author: Guillaume Roguez <guillaume.roguez@savoirfairelinux.com> |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 3 of the License, or |
| * (at your option) any later version. |
| * |
| * This program 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 General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| */ |
| |
| #pragma once |
| |
| #include "ice_socket.h" |
| #include "ip_utils.h" |
| |
| #include <pjnath.h> |
| #include <pjlib.h> |
| #include <pjlib-util.h> |
| |
| #include <functional> |
| #include <memory> |
| #include <msgpack.hpp> |
| #include <vector> |
| |
| namespace jami { |
| |
| namespace upnp { |
| class Controller; |
| } |
| |
| class IceTransport; |
| |
| using IceTransportCompleteCb = std::function<void(bool)>; |
| using IceRecvCb = std::function<ssize_t(unsigned char* buf, size_t len)>; |
| using IceCandidate = pj_ice_sess_cand; |
| using onShutdownCb = std::function<void(void)>; |
| |
| struct ICESDP |
| { |
| std::vector<IceCandidate> rem_candidates; |
| std::string rem_ufrag; |
| std::string rem_pwd; |
| }; |
| |
| struct StunServerInfo |
| { |
| StunServerInfo& setUri(const std::string& args) |
| { |
| uri = args; |
| return *this; |
| } |
| |
| std::string uri; // server URI, mandatory |
| }; |
| |
| struct TurnServerInfo |
| { |
| TurnServerInfo& setUri(const std::string& args) |
| { |
| uri = args; |
| return *this; |
| } |
| TurnServerInfo& setUsername(const std::string& args) |
| { |
| username = args; |
| return *this; |
| } |
| TurnServerInfo& setPassword(const std::string& args) |
| { |
| password = args; |
| return *this; |
| } |
| TurnServerInfo& setRealm(const std::string& args) |
| { |
| realm = args; |
| return *this; |
| } |
| |
| std::string uri; // server URI, mandatory |
| std::string username; // credentials username (optional, empty if not used) |
| std::string password; // credentials password (optional, empty if not used) |
| std::string realm; // credentials realm (optional, empty if not used) |
| }; |
| |
| struct IceTransportOptions |
| { |
| bool master {true}; |
| unsigned streamsCount {1}; |
| unsigned compCountPerStream {1}; |
| bool upnpEnable {false}; |
| IceTransportCompleteCb onInitDone {}; |
| IceTransportCompleteCb onNegoDone {}; |
| std::vector<StunServerInfo> stunServers; |
| std::vector<TurnServerInfo> turnServers; |
| bool tcpEnable {false}; |
| // Addresses used by the account owning the transport instance. |
| IpAddr accountLocalAddr {}; |
| IpAddr accountPublicAddr {}; |
| }; |
| |
| struct SDP |
| { |
| std::string ufrag; |
| std::string pwd; |
| |
| std::vector<std::string> candidates; |
| MSGPACK_DEFINE(ufrag, pwd, candidates) |
| }; |
| |
| class IceTransport |
| { |
| public: |
| using Attribute = struct |
| { |
| std::string ufrag; |
| std::string pwd; |
| }; |
| |
| /** |
| * Constructor |
| */ |
| IceTransport(const char* name); |
| ~IceTransport(); |
| |
| void initIceInstance(const IceTransportOptions& options); |
| |
| /** |
| * Get current state |
| */ |
| bool isInitiator() const; |
| |
| /** |
| * Start tranport negotiation between local candidates and given remote |
| * to find the right candidate pair. |
| * This function doesn't block, the callback on_negodone_cb will be called |
| * with the negotiation result when operation is really done. |
| * Return false if negotiation cannot be started else true. |
| */ |
| bool startIce(const Attribute& rem_attrs, std::vector<IceCandidate>&& rem_candidates); |
| bool startIce(const SDP& sdp); |
| |
| /** |
| * Stop a started or completed transport. |
| */ |
| bool stop(); |
| |
| /** |
| * Cancel operations |
| */ |
| void cancelOperations(); |
| |
| /** |
| * Returns true if ICE transport has been initialized |
| * [mutex protected] |
| */ |
| bool isInitialized() const; |
| |
| /** |
| * Returns true if ICE negotiation has been started |
| * [mutex protected] |
| */ |
| bool isStarted() const; |
| |
| /** |
| * Returns true if ICE negotiation has completed with success |
| * [mutex protected] |
| */ |
| bool isRunning() const; |
| |
| /** |
| * Return true if a start operations fails or if stop() has been called |
| * [mutex protected] |
| */ |
| bool isStopped() const; |
| |
| /** |
| * Returns true if ICE transport is in failure state |
| * [mutex protected] |
| */ |
| bool isFailed() const; |
| |
| IpAddr getLocalAddress(unsigned comp_id) const; |
| |
| IpAddr getRemoteAddress(unsigned comp_id) const; |
| |
| std::string getLastErrMsg() const; |
| |
| IpAddr getDefaultLocalAddress() const { return getLocalAddress(1); } |
| |
| /** |
| * Return ICE session attributes |
| */ |
| const Attribute getLocalAttributes() const; |
| |
| /** |
| * Return ICE session attributes |
| */ |
| std::vector<std::string> getLocalCandidates(unsigned comp_id) const; |
| |
| /** |
| * Return ICE session attributes |
| */ |
| std::vector<std::string> getLocalCandidates(unsigned streamIdx, unsigned compId) const; |
| |
| bool parseIceAttributeLine(unsigned streamIdx, |
| const std::string& line, |
| IceCandidate& cand) const; |
| |
| bool getCandidateFromSDP(const std::string& line, IceCandidate& cand) const; |
| |
| // I/O methods |
| |
| void setOnRecv(unsigned comp_id, IceRecvCb cb); |
| void setOnShutdown(onShutdownCb&& cb); |
| |
| ssize_t recv(unsigned comp_id, unsigned char* buf, size_t len, std::error_code& ec); |
| ssize_t recvfrom(unsigned comp_id, char* buf, size_t len, std::error_code& ec); |
| |
| ssize_t send(unsigned comp_id, const unsigned char* buf, size_t len); |
| |
| int waitForInitialization(std::chrono::milliseconds timeout); |
| |
| int waitForNegotiation(std::chrono::milliseconds timeout); |
| |
| ssize_t waitForData(unsigned comp_id, std::chrono::milliseconds timeout, std::error_code& ec); |
| |
| unsigned getComponentCount() const; |
| |
| // Set session state |
| bool setSlaveSession(); |
| bool setInitiatorSession(); |
| |
| /** |
| * Get SDP messages list |
| * @param msg The payload to parse |
| * @return the list of SDP messages |
| */ |
| static std::vector<SDP> parseSDPList(const std::vector<uint8_t>& msg); |
| |
| bool isTCPEnabled(); |
| |
| ICESDP parseIceCandidates(std::string_view sdp_msg); |
| |
| void setDefaultRemoteAddress(unsigned comp_id, const IpAddr& addr); |
| |
| std::string link() const; |
| |
| private: |
| class Impl; |
| bool isStopped_ {false}; |
| bool isCancelled_ {false}; |
| std::unique_ptr<Impl> pimpl_; |
| }; |
| |
| class IceTransportFactory |
| { |
| public: |
| IceTransportFactory(); |
| ~IceTransportFactory(); |
| |
| std::shared_ptr<IceTransport> createTransport(const char* name); |
| |
| std::unique_ptr<IceTransport> createUTransport(const char* name); |
| |
| /** |
| * PJSIP specifics |
| */ |
| pj_ice_strans_cfg getIceCfg() const { return ice_cfg_; } |
| pj_pool_factory* getPoolFactory() { return &cp_->factory; } |
| std::shared_ptr<pj_caching_pool> getPoolCaching() { return cp_; } |
| |
| private: |
| std::shared_ptr<pj_caching_pool> cp_; |
| pj_ice_strans_cfg ice_cfg_; |
| }; |
| |
| }; // namespace jami |