ssh: make ssh_algorithms_SUITE test public user and host keys

Conflicts:
	lib/ssh/src/ssh_transport.erl

2 files changed, 114 insertions, 55 deletions

diff --git a/lib/ssh/test/ssh_algorithms_SUITE.erl b/lib/ssh/test/ssh_algorithms_SUITE.erl
index 6e6269d3e0..cebd1296fa 100644
--- a/lib/ssh/test/ssh_algorithms_SUITE.erl
+++ b/lib/ssh/test/ssh_algorithms_SUITE.erl
@@ -68,7 +68,7 @@ groups() ->
 
     TagGroupSet ++ AlgoTcSet.
 
-tags() -> [kex,cipher,mac,compression].
+tags() -> [kex,cipher,mac,compression,public_key].
 two_way_tags() -> [cipher,mac,compression].
     
 %%--------------------------------------------------------------------
@@ -123,20 +123,35 @@ init_per_group(Group, Config) ->
 	    Tag = proplists:get_value(name,
 				      hd(proplists:get_value(tc_group_path, Config))),
 	    Alg = Group,
-	    PA =
-		case split(Alg) of
-		    [_] ->
-			[Alg];
-		    [A1,A2] ->
-			[{client2server,[A1]},
-			 {server2client,[A2]}]
-		end,
-	    ct:log("Init tests for tag=~p alg=~p",[Tag,PA]),
-	    PrefAlgs = {preferred_algorithms,[{Tag,PA}]},
-	    start_std_daemon([PrefAlgs],
-			     [{pref_algs,PrefAlgs} | Config])
+            init_per_group(Tag, Alg, Config)
     end.
 
+
+init_per_group(public_key=Tag, Alg, Config) ->
+    ct:log("Init tests for public_key ~p",[Alg]),
+    PrefAlgs = {preferred_algorithms,[{Tag,[Alg]}]},
+    %% Daemon started later in init_per_testcase
+    [{pref_algs,PrefAlgs},
+     {tag_alg,{Tag,Alg}}
+     | Config];
+
+init_per_group(Tag, Alg, Config) ->
+    PA =
+        case split(Alg) of
+            [_] ->
+                [Alg];
+            [A1,A2] ->
+                [{client2server,[A1]},
+                 {server2client,[A2]}]
+        end,
+    ct:log("Init tests for tag=~p alg=~p",[Tag,PA]),
+    PrefAlgs = {preferred_algorithms,[{Tag,PA}]},
+    start_std_daemon([PrefAlgs],
+                     [{pref_algs,PrefAlgs},
+                      {tag_alg,{Tag,Alg}}
+                      | Config]).
+
+
 end_per_group(_Alg, Config) ->
     case proplists:get_value(srvr_pid,Config) of
 	Pid when is_pid(Pid) ->
@@ -148,23 +163,49 @@ end_per_group(_Alg, Config) ->
 
 
 
-init_per_testcase(sshc_simple_exec_os_cmd, Config) ->
-    start_pubkey_daemon([proplists:get_value(pref_algs,Config)], Config);
-init_per_testcase(_TC, Config) ->
-    Config.
+init_per_testcase(TC, Config) ->
+    init_per_testcase(TC, proplists:get_value(tag_alg,Config), Config).
 
 
-end_per_testcase(sshc_simple_exec_os_cmd, Config) ->
-    case proplists:get_value(srvr_pid,Config) of
-	Pid when is_pid(Pid) ->
-	    ssh:stop_daemon(Pid),
-	    ct:log("stopped ~p",[proplists:get_value(srvr_addr,Config)]);
-	_ ->
-	    ok
+init_per_testcase(_, {public_key,Alg}, Config) ->
+    Opts = pubkey_opts(Config),
+    case {ssh_file:user_key(Alg,Opts), ssh_file:host_key(Alg,Opts)} of
+        {{ok,_}, {ok,_}} ->
+            start_pubkey_daemon([proplists:get_value(pref_algs,Config)],
+                                [{extra_daemon,true}|Config]);
+        {{ok,_}, _} ->
+            {skip, "No host key"};
+        
+        {_, {ok,_}} ->
+            {skip, "No user key"};
+        
+        _ ->
+            {skip, "Neither host nor user key"}
     end;
-end_per_testcase(_TC, Config) ->
+
+init_per_testcase(sshc_simple_exec_os_cmd, _, Config) ->
+     start_pubkey_daemon([proplists:get_value(pref_algs,Config)],
+                         [{extra_daemon,true}|Config]);
+
+init_per_testcase(_, _, Config) ->
     Config.
 
+
+end_per_testcase(_TC, Config) ->
+    case proplists:get_value(extra_daemon, Config, false) of
+        true ->
+            case proplists:get_value(srvr_pid,Config) of
+                Pid when is_pid(Pid) ->
+                    ssh:stop_daemon(Pid),
+                    ct:log("stopped ~p",[proplists:get_value(srvr_addr,Config)]),
+                    Config;
+                _ ->
+                    Config
+            end;
+        _ ->
+            Config
+    end.
+
 %%--------------------------------------------------------------------
 %% Test Cases --------------------------------------------------------
 %%--------------------------------------------------------------------
