check so arch in e2e

Author: xangcastle

e2e/cases/uv-deps-650/crossbuild/BUILD.bazel

@@ -185,3 +185,56 @@ platform_transition_filegroup(
     srcs = [":geohash_bin"],
     target_platform = ":arm64_linux",
 )
+
+# ── Verify .so architecture ──
+
+py_image_layer(
+    name = "geohash_layers",
+    binary = ":geohash_bin",
+)
+
+platform_transition_filegroup(
+    name = "geohash_amd64_layers",
+    srcs = [":geohash_layers"],
+    target_platform = ":amd64_linux",
+)
+
+platform_transition_filegroup(
+    name = "geohash_arm64_layers",
+    srcs = [":geohash_layers"],
+    target_platform = ":arm64_linux",
+)
+
+genrule(
+    name = "_geohash_amd64_so_arch_check",
+    srcs = [":geohash_amd64_layers"],
+    outs = ["_geohash_amd64_so_arch.ok"],
+    cmd = """
+tar_file=$$(echo $(SRCS) | tr ' ' '\\n' | grep 'geohash_layers.*\\.tar' | head -1)
+$(location check_so_arch.sh) $$tar_file 3e00
+touch $@
+""",
+    tools = [":check_so_arch.sh"],
+)
+
+genrule(
+    name = "_geohash_arm64_so_arch_check",
+    srcs = [":geohash_arm64_layers"],
+    outs = ["_geohash_arm64_so_arch.ok"],
+    cmd = """
+tar_file=$$(echo $(SRCS) | tr ' ' '\\n' | grep 'geohash_layers.*\\.tar' | head -1)
+$(location check_so_arch.sh) $$tar_file b700
+touch $@
+""",
+    tools = [":check_so_arch.sh"],
+)
+
+build_test(
+    name = "geohash_so_arch_amd64_test",
+    targets = [":_geohash_amd64_so_arch_check"],
+)
+
+build_test(
+    name = "geohash_so_arch_arm64_test",
+    targets = [":_geohash_arm64_so_arch_check"],
+)

e2e/cases/uv-deps-650/crossbuild/check_so_arch.sh

@@ -0,0 +1,37 @@
+#!/usr/bin/env bash
+# Check that _geohash.so inside a py_image_layer tar is ELF for the expected arch.
+#
+# Uses only POSIX utilities (tar, od, find) — no Python dependency.
+#
+# Usage: check_so_arch.sh <tar_file> <expected_machine_le_hex>
+#   expected_machine_le_hex: ELF e_machine as little-endian hex bytes
+#     x86_64  (EM_X86_64  = 62  = 0x3e) → "3e00"
+#     aarch64 (EM_AARCH64 = 183 = 0xb7) → "b700"
+set -euo pipefail
+
+tar_file="${1:?usage: check_so_arch.sh <tar_file> <expected_machine_le_hex>}"
+expected="${2:?expected ELF machine as little-endian hex}"
+
+tmp="$(mktemp -d)"
+trap 'rm -rf "$tmp"' EXIT
+tar xf "$tar_file" -C "$tmp" 2>/dev/null || true
+
+so="$(find "$tmp" -name '_geohash*.so' -type f | head -1)"
+if [ -z "$so" ]; then
+    echo "FAIL: no _geohash*.so found in $tar_file"
+    exit 1
+fi
+
+magic="$(od -A n -t x1 -N 4 "$so" | tr -d ' ')"
+if [ "$magic" != "7f454c46" ]; then
+    echo "FAIL: $(basename "$so") is not ELF (magic=0x${magic})"
+    exit 1
+fi
+
+machine="$(od -A n -t x1 -j 18 -N 2 "$so" | tr -d ' ')"
+if [ "$machine" != "$expected" ]; then
+    echo "FAIL: $(basename "$so") ELF machine=0x${machine} expected=0x${expected}"
+    exit 1
+fi
+
+echo "PASS: $(basename "$so") is ELF machine=0x${machine}"