diff options
Diffstat (limited to 'rlz/win/lib/machine_deal.cc')
-rw-r--r-- | rlz/win/lib/machine_deal.cc | 300 |
1 files changed, 0 insertions, 300 deletions
diff --git a/rlz/win/lib/machine_deal.cc b/rlz/win/lib/machine_deal.cc deleted file mode 100644 index 3216260..0000000 --- a/rlz/win/lib/machine_deal.cc +++ /dev/null @@ -1,300 +0,0 @@ -// Copyright (c) 2012 The Chromium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. -// -// Library functions related to the OEM Deal Confirmation Code. - -#include "rlz/win/lib/machine_deal.h" - -#include <windows.h> -#include <vector> - -#include "base/basictypes.h" -#include "base/memory/scoped_ptr.h" -#include "base/string_split.h" -#include "base/string_util.h" -#include "base/stringprintf.h" -#include "base/win/registry.h" -#include "rlz/lib/assert.h" -#include "rlz/lib/lib_values.h" -#include "rlz/win/lib/lib_mutex.h" -#include "rlz/win/lib/registry_util.h" -#include "rlz/win/lib/rlz_value_store_registry.h" - -const wchar_t kDccValueName[] = L"DCC"; - -namespace { - -// Current DCC can only uses [a-zA-Z0-9_-!@$*();.<>,:] -// We will be more liberal and allow some additional chars, but not url meta -// chars. -bool IsGoodDccChar(char ch) { - if (IsAsciiAlpha(ch) || IsAsciiDigit(ch)) - return true; - - switch (ch) { - case '_': - case '-': - case '!': - case '@': - case '$': - case '*': - case '(': - case ')': - case ';': - case '.': - case '<': - case '>': - case ',': - case ':': - return true; - } - - return false; -} - -// This function will remove bad rlz chars and also limit the max rlz to some -// reasonable size. It also assumes that normalized_dcc is at least -// kMaxDccLength+1 long. -void NormalizeDcc(const char* raw_dcc, char* normalized_dcc) { - int index = 0; - for (; raw_dcc[index] != 0 && index < rlz_lib::kMaxDccLength; ++index) { - char current = raw_dcc[index]; - if (IsGoodDccChar(current)) { - normalized_dcc[index] = current; - } else { - normalized_dcc[index] = '.'; - } - } - - normalized_dcc[index] = 0; -} - -bool GetResponseLine(const char* response_text, int response_length, - int* search_index, std::string* response_line) { - if (!response_line || !search_index || *search_index > response_length) - return false; - - response_line->clear(); - - if (*search_index < 0) - return false; - - int line_begin = *search_index; - const char* line_end = strchr(response_text + line_begin, '\n'); - - if (line_end == NULL || line_end - response_text > response_length) { - line_end = response_text + response_length; - *search_index = -1; - } else { - *search_index = line_end - response_text + 1; - } - - response_line->assign(response_text + line_begin, - line_end - response_text - line_begin); - return true; -} - -bool GetResponseValue(const std::string& response_line, - const std::string& response_key, - std::string* value) { - if (!value) - return false; - - value->clear(); - - if (!StartsWithASCII(response_line, response_key, true)) - return false; - - std::vector<std::string> tokens; - base::SplitString(response_line, ':', &tokens); - if (tokens.size() != 2) - return false; - - // The first token is the key, the second is the value. The value is already - // trimmed for whitespace. - *value = tokens[1]; - return true; -} - -} // namespace anonymous - -namespace rlz_lib { - -bool MachineDealCode::Set(const char* dcc) { - LibMutex lock; - if (lock.failed()) - return false; - - // TODO: if (!ProcessInfo::CanWriteMachineKey()) return false; - - // Validate the new dcc value. - size_t length = strlen(dcc); - if (length > kMaxDccLength) { - ASSERT_STRING("MachineDealCode::Set: DCC length is exceeds max allowed."); - return false; - } - - base::win::RegKey hklm_key(HKEY_LOCAL_MACHINE, - RlzValueStoreRegistry::GetWideLibKeyName().c_str(), - KEY_READ | KEY_WRITE | KEY_WOW64_32KEY); - if (!hklm_key.Valid()) { - ASSERT_STRING("MachineDealCode::Set: Unable to create / open machine key." - " Did you call rlz_lib::CreateMachineState()?"); - return false; - } - - char normalized_dcc[kMaxDccLength + 1]; - NormalizeDcc(dcc, normalized_dcc); - VERIFY(length == strlen(normalized_dcc)); - - // Write the DCC to HKLM. Note that we need to include the null character - // when writing the string. - if (!RegKeyWriteValue(hklm_key, kDccValueName, normalized_dcc)) { - ASSERT_STRING("MachineDealCode::Set: Could not write the DCC value"); - return false; - } - - return true; -} - -bool MachineDealCode::GetNewCodeFromPingResponse(const char* response, - bool* has_new_dcc, char* new_dcc, int new_dcc_size) { - if (!has_new_dcc || !new_dcc || !new_dcc_size) - return false; - - *has_new_dcc = false; - new_dcc[0] = 0; - - int response_length = -1; - if (!IsPingResponseValid(response, &response_length)) - return false; - - // Get the current DCC value to compare to later) - char stored_dcc[kMaxDccLength + 1]; - if (!Get(stored_dcc, arraysize(stored_dcc))) - stored_dcc[0] = 0; - - int search_index = 0; - std::string response_line; - std::string new_dcc_value; - bool old_dcc_confirmed = false; - const std::string dcc_cgi(kDccCgiVariable); - const std::string dcc_cgi_response(kSetDccResponseVariable); - while (GetResponseLine(response, response_length, &search_index, - &response_line)) { - std::string value; - - if (!old_dcc_confirmed && - GetResponseValue(response_line, dcc_cgi, &value)) { - // This is the old DCC confirmation - should match value in registry. - if (value != stored_dcc) - return false; // Corrupted DCC - ignore this response. - else - old_dcc_confirmed = true; - continue; - } - - if (!(*has_new_dcc) && - GetResponseValue(response_line, dcc_cgi_response, &value)) { - // This is the new DCC. - if (value.size() > kMaxDccLength) continue; // Too long - *has_new_dcc = true; - new_dcc_value = value; - } - } - - old_dcc_confirmed |= (NULL == stored_dcc[0]); - - base::strlcpy(new_dcc, new_dcc_value.c_str(), new_dcc_size); - return old_dcc_confirmed; -} - -bool MachineDealCode::SetFromPingResponse(const char* response) { - bool has_new_dcc = false; - char new_dcc[kMaxDccLength + 1]; - - bool response_valid = GetNewCodeFromPingResponse( - response, &has_new_dcc, new_dcc, arraysize(new_dcc)); - - if (response_valid && has_new_dcc) - return Set(new_dcc); - - return response_valid; -} - -bool MachineDealCode::GetAsCgi(char* cgi, int cgi_size) { - if (!cgi || cgi_size <= 0) { - ASSERT_STRING("MachineDealCode::GetAsCgi: Invalid buffer"); - return false; - } - - cgi[0] = 0; - - std::string cgi_arg; - base::StringAppendF(&cgi_arg, "%s=", kDccCgiVariable); - int cgi_arg_length = cgi_arg.size(); - - if (cgi_arg_length >= cgi_size) { - ASSERT_STRING("MachineDealCode::GetAsCgi: Insufficient buffer size"); - return false; - } - - base::strlcpy(cgi, cgi_arg.c_str(), cgi_size); - - if (!Get(cgi + cgi_arg_length, cgi_size - cgi_arg_length)) { - cgi[0] = 0; - return false; - } - return true; -} - -bool MachineDealCode::Get(char* dcc, int dcc_size) { - LibMutex lock; - if (lock.failed()) - return false; - - if (!dcc || dcc_size <= 0) { - ASSERT_STRING("MachineDealCode::Get: Invalid buffer"); - return false; - } - - dcc[0] = 0; - - base::win::RegKey dcc_key(HKEY_LOCAL_MACHINE, - RlzValueStoreRegistry::GetWideLibKeyName().c_str(), - KEY_READ | KEY_WOW64_32KEY); - if (!dcc_key.Valid()) - return false; // no DCC key. - - size_t size = dcc_size; - if (!RegKeyReadValue(dcc_key, kDccValueName, dcc, &size)) { - ASSERT_STRING("MachineDealCode::Get: Insufficient buffer size"); - dcc[0] = 0; - return false; - } - - return true; -} - -bool MachineDealCode::Clear() { - base::win::RegKey dcc_key(HKEY_LOCAL_MACHINE, - RlzValueStoreRegistry::GetWideLibKeyName().c_str(), - KEY_READ | KEY_WRITE | KEY_WOW64_32KEY); - if (!dcc_key.Valid()) - return false; // no DCC key. - - dcc_key.DeleteValue(kDccValueName); - - // Verify deletion. - wchar_t dcc[kMaxDccLength + 1]; - DWORD dcc_size = arraysize(dcc); - if (dcc_key.ReadValue(kDccValueName, dcc, &dcc_size, NULL) == ERROR_SUCCESS) { - ASSERT_STRING("MachineDealCode::Clear: Could not delete the DCC value."); - return false; - } - - return true; -} - -} // namespace rlz_lib |