Monitor/RecordSession.cpp at main · WebRTSP/Monitor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#include "RecordSession.h"

#include <glib.h>

#include "RtspParser/RtspParser.h"

#include "Log.h"


static const auto Log = MonitorLog;

RecordSession::RecordSession(
    const Config* config,
    const CreatePeer& createPeer,
    const rtsp::Session::SendRequest& sendRequest,
    const rtsp::Session::SendResponse& sendResponse) noexcept :
    ServerSession(config->webRTCConfig, createPeer, sendRequest, sendResponse),
    _config(config)
{
}

RecordSession::RecordSession(
    const Config* config,
    const CreatePeer& createPeer,
    const CreatePeer& createRecordPeer,
    const rtsp::Session::SendRequest& sendRequest,
    const rtsp::Session::SendResponse& sendResponse) noexcept :
    ServerSession(config->webRTCConfig, createPeer, createRecordPeer, sendRequest, sendResponse),
    _config(config)
{
}

RecordSession::~RecordSession() {
}

bool RecordSession::recordEnabled(const std::string& uri) noexcept
{
    return
        _config->source &&
        _config->source->type == StreamSource::Type::WebRTSP &&
        _config->source->localServer;
}

bool RecordSession::authorizeRecorder(const std::unique_ptr<rtsp::Request>& requestPtr) noexcept
{
    if(!_config->source)
        return false;

    const StreamSource& source = _config->source.value();

    if(source.type != StreamSource::Type::WebRTSP || !source.localServer)
        return false;

    if(source.token.empty())
        return true;

    const std::pair<rtsp::Authentication, std::string> authPair =
        rtsp::ParseAuthentication(*requestPtr);

    if(authPair.first != rtsp::Authentication::Bearer) // FIXME? only Bearer supported atm
        return false;

    return authPair.second == source.token;
}

bool RecordSession::authorize(const std::unique_ptr<rtsp::Request>& requestPtr) noexcept
{
    switch(requestPtr->method) {
    case rtsp::Method::OPTIONS:
        break;
    case rtsp::Method::LIST:
    case rtsp::Method::DESCRIBE:
        return false;
    case rtsp::Method::SETUP:
        break;
    case rtsp::Method::PLAY:
    case rtsp::Method::SUBSCRIBE:
        return false;
    case rtsp::Method::RECORD:
        return authorizeRecorder(requestPtr);
    case rtsp::Method::TEARDOWN:
        break;
    case rtsp::Method::GET_PARAMETER:
    case rtsp::Method::SET_PARAMETER:
        return false;
    }

    return ServerSession::authorize(requestPtr);
}