specinfra-cisco 2.40.2

data/lib/specinfra/backend/powershell/support/check_file_access_rules.ps1

@@ -0,0 +1,8 @@
+function CheckFileAccessRules
+{
+  param($path, $identity, $rules)
+  
+  $accessRules = @((Get-Acl $path).access | Where-Object {$_.AccessControlType -eq 'Allow' -and $_.IdentityReference -eq $identity })
+  $match = @($accessRules | Where-Object {($_.FileSystemRights.ToString().Split(',') | % {$_.trim()} | ? {$rules -contains $_}) -ne $null})
+  $match.count -gt 0
+}

data/lib/specinfra/backend/powershell/support/crop_text.ps1

@@ -0,0 +1,11 @@
+function CropText
+{
+  param($text, $fromPattern, $toPattern)
+  
+  $from, $to = ([regex]::matches($text, $fromPattern)), ([regex]::matches($text, $toPattern))
+  if ($from.count -gt 0 -and $to.count -gt 0) {
+    $text.substring($from[0].index, $to[0].index + $to[0].length - $from[0].index)
+  } else {
+    ""
+  }
+}

data/lib/specinfra/backend/powershell/support/find_group.ps1

@@ -0,0 +1,8 @@
+function FindGroup
+{
+  param($groupName, $domain)
+  if ($domain -eq $null) {$selectionCriteria = " and LocalAccount = true"}
+  else {$selectionCriteria = " and Domain = '$domain'"}
+
+  Get-WmiObject Win32_Group -filter "Name = '$groupName' $selectionCriteria"
+}

data/lib/specinfra/backend/powershell/support/find_iis_component.ps1

@@ -0,0 +1,90 @@
+function FindIISWebsite 
+{
+    param($name)
+
+    Import-Module WebAdministration
+
+    Try {
+        Get-Item "IIS:\Sites\$name" -Erroraction silentlycontinue
+    }
+    Catch [System.IO.FileNotFoundException] {
+        Get-Item "IIS:\Sites\$name" -Erroraction silentlycontinue
+    }
+}
+
+function FindIISAppPool
+{
+    param($name)
+    
+    Import-Module WebAdministration
+
+    Try {
+        Get-Item "IIS:\AppPools\$name" -Erroraction silentlycontinue
+    }
+    Catch [System.IO.FileNotFoundException] {
+        Get-Item "IIS:\AppPools\$name" -Erroraction silentlycontinue
+
+    }
+}
+
+function FindSiteBindings
+{
+    param($name, $protocol, $hostHeader, $port, $ipAddress)
+
+    Import-Module WebAdministration
+    Try {
+        Get-WebBinding -Name $name -Protocol $protocol -HostHeader $hostHeader -Port $port -IPAddress $ipAddress
+    }
+    Catch [System.IO.FileNotFoundException] {
+        Get-WebBinding -Name $name -Protocol $protocol -HostHeader $hostHeader -Port $port -IPAddress $ipAddress
+    }
+}
+
+function FindSiteVirtualDir
+{
+    param($name, $vdir, $path)
+
+    Import-Module WebAdministration
+
+    $webVirtDirPath = [string]::Format('IIS:\Sites\{0}\{1}',$name, $vdir);
+    if (Test-Path $webVirtDirPath)
+    {
+        if ([string]::IsNullOrEmpty($path))
+        {
+            $true
+        }
+        else
+        {
+            (Get-Item $webVirtDirPath).physicalPath -eq $path
+        }
+    }
+    else
+    {
+        $false
+    }
+}
+
+function FindSiteApplication
+{
+    param($name, $app, $pool, $physicalPath)
+
+    Import-Module WebAdministration
+
+    $path = "IIS:\Sites\${name}\${app}"
+    $result = $false
+    if (Test-Path $path)
+    {
+        $result = $true
+        if ([string]::IsNullOrEmpty($pool) -eq $false)
+        {
+            $result = $result -and (Get-Item $path).applicationPool -eq $pool
+        }
+
+        if ([string]::IsNullOrEmpty($physicalPath) -eq $false)
+        {
+            $result = $result -and (Get-Item $path).physicalPath -eq $physicalPath
+        }
+    }
+
+    $result
+}

data/lib/specinfra/backend/powershell/support/find_installed_application.ps1

