sync_reader_unittest.cc 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "services/audio/sync_reader.h"
  5. #include <limits>
  6. #include <memory>
  7. #include <string>
  8. #include <type_traits>
  9. #include <utility>
  10. #include "base/bind.h"
  11. #include "base/memory/unsafe_shared_memory_region.h"
  12. #include "base/sync_socket.h"
  13. #include "base/test/task_environment.h"
  14. #include "base/time/time.h"
  15. #include "media/base/audio_bus.h"
  16. #include "media/base/audio_parameters.h"
  17. #include "testing/gmock/include/gmock/gmock.h"
  18. #include "testing/gtest/include/gtest/gtest.h"
  19. using ::testing::TestWithParam;
  20. using media::AudioBus;
  21. using media::AudioOutputBuffer;
  22. using media::AudioOutputBufferParameters;
  23. using media::AudioParameters;
  24. namespace audio {
  25. void NoLog(const std::string&) {}
  26. static_assert(
  27. std::is_unsigned<
  28. decltype(AudioOutputBufferParameters::bitstream_data_size)>::value,
  29. "If |bitstream_data_size| is ever made signed, add tests for negative "
  30. "buffer sizes.");
  31. enum OverflowTestCase {
  32. kZero,
  33. kNoOverflow,
  34. kOverflowByOne,
  35. kOverflowByOneThousand,
  36. kOverflowByMax
  37. };
  38. static const OverflowTestCase overflow_test_case_values[]{
  39. kZero, kNoOverflow, kOverflowByOne, kOverflowByOneThousand, kOverflowByMax};
  40. class SyncReaderBitstreamTest : public TestWithParam<OverflowTestCase> {
  41. public:
  42. SyncReaderBitstreamTest() {}
  43. ~SyncReaderBitstreamTest() override {}
  44. private:
  45. base::test::TaskEnvironment env_;
  46. };
  47. TEST_P(SyncReaderBitstreamTest, BitstreamBufferOverflow_DoesNotWriteOOB) {
  48. const int kSampleRate = 44100;
  49. const int kFramesPerBuffer = 1;
  50. AudioParameters params(AudioParameters::AUDIO_BITSTREAM_AC3,
  51. media::CHANNEL_LAYOUT_STEREO, kSampleRate,
  52. kFramesPerBuffer);
  53. auto socket = std::make_unique<base::CancelableSyncSocket>();
  54. SyncReader reader(base::BindRepeating(&NoLog), params, socket.get());
  55. ASSERT_TRUE(reader.IsValid());
  56. const base::WritableSharedMemoryMapping shmem =
  57. reader.TakeSharedMemoryRegion().Map();
  58. ASSERT_TRUE(shmem.IsValid());
  59. auto* const buffer =
  60. reinterpret_cast<media::AudioOutputBuffer*>(shmem.memory());
  61. ASSERT_TRUE(buffer);
  62. reader.RequestMoreData(base::TimeDelta(), base::TimeTicks(), 0);
  63. uint32_t signal;
  64. EXPECT_EQ(socket->Receive(&signal, sizeof(signal)), sizeof(signal));
  65. // So far, this is an ordinary stream.
  66. // Now |reader| expects data to be written to the shared memory. The renderer
  67. // says how much data was written.
  68. switch (GetParam()) {
  69. case kZero:
  70. buffer->params.bitstream_data_size = 0;
  71. break;
  72. case kNoOverflow:
  73. buffer->params.bitstream_data_size =
  74. shmem.mapped_size() - sizeof(AudioOutputBufferParameters);
  75. break;
  76. case kOverflowByOne:
  77. buffer->params.bitstream_data_size =
  78. shmem.mapped_size() - sizeof(AudioOutputBufferParameters) + 1;
  79. break;
  80. case kOverflowByOneThousand:
  81. buffer->params.bitstream_data_size =
  82. shmem.mapped_size() - sizeof(AudioOutputBufferParameters) + 1000;
  83. break;
  84. case kOverflowByMax:
  85. buffer->params.bitstream_data_size = std::numeric_limits<decltype(
  86. buffer->params.bitstream_data_size)>::max();
  87. break;
  88. }
  89. ++signal;
  90. EXPECT_EQ(socket->Send(&signal, sizeof(signal)), sizeof(signal));
  91. // The purpose of the test is to ensure this call doesn't result in undefined
  92. // behavior, which should be verified by sanitizers.
  93. std::unique_ptr<AudioBus> output_bus = AudioBus::Create(params);
  94. reader.Read(output_bus.get(), false);
  95. }
  96. INSTANTIATE_TEST_SUITE_P(All,
  97. SyncReaderBitstreamTest,
  98. ::testing::ValuesIn(overflow_test_case_values));
  99. } // namespace audio