Files
android/libwvdrmengine/level3/src/level3_file_system_android.cpp
Srujan Gaddam 896ce2b5aa Refactor file_store to use smart pointers
Bug: b/119276649
Merge from: http://go/wvgerrit/66367
Test: Android, CE CDM, Linux unit tests

The FileSystem interface as it exists expects an Open for a file and
then a Close when finished. However, the Close doesn't delete the file
itself and depending on the platform, the underlying impl_ as well,
leading to a memory leak. To fix this leak as well as harden against
future memory issues, this change refactors the interface to shift away
from raw pointers and towards smart pointers.

Change-Id: I7a7132ea95cd3775796a540f510b698f4f27dd24
2018-11-30 16:25:58 -08:00

82 lines
2.7 KiB
C++

#include "level3_file_system_android.h"
#include <errno.h>
#include <stdio.h>
#include <sys/stat.h>
#include <cstring>
#include <string>
#include "log.h"
#include "properties.h"
#include "wv_cdm_types.h"
using wvcdm::File;
using wvcdm::FileSystem;
namespace wvoec3 {
OEMCrypto_Level3AndroidFileSystem::OEMCrypto_Level3AndroidFileSystem()
: file_system_(new FileSystem()) {
const char kDirectoryDelimiter = '/';
wvcdm::Properties::GetDeviceFilesBasePath(wvcdm::kSecurityLevelL3,
&base_path_);
size_t size = base_path_.size();
if (size < 2) {
// Default value is set to make sure unit tests pass, running as root.
base_path_ = "/data/";
size = base_path_.size();
} else if (base_path_[size - 1] != kDirectoryDelimiter) {
base_path_ = base_path_ + "/";
}
size_t pos = base_path_.find(kDirectoryDelimiter, 1);
while (pos < size) {
base_path_[pos] = '\0';
if (mkdir(base_path_.c_str(), 0775) != 0 && errno != EEXIST) {
wvcdm::Log(
"", "", 0, wvcdm::LOG_ERROR,
"Could not create base directories for Level3FileSystem, error: %s\n",
strerror(errno));
}
base_path_[pos] = kDirectoryDelimiter;
pos = base_path_.find(kDirectoryDelimiter, pos + 1);
}
}
OEMCrypto_Level3AndroidFileSystem::~OEMCrypto_Level3AndroidFileSystem() {}
ssize_t OEMCrypto_Level3AndroidFileSystem::Read(const char *filename,
void *buffer, size_t size) {
auto file = file_system_->Open(base_path_ + std::string(filename),
FileSystem::kReadOnly);
ssize_t bytes_read = 0;
if (file) {
bytes_read = file->Read(static_cast<char *>(buffer), size);
}
return bytes_read;
}
ssize_t OEMCrypto_Level3AndroidFileSystem::Write(const char *filename,
const void *buffer,
size_t size) {
auto file = file_system_->Open(base_path_ + std::string(filename),
FileSystem::kCreate | FileSystem::kTruncate);
ssize_t bytes_written = 0;
if (file) {
bytes_written = file->Write(static_cast<const char *>(buffer), size);
}
return bytes_written;
}
bool OEMCrypto_Level3AndroidFileSystem::Exists(const char *filename) {
return file_system_->Exists(base_path_ + std::string(filename));
}
ssize_t OEMCrypto_Level3AndroidFileSystem::FileSize(const char *filename) {
return file_system_->FileSize(base_path_ + std::string(filename));
}
bool OEMCrypto_Level3AndroidFileSystem::Remove(const char *filename) {
return file_system_->Remove(base_path_ + std::string(filename));
}
} // namespace wvoec3