1296633Sdes#	$OpenBSD: dhgex.sh,v 1.3 2015/10/23 02:22:01 dtucker Exp $
2263635Sdes#	Placed in the Public Domain.
3263635Sdes
4263635Sdestid="dhgex"
5263635Sdes
6263635SdesLOG=${TEST_SSH_LOGFILE}
7263635Sdesrm -f ${LOG}
8294328Sdescp $OBJ/sshd_proxy $OBJ/sshd_proxy_bak
9263635Sdes
10263635Sdeskexs=`${SSH} -Q kex | grep diffie-hellman-group-exchange`
11263635Sdes
12263635Sdesssh_test_dhgex()
13263635Sdes{
14263635Sdes	bits="$1"; shift
15263635Sdes	cipher="$1"; shift
16263635Sdes	kex="$1"; shift
17263635Sdes
18294328Sdes	cp $OBJ/sshd_proxy_bak $OBJ/sshd_proxy
19294328Sdes	echo "KexAlgorithms=$kex" >> $OBJ/sshd_proxy
20294328Sdes	echo "Ciphers=$cipher" >> $OBJ/sshd_proxy
21263635Sdes	rm -f ${LOG}
22263635Sdes	opts="-oKexAlgorithms=$kex -oCiphers=$cipher"
23296633Sdes	min=2048
24296633Sdes	max=8192
25296633Sdes	groupsz="$min<$bits<$max"
26263635Sdes	verbose "$tid bits $bits $kex $cipher"
27263635Sdes	${SSH} ${opts} $@ -vvv -F ${OBJ}/ssh_proxy somehost true
28263635Sdes	if [ $? -ne 0 ]; then
29263635Sdes		fail "ssh failed ($@)"
30263635Sdes	fi
31263635Sdes	# check what we request
32263635Sdes	grep "SSH2_MSG_KEX_DH_GEX_REQUEST($groupsz) sent" ${LOG} >/dev/null
33263635Sdes	if [ $? != 0 ]; then
34263635Sdes		got=`egrep "SSH2_MSG_KEX_DH_GEX_REQUEST(.*) sent" ${LOG}`
35263635Sdes		fail "$tid unexpected GEX sizes, expected $groupsz, got $got"
36263635Sdes	fi
37263635Sdes	# check what we got (depends on contents of system moduli file)
38263635Sdes	gotbits="`awk '/bits set:/{print $4}' ${LOG} | head -1 | cut -f2 -d/`"
39263635Sdes	if [ "$gotbits" -lt "$bits" ]; then
40263635Sdes		fatal "$tid expected $bits bit group, got $gotbits"
41263635Sdes	fi
42263635Sdes}
43263635Sdes
44263635Sdescheck()
45263635Sdes{
46263635Sdes	bits="$1"; shift
47263635Sdes
48263635Sdes	for c in $@; do
49263635Sdes		for k in $kexs; do
50263635Sdes			ssh_test_dhgex $bits $c $k
51263635Sdes		done
52263635Sdes	done
53263635Sdes}
54263635Sdes
55263635Sdes#check 2048 3des-cbc
56263635Sdescheck 3072 `${SSH} -Q cipher | grep 128`
57263635Sdescheck 3072 arcfour blowfish-cbc
58263635Sdescheck 7680 `${SSH} -Q cipher | grep 192`
59263635Sdescheck 8192 `${SSH} -Q cipher | grep 256`
60263635Sdescheck 8192 rijndael-cbc@lysator.liu.se chacha20-poly1305@openssh.com
61