/* $OpenBSD: kexgexs.c,v 1.47 2024/05/17 00:30:23 djm Exp $ */
/*
 * Copyright (c) 2000 Niels Provos.  All rights reserved.
 * Copyright (c) 2001 Markus Friedl.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "includes.h"

#ifdef WITH_OPENSSL


#include <stdarg.h>
#include <stdio.h>
#include <string.h>
#include <signal.h>

#include <openssl/dh.h>

#include "openbsd-compat/openssl-compat.h"

#include "sshkey.h"
#include "cipher.h"
#include "digest.h"
#include "kex.h"
#include "log.h"
#include "packet.h"
#include "dh.h"
#include "ssh2.h"
#ifdef GSSAPI
#include "ssh-gss.h"
#endif
#include "monitor_wrap.h"
#include "dispatch.h"
#include "ssherr.h"
#include "sshbuf.h"
#include "misc.h"

static int input_kex_dh_gex_request(int, u_int32_t, struct ssh *);
static int input_kex_dh_gex_init(int, u_int32_t, struct ssh *);

int
kexgex_server(struct ssh *ssh)
{
        ssh_dispatch_set(ssh, SSH2_MSG_KEX_DH_GEX_REQUEST,
            &input_kex_dh_gex_request);
        debug("expecting SSH2_MSG_KEX_DH_GEX_REQUEST");
        return 0;
}

static int
input_kex_dh_gex_request(int type, u_int32_t seq, struct ssh *ssh)
{
        struct kex *kex = ssh->kex;
        int r;
        u_int min = 0, max = 0, nbits = 0;
        const BIGNUM *dh_p, *dh_g;

        debug("SSH2_MSG_KEX_DH_GEX_REQUEST received");
        ssh_dispatch_set(ssh, SSH2_MSG_KEX_DH_GEX_REQUEST, &kex_protocol_error);

        if ((r = sshpkt_get_u32(ssh, &min)) != 0 ||
            (r = sshpkt_get_u32(ssh, &nbits)) != 0 ||
            (r = sshpkt_get_u32(ssh, &max)) != 0 ||
            (r = sshpkt_get_end(ssh)) != 0)
                goto out;
        kex->nbits = nbits;
        kex->min = min;
        kex->max = max;
        min = MAXIMUM(DH_GRP_MIN, min);
        max = MINIMUM(DH_GRP_MAX, max);
        nbits = MAXIMUM(DH_GRP_MIN, nbits);
        nbits = MINIMUM(DH_GRP_MAX, nbits);

        if (kex->max < kex->min || kex->nbits < kex->min ||
            kex->max < kex->nbits || kex->max < DH_GRP_MIN) {
                r = SSH_ERR_DH_GEX_OUT_OF_RANGE;
                goto out;
        }

        /* Contact privileged parent */
        kex->dh = mm_choose_dh(min, nbits, max);
        if (kex->dh == NULL) {
                (void)sshpkt_disconnect(ssh, "no matching DH grp found");
                r = SSH_ERR_ALLOC_FAIL;
                goto out;
        }
        debug("SSH2_MSG_KEX_DH_GEX_GROUP sent");
        DH_get0_pqg(kex->dh, &dh_p, NULL, &dh_g);
        if ((r = sshpkt_start(ssh, SSH2_MSG_KEX_DH_GEX_GROUP)) != 0 ||
            (r = sshpkt_put_bignum2(ssh, dh_p)) != 0 ||
            (r = sshpkt_put_bignum2(ssh, dh_g)) != 0 ||
            (r = sshpkt_send(ssh)) != 0)
                goto out;

        /* Compute our exchange value in parallel with the client */
        if ((r = dh_gen_key(kex->dh, kex->we_need * 8)) != 0)
                goto out;

        debug("expecting SSH2_MSG_KEX_DH_GEX_INIT");
        ssh_dispatch_set(ssh, SSH2_MSG_KEX_DH_GEX_INIT, &input_kex_dh_gex_init);
        r = 0;
 out:
        return r;
}

