Merge pull request #2 from lanets/feat/resize_root_partition

Feat/resize root partition

main.tf

@@ -123,4 +123,35 @@ resource "vsphere_virtual_machine" "vm" {
       dns_server_list = var.nameservers
     }
   }
+
+  # Upload partition extend script
+  provisioner "file" {
+    source      = "${path.module}/scripts/partresize.sh"
+    destination = "/tmp/partresize.sh"
+
+    connection {
+      type     = "ssh"
+      user     = var.ssh_user
+      password = var.ssh_password
+      host     = var.networking.interfaces[0].ipv4_address
+    }
+
+  }
+
+  # Extend system partition
+  provisioner "remote-exec" {
+
+    connection {
+      type     = "ssh"
+      user     = var.ssh_user
+      password = var.ssh_password
+      host     = var.networking.interfaces[0].ipv4_address
+    }
+
+    inline = [
+      "sudo chmod +x /tmp/partresize.sh",
+      "echo 'y' | sudo /tmp/partresize.sh -p /dev/sda5 -l -f",
+      "sudo shutdown -r",
+    ]
+  }
 }

scripts/partresize.sh

@@ -0,0 +1,140 @@
+#!/bin/bash
+##########
+# Script to resize a LVM Partition after extending the underlying disk device. Can be used on physical or virtual machines alike.
+# Tested with CentOS6, RHEL6, CentOS7, RHEL7. This script is only intended for MBR partitioned disks and not for GPT.
+#
+# The script will first resize the partition by changing the partition end sector of the selected partition, and then after a reboot resize the filesystem.
+# By default it rescans the SCSI bus to check a change in disk size if the disk was hot-extended, which is easy with VMs, and only then proceeds.
+# If the extended disk size is recognized by the OS already, you can force resizing with the -f flag.
+#
+# Github: https://github.com/alpacacode/Homebrewn-Scripts
+########
+
+extenddisk_parted() {
+  # Use parted because fdisk behavior can vary between OSes and scripting fdisk is non-deterministic.
+  # Using parted resizepart would be easer, but RHEL/CentOS6 parted doesn't support resizepart
+  echo -e "\nThis will now extend partition number $partitionnum on disk $disk using start sector $startsector.\nWARNING: Make sure you backup your boot sector prior to this."
+  read -r -p "Are you sure? [y/N] " response
+  if [[ $response =~ ^([yY][eE][sS]|[yY])$ ]]
+  then
+    echo -e "\n+++Current partition layout of $disk:+++"
+    parted $disk --script unit s print
+    if [ $logical == 1 ]
+    then
+      parted $disk --script rm $ext_partitionnum
+      parted $disk --script "mkpart extended ${ext_startsector}s -1s"
+      parted $disk --script "set $ext_partitionnum lba off"
+      parted $disk --script "mkpart logical ext2 ${startsector}s -1s"
+    else
+      parted $disk --script rm $partitionnum
+      parted $disk --script "mkpart primary ext2 ${startsector}s -1s"
+    fi
+    parted $disk --script set $partitionnum lvm on
+    echo -e "\n\n+++New partition layout of $disk:+++"
+    parted $disk --script unit s print
+    # The 2nd script to expand the filesystem will be automatically executed on the next reboot.
+    echo "#!/bin/bash
+#Extend Physical Volume first
+pvresize $p
+
+#Extend LVM, using 100% of the free allocation units and resize filesystem
+lvextend --extents +100%FREE $l --resizefs
+chmod -x \$0" > /root/fsresize.sh
+    chmod +x /root/fsresize.sh
+    # Use a temporary systemd service or a rc.local script for extending the filesystem during next reboot, depending on what the OS is running.
+    if(pidof systemd)
+    then
+      resizefs_systemd
+    else
+      resizefs_rclocal
+    fi
+
+    echo -e "Done. The system need a reboot.\n"
+  else
+    echo -e "Aborted by user.\n"
+    exit 1
+  fi
+}
+
+resizefs_rclocal() {
+  # Resize the filesystem using a script in rc.local if the OS run with sysvinit.
+  echo "#Cleanup rc.local again
+sed -i /etc/rc.local -e '/\/root\/fsresize\.sh/d' --follow-symlinks
+sed -i /etc/rc.local -re 's/^#(exit 0)$/\1/' --follow-symlinks" >> /root/fsresize.sh
+
+  sed -i /etc/rc.local -re 's/^(exit 0)$/#\1/' --follow-symlinks
+  echo "/root/fsresize.sh" >> /etc/rc.local
+}
+
+resizefs_systemd() {
+  # Resize the filesystem using a script called by a temporary systemd service file if the OS runs with systemd.
+  echo "#Cleanup systemd autostart script again.
+systemctl disable fsresize.service
+rm -f /etc/systemd/system/fsresize.service" >> /root/fsresize.sh
+
+  echo "[Unit]
+Description=Filesystem resize script for LVM volume $l
+
+[Service]
+ExecStart=/root/fsresize.sh
+
+[Install]
+WantedBy=multi-user.target" > /etc/systemd/system/fsresize.service
+  systemctl enable fsresize.service
+}
+
+l=$(sudo lvdisplay | grep 'LV Path' | grep -o '\/[a-z0-9/-]*/root')
+p=$(sudo pvdisplay | grep 'PV Name' | grep -o '\/[a-z0-9/-]*')
+
+command -v fdisk >/dev/null 2>&1 && command -v parted >/dev/null 2>&1 && command -v pvresize >/dev/null 2>&1 || {
+  echo -e "Error: Some of the required utilities (fdisk, parted, lvm tools etc) don't seem to be installed on this system.  Aborting.\n" >&2
+  exit 1
+}
+
+# Check if a valid LVM physical volume was supplied by verifying the pvdisplay exit code ($?).
+pvdisplay $p > /dev/null
+if [ $? != 0 ] || ( ! (file $p | grep -q "block special"))
+then
+  echo -e "Error: $p does not look like a block device or LVM physical volume. Aborting.\n"
+  usage
+fi
+
+# Check if a valid LVM logical volume was supplied by verifying the lvdisplay exit code ($?).
+lvdisplay $l > /dev/null
+if [ $? != 0 ]
+then
+  echo -e "Error: $l does not look like a LVM logical volume. Aborting.\n"
+  usage
+fi
+
+# Fill variables for later use.
+disk=$(echo $p | rev | cut -c 2- | rev) # /dev/sda
+diskshort=$(echo $disk | grep -Po '[^\/]+$') # sda
+partitionnum=$(echo $p | grep -Po '\d$') # 2
+startsector=$(fdisk -u -l $disk | grep $p | awk '{print $2}')
+
+# Detect LVM on logical/extended partition
+layout=$(parted $disk --script unit s print)
+if grep -Pq "^\s$partitionnum\s+.+?logical.+$" <<< "$layout"
+then
+  echo -e "Detected LVM residing on a logical partition.\n"
+  logical=1
+  ext_partitionnum=$(parted $disk --script unit s print | grep extended | grep -Po '^\s\d\s' | tr -d ' ')
+  ext_startsector=$(parted $disk --script unit s print | grep extended | awk '{print $2}' | tr -d 's')
+else
+  logical=0
+fi
+
+parted $disk --script unit s print | if ! grep -Pq "^\s$partitionnum\s+.+?[^,]+?lvm\s*$"
+then
+  echo -e "Error: $p seems to have some flags other than the lvm flag set. Other flags are not supported."
+  usage
+fi
+
+if ! (fdisk -u -l $disk | grep $disk | tail -1 | grep $p | grep -q "Linux LVM")
+then
+  echo -e "Error: $p is not the last LVM volume on disk $disk. Cannot expand.\n"
+  usage
+fi
+
+extenddisk_parted

variables.tf

@@ -16,6 +16,16 @@ variable "template" {
   description = "Name of the template available in the vSphere"
 }
 
+variable "ssh_user" {
+  description = "Username to connect to ssh"
+  type        = string
+}
+
+variable "ssh_password" {
+  description = "Datastore to deploy the VM (With DRS only!)."
+   type        = string
+}
+
 variable "annotation" {
   description = "A user-provided description of the virtual machine. The default is no annotation."
   default     = "Managed by Terraform. NEVER EDIT THE VM MANUALY!"