@@ -318,29 +359,32 @@ concat(A1, A2) -> list_to_atom(lists:concat([A1," + ",A2])).
 split(Alg) -> ssh_test_lib:to_atoms(string:tokens(atom_to_list(Alg), " + ")).
 
 specific_test_cases(Tag, Alg, SshcAlgos, SshdAlgos, TypeSSH) -> 
-    [simple_exec, simple_sftp] ++
-	case supports(Tag, Alg, SshcAlgos) of
-	    true when TypeSSH == openSSH ->
-                [sshc_simple_exec_os_cmd];
-            _ ->
-                []
-	end ++
-	case supports(Tag, Alg, SshdAlgos) of
-	    true ->
-		[sshd_simple_exec];
-	    _ ->
-		[]
-	end ++
-	case {Tag,Alg} of
-	    {kex,_} when Alg == 'diffie-hellman-group-exchange-sha1' ;
-			 Alg == 'diffie-hellman-group-exchange-sha256' ->
-		[simple_exec_groups,
-		 simple_exec_groups_no_match_too_large,
-		 simple_exec_groups_no_match_too_small
-		];
-	    _ ->
-		[]
-	end.
+    case Tag of
+        public_key -> [];
+        _ -> [simple_exec, simple_sftp]
+    end 
+    ++ case supports(Tag, Alg, SshcAlgos) of
+           true when TypeSSH == openSSH ->
+               [sshc_simple_exec_os_cmd];
+           _ ->
+               []
+       end ++
+       case supports(Tag, Alg, SshdAlgos) of
+           true ->
+               [sshd_simple_exec];
+           _ ->
+               []
+       end ++
+       case {Tag,Alg} of
+           {kex,_} when Alg == 'diffie-hellman-group-exchange-sha1' ;
+                        Alg == 'diffie-hellman-group-exchange-sha256' ->
+               [simple_exec_groups,
+                simple_exec_groups_no_match_too_large,
+                simple_exec_groups_no_match_too_small
+               ];
+           _ ->
+               []
+       end.
 
 supports(Tag, Alg, Algos) ->
     lists:all(fun(A) ->
@@ -370,19 +414,30 @@ start_std_daemon(Opts, Config) ->
     ct:log("started ~p:~p  ~p",[Host,Port,Opts]),
     [{srvr_pid,Pid},{srvr_addr,{Host,Port}} | Config].
 
+
 start_pubkey_daemon(Opts0, Config) ->
-    Opts = [{auth_methods,"publickey"}|Opts0],
-    {Pid, Host, Port} = ssh_test_lib:std_daemon1(Config, Opts),
-    ct:log("started pubkey_daemon ~p:~p  ~p",[Host,Port,Opts]),
+    ct:log("starting pubkey_daemon",[]),
+    Opts = pubkey_opts(Config) ++ Opts0,
+    {Pid, Host, Port} = ssh_test_lib:daemon([{failfun, fun ssh_test_lib:failfun/2}
+                                             | Opts]),
+    ct:log("started ~p:~p  ~p",[Host,Port,Opts]),
     [{srvr_pid,Pid},{srvr_addr,{Host,Port}} | Config].
 
 
+pubkey_opts(Config) ->
+    SystemDir = filename:join(proplists:get_value(priv_dir,Config), "system"),
+    [{auth_methods,"publickey"},
+     {system_dir, SystemDir}].
+
+
 setup_pubkey(Config) ->
     DataDir = proplists:get_value(data_dir, Config),
     UserDir = proplists:get_value(priv_dir, Config),
-    ssh_test_lib:setup_dsa(DataDir, UserDir),
-    ssh_test_lib:setup_rsa(DataDir, UserDir),
-    ssh_test_lib:setup_ecdsa("256", DataDir, UserDir),
+    Keys =
+        [ssh_test_lib:setup_dsa(DataDir, UserDir),
+         ssh_test_lib:setup_rsa(DataDir, UserDir),
+         ssh_test_lib:setup_ecdsa("256", DataDir, UserDir)],
+    ssh_test_lib:write_auth_keys(Keys, UserDir), % 'authorized_keys' shall contain ALL pub keys
     Config.
 
 
diff --git a/lib/ssh/test/ssh_test_lib.erl b/lib/ssh/test/ssh_test_lib.erl
index 36ae2525da..7b273fecef 100644
--- a/lib/ssh/test/ssh_test_lib.erl
+++ b/lib/ssh/test/ssh_test_lib.erl
@@ -500,8 +500,12 @@ setup_ecdsa_auth_keys(_Size, Dir, UserDir) ->
 setup_auth_keys(Keys, Dir) ->
     AuthKeys = public_key:ssh_encode(Keys, auth_keys),
     AuthKeysFile = filename:join(Dir, "authorized_keys"),
-    file:write_file(AuthKeysFile, AuthKeys).
+    ok = file:write_file(AuthKeysFile, AuthKeys),
+    AuthKeys.
 
+write_auth_keys(Keys, Dir) ->
+    AuthKeysFile = filename:join(Dir, "authorized_keys"),
+    file:write_file(AuthKeysFile, Keys).
 
 del_dirs(Dir) ->
     case file:list_dir(Dir) of