add ykpiv_set_mgmkey()

lib/error.c

@@ -48,6 +48,7 @@ static const err_t errors[] = {
   ERR (YKPIV_APPLET_ERROR, "No PIV applet found"),
   ERR (YKPIV_AUTHENTICATION_ERROR, "Error during authentication"),
   ERR (YKPIV_RANDOMNESS_ERROR, "Error getting randomness"),
+  ERR (YKPIV_GENERIC_ERROR, "Something went wrong."),
 };
 
 /**

lib/ykpiv.c

@@ -345,3 +345,41 @@ ykpiv_rc ykpiv_authenticate(ykpiv_state *state, unsigned const char *key) {
     }
   }
 }
+
+ykpiv_rc ykpiv_set_mgmkey(ykpiv_state *state, const unsigned char *new_key) {
+  APDU apdu;
+  unsigned char data[0xff];
+  unsigned long recv_len = sizeof(data);
+  int sw;
+  size_t i;
+  ykpiv_rc res;
+
+  for(i = 0; i < 3; i++) {
+    const_DES_cblock key_tmp;
+    memcpy(key_tmp, new_key + i * 8, 8);
+    if(DES_is_weak_key(&key_tmp) == 1) {
+      if(state->verbose) {
+	fprintf(stderr, "Won't set new key '");
+	dump_hex(new_key + i, 8);
+	fprintf(stderr, "' since it's considered weak.\n");
+      }
+      return YKPIV_GENERIC_ERROR;
+    }
+  }
+
+  memset(apdu.raw, 0, sizeof(apdu));
+  apdu.st.ins = YKPIV_INS_SET_MGMKEY;
+  apdu.st.p1 = 0xff;
+  apdu.st.p2 = 0xff;
+  apdu.st.lc = DES_KEY_SZ * 3 + 3;
+  apdu.st.data[0] = YKPIV_ALGO_3DES;
+  apdu.st.data[1] = YKPIV_KEY_CARDMGM;
+  apdu.st.data[2] = DES_KEY_SZ * 3;
+  memcpy(apdu.st.data + 3, new_key, DES_KEY_SZ * 3);
+  if((res = ykpiv_send_data(state, apdu.raw, data, &recv_len, &sw)) != YKPIV_OK) {
+    return res;
+  } else if(sw == 0x9000) {
+    return YKPIV_OK;
+  }
+  return YKPIV_GENERIC_ERROR;
+}

lib/ykpiv.h

@@ -47,6 +47,7 @@ extern "C"
 		YKPIV_APPLET_ERROR = -4,
 		YKPIV_AUTHENTICATION_ERROR = -5,
 		YKPIV_RANDOMNESS_ERROR = -6,
+		YKPIV_GENERIC_ERROR = -7,
 	} ykpiv_rc;
 
 	const char *ykpiv_strerror(ykpiv_rc err);
@@ -61,6 +62,7 @@ extern "C"
 	ykpiv_rc ykpiv_send_data(ykpiv_state *state, unsigned char *apdu,
 			unsigned char *data, unsigned long *recv_len, int *sw);
 	ykpiv_rc ykpiv_authenticate(ykpiv_state *state, const unsigned char *key);
+	ykpiv_rc ykpiv_set_mgmkey(ykpiv_state *state, const unsigned char *new_key);
 
 #define YKPIV_ALGO_3DES 0x03;
 #define YKPIV_ALGO_RSA1024 0x06;
@@ -75,6 +77,9 @@ extern "C"
 
 #define YKPIV_INS_AUTHENTICATE 0x87;
 
+	/* Yubico vendor specific instructions */
+#define YKPIV_INS_SET_MGMKEY 0xff;
+
 #ifdef __cplusplus
 }
 #endif

lib/ykpiv.map

@@ -36,6 +36,7 @@ global:
 	ykpiv_send_data;
 	ykpiv_transfer_data;
 	ykpiv_authenticate;
+	ykpiv_set_mgmkey;
 
 local:
 	*;