sysadmin/ipxe
1
0
Fork 0
mirror of https://github.com/ipxe/ipxe synced 2026-06-29 00:07:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

[test] Allow for large values in key exchange self-tests

Browse Source 
Allow for the existence of key exchange algorithms where the public
keys and shared secrets may be too large for a stack allocation.

Signed-off-by: Michael Brown <mcb30@ipxe.org>
This commit is contained in:
Michael Brown
2026-06-09 13:35:01 +01:00
parent f2ade220f9
commit 6dcb3b1e67
1 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/tests/exchange_test.c
+18 -8
View File
@@ -47,8 +47,10 @@ FILE_LICENCE ( GPL2_OR_LATER_OR_UBDL );
void exchange_okx ( struct exchange_test *test, const char *file,
unsigned int line ) {
struct exchange_algorithm *exchange = test->exchange;
uint8_t public[ exchange->pubsize ];
uint8_t shared[ exchange->sharedsize ];
struct {
uint8_t public[ exchange->pubsize ];
uint8_t shared[ exchange->sharedsize ];
} *actual;
int rc;
/* Sanity checks */
@@ -58,28 +60,36 @@ void exchange_okx ( struct exchange_test *test, const char *file,
okx ( ( ( test->shared_len == 0 ) ||
( test->shared_len == exchange->sharedsize ) ), file, line );
/* Allocate result buffer */
actual = malloc ( sizeof ( *actual ) );
okx ( actual != NULL, file, line );
/* Verify calculation of public key */
DBGC ( test, "KEX %s private key:\n", exchange->name );
DBGC_HDA ( test, 0, test->private, exchange->privsize );
exchange_public ( exchange, test->private, public );
exchange_public ( exchange, test->private, actual->public );
DBGC ( test, "KEX %s public key:\n", exchange->name );
DBGC_HDA ( test, 0, public, exchange->pubsize );
okx ( memcmp ( public, test->public, exchange->pubsize ) == 0,
DBGC_HDA ( test, 0, actual->public, exchange->pubsize );
okx ( memcmp ( actual->public, test->public, exchange->pubsize ) == 0,
file, line );
/* Verify calculation of shared secret */
DBGC ( test, "KEX %s partner key:\n", exchange->name );
DBGC_HDA ( test, 0, test->partner, exchange->pubsize );
rc = exchange_shared ( exchange, test->private, test->partner, shared );
rc = exchange_shared ( exchange, test->private, test->partner,
actual->shared );
if ( test->shared_len ) {
/* Verify successful calculation */
okx ( rc == 0, file, line );
DBGC ( test, "KEX %s shared secret:\n", exchange->name );
DBGC_HDA ( test, 0, shared, exchange->sharedsize );
okx ( memcmp ( shared, test->shared,
DBGC_HDA ( test, 0, actual->shared, exchange->sharedsize );
okx ( memcmp ( actual->shared, test->shared,
exchange->sharedsize ) == 0, file, line );
} else {
/* Verify failure */
okx ( rc != 0, file, line );
}
/* Free result buffer */
free ( actual );
}