From: Eugene Crosser <crosser@average.org>
Date: Thu, 5 Jun 2014 16:36:22 +0000 (+0400)
Subject: propagate forced TLSA onto subdomains
X-Git-Tag: 0.9.0.1~1
X-Git-Url: http://www.average.org/gitweb/?p=pdns-pipe-nmc.git;a=commitdiff_plain;h=b8ee8aa86f0fa6575653ccdf8c985fa0e33211c2;hp=18e24a6fca605da574ba3add941ee9981e5396af;ds=sidebyside

propagate forced TLSA onto subdomains
---

diff --git a/NmcTransform.hs b/NmcTransform.hs
index 23244a5..4a3e667 100644
--- a/NmcTransform.hs
+++ b/NmcTransform.hs
@@ -2,11 +2,12 @@ module NmcTransform ( seedNmcDom
                     , descendNmcDom
                     ) where
 
-import Prelude hiding (lookup)
+import Prelude hiding (lookup, null)
 import Data.ByteString.Lazy (ByteString)
-import Data.Map.Lazy (empty, lookup, delete, size, singleton
+import Data.Map.Lazy (Map, empty, lookup, delete, null, singleton
                      , foldrWithKey, insert, insertWith)
 import Control.Monad (foldM)
+import Data.Maybe (fromMaybe)
 import Data.Aeson (decode)
 import Data.Default.Class (def)
 
@@ -62,7 +63,7 @@ mergeSelf base =
     map   = domSubmap base
     base' = base {domSubmap = removeSelf map}
     removeSelf Nothing    = Nothing
-    removeSelf (Just map) = if size map' == 0 then Nothing else Just map'
+    removeSelf (Just map) = if null map' then Nothing else Just map'
       where map' = delete "" map
   in
     case map of
@@ -76,10 +77,23 @@ mergeSelf base =
 -- | transfer some elements of `base` into `sub`, notably TLSA
 propagate :: NmcDom -> NmcDom -> NmcDom
 propagate base sub = sub `merge` (pickglobals base)
-  where -- FIXME must do this on the map elements, not on the top level
-    pickglobals dom = def { domTlsa = fmap pickforcedtls (domTlsa dom) }
-    pickforcedtls = filter (\x -> tlsIncSubdoms x)
- 
+  where
+    pickglobals dom = fromMaybe def (siftsubmap (siftsubmap taketlsa) dom)
+    siftsubmap f dom =
+      let
+        sdmap = fromMaybe empty (domSubmap dom)
+        sdmap' = foldrWithKey (\k x -> addifjust k (f x)) empty sdmap
+        addifjust k mdom acc = case mdom of
+          Nothing  -> acc
+          Just dom -> insert k dom acc -- dups are impossible here
+      in
+        if null sdmap' then Nothing else Just $ def { domSubmap = Just sdmap'}
+    taketlsa dom = case domTlsa dom of
+      Nothing   -> Nothing
+      Just tlsa -> case filter (\x -> tlsIncSubdoms x) tlsa of
+        []    -> Nothing
+        tlsa' -> Just $ def { domTlsa = Just tlsa' }
+
 -- | Presence of some elements require removal of some others
 normalizeDom :: NmcDom -> NmcDom
 normalizeDom dom = foldr id dom [ translateNormalizer