vagrant-smartos-zones 0.0.1.pre.21

data/files/zonegate/zonenat

@@ -0,0 +1,153 @@
+#!/usr/node/bin/node
+// -*- mode: javascript; tab-width: 4; indent-tabs-mode: 1; st-rulers: [80] -*-
+
+// require
+var exec = require('child_process').exec;
+
+// global
+var zone;
+var state;
+var adminIp;
+var privateIp;
+var zoneIp;
+
+function executeNat(mapCmd, rdrCmd, cb)
+{
+	exec(mapCmd, function (err, _stdout, _stderr) {
+		if (err) {
+			cb(err);
+		} else {
+			exec(rdrCmd, function (err, _stdout, _stderr) {
+				if (err) {
+					cb(err);
+				} else {
+					cb();
+				}
+			});			
+		}
+	});
+}
+
+function enableAdminNat(cb)
+{
+	var mapCmd = 'echo "map e1000g0 0/0 -> ' + adminIp + '/32" | ipnat -f -';
+	var rdrCmd = 'echo "rdr e1000g0 0/0 -> ' + zoneIp + '" | ipnat -f -';
+
+	executeNat(mapCmd, rdrCmd, cb);
+}
+
+function disableAdminNat(cb)
+{
+	var mapCmd = 'echo "map e1000g0 0/0 -> ' + adminIp + '/32" | ipnat -r -f -';
+	var rdrCmd = 'echo "rdr e1000g0 0/0 -> ' + zoneIp + '" | ipnat -r -f -';
+
+	executeNat(mapCmd, rdrCmd, cb);
+}
+
+function enablePrivateNat(cb)
+{
+	if (!privateIp) {
+		cb();
+		return;
+	}
+
+	var mapCmd = 'echo "map e1000g1 0/0 -> ' + privateIp + '/32" | ipnat -f -';
+	var rdrCmd = 'echo "rdr e1000g1 0/0 -> ' + zoneIp + '" | ipnat -f -';
+
+	executeNat(mapCmd, rdrCmd, cb);
+}
+
+function disablePrivateNat(cb)
+{
+	if (!privateIp) {
+		cb();
+		return;
+	}
+
+	var mapCmd = 'echo "map e1000g1 0/0 -> ' + privateIp + '/32" | ipnat -r -f -';
+	var rdrCmd = 'echo "rdr e1000g1 0/0 -> ' + zoneIp + '" | ipnat -r -f -';
+
+	executeNat(mapCmd, rdrCmd, cb);
+}
+
+function probeAddresses(cb)
+{
+	var finished = function () {
+		if (adminIp != null && privateIp != null && zoneIp != null) {
+			cb();
+		}
+	}
+
+	// probe adminIp
+	exec("ipadm show-addr e1000g0/ -p -o ADDR | sed 's/\\\/.*//'", function (err, stdout, stderr) {
+		if (err) {
+			adminIp = false;
+		} else {
+			adminIp = stdout.trim();
+		}
+		finished();
+	});
+
+	// probe privateIP
+	exec("ipadm show-addr e1000g1/ -p -o ADDR | sed 's/\\\/.*//'", function (err, stdout, stderr) {
+		if (err) {
+			privateIp = false;
+		} else {
+			privateIp = stdout.trim();
+		}
+		finished();
+	});
+
+	// probe zoneIP
+	exec('vmadm get ' + zone + ' | json nics[0].ip', function (err, stdout, stderr) {
+		if (err) {
+			zoneIp = false;
+		} else {
+			zoneIp = stdout.trim();
+		}
+		finished();
+	});
+}
+
+function main()
+{
+	var data = process.argv[2];
+
+	if (!data) {
+		process.exit(1);
+	}
+
+	data = data.trim().split(':');
+
+	zone  = data[0];
+	state = data[1];
+
+	probeAddresses(function () {
+		if (state == 'up') {
+			enableAdminNat(function (err) {
+				if (err) {
+					console.log('error enabling admin nat: ' + err.message);
+				}
+			});
+			enablePrivateNat(function (err) {
+				if (err) {
+					console.log('error enabling private nat: ' + err.message);
+				}
+			});
+		} else {
+			disableAdminNat(function (err) {
+				if (err) {
+					console.log('error disabling admin nat: ' + err.message);
+				}
+			});
+			disablePrivateNat(function (err) {
+				if (err) {
+					console.log('error disabling private nat: ' + err.message);
+				}
+			});
+		}
+	});
+
+}
+
+main();

