ssh: diffie-hellman-group14-sha1

2 files changed, 176 insertions, 47 deletions

diff --git a/lib/ssh/test/ssh_basic_SUITE.erl b/lib/ssh/test/ssh_basic_SUITE.erl
index c71463db30..39ea2c9609 100644
--- a/lib/ssh/test/ssh_basic_SUITE.erl
+++ b/lib/ssh/test/ssh_basic_SUITE.erl
@@ -42,7 +42,7 @@ suite() ->
 all() -> 
     [app_test,
      appup_test,
-     {group, 'diffie-hellman-group-exchange-sha1'},
+     {group, key_exchange},
      {group, dsa_key},
      {group, rsa_key},
      {group, dsa_pass_key},
@@ -93,8 +93,11 @@ groups() ->
 			    max_sessions_sftp_start_channel_parallel,
 			    max_sessions_sftp_start_channel_sequential
 			   ]},
-     {'diffie-hellman-group-exchange-sha1', [], ['diffie-hellman-group-exchange-sha1'
-						]},
+     {key_exchange, [], ['diffie-hellman-group-exchange-sha1',
+			 'diffie-hellman-group-exchange-sha256',
+			 'diffie-hellman-group1-sha1',
+			 'diffie-hellman-group14-sha1'
+			]},
      {dir_options, [], [user_dir_option,
 			system_dir_option]}
     ].
@@ -149,17 +152,11 @@ init_per_group(internal_error, Config) ->
     ssh_test_lib:setup_dsa(DataDir, PrivDir),
     file:delete(filename:join(PrivDir, "system/ssh_host_dsa_key")),
     Config;
-init_per_group('diffie-hellman-group-exchange-sha1', Config) ->
-    case lists:member('diffie-hellman-group-exchange-sha1',
-		      ssh_transport:supported_algorithms(kex)) of
-	true ->
-	    DataDir = ?config(data_dir, Config),
-	    PrivDir = ?config(priv_dir, Config),
-	    ssh_test_lib:setup_rsa(DataDir, PrivDir),
-	    Config;
-	false ->
-	    {skip,"diffie-hellman-group-exchange-sha1 is not supported"}
-    end;
+init_per_group(key_exchange, Config) ->
+    DataDir = ?config(data_dir, Config),
+    PrivDir = ?config(priv_dir, Config),
+    ssh_test_lib:setup_rsa(DataDir, PrivDir),
+    Config;
 init_per_group(dir_options, Config) ->
     PrivDir = ?config(priv_dir, Config),
     %% Make unreadable dir:
@@ -207,6 +204,8 @@ init_per_group(_, Config) ->
 
 end_per_group(hardening_tests, Config) ->
     end_per_group(dsa_key, Config);
+end_per_group(key_exchange, Config) ->
+    end_per_group(rsa_key, Config);
 end_per_group(dsa_key, Config) ->
     PrivDir = ?config(priv_dir, Config),
     ssh_test_lib:clean_dsa(PrivDir),
@@ -833,40 +832,56 @@ ssh_msg_debug_fun_option_client(Config) ->
 
 %%--------------------------------------------------------------------
 'diffie-hellman-group-exchange-sha1'(Config) ->
-    process_flag(trap_exit, true),
-    SystemDir = filename:join(?config(priv_dir, Config), system),
-    UserDir = ?config(priv_dir, Config), 
+    kextest('diffie-hellman-group-exchange-sha1',Config).
 
-    {Pid, Host, Port} = ssh_test_lib:daemon([{system_dir, SystemDir},
-					     {user_dir, UserDir},
-					     {user_passwords, [{"foo", "bar"}]},
-					     {preferred_algorithms,
-					      [{kex, ['diffie-hellman-group-exchange-sha1']}]},
-					     {failfun, fun ssh_test_lib:failfun/2}]),
-    
-    ConnectionRef =
-	ssh_test_lib:connect(Host, Port, [{silently_accept_hosts, true},
-					  {user, "foo"},
-					  {password, "bar"},
-					  {user_dir, UserDir},
-					  {preferred_algorithms,
-					   [{kex, ['diffie-hellman-group-exchange-sha1']}]},
-					  {user_interaction, false}]),
-    check(ConnectionRef, Pid).
+'diffie-hellman-group-exchange-sha256'(Config) ->
+    kextest('diffie-hellman-group-exchange-sha256',Config).
 
