Added mutex for reserved_license_ids_.
[ Merge of http://go/wvgerrit/150849 ] [ Cherry-pick of http://ag/17919183 ] DeviceFiles uses a static variable for tracking license IDs which have been reserved by a CDM session before officially storing the license on the device. This variable was not protected by a mutex, and a rare race condition would arise, either crashing the service or getting it stuck in a loop. This CL adds a mutex for protecting the set of reserved IDs. Bug: 226555704 Test: device_files_unittest Change-Id: Icdea88673c76c267b4b7db79697ec52ae8e2581e
This commit is contained in:
@@ -5,6 +5,7 @@
|
||||
#ifndef WVCDM_CORE_DEVICE_FILES_H_
|
||||
#define WVCDM_CORE_DEVICE_FILES_H_
|
||||
|
||||
#include <mutex>
|
||||
#include <set>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
@@ -357,6 +358,7 @@ class DeviceFiles {
|
||||
#endif
|
||||
|
||||
static std::set<std::string> reserved_license_ids_;
|
||||
static std::mutex reserved_license_ids_mutex_;
|
||||
|
||||
FileSystem* file_system_;
|
||||
CdmSecurityLevel security_level_;
|
||||
|
||||
@@ -86,6 +86,8 @@ using video_widevine_client::sdk::
|
||||
}
|
||||
|
||||
namespace wvcdm {
|
||||
using UniqueLock = std::unique_lock<std::mutex>;
|
||||
|
||||
namespace {
|
||||
const char kEmptyFileName[] = "";
|
||||
const char kFalse[] = "false";
|
||||
@@ -363,6 +365,7 @@ const char* DeviceFiles::ResponseTypeToString(ResponseType type) {
|
||||
|
||||
// static
|
||||
std::set<std::string> DeviceFiles::reserved_license_ids_;
|
||||
std::mutex DeviceFiles::reserved_license_ids_mutex_;
|
||||
|
||||
DeviceFiles::DeviceFiles(FileSystem* file_system)
|
||||
: file_system_(file_system),
|
||||
@@ -726,6 +729,7 @@ bool DeviceFiles::StoreLicense(const CdmLicenseData& license_data,
|
||||
std::string serialized_file;
|
||||
file.SerializeToString(&serialized_file);
|
||||
|
||||
UniqueLock lock(reserved_license_ids_mutex_);
|
||||
reserved_license_ids_.erase(license_data.key_set_id);
|
||||
*result = StoreFileWithHash(license_data.key_set_id + kLicenseFileNameExt,
|
||||
serialized_file);
|
||||
@@ -862,18 +866,21 @@ bool DeviceFiles::DeleteAllFiles() {
|
||||
|
||||
bool DeviceFiles::LicenseExists(const std::string& key_set_id) {
|
||||
RETURN_FALSE_IF_UNINITIALIZED();
|
||||
UniqueLock lock(reserved_license_ids_mutex_);
|
||||
return reserved_license_ids_.count(key_set_id) ||
|
||||
FileExists(key_set_id + kLicenseFileNameExt);
|
||||
}
|
||||
|
||||
bool DeviceFiles::ReserveLicenseId(const std::string& key_set_id) {
|
||||
RETURN_FALSE_IF_UNINITIALIZED();
|
||||
UniqueLock lock(reserved_license_ids_mutex_);
|
||||
reserved_license_ids_.insert(key_set_id);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool DeviceFiles::UnreserveLicenseId(const std::string& key_set_id) {
|
||||
RETURN_FALSE_IF_UNINITIALIZED();
|
||||
UniqueLock lock(reserved_license_ids_mutex_);
|
||||
reserved_license_ids_.erase(key_set_id);
|
||||
return true;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user