/*-
 * Copyright 2016 Vsevolod Stakhov
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef KEYPAIR_PRIVATE_H_
#define KEYPAIR_PRIVATE_H_

#include "config.h"
#include "ref.h"
#include "cryptobox.h"

/*
 * KEX cached data
 */
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_nm {
	guchar RSPAMD_ALIGNED(32) nm[rspamd_cryptobox_MAX_NMBYTES];
	ref_entry_t ref;
};

/*
 * Generic keypair
 */
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_keypair  {
	guchar id[rspamd_cryptobox_HASHBYTES];
	enum rspamd_cryptobox_keypair_type type;
	enum rspamd_cryptobox_mode alg;
	ref_entry_t ref;
};

/*
 * NIST p256 ecdh keypair
 */
#define RSPAMD_CRYPTOBOX_KEYPAIR_NIST(x) ((struct rspamd_cryptobox_keypair_nist *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_keypair_nist {
	struct rspamd_cryptobox_keypair parent;
	guchar RSPAMD_ALIGNED(32) sk[32];
	guchar RSPAMD_ALIGNED(32) pk[65];
};

/*
 * Curve25519 ecdh keypair
 */
#define RSPAMD_CRYPTOBOX_KEYPAIR_25519(x) ((struct rspamd_cryptobox_keypair_25519 *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_keypair_25519 {
	struct rspamd_cryptobox_keypair parent;
	guchar RSPAMD_ALIGNED(32) sk[32];
	guchar RSPAMD_ALIGNED(32) pk[32];
};

/*
 * NIST p256 ecdsa keypair
 */
#define RSPAMD_CRYPTOBOX_KEYPAIR_SIG_NIST(x) ((struct rspamd_cryptobox_keypair_sig_nist *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_keypair_sig_nist {
	struct rspamd_cryptobox_keypair parent;
	guchar RSPAMD_ALIGNED(32) sk[32];
	guchar RSPAMD_ALIGNED(32) pk[65];
};

/*
 * Ed25519 keypair
 */
#define RSPAMD_CRYPTOBOX_KEYPAIR_SIG_25519(x) ((struct rspamd_cryptobox_keypair_sig_25519 *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_keypair_sig_25519 {
	struct rspamd_cryptobox_keypair parent;
	guchar RSPAMD_ALIGNED(32) sk[64];
	guchar RSPAMD_ALIGNED(32) pk[32];
};

/*
 * Public component of the keypair
 */
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_pubkey {
	guchar id[rspamd_cryptobox_HASHBYTES];
	struct rspamd_cryptobox_nm *nm;
	enum rspamd_cryptobox_keypair_type type;
	enum rspamd_cryptobox_mode alg;
	ref_entry_t ref;
};

/*
 * Public p256 ecdh
 */
#define RSPAMD_CRYPTOBOX_PUBKEY_NIST(x) ((struct rspamd_cryptobox_pubkey_nist *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_pubkey_nist {
	struct rspamd_cryptobox_pubkey parent;
	guchar RSPAMD_ALIGNED(32) pk[65];
};

/*
 * Public curve25519 ecdh
 */
#define RSPAMD_CRYPTOBOX_PUBKEY_25519(x) ((struct rspamd_cryptobox_pubkey_25519 *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_pubkey_25519 {
	struct rspamd_cryptobox_pubkey parent;
	guchar RSPAMD_ALIGNED(32) pk[32];
};

/*
 * Public p256 ecdsa
 */
#define RSPAMD_CRYPTOBOX_PUBKEY_SIG_NIST(x) ((struct rspamd_cryptobox_pubkey_sig_nist *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_pubkey_sig_nist {
	struct rspamd_cryptobox_pubkey parent;
	guchar RSPAMD_ALIGNED(32) pk[65];
};

/*
 * Public ed25519
 */
#define RSPAMD_CRYPTOBOX_PUBKEY_SIG_25519(x) ((struct rspamd_cryptobox_pubkey_sig_25519 *)(x))
struct RSPAMD_ALIGNED(32) rspamd_cryptobox_pubkey_sig_25519 {
	struct rspamd_cryptobox_pubkey parent;
	guchar RSPAMD_ALIGNED(32) pk[32];
};

void rspamd_cryptobox_nm_dtor (struct rspamd_cryptobox_nm *nm);
void rspamd_cryptobox_keypair_dtor (struct rspamd_cryptobox_keypair *kp);
void rspamd_cryptobox_pubkey_dtor (struct rspamd_cryptobox_pubkey *p);

#endif /* KEYPAIR_PRIVATE_H_ */
