From 9ffdc70d2fbeabbcefd5295cb4ab4f42a228b949 Mon Sep 17 00:00:00 2001
From: brave-builds <brave-builds+devops@brave.com>
Date: Tue, 19 Sep 2023 00:43:49 +0000
Subject: [PATCH] Uplift of #20205 (squashed) to beta

---
 .../chrome/installer/setup/setup_main.cc      | 33 +++++++++++++------
 1 file changed, 23 insertions(+), 10 deletions(-)

diff --git a/chromium_src/chrome/installer/setup/setup_main.cc b/chromium_src/chrome/installer/setup/setup_main.cc
index 3b30bc30af3e..e8124163f606 100644
--- a/chromium_src/chrome/installer/setup/setup_main.cc
+++ b/chromium_src/chrome/installer/setup/setup_main.cc
@@ -3,17 +3,24 @@
  * 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/. */
 
-#define wWinMain wWinMain_ChromiumImpl
-#include "src/chrome/installer/setup/setup_main.cc"
-#undef wWinMain
+#include <string>
+
+#include "base/command_line.h"
+#include "base/files/file_path.h"
+#include "base/files/file_util.h"
+#include "base/logging.h"
+#include "base/path_service.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/installer/setup/setup_util.h"
+#include "chrome/installer/util/install_util.h"
+#include "chrome/installer/util/util_constants.h"
+
+namespace {
 
 const char kBraveReferralCode[] = "brave-referral-code";
 
-int WINAPI wWinMain(HINSTANCE instance, HINSTANCE prev_instance,
-                    wchar_t* command_line, int show_command) {
-  int return_code = wWinMain_ChromiumImpl(instance, prev_instance, command_line,
-                                          show_command);
-  if (!return_code) {
+void SavePromoCode(installer::InstallStatus install_status) {
+  if (!InstallUtil::GetInstallReturnCode(install_status)) {
     const base::CommandLine& cmd_line = *base::CommandLine::ForCurrentProcess();
     if (cmd_line.HasSwitch(kBraveReferralCode)) {
       const std::string referral_code =
@@ -30,6 +37,12 @@ int WINAPI wWinMain(HINSTANCE instance, HINSTANCE prev_instance,
       }
     }
   }
-
-  return return_code;
 }
+
+}  // namespace
+
+#define DoLegacyCleanups         \
+  SavePromoCode(install_status); \
+  DoLegacyCleanups
+#include "src/chrome/installer/setup/setup_main.cc"
+#undef DoLegacyCleanups