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 |