From 7899fc067f68c020519f38db061ffd024dbb274b Mon Sep 17 00:00:00 2001
From: "Z. Cliffe Schreuders" <code.cliffe@schreuders.org>
Date: Thu, 12 Nov 2020 23:23:18 +0000
Subject: [PATCH] lab updates

---
 .../utilities/unix/audit_tools/lime/manifests/install.pp | 9 +++++----
 .../utilities/unix/audit_tools/lime/secgen_metadata.xml  | 3 +++
 .../unix/ctf/zip_file/manifests/install.pp               | 3 +++
 3 files changed, 11 insertions(+), 4 deletions(-)
 create mode 100644 modules/vulnerabilities/unix/ctf/zip_file/manifests/install.pp

diff --git a/modules/utilities/unix/audit_tools/lime/manifests/install.pp b/modules/utilities/unix/audit_tools/lime/manifests/install.pp
index 4d1873357..e74b6e084 100644
--- a/modules/utilities/unix/audit_tools/lime/manifests/install.pp
+++ b/modules/utilities/unix/audit_tools/lime/manifests/install.pp
@@ -1,11 +1,12 @@
 class lime::install{
   Exec { path => ['/bin','/sbin','/usr/bin', '/usr/sbin'] }
 
-  ensure_packages(['git','build-essentials'])
+  ensure_packages(['git','build-essential', "linux-headers-${facts['kernelrelease']}"])
 
-  exec { 'install linux-headers for LiME':
-    command => 'apt-get install linux-headers-$(uname -r)',
-  }->
+  # exec { 'install linux-headers for LiME':
+  #   command => "apt-get install linux-headers-${facts['kernelrelease']}",
+  #   provider => 'shell',
+  # }->
   exec {'clone LiME repo':
     command => 'git clone https://github.com/504ensicsLabs/LiME.git',
     cwd => '/root/'
diff --git a/modules/utilities/unix/audit_tools/lime/secgen_metadata.xml b/modules/utilities/unix/audit_tools/lime/secgen_metadata.xml
index 61481f9e2..af361d8fc 100644
--- a/modules/utilities/unix/audit_tools/lime/secgen_metadata.xml
+++ b/modules/utilities/unix/audit_tools/lime/secgen_metadata.xml
@@ -13,4 +13,7 @@
 
   <reference>https://github.com/504ensicsLabs/LiME</reference>
 
+  <requires>
+    <type>update</type>
+  </requires>
 </utility>
diff --git a/modules/vulnerabilities/unix/ctf/zip_file/manifests/install.pp b/modules/vulnerabilities/unix/ctf/zip_file/manifests/install.pp
new file mode 100644
index 000000000..c3ac4b783
--- /dev/null
+++ b/modules/vulnerabilities/unix/ctf/zip_file/manifests/install.pp
@@ -0,0 +1,3 @@
+class zip_file::install {
+  ensure_packages(['zip'])
+}