Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 /* | |
| 2 * Copyright (c) 2017 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 | |
| 11 #include <math.h> | |
| 12 #include <string.h> | |
| 13 | |
| 14 #include <algorithm> | |
| 15 #include <bitset> | |
| 16 #include <vector> | |
| 17 | |
| 18 #include "webrtc/base/checks.h" | |
| 19 #include "webrtc/modules/audio_processing/residual_echo_detector.h" | |
| 20 | |
| 21 namespace webrtc { | |
| 22 | |
| 23 void FuzzOneInput(const uint8_t* data, size_t size) { | |
| 24 RTC_DCHECK_EQ(sizeof(float), sizeof(unsigned int)); | |
|
hlundin-webrtc
2017/05/12 07:32:31
Make this a static_assert.
And, do you want to ma
ivoc
2017/05/12 09:43:22
Good question. I added this DCHECK earlier when I
hlundin-webrtc
2017/05/12 11:39:48
Acknowledged.
| |
| 25 // Number of times to update the echo detector. | |
| 26 constexpr size_t kNrOfUpdates = 7; | |
| 27 // Each round of updates requires a call to both AnalyzeRender and | |
| 28 // AnalyzeCapture, so the amount of needed input bytes doubles. Also, two | |
| 29 // bytes are used to set the call order. | |
| 30 constexpr size_t kNrOfNeededInputBytes = 2 * kNrOfUpdates * sizeof(float) + 2; | |
| 31 // The maximum audio energy that an audio frame can have is equal to the | |
| 32 // number of samples in the frame multiplied by 2^30. We use a single sample | |
| 33 // to represent an audio frame in this test, so it should have a maximum value | |
| 34 // equal to the square root of that value. | |
| 35 const float maxFuzzedValue = sqrtf(20 * 48) * 32768; | |
| 36 if (size < kNrOfNeededInputBytes) { | |
| 37 return; | |
| 38 } | |
| 39 // Use the first two bytes to choose the call order. | |
| 40 uint16_t call_order_int; | |
| 41 memcpy(&call_order_int, &data[0], 2); | |
|
hlundin-webrtc
2017/05/12 07:32:31
I would like to see either a read index (size_t) o
ivoc
2017/05/12 09:43:22
Good idea, I updated the code to use a read_ix.
| |
| 42 std::bitset<16> call_order(call_order_int); | |
| 43 | |
| 44 ResidualEchoDetector echo_detector; | |
| 45 std::vector<float> input(1); | |
| 46 // Call AnalyzeCaptureAudio once to prevent the flushing of the buffer. | |
| 47 echo_detector.AnalyzeCaptureAudio(input); | |
| 48 for (size_t i = 0; i < 2 * kNrOfUpdates; ++i) { | |
| 49 // Convert 4 input bytes to a float. | |
| 50 memcpy(input.data(), &data[i * 4 + 2], 4); | |
|
hlundin-webrtc
2017/05/12 07:32:31
Add a DCHECK that you are not reading outside of t
ivoc
2017/05/12 09:43:22
Done.
| |
| 51 if (!isfinite(input[0]) || fabs(input[0]) > maxFuzzedValue) { | |
| 52 // Ignore infinity, nan values and values that are unrealistically large. | |
| 53 continue; | |
| 54 } | |
| 55 if (call_order[i]) { | |
| 56 echo_detector.AnalyzeRenderAudio(input); | |
| 57 } else { | |
| 58 echo_detector.AnalyzeCaptureAudio(input); | |
| 59 } | |
| 60 } | |
| 61 } | |
| 62 | |
| 63 } // namespace webrtc | |
| OLD | NEW |