getch 0.1.0 → 0.1.6

data/lib/getch/filesystem/zfs/encrypt/void.rb

@@ -0,0 +1,96 @@
+require_relative '../../../helpers'
+
+module Getch
+  module FileSystem
+    module Zfs
+      module Encrypt
+        class Void < Device
+          include Helpers::Void
+          attr_reader :boot_disk
+
+          def initialize
+            super
+            hostid
+            zfs_zed # mountpoint for zfs
+          end
+
+          # Fstab contain:
+          # > /efi noauto
+          # > swap
+          # > /boot zfs-legacy
+          # > /tmp
+          def fstab
+            conf = "#{MOUNTPOINT}/etc/fstab"
+            File.write(conf, "\n", mode: 'w', chmod: 0644)
+            line_fstab(@dev_esp, "/efi vfat noauto,rw,relatime 0 0") if @dev_esp
+            add_line(conf, "/dev/mapper/cryptswap none swap sw 0 0")
+            add_line(conf, "##{@boot_pool_name}/BOOT/#{@n} /boot zfs defaults 0 0") if @dev_boot
+            add_line(conf, "tmpfs /tmp tmpfs defaults,nosuid,nodev 0 0")
+          end
+
+          def config_dracut
+            conf = "#{MOUNTPOINT}/etc/dracut.conf.d/zfs.conf"
+            # dracut: value+= should be surrounding by white space
+            content = [
+              "hostonly=\"yes\"",
+              "omit_dracutmodules+=\" btrfs lvm \"",
+              "install_items+=\" /etc/crypttab \"",
+              ""
+            ]
+            File.write(conf, content.join("\n"), mode: 'w', chmod: 0644)
+          end
+
+          def kernel_cmdline_dracut
+            #command "zfs set mountpoint=legacy #{@boot_pool_name}/BOOT/#{@n}"
+          end
+
+          def config_grub
+            grub_cmdline("root=zfs:#{@pool_name}/ROOT/#{@n}", "zfs_force=1")
+          end
+
+          def finish
+            zed_update_path
+            puts "+ Enter in your system: chroot /mnt /bin/bash"
+            puts "+ Reboot with: shutdown -r now"
+          end
+
+          def crypttab
+            line_crypttab("cryptswap", @dev_swap, "/dev/urandom", "swap,discard,cipher=aes-xts-plain64:sha256,size=512")
+          end
+
+          private
+
+          def line_crypttab(mapname, dev, point, rest)
+            conf = "#{MOUNTPOINT}/etc/crypttab"
+            device = s_uuid(dev)
+            raise "No partuuid for #{dev} #{device}" if !device
+            raise "Bad partuuid for #{dev} #{device}" if device.kind_of? Array
+            add_line(conf, "#{mapname} PARTUUID=#{device} #{point} #{rest}")
+          end
+
+          def zfs_zed
+            service_dir = "/etc/runit/runsvdir/default/"
+
+            Helpers::mkdir("#{MOUNTPOINT}/etc/zfs/zfs-list.cache")
+            Helpers::touch("#{MOUNTPOINT}/etc/zfs/zfs-list.cache/#{@boot_pool_name}") if @dev_boot
+            Helpers::touch("#{MOUNTPOINT}/etc/zfs/zfs-list.cache/#{@pool_name}")
+            fork { command "/etc/sv/zed/run" }
+            command "ln -fs /etc/sv/zed #{service_dir}"
+          end
+
+          def zed_update_path
+            Dir.glob("#{MOUNTPOINT}/etc/zfs/zfs-list.cache/*").each { |f|
+              if !system("sed", "-Ei", "s|#{MOUNTPOINT}/?|/|", f)
+                raise "System exec sed"
+              end
+            }
+          end
+
+          def hostid
+            command "zgenhostid $(hostid)"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/getch/filesystem/zfs/format.rb

@@ -1,19 +1,112 @@
 module Getch
   module FileSystem
     module Zfs
-      class Format < Getch::FileSystem::Zfs::Device
+      class Format < Device
         def initialize
           super
-          @state = Getch::States.new()
+          @log = Getch::Log.new
+          @state = Getch::States.new
+          if ! @id
+            @log.info "Research pool id for #{@dev_root}..."
+            @id = Helpers::pool_id(@dev_root)
+            @boot_pool_name = "bpool-#{@id}"
+            @pool_name = "rpool-#{@id}"
+          end
           format
         end
 
+        private
+
         def format
           return if STATES[:format]
-          system("mkfs.fat -F32 #{@dev_boot_efi}") if @dev_boot_efi
-          system("mkswap -f #{@dev_swap}")
+          raise "Error, no id found for #{@dev_root}." if ! @id
+          @log.info "Create #{@id} for #{@pool_name}"
+          system("mkfs.fat -F32 #{@dev_esp}") if @dev_esp
+          zfs
+          cache
+          datasets
           @state.format
         end
