brave · emerick · Jan 9, 2019 · Jan 7, 2019 · emerick · Jan 7, 2019
diff --git a/omaha/common/brave_stats_updater.cc b/omaha/common/brave_stats_updater.cc
@@ -0,0 +1,314 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "omaha/common/brave_stats_updater.h"
+
+#include "omaha/base/omaha_version.h"
+#include "omaha/base/safe_format.h"
+#include "omaha/base/utils.h"
+#include "omaha/net/network_config.h"
+#include "omaha/net/network_request.h"
+#include "omaha/net/simple_request.h"
+
+namespace omaha {
+
+namespace {
+
+bool ParseStandardReferralCode(const TCHAR* filename, CString& referral_code) {
+  // Scan backwards for last dash in filename.
+  const TCHAR* scan = filename + lstrlen(filename) - 1;
+  while (scan != filename && *scan != _T('-'))
+    --scan;
+
+  if (*scan++ != _T('-'))
+    return false;
+
+  // Standard referral code is 6 characters.
+  const TCHAR* ref_code = scan;
+  if (lstrlen(ref_code) != 6)
+    return false;
+
+  // Ensure that first half of referral code is alphabetic.
+  for (int i = 0; i < 3; ++i) {
+    if ((ref_code[i] < _T('a') || ref_code[i] > _T('z')) &&
+        (ref_code[i] < _T('A') || ref_code[i] > _T('Z')))
+      return false;
+  }
+
+  // Ensure that second half of referral code is numeric.
+  for (int i = 3; i < 6; ++i) {
+    if (ref_code[i] < _T('0') || ref_code[i] > _T('9'))
+      return false;
+  }
+
+  referral_code = ref_code;
+
+  return true;
+}
+
+bool ParseExtendedReferralCode(const TCHAR* filename, CString& referral_code) {
+  // Scan backwards for second-to-last dash in filename, since this
+  // type of referral code has an embedded dash.
+  const TCHAR* scan = filename + lstrlen(filename) - 1;
+  while (scan != filename && *scan != _T('-'))
+    --scan;
+
+  if (*scan-- != _T('-'))
+    return false;
+
+  while (scan != filename && *scan != _T('-'))
+    --scan;
+
+  if (*scan++ != _T('-'))
+    return false;
+
+  // Ensure that referral code is alphabetic.
+  const TCHAR* ref_code = scan;
+  int dashes = 0;
+  for (int i = 0; i < lstrlen(ref_code); ++i) {
+    if ((ref_code[i] < _T('a') || ref_code[i] > _T('z')) &&
+        (ref_code[i] < _T('A') || ref_code[i] > _T('Z')) &&
+        (ref_code[i] != _T('-')))
+      return NULL;
+    if (ref_code[i] == _T('-'))
+      ++dashes;
+  }
+
+  // Ensure that referral code contains exactly one dash.
+  if (dashes != 1)
+    return false;
+
+  referral_code = ref_code;
+  return true;
+}
+
+bool ParseReferralCode(const TCHAR* installer_filename,
+                       CString& referral_code) {
+  TCHAR filename[MAX_PATH];
+  lstrcpyn(filename, installer_filename, MAX_PATH);
+
+  // Strip path from filename.
+  PathStripPath(filename);
+
+  // Strip extension from filename.
+  PathRemoveExtension(filename);
+
+  // Strip any de-duplicating suffix from filename, e.g. "(1)".
+  const TCHAR* scan = filename + lstrlen(filename) - 1;
+  if (*scan == _T(')')) {
+    --scan;
+    while (scan != filename && *scan >= _T('0') && *scan <= _T('9'))
+      --scan;
+    if (*scan == _T('('))
+      filename[scan - filename] = '\0';
+  }
+
+  // Strip trailing spaces from filename.
+  scan = filename + lstrlen(filename) - 1;
+  while (scan != filename && *scan == _T(' '))
+    --scan;
+
+  if (scan != filename && (scan != filename + lstrlen(filename)))
+    filename[scan - filename + 1];
+
+  // First check for 6-character standard referral code XXXDDD, where
+  // X is an alphabetic character and D is a numeric character. If not
+  // found, check for an alphabetic referral code of any length in the
+  // form XXX-XXX.
+  if (!ParseStandardReferralCode(filename, referral_code) &&
+      !ParseExtendedReferralCode(filename, referral_code)) {
+    return false;
+  }
+
+  return true;
+}
+
+CString ReadAppGuidFromReg() {
+  HKEY key;
+  int res =
+      RegOpenKeyEx(HKEY_CURRENT_USER, _T("Software\\BraveSoftware\\Installer"),
+                   0, KEY_READ, &key);
+  if (res != ERROR_SUCCESS) {
+    return _T("");
+  }
+
+  DWORD bytes = 0;
+  DWORD type = REG_SZ;
+  res = ::RegQueryValueEx(key, _T("AppGuid"), 0, &type, NULL, &bytes);
+  if (!(res == ERROR_SUCCESS && type == REG_SZ)) {
+    return _T("");
+  }
+
+  scoped_array<byte> buf(new byte[bytes]);
+  memset(buf.get(), 0, bytes);
+  res = ::RegQueryValueEx(key, _T("AppGuid"), 0, NULL, buf.get(), &bytes);
+  if (res != ERROR_SUCCESS) {
+    return _T("");
+  }
+
+  RegCloseKey(key);
+
+  return reinterpret_cast<const wchar_t*>(buf.get());
+}
+
+CString ReadStubInstallerPathFromReg() {
+  HKEY key;
+  int res =
+      RegOpenKeyEx(HKEY_CURRENT_USER, _T("Software\\BraveSoftware\\Promo"), 0,
+                   KEY_READ, &key);
+  if (res != ERROR_SUCCESS) {
+    return _T("");
+  }
+
+  DWORD bytes = 0;
+  DWORD type = REG_SZ;
+  res = ::RegQueryValueEx(key, _T("StubInstallerPath"), 0, &type, NULL, &bytes);
+  if (!(res == ERROR_SUCCESS && type == REG_SZ)) {
+    return _T("");
+  }
+
+  scoped_array<byte> buf(new byte[bytes]);
+  memset(buf.get(), 0, bytes);
+  res = ::RegQueryValueEx(key, _T("StubInstallerPath"), 0, NULL, buf.get(),
+                          &bytes);
+  if (res != ERROR_SUCCESS) {
+    return _T("");
+  }
+
+  RegCloseKey(key);
+
+  return reinterpret_cast<const wchar_t*>(buf.get());
+}
+
+bool IsReferralCodeWhiteListed(const CString& referral_code) {
+  static std::vector<CString> whitelist(
+      {// ISOptOut2
+       CString(_T("CTD238")), CString(_T("OJV851")), CString(_T("DPM796")),
+       CString(_T("BNI512")), CString(_T("FEB604")), CString(_T("YDQ106")),
+       CString(_T("NFF966")), CString(_T("FQW627")), CString(_T("AMV588")),
+       CString(_T("YZR853")), CString(_T("IUX155")), CString(_T("DFQ107")),
+       CString(_T("ZWV410")), CString(_T("ZTT758")), CString(_T("EPH628")),
+       CString(_T("DKD200")), CString(_T("WIR635")), CString(_T("YHC941")),
+       CString(_T("OPV062")), CString(_T("MTS962")), CString(_T("PSN487")),
+       CString(_T("MZJ289")), CString(_T("EBC998")), CString(_T("KWZ833")),
+       CString(_T("YQI537")), CString(_T("BUE553")), CString(_T("UGO473")),
+       CString(_T("CFT923")), CString(_T("WKG906")), CString(_T("OXU789")),
+       CString(_T("VGT997")), CString(_T("MOU235")), CString(_T("WLJ467")),
+       CString(_T("YXG330")), CString(_T("RNH069")), CString(_T("LRT088")),
+       CString(_T("BEM856")), CString(_T("DSL157")), CString(_T("ZGL739")),
+       CString(_T("MZX864")), CString(_T("GTW616")), CString(_T("YBX790")),
+       CString(_T("LJT611")), CString(_T("FST304")), CString(_T("TVJ301")),
+       CString(_T("RMB962")), CString(_T("NKZ324")),
+
+       // STTier1
+       CString(_T("FBI093")), CString(_T("VNI569")), CString(_T("GMM900")),
+       CString(_T("NOQ491")), CString(_T("IOB292")), CString(_T("JJE625")),
+       CString(_T("FUX638")), CString(_T("MPO928")), CString(_T("LYD303")),
+       CString(_T("TJF413")), CString(_T("MQP631")), CString(_T("ILY758")),
+       CString(_T("VBQ225")), CString(_T("UQS362")),
+
+       // GS
+       CString(_T("PJJ948")), CString(_T("UIH408")), CString(_T("KXC376")),
+       CString(_T("OZD582")), CString(_T("JDT909")), CString(_T("IJZ122")),
+       CString(_T("OIE359")), CString(_T("DLQ981")), CString(_T("WIX525")),
+       CString(_T("PZH825")), CString(_T("NGY511")), CString(_T("QXS120")),
+       CString(_T("XTA152")), CString(_T("EZM037")),
+
+       // JAc
+       CString(_T("DHU083")), CString(_T("UGI415")), CString(_T("RQH046")),
+       CString(_T("MEB961"))});
+  return std::any_of(whitelist.begin(), whitelist.end(),
+                     [&referral_code](CString candidate) {
+                       return candidate == referral_code;
+                     });
+}
+
+CString GetChannelName(const CString& app_guid) {
+  CString app_guid_mutable = app_guid;
+  if (app_guid_mutable.IsEmpty())
+    app_guid_mutable = ReadAppGuidFromReg();
+
+  if (app_guid_mutable == _T("{AFE6A462-C574-4B8A-AF43-4CC60DF4563B}"))
+    return _T("release");
+  else if (app_guid_mutable == _T("{103BD053-949B-43A8-9120-2E424887DE11}"))
+    return _T("beta");
+  else if (app_guid_mutable == _T("{CB2150F2-595F-4633-891A-E39720CE0531}"))
+    return _T("dev");
+  else if (app_guid_mutable == _T("{C6CB981E-DB30-4876-8639-109F8933582C}"))
+    return _T("nightly");
+  else
+    return _T("");
+}
+
+CString GetBraveVersion(const CString& version) {
+  // Strip first segment from version number.
+  int pos = version.Find(_T('.'));
+  if (pos != -1)
+    return version.Right(version.GetLength() - pos - 1);
+  return _T("0.0.0");
+}
+
+CString GetReferralCode() {
+  CString referral_code;
+  CString stub_installer_path = ReadStubInstallerPathFromReg();
+  if (!ParseReferralCode(stub_installer_path, referral_code))
+    return _T("none");
+
+  if (!IsReferralCodeWhiteListed(referral_code))
+    return _T("none");
+
+  return referral_code;
+}
+
+}  // namespace
+
+HRESULT BraveSendStatsPing(const CString& event, const CString& version) {
+  // Retrieve the update server name.
+  CString update_server =
+      omaha::GetEnvironmentVariableAsString(_T("BRAVE_UPDATE_SERVER"));
+  if (update_server.IsEmpty())
+    update_server = _T("laptop-updates.brave.com");
+
+  _SYSTEM_INFO system_info = {};
+  ::GetNativeSystemInfo(&system_info);
+
+  // Ensure that channel is valid/supported.
+  CString app_guid = ReadAppGuidFromReg();
+  CString channel_name = GetChannelName(app_guid);
+  if (channel_name.IsEmpty())
+    return E_INVALIDARG;
+
+  // Build the update url.
+  CString url;
+  omaha::SafeCStringAppendFormat(
+      &url,
+      _T("https://%s/1/")
+      _T("installerEvent?platform=%s&version=%s&channel=%s&ref=%s&event=%s"),
+      update_server,
+      system_info.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_INTEL
+          ? _T("winia32-bc")
+          : _T("winx64-bc"),
+      GetBraveVersion(version), channel_name, GetReferralCode(), event);
+
+  omaha::NetworkConfig* network_config = NULL;
+  omaha::NetworkConfigManager& network_manager =
+      omaha::NetworkConfigManager::Instance();
+  HRESULT hr = network_manager.GetUserNetworkConfig(&network_config);
+  if (FAILED(hr))
+    return hr;
+
+  omaha::NetworkRequest network_request(network_config->session());
+
+  network_request.set_num_retries(1);
+  network_request.AddHttpRequest(new omaha::SimpleRequest);
+
+  std::vector<uint8> response_buffer;
+  hr = network_request.Get(url, &response_buffer);
+  if (FAILED(hr))
+    return hr;
+
+  return S_OK;
+}
+
+}  // namespace omaha
diff --git a/omaha/common/brave_stats_updater.h b/omaha/common/brave_stats_updater.h
@@ -0,0 +1,14 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef OMAHA_COMMON_BRAVE_STATS_UPDATER_H__
+#define OMAHA_COMMON_BRAVE_STATS_UPDATER_H__
+
+namespace omaha {
+
+HRESULT BraveSendStatsPing(const CString& event, const CString& version);
+
+} // namespace omaha
+
+#endif  // OMAHA_COMMON_BRAVE_STATS_UPDATER_H__
diff --git a/omaha/common/build.scons b/omaha/common/build.scons
@@ -24,6 +24,7 @@ def BuildCommonLib(env):
 
   inputs = [
       'app_registry_utils.cc',
+      'brave_stats_updater.cc',
       'command_line.cc',
       'command_line_builder.cc',
       'config_manager.cc',

diff --git a/omaha/goopdate/download_manager.cc b/omaha/goopdate/download_manager.cc
@@ -39,6 +39,7 @@
 #include "omaha/base/synchronized.h"
 #include "omaha/base/user_rights.h"
 #include "omaha/base/utils.h"
+#include "omaha/common/brave_stats_updater.h"
 #include "omaha/common/config_manager.h"
 #include "omaha/common/const_goopdate.h"
 #include "omaha/goopdate/model.h"
@@ -276,6 +277,8 @@ HRESULT DownloadManager::DownloadApp(App* app) {
   }
 
   if (SUCCEEDED(hr)) {
+    omaha::BraveSendStatsPing(_T("download-complete"),
+                              app_version->app()->next_version()->version());
     app->DownloadComplete();
     app->MarkReadyToInstall();
   } else {

diff --git a/omaha/goopdate/install_manager.cc b/omaha/goopdate/install_manager.cc
@@ -23,6 +23,7 @@
 #include "omaha/base/scope_guard.h"
 #include "omaha/base/synchronized.h"
 #include "omaha/base/utils.h"
+#include "omaha/common/brave_stats_updater.h"
 #include "omaha/common/config_manager.h"
 #include "omaha/common/const_cmd_line.h"
 #include "omaha/common/install_manifest.h"
@@ -119,6 +120,8 @@ void InstallManager::InstallApp(App* app,  const CString& dir) {
 
   HANDLE primary_token(app->app_bundle()->primary_token());
 
+  omaha::BraveSendStatsPing(_T("installer-run"), next_version->version());
+
   HRESULT hr = InstallApp(is_machine_,
                           primary_token,
                           current_version_string,

diff --git a/omaha/mi_exe_stub/build.scons b/omaha/mi_exe_stub/build.scons
@@ -40,12 +40,22 @@ for omaha_version_info in env['omaha_versions_info']:
           '/FI$MAIN_DIR/mi_exe_stub/ptrdiff_t.h',
       ],
       LIBS = [
+          '$LIB_DIR/base.lib',
+          '$LIB_DIR/common.lib',
+          '$LIB_DIR/net.lib',
+          '$LIB_DIR/security.lib',
           'lzma',
           'mi_exe_stub_lib',
           temp_env['atls_libs'][temp_env.Bit('debug')],
           temp_env['crt_libs'][temp_env.Bit('debug')],
+          'crypt32.lib',
+          'iphlpapi.lib',
+          'netapi32.lib',
+          'psapi.lib',
           'shlwapi',
+          'userenv.lib',
           'version',
+          'wtsapi32.lib',
           ],
       RCFLAGS = [
           '/DVERSION_MAJOR=%d' % omaha_version_info.version_major,