vagrant-bhyve 0.1.0

data/lib/vagrant-bhyve/action/boot.rb

@@ -0,0 +1,26 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class Boot
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::boot")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine 	= env[:machine]
+	  @driver	= @machine.provider.driver
+
+          env[:ui].detail I18n.t('vagrant_bhyve.action.vm.boot.booting')
+	  @driver.boot(@machine, env[:ui])
+
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/cleanup.rb

@@ -0,0 +1,23 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class Cleanup
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::cleanup")
+	  @app = app
+	end
+
+	def call(env)
+	  env[:ui].info I18n.t('vagrant_bhyve.action.vm.halt.cleaning_up')
+	  env[:machine].provider.driver.cleanup
+
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/create_bridge.rb

@@ -0,0 +1,33 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class CreateBridge
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::create_bridge")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine 	= env[:machine]
+	  @ui		= env[:ui]
+	  @driver	= @machine.provider.driver
+	  
+          @ui.info I18n.t('vagrant.actions.vm.boot.booting')
+	  @ui.detail I18n.t('vagrant_bhyve.action.vm.boot.setup_nat')
+
+	  bridge_list 	= %w(vagrant_bhyve_default_bridge)
+	  # The bridge name is used as created bridge device's description
+	  bridge_list.each do |bridge|
+	    @driver.create_network_device(bridge, "bridge")
+	    @driver.enable_nat(bridge, @ui)
+	  end
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/create_tap.rb

@@ -0,0 +1,31 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class CreateTap
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::create_tap")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine	= env[:machine]
+	  @driver	= @machine.provider.driver
+
+	  env[:ui].detail I18n.t('vagrant_bhyve.action.vm.boot.create_tap')
+	  vm_name	= @driver.get_attr('vm_name')
+	  tap_name	= "vagrant_bhyve_#{vm_name}"
+	  tap_list 	= [tap_name]
+	  # The switch name is used as created bridge device's description
+	  tap_list.each do |tap|
+	    @driver.create_network_device(tap, "tap")
+	  end
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/destroy.rb

@@ -0,0 +1,23 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class Destroy
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::destroy")
+	  @app = app
+	end
+
+	def call(env)
+	  env[:ui].info I18n.t('vagrant_bhyve.action.vm.destroying')
+	  env[:machine].provider.driver.destroy
+
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/forward_ports.rb

@@ -0,0 +1,53 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class ForwardPorts
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::forward_ports")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine              = env[:machine]
+	  @driver               = @machine.provider.driver
+
+	  env[:ui].info I18n.t('vagrant_bhyve.action.vm.forward_ports')
+
+	  env[:forwarded_ports]  = compile_forwarded_ports(@machine.config)
+	  tap_device            = @driver.get_attr('tap')
+	  gateway               = @driver.get_attr('gateway')
+	  env[:forwarded_ports].each do |item|
+	    forward_information = {
+	      adapter: item[:adapter] || gateway,
+	      guest_port: item[:guest],
+	      host_port: item[:host]
+	    }
+	    @driver.forward_port(forward_information, tap_device)
+	  end
+	  @app.call(env)
+	end
+
+	private
+
+	def compile_forwarded_ports(config)
+	  mappings = {}
+	  config.vm.networks.each do |type, options|
+	    next if options[:disabled]
+
+	    if type == :forwarded_port && options[:id] != 'ssh'
+	      if options.fetch(:host_ip, '').to_s.strip.empty?
+		options.delete(:host_ip)
+	      end
+	      mappings[options[:host]] = options
+	    end
+	  end
+	  mappings.values
+
+	end
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/import.rb

@@ -0,0 +1,32 @@
+require "log4r"
+require "securerandom"
+require "digest/md5"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class Import
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::import")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine 	= env[:machine]
+	  @driver	= @machine.provider.driver
+
+	  env[:ui].info I18n.t('vagrant_bhyve.action.vm.import_box')
+	  @machine.id 	= SecureRandom.uuid
+	  vm_name   = @machine.id.gsub('-', '')[0..30]
+	  mac       = @driver.get_mac_address(vm_name)
+	  @driver.store_attr('vm_name', vm_name)
+	  @driver.store_attr('mac', mac)
+	  @driver.import(@machine, env[:ui])
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/load.rb

@@ -0,0 +1,27 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class Load
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::load")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine 	= env[:machine]
+	  @driver	= @machine.provider.driver
+	  firmware	= @driver.get_attr('firmware')
+
+          env[:ui].detail I18n.t('vagrant_bhyve.action.vm.boot.load_kernel')
+	  @driver.load(@machine, env[:ui]) if firmware == 'bios'
+
+	  @app.call(env)
+	end
+	
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/prepare_nfs_settings.rb

@@ -0,0 +1,50 @@
+require "log4r"
+require "securerandom"
+require "digest/md5"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class PrepareNFSSettings
+	include Vagrant::Action::Builtin::MixinSyncedFolders
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::prepare_nfs_settings")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine	= env[:machine]
+	  @driver	= @machine.provider.driver
+	  @app.call(env)
+
+	  if using_nfs?
+	    tap_device	= @driver.get_attr('tap')
+	    #host_ip	= @driver.get_ip_address(nil, :host)
+	    guest_ip 	= @driver.get_ip_address(tap_device, :guest)
+	    host_ip	= read_host_ip(guest_ip)
+	    env[:nfs_machine_ip]	= guest_ip
+	    env[:nfs_host_ip]		= host_ip
+	  end
+	end
+
+	def using_nfs?
+	  !!synced_folders(@machine)[:nfs]
+	end
+
+	# Ruby way to get host ip
+	def read_host_ip(ip)
+	  UDPSocket.open do |s|
+	    if(ip.kind_of?(Array))
+	      s.connect(ip.last, 1)
+	    else
+	      s.connect(ip, 1)
+	    end
+	    s.addr.last
+	  end
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/prepare_nfs_valid_ids.rb

