blob: 9e23b24cc542f1a7b5cbd2edd54455a381a0bb7e [file] [log] [blame]
pbos@webrtc.org29d58392013-05-16 12:08:03 +00001/*
2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
pbos@webrtc.org12dc1a32013-08-05 16:22:53 +000011#include <assert.h>
12#include <string.h>
13
pbos@webrtc.org29d58392013-05-16 12:08:03 +000014#include <map>
15#include <vector>
16
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000017#include "webrtc/call.h"
18#include "webrtc/modules/rtp_rtcp/interface/rtp_header_parser.h"
pbos@webrtc.orgde74b642013-10-02 13:36:09 +000019#include "webrtc/system_wrappers/interface/critical_section_wrapper.h"
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000020#include "webrtc/system_wrappers/interface/rw_lock_wrapper.h"
pbos@webrtc.orgde74b642013-10-02 13:36:09 +000021#include "webrtc/system_wrappers/interface/scoped_ptr.h"
22#include "webrtc/system_wrappers/interface/trace.h"
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000023#include "webrtc/video/video_receive_stream.h"
24#include "webrtc/video/video_send_stream.h"
pbos@webrtc.org29d58392013-05-16 12:08:03 +000025#include "webrtc/video_engine/include/vie_base.h"
26#include "webrtc/video_engine/include/vie_codec.h"
27#include "webrtc/video_engine/include/vie_rtp_rtcp.h"
pbos@webrtc.org29d58392013-05-16 12:08:03 +000028
29namespace webrtc {
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000030namespace internal {
31class Call : public webrtc::Call, public PacketReceiver {
32 public:
33 Call(webrtc::VideoEngine* video_engine, const Call::Config& config);
34 virtual ~Call();
35
36 virtual PacketReceiver* Receiver() OVERRIDE;
37 virtual std::vector<VideoCodec> GetVideoCodecs() OVERRIDE;
38
39 virtual VideoSendStream::Config GetDefaultSendConfig() OVERRIDE;
40
pbos@webrtc.org5a636552013-11-20 10:40:25 +000041 virtual VideoSendStream* CreateVideoSendStream(
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000042 const VideoSendStream::Config& config) OVERRIDE;
43
pbos@webrtc.org2c46f8d2013-11-21 13:49:43 +000044 virtual void DestroyVideoSendStream(webrtc::VideoSendStream* send_stream)
45 OVERRIDE;
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000046
47 virtual VideoReceiveStream::Config GetDefaultReceiveConfig() OVERRIDE;
48
pbos@webrtc.org5a636552013-11-20 10:40:25 +000049 virtual VideoReceiveStream* CreateVideoReceiveStream(
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000050 const VideoReceiveStream::Config& config) OVERRIDE;
51
pbos@webrtc.org2c46f8d2013-11-21 13:49:43 +000052 virtual void DestroyVideoReceiveStream(
pbos@webrtc.org16e03b72013-10-28 16:32:01 +000053 webrtc::VideoReceiveStream* receive_stream) OVERRIDE;
54
55 virtual uint32_t SendBitrateEstimate() OVERRIDE;
56 virtual uint32_t ReceiveBitrateEstimate() OVERRIDE;
57
58 virtual bool DeliverPacket(const uint8_t* packet, size_t length) OVERRIDE;
59
60 private:
61 bool DeliverRtcp(const uint8_t* packet, size_t length);
62 bool DeliverRtp(const RTPHeader& header,
63 const uint8_t* packet,
64 size_t length);
65
66 Call::Config config_;
67
68 std::map<uint32_t, VideoReceiveStream*> receive_ssrcs_;
69 scoped_ptr<RWLockWrapper> receive_lock_;
70
71 std::map<uint32_t, VideoSendStream*> send_ssrcs_;
72 scoped_ptr<RWLockWrapper> send_lock_;
73
74 scoped_ptr<RtpHeaderParser> rtp_header_parser_;
75
76 webrtc::VideoEngine* video_engine_;
77 ViERTP_RTCP* rtp_rtcp_;
78 ViECodec* codec_;
79
80 DISALLOW_COPY_AND_ASSIGN(Call);
81};
82} // internal
pbos@webrtc.orgfd39e132013-08-14 13:52:52 +000083
pbos@webrtc.orgde74b642013-10-02 13:36:09 +000084class TraceDispatcher : public TraceCallback {
85 public:
86 TraceDispatcher()
87 : crit_(CriticalSectionWrapper::CreateCriticalSection()),
pbos@webrtc.org9b5c8072013-10-02 16:22:18 +000088 initialized_(false),
pbos@webrtc.orgde74b642013-10-02 13:36:09 +000089 filter_(kTraceNone) {}
90
pbos@webrtc.org9b5c8072013-10-02 16:22:18 +000091 ~TraceDispatcher() {
92 if (initialized_) {
93 Trace::ReturnTrace();
94 VideoEngine::SetTraceCallback(NULL);
95 }
96 }
97
pbos@webrtc.orgde74b642013-10-02 13:36:09 +000098 virtual void Print(TraceLevel level,
99 const char* message,
100 int length) OVERRIDE {
101 CriticalSectionScoped lock(crit_.get());
102 for (std::map<Call*, Call::Config*>::iterator it = callbacks_.begin();
103 it != callbacks_.end();
104 ++it) {
105 if ((level & it->second->trace_filter) != kTraceNone)
106 it->second->trace_callback->Print(level, message, length);
107 }
108 }
109
110 void RegisterCallback(Call* call, Call::Config* config) {
111 if (config->trace_callback == NULL)
112 return;
113
114 CriticalSectionScoped lock(crit_.get());
115 callbacks_[call] = config;
116
pbos@webrtc.org9b5c8072013-10-02 16:22:18 +0000117 filter_ |= config->trace_filter;
118 if (filter_ != kTraceNone && !initialized_) {
119 initialized_ = true;
120 Trace::CreateTrace();
121 VideoEngine::SetTraceCallback(this);
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000122 }
pbos@webrtc.org9b5c8072013-10-02 16:22:18 +0000123 VideoEngine::SetTraceFilter(filter_);
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000124 }
125
126 void DeregisterCallback(Call* call) {
127 CriticalSectionScoped lock(crit_.get());
128 callbacks_.erase(call);
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000129
130 filter_ = kTraceNone;
131 for (std::map<Call*, Call::Config*>::iterator it = callbacks_.begin();
132 it != callbacks_.end();
133 ++it) {
134 filter_ |= it->second->trace_filter;
135 }
136
137 VideoEngine::SetTraceFilter(filter_);
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000138 }
139
140 private:
141 scoped_ptr<CriticalSectionWrapper> crit_;
pbos@webrtc.org9b5c8072013-10-02 16:22:18 +0000142 bool initialized_;
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000143 unsigned int filter_;
144 std::map<Call*, Call::Config*> callbacks_;
145};
146
147namespace internal {
pbos@webrtc.org16e03b72013-10-28 16:32:01 +0000148TraceDispatcher* global_trace_dispatcher = NULL;
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000149} // internal
150
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000151Call* Call::Create(const Call::Config& config) {
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000152 if (internal::global_trace_dispatcher == NULL) {
153 TraceDispatcher* dispatcher = new TraceDispatcher();
154 // TODO(pbos): Atomic compare and exchange.
155 if (internal::global_trace_dispatcher == NULL) {
156 internal::global_trace_dispatcher = dispatcher;
157 } else {
158 delete dispatcher;
159 }
160 }
161
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000162 VideoEngine* video_engine = VideoEngine::Create();
pbos@webrtc.orgfd39e132013-08-14 13:52:52 +0000163 assert(video_engine != NULL);
164
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000165 return new internal::Call(video_engine, config);
pbos@webrtc.orgfd39e132013-08-14 13:52:52 +0000166}
pbos@webrtc.orgfd39e132013-08-14 13:52:52 +0000167
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000168namespace internal {
169
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000170Call::Call(webrtc::VideoEngine* video_engine, const Call::Config& config)
mflodman@webrtc.org6879c8a2013-07-23 11:35:00 +0000171 : config_(config),
pbos@webrtc.org1819fd72013-06-10 13:48:26 +0000172 receive_lock_(RWLockWrapper::CreateRWLock()),
173 send_lock_(RWLockWrapper::CreateRWLock()),
pbos@webrtc.org40523702013-08-05 12:49:22 +0000174 rtp_header_parser_(RtpHeaderParser::Create()),
pbos@webrtc.org1819fd72013-06-10 13:48:26 +0000175 video_engine_(video_engine) {
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000176 assert(video_engine != NULL);
mflodman@webrtc.org6879c8a2013-07-23 11:35:00 +0000177 assert(config.send_transport != NULL);
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000178
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000179 global_trace_dispatcher->RegisterCallback(this, &config_);
180
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000181 rtp_rtcp_ = ViERTP_RTCP::GetInterface(video_engine_);
182 assert(rtp_rtcp_ != NULL);
183
184 codec_ = ViECodec::GetInterface(video_engine_);
185 assert(codec_ != NULL);
186}
187
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000188Call::~Call() {
pbos@webrtc.orgde74b642013-10-02 13:36:09 +0000189 global_trace_dispatcher->DeregisterCallback(this);
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000190 codec_->Release();
pbos@webrtc.orgfd39e132013-08-14 13:52:52 +0000191 rtp_rtcp_->Release();
192 webrtc::VideoEngine::Delete(video_engine_);
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000193}
194
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000195PacketReceiver* Call::Receiver() { return this; }
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000196
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000197std::vector<VideoCodec> Call::GetVideoCodecs() {
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000198 std::vector<VideoCodec> codecs;
199
200 VideoCodec codec;
201 for (size_t i = 0; i < static_cast<size_t>(codec_->NumberOfCodecs()); ++i) {
pbos@webrtc.org16e03b72013-10-28 16:32:01 +0000202 if (codec_->GetCodec(static_cast<unsigned char>(i), codec) == 0) {
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000203 codecs.push_back(codec);
204 }
205 }
206 return codecs;
207}
208
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000209VideoSendStream::Config Call::GetDefaultSendConfig() {
pbos@webrtc.org025f4f12013-06-05 11:33:21 +0000210 VideoSendStream::Config config;
211 codec_->GetCodec(0, config.codec);
212 return config;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000213}
214
pbos@webrtc.org5a636552013-11-20 10:40:25 +0000215VideoSendStream* Call::CreateVideoSendStream(
216 const VideoSendStream::Config& config) {
pbos@webrtc.org1819fd72013-06-10 13:48:26 +0000217 assert(config.rtp.ssrcs.size() > 0);
pbos@webrtc.org13d38a12013-11-28 11:59:31 +0000218 assert(config.rtp.ssrcs.size() >= config.codec.numberOfSimulcastStreams);
pbos@webrtc.org1819fd72013-06-10 13:48:26 +0000219
pbos@webrtc.org40523702013-08-05 12:49:22 +0000220 VideoSendStream* send_stream = new VideoSendStream(
221 config_.send_transport, config_.overuse_detection, video_engine_, config);
pbos@webrtc.org1819fd72013-06-10 13:48:26 +0000222
223 WriteLockScoped write_lock(*send_lock_);
224 for (size_t i = 0; i < config.rtp.ssrcs.size(); ++i) {
225 assert(send_ssrcs_.find(config.rtp.ssrcs[i]) == send_ssrcs_.end());
226 send_ssrcs_[config.rtp.ssrcs[i]] = send_stream;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000227 }
228 return send_stream;
229}
230
pbos@webrtc.org2c46f8d2013-11-21 13:49:43 +0000231void Call::DestroyVideoSendStream(webrtc::VideoSendStream* send_stream) {
pbos@webrtc.org95e51f52013-09-05 12:38:54 +0000232 assert(send_stream != NULL);
233
234 VideoSendStream* send_stream_impl = NULL;
235 {
236 WriteLockScoped write_lock(*send_lock_);
237 for (std::map<uint32_t, VideoSendStream*>::iterator it =
238 send_ssrcs_.begin();
239 it != send_ssrcs_.end();
240 ++it) {
241 if (it->second == static_cast<VideoSendStream*>(send_stream)) {
242 send_stream_impl = it->second;
243 send_ssrcs_.erase(it);
244 break;
245 }
246 }
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000247 }
pbos@webrtc.org95e51f52013-09-05 12:38:54 +0000248
249 assert(send_stream_impl != NULL);
250 delete send_stream_impl;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000251}
252
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000253VideoReceiveStream::Config Call::GetDefaultReceiveConfig() {
pbos@webrtc.org74fa4892013-08-23 09:19:30 +0000254 return VideoReceiveStream::Config();
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000255}
256
pbos@webrtc.org5a636552013-11-20 10:40:25 +0000257VideoReceiveStream* Call::CreateVideoReceiveStream(
pbos@webrtc.org74fa4892013-08-23 09:19:30 +0000258 const VideoReceiveStream::Config& config) {
stefan@webrtc.orgb082ade2013-11-18 11:45:11 +0000259 VideoReceiveStream* receive_stream = new VideoReceiveStream(
260 video_engine_, config, config_.send_transport, config_.voice_engine);
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000261
pbos@webrtc.org1819fd72013-06-10 13:48:26 +0000262 WriteLockScoped write_lock(*receive_lock_);
263 assert(receive_ssrcs_.find(config.rtp.ssrc) == receive_ssrcs_.end());
264 receive_ssrcs_[config.rtp.ssrc] = receive_stream;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000265 return receive_stream;
266}
267
pbos@webrtc.org2c46f8d2013-11-21 13:49:43 +0000268void Call::DestroyVideoReceiveStream(
269 webrtc::VideoReceiveStream* receive_stream) {
pbos@webrtc.org95e51f52013-09-05 12:38:54 +0000270 assert(receive_stream != NULL);
271
272 VideoReceiveStream* receive_stream_impl = NULL;
273 {
274 WriteLockScoped write_lock(*receive_lock_);
275 for (std::map<uint32_t, VideoReceiveStream*>::iterator it =
276 receive_ssrcs_.begin();
277 it != receive_ssrcs_.end();
278 ++it) {
279 if (it->second == static_cast<VideoReceiveStream*>(receive_stream)) {
280 receive_stream_impl = it->second;
281 receive_ssrcs_.erase(it);
282 break;
283 }
284 }
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000285 }
pbos@webrtc.org95e51f52013-09-05 12:38:54 +0000286
287 assert(receive_stream_impl != NULL);
288 delete receive_stream_impl;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000289}
290
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000291uint32_t Call::SendBitrateEstimate() {
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000292 // TODO(pbos): Return send-bitrate estimate
293 return 0;
294}
295
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000296uint32_t Call::ReceiveBitrateEstimate() {
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000297 // TODO(pbos): Return receive-bitrate estimate
298 return 0;
299}
300
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000301bool Call::DeliverRtcp(const uint8_t* packet, size_t length) {
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000302 // TODO(pbos): Figure out what channel needs it actually.
303 // Do NOT broadcast! Also make sure it's a valid packet.
304 bool rtcp_delivered = false;
pbos@webrtc.org40523702013-08-05 12:49:22 +0000305 {
306 ReadLockScoped read_lock(*receive_lock_);
307 for (std::map<uint32_t, VideoReceiveStream*>::iterator it =
308 receive_ssrcs_.begin();
309 it != receive_ssrcs_.end();
310 ++it) {
pbos@webrtc.org0e63e762013-09-20 11:56:26 +0000311 if (it->second->DeliverRtcp(packet, length))
pbos@webrtc.org40523702013-08-05 12:49:22 +0000312 rtcp_delivered = true;
pbos@webrtc.orgbbb07e62013-08-05 12:01:36 +0000313 }
314 }
315
pbos@webrtc.org40523702013-08-05 12:49:22 +0000316 {
317 ReadLockScoped read_lock(*send_lock_);
318 for (std::map<uint32_t, VideoSendStream*>::iterator it =
319 send_ssrcs_.begin();
320 it != send_ssrcs_.end();
321 ++it) {
pbos@webrtc.org0e63e762013-09-20 11:56:26 +0000322 if (it->second->DeliverRtcp(packet, length))
pbos@webrtc.org40523702013-08-05 12:49:22 +0000323 rtcp_delivered = true;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000324 }
325 }
326 return rtcp_delivered;
327}
328
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000329bool Call::DeliverRtp(const RTPHeader& header,
330 const uint8_t* packet,
331 size_t length) {
pbos@webrtc.org40523702013-08-05 12:49:22 +0000332 VideoReceiveStream* receiver;
333 {
334 ReadLockScoped read_lock(*receive_lock_);
335 std::map<uint32_t, VideoReceiveStream*>::iterator it =
336 receive_ssrcs_.find(header.ssrc);
337 if (it == receive_ssrcs_.end()) {
338 // TODO(pbos): Log some warning, SSRC without receiver.
339 return false;
340 }
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000341
pbos@webrtc.org40523702013-08-05 12:49:22 +0000342 receiver = it->second;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000343 }
pbos@webrtc.orgbbb07e62013-08-05 12:01:36 +0000344 return receiver->DeliverRtp(static_cast<const uint8_t*>(packet), length);
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000345}
346
pbos@webrtc.org841c8a42013-09-09 15:04:25 +0000347bool Call::DeliverPacket(const uint8_t* packet, size_t length) {
pbos@webrtc.org40523702013-08-05 12:49:22 +0000348 // TODO(pbos): ExtensionMap if there are extensions.
349 if (RtpHeaderParser::IsRtcp(packet, static_cast<int>(length)))
350 return DeliverRtcp(packet, length);
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000351
pbos@webrtc.org40523702013-08-05 12:49:22 +0000352 RTPHeader rtp_header;
353 if (!rtp_header_parser_->Parse(packet, static_cast<int>(length), &rtp_header))
354 return false;
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000355
pbos@webrtc.org40523702013-08-05 12:49:22 +0000356 return DeliverRtp(rtp_header, packet, length);
pbos@webrtc.org29d58392013-05-16 12:08:03 +0000357}
358
359} // namespace internal
360} // namespace webrtc