Feat/semantic handler generation

.github/workflows/validate-examples.yml

@@ -9,24 +9,8 @@ on:
       - main
     types: [opened, synchronize]
 jobs:
-  discover:
-    runs-on: ubuntu-latest
-    outputs:
-      examples: ${{ steps.discover.outputs.examples }}
-    steps:
-      - uses: actions/checkout@v4
-
-      - id: discover
-        run: |
-          examples=$(find examples -mindepth 2 -maxdepth 2 -type d | jq -R -s -c 'split("\n") | map(select(length > 0))')
-          echo "examples=$examples" >> $GITHUB_OUTPUT
-
   validate:
-    needs: [discover]
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        example: ${{ fromJson(needs.discover.outputs.examples) }}
     steps:
       - uses: actions/checkout@v4
       - uses: dtolnay/rust-toolchain@stable
@@ -44,45 +28,14 @@ jobs:
           sudo apt-get update
           sudo apt-get install -y mingw-w64 wine wine32 wine64
 
-      - name: Generate templates
-        env:
-          OUTPUT_DIRECTORY: ${{ github.workspace }}/api/generated
-          TEMPLATES_DIRECTORY: ${{ github.workspace }}/api/templates
-        run: cargo build --package markers
-
       - name: Build for Windows
         run: cargo build --bin obfuscator --release --target x86_64-pc-windows-gnu
 
-      - name: Build examples
-        run: |
-          language=$(echo "${{ matrix.example }}" | cut -d/ -f2)
-          case $language in
-            rust) cargo build --manifest-path ${{ matrix.example }}/Cargo.toml --release --target x86_64-pc-windows-gnu ;;
-            cpp)  x86_64-w64-mingw32-g++ ${{ matrix.example }}/main.cpp -o ${{ matrix.example }}/example.exe -static-libgcc -static-libstdc++ ;;
-          esac
+      - name: Build example
+        run: x86_64-w64-mingw32-g++ examples/cpp/hello/main.cpp -o example.exe
 
-      - name: Obfuscate examples
-        run: |
-          language=$(echo "${{ matrix.example }}" | cut -d/ -f2)
-          case $language in
-            rust)
-              name=$(cargo metadata --manifest-path ${{ matrix.example }}/Cargo.toml --no-deps --format-version 1 | jq -r '.packages[0].name')
-              wine target/x86_64-pc-windows-gnu/release/obfuscator.exe ${{ matrix.example }}/target/x86_64-pc-windows-gnu/release/$name.exe --virtualization
-              ;;
-            cpp)
-              wine target/x86_64-pc-windows-gnu/release/obfuscator.exe ${{ matrix.example }}/example.exe --virtualization
-              ;;
-          esac
+      - name: Obfuscate example
+        run: wine target/x86_64-pc-windows-gnu/release/obfuscator.exe example.exe --virtualization
 
-      - name: Validate examples
-        run: |
-          language=$(echo "${{ matrix.example }}" | cut -d/ -f2)
-          case $language in
-            rust)
-              name=$(cargo metadata --manifest-path ${{ matrix.example }}/Cargo.toml --no-deps --format-version 1 | jq -r '.packages[0].name')
-              wine ${{ matrix.example }}/target/x86_64-pc-windows-gnu/release/$name.exe
-              ;;
-            cpp)
-              wine ${{ matrix.example }}/example.exe
-              ;;
-          esac
+      - name: Validate example
+        run: wine example.exe

Cargo.toml

@@ -1,4 +1,4 @@
 [workspace]
-members = ["logger", "obfuscator", "runtime", "tests", "markers"]
+members = ["logger", "obfuscator", "runtime", "tests"]
 resolver = "2"
 exclude = ["examples"]

README.md

@@ -7,7 +7,7 @@ Obfuscator for compiled 64-bit portable executables.
 ### 1. Obfuscator
 
 #### Analysis
-- **Disassembly**: the binary's code section is disassembled and partitioned into basic blocks by tracing all reachable control flow or by scanning for explicit inline markers as shown in the [examples](https://github.com/RadonCoding/binsafe/tree/main/examples).
+- **Disassembly**: the binary's code section is disassembled and partitioned into basic blocks by tracing all reachable control flow or by scanning the `.binsafe` section that [marked functions](https://github.com/RadonCoding/binsafe/tree/main/examples) are placed in.
 
 #### Virtualization
 - **Lifting**: instructions are translated into bytecode that the runtime interprets.

api/binsafe.hpp

