widevine-partner/android
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated buffer_reader_fuzzer

Browse Source 
Implemented google c++ code style changes for buffer_reader_fuzzer

exec/s: 3649
Test: ./buffer_reader_fuzzer
Bug: 312374669

Change-Id: I53bd247f6c95202d7a5e86115aa060c61821d49c
This commit is contained in:
Onkar Shinde
2023-12-08 10:26:23 +00:00
committed by Aditya Wazir
parent 2b3de00a36
commit 363447d00c
1 changed files with 109 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

182
fuzzer/buffer_reader_fuzzer.cpp
View File

@@ -93,7 +93,7 @@ const std::string kMultipleWidevinePsshBox = wvutil::a2bs_hex(
"3c8f389223c0f3621522101946f0c2a3" "3c8f389223c0f3621522101946f0c2a3"
"d543101cc842bbec2d0b30"); "d543101cc842bbec2d0b30");
const std::string kHLSData = const std::string kHlsData =
"EXT-X-KEY: METHOD=SAMPLE-AES, " "EXT-X-KEY: METHOD=SAMPLE-AES, "
"URI=”data:text/plain;base64,eyANCiAgICJwcm92aWRlciI6Im1sYmFtaGJvIiwNCiAg" "URI=”data:text/plain;base64,eyANCiAgICJwcm92aWRlciI6Im1sYmFtaGJvIiwNCiAg"
"ICJjb250ZW50X2lkIjoiMjAxNV9UZWFycyIsDQogICAia2V5X2lkcyI6DQogICBbDQo" "ICJjb250ZW50X2lkIjoiMjAxNV9UZWFycyIsDQogICAia2V5X2lkcyI6DQogICBbDQo"
@@ -103,179 +103,189 @@ const std::string kHLSData =
"KEYFORMAT=”com.widevine”, " "KEYFORMAT=”com.widevine”, "
"KEYFORMATVERSIONS=”1”"; "KEYFORMATVERSIONS=”1”";
const std::string kHLSMethodParse = "EXT-X-KEY: METHOD="; const std::string kHlsMethodParse = "EXT-X-KEY: METHOD=";
const std::string KUriParse = "\"URI=\"data:text/plain;base64,"; const std::string KUriParse = "\"URI=\"data:text/plain;base64,";
const std::string kIVParse = "\"IV="; const std::string kIvParse = "\"IV=";
const std::string kKeyFormatParse = "\"KEYFORMAT=\""; const std::string kKeyFormatParse = "\"KEYFORMAT=\"";
const std::string kKeyFormatVersionParse = "\"KEYFORMATVERSIONS=\""; const std::string kKeyFormatVersionParse = "\"KEYFORMATVERSIONS=\"";
class BufferReaderFuzzer { class BufferReaderFuzzer {
public: public:
BufferReaderFuzzer(const uint8_t *data, size_t size) : mFdp(data, size){}; BufferReaderFuzzer(const uint8_t *data, size_t size) : fdp_(data, size){};
void process(); void Process();
std::string setPartialRandomHLSData(); std::string SetPartialRandomHlsData();
std::string setRandomHLSData(); std::string SetRandomHlsData();
private: private:
FuzzedDataProvider mFdp; FuzzedDataProvider fdp_;
}; };
std::string BufferReaderFuzzer::setPartialRandomHLSData() { std::string BufferReaderFuzzer::SetPartialRandomHlsData() {
std::string hlsMethod = std::string hls_method =
kHLSMethodParse + kHlsMethodParse +
(mFdp.ConsumeBool() ? mFdp.ConsumeRandomLengthString(kMaxByte) (fdp_.ConsumeBool() ? fdp_.ConsumeRandomLengthString(kMaxByte)
: mFdp.PickValueInArray(kHlsMethod)) + : fdp_.PickValueInArray(kHlsMethod)) +
", \""; ", \"";
std::string uri = KUriParse + std::string uri = KUriParse +
(mFdp.ConsumeBool() (fdp_.ConsumeBool()
? mFdp.ConsumeRandomLengthString(kMaxByte) ? fdp_.ConsumeRandomLengthString(kMaxByte)
: "eyANCiAgICJwcm92aWRlciI6Im1sYmFtaGJvIiwNCiAg\" \ : "eyANCiAgICJwcm92aWRlciI6Im1sYmFtaGJvIiwNCiAg\" \
\"ICJjb250ZW50X2lkIjoiMjAxNV9UZWFycyIsDQogICAia2V5X2lkcyI6DQogICBbDQo\" \ \"ICJjb250ZW50X2lkIjoiMjAxNV9UZWFycyIsDQogICAia2V5X2lkcyI6DQogICBbDQo\" \
\"gICAgICAiMzcxZTEzNWUxYTk4NWQ3NWQxOThhN2Y0MTAyMGRjMjMiDQogICBdDQp9DQ\" \ \"gICAgICAiMzcxZTEzNWUxYTk4NWQ3NWQxOThhN2Y0MTAyMGRjMjMiDQogICBdDQp9DQ\" \
\"o=") + ", \""; \"o=") + ", \"";
std::string iv = std::string iv =
kIVParse + kIvParse +
((mFdp.ConsumeBool() ? mFdp.ConsumeRandomLengthString(kMaxByte) ((fdp_.ConsumeBool() ? fdp_.ConsumeRandomLengthString(kMaxByte)
: "0x6df49213a781e338628d0e9c812d328e,")) + : "0x6df49213a781e338628d0e9c812d328e,")) +
"\""; "\"";
std::string keyFormat = std::string key_format =
kKeyFormatParse + kKeyFormatParse +
(mFdp.ConsumeBool() (fdp_.ConsumeBool()
? ("\"" + mFdp.ConsumeRandomLengthString(kMaxByte) + "\"") ? ("\"" + fdp_.ConsumeRandomLengthString(kMaxByte) + "\"")
: "com.widevine\", \""); : "com.widevine\", \"");
std::string keyFormatVersion = std::string key_format_version =
kKeyFormatVersionParse + kKeyFormatVersionParse +
(mFdp.ConsumeBool() ? mFdp.ConsumeRandomLengthString(kMaxByte) : "\"1\""); (fdp_.ConsumeBool() ? fdp_.ConsumeRandomLengthString(kMaxByte) : "\"1\"");
std::string result = hlsMethod + uri + iv + keyFormat + keyFormatVersion; std::string result = hls_method + uri + iv + key_format + key_format_version;
return result; return result;
} }
std::string BufferReaderFuzzer::setRandomHLSData() { std::string BufferReaderFuzzer::SetRandomHlsData() {
std::string hlsMethod; std::string hls_method;
if(mFdp.ConsumeBool()){ if (fdp_.ConsumeBool()) {
hlsMethod = kHLSMethodParse; hls_method = kHlsMethodParse;
} }
hlsMethod += mFdp.ConsumeRandomLengthString(kMaxByte) + ", \""; hls_method += fdp_.ConsumeRandomLengthString(kMaxByte) + ", \"";
std::string uri; std::string uri;
if(mFdp.ConsumeBool()){ if (fdp_.ConsumeBool()) {
uri = KUriParse; uri = KUriParse;
} }
uri += mFdp.ConsumeRandomLengthString(kMaxByte) + ", \""; uri += fdp_.ConsumeRandomLengthString(kMaxByte) + ", \"";
std::string iv; std::string iv;
if(mFdp.ConsumeBool()){ if (fdp_.ConsumeBool()) {
iv = kIVParse; iv = kIvParse;
} }
iv += mFdp.ConsumeRandomLengthString(kMaxByte) + "\""; iv += fdp_.ConsumeRandomLengthString(kMaxByte) + "\"";
std::string keyFormat; std::string key_format;
if(mFdp.ConsumeBool()){ if (fdp_.ConsumeBool()) {
keyFormat = kKeyFormatParse; key_format = kKeyFormatParse;
} }
keyFormat += mFdp.ConsumeRandomLengthString(kMaxByte) + "\""; key_format += fdp_.ConsumeRandomLengthString(kMaxByte) + "\"";
std::string keyFormatVersion; std::string key_format_version;
if(mFdp.ConsumeBool()){ if (fdp_.ConsumeBool()) {
keyFormatVersion = kKeyFormatVersionParse; key_format_version = kKeyFormatVersionParse;
} }
keyFormatVersion += mFdp.ConsumeRandomLengthString(kMaxByte); key_format_version += fdp_.ConsumeRandomLengthString(kMaxByte);
std::string result = hlsMethod + uri + iv + keyFormat + keyFormatVersion; std::string result = hls_method + uri + iv + key_format + key_format_version;
return result; return result;
} }
void BufferReaderFuzzer::process() { void BufferReaderFuzzer::Process() {
std::vector<uint8_t> raw_data =
fdp_.ConsumeBytes<uint8_t>(fdp_.ConsumeIntegral<size_t>());
BufferReader reader(raw_data.data(), raw_data.size());
std::vector<uint8_t> rawData = std::string type = fdp_.PickValueInArray(kInitDataTypes);
mFdp.ConsumeBytes<uint8_t>(mFdp.ConsumeIntegral<size_t>());
BufferReader reader(rawData.data(), rawData.size());
std::string type = mFdp.PickValueInArray(kInitDataTypes);
std::string data; std::string data;
if (type == ISO_BMFF_VIDEO_MIME_TYPE || type == ISO_BMFF_AUDIO_MIME_TYPE || if (type == ISO_BMFF_VIDEO_MIME_TYPE || type == ISO_BMFF_AUDIO_MIME_TYPE ||
type == CENC_INIT_DATA_FORMAT) { type == CENC_INIT_DATA_FORMAT) {
data = mFdp.ConsumeBool() data = fdp_.ConsumeBool()
? wvutil::a2bs_hex(mFdp.ConsumeRandomLengthString(kMaxByte)) ? wvutil::a2bs_hex(fdp_.ConsumeRandomLengthString(kMaxByte))
: kMultipleWidevinePsshBox; : kMultipleWidevinePsshBox;
} else if (type == HLS_INIT_DATA_FORMAT) { } else if (type == HLS_INIT_DATA_FORMAT) {
data = mFdp.PickValueInArray({kHLSData, setPartialRandomHLSData(), data = fdp_.PickValueInArray(
setRandomHLSData()}); {kHlsData, SetPartialRandomHlsData(), SetRandomHlsData()});
} else { } else {
data = mFdp.ConsumeRandomLengthString(kMaxByte); data = fdp_.ConsumeRandomLengthString(kMaxByte);
} }
std::string initDataType = std::string init_data_type =
mFdp.ConsumeBool() ? mFdp.ConsumeRandomLengthString(kMaxByte) : type; fdp_.ConsumeBool() ? fdp_.ConsumeRandomLengthString(kMaxByte) : type;
std::string version = std::string version =
mFdp.ConsumeBool() ? mFdp.PickValueInArray(kInitDataVersions) : ""; fdp_.ConsumeBool() ? fdp_.PickValueInArray(kInitDataVersions) : "";
InitializationData initData(initDataType, data, version); InitializationData init_data(init_data_type, data, version);
while (mFdp.remaining_bytes()) { while (fdp_.remaining_bytes()) {
auto invokeBufferReaderAPIs = auto invoke_buffer_reader_API =
mFdp.PickValueInArray<const std::function<void()>>({ fdp_.PickValueInArray<const std::function<void()>>({
[&]() { [&]() {
uint8_t placeholder; uint8_t place_holder;
mFdp.ConsumeBool() ? reader.Read1(&placeholder) fdp_.ConsumeBool() ? reader.Read1(&place_holder)
: reader.Read1(nullptr); : reader.Read1(nullptr);
}, },
[&]() { [&]() {
uint16_t placeholder; uint16_t place_holder;
mFdp.ConsumeBool() ? reader.Read2(&placeholder) : reader.Read2(nullptr); fdp_.ConsumeBool() ? reader.Read2(&place_holder)
: reader.Read2(nullptr);
}, },
[&]() { [&]() {
int16_t placeholder; int16_t place_holder;
mFdp.ConsumeBool() ? reader.Read2s(&placeholder) fdp_.ConsumeBool() ? reader.Read2s(&place_holder)
: reader.Read2s(nullptr); : reader.Read2s(nullptr);
}, },
[&]() { [&]() {
int64_t placeholder; int64_t place_holder;
mFdp.ConsumeBool() ? reader.Read8s(&placeholder) fdp_.ConsumeBool() ? reader.Read8s(&place_holder)
: reader.Read8s(nullptr); : reader.Read8s(nullptr);
}, },
[&]() { [&]() {
uint64_t placeholder; uint64_t place_holder;
mFdp.ConsumeBool() ? reader.Read8(&placeholder) : reader.Read8(nullptr); fdp_.ConsumeBool() ? reader.Read8(&place_holder)
: reader.Read8(nullptr);
}, },
[&]() { [&]() {
int64_t placeholder; int64_t place_holder;
mFdp.ConsumeBool() ? reader.Read4sInto8s(&placeholder) fdp_.ConsumeBool() ? reader.Read4sInto8s(&place_holder)
: reader.Read4sInto8s(nullptr); : reader.Read4sInto8s(nullptr);
}, },
[&]() { [&]() {
std::string placeholder; std::string place_holder;
mFdp.ConsumeBool() ? reader.ReadString(&placeholder, mFdp.ConsumeIntegralInRange<size_t>(kMinSize, kMaxSize)) fdp_.ConsumeBool()
: reader.ReadString(nullptr, mFdp.ConsumeIntegralInRange<size_t>(kMinSize, kMaxSize)); ? reader.ReadString(
&place_holder,
fdp_.ConsumeIntegralInRange<size_t>(kMinSize, kMaxSize))
: reader.ReadString(
nullptr, fdp_.ConsumeIntegralInRange<size_t>(kMinSize,
kMaxSize));
}, },
[&]() { [&]() {
std::vector<uint8_t> placeholder; std::vector<uint8_t> place_holder;
mFdp.ConsumeBool() ? reader.ReadVec(&placeholder, mFdp.ConsumeIntegralInRange<size_t>(kMinSize, kMaxSize)) fdp_.ConsumeBool()
: reader.ReadVec(nullptr, mFdp.ConsumeIntegralInRange<size_t>(kMinSize, kMaxSize)); ? reader.ReadVec(
&place_holder,
fdp_.ConsumeIntegralInRange<size_t>(kMinSize, kMaxSize))
: reader.ReadVec(nullptr, fdp_.ConsumeIntegralInRange<size_t>(
kMinSize, kMaxSize));
}, },
[&]() { initData.ExtractWrappedKeys(); }, [&]() { init_data.ExtractWrappedKeys(); },
[&]() { initData.DumpToLogs(); }, [&]() { init_data.DumpToLogs(); },
[&]() { initData.WidevineSystemID(); }, [&]() { init_data.WidevineSystemID(); },
}); });
invokeBufferReaderAPIs(); invoke_buffer_reader_API();
} }
} }
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
BufferReaderFuzzer bufferReaderFuzzer(data, size); BufferReaderFuzzer buffer_reader_fuzzer(data, size);
bufferReaderFuzzer.process(); buffer_reader_fuzzer.Process();
return 0; return 0;
} }