@@ -0,0 +1,37 @@
+function FindInstalledApplication
+{
+  param($appName, $appVersion)
+    
+  if ((Get-WmiObject win32_operatingsystem).OSArchitecture -notmatch '64')  
+  { 
+    $keys= (Get-ItemProperty 'HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\*')
+    $possible_path= 'HKCU:\Software\Microsoft\Windows\CurrentVersion\Uninstall\*'
+    if (Test-Path $possible_path)
+    {
+      $keys+= (Get-ItemProperty $possible_path)
+    }
+  }  
+    else  
+  { 
+    $keys = (Get-ItemProperty 'HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\*','HKLM:\SOFTWARE\Wow6432Node\Microsoft\Windows\CurrentVersion\Uninstall\*')
+    $possible_path= 'HKCU:\Software\Microsoft\Windows\CurrentVersion\Uninstall\*'
+    if (Test-Path $possible_path)
+    {
+      $keys+= (Get-ItemProperty $possible_path)
+    }
+    $possible_path= 'HKCU:\Software\Wow6432Node\Microsoft\Windows\CurrentVersion\Uninstall\*'
+    if (Test-Path $possible_path)
+    {
+      $keys+= (Get-ItemProperty $possible_path)
+    }
+  }
+
+  if ($appVersion -eq $null) { 
+    @($keys | Where-Object {$_.DisplayName -like $appName -or $_.PSChildName -like $appName}).Length -gt 0
+  }
+  else{
+    @($keys | Where-Object {$_.DisplayName -like $appName -or $_.PSChildName -like $appName  } | Where-Object {$_.DisplayVersion -eq $appVersion} ).Length -gt 0
+  }
+
+}
+

data/lib/specinfra/backend/powershell/support/find_installed_hot_fix.ps1

@@ -0,0 +1,18 @@
+function FindInstalledHotFix
+{
+  param($description, $hotFixId)
+
+Write-Host "Num Args:" $args.Length;
+foreach ($arg in $args)
+{
+  Write-Host "Arg: $arg";
+}
+
+  Write-Host $description
+  Write-Host $hotFixId
+
+  $keys= (Get-WmiObject -Class WIN32_QuickFixEngineering)
+  
+  @($keys | Where-Object {$_.HotFixID -like $hotFixId}).Length -gt 0
+
+}

data/lib/specinfra/backend/powershell/support/find_scheduled_task.ps1

@@ -0,0 +1,7 @@
+function FindScheduledTask
+{
+  param($name)
+
+  $task = schtasks /query /v /fo csv /TN "$name" | ConvertFrom-CSV
+  return $task
+}

data/lib/specinfra/backend/powershell/support/find_service.ps1

@@ -0,0 +1,5 @@
+function FindService
+{
+  param($name)
+  Get-WmiObject Win32_Service | Where-Object {$_.Name -eq $name -or $_.DisplayName -eq $name}
+}

data/lib/specinfra/backend/powershell/support/find_user.ps1

@@ -0,0 +1,8 @@
+function FindUser
+{
+  param($userName, $domain)
+  if ($domain -eq $null) {$selectionCriteria = " and LocalAccount = true"}
+  else {$selectionCriteria = " and Domain = '$domain'"}
+
+  Get-WmiObject Win32_UserAccount -filter "Name = '$userName' $selectionCriteria"
+}

data/lib/specinfra/backend/powershell/support/find_usergroup.ps1