-check(ConnectionRef, Pid) ->
-    {ok, ChannelId} = ssh_connection:session_channel(ConnectionRef, infinity),
-    success = ssh_connection:exec(ConnectionRef, ChannelId,
-				  "1+1.", infinity),
-    Data = {ssh_cm, ConnectionRef, {data, ChannelId, 0, <<"2\n">>}},
-    case ssh_test_lib:receive_exec_result(Data) of
-	expected ->
-	    ok;
-	Other ->
-	    ct:fail(Other)
-    end,
-    ssh_test_lib:receive_exec_end(ConnectionRef, ChannelId),
-    ssh:stop_daemon(Pid).
+'diffie-hellman-group1-sha1'(Config) ->
+    kextest('diffie-hellman-group1-sha1',Config).
+
+'diffie-hellman-group14-sha1'(Config) ->
+    kextest('diffie-hellman-group14-sha1',Config).
+
+
+kextest(Kex, Config) ->
+    case lists:member(Kex, ssh_transport:supported_algorithms(kex)) of
+	true ->
+	    process_flag(trap_exit, true),
+	    SystemDir = filename:join(?config(priv_dir, Config), system),
+	    UserDir = ?config(priv_dir, Config), 
+	    
+	    {Pid, Host, Port} = ssh_test_lib:daemon([{system_dir, SystemDir},
+						     {user_dir, UserDir},
+						     {user_passwords, [{"foo", "bar"}]},
+						     {preferred_algorithms,
+						      [{kex, [Kex]}]},
+						     {failfun, fun ssh_test_lib:failfun/2}]),
+    
+	    ConnectionRef =
+		ssh_test_lib:connect(Host, Port, [{silently_accept_hosts, true},
+						  {user, "foo"},
+						  {password, "bar"},
+						  {user_dir, UserDir},
+						  {preferred_algorithms,
+						   [{kex, [Kex]}]},
+						  {user_interaction, false}]),
+	    
+	    {ok, ChannelId} = ssh_connection:session_channel(ConnectionRef, infinity),
+	    success = ssh_connection:exec(ConnectionRef, ChannelId,
+					  "1+1.", infinity),
+	    Data = {ssh_cm, ConnectionRef, {data, ChannelId, 0, <<"2\n">>}},
+	    case ssh_test_lib:receive_exec_result(Data) of
+		expected ->
+		    ok;
+		Other ->
+		    ct:fail(Other)
+	    end,
+	    ssh_test_lib:receive_exec_end(ConnectionRef, ChannelId),
+	    ssh:stop_daemon(Pid);
+	false ->
+	    {skip, lists:concat([Kex, " is not supported"])}
+    end.
 
 %%--------------------------------------------------------------------
 connectfun_disconnectfun_server(Config) ->
diff --git a/lib/ssh/test/ssh_to_openssh_SUITE.erl b/lib/ssh/test/ssh_to_openssh_SUITE.erl
index b7283202a3..fb1c6a1b61 100644
--- a/lib/ssh/test/ssh_to_openssh_SUITE.erl
+++ b/lib/ssh/test/ssh_to_openssh_SUITE.erl
@@ -51,13 +51,15 @@ groups() ->
 			  erlang_client_openssh_server_publickey_rsa,
 			  erlang_client_openssh_server_publickey_dsa,
 			  erlang_client_openssh_server_password,
+			  erlang_client_openssh_server_kexs,
 			  erlang_client_openssh_server_nonexistent_subsystem
 			 ]},
      {erlang_server, [], [erlang_server_openssh_client_exec,
 			  erlang_server_openssh_client_exec_compressed,
 			  erlang_server_openssh_client_pulic_key_dsa,
 			  erlang_server_openssh_client_cipher_suites,
-			  erlang_server_openssh_client_macs]}
+			  erlang_server_openssh_client_macs,
+			  erlang_server_openssh_client_kexs]}
     ].
 
 init_per_suite(Config) ->
@@ -99,6 +101,12 @@ init_per_testcase(erlang_server_openssh_client_cipher_suites, Config) ->
 init_per_testcase(erlang_server_openssh_client_macs, Config) ->
     check_ssh_client_support(Config);
 
+init_per_testcase(erlang_server_openssh_client_kexs, Config) ->
+    check_ssh_client_support(Config);
+
+init_per_testcase(erlang_client_openssh_server_kexs, Config) ->
+    check_ssh_client_support(Config);
+
 init_per_testcase(_TestCase, Config) ->
     ssh:start(),
     Config.
@@ -189,6 +197,48 @@ erlang_client_openssh_server_exec_compressed(Config) when is_list(Config) ->
     end.
 
 %%--------------------------------------------------------------------
