From 37fbc1ddfff7e69c7871bc9e203cf4323bb55934 Mon Sep 17 00:00:00 2001
From: Anthony Hu <anthony@wolfssl.com>
Date: Wed, 22 Jun 2022 15:24:43 -0400
Subject: [PATCH] Adding CSharp wrapper for `wolfTPM2_CreatePrimaryKey()`.

---
 wrapper/CSharp/wolfTPM.cs | 112 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 111 insertions(+), 1 deletion(-)

diff --git a/wrapper/CSharp/wolfTPM.cs b/wrapper/CSharp/wolfTPM.cs
index 9eda1ac..50ba183 100644
--- a/wrapper/CSharp/wolfTPM.cs
+++ b/wrapper/CSharp/wolfTPM.cs
@@ -87,6 +87,19 @@ namespace wolfTPM
         ECB = 0x0044,
     }
 
+    public enum TPM2_ECC : uint
+    {
+        NONE        = 0x0000,
+        NIST_P192   = 0x0001,
+        NIST_P224   = 0x0002,
+        NIST_P256   = 0x0003,
+        NIST_P384   = 0x0004,
+        NIST_P521   = 0x0005,
+        BN_P256     = 0x0010,
+        BN_P638     = 0x0011,
+        SM2_P256    = 0x0020,
+    }
+
     public enum SE : byte
     {
         HMAC = 0x00,
@@ -104,6 +117,28 @@ namespace wolfTPM
         audit           = 0x80,
     }
 
+    public enum TPM_RH : ulong
+    {
+        FIRST        = 0x40000000,
+        SRK          = FIRST,
+        OWNER        = 0x40000001,
+        REVOKE       = 0x40000002,
+        TRANSPORT    = 0x40000003,
+        OPERATOR     = 0x40000004,
+        ADMIN        = 0x40000005,
+        EK           = 0x40000006,
+        NULL         = 0x40000007,
+        UNASSIGNED   = 0x40000008,
+        PW           = 0x40000009,
+        LOCKOUT      = 0x4000000A,
+        ENDORSEMENT  = 0x4000000B,
+        PLATFORM     = 0x4000000C,
+        PLATFORM_NV  = 0x4000000D,
+        AUTH_00      = 0x40000010,
+        AUTH_FF      = 0x4000010F,
+        LAST         = AUTH_FF,
+    }
+
     public class KeyBlob
     {
         const string DLLNAME = "wolftpm";
@@ -248,11 +283,22 @@ namespace wolfTPM
                                                objectAttributes);
         }
 
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_ECC")]
+        private static extern int wolfTPM2_GetKeyTemplate_ECC(IntPtr publicTemplate,
+                                                              ulong objectAttributes,
+                                                              uint curve,
+                                                              uint sigScheme);
+        public int GetKeyTemplate_ECC(ulong objectAttributes, TPM2_ECC curve,
+            TPM2_Alg sigScheme)
+        {
+            return wolfTPM2_GetKeyTemplate_ECC(template, objectAttributes,
+                (uint)curve, (uint)sigScheme);
+        }
+
         [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_Symmetric")]
         private static extern int wolfTPM2_GetKeyTemplate_Symmetric(
             IntPtr publicTemplate, int keyBits, uint algMode, int isSign,
             int isDecrypt);
-
         public int GetKeyTemplate_Symmetric(int keyBits,
                                             TPM2_Alg algMode,
                                             bool isSign,
@@ -265,6 +311,47 @@ namespace wolfTPM
                                                      isDecrypt ? 1 : 0);
         }
 
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_RSA_EK")]
+        private static extern int wolfTPM2_GetKeyTemplate_RSA_EK(IntPtr publicTemplate);
+        public int GetKeyTemplate_RSA_EK()
+        {
+            return wolfTPM2_GetKeyTemplate_RSA_EK(template);
+        }
+
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_ECC_EK")]
+        private static extern int wolfTPM2_GetKeyTemplate_ECC_EK(IntPtr publicTemplate);
+        public int GetKeyTemplate_ECC_EK()
+        {
+            return wolfTPM2_GetKeyTemplate_ECC_EK(template);
+        }
+
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_RSA_SRK")]
+        private static extern int wolfTPM2_GetKeyTemplate_RSA_SRK(IntPtr publicTemplate);
+        public int GetKeyTemplate_RSA_SRK()
+        {
+            return wolfTPM2_GetKeyTemplate_RSA_SRK(template);
+        }
+
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_ECC_SRK")]
+        private static extern int wolfTPM2_GetKeyTemplate_ECC_SRK(IntPtr publicTemplate);
+        public int GetKeyTemplate_ECC_SRK()
+        {
+            return wolfTPM2_GetKeyTemplate_ECC_SRK(template);
+        }
+
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_RSA_AIK")]
+        private static extern int wolfTPM2_GetKeyTemplate_RSA_AIK(IntPtr publicTemplate);
+        public int GetKeyTemplate_RSA_AIK()
+        {
+            return wolfTPM2_GetKeyTemplate_RSA_AIK(template);
+        }
+
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_GetKeyTemplate_ECC_AIK")]
+        private static extern int wolfTPM2_GetKeyTemplate_ECC_AIK(IntPtr publicTemplate);
+        public int GetKeyTemplate_ECC_AIK()
+        {
+            return wolfTPM2_GetKeyTemplate_ECC_AIK(template);
+        }
     }
 
     public class Session
@@ -603,6 +690,29 @@ namespace wolfTPM
                 rsaPriv.Length);
         }
 
+        [DllImport(DLLNAME, EntryPoint = "wolfTPM2_CreatePrimaryKey")]
+        private static extern int wolfTPM2_CreatePrimaryKey(
+            IntPtr dev,
+            IntPtr key,
+            ulong primaryHandle,
+            IntPtr publicTemplate,
+            string auth,
+            int authSz);
+        public int CreatePrimaryKey(
+            Key key,
+            TPM_RH primaryHandle,
+            Template publicTemplate,
+            string auth)
+        {
+            return wolfTPM2_CreatePrimaryKey(
+                device,
+                key.key,
+                (ulong)primaryHandle,
+                publicTemplate.template,
+                auth,
+                !string.IsNullOrEmpty(auth) ? auth.Length : 0);
+        }
+
         [DllImport(DLLNAME, EntryPoint = "wolfTPM2_UnloadHandle")]
         private static extern int wolfTPM2_UnloadHandle(IntPtr dev, IntPtr handle);
         public int UnloadHandle(Key key)