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
90
91
92
93
|
/*
* SRT - Secure, Reliable, Transport
* Copyright (c) 2018 Haivision Systems Inc.
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
*/
#ifndef INC_SRT_COMMON_TRANMITBASE_HPP
#define INC_SRT_COMMON_TRANMITBASE_HPP
#include <string>
#include <memory>
#include <vector>
#include <iostream>
#include <stdexcept>
#include "srt.h"
#include "uriparser.hpp"
#include "apputil.hpp"
#include "statswriter.hpp"
typedef std::vector<char> bytevector;
extern bool transmit_total_stats;
extern bool g_stats_are_printed_to_stdout;
extern unsigned long transmit_bw_report;
extern unsigned long transmit_stats_report;
extern unsigned long transmit_chunk_size;
struct MediaPacket
{
bytevector payload;
int64_t time = 0;
MediaPacket(bytevector&& src) : payload(std::move(src)) {}
MediaPacket(bytevector&& src, int64_t stime) : payload(std::move(src)), time(stime) {}
MediaPacket(size_t payload_size) : payload(payload_size), time(0) {}
MediaPacket(const bytevector& src) : payload(src) {}
MediaPacket(const bytevector& src, int64_t stime) : payload(src), time(stime) {}
MediaPacket() {}
};
extern std::shared_ptr<SrtStatsWriter> transmit_stats_writer;
class Location
{
public:
UriParser uri;
Location() {}
};
class Source: public Location
{
public:
virtual int Read(size_t chunk, MediaPacket& pkt, std::ostream &out_stats = std::cout) = 0;
virtual bool IsOpen() = 0;
virtual bool End() = 0;
static std::unique_ptr<Source> Create(const std::string& url);
virtual void Close() {}
virtual ~Source() {}
class ReadEOF: public std::runtime_error
{
public:
ReadEOF(const std::string& fn): std::runtime_error( "EOF while reading file: " + fn )
{
}
};
virtual SRTSOCKET GetSRTSocket() const { return SRT_INVALID_SOCK; }
virtual int GetSysSocket() const { return -1; }
virtual bool AcceptNewClient() { return false; }
};
class Target: public Location
{
public:
virtual int Write(const char* data, size_t size, int64_t src_time, std::ostream &out_stats = std::cout) = 0;
virtual bool IsOpen() = 0;
virtual bool Broken() = 0;
virtual void Close() {}
virtual size_t Still() { return 0; }
static std::unique_ptr<Target> Create(const std::string& url);
virtual ~Target() {}
virtual SRTSOCKET GetSRTSocket() const { return SRT_INVALID_SOCK; }
virtual int GetSysSocket() const { return -1; }
virtual bool AcceptNewClient() { return false; }
};
#endif
|
