crosscompiler (#21)

Author: vaporif

flake.nix

@@ -87,6 +87,34 @@
         "rust-analyzer"
         "llvm-tools"
       ];
+
+      devToolchain =
+        if pkgs.stdenv.isLinux
+        then
+          fenixPkgs.combine [
+            fenixPkgs.stable.cargo
+            fenixPkgs.stable.clippy
+            fenixPkgs.stable.rustc
+            fenixPkgs.stable.rustfmt
+            fenixPkgs.stable.rust-src
+            fenixPkgs.stable.rust-analyzer
+            fenixPkgs.stable.llvm-tools
+            fenixPkgs.targets."x86_64-unknown-linux-musl".stable.rust-std
+            fenixPkgs.targets."aarch64-unknown-linux-musl".stable.rust-std
+          ]
+        else if pkgs.stdenv.isDarwin && pkgs.stdenv.isAarch64
+        then
+          fenixPkgs.combine [
+            fenixPkgs.stable.cargo
+            fenixPkgs.stable.clippy
+            fenixPkgs.stable.rustc
+            fenixPkgs.stable.rustfmt
+            fenixPkgs.stable.rust-src
+            fenixPkgs.stable.rust-analyzer
+            fenixPkgs.stable.llvm-tools
+            fenixPkgs.targets."x86_64-apple-darwin".stable.rust-std
+          ]
+        else toolchain;
     in {
       packages = {
         inherit pkg cargoArtifacts;
@@ -128,7 +156,7 @@
       devShells.default = pkgs.mkShell {
         packages =
           [
-            toolchain
+            devToolchain
             pkgs.cargo-nextest
             pkgs.cargo-llvm-cov
             pkgs.cargo-deny
@@ -142,6 +170,8 @@
           ++ pkgs.lib.optionals pkgs.stdenv.isLinux [
             pkgs.pkg-config
             pkgs.openssl
+            pkgs.pkgsCross.musl64.stdenv.cc
+            pkgs.pkgsCross.aarch64-multiplatform-musl.stdenv.cc
           ]
           ++ pkgs.lib.optionals pkgs.stdenv.isDarwin [
             pkgs.apple-sdk_26
@@ -150,11 +180,23 @@
         env =
           {
             RUST_BACKTRACE = "1";
-            RUST_SRC_PATH = "${toolchain}/lib/rustlib/src/rust/library";
+            RUST_SRC_PATH = "${devToolchain}/lib/rustlib/src/rust/library";
             ORT_DYLIB_PATH = "${pkgs.onnxruntime}/lib/libonnxruntime${pkgs.stdenv.hostPlatform.extensions.sharedLibrary}";
           }
           // pkgs.lib.optionalAttrs pkgs.stdenv.isLinux {
             LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath [pkgs.openssl pkgs.stdenv.cc.cc.lib];
+            CARGO_TARGET_X86_64_UNKNOWN_LINUX_MUSL_LINKER = "${pkgs.pkgsCross.musl64.stdenv.cc}/bin/${pkgs.pkgsCross.musl64.stdenv.cc.targetPrefix}cc";
+            CC_x86_64_unknown_linux_musl = "${pkgs.pkgsCross.musl64.stdenv.cc}/bin/${pkgs.pkgsCross.musl64.stdenv.cc.targetPrefix}cc";
+            CFLAGS_x86_64_unknown_linux_musl = "-U_FORTIFY_SOURCE";
+            CARGO_TARGET_AARCH64_UNKNOWN_LINUX_MUSL_LINKER = "${pkgs.pkgsCross.aarch64-multiplatform-musl.stdenv.cc}/bin/${pkgs.pkgsCross.aarch64-multiplatform-musl.stdenv.cc.targetPrefix}cc";
+            CC_aarch64_unknown_linux_musl = "${pkgs.pkgsCross.aarch64-multiplatform-musl.stdenv.cc}/bin/${pkgs.pkgsCross.aarch64-multiplatform-musl.stdenv.cc.targetPrefix}cc";
+            CFLAGS_aarch64_unknown_linux_musl = "-U_FORTIFY_SOURCE";
+            X86_64_UNKNOWN_LINUX_MUSL_OPENSSL_STATIC = "1";
+            X86_64_UNKNOWN_LINUX_MUSL_OPENSSL_LIB_DIR = "${pkgs.pkgsCross.musl64.openssl.out}/lib";
+            X86_64_UNKNOWN_LINUX_MUSL_OPENSSL_INCLUDE_DIR = "${pkgs.pkgsCross.musl64.openssl.dev}/include";
+            AARCH64_UNKNOWN_LINUX_MUSL_OPENSSL_STATIC = "1";
+            AARCH64_UNKNOWN_LINUX_MUSL_OPENSSL_LIB_DIR = "${pkgs.pkgsCross.aarch64-multiplatform-musl.openssl.out}/lib";
+            AARCH64_UNKNOWN_LINUX_MUSL_OPENSSL_INCLUDE_DIR = "${pkgs.pkgsCross.aarch64-multiplatform-musl.openssl.dev}/include";
           };
       };
     });