mplang-nightly 0.1.dev259__py3-none-any.whl → 0.1.dev260__py3-none-any.whl

mplang/v2/backends/tee_impl.py

@@ -155,7 +155,8 @@ def _quote_gen_impl(
             f"got {type(pk).__name__}"
         )
 
-    platform = op.attrs.get("platform", "mock")
+    # In a real implementation, the platform would be detected from the environment
+    platform = "mock"
 
     return MockQuoteValue(
         platform=platform,

mplang/v2/dialects/tee.py

@@ -22,11 +22,11 @@ computation where:
 
 Architecture:
     PublicKey (from crypto.kem_keygen)
-        ↓ quote_gen(pk, platform)
-    Quote[platform]  (cryptographic attestation proof)
+        ↓ quote_gen(pk)
+    Quote  (cryptographic attestation proof)
         ↓ (transfer to verifier)
         ↓ attest(quote)
-    AttestedKey[platform, curve]  (verified TEE public key)
+    AttestedKey[curve]  (verified TEE public key)
         ↓ crypto.kem_derive(local_sk, attested_pk)
     SharedSecret  (secure channel with TEE)
 
@@ -71,7 +71,6 @@ from mplang.v2.edsl import serde
 # --- Type Definitions
 # ==============================================================================
 
-Platform = Literal["mock", "sgx", "tdx", "sev"]
 KeyCurve = Literal["x25519", "secp256k1"]
 
 
@@ -86,37 +85,29 @@ class QuoteType(elt.BaseType):
 
     The quote can be verified by anyone with access to the TEE vendor's
     root certificates (Intel, AMD, etc.).
-
-    Attributes:
-        platform: TEE platform identifier
     """
 
-    def __init__(self, platform: str = "mock"):
-        self.platform: Platform = platform  # type: ignore[assignment]
-
     def __str__(self) -> str:
-        return f"TEEQuote[{self.platform}]"
+        return "TEEQuote"
 
     def __repr__(self) -> str:
-        return f"QuoteType(platform={self.platform!r})"
+        return "QuoteType()"
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, QuoteType):
-            return False
-        return self.platform == other.platform
+        return isinstance(other, QuoteType)
 
     def __hash__(self) -> int:
-        return hash(("QuoteType", self.platform))
+        return hash("QuoteType")
 
     # --- Serde methods ---
     _serde_kind: ClassVar[str] = "tee.QuoteType"
 
     def to_json(self) -> dict[str, Any]:
-        return {"platform": self.platform}
+        return {}
 
     @classmethod
     def from_json(cls, data: dict[str, Any]) -> QuoteType:
-        return cls(platform=data["platform"])
+        return cls()
 
 
 @serde.register_class
@@ -134,37 +125,35 @@ class AttestedKeyType(elt.BaseType):
     3. The public key is bound in the quote's report_data
 
     Attributes:
-        platform: TEE platform that generated the original quote
         curve: Cryptographic curve of the key
     """
 
-    def __init__(self, platform: str = "mock", curve: str = "x25519"):
-        self.platform: Platform = platform  # type: ignore[assignment]
+    def __init__(self, curve: str = "x25519"):
         self.curve: KeyCurve = curve  # type: ignore[assignment]
 
     def __str__(self) -> str:
-        return f"AttestedKey[{self.platform}, {self.curve}]"
+        return f"AttestedKey[{self.curve}]"
 
     def __repr__(self) -> str:
-        return f"AttestedKeyType(platform={self.platform!r}, curve={self.curve!r})"
+        return f"AttestedKeyType(curve={self.curve!r})"
 
     def __eq__(self, other: object) -> bool:
         if not isinstance(other, AttestedKeyType):
             return False
-        return self.platform == other.platform and self.curve == other.curve
+        return self.curve == other.curve
 
     def __hash__(self) -> int:
-        return hash(("AttestedKeyType", self.platform, self.curve))
+        return hash(("AttestedKeyType", self.curve))
 
     # --- Serde methods ---
     _serde_kind: ClassVar[str] = "tee.AttestedKeyType"
 
     def to_json(self) -> dict[str, Any]:
-        return {"platform": self.platform, "curve": self.curve}
+        return {"curve": self.curve}
 
     @classmethod
     def from_json(cls, data: dict[str, Any]) -> AttestedKeyType:
-        return cls(platform=data["platform"], curve=data["curve"])
+        return cls(curve=data["curve"])
 
 
 @serde.register_class
@@ -173,37 +162,29 @@ class MeasurementType(elt.BaseType):
 
     Represents a cryptographic hash of the code and initial configuration
     running inside the TEE. Used to verify the TEE is running expected code.
-
-    Attributes:
-        platform: TEE platform
     """
 
-    def __init__(self, platform: str = "mock"):
-        self.platform: Platform = platform  # type: ignore[assignment]
-
     def __str__(self) -> str:
-        return f"TEEMeasurement[{self.platform}]"
+        return "TEEMeasurement"
 
     def __repr__(self) -> str:
-        return f"MeasurementType(platform={self.platform!r})"
+        return "MeasurementType()"
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, MeasurementType):
-            return False
-        return self.platform == other.platform
+        return isinstance(other, MeasurementType)
 
     def __hash__(self) -> int:
-        return hash(("MeasurementType", self.platform))
+        return hash("MeasurementType")
 
     # --- Serde methods ---
     _serde_kind: ClassVar[str] = "tee.MeasurementType"
 
     def to_json(self) -> dict[str, Any]:
-        return {"platform": self.platform}
+        return {}
 
     @classmethod
     def from_json(cls, data: dict[str, Any]) -> MeasurementType:
-        return cls(platform=data["platform"])
+        return cls()
 
 
 # ==============================================================================
@@ -223,14 +204,11 @@ get_measurement_p = el.Primitive[el.Object]("tee.get_measurement")
 @quote_gen_p.def_abstract_eval
 def _quote_gen_ae(
     pk: elt.BaseType,
-    *,
-    platform: Platform = "mock",
 ) -> QuoteType:
     """Generate a TEE quote binding the provided public key.
 
     Args:
         pk: Public key to bind in the quote (must be PublicKeyType from crypto.kem_keygen)
-        platform: TEE platform to use
 
     Returns:
         QuoteType representing the attestation proof
@@ -243,7 +221,7 @@ def _quote_gen_ae(
             f"quote_gen expects PublicKeyType (from crypto.kem_keygen), "
             f"got {type(pk).__name__}"
         )
-    return QuoteType(platform=platform)
+    return QuoteType()
 
 
 @attest_p.def_abstract_eval
@@ -261,7 +239,7 @@ def _attest_ae(
     Returns:
         AttestedKeyType containing the verified public key
     """
-    return AttestedKeyType(platform=quote.platform, curve=expected_curve)
+    return AttestedKeyType(curve=expected_curve)
 
 
 @get_measurement_p.def_abstract_eval
@@ -276,7 +254,7 @@ def _get_measurement_ae(
     Returns:
         MeasurementType containing the code measurement
     """
-    return MeasurementType(platform=quote.platform)
+    return MeasurementType()
 
 
 # ==============================================================================
@@ -286,7 +264,6 @@ def _get_measurement_ae(
 
 def quote_gen(
     pk: el.Object,
-    platform: Platform = "mock",
 ) -> el.Object:
     """Generate a TEE attestation quote binding the provided public key.
 
@@ -297,7 +274,6 @@ def quote_gen(
 
     Args:
         pk: Public key to bind (typically from crypto.kem_keygen)
-        platform: TEE platform ("mock", "sgx", "tdx", "sev")
 
     Returns:
         Object[QuoteType] - The attestation quote
@@ -306,7 +282,7 @@ def quote_gen(
         >>> sk, pk = crypto.kem_keygen("x25519")
         >>> quote = tee.quote_gen(pk)  # Bind pk in attestation
     """
-    return quote_gen_p.bind(pk, platform=platform)
+    return quote_gen_p.bind(pk)
 
 
 def attest(
@@ -360,7 +336,6 @@ __all__ = [
     "AttestedKeyType",
     "KeyCurve",
     "MeasurementType",
-    "Platform",
     "QuoteType",
     "attest",
     "attest_p",

{mplang_nightly-0.1.dev259.dist-info → mplang_nightly-0.1.dev260.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mplang-nightly
-Version: 0.1.dev259
+Version: 0.1.dev260
 Summary: Multi-Party Programming Language
 Author-email: SecretFlow Team <secretflow-contact@service.alipay.com>
 License:                                  Apache License

{mplang_nightly-0.1.dev259.dist-info → mplang_nightly-0.1.dev260.dist-info}/RECORD

@@ -91,7 +91,7 @@ mplang/v2/backends/spu_impl.py,sha256=nDmpntXMKlFhaOUMXAOO_-RZTzqGLsgxEvwJuVA6h1
 mplang/v2/backends/spu_state.py,sha256=wj876IvNPhKyWISN6WwKBYoaDQFFJ8jemdJUVeH5IfA,4144
 mplang/v2/backends/store_impl.py,sha256=RyhADTNsnnNnwsatAMr7eeewXkVXtfNWA1oFiLXg8H0,2222
 mplang/v2/backends/table_impl.py,sha256=c36gyBCWLQbV3g0hkJeTnMXUqT0nxgu74k2sLondTio,8784
-mplang/v2/backends/tee_impl.py,sha256=5PzzQ6mibd6-Wyvvt_8DD6G-CzA4VAmqGk6H1Z9risI,6986
+mplang/v2/backends/tee_impl.py,sha256=Gp-vqqJPtEMNqP7y68tLhL3a-EW3BQwpo_qCJOSHqKs,7044
 mplang/v2/backends/tensor_impl.py,sha256=8f9f4-_e-m4JWGZSbXLmSSHcgPykRBc1sAYrA3OIxEg,18906
 mplang/v2/backends/simp_driver/__init__.py,sha256=ahOPYYvtFVwqxiFxkpSNP8BCTao_MfCXmtt5zsMaJxg,1258
 mplang/v2/backends/simp_driver/http.py,sha256=nl7ny7f8bzhy1ubNIDXhMgA5P_WA8dhhgFNHvcmfSKk,5548
@@ -115,7 +115,7 @@ mplang/v2/dialects/simp.py,sha256=ON7iegkHp3um5UX8V4Y5I-fGgFJ3YVwmFsXsleiqqUE,32
 mplang/v2/dialects/spu.py,sha256=3JO-D394TKNH2VdFDRp5ohmG0uOcOHEs_ivFHbMZIgA,11385
 mplang/v2/dialects/store.py,sha256=RqUBzMAgtEMBmdT8axV5lVCv1hp5w0ZZM0Tu4iOZt-c,2114
 mplang/v2/dialects/table.py,sha256=ax9Yjvcb8jJ8fqNJodMQ_mrS8tf-xECHQFvUKUWPp70,12714
-mplang/v2/dialects/tee.py,sha256=R_XySt-JNhZJNdjRiUw-nU4BjwBNMtV4sYFMOtvJZzc,11506
+mplang/v2/dialects/tee.py,sha256=oj_G8ebhtuz9_HarK8rKoaJNJ9ZkRbqcIxhp3m0xsjQ,10129
 mplang/v2/dialects/tensor.py,sha256=FxPKsiNi2vFb-R2hqRgR7zYYO5LdRu_rdDHfjE3_2Lw,40003
 mplang/v2/edsl/README.md,sha256=viflvdRojOa6Xk_UMRPqpuPGXcPGmdlv2-XR6LO7B58,7592
 mplang/v2/edsl/__init__.py,sha256=YqmtrJXD1NLKS-_Ofnxtiv77muokTZnrAiV7dXUZVyo,2607
@@ -170,8 +170,8 @@ mplang/v2/runtime/dialect_state.py,sha256=HxO1i4kSOujS2tQzAF9-WmI3nChSaGgupf2_07
 mplang/v2/runtime/interpreter.py,sha256=UzrM5oepka6H0YKRZncNXhsuwKVm4pliG5J92fFRZMI,32300
 mplang/v2/runtime/object_store.py,sha256=yT6jtKG2GUEJVmpq3gnQ8mCMvUFYzgBciC5A-J5KRdk,5998
 mplang/v2/runtime/value.py,sha256=CMOxElJP78v7pjasPhEpbxWbSgB2KsLbpPmzz0mQX0E,4317
-mplang_nightly-0.1.dev259.dist-info/METADATA,sha256=fh4TjDPfvnmmK5d-HNBXFR447PIg-i9BDtpaayzDCs0,16768
-mplang_nightly-0.1.dev259.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mplang_nightly-0.1.dev259.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev259.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev259.dist-info/RECORD,,
+mplang_nightly-0.1.dev260.dist-info/METADATA,sha256=16iSjw54jMsEvkUSk4HqpmDOh6BCd5O6lxAdiRV-GWc,16768
+mplang_nightly-0.1.dev260.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mplang_nightly-0.1.dev260.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev260.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev260.dist-info/RECORD,,