From 90f63680bf19781ba09e9bbfaad26283cc7787b1 Mon Sep 17 00:00:00 2001
From: sachin-ac <sachin_a@live.com>
Date: Thu, 9 May 2019 12:44:46 -0700
Subject: [PATCH] Update install script to check if containers feature is
 enabled on Iot Core Image (#1158)

---
 scripts/windows/setup/IotEdgeSecurityDaemon.ps1 | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/scripts/windows/setup/IotEdgeSecurityDaemon.ps1 b/scripts/windows/setup/IotEdgeSecurityDaemon.ps1
index 97b25df3992..4d28b815aed 100644
--- a/scripts/windows/setup/IotEdgeSecurityDaemon.ps1
+++ b/scripts/windows/setup/IotEdgeSecurityDaemon.ps1
@@ -24,6 +24,9 @@ Set-Variable EdgeInstallDirectory -Value "$env:ProgramFiles\iotedge" -Option Con
 Set-Variable EdgeDataDirectory -Value "$env:ProgramData\iotedge" -Option Constant
 Set-Variable EdgeServiceName -Value 'iotedge' -Option Constant
 
+Set-Variable ContainersFeaturePackageName -Value 'Microsoft-IoT-Containers-Server-Package' -Option Constant
+Set-Variable ContainersFeatureLangPackageName -Value 'Microsoft-IoT-Containers-Server-Package_*' -Option Constant
+
 Set-Variable MobyDataRootDirectory -Value "$env:ProgramData\iotedge-moby" -Option Constant
 Set-Variable MobyInstallDirectory -Value "$env:ProgramFiles\iotedge-moby" -Option Constant
 Set-Variable MobyLinuxNamedPipeUrl -Value 'npipe://./pipe/docker_engine' -Option Constant
@@ -863,6 +866,13 @@ function Setup-Environment {
             'See https://aka.ms/iotedge-platsup for more details.')
         $preRequisitesMet = $false
     }
+    
+    if (Test-IoTCore) {
+        if (-not (Get-Service vmcompute -ErrorAction SilentlyContinue) -or (-not [bool] (Get-Package $ContainersFeaturePackageName)) -or (-not [bool] (Get-Package $ContainersFeatureLangPackageName))) {
+            Write-HostRed "The container host does not have 'Containers Feature' enabled. Please build an Iot Core image with 'Containers Feature' enabled."
+            $preRequisitesMet = $false
+        }
+    }
 
     if ($preRequisitesMet) {
         Write-HostGreen "The container host is on supported build version $currentWindowsBuild."