+erlang_client_openssh_server_kexs() ->
+    [{doc, "Test that we can connect with different KEXs."}].
+
+erlang_client_openssh_server_kexs(Config) when is_list(Config) ->
+    Success =
+	lists:foldl(
+	  fun(Kex, Acc) ->
+		  ConnectionRef = 
+		      ssh_test_lib:connect(?SSH_DEFAULT_PORT, [{silently_accept_hosts, true},
+							       {user_interaction, false},
+							       {preferred_algorithms,
+								[{kex,[Kex]}]}]),
+    
+		  {ok, ChannelId} =
+		      ssh_connection:session_channel(ConnectionRef, infinity),
+		  success =
+		      ssh_connection:exec(ConnectionRef, ChannelId,
+					  "echo testing", infinity),
+
+		  ExpectedData = {ssh_cm, ConnectionRef, {data, ChannelId, 0, <<"testing\n">>}},
+		  case ssh_test_lib:receive_exec_result(ExpectedData) of
+		      expected ->
+			  ssh_test_lib:receive_exec_end(ConnectionRef, ChannelId),
+			  Acc;
+		      {unexpected_msg,{ssh_cm, ConnectionRef,
+				       {exit_status, ChannelId, 0}} = ExitStatus} ->
+			  ct:pal("0: Collected data ~p", [ExitStatus]),
+			  ssh_test_lib:receive_exec_result(ExpectedData,  ConnectionRef, ChannelId),
+			  Acc;
+		      Other ->
+			  ct:pal("~p failed: ~p",[Kex,Other]),
+			  false
+		  end
+	  end, true, ssh_transport:supported_algorithms(kex)),
+    case Success of
+	true ->
+	    ok;
+	false ->
+	    {fail, "Kex failed for one or more algos"}
+    end.
+
+%%--------------------------------------------------------------------
 erlang_server_openssh_client_exec() ->
     [{doc, "Test that exec command works."}].
 
@@ -322,6 +372,70 @@ erlang_server_openssh_client_macs(Config) when is_list(Config) ->
      ssh:stop_daemon(Pid).
 
 %%--------------------------------------------------------------------
+erlang_server_openssh_client_kexs() ->
+    [{doc, "Test that we can connect with different KEXs."}].
+
+erlang_server_openssh_client_kexs(Config) when is_list(Config) ->
+    SystemDir = ?config(data_dir, Config),
+    PrivDir = ?config(priv_dir, Config),
+    KnownHosts = filename:join(PrivDir, "known_hosts"),
+
+    {Pid, Host, Port} = ssh_test_lib:daemon([{system_dir, SystemDir},
+					     {failfun, fun ssh_test_lib:failfun/2},
+					     {preferred_algorithms,
+					      [{kex,ssh_transport:supported_algorithms(kex)}]}
+					    ]),
+    ct:sleep(500),
+
+    ErlKexs = lists:map(fun erlang:atom_to_list/1, 
+			ssh_transport:supported_algorithms(kex)),
+    OpenSshKexs = string:tokens(os:cmd("ssh -Q kex"), "\n"),
+    
+    Kexs = [{OpenSshKex,lists:member(OpenSshKex,ErlKexs)} 
+	    || OpenSshKex <- OpenSshKexs],
+
+    Success =
+	lists:foldl(
+	  fun({Kex, Expect}, Acc) ->
+		  Cmd = "ssh -p " ++ integer_to_list(Port) ++
+		      " -o UserKnownHostsFile=" ++ KnownHosts ++ " " ++ Host ++ " " ++
+		      " -o KexAlgorithms=" ++ Kex ++ " 1+1.",
+
+		  ct:pal("Cmd: ~p~n", [Cmd]),
+
+		  SshPort = open_port({spawn, Cmd}, [binary, stderr_to_stdout]),
+
+		  case Expect of
+		      true ->
+			  receive
+			      {SshPort,{data, <<"2\n">>}} ->
+				  Acc
+			  after ?TIMEOUT ->
+				  ct:pal("Did not receive answer for ~p",[Kex]),
+				  false
+			  end;
+		      false ->
+			  receive
+			      {SshPort,{data, <<"Unable to negotiate a key exchange method", _/binary>>}} ->
+				  Acc
+			  after ?TIMEOUT ->
+				  ct:pal("Did not receive no matching kex message for ~p",[Kex]),
+				  false
+			  end
+		  end
+	  end, true, Kexs),
+    
+    ssh:stop_daemon(Pid),
+
+    case Success of
+	true ->
+	    ok;
+	false ->
+	    {fail, "Kex failed for one or more algos"}
+    end.
+	
+
+%%--------------------------------------------------------------------
 erlang_server_openssh_client_exec_compressed() ->
     [{doc, "Test that exec command works."}].