139 lines
4.4 KiB
C++
139 lines
4.4 KiB
C++
// rtsp_server.hpp
|
|
//
|
|
// Copyright (c) 2019 2020 Andrea Bondavalli. All rights reserved.
|
|
//
|
|
// 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
|
|
// 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, see <http://www.gnu.org/licenses/>.
|
|
|
|
#ifndef _RTSP_SERVER_HPP_
|
|
#define _RTSP_SERVER_HPP_
|
|
|
|
#include <boost/algorithm/string.hpp>
|
|
#include <boost/asio.hpp>
|
|
#include <chrono>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <unordered_set>
|
|
#include <vector>
|
|
|
|
#include "session_manager.hpp"
|
|
|
|
using namespace std::chrono;
|
|
using boost::asio::deadline_timer;
|
|
using boost::asio::ip::tcp;
|
|
using second_t = duration<double, std::ratio<1> >;
|
|
|
|
class RtspSession : public std::enable_shared_from_this<RtspSession> {
|
|
public:
|
|
constexpr static uint16_t max_length = 4096; // byte
|
|
constexpr static uint16_t session_tout_secs = 10; // sec
|
|
|
|
RtspSession(std::shared_ptr<Config> config,
|
|
std::shared_ptr<SessionManager> session_manager,
|
|
tcp::socket socket)
|
|
: config_(config),
|
|
session_manager_(session_manager),
|
|
socket_(std::move(socket)) {}
|
|
|
|
virtual ~RtspSession() {
|
|
BOOST_LOG_TRIVIAL(debug) << "rtsp_server:: session end";
|
|
}
|
|
|
|
void start();
|
|
void stop();
|
|
|
|
bool announce(uint8_t source_id,
|
|
const std::string& name,
|
|
const std::string& sdp,
|
|
const std::string& address,
|
|
uint16_t port);
|
|
|
|
private:
|
|
bool process_request();
|
|
void build_response(const std::string& url);
|
|
void read_request();
|
|
void send_error(int status_code, const std::string& description);
|
|
void send_response(const std::string& response);
|
|
std::shared_ptr<Config> config_;
|
|
std::shared_ptr<SessionManager> session_manager_;
|
|
tcp::socket socket_;
|
|
char data_[max_length + 1];
|
|
std::string request_;
|
|
size_t length_{0};
|
|
int32_t cseq_{-1};
|
|
size_t consumed_{0};
|
|
int32_t announce_cseq_{0};
|
|
/* set with the ids described on this session */
|
|
std::unordered_set<uint8_t> source_ids_;
|
|
};
|
|
|
|
class RtspServer {
|
|
public:
|
|
constexpr static uint8_t session_num_max{(SessionManager::stream_id_max + 1) *
|
|
2};
|
|
|
|
RtspServer() = delete;
|
|
explicit RtspServer(std::shared_ptr<SessionManager> session_manager,
|
|
std::shared_ptr<Config> config)
|
|
: session_manager_(session_manager),
|
|
config_(config),
|
|
acceptor_(io_service_,
|
|
tcp::endpoint(boost::asio::ip::address::from_string(
|
|
config_->get_ip_addr_str()),
|
|
config_->get_rtsp_port())) {}
|
|
bool init() {
|
|
accept();
|
|
/* start rtsp server on a separate thread */
|
|
res_ = std::async([this]() { io_service_.run(); });
|
|
|
|
session_manager_->add_source_observer(
|
|
SessionManager::ObserverType::add_source,
|
|
std::bind(&RtspServer::update_source, this, std::placeholders::_1,
|
|
std::placeholders::_2, std::placeholders::_3));
|
|
|
|
session_manager_->add_source_observer(
|
|
SessionManager::ObserverType::update_source,
|
|
std::bind(&RtspServer::update_source, this, std::placeholders::_1,
|
|
std::placeholders::_2, std::placeholders::_3));
|
|
|
|
return true;
|
|
}
|
|
|
|
bool terminate() {
|
|
BOOST_LOG_TRIVIAL(info) << "rtsp_server: stopping ... ";
|
|
io_service_.stop();
|
|
res_.get();
|
|
return true;
|
|
}
|
|
|
|
private:
|
|
/* a source was updated */
|
|
bool update_source(uint8_t id,
|
|
const std::string& name,
|
|
const std::string& sdp);
|
|
void accept();
|
|
|
|
std::mutex mutex_;
|
|
boost::asio::io_service io_service_;
|
|
std::shared_ptr<SessionManager> session_manager_;
|
|
std::shared_ptr<Config> config_;
|
|
std::vector<std::weak_ptr<RtspSession> > sessions_{session_num_max};
|
|
std::vector<time_point<steady_clock> > sessions_start_point_{session_num_max};
|
|
tcp::acceptor acceptor_;
|
|
tcp::socket socket_{io_service_};
|
|
std::future<void> res_;
|
|
};
|
|
|
|
#endif
|