@@ -0,0 +1,9 @@
+#pragma once
+
+#if defined(__GNUC__) || defined(__clang__)
+#define BINSAFE __attribute__((section(".binsafe"), noinline))
+#elif defined(_MSC_VER)
+#define BINSAFE __declspec(code_seg(".binsafe")) __declspec(noinline)
+#else
+#define BINSAFE
+#endif

api/binsafe.rs

@@ -0,0 +1,8 @@
+#[macro_export]
+macro_rules! binsafe {
+    ($item:item) => {
+        #[link_section = ".binsafe"]
+        #[inline(never)]
+        $item
+    };
+}

example.ps1

@@ -1,50 +1,103 @@
+using namespace System.IO
+
 param(
-    [Parameter(Mandatory)]
-    [string]$Example
-)
+    [Parameter(Mandatory, Position = 0)]
+    [DirectoryInfo]$Example,
 
-$ErrorActionPreference = "Stop"
+    [Parameter(Position = 1)]
+    [ValidateSet("debug", "release")]
+    [string]$Configuration = "debug",
 
-$language = $Example.Split("/")[1]
+    [Parameter(Position = 2)]
+    [string[]]$Arguments
+)
 
-$env:OUTPUT_DIRECTORY = "$PWD/api/generated"
-$env:TEMPLATES_DIRECTORY = "$PWD/api/templates"
+$ErrorActionPreference = "Stop"
 
-cargo build --package markers
+$language = Split-Path (Split-Path $Example -Parent) -Leaf
 
-if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
-
-cargo build --bin obfuscator
+if ($Configuration -eq "release") {
+    cargo build --bin obfuscator --release $Arguments
+}
+else {
+    cargo build --bin obfuscator $Arguments
+}
 
 if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
 
 switch ($language) {
     "rust" {
-        cargo build --manifest-path "$Example/Cargo.toml"
+        $manifest = Join-Path $Example.FullName "Cargo.toml"
+
+        if ($Configuration -eq "release") {
+            cargo build --manifest-path $manifest --release
+        }
+        else {
+            cargo build --manifest-path $manifest
+        }
 
         if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
 
-        $metadata = cargo metadata --manifest-path "$Example/Cargo.toml" --format-version 1 --no-deps | ConvertFrom-Json
+        $metadata = cargo metadata `
+            --manifest-path $manifest `
+            --format-version 1 `
+            --no-deps | ConvertFrom-Json
+
+        $manifest = [Path]::GetFullPath($manifest)
 
-        $package = $metadata.packages[0]
-        $target = $package.targets | Where-Object { $_.kind -contains "bin" } | Select-Object -First 1
+        $package = $metadata.packages |
+            Where-Object {
+                [Path]::GetFullPath($_.manifest_path) -eq $manifest
+            } |
+            Select-Object -First 1
 
-        $source = Join-Path $metadata.target_directory "debug\$($target.name).exe"
+        $target = $package.targets |
+            Where-Object { $_.kind -contains "bin" } |
+            Select-Object -First 1
+
+        $source = Join-Path `
+            $metadata.target_directory `
+            "$Configuration\$($target.name).exe"
     }
+
     "cpp" {
-        g++ "$Example/main.cpp"
-
+        $main = Join-Path $Example.FullName "main.cpp"
+
+        $source = Join-Path `
+            $Example.Parent.Parent.FullName `
+            "$($Example.Name).exe"
+
+        if ($Configuration -eq "release") {
+            g++ $main -o $source -O2
+        }
+        else {
+            g++ $main -o $source -O0 -g
+        }
+
         if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
+    }
 
-        $source = Join-Path $PWD "a.exe"
+    default {
+        throw "Unsupported language: $language"
     }
 }
 
-& "target/debug/obfuscator.exe" --virtualization $source
+$source = [Path]::GetFullPath($source)
+
+if (-not (Test-Path -LiteralPath $source -PathType Leaf)) {
+    throw "Source executable not found: $source"
+}
+
+$obfuscator = Join-Path $PWD "target\$Configuration\obfuscator.exe"
+
+& $obfuscator --virtualization -v $source
 
 if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
 
-$destination = [System.IO.Path]::ChangeExtension($source, ".protected.exe")
+$destination = [Path]::ChangeExtension(
+    $source,
+    ".protected.exe"
+)
 
 & $destination
 
@@ -53,4 +106,4 @@ $code = $LASTEXITCODE
 Remove-Item $source -Force -ErrorAction SilentlyContinue
 Remove-Item $destination -Force -ErrorAction SilentlyContinue
 
-exit $code
+exit $code