Fix C_CopyObject alloc for ECC keys

Author: LinuxJedi

src/crypto.c

@@ -1201,6 +1201,9 @@ CK_RV C_CopyObject(CK_SESSION_HANDLE hSession, CK_OBJECT_HANDLE hObject,
 
     /* Set the object class from the original object */
     ret = WP11_Object_SetClass(newObj, WP11_Object_GetClass(obj));
+    if (ret != 0)
+        return CKR_FUNCTION_FAILED;
+    ret = WP11_Object_SetType(newObj, WP11_Object_GetType(obj));
     if (ret != 0)
         return CKR_FUNCTION_FAILED;
 

src/internal.c

@@ -7939,6 +7939,24 @@ int WP11_Object_SetClass(WP11_Object* object, CK_OBJECT_CLASS objClass)
     return 0;
 }
 
+/**
+ * Set the object's type.
+ *
+ * @param  object    [in]  Object object.
+ * @param  objType   [in]  Object's type.
+ * @return  0 on success.
+ */
+int WP11_Object_SetType(WP11_Object* object, CK_KEY_TYPE objType)
+{
+    if (object->onToken)
+        WP11_Lock_LockRW(object->lock);
+    object->type = objType;
+    if (object->onToken)
+        WP11_Lock_UnlockRW(object->lock);
+
+    return 0;
+}
+
 /**
  * Find an object based on the handle.
  *

wolfpkcs11/internal.h

@@ -401,6 +401,7 @@ WP11_LOCAL int WP11_Object_DataObject(WP11_Object* object, unsigned char** data,
                            CK_ULONG* len);
 
 WP11_LOCAL int WP11_Object_SetClass(WP11_Object* object, CK_OBJECT_CLASS objClass);
+WP11_LOCAL int WP11_Object_SetType(WP11_Object* object, CK_KEY_TYPE objType);
 WP11_LOCAL CK_OBJECT_CLASS WP11_Object_GetClass(WP11_Object* object);
 
 #ifdef WOLFPKCS11_NSS