roborio-vm/create_rootfs.sh at master · robotpy/roborio-vm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/bin/bash -e

NBD_DEVICE=/dev/nbd0

cd `dirname $0`
source params

if [ -f "$IMG_FILE" ]; then
  echo "$IMG_FILE already exists! Please delete it if you wish to build a new image"
  exit 1
fi

if [ "$1" == "" ]; then
  echo "Usage: $0 /path/to/FRC_roboRIO_*.zip"
  exit 1
fi

ROBORIO_ZIP="$1"

QEMU_NBD=`which qemu-nbd`

function abspath {
  return "$(cd "$(dirname "$1")"; pwd)/$(basename "$1")"
}

function rm_mount {
  if [ -d mnt ]; then
    sudo umount mnt || true
    rmdir mnt
  fi
}

function rm_unpacked {
  [ ! -d unpacked ] || rm -rf unpacked
}

function cleanup {
  echo "Cleaning up..."
  rm_unpacked
  rm_mount

  if [ ! -z "$NBD_DEVICE" ]; then
    sudo "$QEMU_NBD" -d $NBD_DEVICE
  fi
}

trap cleanup EXIT

rm_unpacked
rm_mount

# Unpack the roborio image zipfile, ensure we're sane
mkdir unpacked

unzip "$ROBORIO_ZIP" -d unpacked
# there are two files in there, one is a zip file, unzip the zip file
mkdir unpacked/more
unzip unpacked/*.zip -d unpacked/more

# Make sure the file we're looking for is there
if [ ! -f unpacked/more/systemimage.tar.gz ]; then
  echo "Error: Expected to find systemimage.tar.gz, did not find it!"
  exit 1
fi

# Create the qemu image
qemu-img create -f qcow2 "$IMG_FILE" $HDD_SIZE

# Mount it, format it (requires root access!)
sudo modprobe nbd

sudo "$QEMU_NBD" -d /dev/nbd0
sudo "$QEMU_NBD" -c $NBD_DEVICE "$IMG_FILE"

echo "Formatting image, this may take a few minutes..."
# TODO: probably should use a different filesystem
sudo mkfs.ext3 $NBD_DEVICE

mkdir mnt
sudo mount -t ext3 $NBD_DEVICE mnt

# Untar the file onto the image..
echo "Unpacking FRC image..."
sudo tar -xf unpacked/more/systemimage.tar.gz --directory mnt

# Modify the startup configuration to enable SSHD
STARTUP_INI_FILE=mnt/etc/natinst/share/ni-rt.ini
sudo python _modify_ini.py ${STARTUP_INI_FILE} SYSTEMSETTINGS host_name roboRIO-VM
sudo python _modify_ini.py ${STARTUP_INI_FILE} SYSTEMSETTINGS sshd.enabled True
sudo python _modify_ini.py ${STARTUP_INI_FILE} SYSTEMSETTINGS ConsoleOut.enabled True

# Unmount it
rm_mount

# Create a snapshot in case someone wants to revert their VM without rebuilding it
qemu-img snapshot -c initial "$IMG_FILE"

echo "Successfully created $IMG_FILE!"