@@ -0,0 +1,23 @@
+require "log4r"
+require "securerandom"
+require "digest/md5"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class PrepareNFSValidIds
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::prepare_nfs_valid_ids")
+	  @app = app
+	end
+
+	def call(env)
+	  env[:nfs_valid_ids] = [env[:machine].id]
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/setup.rb

@@ -0,0 +1,27 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class Setup
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::setup")
+	  @app = app
+	end
+
+	def call(env)
+	  @driver	= env[:machine].provider.driver
+	  env[:ui].info I18n.t('vagrant_bhyve.action.vm.setup_environment')
+	  @driver.check_bhyve_support
+	  module_list 	= %w(vmm nmdm if_bridge if_tap pf)
+	  for kernel_module in module_list
+	    @driver.load_module(kernel_module)
+	  end
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/shutdown.rb

@@ -0,0 +1,26 @@
+require "log4r"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class Shutdown
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::shutdown")
+	  @app = app
+	end
+
+	def call(env)
+	  @machine 	= env[:machine]
+	  @ui		= env[:ui]
+	  @driver	= @machine.provider.driver
+
+	  @ui.info I18n.t('vagrant_bhyve.action.vm.halt.shutting_down')
+	  @driver.shutdown(@ui)
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/action/wait_until_up.rb

@@ -0,0 +1,48 @@
+require "log4r"
+require "vagrant/util/retryable"
+
+module VagrantPlugins
+  module ProviderBhyve
+    module Action
+      class WaitUntilUP
+	include Vagrant::Util::Retryable
+
+	def initialize(app, env)
+	  @logger = Log4r::Logger.new("vagrant_bhyve::action::wait_until_up")
+	  @app = app
+	end
+
+	def call(env)
+	  @driver = env[:machine].provider.driver
+	  env[:ui].info I18n.t('vagrant_bhyve.action.vm.boot.wait_until_up')
+
+	  vm_name = @driver.get_attr('vm_name')
+	  # Check whether ip is assigned
+	  env[:uncleaned] = false
+	  while !env[:uncleaned]
+	    sleep 1
+	    env[:uncleaned] = true if @driver.state(vm_name) == :uncleaned
+	    if @driver.ip_ready?
+	      sleep 2
+	      env[:uncleaned] = true if @driver.state(vm_name) == :uncleaned
+	      break
+	    end
+	  end
+	  
+	  # Check whether we have ssh access
+	  while !env[:uncleaned]
+	    sleep 1
+	    env[:uncleaned] = true if @driver.state(vm_name) == :uncleaned
+	    if @driver.ssh_ready?(env[:machine].provider.ssh_info)
+	      sleep 2
+	      env[:uncleaned] = true if @driver.state(vm_name) == :uncleaned
+	      break
+	    end
+	  end
+	  @app.call(env)
+	end
+
+      end
+    end
+  end
+end

data/lib/vagrant-bhyve/config.rb

@@ -0,0 +1,75 @@
+require "vagrant"
+
+module VagrantPlugins
+  module ProviderBhyve
+    class Config < Vagrant.plugin('2', :config)
+      
+      # Guest like CentOS-6 requires a customized grub config file
+      #attr_accessor :grub_config_file
+
+      # Some arguments required by grub-bhyve
+      #attr_accessor :grub_run_partition
+      #attr_accessor :grub_run_dir
+      
+      # Specify the number of virtual CPUs.
+      attr_accessor :cpus
+      # Specify the size of guest physical memory.
+      attr_accessor :memory
+      # Specify virtual devices will be attached to bhyve's emulated
+      # PCI bus. Network interface and disk will both attched as this kind
+      # of devices.
+      attr_accessor :pcis
+      # Specify console device which will be attached to the VM
+      attr_accessor :lpc
+      attr_accessor :hostbridge
+      # Addition storage
+      attr_accessor :disks
+      attr_accessor :cdroms
+
+      def initialize
+	@cpus			= UNSET_VALUE
+	@memory			= UNSET_VALUE
+	@pcis			= UNSET_VALUE
+	@lpc			= UNSET_VALUE
+	@hostbridge		= UNSET_VALUE
+	@disks			= []
+	@cdroms			= []
+	#@grub_config_file	= ''
+	#@grub_run_partition	= ''
+	#@grub_run_dir 		= ''
+      end
+
+      def storage(options={})
+	if options[:device] == :cdrom
+	  _handle_cdrom_storage(options)
+	elsif options[:device] == :disk
+	  _handle_disk_storage(options)
+	end
+      end
+
+      def _handle_disk_storage(options={})
+	cdrom = {
+	  path: options[:path]
+	}
+	@cdroms << cdrom
+      end
+
+      def _handle_cdrom_storage(options={})
+	options = {
+	  path: nil,
+	  name: nil,
+	  size: "20G",
+	  format: "raw",
+	}.merge(options)
+
+	disk = {
+	  path: options[:path],
+	  size: options[:size],
+	}
+
+	@disks << disk
+      end
+
+    end
+  end
+end