noise_init_pqc.c

// SPDX-License-Identifier: GPL-2.0
/*
 * WireGuard PQC Extension - Noise Initialization with PQC Support
 */

#include "noise.h"
#include "device.h"
#include "peer.h"
#include "crypto_mode.h"

#include <linux/slab.h>
#include <linux/string.h>

/*
 * Extended handshake initializer that wraps wg_noise_handshake_init() and
 * additionally allocates PQC key storage when PQC mode is active.
 */
int wg_noise_handshake_init_with_pqc(struct noise_handshake *handshake,
				     struct noise_static_identity *static_identity,
				     const u8 peer_public_key[NOISE_PUBLIC_KEY_LEN],
				     const u8 peer_preshared_key[NOISE_SYMMETRIC_KEY_LEN],
				     struct wg_peer *peer)
{
	wg_noise_handshake_init(handshake, static_identity, peer_public_key,
				peer_preshared_key, peer);

	if (wg_mode_pqc_enabled()) {
		int ret = wg_noise_pqc_init_handshake(handshake);
		if (ret < 0) {
			pr_err("WG_PQC: Failed to initialize PQC handshake: %d\n", ret);
			wg_noise_handshake_clear(handshake);
			return ret;
		}
	}

	return 0;
}

/* Extended handshake clear that also frees PQC key material. */
void wg_noise_handshake_clear_with_pqc(struct noise_handshake *handshake)
{
	if (wg_mode_pqc_enabled())
		wg_noise_pqc_clear_handshake(handshake);

	wg_noise_handshake_clear(handshake);
}

/* Initialises PQC key storage on the static identity if PQC mode is active. */
int wg_noise_static_identity_init_pqc(struct noise_static_identity *identity)
{
	if (!wg_mode_pqc_enabled())
		return 0;

	return wg_noise_pqc_init_identity(identity);
}

/* Clears PQC key material from the static identity if PQC mode is active. */
void wg_noise_static_identity_clear_pqc(struct noise_static_identity *identity)
{
	if (wg_mode_pqc_enabled())
		wg_noise_pqc_clear_identity(identity);
}