Add launcher executables for 32-bit and ARM64.

Fixes #67

Author: zooba

_msbuild.py

@@ -107,6 +107,7 @@ def main_exe(name):
         ConfigurationType='Application',
     )
 
+
 def mainw_exe(name):
     return CProject(name,
         VersionInfo(FileDescription="Python Install Manager (windowed)"),
@@ -127,6 +128,26 @@ def mainw_exe(name):
     )
 
 
+def launcher_exe(name, platform, windowed=False):
+    return CProject(name,
+        VersionInfo(
+            FileDescription="Python launcher" + (" (windowed)" if windowed else ""),
+            OriginalFilename=f"{name}.exe"
+        ),
+        CPP_SETTINGS,
+        Property('StaticLibcppLinkage', 'true'),
+        ItemDefinition('Link', SubSystem='WINDOWS' if windowed else 'CONSOLE'),
+        Manifest('default.manifest'),
+        ResourceFile('pywicon.rc' if windowed else 'pyicon.rc'),
+        CSourceFile('launcher.cpp'),
+        CSourceFile('_launch.cpp'),
+        IncludeFile('*.h'),
+        source='src/pymanager',
+        ConfigurationType='Application',
+        Platform=platform,
+    )
+
+
 PACKAGE = Package('python-manager',
     PyprojectTomlFile('pyproject.toml'),
     # MSIX manifest
@@ -150,32 +171,12 @@ def mainw_exe(name):
     Package(
         'templates',
         File('src/pymanager/templates/template.py'),
-        CProject('launcher',
-            VersionInfo(FileDescription="Python launcher", OriginalFilename="launcher.exe"),
-            CPP_SETTINGS,
-            Property('StaticLibcppLinkage', 'true'),
-            ItemDefinition('Link', SubSystem='CONSOLE'),
-            Manifest('default.manifest'),
-            ResourceFile('pyicon.rc'),
-            CSourceFile('launcher.cpp'),
-            CSourceFile('_launch.cpp'),
-            IncludeFile('*.h'),
-            source='src/pymanager',
-            ConfigurationType='Application',
-        ),
-        CProject('launcherw',
-            VersionInfo(FileDescription="Python launcher (windowed)", OriginalFilename="launcherw.exe"),
-            CPP_SETTINGS,
-            Property('StaticLibcppLinkage', 'true'),
-            ItemDefinition('Link', SubSystem='WINDOWS'),
-            Manifest('default.manifest'),
-            ResourceFile('pywicon.rc'),
-            CSourceFile('launcher.cpp'),
-            CSourceFile('_launch.cpp'),
-            IncludeFile('*.h'),
-            source='src/pymanager',
-            ConfigurationType='Application',
-        ),
+        launcher_exe("launcher-64", "x64", windowed=False),
+        launcher_exe("launcherw-64", "x64", windowed=True),
+        launcher_exe("launcher-arm64", "ARM64", windowed=False),
+        launcher_exe("launcherw-arm64", "ARM64", windowed=True),
+        launcher_exe("launcher-32", "Win32", windowed=False),
+        launcher_exe("launcherw-32", "Win32", windowed=True),
     ),
 
     # Directory for MSIX resources

src/manage/install_command.py

@@ -215,13 +215,30 @@ def _calc(prefix, filename, calculate_dest=calculate_dest):
             LOGGER.debug("Attempted to overwrite: %s", dest)
 
 
-def _write_alias(cmd, alias, target):
+def _if_exists(launcher, plat):
+    plat_launcher = launcher.parent / f"{launcher.stem}{plat}.exe"
+    if plat_launcher.is_file():
+        return plat_launcher
+    return launcher
+
+
+def _write_alias(cmd, install, alias, target):
     p = (cmd.global_dir / alias["name"])
     ensure_tree(p)
     unlink(p)
     launcher = cmd.launcher_exe
     if alias.get("windowed"):
         launcher = cmd.launcherw_exe or launcher
+    plat = install["tag"].rpartition("-")[-1]
+    if plat:
+        LOGGER.debug("Checking for launcher for platform -%s", plat)
+        launcher = _if_exists(launcher, f"-{plat}")
+    if not launcher.is_file():
+        LOGGER.debug("Checking for launcher for default platform %s", cmd.default_platform)
+        launcher = _if_exists(launcher, cmd.default_platform)
+    if not launcher.is_file():
+        LOGGER.debug("Checking for launcher for -64")
+        launcher = _if_exists(launcher, "64")
     LOGGER.debug("Create %s linking to %s using %s", alias["name"], target, launcher)
     if not launcher or not launcher.is_file():
         LOGGER.warn("Skipping %s alias because the launcher template was not found.", alias["name"])
@@ -281,7 +298,7 @@ def update_all_shortcuts(cmd, path_warning=True):
                 if not target.is_file():
                     LOGGER.warn("Skipping alias '%s' because target '%s' does not exist", a["name"], a["target"])
                     continue
-                _write_alias(cmd, a, target)
+                _write_alias(cmd, i, a, target)
                 alias_written.add(a["name"].casefold())
 
         for s in i.get("shortcuts", ()):

src/pymanager/_launch.cpp

@@ -14,7 +14,10 @@ ctrl_c_handler(DWORD code)
 static int
 dup_handle(HANDLE input, HANDLE *output)
 {
-    static HANDLE self = GetCurrentProcess();
+    static HANDLE self = NULL;
+    if (self == NULL) {
+        self = GetCurrentProcess();
+    }
     if (input == NULL || input == INVALID_HANDLE_VALUE) {
         *output = input;
         return 0;

src/pymanager/launcher.cpp

@@ -117,7 +117,7 @@ get_executable(wchar_t *executable, unsigned int bufferSize)
         return HRESULT_FROM_WIN32(GetLastError());
     }
 
-    wcscat_s(config, L".__target__");
+    wcscat_s(config, MAXLEN, L".__target__");
 
     HANDLE hFile = CreateFileW(config, GENERIC_READ,
         FILE_SHARE_READ | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, 0, NULL);
@@ -150,7 +150,7 @@ try_load_python3_dll(const wchar_t *executable, unsigned int bufferSize, void **
     return ERROR_DLL_LOAD_DISABLED;
 #else
     wchar_t directory[MAXLEN];
-    wcscpy_s(directory, executable);
+    wcscpy_s(directory, MAXLEN, executable);
     wchar_t *sep = wcsrchr(directory, L'\\');
     if (!sep) {
         return ERROR_RELATIVE_PATH;

src/pymanager/main.cpp

@@ -541,7 +541,8 @@ wmain(int argc, wchar_t **argv)
 
     if (err) {
         // Most 'not found' errors have been handled above. These are internal
-        fprintf(stderr, "INTERNAL ERROR 0x%08X. Please report to https://github.com/python/pymanager\n", err);
+        fprintf(stderr, "[ERROR] Internal error 0x%08X. "
+                "Please report to https://github.com/python/pymanager\n", err);
         goto error;
     }
 
@@ -560,15 +561,15 @@ wmain(int argc, wchar_t **argv)
     case ERROR_EXE_MACHINE_TYPE_MISMATCH:
     case HRESULT_FROM_WIN32(ERROR_EXE_MACHINE_TYPE_MISMATCH):
         fprintf(stderr,
-                "[FATAL ERROR] Executable '%ls' is for a different kind of "
+                "[ERROR] Executable '%ls' is for a different kind of "
                 "processor architecture.\n",
                 executable.c_str());
         fprintf(stderr,
                 "Try using '-V:<version>' to select a different runtime, or use "
                 "'py install' to install one for your CPU.\n");
         break;
     default:
-        fprintf(stderr, "[FATAL ERROR] Failed to launch '%ls' (0x%08X)\n", executable.c_str(), err);
+        fprintf(stderr, "[ERROR] Failed to launch '%ls' (0x%08X)\n", executable.c_str(), err);
         fprintf(stderr, "This may be a corrupt install or a system configuration issue.\n");
         break;
     }

src/pymanager/msi.wxs

@@ -107,8 +107,12 @@
     </Component>
 
     <Component Id="TemplatesComponent" Directory="TEMPLATES" Guid="3ACC110C-5C33-45C8-9843-A446718930A0">
-      <File KeyPath="yes" Source="templates\launcher.exe" Name="launcher.exe" />
-      <File Source="templates\launcherw.exe" Name="launcherw.exe" />
+      <File KeyPath="yes" Source="templates\launcher-64.exe" Name="launcher-64.exe" />
+      <File Source="templates\launcherw-64.exe" Name="launcherw-64.exe" />
+      <File Source="templates\launcher-arm64.exe" Name="launcher-arm64.exe" />
+      <File Source="templates\launcherw-arm64.exe" Name="launcherw-arm64.exe" />
+      <File Source="templates\launcher-32.exe" Name="launcher-32.exe" />
+      <File Source="templates\launcherw-32.exe" Name="launcherw-32.exe" />
     </Component>
 
   </Package>