+
+        def zfs
+          bloc=`blockdev --getpbsz #{@dev_root}`
+          ashift = case bloc
+            when /8096/
+              13
+            when /4096/
+              12
+            else # 512
+              9
+            end
+
+          Helpers::mkdir(MOUNTPOINT)
+          @log.debug("ashift found for #{bloc} - #{ashift}")
+
+          if @dev_boot
+            # https://openzfs.github.io/openzfs-docs/Getting%20Started/Ubuntu/Ubuntu%2020.04%20Root%20on%20ZFS.html
+            @log.info("Creating boot pool on #{@pool_name}")
+            exec("zpool create -f \\
+              -o ashift=#{ashift} -o autotrim=on -d \\
+              -o feature@async_destroy=enabled \\
+              -o feature@bookmarks=enabled \\
+              -o feature@embedded_data=enabled \\
+              -o feature@empty_bpobj=enabled \\
+              -o feature@enabled_txg=enabled \\
+              -o feature@extensible_dataset=enabled \\
+              -o feature@filesystem_limits=enabled \\
+              -o feature@hole_birth=enabled \\
+              -o feature@large_blocks=enabled \\
+              -o feature@lz4_compress=enabled \\
+              -o feature@spacemap_histogram=enabled \\
+              -O acltype=posixacl -O canmount=off -O compression=lz4 \\
+              -O devices=off -O normalization=formD -O atime=off -O xattr=sa \\
+              -O mountpoint=/boot -R #{MOUNTPOINT} \\
+              #{@boot_pool_name} #{@dev_boot}
+            ")
+          end
+
+          exec("zpool create -f -o ashift=#{ashift} -o autotrim=on \\
+            -O acltype=posixacl -O canmount=off -O compression=lz4 \\
+            -O dnodesize=auto -O normalization=formD -O atime=off \\
+            -O xattr=sa -O mountpoint=/ -R #{MOUNTPOINT} \\
+            #{@pool_name} #{@dev_root}
+          ")
+        end
+
+        def cache
+          system("mkswap -f #{@dev_swap}")
+          if @dev_log
+            exec("zpool add #{@pool_name} log #{@dev_log}")
+          end
+          if @dev_cache
+            exec("zpool add #{@pool_name} cache #{@dev_cache}")
+          end
+        end
+
+        def datasets
+          exec("zfs create -o canmount=off -o mountpoint=none #{@pool_name}/ROOT")
+          exec("zfs create -o canmount=off -o mountpoint=none #{@boot_pool_name}/BOOT") if @dev_boot
+
+          exec("zfs create -o canmount=noauto -o mountpoint=/ #{@pool_name}/ROOT/#{@n}")
+          exec("zfs create -o canmount=noauto -o mountpoint=/boot #{@boot_pool_name}/BOOT/#{@n}") if @dev_boot
+
+          exec("zfs create -o canmount=off #{@pool_name}/ROOT/#{@n}/usr")
+          exec("zfs create #{@pool_name}/ROOT/#{@n}/usr/src")
+          exec("zfs create -o canmount=off #{@pool_name}/ROOT/#{@n}/var")
+          exec("zfs create #{@pool_name}/ROOT/#{@n}/var/log")
+          exec("zfs create #{@pool_name}/ROOT/#{@n}/var/db")
+          exec("zfs create #{@pool_name}/ROOT/#{@n}/var/tmp")
+
+          exec("zfs create -o canmount=off -o mountpoint=/ #{@pool_name}/USERDATA")
+          exec("zfs create -o canmount=on -o mountpoint=/root #{@pool_name}/USERDATA/root")
+          if @user
+            exec("zfs create -o canmount=on -o mountpoint=/home/#{@user} #{@pool_name}/USERDATA/#{@user}")
+          end
+        end
+
+        def exec(cmd)
+          Getch::Command.new(cmd).run!
+        end
       end
     end
   end

data/lib/getch/filesystem/zfs/mount.rb

@@ -1,15 +1,11 @@
-require 'fileutils'
-
 module Getch
   module FileSystem
     module Zfs
-      class Mount < Getch::FileSystem::Zfs::Device
+      class Mount < Device
         def initialize
           super
-          @root_dir = MOUNTPOINT
-          @boot_dir = "#{@root_dir}/boot"
-          @boot_efi_dir = "#{@root_dir}/boot/efi"
-          @state = Getch::States.new()
+          @mount = Getch::FileSystem::Mount.new
+          @state = Getch::States.new
           @log = Getch::Log.new
         end
 
@@ -18,48 +14,30 @@ module Getch
           exec("zpool export -a")
           exec("rm -rf #{MOUNTPOINT}/*")
           exec("zpool import -N -R #{MOUNTPOINT} #{@pool_name}")
-          exec("zpool import -N -R #{MOUNTPOINT} #{@boot_pool_name}") if @dev_boot
-          mount_swap
+          exec("zpool import -f -N -R #{MOUNTPOINT} #{@boot_pool_name}") if @dev_boot
+          @mount.swap(@dev_swap)
           mount_root
           mount_boot
-          mount_boot_efi
+          @mount.esp(@dev_esp)
           exec("zfs mount -a")
           @state.mount
         end
 
         private
 
-        def mount_swap
-          if Helpers::grep?('/proc/swaps', /^\/dev/)
-            exec("swapoff #{@dev_swap}")
-          end
-
-          exec("swapon #{@dev_swap}")
-        end
-
         def mount_root
-          Helpers::mkdir(@root_dir)
-          exec("zfs mount #{@pool_name}/ROOT/gentoo")
-        end
-
-        def mount_boot_efi
-          return if ! @dev_boot_efi
-          Helpers::mkdir(@boot_efi_dir)
-          exec("mount #{@dev_boot_efi} #{@boot_efi_dir}")
+          Helpers::mkdir(MOUNTPOINT)
+          exec("zfs mount #{@pool_name}/ROOT/#{@n}")
         end
 
         def mount_boot
           return if ! @dev_boot
-          Helpers::mkdir(@boot_dir)
-          exec("zfs mount #{@boot_pool_name}/BOOT/gentoo")
+          exec("zfs mount #{@boot_pool_name}/BOOT/#{@n}")
         end
 
         def exec(cmd)
           @log.info("==> #{cmd}")
-          system(cmd)
-          unless $?.success?
-            raise "Error with #{cmd}"
-          end
+          Helpers::sys(cmd)
         end
       end
     end

data/lib/getch/filesystem/zfs/partition.rb

@@ -1,9 +1,11 @@
 module Getch
   module FileSystem
     module Zfs
-      class Partition < Getch::FileSystem::Zfs::Device
+      class Partition < Device
         def initialize
           super
+          @clean = Getch::FileSystem::Clean
+          @partition = Getch::FileSystem::Partition.new
           @state = Getch::States.new()
           @log = Getch::Log.new()
           run_partition
@@ -11,126 +13,43 @@ module Getch
 
         def run_partition
           return if STATES[:partition ]
-          clear_struct
-          cleaning
+          @clean.old_zpool
+          @clean.hdd(@disk)
+          @clean.external_disk(@disk, @boot_disk, @cache_disk, @home_disk)
+
           partition
-          zfs
+          cache
           @state.partition
         end
 
         private
 
-        def clear_struct
-          oldvg = `vgdisplay | grep #{@vg}`.chomp
-          oldzpool = `zpool status | grep pool:`.gsub(/pool: /, '').delete(' ').split("\n")
-          if oldzpool[0] != "" and $?.success?
-            oldzpool.each { |p| exec("zpool destroy #{p}") if p }
-          end
-          exec("vgremove -f #{@vg}") if oldvg != '' # remove older volume group
-          exec("pvremove -f #{@dev_root}") if oldvg != '' and File.exist? @dev_root # remove older volume group
-
-          exec("sgdisk -Z /dev/#{@disk}")
-          exec("wipefs -a /dev/#{@disk}")
-        end
-
-        # See https://wiki.archlinux.org/index.php/Solid_state_drive/Memory_cell_clearing
-        # for SSD
-        def cleaning
-          @bloc=`blockdev --getbsz /dev/#{@disk}`.chomp
-          puts
-          print "Cleaning data on #{@disk}, can be long, avoid this on Flash Memory (SSD,USB,...) ? (n,y) "
-          case gets.chomp
-          when /^y|^Y/
-            exec("dd if=/dev/urandom of=/dev/#{@disk} bs=#{@bloc} status=progress")
-          else
-            return
-          end
-        end
-
         def partition
-          mem=`awk '/MemTotal/ {print $2}' /proc/meminfo`.chomp + 'K'
-          if Helpers::efi?
-            exec("sgdisk -n1:1M:+260M -t1:EF00 /dev/#{@disk}")
-            exec("sgdisk -n2:0:+#{mem} -t2:8200 /dev/#{@disk}")
-            exec("sgdisk -n3:0:+0 -t3:BF00 /dev/#{@disk}")
+          if @efi
+            @partition.efi(@dev_esp)
+            @partition.boot(@dev_boot) if Getch::OPTIONS[:os] == 'void'
+            @partition.swap(@dev_swap) if !@cache_disk
+            @partition.root(@dev_root, "BF00") if @root_part != 1
           else
-            exec("sgdisk -n1:1MiB:+1MiB -t1:EF02 /dev/#{@disk}")
-            exec("sgdisk -n2:0:+2G -t2:BE00 /dev/#{@disk}") # boot pool GRUB
-            exec("sgdisk -n3:0:+#{mem} -t3:8200 /dev/#{@disk}")
-            exec("sgdisk -n4:0:+0 -t4:BF00 /dev/#{@disk}")
+            @partition.gpt(@dev_gpt)
+            # Boot pool for GRUB2
+            @partition.boot(@dev_boot)
+            @partition.swap(@dev_swap) if !@cache_disk
+            @partition.root(@dev_root, "BF00") if @root_part != 1
           end
         end
 
-        def zfs
-          ashift = case @bloc
-            when 8096
-              13
-            when 4096
-              12
-            else # 512
-              9
-            end
-
-          Helpers::mkdir(MOUNTPOINT)
-            
-          @log.debug("ashift found for #{@bloc} - #{ashift}")
-          if ! Helpers::efi? 
-            # https://openzfs.github.io/openzfs-docs/Getting%20Started/Ubuntu/Ubuntu%2020.04%20Root%20on%20ZFS.html
-            @log.info("Creating boot pool on #{@pool_name}")
-            exec("zpool create -f \\
-              -o ashift=#{ashift} -d \\
-              -o feature@async_destroy=enabled \\
-              -o feature@bookmarks=enabled \\
-              -o feature@embedded_data=enabled \\
-              -o feature@empty_bpobj=enabled \\
-              -o feature@enabled_txg=enabled \\
-              -o feature@extensible_dataset=enabled \\
-              -o feature@filesystem_limits=enabled \\
-              -o feature@hole_birth=enabled \\
-              -o feature@large_blocks=enabled \\
-              -o feature@lz4_compress=enabled \\
-              -o feature@spacemap_histogram=enabled \\
-              -O acltype=posixacl -O canmount=off -O compression=lz4 \\
-              -O devices=off -O normalization=formD -O atime=off -O xattr=sa \\
-              -O mountpoint=/boot -R #{MOUNTPOINT} \\
-              #{@boot_pool_name} #{@dev_boot}
-            ")
+        def cache
+          if @cache_disk
+            mem=`awk '/MemTotal/ {print $2}' /proc/meminfo`.chomp + 'K'
+            exec("sgdisk -n1:0:+#{mem} -t1:8200 /dev/#{@cache_disk}")
+            exec("sgdisk -n2:0:+4G -t2:BF07 /dev/#{@cache_disk}")
+            exec("sgdisk -n3:0:0 -t3:BF00 /dev/#{@cache_disk}")
           end
-
-          exec("zpool create -f -o ashift=#{ashift} \\
-            -O acltype=posixacl -O canmount=off -O compression=lz4 \\
-            -O dnodesize=auto -O normalization=formD -O atime=off \\
-            -O xattr=sa -O mountpoint=/ -R #{MOUNTPOINT} \\
-            #{@pool_name} #{@dev_root}
-          ")
-
-          add_datasets
-        end
-
-        def add_datasets
-          exec("zfs create -o canmount=off -o mountpoint=none #{@pool_name}/ROOT")
-          exec("zfs create -o canmount=off -o mountpoint=none #{@boot_pool_name}/BOOT") if @dev_boot
-
-          exec("zfs create -o canmount=noauto -o mountpoint=/ #{@pool_name}/ROOT/gentoo")
-          # set bootfs
-          #exec("zpool set bootfs=#{@pool_name}/ROOT/gentoo #{@pool_name}")
-          exec("zfs create -o canmount=noauto -o mountpoint=/boot #{@boot_pool_name}/BOOT/gentoo") if @dev_boot
-
-          exec("zfs create -o canmount=off #{@pool_name}/ROOT/gentoo/usr")
-          exec("zfs create #{@pool_name}/ROOT/gentoo/usr/src")
-          exec("zfs create -o canmount=off #{@pool_name}/ROOT/gentoo/var")
-          exec("zfs create #{@pool_name}/ROOT/gentoo/var/log")
-          exec("zfs create #{@pool_name}/ROOT/gentoo/var/db")
-          exec("zfs create #{@pool_name}/ROOT/gentoo/var/tmp")
-
-          exec("zfs create -o canmount=off -o mountpoint=/ #{@pool_name}/USERDATA")
-          exec("zfs create -o canmount=on -o mountpoint=/root #{@pool_name}/USERDATA/root")
-          exec("zfs create -o canmount=on -o mountpoint=/home/#{@user} #{@pool_name}/USERDATA/#{@user}") if @user
         end
 
-        # Follow https://wiki.archlinux.org/index.php/Partitioning
         # Partition_efi
-        # /boot/efi - EFI system partition - 260MB
+        # /efi      - EFI system partition - 260MB
         # /         - Root
 
         # Partition_bios