+static int test_set_defaults(void)
+{
+ RAND_DRBG *master = NULL, *public = NULL, *private = NULL;
+
+ /* Check the default type and flags for master, public and private */
+ return TEST_ptr(master = RAND_DRBG_get0_master())
+ && TEST_ptr(public = RAND_DRBG_get0_public())
+ && TEST_ptr(private = RAND_DRBG_get0_private())
+ && TEST_int_eq(master->type, RAND_DRBG_TYPE)
+ && TEST_int_eq(master->flags,
+ RAND_DRBG_FLAGS | RAND_DRBG_FLAG_MASTER)
+ && TEST_int_eq(public->type, RAND_DRBG_TYPE)
+ && TEST_int_eq(public->flags,
+ RAND_DRBG_FLAGS | RAND_DRBG_FLAG_PUBLIC)
+ && TEST_int_eq(private->type, RAND_DRBG_TYPE)
+ && TEST_int_eq(private->flags,
+ RAND_DRBG_FLAGS | RAND_DRBG_FLAG_PRIVATE)
+
+ /* change master DRBG and check again */
+ && TEST_true(RAND_DRBG_set_defaults(NID_sha256,
+ RAND_DRBG_FLAG_MASTER))
+ && TEST_true(RAND_DRBG_uninstantiate(master))
+ && TEST_int_eq(master->type, NID_sha256)
+ && TEST_int_eq(master->flags, RAND_DRBG_FLAG_MASTER)
+ && TEST_int_eq(public->type, RAND_DRBG_TYPE)
+ && TEST_int_eq(public->flags,
+ RAND_DRBG_FLAGS | RAND_DRBG_FLAG_PUBLIC)
+ && TEST_int_eq(private->type, RAND_DRBG_TYPE)
+ && TEST_int_eq(private->flags,
+ RAND_DRBG_FLAGS | RAND_DRBG_FLAG_PRIVATE)
+ /* change private DRBG and check again */
+ && TEST_true(RAND_DRBG_set_defaults(NID_sha256,
+ RAND_DRBG_FLAG_PRIVATE|RAND_DRBG_FLAG_HMAC))
+ && TEST_true(RAND_DRBG_uninstantiate(private))
+ && TEST_int_eq(master->type, NID_sha256)
+ && TEST_int_eq(master->flags, RAND_DRBG_FLAG_MASTER)
+ && TEST_int_eq(public->type, RAND_DRBG_TYPE)
+ && TEST_int_eq(public->flags,
+ RAND_DRBG_FLAGS | RAND_DRBG_FLAG_PUBLIC)
+ && TEST_int_eq(private->type, NID_sha256)
+ && TEST_int_eq(private->flags,
+ RAND_DRBG_FLAG_PRIVATE | RAND_DRBG_FLAG_HMAC)
+ /* change public DRBG and check again */
+ && TEST_true(RAND_DRBG_set_defaults(NID_sha1,
+ RAND_DRBG_FLAG_PUBLIC
+ | RAND_DRBG_FLAG_HMAC))
+ && TEST_true(RAND_DRBG_uninstantiate(public))
+ && TEST_int_eq(master->type, NID_sha256)
+ && TEST_int_eq(master->flags, RAND_DRBG_FLAG_MASTER)
+ && TEST_int_eq(public->type, NID_sha1)
+ && TEST_int_eq(public->flags,
+ RAND_DRBG_FLAG_PUBLIC | RAND_DRBG_FLAG_HMAC)
+ && TEST_int_eq(private->type, NID_sha256)
+ && TEST_int_eq(private->flags,
+ RAND_DRBG_FLAG_PRIVATE | RAND_DRBG_FLAG_HMAC)
+ /* Change DRBG defaults and change public and check again */
+ && TEST_true(RAND_DRBG_set_defaults(NID_sha256, 0))
+ && TEST_true(RAND_DRBG_uninstantiate(public))
+ && TEST_int_eq(public->type, NID_sha256)
+ && TEST_int_eq(public->flags, RAND_DRBG_FLAG_PUBLIC)
+
+ /* Change DRBG defaults and change master and check again */
+ && TEST_true(RAND_DRBG_set_defaults(NID_aes_256_ctr,
+ RAND_DRBG_FLAG_CTR_NO_DF))
+ && TEST_true(RAND_DRBG_uninstantiate(master))
+ && TEST_int_eq(master->type, NID_aes_256_ctr)
+ && TEST_int_eq(master->flags,
+ RAND_DRBG_FLAG_MASTER|RAND_DRBG_FLAG_CTR_NO_DF)
+
+ /* Reset back to the standard defaults */
+ && TEST_true(RAND_DRBG_set_defaults(RAND_DRBG_TYPE,
+ RAND_DRBG_FLAGS
+ | RAND_DRBG_FLAG_MASTER
+ | RAND_DRBG_FLAG_PUBLIC
+ | RAND_DRBG_FLAG_PRIVATE))
+ && TEST_true(RAND_DRBG_uninstantiate(master))
+ && TEST_true(RAND_DRBG_uninstantiate(public))
+ && TEST_true(RAND_DRBG_uninstantiate(private));
+}
+