@@ -0,0 +1,9 @@
+function FindUserGroup
+{
+  param($userName, $groupName, $userDomain, $groupDomain)
+  $user = FindUser -userName $userName -domain $userDomain
+  $group = FindGroup -groupName $groupName -domain $groupDomain
+  if ($user -and $group) {
+    Get-WmiObject Win32_GroupUser -filter ("GroupComponent = 'Win32_Group.Domain=`"" + $group.domain + "`",Name=`"" + $group.name + "`"' and PartComponent = 'Win32_UserAccount.Domain=`"" + $user.domain + "`",Name=`"" + $user.name + "`"'")
+  }
+}

data/lib/specinfra/backend/powershell/support/is_port_listening.ps1

@@ -0,0 +1,14 @@
+function IsPortListening
+{
+  param($portNumber, $protocol)
+  $netstatOutput = netstat -an | Out-String
+  $networkIPs = (Get-WmiObject Win32_NetworkAdapterConfiguration | ? {$_.IPEnabled}) | %{ $_.IPAddress[0] }
+  [array] $networkIPs += "0.0.0.0"
+  foreach ($ipaddress in $networkIPs)
+  {
+    $matchExpression = ("$ipaddress" + ":" + $portNumber)
+    if ($protocol) { $matchExpression = ($protocol.toUpper() + "\s+$matchExpression") }
+    if ($netstatOutput -match $matchExpression) { return $true }
+  }
+  $false
+}

data/lib/specinfra/backend/powershell/support/is_remote_port_listening.ps1

@@ -0,0 +1,51 @@
+function IsRemotePortListening{
+	param(
+		[string]$hostname,
+		[string]$proto="tcp",
+		[int]$port,
+		[int]$timeout)
+
+
+	If ($proto -eq "tcp") {     
+		$tcpobject = new-Object system.Net.Sockets.TcpClient                   
+		$connect = $tcpobject.BeginConnect($hostname,$port,$null,$null)   
+		$wait = $connect.AsyncWaitHandle.WaitOne($timeout * 1000,$false)   
+		#If timeout   
+		If(!$wait) {   
+				$tcpobject.Close()   
+				return $false
+			} 
+			else{    
+				$result = $tcpobject.Connected
+				$tcpobject.Close()
+				return $result
+			}      
+		}       
+		elseif ($proto -eq "udp") {                               
+			$udpobject = new-Object system.Net.Sockets.Udpclient 
+			$udpobject.client.ReceiveTimeout = $timeout  * 1000
+			$udpobject.Connect($hostname,$port)  
+			$a = new-object system.text.asciiencoding  
+			$byte = $a.GetBytes("$(Get-Date)")  
+			[void]$udpobject.Send($byte,$byte.length)  
+			$remoteendpoint = New-Object system.net.ipendpoint([system.net.ipaddress]::Any,0)  
+			try{
+				#Blocks until a message returns on this socket from a remote host.  
+				$receivebytes = $udpobject.Receive([ref]$remoteendpoint)  
+				[string]$returndata = $a.GetString($receivebytes) 
+				If ($returndata) { 
+					$udpobject.close()    
+					return $true
+				} 
+				else{
+					return $false
+				}                        
+			}
+			catch{
+				return $false
+			}
+		}         
+		else{
+			throw "Protocol ${proto} not supported"
+		}                          
+	}   

data/lib/specinfra/backend/powershell/support/list_windows_features.ps1

@@ -0,0 +1,68 @@
+function ListWindowsFeatures
+{    
+
+param(
+        [string]$feature,
+        [string]$provider="dism"
+         )
+
+    $cachepath =  "${env:temp}/ListWindowsFeatures-${provider}.xml"
+    $maxAge = 2
+
+    $cache =  Get-Item $cachepath -erroraction SilentlyContinue
+
+    if($cache -ne $null -and ((get-date) - $cache.LastWriteTime).minutes -lt $maxage){
+        $features = Import-Clixml $cachepath |  Select *| Where-Object {(($_.name -like $feature) -or ($_.displayName -like $feature)) -and (($_.installed -eq $true) -or ($_.state -eq "Enabled"))}
+        return $features
+    }
+    else{
+
+        switch($provider)
+        {
+            "dism" { return features_dism | Select * | Where-Object {($_.name -eq $feature) -and ($_.state -eq "Enabled")}  }
+            "powershell" { return features_powershell | Select * | Where-Object {(($_.name -like $feature) -or ($_.displayName -like $feature)) -and ($_.installed -eq $true)} }
+            default {throw "Unsupported provider"}
+        }
+
+    }
+}
+
+function features_dism{
+      try
+        {
+            $out = DISM /Online /Get-Features /Format:List | Where-Object {$_}     
+
+            if($LASTEXITCODE -ne 0)
+            {
+                Write-Error $out
+                Break
+            }
+
+            $f = $out[4..($out.length-2)]
+            $features = for($i=0; $i -lt $f.length;$i+=2)
+            {
+                $tmp = $f[$i],$f[$i+1] -replace '^([^:]+:\s)'
+                
+                New-Object PSObject -Property @{
+                    Name = $tmp[0]
+                    State = $tmp[1]
+                }
+            }
+
+            $features | Export-Clixml $cachepath
+
+            return $features
+        }
+        catch
+        {
+            Throw
+        }
+}
+
+function features_powershell{
+    $ProgressPreference = "SilentlyContinue"
+     import-module servermanager
+     $features = Get-WindowsFeature
+     $features | Export-Clixml $cachepath
+     return Get-WindowsFeature 
+}

data/lib/specinfra/backend/shell_script.rb

@@ -0,0 +1,29 @@
+require 'singleton'
+
+module Specinfra
+  module Backend
+    class ShellScript < Base
+      def initialize(config = {})
+        super
+
+        @lines = [ "#!/bin/sh", "" ]
+        ObjectSpace.define_finalizer(self, Writer.new(@lines))
+      end
+
+      def run_command(cmd, opts={})
+        @lines << cmd
+        CommandResult.new
+      end
+
+      class Writer
+        def initialize(lines)
+          @lines = lines
+        end
+
+        def call(*args)
+          puts @lines
+        end
+      end
+    end
+  end
+end

data/lib/specinfra/backend/ssh.rb

@@ -0,0 +1,182 @@
+# -*- coding: utf-8 -*-
+require 'specinfra/backend/exec'
+require 'net/ssh'
+require 'net/scp'
+
+module Specinfra
+  module Backend
+    class Ssh < Exec
+      def run_command(cmd, opt={})
+        cmd = build_command(cmd)
+        cmd = add_pre_command(cmd)
+
+        if get_config(:ssh_without_env)
+          ret = ssh_exec!(cmd)
+        else
+          ret = with_env do
+            ssh_exec!(cmd)
+          end
+        end
+
+        ret[:stdout].gsub!(/\r\n/, "\n")
+        ret[:stdout].gsub!(/\A\n/, "") if sudo?
+
+        if @example
+          @example.metadata[:command] = cmd
+          @example.metadata[:stdout]  = ret[:stdout]
+        end
+
+        CommandResult.new ret
+      end
+
+      def send_file(from, to)
+        scp_upload!(from, to)
+      end
+
+      def send_directory(from, to)
+        scp_upload!(from, to, :recursive => true)
+      end
+
+      def build_command(cmd)
+        cmd = super(cmd)
+        if sudo?
+          cmd = "#{sudo} -p '#{prompt}' #{cmd}"
+        end
+        cmd
+      end
+
+      private
+      def prompt
+        'Password: '
+      end
+
+      def with_env
+        env = get_config(:env) || {}
+        env[:LANG] ||= 'C'
+
+        ssh_options = get_config(:ssh_options) || {}
+        ssh_options[:send_env] ||= []
+
+        env.each do |key, value|
+          key = key.to_s
+          ENV["_SPECINFRA_#{key}"] = ENV[key];
+          ENV[key] = value
+          ssh_options[:send_env] << key
+        end
+
+        yield
+      ensure
+        env.each do |key, value|
+          key = key.to_s
+          ENV[key] = ENV.delete("_SPECINFRA_#{key}");
+        end
+      end
+
+      def create_ssh
+        Net::SSH.start(
+          get_config(:host),
+          get_config(:ssh_options)[:user],
+          get_config(:ssh_options)
+        )
+      end
+
+      def create_scp
+        ssh = get_config(:ssh)
+        if ssh.nil?
+          ssh = create_ssh
+        end
+        Net::SCP.new(ssh)
+      end
+
+      def scp_upload!(from, to, opt={})
+        if get_config(:scp).nil?
+          set_config(:scp, create_scp)
+        end
+
+        tmp = File.join('/tmp', File.basename(to))
+        scp = get_config(:scp)
+        scp.upload!(from, tmp, opt)
+        run_command(command.get(:move_file, tmp, to))
+      end
+
+      def ssh_exec!(command)
+        stdout_data = ''
+        stderr_data = ''
+        exit_status = nil
+        exit_signal = nil
+        retry_prompt = /^Sorry, try again/
+
+        if get_config(:ssh).nil?
+          set_config(:ssh, create_ssh)
+        end
+
+        ssh = get_config(:ssh)
+        ssh.open_channel do |channel|
+          if get_config(:sudo_password) or get_config(:request_pty)
+            channel.request_pty do |ch, success|
+              abort "Could not obtain pty " if !success
+            end
+          end
+          channel.exec("#{command}") do |ch, success|
+            abort "FAILED: couldn't execute command (ssh.channel.exec)" if !success
+            channel.on_data do |ch, data|
+              if data.match retry_prompt
+                abort 'Wrong sudo password! Please confirm your password.'
+              elsif data.match /^#{prompt}/
+                channel.send_data "#{get_config(:sudo_password)}\n"
+              else
+                stdout_data += data
+                @stdout_handler.call(data) if @stdout_handler
+              end
+            end
+
+            channel.on_extended_data do |ch, type, data|
+              if data.match /you must have a tty to run sudo/
+                abort 'Please write "set :request_pty, true" in your spec_helper.rb or other appropriate file.'
+              end
+
+              if data.match /^sudo: no tty present and no askpass program specified/
+                abort 'Please set sudo password to Specinfra.configuration.sudo_password.'
+              else
+                stderr_data += data
+                @stderr_handler.call(data) if @stderr_handler
+              end
+            end
+
+            channel.on_request("exit-status") do |ch, data|
+              exit_status = data.read_long
+            end
+
+            channel.on_request("exit-signal") do |ch, data|
+              exit_signal = data.read_long
+            end
+          end
+        end
+        ssh.loop
+        { :stdout => stdout_data, :stderr => stderr_data, :exit_status => exit_status, :exit_signal => exit_signal }
+      end
+
+      def sudo
+        if sudo_path = get_config(:sudo_path)
+          sudo_path += '/sudo'
+        else
+          sudo_path = 'sudo'
+        end
+
+        sudo_options = get_config(:sudo_options)
+        if sudo_options
+          sudo_options = sudo_options.shelljoin if sudo_options.is_a?(Array)
+          sudo_options = ' ' + sudo_options
+        end
+
+        "#{sudo_path.shellescape}#{sudo_options}"
+      end
+
+      def sudo?
+        user = get_config(:ssh_options)[:user]
+        disable_sudo = get_config(:disable_sudo)
+        user != 'root' && !disable_sudo
+      end
+    end
+  end
+end