Update CCUserDefault.cpp

This commit is contained in:
HALX99 2020-11-15 13:08:21 +08:00 committed by GitHub
parent 9ba25c0fb1
commit f1262c521f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions

View File

@ -82,13 +82,13 @@ static void ud_setkey(std::string& lhs, const cxx17::string_view& rhs) {
lhs.assign(keyLen, '\0'); lhs.assign(keyLen, '\0');
} }
static void ud_write_v_s(yasio::obstream& obs, const cxx17::string_view value) static void ud_write_v_s(UserDefault* ud, yasio::obstream& obs, const cxx17::string_view value)
{ {
size_t value_offset = obs.length(); size_t value_offset = obs.length();
obs.write_v(value); obs.write_v(value);
value_offset += (obs.length() - value_offset - value.length()); value_offset += (obs.length() - value_offset - value.length());
if(!value.empty()) if(!value.empty())
UserDefault::getInstance()->encrypt(obs.wptr(value_offset), value.length(), AES_ENCRYPT); ud->encrypt(obs.wptr(value_offset), value.length(), AES_ENCRYPT);
} }
void UserDefault::setEncryptEnabled(bool enabled, const std::string& key, const std::string& iv) void UserDefault::setEncryptEnabled(bool enabled, const std::string& key, const std::string& iv)
@ -99,7 +99,7 @@ void UserDefault::setEncryptEnabled(bool enabled, const std::string& key, const
ud_setkey(_iv, iv); ud_setkey(_iv, iv);
} }
} }
a
void UserDefault::encrypt(std::string& inout, int enc) void UserDefault::encrypt(std::string& inout, int enc)
{ {
if (!inout.empty()) if (!inout.empty())
@ -302,8 +302,8 @@ void UserDefault::setStringForKey(const char* pKey, const std::string & value)
yasio::obstream obs; yasio::obstream obs;
if (_encryptEnabled) if (_encryptEnabled)
{ {
ud_write_v_s(obs, pKey); ud_write_v_s(this, obs, pKey);
ud_write_v_s(obs, value); ud_write_v_s(this, obs, value);
} }
else { else {
obs.write_v(pKey); obs.write_v(pKey);
@ -311,7 +311,7 @@ void UserDefault::setStringForKey(const char* pKey, const std::string & value)
} }
if ((_realSize + obs.length() + sizeof(udflen_t)) < _curMapSize) if ((_realSize + obs.length() + sizeof(udflen_t)) < _curMapSize)
{ { // fast append entity without flush
// increase entities count // increase entities count
yasio::obstream::swrite_ix(_rwmmap->data(), 1 + yasio::ibstream::sread_ix<udflen_t>(_rwmmap->data())); yasio::obstream::swrite_ix(_rwmmap->data(), 1 + yasio::ibstream::sread_ix<udflen_t>(_rwmmap->data()));
@ -437,8 +437,8 @@ void UserDefault::flush()
for (auto& item : this->_values) { for (auto& item : this->_values) {
if (_encryptEnabled) if (_encryptEnabled)
{ {
ud_write_v_s(obs, item.first); ud_write_v_s(this, obs, item.first);
ud_write_v_s(obs, item.second); ud_write_v_s(this, obs, item.second);
} }
else { else {
obs.write_v(item.first); obs.write_v(item.first);