data/lib/vagrant/smartos/locales/en.yml

@@ -0,0 +1,41 @@
+en:
+  vagrant:
+    smartos:
+      zones:
+        commands:
+          zones:
+            list: |-
+              name                      state      uuid 
+              %{zones}
+            create: |-
+              Created zone with name %{name}
+                uuid: %{uuid}
+                state: %{state}
+                brand: %{brand}
+                image: %{image}
+            destroyed: |-
+              Zone destroyed %{name} %{uuid}
+            show: |-
+              name:      %{name}
+              state:     %{state}
+              uuid:      %{uuid}
+              brand:     %{brand}
+              image:     %{image}
+            snapshot:
+              list: |-
+                name         created at              used zone
+                %{snapshots}
+              create: |-
+                Creating snapshot with name %{name}
+              destroy: |-
+                Destroying snapshot with name %{name}
+              rollback: |-
+                Rolling back zone %{zonename} to snapshot %{name}
+            start: |-
+              Started zone %{name} %{uuid}
+            stop: |-
+              Stopped zone %{name} %{uuid}
+        warning:
+          zone_not_found: |-
+            Unable to find zone with name %{name}
+

data/lib/vagrant/smartos/zones/action/configure_zone_synced_folders.rb

@@ -0,0 +1,53 @@
+require 'vagrant/smartos/zones/models/zone'
+require 'vagrant/smartos/zones/util/zone_user'
+require 'vagrant/smartos/zones/util/zone_group'
+require 'vagrant/smartos/zones/action/helper'
+
+# rubocop:disable all
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        class ConfigureZoneSyncedFolders
+          include Helper
+
+          def initialize(app, env)
+            @app = app
+            @env = env
+          end
+
+          def call(env)
+            app.call(env)
+
+            if zones_supported?
+              zone = Vagrant::Smartos::Zones::Models::Zone.find(machine, machine.config.zone.name)
+              machine.config.zone.synced_folders.each do |folder|
+                configure_synced_folder(zone, folder)
+              end
+            end
+          end
+
+          def configure_synced_folder(zone, folder)
+            host_path, guest_path, args = folder
+            path = "/zones/#{zone.uuid}/root#{guest_path}"
+            type = args.delete(:type) { |el| :rsync }
+
+            username = args[:owner] || 'vagrant'
+            groupname = args[:group] || 'vagrant'
+            user = Vagrant::Smartos::Zones::Util::ZoneUser.new(machine, zone).find(username)
+            group = Vagrant::Smartos::Zones::Util::ZoneGroup.new(machine, zone).find(groupname)
+
+            machine.config.vm.synced_folders[path] = {
+              disabled: false,
+              guestpath: path,
+              hostpath: host_path,
+              type: type.to_sym,
+              owner: user.uid,
+              group: group.gid
+            }.merge(args || {})
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/create_gz_vnic.rb

@@ -0,0 +1,28 @@
+require 'vagrant/smartos/zones/action/helper'
+
+# This action installs a vnic in the global zone, used by
+# local zones. The executed code needs to happen after
+# networks have been configured in the global zone.
+#
+# Zone creation is hooked onto this.
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        class CreateGZVnic
+          include Helper
+
+          def initialize(app, env)
+            @app = app
+            @env = env
+          end
+
+          def call(env)
+            app.call(env)
+            guest.capability(:create_gz_vnic) if zones_supported?
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/forward_gz_ports.rb

