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

Simplify fuzz tests using LicenseWithUsageEntryFuzz

Browse Source 
Merged from https://widevine-internal-review.googlesource.com/175060

Change-Id: I2f245ac998883e2a31cde361e30b36f00681262c
This commit is contained in:
Ian Benz
2023-05-12 17:52:38 +00:00
committed by Robert Shih
parent 5c5433a8d8
commit 078682a897
8 changed files with 72 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
libwvdrmengine/oemcrypto/test/fuzz_tests/oemcrypto_fuzz_helper.h
View File

@@ -143,33 +143,36 @@ class OEMCryptoRenewalAPIFuzz {
class LicenseWithUsageEntryFuzz {
public:
LicenseWithUsageEntryFuzz() : license_messages_(&session_) {
LicenseWithUsageEntryFuzz() : license_messages_(&session_fuzz_.session()) {
license_messages_.set_pst("my_pst");
}
void Initialize() { InitializeFuzz(session_util_); }
void Initialize() { session_fuzz_.Initialize(); }
void InstallTestDrmKey(Session* session) {
session_util_.InstallTestDrmKey(session);
}
void Terminate() { session_fuzz_.Terminate(); }
void CreateUsageTableHeader();
void InstallTestDrmKey() { session_fuzz_.InstallTestDrmKey(); }
void LoadLicense();
LicenseRoundTrip& license_messages() { return license_messages_; }
const LicenseRoundTrip& license_messages() const { return license_messages_; }
const vector<uint8_t>& encrypted_usage_header() const {
const std::vector<uint8_t>& encrypted_usage_header() const {
return encrypted_usage_header_;
}
Session& session() { return session_fuzz_.session(); }
const Session& session() const { return session_fuzz_.session(); }
private:
SessionUtil session_util_;
vector<uint8_t> encrypted_usage_header_;
SessionFuzz session_fuzz_;
LicenseRoundTrip license_messages_;
Session session_;
std::vector<uint8_t> encrypted_usage_header_;
};
// Convert data from FuzzedDataProvider to valid enum value.