Added API to get enclave platform ID

lsd-ucsc · Jul 4, 2024 · 3538fe5 · 3538fe5
1 parent 7715e1a
commit 3538fe5
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 0 deletions.
diff --git a/contracts/DecentServerCertMgr.sol b/contracts/DecentServerCertMgr.sol
@@ -88,4 +88,22 @@ contract DecentServerCertMgr {
             bytes32(0);
     }
 
+    /**
+     * Get the platform ID of a Decent Server
+     * @param svrKeyAddr Address derived from the Decent Server public key
+     * @return bytes32 Platform ID of the Decent Server
+     */
+     function getPlatformId(address svrKeyAddr)
+        external
+        view
+        returns (bytes32)
+    {
+        DecentServerCert.DecentServerCertObj storage cert =
+            m_serverCerts[svrKeyAddr];
+
+        require(cert.isVerified, "Server not verified");
+
+        return cert.platformId;
+    }
+
 }
diff --git a/contracts/Interface_DecentServerCertMgr.sol b/contracts/Interface_DecentServerCertMgr.sol
@@ -35,4 +35,14 @@ interface Interface_DecentServerCertMgr {
         view
         returns (bytes32);
 
+    /**
+     * Get the platform ID of a Decent Server
+     * @param svrKeyAddr Address derived from the Decent Server public key
+     * @return bytes32 Platform ID of the Decent Server
+     */
+     function getPlatformId(address svrKeyAddr)
+        external
+        view
+        returns (bytes32);
+
 }
diff --git a/tests/DecentServer/03_DecentServerCertMgr_test.sol b/tests/DecentServer/03_DecentServerCertMgr_test.sol
@@ -69,6 +69,14 @@ contract DecentServerCertMgr_testSuit {
             TestCerts.DECENT_SVR_CERT_ENCL_HASH,
             "encHash not match"
         );
+
+        bytes32 platId = decentServerCertMgr.getPlatformId(TestCerts.DECENT_SVR_CERT_KEY_ADDR);
+
+        Assert.equal(
+            platId,
+            TestCerts.DECENT_SVR_CERT_PLAT_ID,
+            "platform ID not match"
+        );
     }
 
     function verifyCertTest() public {
@@ -93,6 +101,14 @@ contract DecentServerCertMgr_testSuit {
             TestCerts.DECENT_SVR_CERT_ENCL_HASH,
             "encHash not match"
         );
+
+        bytes32 platId = decentServerCertMgr.getPlatformId(TestCerts.DECENT_SVR_CERT_KEY_ADDR);
+
+        Assert.equal(
+            platId,
+            TestCerts.DECENT_SVR_CERT_PLAT_ID,
+            "platform ID not match"
+        );
     }
 
     function verifyInvalidCertTest() public {