static int
input_kex_dh_gex_init(int type, u_int32_t seq, struct ssh *ssh)
{
        struct kex *kex = ssh->kex;
        BIGNUM *dh_client_pub = NULL;
        const BIGNUM *pub_key, *dh_p, *dh_g;
        struct sshbuf *shared_secret = NULL;
        struct sshbuf *server_host_key_blob = NULL;
        struct sshkey *server_host_public, *server_host_private;
        u_char *signature = NULL;
        u_char hash[SSH_DIGEST_MAX_LENGTH];
        size_t slen, hashlen;
        int r;

        debug("SSH2_MSG_KEX_DH_GEX_INIT received");
        ssh_dispatch_set(ssh, SSH2_MSG_KEX_DH_GEX_INIT, &kex_protocol_error);

        if ((r = kex_load_hostkey(ssh, &server_host_private,
            &server_host_public)) != 0)
                goto out;

        /* key, cert */
        if ((r = sshpkt_get_bignum2(ssh, &dh_client_pub)) != 0 ||
            (r = sshpkt_get_end(ssh)) != 0)
                goto out;
        if ((shared_secret = sshbuf_new()) == NULL) {
                r = SSH_ERR_ALLOC_FAIL;
                goto out;
        }
        if ((r = kex_dh_compute_key(kex, dh_client_pub, shared_secret)) != 0)
                goto out;
        if ((server_host_key_blob = sshbuf_new()) == NULL) {
                r = SSH_ERR_ALLOC_FAIL;
                goto out;
        }
        if ((r = sshkey_putb(server_host_public, server_host_key_blob)) != 0)
                goto out;

        /* calc H */
        DH_get0_key(kex->dh, &pub_key, NULL);
        DH_get0_pqg(kex->dh, &dh_p, NULL, &dh_g);
        hashlen = sizeof(hash);
        if ((r = kexgex_hash(
            kex->hash_alg,
            kex->client_version,
            kex->server_version,
            kex->peer,
            kex->my,
            server_host_key_blob,
            kex->min, kex->nbits, kex->max,
            dh_p, dh_g,
            dh_client_pub,
            pub_key,
            sshbuf_ptr(shared_secret), sshbuf_len(shared_secret),
            hash, &hashlen)) != 0)
                goto out;

        /* sign H */
        if ((r = kex->sign(ssh, server_host_private, server_host_public,
            &signature, &slen, hash, hashlen, kex->hostkey_alg)) < 0)
                goto out;

        /* send server hostkey, DH pubkey 'f' and signed H */
        if ((r = sshpkt_start(ssh, SSH2_MSG_KEX_DH_GEX_REPLY)) != 0 ||
            (r = sshpkt_put_stringb(ssh, server_host_key_blob)) != 0 ||
            (r = sshpkt_put_bignum2(ssh, pub_key)) != 0 ||     /* f */
            (r = sshpkt_put_string(ssh, signature, slen)) != 0 ||
            (r = sshpkt_send(ssh)) != 0)
                goto out;

        if ((r = kex_derive_keys(ssh, hash, hashlen, shared_secret)) != 0 ||
            (r = kex_send_newkeys(ssh)) != 0)
                goto out;

        /* retain copy of hostkey used at initial KEX */
        if (kex->initial_hostkey == NULL &&
            (r = sshkey_from_private(server_host_public,
            &kex->initial_hostkey)) != 0)
                goto out;
        /* success */
 out:
        explicit_bzero(hash, sizeof(hash));
        DH_free(kex->dh);
        kex->dh = NULL;
        BN_clear_free(dh_client_pub);
        sshbuf_free(shared_secret);
        sshbuf_free(server_host_key_blob);
        free(signature);
        return r;
}
#endif /* WITH_OPENSSL */