From 0122b0b83d118107f53205c294030be214966591 Mon Sep 17 00:00:00 2001
From: Jed <lejosnej@ainfosec.com>
Date: Thu, 25 Aug 2016 11:32:01 -0400
Subject: [PATCH] Add migration script to fix the guest v4v firewall rules

Signed-off-by: Jed <lejosnej@ainfosec.com>
(cherry picked from commit 4be66ab1e066f0915d1842e00639bfcd79ccd551)
Signed-off-by: Jed <lejosnej@ainfosec.com>
---
 upgrade-db/Migrations.hs      |  2 ++
 upgrade-db/Migrations/M_31.hs | 51 +++++++++++++++++++++++++++++++++++
 upgrade-db/Upgrade.hs         |  2 +-
 3 files changed, 54 insertions(+), 1 deletion(-)
 create mode 100644 upgrade-db/Migrations/M_31.hs

diff --git a/upgrade-db/Migrations.hs b/upgrade-db/Migrations.hs
index 93c5ddb4..b5c97989 100644
--- a/upgrade-db/Migrations.hs
+++ b/upgrade-db/Migrations.hs
@@ -51,6 +51,7 @@ import qualified Migrations.M_27
 import qualified Migrations.M_28
 import qualified Migrations.M_29
 import qualified Migrations.M_30
+import qualified Migrations.M_31
 
 migrations :: [Migration]
 migrations = [ Migrations.M_1.migration
@@ -83,6 +84,7 @@ migrations = [ Migrations.M_1.migration
              , Migrations.M_28.migration
              , Migrations.M_29.migration
              , Migrations.M_30.migration
+             , Migrations.M_31.migration
              ]
 
 getMigrationFromVer :: Int -> Migration
diff --git a/upgrade-db/Migrations/M_31.hs b/upgrade-db/Migrations/M_31.hs
new file mode 100644
index 00000000..8ac18e6c
--- /dev/null
+++ b/upgrade-db/Migrations/M_31.hs
@@ -0,0 +1,51 @@
+--
+-- Copyright (c) 2016 Assured Information Security, Inc. <lejosnej@ainfosec.com>
+-- 
+-- This program is free software; you can redistribute it and/or modify
+-- it under the terms of the GNU General Public License as published by
+-- the Free Software Foundation; either version 2 of the License, or
+-- (at your option) any later version.
+-- 
+-- This program is distributed in the hope that it will be useful,
+-- but WITHOUT ANY WARRANTY; without even the implied warranty of
+-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+-- GNU General Public License for more details.
+-- 
+-- You should have received a copy of the GNU General Public License
+-- along with this program; if not, write to the Free Software
+-- Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+--
+
+{-# LANGUAGE PatternGuards #-}
+
+-- description: Fix the guest v4v firewall rules
+-- date: 08/25/2016
+
+module Migrations.M_31 (migration) where
+
+import UpgradeEngine
+import Data.List
+
+migration = Migration {
+              sourceVersion = 31
+            , targetVersion = 32
+            , actions = act
+            }
+
+act :: IO ()
+act = do
+  v4vRules
+
+v4vRules = xformVmJSON xform where
+    xform tree
+        | Just t <- typ, t `elem` ["svm"] = add_rules tree
+        | otherwise                       = tree
+        where
+          typ = jsUnboxString `fmap` (jsGet "/type" tree)
+
+    add_rules tree = foldl' (\t (p,v) -> jsSet p (jsBoxString v) t) (jsRm "/v4v-firewall-rules" tree) rules
+    rules = [ ("/v4v-firewall-rules/0", "my-stubdom -> 0:4001")
+            , ("/v4v-firewall-rules/1", "my-stubdom -> 0:4002")
+            , ("/v4v-firewall-rules/2", "my-stubdom -> 0:5000")
+            , ("/v4v-firewall-rules/3", "my-stubdom -> 0:5001")
+            ]
diff --git a/upgrade-db/Upgrade.hs b/upgrade-db/Upgrade.hs
index 75200674..6193bd19 100644
--- a/upgrade-db/Upgrade.hs
+++ b/upgrade-db/Upgrade.hs
@@ -40,7 +40,7 @@ import qualified Data.Text as T
 
 -- MODIFY THIS WHEN FORMAT CHANGES
 latestVersion :: Int
-latestVersion = 31
+latestVersion = 32
 ----------------------------------
 
 dbdRunning :: IO Bool