From a6d8fed9e4614653d276638784c16db370bb0af5 Mon Sep 17 00:00:00 2001
From: Simonas Kazlauskas <git@kazlauskas.me>
Date: Mon, 18 Sep 2023 15:02:27 +0300
Subject: [PATCH] Add a zkASM architecture

This is a very initial take on this if only to be able to specify
`zkasm-unknown-unknown` target in tooling that uses `target-lexicon`.
---
 Cargo.toml     |  2 ++
 src/targets.rs | 15 +++++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/Cargo.toml b/Cargo.toml
index dec31f1..9838685 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -21,6 +21,8 @@ serde_json = "1.0"
 default = []
 serde_support = ["serde", "std"]
 std = []
+# Enable (unstable) support for the zkasm architecture.
+arch_zkasm = []
 
 [badges]
 maintenance = { status = "passively-maintained" }
diff --git a/src/targets.rs b/src/targets.rs
index d2a649f..d03d202 100644
--- a/src/targets.rs
+++ b/src/targets.rs
@@ -47,6 +47,11 @@ pub enum Architecture {
     X86_64h,
     XTensa,
     Clever(CleverArchitecture),
+    /// A software machine that produces zero-knowledge proofs of the execution.
+    ///
+    /// See https://wiki.polygon.technology/docs/category/zk-assembly/
+    #[cfg(feature = "arch_zkasm")]
+    ZkAsm,
 }
 
 #[cfg_attr(feature = "rust_1_40", non_exhaustive)]
@@ -856,6 +861,8 @@ impl Architecture {
             | Sparc
             | Sparc64
             | Sparcv9 => Ok(Endianness::Big),
+            #[cfg(feature="arch_zkasm")]
+            ZkAsm => Ok(Endianness::Big),
         }
 
     }
@@ -896,6 +903,8 @@ impl Architecture {
             | LoongArch64
             | Wasm64
             | Clever(_) => Ok(PointerWidth::U64),
+            #[cfg(feature="arch_zkasm")]
+            ZkAsm => Ok(PointerWidth::U64),
         }
     }
 
@@ -943,6 +952,8 @@ impl Architecture {
             X86_64h => Cow::Borrowed("x86_64h"),
             XTensa => Cow::Borrowed("xtensa"),
             Clever(ver) => ver.into_str(),
+            #[cfg(feature = "arch_zkasm")]
+            ZkAsm => Cow::Borrowed("zkasm"),
         }
     }
 }
@@ -1218,6 +1229,8 @@ impl FromStr for Architecture {
             "x86_64" => X86_64,
             "x86_64h" => X86_64h,
             "xtensa" => XTensa,
+            #[cfg(feature = "arch_zkasm")]
+            "zkasm" => ZkAsm,
             _ => {
                 if let Ok(arm) = ArmArchitecture::from_str(s) {
                     Arm(arm)
@@ -1709,6 +1722,8 @@ mod tests {
             "x86_64-wrs-vxworks",
             "xtensa-esp32-espidf",
             "clever-unknown-elf",
+            #[cfg(feature = "arch_zkasm")]
+            "zkasm-unknown-unknown",
         ];
 
         for target in targets.iter() {