From ce5fca5d71d1106a3cb48de30a6618efba503efe Mon Sep 17 00:00:00 2001
From: Tharsan Ponnampalam <tharsan@kn0x.io>
Date: Fri, 12 Jul 2019 12:42:25 -0400
Subject: [PATCH] lib: Implemented change pin for the management key

---
 ykcs11/yubico_token.c | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/ykcs11/yubico_token.c b/ykcs11/yubico_token.c
index 7c799d3..08040e6 100644
--- a/ykcs11/yubico_token.c
+++ b/ykcs11/yubico_token.c
@@ -368,11 +368,26 @@ CK_RV YUBICO_get_token_raw_certificate(ykpiv_state *state, piv_obj_id_t obj, CK_
 CK_RV YUBICO_token_change_pin(ykpiv_state *state, CK_USER_TYPE user_type, CK_UTF8CHAR_PTR pOldPin, CK_ULONG ulOldLen, CK_UTF8CHAR_PTR pNewPin, CK_ULONG ulNewLen) {
   int tries;
   ykpiv_rc res;
-  if (user_type != CKU_USER) {
-    DBG("TODO implement other users pin change");
-    return CKR_FUNCTION_FAILED;
+
+  switch(user_type){
+    case CKU_SO:{
+      unsigned char new_key[24];
+      size_t new_key_len = sizeof(new_key);
+      if(ykpiv_hex_decode((const char*)pNewPin, ulNewLen, new_key, &new_key_len) != YKPIV_OK) {
+        DBG("Failed to decode new pin")
+        return CKR_ARGUMENTS_BAD;
+      }
+      res = ykpiv_set_mgmkey(state, new_key);
+      break;
+    }
+    case CKU_USER:
+      res = ykpiv_change_pin(state, (const char*)pOldPin, ulOldLen, (const char*)pNewPin, ulNewLen, &tries);
+      break;
+    default:
+      DBG("TODO implement other context specific pin change");
+      return CKR_FUNCTION_FAILED;
   }
-  res = ykpiv_change_pin(state, (const char*)pOldPin, ulOldLen, (const char*)pNewPin, ulNewLen, &tries);
+
   switch (res) {
     case YKPIV_OK:
       return CKR_OK;