/* Test for Phase 6 get_public_key response format (dispatcher.c). * * Verifies: * - secp256k1: get_public_key returns plain hex string (backward compat) * - secp256k1 with {"format":"structured"}: returns structured JSON * - ed25519: returns structured JSON with algorithm "ed25519" * - ML-DSA-65: returns structured JSON with algorithm "ml-dsa-65" and large pubkey * - ML-KEM-768: returns structured JSON with algorithm "ml-kem-768" * - Structured result contains algorithm, public_key, and key_id fields * - key_id is the first 16 hex chars of the public key */ /* NSIGNER_HEADERLESS_DECLS_BEGIN */ #include #include #include #include /* from secure_mem.h */ typedef struct { void *data; size_t size; int locked; } secure_buf_t; int secure_buf_alloc(secure_buf_t *buf, size_t size); void secure_buf_free(secure_buf_t *buf); void secure_memzero(void *ptr, size_t len); /* from mnemonic.h */ #define MNEMONIC_MAX_LEN 256 typedef struct { secure_buf_t buf; int loaded; int word_count; } mnemonic_state_t; void mnemonic_init(mnemonic_state_t *state); int mnemonic_load(mnemonic_state_t *state, const char *phrase); void mnemonic_unload(mnemonic_state_t *state); int mnemonic_is_loaded(const mnemonic_state_t *state); const char *mnemonic_get_phrase(const mnemonic_state_t *state); int mnemonic_generate(int word_count, char *out, size_t out_len); /* from role_table.h */ #define ROLE_NAME_MAX 64 #define ROLE_PATH_MAX 128 #define ROLE_PURPOSE_MAX 32 #define ROLE_CURVE_MAX 16 #define ROLE_PUBKEY_HEX_MAX 66 #define ROLE_TABLE_MAX_ENTRIES 256 typedef enum { PURPOSE_NOSTR = 0, PURPOSE_BITCOIN, PURPOSE_SSH, PURPOSE_AGE, PURPOSE_FIPS, PURPOSE_PQ_SIG, PURPOSE_PQ_KEM, PURPOSE_UNKNOWN } role_purpose_t; typedef enum { CURVE_SECP256K1 = 0, CURVE_ED25519, CURVE_X25519, CURVE_ML_DSA_65, CURVE_SLH_DSA_128S, CURVE_ML_KEM_768, CURVE_UNKNOWN } role_curve_t; typedef enum { SELECTOR_NOSTR_INDEX, SELECTOR_ROLE_PATH } role_selector_type_t; typedef struct { char name[ROLE_NAME_MAX]; char purpose_str[ROLE_PURPOSE_MAX]; char curve_str[ROLE_CURVE_MAX]; role_purpose_t purpose; role_curve_t curve; role_selector_type_t selector_type; int nostr_index; char role_path[ROLE_PATH_MAX]; char pubkey_hex[ROLE_PUBKEY_HEX_MAX]; int derived; } role_entry_t; typedef struct { role_entry_t entries[ROLE_TABLE_MAX_ENTRIES]; int count; } role_table_t; void role_table_init(role_table_t *table); int role_table_add(role_table_t *table, const role_entry_t *entry); role_entry_t *role_table_find_by_name(role_table_t *table, const char *name); role_entry_t *role_table_find_by_nostr_index(role_table_t *table, int index); role_entry_t *role_table_find_by_path(role_table_t *table, const char *path); role_entry_t *role_table_get_default(role_table_t *table); role_purpose_t role_purpose_from_str(const char *s); role_curve_t role_curve_from_str(const char *s); const char *role_purpose_to_str(role_purpose_t p); const char *role_curve_to_str(role_curve_t c); int role_table_register_nostr_index(role_table_t *table, int nostr_index); /* from selector.h */ #define SELECTOR_OK 0 #define SELECTOR_ERR_AMBIGUOUS -1 #define SELECTOR_ERR_NOT_FOUND -2 #define SELECTOR_ERR_NO_DEFAULT -3 typedef struct { int has_role; char role_name[ROLE_NAME_MAX]; int has_nostr_index; int nostr_index; int has_role_path; char role_path[ROLE_PATH_MAX]; } selector_request_t; void selector_request_init(selector_request_t *req); int selector_resolve(const selector_request_t *req, role_table_t *table, role_entry_t **out); const char *selector_strerror(int err); /* from enforcement.h */ #define ENFORCE_OK 0 #define ENFORCE_ERR_PURPOSE -1 #define ENFORCE_ERR_CURVE -2 #define ENFORCE_ERR_UNKNOWN_VERB -3 #define VERB_GET_PUBLIC_KEY "get_public_key" #define VERB_NOSTR_GET_PUBLIC_KEY "nostr_get_public_key" #define VERB_NOSTR_SIGN_EVENT "nostr_sign_event" #define VERB_NOSTR_MINE_EVENT "nostr_mine_event" #define VERB_NOSTR_NIP44_ENCRYPT "nostr_nip44_encrypt" #define VERB_NOSTR_NIP44_DECRYPT "nostr_nip44_decrypt" #define VERB_NOSTR_NIP04_ENCRYPT "nostr_nip04_encrypt" #define VERB_NOSTR_NIP04_DECRYPT "nostr_nip04_decrypt" int enforce_verb_role(const char *verb, const role_entry_t *role); const char *enforce_strerror(int err); /* from policy.h */ #define POLICY_MAX_ENTRIES 32 #define POLICY_MAX_VERBS 16 #define POLICY_MAX_ROLES 16 #define POLICY_MAX_PURPOSES 8 #define POLICY_VERB_MAX_LEN 32 #define POLICY_CALLER_MAX_LEN 160 #define POLICY_MAX_ALGS 16 #define POLICY_MAX_ALGS 16 typedef enum { PROMPT_NEVER = 0, PROMPT_FIRST_PER_BOOT, PROMPT_EVERY_REQUEST, PROMPT_DENY } prompt_mode_t; typedef enum { POLICY_SOURCE_DEFAULT = 0, POLICY_SOURCE_PREAPPROVE, POLICY_SOURCE_SESSION_GRANT } policy_source_t; typedef struct { char caller[POLICY_CALLER_MAX_LEN]; char verbs[POLICY_MAX_VERBS][POLICY_VERB_MAX_LEN]; int verb_count; char roles[POLICY_MAX_ROLES][ROLE_NAME_MAX]; int role_count; char purposes[POLICY_MAX_PURPOSES][ROLE_PURPOSE_MAX]; int purpose_count; /* Algorithm-based (new) */ char algorithms[16][32]; /* algorithm names; POLICY_MAX_ALGS */ int alg_count; int index_min; /* -1 = any */ int index_max; /* -1 = any */ /* Common */ prompt_mode_t prompt; policy_source_t source; } policy_entry_t; typedef struct { policy_entry_t entries[POLICY_MAX_ENTRIES]; int count; } policy_table_t; #define POLICY_ALLOW 0 #define POLICY_DENY -1 #define POLICY_PROMPT -2 #define POLICY_NO_MATCH -3 void policy_table_init(policy_table_t *table); void policy_init_default(policy_table_t *table, uid_t owner_uid); int policy_table_add(policy_table_t *table, const policy_entry_t *entry); int policy_check(const policy_table_t *table, const char *caller_id, const char *verb, const char *role_name, const char *purpose, policy_source_t *out_source); /* Check whether caller_id is allowed to invoke `verb` with the given * algorithm and index (algorithm-based policy). Returns POLICY_ALLOW, * POLICY_DENY, POLICY_PROMPT, or POLICY_NO_MATCH. */ int policy_check_algorithm(const policy_table_t *table, const char *caller_id, const char *verb, const char *algorithm, int index, policy_source_t *out_source); /* Check whether caller_id is allowed to invoke `verb` with the given * algorithm and index (algorithm-based policy). Returns POLICY_ALLOW, * POLICY_DENY, POLICY_PROMPT, or POLICY_NO_MATCH. */ int policy_check_algorithm(const policy_table_t *table, const char *caller_id, const char *verb, const char *algorithm, int index, policy_source_t *out_source); prompt_mode_t prompt_mode_from_str(const char *s); const char *prompt_mode_to_str(prompt_mode_t m); /* from pq_crypto.h */ typedef enum { CRYPTO_ALG_SECP256K1 = 0, CRYPTO_ALG_ED25519, CRYPTO_ALG_X25519, CRYPTO_ALG_ML_DSA_65, CRYPTO_ALG_SLH_DSA_128S, CRYPTO_ALG_ML_KEM_768, CRYPTO_ALG_UNKNOWN } crypto_alg_t; typedef struct { size_t priv_key_len; size_t pub_key_len; size_t sig_len; size_t ciphertext_len; size_t shared_secret_len; } crypto_alg_sizes_t; const crypto_alg_sizes_t *crypto_alg_get_sizes(crypto_alg_t alg); crypto_alg_t crypto_alg_from_role(role_curve_t curve, role_purpose_t purpose); const char *crypto_alg_to_str(crypto_alg_t alg); crypto_alg_t crypto_alg_from_str(const char *s); int crypto_ed25519_keygen_from_seed(const unsigned char *seed, size_t seed_len, unsigned char *priv_out, unsigned char *pub_out); int crypto_ed25519_sign(const unsigned char *priv, size_t priv_len, const unsigned char *msg, size_t msg_len, unsigned char *sig_out, size_t *sig_out_len); int crypto_ed25519_verify(const unsigned char *pub, size_t pub_len, const unsigned char *msg, size_t msg_len, const unsigned char *sig, size_t sig_len); int crypto_x25519_keygen_from_seed(const unsigned char *seed, size_t seed_len, unsigned char *priv_out, unsigned char *pub_out); int crypto_x25519_ecdh(const unsigned char *our_priv, size_t priv_len, const unsigned char *peer_pub, size_t pub_len, unsigned char *shared_out, size_t *shared_out_len); int crypto_derive_seed_from_mnemonic(const char *mnemonic, const char *path, unsigned char *seed_out, size_t seed_out_len); int crypto_ml_dsa_65_keygen_from_seed(const unsigned char *seed, size_t seed_len, unsigned char *priv_out, unsigned char *pub_out); int crypto_ml_dsa_65_sign(const unsigned char *priv, size_t priv_len, const unsigned char *msg, size_t msg_len, unsigned char *sig_out, size_t *sig_out_len); int crypto_ml_dsa_65_verify(const unsigned char *pub, size_t pub_len, const unsigned char *msg, size_t msg_len, const unsigned char *sig, size_t sig_len); int crypto_slh_dsa_128s_keygen_from_seed(const unsigned char *seed, size_t seed_len, unsigned char *priv_out, unsigned char *pub_out); int crypto_slh_dsa_128s_sign(const unsigned char *priv, size_t priv_len, const unsigned char *msg, size_t msg_len, unsigned char *sig_out, size_t *sig_out_len); int crypto_slh_dsa_128s_verify(const unsigned char *pub, size_t pub_len, const unsigned char *msg, size_t msg_len, const unsigned char *sig, size_t sig_len); int crypto_ml_kem_768_keygen_from_seed(const unsigned char *seed, size_t seed_len, unsigned char *priv_out, unsigned char *pub_out); int crypto_ml_kem_768_encaps(const unsigned char *pub, size_t pub_len, unsigned char *ct_out, unsigned char *ss_out); int crypto_ml_kem_768_decaps(const unsigned char *priv, size_t priv_len, const unsigned char *ct, size_t ct_len, unsigned char *ss_out); void pq_drbg_init(const unsigned char *seed, size_t seed_len); int pq_drbg_randombytes(unsigned char *buf, size_t len); void pq_drbg_zeroize(void); /* from crypto.h */ typedef struct { secure_buf_t private_key; secure_buf_t public_key; char pubkey_hex[8192]; char npub[128]; crypto_alg_t alg; int valid; } derived_key_t; typedef struct { derived_key_t keys[ROLE_TABLE_MAX_ENTRIES]; int count; } key_store_t; int crypto_derive_all(key_store_t *store, role_table_t *table, const mnemonic_state_t *mnemonic); int crypto_derive_one(key_store_t *store, role_table_t *table, const mnemonic_state_t *mnemonic, int role_index); const unsigned char *crypto_get_private_key(const key_store_t *store, int role_index); const char *crypto_get_pubkey_hex(const key_store_t *store, int role_index); char *crypto_sign_event(const key_store_t *store, int role_index, const char *event_json); void crypto_wipe(key_store_t *store); /* from alg_api.h */ /* Check whether a verb is valid for an algorithm (algorithm-based enforcement). * Returns ENFORCE_OK, ENFORCE_ERR_ALGORITHM, or ENFORCE_ERR_UNKNOWN_VERB. * Does NOT check purpose — purpose is irrelevant for the new verbs. */ int enforce_verb_algorithm(const char *verb, crypto_alg_t alg); /* secp256k1 Schnorr (BIP-340) sign arbitrary bytes. * priv is 32-byte scalar, pub is 32-byte x-only pubkey, sig_out is 64 bytes. * Hashes the message with SHA-256 before signing (like Nostr event signing). * Returns 0 on success, -1 on error. */ int crypto_secp256k1_schnorr_sign(const unsigned char *priv, size_t priv_len, const unsigned char *msg, size_t msg_len, unsigned char *sig_out, size_t *sig_out_len); /* secp256k1 Schnorr (BIP-340) verify. * pub is 32-byte x-only pubkey, sig is 64 bytes. * Returns 0 on valid, 1 on invalid, -1 on error. */ int crypto_secp256k1_schnorr_verify(const unsigned char *pub, size_t pub_len, const unsigned char *msg, size_t msg_len, const unsigned char *sig, size_t sig_len); /* secp256k1 ECDSA sign arbitrary bytes. * priv is 32-byte scalar, sig_out must be at least 64 bytes (compact DER r||s). * Hashes the message with SHA-256 before signing. * Returns 0 on success, -1 on error. */ int crypto_secp256k1_ecdsa_sign(const unsigned char *priv, size_t priv_len, const unsigned char *msg, size_t msg_len, unsigned char *sig_out, size_t *sig_out_len); /* secp256k1 ECDSA verify. * pub is 32-byte x-only pubkey (converted internally to compressed form). * sig is 64-byte compact (r||s). Returns 0 on valid, 1 on invalid, -1 on error. */ int crypto_secp256k1_ecdsa_verify(const unsigned char *pub, size_t pub_len, const unsigned char *msg, size_t msg_len, const unsigned char *sig, size_t sig_len); /* ---- Algorithm key cache ---- * On-demand key derivation by algorithm+index, separate from the role-based * key_store. Holds up to ALG_KEY_CACHE_MAX derived keys in secure memory. * When full, the oldest entry is evicted (FIFO). */ #define ALG_KEY_CACHE_MAX 32 typedef struct { crypto_alg_t alg; int index; secure_buf_t private_key; secure_buf_t public_key; char pubkey_hex[8192]; char key_id[17]; int valid; } alg_key_entry_t; typedef struct { alg_key_entry_t entries[ALG_KEY_CACHE_MAX]; int count; } algorithm_key_cache_t; /* Initialize an empty cache. */ void alg_key_cache_init(algorithm_key_cache_t *cache); /* Zeroize and free all entries. Idempotent. */ void alg_key_cache_wipe(algorithm_key_cache_t *cache); /* Look up a cached entry by (alg, index). Returns NULL if not present. */ const alg_key_entry_t *alg_key_cache_get(algorithm_key_cache_t *cache, crypto_alg_t alg, int index); /* Derive a key on-demand by (alg, index) and store it in the cache. * Uses the standard derivation path for the algorithm. * Returns 0 on success, -1 on error. */ int alg_key_cache_derive(algorithm_key_cache_t *cache, const mnemonic_state_t *mnemonic, crypto_alg_t alg, int index); /* from dispatcher.h */ typedef struct { role_table_t *role_table; mnemonic_state_t *mnemonic; key_store_t *key_store; } dispatcher_ctx_t; void dispatcher_init(dispatcher_ctx_t *ctx, role_table_t *table, mnemonic_state_t *mnemonic, key_store_t *key_store, algorithm_key_cache_t *alg_key_cache); char *dispatcher_handle_request(dispatcher_ctx_t *ctx, const char *json_request); /* NSIGNER_HEADERLESS_DECLS_END */ static key_store_t g_key_store; static algorithm_key_cache_t g_alg_key_cache; #include #include #include #include #include static int g_passes = 0; static int g_total = 0; static void check_condition(const char *name, int condition) { g_total++; if (condition) { printf("PASS: %s\n", name); g_passes++; } else { printf("FAIL: %s\n", name); } } static int response_has(const char *response, const char *needle) { return (response != NULL && needle != NULL && strstr(response, needle) != NULL); } /* Parse the dispatcher response's "result" field. If result is a JSON string * (plain hex), returns a newly-allocated copy of the string in *out_str and * returns 1. If result is a JSON object serialized as a string, parses it into * *out_obj and returns 2. Returns 0 on failure. Caller frees accordingly. */ static int parse_result(const char *response, char **out_str, cJSON **out_obj) { cJSON *root = NULL; cJSON *result_item = NULL; int ret = 0; if (out_str != NULL) *out_str = NULL; if (out_obj != NULL) *out_obj = NULL; if (response == NULL) return 0; root = cJSON_Parse(response); if (root == NULL) return 0; result_item = cJSON_GetObjectItemCaseSensitive(root, "result"); if (cJSON_IsString(result_item) && result_item->valuestring != NULL) { /* Could be a plain hex string OR a serialized JSON object string. */ cJSON *maybe_obj = cJSON_Parse(result_item->valuestring); if (maybe_obj != NULL && cJSON_IsObject(maybe_obj)) { if (out_obj != NULL) { *out_obj = maybe_obj; maybe_obj = NULL; ret = 2; } else { cJSON_Delete(maybe_obj); ret = 0; } } else { cJSON_Delete(maybe_obj); if (out_str != NULL) { *out_str = strdup(result_item->valuestring); ret = 1; } } } cJSON_Delete(root); return ret; } static role_entry_t make_entry(const char *name, const char *purpose_str, const char *curve_str, int idx) { role_entry_t e; memset(&e, 0, sizeof(e)); strncpy(e.name, name, sizeof(e.name) - 1); strncpy(e.purpose_str, purpose_str, sizeof(e.purpose_str) - 1); strncpy(e.curve_str, curve_str, sizeof(e.curve_str) - 1); e.purpose = role_purpose_from_str(e.purpose_str); e.curve = role_curve_from_str(e.curve_str); e.selector_type = SELECTOR_NOSTR_INDEX; e.nostr_index = idx; e.derived = 0; return e; } int main(void) { const char *mnemonic = "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about"; if (nostr_init() != 0) { fprintf(stderr, "nostr_init failed\n"); return 1; } /* ---- secp256k1: plain hex (backward compat) ---- */ { role_table_t table; role_entry_t nostr_role; static mnemonic_state_t mnemonic_state; dispatcher_ctx_t dispatcher; char *resp = NULL; char *result_str = NULL; cJSON *result_obj = NULL; int pr; role_table_init(&table); nostr_role = make_entry("main", "nostr", "secp256k1", 0); role_table_add(&table, &nostr_role); mnemonic_init(&mnemonic_state); mnemonic_load(&mnemonic_state, mnemonic); memset(&g_key_store, 0, sizeof(g_key_store)); alg_key_cache_init(&g_alg_key_cache); crypto_derive_all(&g_key_store, &table, &mnemonic_state); dispatcher_init(&dispatcher, &table, &mnemonic_state, &g_key_store, &g_alg_key_cache); resp = dispatcher_handle_request(&dispatcher, "{\"id\":\"1\",\"method\":\"get_public_key\",\"params\":[{\"algorithm\":\"secp256k1\",\"index\":0}]}"); check_condition("secp256k1 get_public_key returns a response", resp != NULL); /* Algorithm-based get_public_key always returns a structured object. */ pr = parse_result(resp, &result_str, &result_obj); check_condition("secp256k1 get_public_key result is a JSON object", pr == 2 && result_obj != NULL); if (result_obj != NULL) { cJSON *pk_item = cJSON_GetObjectItemCaseSensitive(result_obj, "public_key"); check_condition("secp256k1 structured has public_key (64 hex chars)", pk_item != NULL && cJSON_IsString(pk_item) && strlen(pk_item->valuestring) == 64); } else { check_condition("secp256k1 structured has public_key (64 hex chars)", 0); } check_condition("secp256k1 structured has algorithm field", response_has(resp, "algorithm")); free(result_str); cJSON_Delete(result_obj); free(resp); crypto_wipe(&g_key_store); mnemonic_unload(&mnemonic_state); } /* ---- secp256k1 with format:structured ---- */ { role_table_t table; role_entry_t nostr_role; static mnemonic_state_t mnemonic_state; dispatcher_ctx_t dispatcher; char *resp = NULL; char *result_str = NULL; cJSON *result_obj = NULL; cJSON *alg_item = NULL; cJSON *pk_item = NULL; cJSON *kid_item = NULL; int pr; role_table_init(&table); nostr_role = make_entry("main", "nostr", "secp256k1", 0); role_table_add(&table, &nostr_role); mnemonic_init(&mnemonic_state); mnemonic_load(&mnemonic_state, mnemonic); memset(&g_key_store, 0, sizeof(g_key_store)); alg_key_cache_init(&g_alg_key_cache); crypto_derive_all(&g_key_store, &table, &mnemonic_state); dispatcher_init(&dispatcher, &table, &mnemonic_state, &g_key_store, &g_alg_key_cache); resp = dispatcher_handle_request(&dispatcher, "{\"id\":\"2\",\"method\":\"get_public_key\",\"params\":[{\"algorithm\":\"secp256k1\",\"index\":0,\"format\":\"structured\"}]}"); check_condition("secp256k1 structured get_public_key returns a response", resp != NULL); pr = parse_result(resp, &result_str, &result_obj); check_condition("secp256k1 structured result is a JSON object", pr == 2 && result_obj != NULL); if (result_obj != NULL) { alg_item = cJSON_GetObjectItemCaseSensitive(result_obj, "algorithm"); pk_item = cJSON_GetObjectItemCaseSensitive(result_obj, "public_key"); kid_item = cJSON_GetObjectItemCaseSensitive(result_obj, "key_id"); check_condition("secp256k1 structured has algorithm=secp256k1", cJSON_IsString(alg_item) && strcmp(alg_item->valuestring, "secp256k1") == 0); check_condition("secp256k1 structured has public_key (64 hex chars)", cJSON_IsString(pk_item) && strlen(pk_item->valuestring) == 64); check_condition("secp256k1 structured has key_id (16 hex chars)", cJSON_IsString(kid_item) && strlen(kid_item->valuestring) == 16); check_condition("secp256k1 key_id == first 16 chars of public_key", cJSON_IsString(pk_item) && cJSON_IsString(kid_item) && strncmp(pk_item->valuestring, kid_item->valuestring, 16) == 0); } else { check_condition("secp256k1 structured has algorithm=secp256k1", 0); check_condition("secp256k1 structured has public_key (64 hex chars)", 0); check_condition("secp256k1 structured has key_id (16 hex chars)", 0); check_condition("secp256k1 key_id == first 16 chars of public_key", 0); } free(result_str); cJSON_Delete(result_obj); free(resp); crypto_wipe(&g_key_store); mnemonic_unload(&mnemonic_state); } /* ---- ed25519: structured JSON ---- */ { role_table_t table; role_entry_t ssh_role; static mnemonic_state_t mnemonic_state; dispatcher_ctx_t dispatcher; char *resp = NULL; char *result_str = NULL; cJSON *result_obj = NULL; cJSON *alg_item = NULL; cJSON *pk_item = NULL; cJSON *kid_item = NULL; int pr; role_table_init(&table); ssh_role = make_entry("ssh_main", "ssh", "ed25519", 0); role_table_add(&table, &ssh_role); mnemonic_init(&mnemonic_state); mnemonic_load(&mnemonic_state, mnemonic); memset(&g_key_store, 0, sizeof(g_key_store)); alg_key_cache_init(&g_alg_key_cache); crypto_derive_all(&g_key_store, &table, &mnemonic_state); dispatcher_init(&dispatcher, &table, &mnemonic_state, &g_key_store, &g_alg_key_cache); resp = dispatcher_handle_request(&dispatcher, "{\"id\":\"3\",\"method\":\"get_public_key\",\"params\":[{\"algorithm\":\"ed25519\",\"index\":0}]}"); check_condition("ed25519 get_public_key returns a response", resp != NULL); pr = parse_result(resp, &result_str, &result_obj); check_condition("ed25519 get_public_key result is a JSON object", pr == 2 && result_obj != NULL); if (result_obj != NULL) { alg_item = cJSON_GetObjectItemCaseSensitive(result_obj, "algorithm"); pk_item = cJSON_GetObjectItemCaseSensitive(result_obj, "public_key"); kid_item = cJSON_GetObjectItemCaseSensitive(result_obj, "key_id"); check_condition("ed25519 structured has algorithm=ed25519", cJSON_IsString(alg_item) && strcmp(alg_item->valuestring, "ed25519") == 0); check_condition("ed25519 structured has public_key (64 hex chars)", cJSON_IsString(pk_item) && strlen(pk_item->valuestring) == 64); check_condition("ed25519 structured has key_id (16 hex chars)", cJSON_IsString(kid_item) && strlen(kid_item->valuestring) == 16); check_condition("ed25519 key_id == first 16 chars of public_key", cJSON_IsString(pk_item) && cJSON_IsString(kid_item) && strncmp(pk_item->valuestring, kid_item->valuestring, 16) == 0); } else { check_condition("ed25519 structured has algorithm=ed25519", 0); check_condition("ed25519 structured has public_key (64 hex chars)", 0); check_condition("ed25519 structured has key_id (16 hex chars)", 0); check_condition("ed25519 key_id == first 16 chars of public_key", 0); } free(result_str); cJSON_Delete(result_obj); free(resp); crypto_wipe(&g_key_store); mnemonic_unload(&mnemonic_state); } /* ---- ML-DSA-65: structured JSON with large pubkey ---- */ { role_table_t table; role_entry_t pq_role; static mnemonic_state_t mnemonic_state; dispatcher_ctx_t dispatcher; char *resp = NULL; char *result_str = NULL; cJSON *result_obj = NULL; cJSON *alg_item = NULL; cJSON *pk_item = NULL; cJSON *kid_item = NULL; const crypto_alg_sizes_t *sz; int pr; role_table_init(&table); pq_role = make_entry("pq_sig", "pq-sig", "ml-dsa-65", 0); role_table_add(&table, &pq_role); mnemonic_init(&mnemonic_state); mnemonic_load(&mnemonic_state, mnemonic); memset(&g_key_store, 0, sizeof(g_key_store)); alg_key_cache_init(&g_alg_key_cache); crypto_derive_all(&g_key_store, &table, &mnemonic_state); dispatcher_init(&dispatcher, &table, &mnemonic_state, &g_key_store, &g_alg_key_cache); sz = crypto_alg_get_sizes(CRYPTO_ALG_ML_DSA_65); check_condition("ML-DSA-65 sizes available", sz != NULL); resp = dispatcher_handle_request(&dispatcher, "{\"id\":\"4\",\"method\":\"get_public_key\",\"params\":[{\"algorithm\":\"ml-dsa-65\",\"index\":0}]}"); check_condition("ML-DSA-65 get_public_key returns a response", resp != NULL); pr = parse_result(resp, &result_str, &result_obj); check_condition("ML-DSA-65 get_public_key result is a JSON object", pr == 2 && result_obj != NULL); if (result_obj != NULL && sz != NULL) { alg_item = cJSON_GetObjectItemCaseSensitive(result_obj, "algorithm"); pk_item = cJSON_GetObjectItemCaseSensitive(result_obj, "public_key"); kid_item = cJSON_GetObjectItemCaseSensitive(result_obj, "key_id"); check_condition("ML-DSA-65 structured has algorithm=ml-dsa-65", cJSON_IsString(alg_item) && strcmp(alg_item->valuestring, "ml-dsa-65") == 0); check_condition("ML-DSA-65 structured has large public_key (3904 hex chars)", cJSON_IsString(pk_item) && strlen(pk_item->valuestring) == sz->pub_key_len * 2); check_condition("ML-DSA-65 structured has key_id (16 hex chars)", cJSON_IsString(kid_item) && strlen(kid_item->valuestring) == 16); check_condition("ML-DSA-65 key_id == first 16 chars of public_key", cJSON_IsString(pk_item) && cJSON_IsString(kid_item) && strncmp(pk_item->valuestring, kid_item->valuestring, 16) == 0); } else { check_condition("ML-DSA-65 structured has algorithm=ml-dsa-65", 0); check_condition("ML-DSA-65 structured has large public_key (3904 hex chars)", 0); check_condition("ML-DSA-65 structured has key_id (16 hex chars)", 0); check_condition("ML-DSA-65 key_id == first 16 chars of public_key", 0); } free(result_str); cJSON_Delete(result_obj); free(resp); crypto_wipe(&g_key_store); mnemonic_unload(&mnemonic_state); } /* ---- ML-KEM-768: structured JSON ---- */ { role_table_t table; role_entry_t kem_role; static mnemonic_state_t mnemonic_state; dispatcher_ctx_t dispatcher; char *resp = NULL; char *result_str = NULL; cJSON *result_obj = NULL; cJSON *alg_item = NULL; cJSON *pk_item = NULL; cJSON *kid_item = NULL; const crypto_alg_sizes_t *sz; int pr; role_table_init(&table); kem_role = make_entry("kem_main", "pq-kem", "ml-kem-768", 0); role_table_add(&table, &kem_role); mnemonic_init(&mnemonic_state); mnemonic_load(&mnemonic_state, mnemonic); memset(&g_key_store, 0, sizeof(g_key_store)); alg_key_cache_init(&g_alg_key_cache); crypto_derive_all(&g_key_store, &table, &mnemonic_state); dispatcher_init(&dispatcher, &table, &mnemonic_state, &g_key_store, &g_alg_key_cache); sz = crypto_alg_get_sizes(CRYPTO_ALG_ML_KEM_768); check_condition("ML-KEM-768 sizes available", sz != NULL); resp = dispatcher_handle_request(&dispatcher, "{\"id\":\"5\",\"method\":\"get_public_key\",\"params\":[{\"algorithm\":\"ml-kem-768\",\"index\":0}]}"); check_condition("ML-KEM-768 get_public_key returns a response", resp != NULL); pr = parse_result(resp, &result_str, &result_obj); check_condition("ML-KEM-768 get_public_key result is a JSON object", pr == 2 && result_obj != NULL); if (result_obj != NULL && sz != NULL) { alg_item = cJSON_GetObjectItemCaseSensitive(result_obj, "algorithm"); pk_item = cJSON_GetObjectItemCaseSensitive(result_obj, "public_key"); kid_item = cJSON_GetObjectItemCaseSensitive(result_obj, "key_id"); check_condition("ML-KEM-768 structured has algorithm=ml-kem-768", cJSON_IsString(alg_item) && strcmp(alg_item->valuestring, "ml-kem-768") == 0); check_condition("ML-KEM-768 structured has public_key (2368 hex chars)", cJSON_IsString(pk_item) && strlen(pk_item->valuestring) == sz->pub_key_len * 2); check_condition("ML-KEM-768 structured has key_id (16 hex chars)", cJSON_IsString(kid_item) && strlen(kid_item->valuestring) == 16); check_condition("ML-KEM-768 key_id == first 16 chars of public_key", cJSON_IsString(pk_item) && cJSON_IsString(kid_item) && strncmp(pk_item->valuestring, kid_item->valuestring, 16) == 0); } else { check_condition("ML-KEM-768 structured has algorithm=ml-kem-768", 0); check_condition("ML-KEM-768 structured has public_key (2368 hex chars)", 0); check_condition("ML-KEM-768 structured has key_id (16 hex chars)", 0); check_condition("ML-KEM-768 key_id == first 16 chars of public_key", 0); } free(result_str); cJSON_Delete(result_obj); free(resp); crypto_wipe(&g_key_store); mnemonic_unload(&mnemonic_state); } printf("%d/%d tests passed\n", g_passes, g_total); return (g_passes == g_total) ? 0 : 1; }