2016-03-14 03:21:33 +03:00
|
|
|
#! /usr/bin/env bash
|
2016-03-14 12:59:18 +03:00
|
|
|
|
2017-03-04 18:22:37 +03:00
|
|
|
# More info at: https://github.com/elitak/nixos-infect
|
2016-03-14 12:59:18 +03:00
|
|
|
|
2017-02-14 12:25:42 +03:00
|
|
|
set -e -o pipefail
|
2017-01-25 02:39:15 +03:00
|
|
|
|
2016-03-14 08:09:30 +03:00
|
|
|
makeConf() {
|
2017-02-14 12:25:42 +03:00
|
|
|
# Skip everything if main config already present
|
|
|
|
[[ -e /etc/nixos/configuration.nix ]] && return 0
|
2016-04-09 22:39:35 +03:00
|
|
|
# NB <<"EOF" quotes / $ ` in heredocs, <<EOF does not
|
2016-03-14 12:59:18 +03:00
|
|
|
mkdir -p /etc/nixos
|
2019-07-07 20:09:47 +03:00
|
|
|
# Prevent grep for sending error code 1 (and halting execution) when no lines are selected : https://www.unix.com/man-page/posix/1P/grep
|
2020-02-20 02:22:02 +03:00
|
|
|
local IFS=$'\n'
|
|
|
|
for trypath in /root/.ssh/authorized_keys $HOME/.ssh/authorized_keys; do
|
|
|
|
[[ -r "$trypath" ]] \
|
|
|
|
&& keys=$(sed -E 's/^.*((ssh|ecdsa)-[^[:space:]]+)[[:space:]]+([^[:space:]]+)([[:space:]]*.*)$/\1 \3\4/' "$trypath") \
|
|
|
|
&& break
|
|
|
|
done
|
2018-12-08 02:16:00 +03:00
|
|
|
local network_import=""
|
|
|
|
|
|
|
|
[ "$PROVIDER" = "digitalocean" ] && network_import="./networking.nix # generated at runtime by nixos-infect"
|
2016-03-14 12:59:18 +03:00
|
|
|
cat > /etc/nixos/configuration.nix << EOF
|
2016-03-14 23:28:14 +03:00
|
|
|
{ ... }: {
|
2016-03-14 08:09:30 +03:00
|
|
|
imports = [
|
|
|
|
./hardware-configuration.nix
|
2018-12-08 02:16:00 +03:00
|
|
|
$network_import
|
2017-04-19 02:25:42 +03:00
|
|
|
$NIXOS_IMPORT
|
2016-03-14 08:09:30 +03:00
|
|
|
];
|
|
|
|
|
|
|
|
boot.cleanTmpDir = true;
|
2016-03-14 12:59:18 +03:00
|
|
|
networking.hostName = "$(hostname)";
|
2016-03-14 23:28:14 +03:00
|
|
|
networking.firewall.allowPing = true;
|
|
|
|
services.openssh.enable = true;
|
2020-02-20 02:22:02 +03:00
|
|
|
users.users.root.openssh.authorizedKeys.keys = [$(while read -r line; do echo -n "
|
|
|
|
\"$line\" "; done <<< "$keys")
|
2016-03-14 23:28:14 +03:00
|
|
|
];
|
2016-03-14 08:09:30 +03:00
|
|
|
}
|
|
|
|
EOF
|
2017-02-17 05:51:05 +03:00
|
|
|
# If you rerun this later, be sure to prune the filesSystems attr
|
2016-03-14 12:59:18 +03:00
|
|
|
cat > /etc/nixos/hardware-configuration.nix << EOF
|
2016-06-22 22:45:38 +03:00
|
|
|
{ ... }:
|
|
|
|
{
|
2016-03-14 08:09:30 +03:00
|
|
|
imports = [ <nixpkgs/nixos/modules/profiles/qemu-guest.nix> ];
|
2020-02-14 04:04:27 +03:00
|
|
|
boot.loader.grub.device = "$grubdev";
|
|
|
|
fileSystems."/" = { device = "$rootfsdev"; fsType = "ext4"; };
|
2016-03-14 08:09:30 +03:00
|
|
|
}
|
2016-04-09 22:39:35 +03:00
|
|
|
EOF
|
2019-05-05 21:23:09 +03:00
|
|
|
|
2018-12-09 01:16:18 +03:00
|
|
|
if [ "$PROVIDER" = "digitalocean" ]
|
|
|
|
then
|
|
|
|
makeNetworkingConf
|
|
|
|
else
|
|
|
|
true
|
|
|
|
fi
|
2018-12-08 02:16:00 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
makeNetworkingConf() {
|
2017-02-14 11:37:47 +03:00
|
|
|
# XXX It'd be better if we used procfs for all this...
|
2016-04-09 22:39:35 +03:00
|
|
|
local IFS=$'\n'
|
2017-04-10 18:50:09 +03:00
|
|
|
eth0_name=$(ip address show | grep '^2:' | awk -F': ' '{print $2}')
|
2017-04-15 19:55:08 +03:00
|
|
|
eth0_ip4s=$(ip address show dev "$eth0_name" | grep 'inet ' | sed -r 's|.*inet ([0-9.]+)/([0-9]+).*|{ address="\1"; prefixLength=\2; }|')
|
|
|
|
eth0_ip6s=$(ip address show dev "$eth0_name" | grep 'inet6 ' | sed -r 's|.*inet6 ([0-9a-f:]+)/([0-9]+).*|{ address="\1"; prefixLength=\2; }|' || '')
|
|
|
|
gateway=$(ip route show dev "$eth0_name" | grep default | sed -r 's|default via ([0-9.]+).*|\1|')
|
2018-07-13 01:02:14 +03:00
|
|
|
gateway6=$(ip -6 route show dev "$eth0_name" | grep default | sed -r 's|default via ([0-9a-f:]+).*|\1|' || true)
|
2017-04-15 19:55:08 +03:00
|
|
|
ether0=$(ip address show dev "$eth0_name" | grep link/ether | sed -r 's|.*link/ether ([0-9a-f:]+) .*|\1|')
|
2017-04-10 18:50:09 +03:00
|
|
|
|
|
|
|
eth1_name=$(ip address show | grep '^3:' | awk -F': ' '{print $2}')||true
|
|
|
|
if [ -n "$eth1_name" ];then
|
2017-04-15 19:55:08 +03:00
|
|
|
eth1_ip4s=$(ip address show dev "$eth1_name" | grep 'inet ' | sed -r 's|.*inet ([0-9.]+)/([0-9]+).*|{ address="\1"; prefixLength=\2; }|')
|
|
|
|
eth1_ip6s=$(ip address show dev "$eth1_name" | grep 'inet6 ' | sed -r 's|.*inet6 ([0-9a-f:]+)/([0-9]+).*|{ address="\1"; prefixLength=\2; }|' || '')
|
|
|
|
ether1=$(ip address show dev "$eth1_name" | grep link/ether | sed -r 's|.*link/ether ([0-9a-f:]+) .*|\1|')
|
2017-04-10 18:50:09 +03:00
|
|
|
interfaces1=<< EOF
|
|
|
|
$eth1_name = {
|
2018-03-23 23:37:49 +03:00
|
|
|
ipv4.addresses = [$(for a in "${eth1_ip4s[@]}"; do echo -n "
|
2017-04-10 18:50:09 +03:00
|
|
|
$a"; done)
|
|
|
|
];
|
2018-03-23 23:37:49 +03:00
|
|
|
ipv6.addresses = [$(for a in "${eth1_ip6s[@]}"; do echo -n "
|
2017-04-10 18:50:09 +03:00
|
|
|
$a"; done)
|
|
|
|
];
|
|
|
|
EOF
|
2018-07-13 01:02:14 +03:00
|
|
|
extraRules1="ATTR{address}==\"${ether1}\", NAME=\"${eth1_name}\""
|
2017-04-10 18:50:09 +03:00
|
|
|
else
|
|
|
|
interfaces1=""
|
|
|
|
extraRules1=""
|
|
|
|
fi
|
|
|
|
|
2016-04-09 22:39:35 +03:00
|
|
|
nameservers=($(grep ^nameserver /etc/resolv.conf | cut -f2 -d' '))
|
2018-07-12 21:38:36 +03:00
|
|
|
if [ "$eth0_name" = eth* ]; then
|
|
|
|
predictable_inames="usePredictableInterfaceNames = lib.mkForce false;"
|
|
|
|
else
|
|
|
|
predictable_inames="usePredictableInterfaceNames = lib.mkForce true;"
|
|
|
|
fi
|
2016-04-09 22:39:35 +03:00
|
|
|
cat > /etc/nixos/networking.nix << EOF
|
2018-07-12 21:38:36 +03:00
|
|
|
{ lib, ... }: {
|
2016-04-09 22:39:35 +03:00
|
|
|
# This file was populated at runtime with the networking
|
|
|
|
# details gathered from the active system.
|
|
|
|
networking = {
|
2017-04-15 19:46:17 +03:00
|
|
|
nameservers = [$(for a in "${nameservers[@]}"; do echo -n "
|
2016-04-09 22:39:35 +03:00
|
|
|
\"$a\""; done)
|
|
|
|
];
|
|
|
|
defaultGateway = "${gateway}";
|
|
|
|
defaultGateway6 = "${gateway6}";
|
2018-06-10 02:25:27 +03:00
|
|
|
dhcpcd.enable = false;
|
2018-07-12 21:38:36 +03:00
|
|
|
$predictable_inames
|
2016-04-09 22:39:35 +03:00
|
|
|
interfaces = {
|
2017-04-10 18:50:09 +03:00
|
|
|
$eth0_name = {
|
2018-03-24 19:29:49 +03:00
|
|
|
ipv4.addresses = [$(for a in "${eth0_ip4s[@]}"; do echo -n "
|
2016-04-09 22:39:35 +03:00
|
|
|
$a"; done)
|
|
|
|
];
|
2018-03-24 19:29:49 +03:00
|
|
|
ipv6.addresses = [$(for a in "${eth0_ip6s[@]}"; do echo -n "
|
2016-04-09 22:39:35 +03:00
|
|
|
$a"; done)
|
|
|
|
];
|
2019-10-11 10:32:54 +03:00
|
|
|
ipv4.routes = [ { address = "${gateway}"; prefixLength = 32; } ];
|
|
|
|
ipv6.routes = [ { address = "${gateway6}"; prefixLength = 32; } ];
|
2016-04-09 22:39:35 +03:00
|
|
|
};
|
2017-04-10 18:50:09 +03:00
|
|
|
$interfaces1
|
2016-04-09 22:39:35 +03:00
|
|
|
};
|
|
|
|
};
|
|
|
|
services.udev.extraRules = ''
|
2018-07-12 21:38:36 +03:00
|
|
|
ATTR{address}=="${ether0}", NAME="${eth0_name}"
|
2017-04-10 18:50:09 +03:00
|
|
|
$extraRules1
|
2016-04-09 22:39:35 +03:00
|
|
|
'';
|
|
|
|
}
|
2016-03-14 08:09:30 +03:00
|
|
|
EOF
|
2016-03-14 12:59:18 +03:00
|
|
|
#! /usr/bin/env bash
|
|
|
|
# NB put your semi-sensitive (not posted to github) configuration in a separate
|
|
|
|
# file and include it via this customConfig() function. e.g.:
|
|
|
|
# customConfig() {
|
|
|
|
# cat > /etc/nixos/custom.nix << EOF
|
|
|
|
# { config, lib, pkgs, ... }: {
|
|
|
|
# }
|
|
|
|
# EOF
|
|
|
|
# }
|
|
|
|
#
|
|
|
|
# then you can add the files in configuration.nix's imports above and run something like:
|
|
|
|
# cat customConfig nixos-infect | root@targethost bash
|
2017-02-14 11:18:59 +03:00
|
|
|
if [[ "$(type -t customConfig)" == "function" ]]; then customConfig; fi
|
2016-03-14 08:09:30 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
makeSwap() {
|
2017-02-14 04:30:42 +03:00
|
|
|
# TODO check currently available swapspace first
|
2017-04-15 19:52:39 +03:00
|
|
|
swapFile=$(mktemp /tmp/nixos-infect.XXXXX.swp)
|
2017-04-15 19:50:01 +03:00
|
|
|
dd if=/dev/zero "of=$swapFile" bs=1M count=$((1*1024))
|
|
|
|
chmod 0600 "$swapFile"
|
|
|
|
mkswap "$swapFile"
|
|
|
|
swapon -v "$swapFile"
|
2017-03-04 18:22:46 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
removeSwap() {
|
2018-07-13 00:27:59 +03:00
|
|
|
swapoff -a
|
|
|
|
rm -vf /tmp/nixos-infect.*.swp
|
2016-03-14 08:09:30 +03:00
|
|
|
}
|
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
prepareEnv() {
|
2020-02-14 04:04:27 +03:00
|
|
|
# $grubdev is used in makeConf()
|
|
|
|
for grubdev in /dev/vda /dev/sda; do [[ -e $grubdev ]] && break; done
|
|
|
|
|
|
|
|
# Retrieve root fs block device
|
|
|
|
# (get root mount) (get partition or logical volume)
|
|
|
|
rootfsdev=$(mount | grep "on / type" | awk '{print $1;}')
|
2016-08-18 15:23:42 +03:00
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
# DigitalOcean doesn't seem to set USER while running user data
|
|
|
|
export USER="root"
|
|
|
|
export HOME="/root"
|
2016-03-14 08:09:30 +03:00
|
|
|
|
2020-04-05 23:31:17 +03:00
|
|
|
# Nix installer tries to use sudo regardless of whether we're already uid 0
|
|
|
|
#which sudo || { sudo() { eval "$@"; }; export -f sudo; }
|
|
|
|
# shellcheck disable=SC2174
|
|
|
|
mkdir -p -m 0755 /nix
|
|
|
|
}
|
|
|
|
|
|
|
|
fakeCurlUsingWget() {
|
2017-02-14 11:18:59 +03:00
|
|
|
# Use adapted wget if curl is missing
|
2020-04-05 23:31:17 +03:00
|
|
|
which wget && { \
|
2017-02-14 11:18:59 +03:00
|
|
|
curl() {
|
|
|
|
eval "wget $(
|
|
|
|
(local isStdout=1
|
|
|
|
for arg in "$@"; do
|
|
|
|
case "$arg" in
|
|
|
|
"-o")
|
|
|
|
echo "-O";
|
|
|
|
isStdout=0
|
|
|
|
;;
|
|
|
|
"-O")
|
|
|
|
isStdout=0
|
|
|
|
;;
|
|
|
|
"-L")
|
|
|
|
;;
|
|
|
|
*)
|
|
|
|
echo "$arg"
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
done;
|
|
|
|
[[ $isStdout -eq 1 ]] && echo "-O-"
|
|
|
|
)| tr '\n' ' '
|
|
|
|
)"
|
|
|
|
}; export -f curl; }
|
|
|
|
}
|
|
|
|
|
|
|
|
req() {
|
|
|
|
type "$1" > /dev/null 2>&1 || which "$1" > /dev/null 2>&1
|
2017-02-14 04:30:42 +03:00
|
|
|
}
|
2016-03-14 08:09:30 +03:00
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
checkEnv() {
|
2017-02-14 11:18:59 +03:00
|
|
|
# Perform some easy fixups before checking
|
2020-04-05 23:31:17 +03:00
|
|
|
# TODO prevent multiple calls to apt-get update
|
2017-02-14 11:18:59 +03:00
|
|
|
which dnf && dnf install -y perl-Digest-SHA # Fedora 24
|
2017-04-10 18:50:09 +03:00
|
|
|
which bzcat || (which yum && yum install -y bzip2) \
|
2019-07-07 20:09:47 +03:00
|
|
|
|| (which apt-get && apt-get update && apt-get install -y bzip2) \
|
2019-05-05 21:23:09 +03:00
|
|
|
|| true
|
2020-04-05 23:31:17 +03:00
|
|
|
which curl || fakeCurlUsingWget \
|
|
|
|
|| (which apt-get && apt-get update && apt-get install -y curl) \
|
|
|
|
|| true
|
2017-02-14 11:18:59 +03:00
|
|
|
|
|
|
|
[[ "$(whoami)" == "root" ]] || { echo "ERROR: Must run as root"; return 1; }
|
|
|
|
|
2017-02-14 12:25:42 +03:00
|
|
|
req curl || req wget || { echo "ERROR: Missing both curl and wget"; return 1; }
|
|
|
|
req bzcat || { echo "ERROR: Missing bzcat"; return 1; }
|
|
|
|
req groupadd || { echo "ERROR: Missing groupadd"; return 1; }
|
|
|
|
req useradd || { echo "ERROR: Missing useradd"; return 1; }
|
|
|
|
req ip || { echo "ERROR: Missing ip"; return 1; }
|
|
|
|
req awk || { echo "ERROR: Missing awk"; return 1; }
|
|
|
|
req cut || { echo "ERROR: Missing cut"; return 1; }
|
2017-02-14 04:30:42 +03:00
|
|
|
}
|
2016-03-14 08:09:30 +03:00
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
infect() {
|
2017-02-14 11:18:59 +03:00
|
|
|
# Add nix build users
|
|
|
|
# FIXME run only if necessary, rather than defaulting true
|
2017-02-17 05:51:05 +03:00
|
|
|
groupadd nixbld -g 30000 || true
|
2017-04-15 19:56:26 +03:00
|
|
|
for i in {1..10}; do useradd -c "Nix build user $i" -d /var/empty -g nixbld -G nixbld -M -N -r -s "$(which nologin)" nixbld$i || true; done
|
2017-02-17 05:51:05 +03:00
|
|
|
# TODO use addgroup and adduser as fallbacks
|
|
|
|
#addgroup nixbld -g 30000 || true
|
|
|
|
#for i in {1..10}; do adduser -DH -G nixbld nixbld$i || true; done
|
2016-03-14 08:09:30 +03:00
|
|
|
|
2017-02-14 11:18:59 +03:00
|
|
|
curl https://nixos.org/nix/install | $SHELL
|
2016-09-07 21:36:01 +03:00
|
|
|
|
2017-04-15 19:57:55 +03:00
|
|
|
# shellcheck disable=SC1090
|
2017-02-14 04:30:42 +03:00
|
|
|
source ~/.nix-profile/etc/profile.d/nix.sh
|
2016-03-14 08:09:30 +03:00
|
|
|
|
2019-11-13 00:14:32 +03:00
|
|
|
[[ -z "$NIX_CHANNEL" ]] && NIX_CHANNEL="nixos-19.09"
|
2017-02-14 04:30:42 +03:00
|
|
|
nix-channel --remove nixpkgs
|
|
|
|
nix-channel --add "https://nixos.org/channels/$NIX_CHANNEL" nixos
|
|
|
|
nix-channel --update
|
2016-03-14 08:09:30 +03:00
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
export NIXOS_CONFIG=/etc/nixos/configuration.nix
|
2016-08-18 15:23:42 +03:00
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
nix-env --set \
|
|
|
|
-I nixpkgs=$HOME/.nix-defexpr/channels/nixos \
|
|
|
|
-f '<nixpkgs/nixos>' \
|
|
|
|
-p /nix/var/nix/profiles/system \
|
|
|
|
-A system
|
2016-08-18 15:23:42 +03:00
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
# Remove nix installed with curl | bash
|
|
|
|
rm -fv /nix/var/nix/profiles/default*
|
|
|
|
/nix/var/nix/profiles/system/sw/bin/nix-collect-garbage
|
2016-08-18 15:23:42 +03:00
|
|
|
|
2017-02-14 11:18:59 +03:00
|
|
|
# Reify resolv.conf
|
|
|
|
[[ -L /etc/resolv.conf ]] && mv -v /etc/resolv.conf /etc/resolv.conf.lnk && cat /etc/resolv.conf.lnk > /etc/resolv.conf
|
2017-02-14 04:30:42 +03:00
|
|
|
|
2017-02-14 11:18:59 +03:00
|
|
|
# Stage the Nix coup d'état
|
2017-02-14 04:30:42 +03:00
|
|
|
touch /etc/NIXOS
|
2017-02-14 11:18:59 +03:00
|
|
|
echo etc/nixos > /etc/NIXOS_LUSTRATE
|
|
|
|
echo etc/resolv.conf >> /etc/NIXOS_LUSTRATE
|
|
|
|
echo root/.nix-defexpr/channels >> /etc/NIXOS_LUSTRATE
|
2017-02-14 04:30:42 +03:00
|
|
|
|
2017-02-14 12:25:42 +03:00
|
|
|
rm -rf /boot.bak
|
|
|
|
mv -v /boot /boot.bak
|
|
|
|
/nix/var/nix/profiles/system/bin/switch-to-configuration boot
|
2017-02-14 04:30:42 +03:00
|
|
|
}
|
2016-08-18 15:23:42 +03:00
|
|
|
|
2018-12-08 02:16:00 +03:00
|
|
|
[ -z "$PROVIDER" ] && PROVIDER="digitalocean" # you may also prepend PROVIDER=vultr to your call instead
|
|
|
|
|
2017-02-14 04:30:42 +03:00
|
|
|
prepareEnv
|
2019-05-11 05:42:20 +03:00
|
|
|
makeSwap # smallest (512MB) droplet needs extra memory!
|
2017-02-14 11:18:59 +03:00
|
|
|
checkEnv
|
|
|
|
makeConf
|
|
|
|
infect
|
2017-03-04 18:22:46 +03:00
|
|
|
removeSwap
|
2019-05-05 21:23:09 +03:00
|
|
|
|
|
|
|
if [[ -z "$NO_REBOOT" ]]; then
|
|
|
|
reboot
|
|
|
|
fi
|