From da006e52ae45e01d0537df44787ecf44f07e5ad2 Mon Sep 17 00:00:00 2001
From: DL6ER <dl6er@dl6er.de>
Date: Mon, 25 May 2020 18:55:21 +0200
Subject: [PATCH] Add checks for the compiled binary for all supported CI
 platforms.

Signed-off-by: DL6ER <dl6er@dl6er.de>
---
 .circleci/config.yml |   3 ++
 test/arch_test.sh    | 105 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 108 insertions(+)
 create mode 100644 test/arch_test.sh

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 36c3841e..0b13a99d 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -16,6 +16,9 @@ version: 2
           [[ $CIRCLE_JOB == *"qemu"* ]] && DOCKERIFNEEDED="docker run --rm -v $(pwd):/workspace -w /workspace pihole/ftl-build:arm-qemu "
           $DOCKERIFNEEDED make GIT_BRANCH="${BRANCH}" GIT_TAG="${CIRCLE_TAG}" STATIC="${STATIC}"
           file pihole-FTL
+    - run:
+        name: "Binary checks"
+        command: bash test/arch_test.sh
     - run:
         name: "Upload"
         command: |
diff --git a/test/arch_test.sh b/test/arch_test.sh
new file mode 100644
index 00000000..3e0906d3
--- /dev/null
+++ b/test/arch_test.sh
@@ -0,0 +1,105 @@
+#!/bin/bash
+# Pi-hole: A black hole for Internet advertisements
+# (c) 2020 Pi-hole, LLC (https://pi-hole.net)
+# Network-wide ad blocking via your own hardware.
+#
+# FTL Engine
+# Binary target tests
+#
+# This file is copyright under the latest version of the EUPL.
+# Please see LICENSE file for your rights under this license. */
+
+check_libs() {
+  mapfile -t libs < <(readelf -d ./pihole-FTL | grep "Shared library" | grep -oE "\[.*\]")
+  if [[ "${libs[*]}" != "${1}" ]]; then
+    echo "Wrong libraries"
+    echo "Found: ${libs[*]}"
+    echo "Expected: ${1}"
+    exit 1
+  fi
+  echo "Library checks: OK (using ${#libs[*]} shared libraries)"
+}
+
+check_machine() {
+  mapfile -t header < <(readelf -h ./pihole-FTL | grep -E "(Class)|(Machine)" | sed "s/.*://;s/ \{2,\}//g;")
+  if [[ "${header[0]}" != "${1}" || "${header[1]}" != "${2}" ]]; then
+    echo "Wrong machine"
+    echo "Expected: Class: ${1} Machine: ${2}"
+    echo "Found: Class: ${header[0]} Machine: ${header[1]}"
+    exit 1
+  fi
+  echo "Machine checks: OK (${1} binary for ${2})"
+}
+
+check_CPU_arch() {
+  cpuarch="$(readelf -A ./pihole-FTL | grep "Tag_CPU_arch:" | sed "s/^ *//")"
+  if [[ "${cpuarch}" != "Tag_CPU_arch: ${1}" ]]; then
+    echo "Wrong CPU arch"
+    echo "Expected: Tag_CPU_arch: ${1}"
+    echo "Found: ${cpuarch}"
+    exit 1
+  fi
+  echo "CPU architecture checks: OK (${1})"
+}
+
+check_FP_arch() {
+  fparch="$(readelf -A ./pihole-FTL | grep "Tag_FP_arch:" | sed "s/^ *//")"
+  if [[ "${fparch}" != "Tag_FP_arch: ${1}" && -n "${1}" ]]; then
+    echo "Wrong FP arch"
+    echo "Expected: Tag_FP_arch: ${1}"
+    echo "Found: ${fparch}"
+    exit 1
+  fi
+  echo "FP architecture checks: OK (${1})"
+}
+
+if [[ "${CIRCLE_JOB}" == "x86_64" ]]; then
+
+  check_machine "ELF64" "Advanced Micro Devices X86-64"
+  check_libs "[librt.so.1] [libpthread.so.0] [libc.so.6]"
+
+elif [[ "${CIRCLE_JOB}" == "x86_64-musl" ]]; then
+
+  check_machine "ELF64" "Advanced Micro Devices X86-64"
+  check_libs "" # No dependency on any shared library is intended
+
+elif [[ "${CIRCLE_JOB}" == "x86_32" ]]; then
+
+  check_machine "ELF32" "Intel 80386"
+  check_libs "[librt.so.1] [libpthread.so.0] [libc.so.6]"
+
+elif [[ "${CIRCLE_JOB}" == "aarch64" ]]; then
+
+  check_machine "ELF64" "AArch64"
+  check_libs "[librt.so.1] [libpthread.so.0] [libc.so.6] [ld-linux-aarch64.so.1]"
+
+elif [[ "${CIRCLE_JOB}" == "arm" ]]; then
+
+  check_machine "ELF32" "ARM"
+  check_libs "[librt.so.1] [libgcc_s.so.1] [libpthread.so.0] [libc.so.6] [ld-linux-armhf.so.3]"
+
+  check_CPU_arch "v6"
+  check_FP_arch "VFPv2"
+
+elif [[ "${CIRCLE_JOB}" == "armhf" ]]; then
+
+  check_machine "ELF32" "ARM"
+  check_libs "[librt.so.1] [libgcc_s.so.1] [libpthread.so.0] [libc.so.6] [ld-linux-armhf.so.3]"
+
+  check_CPU_arch "v7"
+  check_FP_arch "VFPv3-D16"
+
+elif [[ "${CIRCLE_JOB}" == "arm-qemu" ]]; then
+
+  check_machine "ELF32" "ARM"
+  check_libs "[librt.so.1] [libgcc_s.so.1] [libpthread.so.0] [libc.so.6] [ld-linux.so.3]"
+
+  check_CPU_arch "v6"
+  check_FP_arch "" # No specified FP arch
+
+else
+  echo "Invalid job ${CIRCLE_JOB}"
+  exit 1
+fi
+
+exit 0
\ No newline at end of file