1 files changed, 91 insertions, 0 deletions
diff --git a/third_party/libwebrtc/video/end_to_end_tests/frame_encryption_tests.cc b/third_party/libwebrtc/video/end_to_end_tests/frame_encryption_tests.cc
new file mode 100644
index 0000000000..6a1b16927c
--- /dev/null
+++ b/third_party/libwebrtc/video/end_to_end_tests/frame_encryption_tests.cc
@@ -0,0 +1,91 @@
+/*
+ *  Copyright 2018 The WebRTC project authors. All Rights Reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "api/test/fake_frame_decryptor.h"
+#include "api/test/fake_frame_encryptor.h"
+#include "media/engine/internal_decoder_factory.h"
+#include "modules/rtp_rtcp/source/rtp_dependency_descriptor_extension.h"
+#include "modules/video_coding/codecs/vp8/include/vp8.h"
+#include "test/call_test.h"
+#include "test/gtest.h"
+
+namespace webrtc {
+namespace {
+
+using FrameEncryptionEndToEndTest = test::CallTest;
+
+enum : int {  // The first valid value is 1.
+  kGenericDescriptorExtensionId = 1,
+};
+
+class DecryptedFrameObserver : public test::EndToEndTest,
+                               public rtc::VideoSinkInterface<VideoFrame> {
+ public:
+  DecryptedFrameObserver()
+      : EndToEndTest(test::CallTest::kDefaultTimeout),
+        encoder_factory_([] { return VP8Encoder::Create(); }) {}
+
+ private:
+  void ModifyVideoConfigs(
+      VideoSendStream::Config* send_config,
+      std::vector<VideoReceiveStreamInterface::Config>* receive_configs,
+      VideoEncoderConfig* encoder_config) override {
+    // Use VP8 instead of FAKE.
+    send_config->encoder_settings.encoder_factory = &encoder_factory_;
+    send_config->rtp.payload_name = "VP8";
+    send_config->rtp.payload_type = test::CallTest::kVideoSendPayloadType;
+    send_config->frame_encryptor = new FakeFrameEncryptor();
+    send_config->crypto_options.sframe.require_frame_encryption = true;
+    encoder_config->codec_type = kVideoCodecVP8;
+    VideoReceiveStreamInterface::Decoder decoder =
+        test::CreateMatchingDecoder(*send_config);
+    for (auto& recv_config : *receive_configs) {
+      recv_config.decoder_factory = &decoder_factory_;
+      recv_config.decoders.clear();
+      recv_config.decoders.push_back(decoder);
+      recv_config.renderer = this;
+      recv_config.frame_decryptor = rtc::make_ref_counted<FakeFrameDecryptor>();
+      recv_config.crypto_options.sframe.require_frame_encryption = true;
+    }
+  }
+
+  void OnFrame(const VideoFrame& video_frame) override {
+    observation_complete_.Set();
+  }
+
+  void PerformTest() override {
+    EXPECT_TRUE(Wait())
+        << "Timed out waiting for decrypted frames to be rendered.";
+  }
+
+  std::unique_ptr<VideoEncoder> encoder_;
+  test::FunctionVideoEncoderFactory encoder_factory_;
+  InternalDecoderFactory decoder_factory_;
+};
+
+// Validates that payloads cannot be sent without a frame encryptor and frame
+// decryptor attached.
+TEST_F(FrameEncryptionEndToEndTest,
+       WithGenericFrameDescriptorRequireFrameEncryptionEnforced) {
+  RegisterRtpExtension(RtpExtension(RtpExtension::kGenericFrameDescriptorUri00,
+                                    kGenericDescriptorExtensionId));
+  DecryptedFrameObserver test;
+  RunBaseTest(&test);
+}
+
+TEST_F(FrameEncryptionEndToEndTest,
+       WithDependencyDescriptorRequireFrameEncryptionEnforced) {
+  RegisterRtpExtension(RtpExtension(RtpExtension::kDependencyDescriptorUri,
+                                    kGenericDescriptorExtensionId));
+  DecryptedFrameObserver test;
+  RunBaseTest(&test);
+}
+}  // namespace
+}  // namespace webrtc