Native Binary Patching

val soFile = get("lib/arm64-v8a/libchrome.so")
if (!soFile.exists()) {
    return@execute
}

data class HostEntry(
    val offset: Long,
    val hostName: String,
)

val hostEntries = listOf(
    HostEntry(0x001f9329L, "star-randsrv.bsg.brave.com"),
    HostEntry(0x001f935aL, "collector.bsg.brave.com"),
    HostEntry(0x001f9391L, "usage-ping.brave.com"),
)

val redirectionIp = "0.0.0.0".toByteArray(Charsets.US_ASCII)
var writtenHosts = 0

RandomAccessFile(soFile, "rw").use { raf ->
    for (entry in hostEntries) {
        val expectedBytes = entry.hostName.toByteArray(Charsets.US_ASCII)
        val len = expectedBytes.size
        if (entry.offset + len > raf.length()) {
            throw PatchException("Host offset 0x${entry.offset.toString(16)} out of bounds in libchrome.so")
        }
        val buf = ByteArray(len)
        raf.seek(entry.offset)
        raf.readFully(buf)
        if (!buf.contentEquals(expectedBytes)) {
            throw PatchException(
                "Host fingerprint mismatch at 0x${entry.offset.toString(16)}. " +
                    "Expected: ${entry.hostName}, Found: ${String(buf, Charsets.US_ASCII)}",
            )
        }

        // Construct replacement: "0.0.0.0" + null byte + zero padding to original length
        val replacement = ByteArray(len)
        System.arraycopy(redirectionIp, 0, replacement, 0, redirectionIp.size)
        // Remaining bytes default to 0x00 (null padded)

        raf.seek(entry.offset)
        raf.write(replacement)
        writtenHosts++
    }
}

println("[Block Telemetry] Redirected $writtenHosts / ${hostEntries.size} endpoints to 0.0.0.0 in libchrome.so")

val syncOffsets = listOf(0x0031b2b2L, 0x0031b2b5L, 0x0031b34fL)
val targetUrl = "https://endpoint.example.com/v1/telemetry"
val expectedBytes = targetUrl.toByteArray(Charsets.US_ASCII)
val len = expectedBytes.size
val redirectionIp = "0.0.0.0".toByteArray(Charsets.US_ASCII)

RandomAccessFile(soFile, "rw").use { raf ->
    val validOffset = syncOffsets.firstOrNull { offset ->
        if (offset + len > raf.length()) return@firstOrNull false
        val buf = ByteArray(len)
        raf.seek(offset)
        raf.readFully(buf)
        buf.contentEquals(expectedBytes)
    } ?: throw PatchException(
        "Sync fingerprint mismatch for $targetUrl in libchrome.so",
    )

    val replacement = ByteArray(len)
    System.arraycopy(redirectionIp, 0, replacement, 0, redirectionIp.size)

    raf.seek(validOffset)
    raf.write(replacement)
}

private val EMPTY_STUB_BYTES = byteArrayOf()

private val BLOAT_NATIVE_LIBS = listOf(
    "libimpress_api_jni.so",
    "libwg-go.so",
    "libandroidx.xr.arcore.openxr.so",
    "libarcore_sdk_c.so",
)

val libDir = get("lib/arm64-v8a")
if (libDir.exists() && libDir.isDirectory) {
    BLOAT_NATIVE_LIBS.forEach { libName ->
        val libFile = get("lib/arm64-v8a/$libName")
        if (libFile.exists() && libFile.isFile) {
            val originalSize = libFile.length()
            if (originalSize > 0) {
                libFile.writeBytes(EMPTY_STUB_BYTES)
                savedBytes += (originalSize - libFile.length())
                strippedLibs.add(libName)
            }
        }
    }
}