kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2014 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 | |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 11 | #include "common_audio/wav_header.h" |
| 12 | |
Raphael Kubo da Costa | 7ce3091 | 2018-04-16 11:17:10 +0200 | [diff] [blame] | 13 | #include <string.h> |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 14 | |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 15 | #include <limits> |
| 16 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 17 | #include "test/gtest.h" |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 18 | |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 19 | namespace webrtc { |
| 20 | |
| 21 | // Doesn't take ownership of the buffer. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 22 | class WavHeaderBufferReader : public WavHeaderReader { |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 23 | public: |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 24 | WavHeaderBufferReader(const uint8_t* buf, size_t size, bool check_read_size) |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 25 | : buf_(buf), |
| 26 | size_(size), |
| 27 | pos_(0), |
| 28 | buf_exhausted_(false), |
| 29 | check_read_size_(check_read_size) {} |
| 30 | |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 31 | ~WavHeaderBufferReader() override { |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 32 | // Verify the entire buffer has been read. |
| 33 | if (check_read_size_) |
| 34 | EXPECT_EQ(size_, pos_); |
| 35 | } |
| 36 | |
Mirko Bonadei | 91df091 | 2018-07-17 11:08:15 +0200 | [diff] [blame] | 37 | size_t Read(void* buf, size_t num_bytes) override { |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 38 | EXPECT_FALSE(buf_exhausted_); |
| 39 | |
| 40 | const size_t bytes_remaining = size_ - pos_; |
| 41 | if (num_bytes > bytes_remaining) { |
| 42 | // The caller is signalled about an exhausted buffer when we return fewer |
| 43 | // bytes than requested. There should not be another read attempt after |
| 44 | // this point. |
| 45 | buf_exhausted_ = true; |
| 46 | num_bytes = bytes_remaining; |
| 47 | } |
| 48 | memcpy(buf, &buf_[pos_], num_bytes); |
| 49 | pos_ += num_bytes; |
| 50 | return num_bytes; |
| 51 | } |
| 52 | |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 53 | bool SeekForward(uint32_t num_bytes) override { |
| 54 | // Verify we don't try to read outside of a properly sized header. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 55 | if (size_ >= kPcmWavHeaderSize) |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 56 | EXPECT_GE(size_, pos_ + num_bytes); |
| 57 | EXPECT_FALSE(buf_exhausted_); |
| 58 | |
| 59 | const size_t bytes_remaining = size_ - pos_; |
| 60 | if (num_bytes > bytes_remaining) { |
| 61 | // Error: cannot seek beyond EOF. |
| 62 | return false; |
| 63 | } |
| 64 | if (num_bytes == bytes_remaining) { |
| 65 | // There should not be another read attempt after this point. |
| 66 | buf_exhausted_ = true; |
| 67 | } |
| 68 | pos_ += num_bytes; |
| 69 | return true; |
| 70 | } |
| 71 | |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 72 | int64_t GetPosition() override { return pos_; } |
| 73 | |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 74 | private: |
| 75 | const uint8_t* buf_; |
| 76 | const size_t size_; |
| 77 | size_t pos_; |
| 78 | bool buf_exhausted_; |
| 79 | const bool check_read_size_; |
| 80 | }; |
| 81 | |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 82 | // Try various choices of WAV header parameters, and make sure that the good |
| 83 | // ones are accepted and the bad ones rejected. |
| 84 | TEST(WavHeaderTest, CheckWavParameters) { |
| 85 | // Try some really stupid values for one parameter at a time. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 86 | EXPECT_TRUE(CheckWavParameters(1, 8000, WavFormat::kWavFormatPcm, 0)); |
| 87 | EXPECT_FALSE(CheckWavParameters(0, 8000, WavFormat::kWavFormatPcm, 0)); |
| 88 | EXPECT_FALSE(CheckWavParameters(0x10000, 8000, WavFormat::kWavFormatPcm, 0)); |
| 89 | EXPECT_FALSE(CheckWavParameters(1, 0, WavFormat::kWavFormatPcm, 0)); |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 90 | |
| 91 | // Too large values. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 92 | EXPECT_FALSE( |
| 93 | CheckWavParameters(1 << 20, 1 << 20, WavFormat::kWavFormatPcm, 0)); |
| 94 | EXPECT_FALSE(CheckWavParameters(1, 8000, WavFormat::kWavFormatPcm, |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 95 | std::numeric_limits<uint32_t>::max())); |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 96 | |
| 97 | // Not the same number of samples for each channel. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 98 | EXPECT_FALSE(CheckWavParameters(3, 8000, WavFormat::kWavFormatPcm, 5)); |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 99 | } |
| 100 | |
andrew@webrtc.org | f866b2d | 2014-11-03 18:20:06 +0000 | [diff] [blame] | 101 | TEST(WavHeaderTest, ReadWavHeaderWithErrors) { |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 102 | size_t num_channels = 0; |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 103 | int sample_rate = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 104 | WavFormat format = WavFormat::kWavFormatPcm; |
pkasting | 25702cb | 2016-01-08 13:50:27 -0800 | [diff] [blame] | 105 | size_t bytes_per_sample = 0; |
| 106 | size_t num_samples = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 107 | int64_t data_start_pos = 0; |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 108 | |
| 109 | // Test a few ways the header can be invalid. We start with the valid header |
| 110 | // used in WriteAndReadWavHeader, and invalidate one field per test. The |
| 111 | // invalid field is indicated in the array name, and in the comments with |
| 112 | // *BAD*. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 113 | { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 114 | constexpr uint8_t kBadRiffID[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 115 | // clang-format off |
| 116 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 117 | 'R', 'i', 'f', 'f', // *BAD* |
| 118 | 0xbd, 0xd0, 0x5b, 0x07, // size of whole file - 8: 123457689 + 44 - 8 |
| 119 | 'W', 'A', 'V', 'E', |
| 120 | 'f', 'm', 't', ' ', |
| 121 | 16, 0, 0, 0, // size of fmt block - 8: 24 - 8 |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 122 | 1, 0, // format: PCM (1) |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 123 | 17, 0, // channels: 17 |
| 124 | 0x39, 0x30, 0, 0, // sample rate: 12345 |
| 125 | 0xc9, 0x33, 0x03, 0, // byte rate: 1 * 17 * 12345 |
| 126 | 17, 0, // block align: NumChannels * BytesPerSample |
| 127 | 8, 0, // bits per sample: 1 * 8 |
| 128 | 'd', 'a', 't', 'a', |
| 129 | 0x99, 0xd0, 0x5b, 0x07, // size of payload: 123457689 |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 130 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 131 | }; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 132 | WavHeaderBufferReader r(kBadRiffID, sizeof(kBadRiffID), |
| 133 | /*check_read_size=*/false); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 134 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 135 | &bytes_per_sample, &num_samples, |
| 136 | &data_start_pos)); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 137 | } |
| 138 | { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 139 | constexpr uint8_t kBadBitsPerSample[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 140 | // clang-format off |
| 141 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 142 | 'R', 'I', 'F', 'F', |
| 143 | 0xbd, 0xd0, 0x5b, 0x07, // size of whole file - 8: 123457689 + 44 - 8 |
| 144 | 'W', 'A', 'V', 'E', |
| 145 | 'f', 'm', 't', ' ', |
| 146 | 16, 0, 0, 0, // size of fmt block - 8: 24 - 8 |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 147 | 1, 0, // format: PCM (1) |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 148 | 17, 0, // channels: 17 |
| 149 | 0x39, 0x30, 0, 0, // sample rate: 12345 |
| 150 | 0xc9, 0x33, 0x03, 0, // byte rate: 1 * 17 * 12345 |
| 151 | 17, 0, // block align: NumChannels * BytesPerSample |
| 152 | 1, 0, // bits per sample: *BAD* |
| 153 | 'd', 'a', 't', 'a', |
| 154 | 0x99, 0xd0, 0x5b, 0x07, // size of payload: 123457689 |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 155 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 156 | }; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 157 | WavHeaderBufferReader r(kBadBitsPerSample, sizeof(kBadBitsPerSample), |
| 158 | /*check_read_size=*/true); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 159 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 160 | &bytes_per_sample, &num_samples, |
| 161 | &data_start_pos)); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 162 | } |
| 163 | { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 164 | constexpr uint8_t kBadByteRate[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 165 | // clang-format off |
| 166 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 167 | 'R', 'I', 'F', 'F', |
| 168 | 0xbd, 0xd0, 0x5b, 0x07, // size of whole file - 8: 123457689 + 44 - 8 |
| 169 | 'W', 'A', 'V', 'E', |
| 170 | 'f', 'm', 't', ' ', |
| 171 | 16, 0, 0, 0, // size of fmt block - 8: 24 - 8 |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 172 | 1, 0, // format: PCM (1) |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 173 | 17, 0, // channels: 17 |
| 174 | 0x39, 0x30, 0, 0, // sample rate: 12345 |
| 175 | 0x00, 0x33, 0x03, 0, // byte rate: *BAD* |
| 176 | 17, 0, // block align: NumChannels * BytesPerSample |
| 177 | 8, 0, // bits per sample: 1 * 8 |
| 178 | 'd', 'a', 't', 'a', |
| 179 | 0x99, 0xd0, 0x5b, 0x07, // size of payload: 123457689 |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 180 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 181 | }; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 182 | WavHeaderBufferReader r(kBadByteRate, sizeof(kBadByteRate), |
| 183 | /*check_read_size=*/true); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 184 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 185 | &bytes_per_sample, &num_samples, |
| 186 | &data_start_pos)); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 187 | } |
| 188 | { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 189 | constexpr uint8_t kBadFmtHeaderSize[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 190 | // clang-format off |
| 191 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 192 | 'R', 'I', 'F', 'F', |
| 193 | 0xbd, 0xd0, 0x5b, 0x07, // size of whole file - 8: 123457689 + 44 - 8 |
| 194 | 'W', 'A', 'V', 'E', |
| 195 | 'f', 'm', 't', ' ', |
| 196 | 17, 0, 0, 0, // size of fmt block *BAD*. Only 16 and 18 permitted. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 197 | 1, 0, // format: PCM (1) |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 198 | 17, 0, // channels: 17 |
| 199 | 0x39, 0x30, 0, 0, // sample rate: 12345 |
| 200 | 0xc9, 0x33, 0x03, 0, // byte rate: 1 * 17 * 12345 |
| 201 | 17, 0, // block align: NumChannels * BytesPerSample |
| 202 | 8, 0, // bits per sample: 1 * 8 |
| 203 | 0, // extra (though invalid) header byte |
| 204 | 'd', 'a', 't', 'a', |
| 205 | 0x99, 0xd0, 0x5b, 0x07, // size of payload: 123457689 |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 206 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 207 | }; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 208 | WavHeaderBufferReader r(kBadFmtHeaderSize, sizeof(kBadFmtHeaderSize), |
| 209 | /*check_read_size=*/false); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 210 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 211 | &bytes_per_sample, &num_samples, |
| 212 | &data_start_pos)); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 213 | } |
| 214 | { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 215 | constexpr uint8_t kNonZeroExtensionField[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 216 | // clang-format off |
| 217 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 218 | 'R', 'I', 'F', 'F', |
| 219 | 0xbd, 0xd0, 0x5b, 0x07, // size of whole file - 8: 123457689 + 44 - 8 |
| 220 | 'W', 'A', 'V', 'E', |
| 221 | 'f', 'm', 't', ' ', |
| 222 | 18, 0, 0, 0, // size of fmt block - 8: 24 - 8 |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 223 | 1, 0, // format: PCM (1) |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 224 | 17, 0, // channels: 17 |
| 225 | 0x39, 0x30, 0, 0, // sample rate: 12345 |
| 226 | 0xc9, 0x33, 0x03, 0, // byte rate: 1 * 17 * 12345 |
| 227 | 17, 0, // block align: NumChannels * BytesPerSample |
| 228 | 8, 0, // bits per sample: 1 * 8 |
| 229 | 1, 0, // non-zero extension field *BAD* |
| 230 | 'd', 'a', 't', 'a', |
| 231 | 0x99, 0xd0, 0x5b, 0x07, // size of payload: 123457689 |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 232 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 233 | }; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 234 | WavHeaderBufferReader r(kNonZeroExtensionField, |
| 235 | sizeof(kNonZeroExtensionField), |
| 236 | /*check_read_size=*/false); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 237 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 238 | &bytes_per_sample, &num_samples, |
| 239 | &data_start_pos)); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 240 | } |
| 241 | { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 242 | constexpr uint8_t kMissingDataChunk[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 243 | // clang-format off |
| 244 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 245 | 'R', 'I', 'F', 'F', |
| 246 | 0xbd, 0xd0, 0x5b, 0x07, // size of whole file - 8: 123457689 + 44 - 8 |
| 247 | 'W', 'A', 'V', 'E', |
| 248 | 'f', 'm', 't', ' ', |
| 249 | 16, 0, 0, 0, // size of fmt block - 8: 24 - 8 |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 250 | 1, 0, // format: PCM (1) |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 251 | 17, 0, // channels: 17 |
| 252 | 0x39, 0x30, 0, 0, // sample rate: 12345 |
| 253 | 0xc9, 0x33, 0x03, 0, // byte rate: 1 * 17 * 12345 |
| 254 | 17, 0, // block align: NumChannels * BytesPerSample |
| 255 | 8, 0, // bits per sample: 1 * 8 |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 256 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 257 | }; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 258 | WavHeaderBufferReader r(kMissingDataChunk, sizeof(kMissingDataChunk), |
| 259 | /*check_read_size=*/true); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 260 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 261 | &bytes_per_sample, &num_samples, |
| 262 | &data_start_pos)); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 263 | } |
| 264 | { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 265 | constexpr uint8_t kMissingFmtAndDataChunks[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 266 | // clang-format off |
| 267 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 268 | 'R', 'I', 'F', 'F', |
| 269 | 0xbd, 0xd0, 0x5b, 0x07, // size of whole file - 8: 123457689 + 44 - 8 |
| 270 | 'W', 'A', 'V', 'E', |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 271 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 272 | }; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 273 | WavHeaderBufferReader r(kMissingFmtAndDataChunks, |
| 274 | sizeof(kMissingFmtAndDataChunks), |
| 275 | /*check_read_size=*/true); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 276 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 277 | &bytes_per_sample, &num_samples, |
| 278 | &data_start_pos)); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 279 | } |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 280 | } |
| 281 | |
andrew@webrtc.org | f866b2d | 2014-11-03 18:20:06 +0000 | [diff] [blame] | 282 | // Try writing and reading a valid WAV header and make sure it looks OK. |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 283 | TEST(WavHeaderTest, WriteAndReadWavHeader) { |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 284 | constexpr int kSize = 4 + kPcmWavHeaderSize + 4; |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 285 | uint8_t buf[kSize]; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 286 | size_t header_size; |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 287 | memset(buf, 0xa4, sizeof(buf)); |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 288 | WriteWavHeader(17, 12345, WavFormat::kWavFormatPcm, 123457689, buf + 4, |
| 289 | &header_size); |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 290 | constexpr uint8_t kExpectedBuf[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 291 | // clang-format off |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 292 | // clang formatting doesn't respect inline comments. |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 293 | 0xa4, 0xa4, 0xa4, 0xa4, // untouched bytes before header |
| 294 | 'R', 'I', 'F', 'F', |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 295 | 0x56, 0xa1, 0xb7, 0x0e, // size of whole file - 8: 123457689 + 44 - 8 |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 296 | 'W', 'A', 'V', 'E', |
| 297 | 'f', 'm', 't', ' ', |
| 298 | 16, 0, 0, 0, // size of fmt block - 8: 24 - 8 |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 299 | 1, 0, // format: PCM (1) |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 300 | 17, 0, // channels: 17 |
| 301 | 0x39, 0x30, 0, 0, // sample rate: 12345 |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 302 | 0x92, 0x67, 0x06, 0, // byte rate: 2 * 17 * 12345 |
| 303 | 34, 0, // block align: NumChannels * BytesPerSample |
| 304 | 16, 0, // bits per sample: 2 * 8 |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 305 | 'd', 'a', 't', 'a', |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 306 | 0x32, 0xa1, 0xb7, 0x0e, // size of payload: 2 * 123457689 |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 307 | 0xa4, 0xa4, 0xa4, 0xa4, // untouched bytes after header |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 308 | // clang-format on |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 309 | }; |
kwiberg@webrtc.org | 2ebfac5 | 2015-01-14 10:51:54 +0000 | [diff] [blame] | 310 | static_assert(sizeof(kExpectedBuf) == kSize, "buffer size"); |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 311 | EXPECT_EQ(0, memcmp(kExpectedBuf, buf, kSize)); |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 312 | |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 313 | size_t num_channels = 0; |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 314 | int sample_rate = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 315 | WavFormat format = WavFormat::kWavFormatPcm; |
pkasting | 25702cb | 2016-01-08 13:50:27 -0800 | [diff] [blame] | 316 | size_t bytes_per_sample = 0; |
| 317 | size_t num_samples = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 318 | int64_t data_start_pos = 0; |
| 319 | WavHeaderBufferReader r(buf + 4, sizeof(buf) - 8, |
| 320 | /*check_read_size=*/true); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 321 | EXPECT_TRUE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 322 | &bytes_per_sample, &num_samples, &data_start_pos)); |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 323 | EXPECT_EQ(17u, num_channels); |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 324 | EXPECT_EQ(12345, sample_rate); |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 325 | EXPECT_EQ(WavFormat::kWavFormatPcm, format); |
| 326 | EXPECT_EQ(2u, bytes_per_sample); |
andrew@webrtc.org | a3ed713 | 2014-10-31 21:51:03 +0000 | [diff] [blame] | 327 | EXPECT_EQ(123457689u, num_samples); |
kwiberg@webrtc.org | 877083c | 2014-08-20 07:42:46 +0000 | [diff] [blame] | 328 | } |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 329 | |
| 330 | // Try reading an atypical but valid WAV header and make sure it's parsed OK. |
| 331 | TEST(WavHeaderTest, ReadAtypicalWavHeader) { |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 332 | constexpr uint8_t kBuf[] = { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 333 | // clang-format off |
| 334 | // clang formatting doesn't respect inline comments. |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 335 | 'R', 'I', 'F', 'F', |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 336 | 0xbf, 0xd0, 0x5b, 0x07, // Size of whole file - 8 + extra 2 bytes of zero |
| 337 | // extension: 123457689 + 44 - 8 + 2 (atypical). |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 338 | 'W', 'A', 'V', 'E', |
| 339 | 'f', 'm', 't', ' ', |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 340 | 18, 0, 0, 0, // Size of fmt block (with an atypical extension |
| 341 | // size field). |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 342 | 1, 0, // Format: PCM (1). |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 343 | 17, 0, // Channels: 17. |
| 344 | 0x39, 0x30, 0, 0, // Sample rate: 12345. |
| 345 | 0xc9, 0x33, 0x03, 0, // Byte rate: 1 * 17 * 12345. |
| 346 | 17, 0, // Block align: NumChannels * BytesPerSample. |
| 347 | 8, 0, // Bits per sample: 1 * 8. |
| 348 | 0, 0, // Zero extension size field (atypical). |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 349 | 'd', 'a', 't', 'a', |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 350 | 0x99, 0xd0, 0x5b, 0x07, // Size of payload: 123457689. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 351 | // clang-format on |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 352 | }; |
| 353 | |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 354 | size_t num_channels = 0; |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 355 | int sample_rate = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 356 | WavFormat format = WavFormat::kWavFormatPcm; |
pkasting | 25702cb | 2016-01-08 13:50:27 -0800 | [diff] [blame] | 357 | size_t bytes_per_sample = 0; |
| 358 | size_t num_samples = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 359 | int64_t data_start_pos = 0; |
| 360 | WavHeaderBufferReader r(kBuf, sizeof(kBuf), /*check_read_size=*/true); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 361 | EXPECT_TRUE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 362 | &bytes_per_sample, &num_samples, &data_start_pos)); |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 363 | EXPECT_EQ(17u, num_channels); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 364 | EXPECT_EQ(12345, sample_rate); |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 365 | EXPECT_EQ(WavFormat::kWavFormatPcm, format); |
pkasting | 25702cb | 2016-01-08 13:50:27 -0800 | [diff] [blame] | 366 | EXPECT_EQ(1u, bytes_per_sample); |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 367 | EXPECT_EQ(123457689u, num_samples); |
| 368 | } |
| 369 | |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 370 | // Try reading a valid WAV header which contains an optional chunk and make sure |
| 371 | // it's parsed OK. |
| 372 | TEST(WavHeaderTest, ReadWavHeaderWithOptionalChunk) { |
| 373 | constexpr uint8_t kBuf[] = { |
| 374 | // clang-format off |
| 375 | // clang formatting doesn't respect inline comments. |
| 376 | 'R', 'I', 'F', 'F', |
| 377 | 0xcd, 0xd0, 0x5b, 0x07, // Size of whole file - 8 + an extra 16 bytes of |
| 378 | // "metadata" (8 bytes header, 16 bytes payload): |
| 379 | // 123457689 + 44 - 8 + 16. |
| 380 | 'W', 'A', 'V', 'E', |
| 381 | 'f', 'm', 't', ' ', |
| 382 | 16, 0, 0, 0, // Size of fmt block. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 383 | 1, 0, // Format: PCM (1). |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 384 | 17, 0, // Channels: 17. |
| 385 | 0x39, 0x30, 0, 0, // Sample rate: 12345. |
| 386 | 0xc9, 0x33, 0x03, 0, // Byte rate: 1 * 17 * 12345. |
| 387 | 17, 0, // Block align: NumChannels * BytesPerSample. |
| 388 | 8, 0, // Bits per sample: 1 * 8. |
| 389 | 'L', 'I', 'S', 'T', // Metadata chunk ID. |
| 390 | 16, 0, 0, 0, // Metadata chunk payload size. |
| 391 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // Metadata (16 bytes). |
| 392 | 'd', 'a', 't', 'a', |
| 393 | 0x99, 0xd0, 0x5b, 0x07, // Size of payload: 123457689. |
| 394 | // clang-format on |
| 395 | }; |
| 396 | |
| 397 | size_t num_channels = 0; |
| 398 | int sample_rate = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 399 | WavFormat format = WavFormat::kWavFormatPcm; |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 400 | size_t bytes_per_sample = 0; |
| 401 | size_t num_samples = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 402 | int64_t data_start_pos = 0; |
| 403 | WavHeaderBufferReader r(kBuf, sizeof(kBuf), /*check_read_size=*/true); |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 404 | EXPECT_TRUE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 405 | &bytes_per_sample, &num_samples, &data_start_pos)); |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 406 | EXPECT_EQ(17u, num_channels); |
| 407 | EXPECT_EQ(12345, sample_rate); |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 408 | EXPECT_EQ(WavFormat::kWavFormatPcm, format); |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 409 | EXPECT_EQ(1u, bytes_per_sample); |
| 410 | EXPECT_EQ(123457689u, num_samples); |
| 411 | } |
| 412 | |
| 413 | // Try reading an invalid WAV header which has the the data chunk before the |
| 414 | // format one and make sure it's not parsed. |
| 415 | TEST(WavHeaderTest, ReadWavHeaderWithDataBeforeFormat) { |
| 416 | constexpr uint8_t kBuf[] = { |
| 417 | // clang-format off |
| 418 | // clang formatting doesn't respect inline comments. |
| 419 | 'R', 'I', 'F', 'F', |
| 420 | 52, 0, 0, 0, // Size of whole file - 8: 16 + 44 - 8. |
| 421 | 'W', 'A', 'V', 'E', |
| 422 | 'd', 'a', 't', 'a', |
| 423 | 16, 0, 0, 0, // Data chunk payload size. |
| 424 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // Data 16 bytes. |
| 425 | 'f', 'm', 't', ' ', |
| 426 | 16, 0, 0, 0, // Size of fmt block. |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 427 | 1, 0, // Format: Pcm (1). |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 428 | 1, 0, // Channels: 1. |
| 429 | 60, 0, 0, 0, // Sample rate: 60. |
| 430 | 60, 0, 0, 0, // Byte rate: 1 * 1 * 60. |
| 431 | 1, 0, // Block align: NumChannels * BytesPerSample. |
| 432 | 8, 0, // Bits per sample: 1 * 8. |
| 433 | // clang-format on |
| 434 | }; |
| 435 | |
| 436 | size_t num_channels = 0; |
| 437 | int sample_rate = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 438 | WavFormat format = WavFormat::kWavFormatPcm; |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 439 | size_t bytes_per_sample = 0; |
| 440 | size_t num_samples = 0; |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 441 | int64_t data_start_pos = 0; |
| 442 | WavHeaderBufferReader r(kBuf, sizeof(kBuf), /*check_read_size=*/false); |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 443 | EXPECT_FALSE(ReadWavHeader(&r, &num_channels, &sample_rate, &format, |
Per Åhgren | 5dca3f1 | 2020-01-28 09:08:11 +0100 | [diff] [blame] | 444 | &bytes_per_sample, &num_samples, &data_start_pos)); |
Alessio Bazzica | a33c7af | 2018-11-08 12:16:11 +0100 | [diff] [blame] | 445 | } |
| 446 | |
andrew@webrtc.org | 048c502 | 2014-12-16 20:17:21 +0000 | [diff] [blame] | 447 | } // namespace webrtc |