1 /*
2  * libjingle
3  * Copyright 2004 Google Inc.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  *
8  *  1. Redistributions of source code must retain the above copyright notice,
9  *     this list of conditions and the following disclaimer.
10  *  2. Redistributions in binary form must reproduce the above copyright notice,
11  *     this list of conditions and the following disclaimer in the documentation
12  *     and/or other materials provided with the distribution.
13  *  3. The name of the author may not be used to endorse or promote products
14  *     derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
17  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
18  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
19  * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #ifndef TALK_MEDIA_WEBRTCVOICEENGINE_H_
29 #define TALK_MEDIA_WEBRTCVOICEENGINE_H_
30 
31 #include <map>
32 #include <string>
33 #include <vector>
34 
35 #include "talk/media/base/rtputils.h"
36 #include "talk/media/webrtc/webrtccommon.h"
37 #include "talk/media/webrtc/webrtcvoe.h"
38 #include "talk/session/media/channel.h"
39 #include "webrtc/audio_state.h"
40 #include "webrtc/base/buffer.h"
41 #include "webrtc/base/scoped_ptr.h"
42 #include "webrtc/base/stream.h"
43 #include "webrtc/base/thread_checker.h"
44 #include "webrtc/call.h"
45 #include "webrtc/common.h"
46 #include "webrtc/config.h"
47 
48 namespace cricket {
49 
50 class AudioDeviceModule;
51 class AudioRenderer;
52 class VoEWrapper;
53 class WebRtcVoiceMediaChannel;
54 
55 // WebRtcVoiceEngine is a class to be used with CompositeMediaEngine.
56 // It uses the WebRtc VoiceEngine library for audio handling.
57 class WebRtcVoiceEngine final : public webrtc::TraceCallback  {
58   friend class WebRtcVoiceMediaChannel;
59  public:
60   // Exposed for the WVoE/MC unit test.
61   static bool ToCodecInst(const AudioCodec& in, webrtc::CodecInst* out);
62 
63   WebRtcVoiceEngine();
64   // Dependency injection for testing.
65   explicit WebRtcVoiceEngine(VoEWrapper* voe_wrapper);
66   ~WebRtcVoiceEngine();
67   bool Init(rtc::Thread* worker_thread);
68   void Terminate();
69 
70   rtc::scoped_refptr<webrtc::AudioState> GetAudioState() const;
71   VoiceMediaChannel* CreateChannel(webrtc::Call* call,
72                                    const AudioOptions& options);
73 
74   bool GetOutputVolume(int* level);
75   bool SetOutputVolume(int level);
76   int GetInputLevel();
77 
78   const std::vector<AudioCodec>& codecs();
79   RtpCapabilities GetCapabilities() const;
80 
81   // For tracking WebRtc channels. Needed because we have to pause them
82   // all when switching devices.
83   // May only be called by WebRtcVoiceMediaChannel.
84   void RegisterChannel(WebRtcVoiceMediaChannel* channel);
85   void UnregisterChannel(WebRtcVoiceMediaChannel* channel);
86 
87   // Called by WebRtcVoiceMediaChannel to set a gain offset from
88   // the default AGC target level.
89   bool AdjustAgcLevel(int delta);
90 
voe()91   VoEWrapper* voe() { return voe_wrapper_.get(); }
92   int GetLastEngineError();
93 
94   // Set the external ADM. This can only be called before Init.
95   bool SetAudioDeviceModule(webrtc::AudioDeviceModule* adm);
96 
97   // Starts AEC dump using existing file.
98   bool StartAecDump(rtc::PlatformFile file);
99 
100   // Stops AEC dump.
101   void StopAecDump();
102 
103   // Starts recording an RtcEventLog using an existing file until 10 minutes
104   // pass or the StopRtcEventLog function is called.
105   bool StartRtcEventLog(rtc::PlatformFile file);
106 
107   // Stops recording the RtcEventLog.
108   void StopRtcEventLog();
109 
110  private:
111   void Construct();
112   bool InitInternal();
113   // Every option that is "set" will be applied. Every option not "set" will be
114   // ignored. This allows us to selectively turn on and off different options
115   // easily at any time.
116   bool ApplyOptions(const AudioOptions& options);
117   void SetDefaultDevices();
118 
119   // webrtc::TraceCallback:
120   void Print(webrtc::TraceLevel level, const char* trace, int length) override;
121 
122   void StartAecDump(const std::string& filename);
123   int CreateVoEChannel();
124 
125   rtc::ThreadChecker signal_thread_checker_;
126   rtc::ThreadChecker worker_thread_checker_;
127 
128   // The primary instance of WebRtc VoiceEngine.
129   rtc::scoped_ptr<VoEWrapper> voe_wrapper_;
130   rtc::scoped_refptr<webrtc::AudioState> audio_state_;
131   // The external audio device manager
132   webrtc::AudioDeviceModule* adm_ = nullptr;
133   std::vector<AudioCodec> codecs_;
134   std::vector<WebRtcVoiceMediaChannel*> channels_;
135   webrtc::Config voe_config_;
136   bool initialized_ = false;
137   bool is_dumping_aec_ = false;
138 
139   webrtc::AgcConfig default_agc_config_;
140   // Cache received extended_filter_aec, delay_agnostic_aec and experimental_ns
141   // values, and apply them in case they are missing in the audio options. We
142   // need to do this because SetExtraOptions() will revert to defaults for
143   // options which are not provided.
144   rtc::Optional<bool> extended_filter_aec_;
145   rtc::Optional<bool> delay_agnostic_aec_;
146   rtc::Optional<bool> experimental_ns_;
147 
148   RTC_DISALLOW_COPY_AND_ASSIGN(WebRtcVoiceEngine);
149 };
150 
151 // WebRtcVoiceMediaChannel is an implementation of VoiceMediaChannel that uses
152 // WebRtc Voice Engine.
153 class WebRtcVoiceMediaChannel final : public VoiceMediaChannel,
154                                       public webrtc::Transport {
155  public:
156   WebRtcVoiceMediaChannel(WebRtcVoiceEngine* engine,
157                           const AudioOptions& options,
158                           webrtc::Call* call);
159   ~WebRtcVoiceMediaChannel() override;
160 
options()161   const AudioOptions& options() const { return options_; }
162 
163   bool SetSendParameters(const AudioSendParameters& params) override;
164   bool SetRecvParameters(const AudioRecvParameters& params) override;
165   bool SetPlayout(bool playout) override;
166   bool PausePlayout();
167   bool ResumePlayout();
168   bool SetSend(SendFlags send) override;
169   bool PauseSend();
170   bool ResumeSend();
171   bool SetAudioSend(uint32_t ssrc,
172                     bool enable,
173                     const AudioOptions* options,
174                     AudioRenderer* renderer) override;
175   bool AddSendStream(const StreamParams& sp) override;
176   bool RemoveSendStream(uint32_t ssrc) override;
177   bool AddRecvStream(const StreamParams& sp) override;
178   bool RemoveRecvStream(uint32_t ssrc) override;
179   bool GetActiveStreams(AudioInfo::StreamList* actives) override;
180   int GetOutputLevel() override;
181   int GetTimeSinceLastTyping() override;
182   void SetTypingDetectionParameters(int time_window,
183                                     int cost_per_typing,
184                                     int reporting_threshold,
185                                     int penalty_decay,
186                                     int type_event_delay) override;
187   bool SetOutputVolume(uint32_t ssrc, double volume) override;
188 
189   bool CanInsertDtmf() override;
190   bool InsertDtmf(uint32_t ssrc, int event, int duration) override;
191 
192   void OnPacketReceived(rtc::Buffer* packet,
193                         const rtc::PacketTime& packet_time) override;
194   void OnRtcpReceived(rtc::Buffer* packet,
195                       const rtc::PacketTime& packet_time) override;
OnReadyToSend(bool ready)196   void OnReadyToSend(bool ready) override {}
197   bool GetStats(VoiceMediaInfo* info) override;
198 
199   void SetRawAudioSink(
200       uint32_t ssrc,
201       rtc::scoped_ptr<webrtc::AudioSinkInterface> sink) override;
202 
203   // implements Transport interface
SendRtp(const uint8_t * data,size_t len,const webrtc::PacketOptions & options)204   bool SendRtp(const uint8_t* data,
205                size_t len,
206                const webrtc::PacketOptions& options) override {
207     rtc::Buffer packet(reinterpret_cast<const uint8_t*>(data), len,
208                        kMaxRtpPacketLen);
209     rtc::PacketOptions rtc_options;
210     rtc_options.packet_id = options.packet_id;
211     return VoiceMediaChannel::SendPacket(&packet, rtc_options);
212   }
213 
SendRtcp(const uint8_t * data,size_t len)214   bool SendRtcp(const uint8_t* data, size_t len) override {
215     rtc::Buffer packet(reinterpret_cast<const uint8_t*>(data), len,
216                        kMaxRtpPacketLen);
217     return VoiceMediaChannel::SendRtcp(&packet, rtc::PacketOptions());
218   }
219 
220   int GetReceiveChannelId(uint32_t ssrc) const;
221   int GetSendChannelId(uint32_t ssrc) const;
222 
223  private:
224   bool SetSendCodecs(const std::vector<AudioCodec>& codecs);
225   bool SetOptions(const AudioOptions& options);
226   bool SetMaxSendBandwidth(int bps);
227   bool SetRecvCodecs(const std::vector<AudioCodec>& codecs);
228   bool SetLocalRenderer(uint32_t ssrc, AudioRenderer* renderer);
229   bool MuteStream(uint32_t ssrc, bool mute);
230 
engine()231   WebRtcVoiceEngine* engine() { return engine_; }
GetLastEngineError()232   int GetLastEngineError() { return engine()->GetLastEngineError(); }
233   int GetOutputLevel(int channel);
234   bool GetRedSendCodec(const AudioCodec& red_codec,
235                        const std::vector<AudioCodec>& all_codecs,
236                        webrtc::CodecInst* send_codec);
237   bool SetPlayout(int channel, bool playout);
238   void SetNack(int channel, bool nack_enabled);
239   bool SetSendCodec(int channel, const webrtc::CodecInst& send_codec);
240   bool ChangePlayout(bool playout);
241   bool ChangeSend(SendFlags send);
242   bool ChangeSend(int channel, SendFlags send);
243   int CreateVoEChannel();
244   bool DeleteVoEChannel(int channel);
IsDefaultRecvStream(uint32_t ssrc)245   bool IsDefaultRecvStream(uint32_t ssrc) {
246     return default_recv_ssrc_ == static_cast<int64_t>(ssrc);
247   }
248   bool SetSendCodecs(int channel, const std::vector<AudioCodec>& codecs);
249   bool SetSendBitrateInternal(int bps);
250 
251   rtc::ThreadChecker worker_thread_checker_;
252 
253   WebRtcVoiceEngine* const engine_ = nullptr;
254   std::vector<AudioCodec> recv_codecs_;
255   std::vector<AudioCodec> send_codecs_;
256   rtc::scoped_ptr<webrtc::CodecInst> send_codec_;
257   bool send_bitrate_setting_ = false;
258   int send_bitrate_bps_ = 0;
259   AudioOptions options_;
260   rtc::Optional<int> dtmf_payload_type_;
261   bool desired_playout_ = false;
262   bool nack_enabled_ = false;
263   bool playout_ = false;
264   SendFlags desired_send_ = SEND_NOTHING;
265   SendFlags send_ = SEND_NOTHING;
266   webrtc::Call* const call_ = nullptr;
267 
268   // SSRC of unsignalled receive stream, or -1 if there isn't one.
269   int64_t default_recv_ssrc_ = -1;
270   // Volume for unsignalled stream, which may be set before the stream exists.
271   double default_recv_volume_ = 1.0;
272   // Default SSRC to use for RTCP receiver reports in case of no signaled
273   // send streams. See: https://code.google.com/p/webrtc/issues/detail?id=4740
274   // and https://code.google.com/p/chromium/issues/detail?id=547661
275   uint32_t receiver_reports_ssrc_ = 0xFA17FA17u;
276 
277   class WebRtcAudioSendStream;
278   std::map<uint32_t, WebRtcAudioSendStream*> send_streams_;
279   std::vector<webrtc::RtpExtension> send_rtp_extensions_;
280 
281   class WebRtcAudioReceiveStream;
282   std::map<uint32_t, WebRtcAudioReceiveStream*> recv_streams_;
283   std::vector<webrtc::RtpExtension> recv_rtp_extensions_;
284 
285   RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(WebRtcVoiceMediaChannel);
286 };
287 }  // namespace cricket
288 
289 #endif  // TALK_MEDIA_WEBRTCVOICEENGINE_H_
290