mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-12-25 20:34:52 +03:00
9198ad65ef
starts two VMs: - one with dropbear listening from initrd, waiting for a file - another connecting via ssh, creating the file
75 lines
1.9 KiB
Nix
75 lines
1.9 KiB
Nix
import ./make-test.nix ({ pkgs, lib, ... }:
|
|
|
|
let
|
|
keys = pkgs.runCommand "gen-keys" {
|
|
outputs = [ "out" "dbPub" "dbPriv" "sshPub" "sshPriv" ];
|
|
buildInputs = with pkgs; [ dropbear openssh ];
|
|
}
|
|
''
|
|
touch $out
|
|
dropbearkey -t rsa -f $dbPriv -s 4096 | sed -n 2p > $dbPub
|
|
ssh-keygen -q -t rsa -b 4096 -N "" -f client
|
|
mv client $sshPriv
|
|
mv client.pub $sshPub
|
|
'';
|
|
|
|
in {
|
|
name = "initrd-network-ssh";
|
|
meta = with lib.maintainers; {
|
|
maintainers = [ willibutz ];
|
|
};
|
|
|
|
nodes = with lib; rec {
|
|
server =
|
|
{ config, pkgs, ... }:
|
|
{
|
|
boot.kernelParams = [
|
|
"ip=${
|
|
(head config.networking.interfaces.eth1.ip4).address
|
|
}:::255.255.255.0::eth1:none"
|
|
];
|
|
boot.initrd.network = {
|
|
enable = true;
|
|
ssh = {
|
|
enable = true;
|
|
authorizedKeys = [ "${readFile keys.sshPub}" ];
|
|
port = 22;
|
|
hostRSAKey = keys.dbPriv;
|
|
};
|
|
};
|
|
boot.initrd.preLVMCommands = ''
|
|
while true; do
|
|
if [ -f fnord ]; then
|
|
poweroff
|
|
fi
|
|
sleep 1
|
|
done
|
|
'';
|
|
};
|
|
|
|
client =
|
|
{ config, pkgs, ... }:
|
|
{
|
|
environment.etc.knownHosts = {
|
|
text = concatStrings [
|
|
"server,"
|
|
"${toString (head (splitString " " (
|
|
toString (elemAt (splitString "\n" config.networking.extraHosts) 2)
|
|
)))} "
|
|
"${readFile keys.dbPub}"
|
|
];
|
|
};
|
|
};
|
|
};
|
|
|
|
testScript = ''
|
|
startAll;
|
|
$client->waitForUnit("network.target");
|
|
$client->copyFileFromHost("${keys.sshPriv}","/etc/sshKey");
|
|
$client->succeed("chmod 0600 /etc/sshKey");
|
|
$client->waitUntilSucceeds("ping -c 1 server");
|
|
$client->succeed("ssh -i /etc/sshKey -o UserKnownHostsFile=/etc/knownHosts server 'touch /fnord'");
|
|
$client->shutdown;
|
|
'';
|
|
})
|