openssh-portable/kem.h

#ifndef KEM_H
#define KEM_H

#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <openssl/rsa.h>
#include <openssl/evp.h>
#include <openssl/sha.h>
#include <openssl/err.h>
#include <openssl/param_build.h>
#include "kex.h"

struct kex;

#define MAX_KEM 32
#define KEY_SIZE 32

struct kem {
    //int (*kem[MAX_KEM])(struct ssh *);
    // RSA *pubkey;
};

struct kem_init {
    struct rsa_keys *keys_mine;
    struct rsa_keys *keys_peer;
};

struct rsa_keys {
	EVP_PKEY *pkey;
	unsigned char *n;
	size_t len_n;
	unsigned char *e;
	size_t len_e;
	unsigned char *d;
	size_t len_d;
};

struct ssh;

int kem_rsa_keypair(struct kex *);
int kem_rsa_enc(struct kex *, const struct sshbuf *, struct sshbuf **, struct sshbuf **);
int kem_rsa_dec(struct kex *, struct sshbuf *, struct sshbuf **);
int init_rsa_keys(struct rsa_keys **);
int kem_gen_keys(struct rsa_keys *);
int generate_rand_key(char **, size_t *);
int build_pkey(struct rsa_keys *);
int decrypt_key(const char *, const size_t, struct rsa_keys *, unsigned char **, size_t *);
int encrypt_key(const char *, const size_t, const struct rsa_keys *, unsigned char **, size_t *);
int derive_key(const char *, const size_t, char *);
void read_buffer_kem(struct rsa_keys *, char *, size_t);

#endif