@@ -0,0 +1,46 @@
+require 'vagrant/smartos/zones/action/helper'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        class ForwardGZPorts
+          include VagrantPlugins::ProviderVirtualBox::Util::CompileForwardedPorts
+          include Helper
+
+          def initialize(app, env)
+            @app = app
+            @env = env
+          end
+
+          # This middleware sets up a VirtualBox port forward to 2222 in the global zone.
+          # ipnat in the global zone is configured to route this to 22 in the global zone,
+          # even when all other ports are forwarded to a zone.
+          def call(env)
+            configure_gz_port_forward if zone_configured?
+
+            app.call(env)
+          end
+
+          def configure_gz_port_forward
+            machine.ui.info 'Configuring a port forward to talk to the Global Zone'
+            machine.config.vm.network :forwarded_port, port_forward_options
+          end
+
+          def network
+            machine.config.vm.networks.first
+          end
+
+          def port_forward_options
+            scoped_hash_override(network[1], :virtualbox).merge(
+              id: 'gz_ssh',
+              guest: machine.config.global_zone.ssh_port,
+              host: machine.config.global_zone.forwarded_ssh_port,
+              auto_correct: true
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/helper.rb

@@ -0,0 +1,33 @@
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        module Helper
+          attr_reader :app, :env
+
+          private
+
+          def machine
+            @machine ||= env[:machine]
+          end
+
+          def guest
+            @guest ||= machine.guest
+          end
+
+          def zones_supported?
+            @zones_supported ||= guest.capability?(:imgadm_import)
+          end
+
+          def zone_configured?
+            !!machine.config.zone.name
+          end
+
+          def single_zone_mode?
+            !!machine.config.zone.name
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/imgadm_import.rb

@@ -0,0 +1,24 @@
+require 'vagrant/smartos/zones/action/helper'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        class ImgadmImport
+          include Helper
+
+          def initialize(app, env)
+            @app = app
+            @env = env
+          end
+
+          def call(env)
+            app.call(env)
+            env[:ui].info "Checking if machine supports zones: #{zones_supported? ? 'yes' : 'no'}"
+            guest.capability(:imgadm_import) if zones_supported?
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/virtualbox/platform_iso.rb

@@ -0,0 +1,57 @@
+require 'vagrant/smartos/zones/action/helper'
+require 'vagrant/smartos/zones/util/platform_images'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        module VirtualBox
+          class PlatformISO
+            include Helper
+
+            attr_accessor :app, :env, :machine
+            def initialize(app, _env)
+              @app = app
+            end
+
+            def call(env)
+              @env = env
+              @machine = env[:machine]
+
+              if requires_platform_iso?
+                env[:ui].warn 'Remapping platform ISO'
+                machine.provider_config.customizations << remove_dvddrive
+                machine.provider_config.customizations << add_platform_image
+              end
+
+              @app.call(env)
+            end
+
+            private
+
+            def requires_platform_iso?
+              machine.config.global_zone.platform_image &&
+                machine.config.vm.communicator == :smartos
+            end
+
+            def remove_dvddrive
+              ['pre-boot',
+               ['storageattach', :id, '--storagectl', 'IDE Controller', '--port', 1, '--device',
+                0, '--medium', 'none']]
+            end
+
+            def add_platform_image
+              ['pre-boot', ['storageattach', :id, '--storagectl', 'IDE Controller',
+                            '--port', 1, '--device', 0, '--type', 'dvddrive',
+                            '--medium', platform_image.to_s]]
+            end
+
+            def platform_image
+              Util::PlatformImages.new(env, machine).get_platform_image(machine.config.global_zone.platform_image)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/zone/create.rb

@@ -0,0 +1,25 @@
+require 'vagrant/smartos/zones/action/helper'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        module Zone
+          class Create
+            include Helper
+
+            def initialize(app, env)
+              @app = app
+              @env = env
+            end
+
+            def call(env)
+              app.call(env)
+              guest.capability(:zone__create) if zones_supported?
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/zone/start.rb

@@ -0,0 +1,25 @@
+require 'vagrant/smartos/zones/action/helper'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        module Zone
+          class Start
+            include Helper
+
+            def initialize(app, env)
+              @app = app
+              @env = env
+            end
+
+            def call(env)
+              app.call(env)
+              guest.capability(:zone__start) if zones_supported?
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/zone/stop.rb

@@ -0,0 +1,25 @@
+require 'vagrant/smartos/zones/action/helper'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        module Zone
+          class Stop
+            include Helper
+
+            def initialize(app, env)
+              @app = app
+              @env = env
+            end
+
+            def call(env)
+              guest.capability(:zone__stop) if zones_supported?
+              app.call(env)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/zone_gate/enable.rb

@@ -0,0 +1,26 @@
+require 'vagrant/smartos/zones/action/helper'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        module ZoneGate
+          class Enable
+            include Helper
+
+            def initialize(app, env)
+              @app = app
+              @env = env
+            end
+
+            def call(env)
+              app.call(env)
+              return unless guest.capability?(:zone_gate_enable)
+              guest.capability(:zone_gate_enable) if single_zone_mode?
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action/zone_gate/install.rb

@@ -0,0 +1,26 @@
+require 'vagrant/smartos/zones/action/helper'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        module ZoneGate
+          class Install
+            include Helper
+
+            def initialize(app, env)
+              @app = app
+              @env = env
+            end
+
+            def call(env)
+              app.call(env)
+              return unless guest.capability?(:zone_gate_install)
+              guest.capability(:zone_gate_install) if zones_supported?
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/action.rb

@@ -0,0 +1,75 @@
+module Vagrant
+  module Smartos
+    module Zones
+      module Action
+        require_relative 'action/configure_zone_synced_folders'
+        require_relative 'action/create_gz_vnic'
+        require_relative 'action/forward_gz_ports'
+        require_relative 'action/imgadm_import'
+        require_relative 'action/zone/create'
+        require_relative 'action/zone/start'
+        require_relative 'action/zone/stop'
+        require_relative 'action/zone_gate/enable'
+        require_relative 'action/zone_gate/install'
+        require_relative 'action/virtualbox/platform_iso'
+
+        class << self
+          def configure_zone_synced_folders
+            @configure_zone_synced_folders ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::ConfigureZoneSyncedFolders
+            end
+          end
+
+          def create_gz_vnic
+            @configure_zone_synced_folders ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::CreateGZVnic
+            end
+          end
+
+          def enable_zone_gate
+            @enable_zone_gate ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::ZoneGate::Enable
+            end
+          end
+
+          def forward_gz_ports
+            @forward_gz_ports ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::ForwardGZPorts
+            end
+          end
+
+          def install_zone_gate
+            @install_zone_gate ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::ZoneGate::Install
+            end
+          end
+
+          def virtualbox_platform_iso
+            @virtualbox_platform_iso ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::VirtualBox::PlatformISO
+            end
+          end
+
+          def zone_create
+            @zone_create ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::Zone::Create
+              b.use Vagrant::Smartos::Zones::Action::ImgadmImport
+            end
+          end
+
+          def zone_start
+            @zone_start ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::Zone::Start
+            end
+          end
+
+          def zone_stop
+            @zone_stop ||= ::Vagrant::Action::Builder.new.tap do |b|
+              b.use Vagrant::Smartos::Zones::Action::Zone::Stop
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/base.rb

@@ -0,0 +1,29 @@
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        class Base
+          attr_reader :machine
+
+          def self.cap_method(method_name)
+            self.class.send :define_method, method_name do |machine|
+              new(machine).execute
+            end
+          end
+
+          def initialize(machine)
+            @machine = machine
+          end
+
+          def ui
+            machine.ui
+          end
+
+          def sudo
+            machine.config.smartos.suexec_cmd
+          end
+        end
+      end
+    end
+  end
+end