Skip to content

dhall freeze: Remove cache import expression before a secure freeze #2486

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Feb 1, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 23 additions & 2 deletions dhall/src/Dhall/Freeze.hs
Original file line number Diff line number Diff line change
Expand Up @@ -401,9 +401,30 @@ freezeExpressionWithSettings settings directory scope intent expression = do
cache expression_ =
return expression_

case intent of
let uncache
(ImportAlt
(Core.shallowDenote -> Embed
Import{ importHashed = ImportHashed { hash = Just expectedHash, importType = Missing } }
)
(Core.shallowDenote -> Embed
import_@Import{ importHashed = ImportHashed{ hash = Nothing } }
)
) = Embed
(import_
{ importHashed = (importHashed import_)
{ hash = Just expectedHash
}
}
)
uncache expression_ = expression_

let simplify (ImportAlt (Core.shallowDenote -> Embed import1) (Core.shallowDenote -> Embed import2))
| import1 == import2 = Embed import1
simplify expression_ = expression_

Dhall.Optics.transformOf Core.subExpressions simplify <$> case intent of
Secure ->
traverse freezeFunction expression
traverse freezeFunction (Dhall.Optics.transformOf Core.subExpressions uncache expression)
Cache ->
Dhall.Optics.transformMOf Core.subExpressions cache expression

Expand Down
27 changes: 19 additions & 8 deletions dhall/tests/Dhall/Test/Freeze.hs
Original file line number Diff line number Diff line change
Expand Up @@ -21,19 +21,30 @@ import qualified Test.Tasty as Tasty
import qualified Test.Tasty.HUnit as Tasty.HUnit
import qualified Turtle

freezeDirectory :: FilePath
freezeDirectory = "./tests/freeze"
freezeDirectoryCached :: FilePath
freezeDirectoryCached = "./tests/freeze/cached"

freezeDirectorySecure :: FilePath
freezeDirectorySecure = "./tests/freeze/secure"

getTests :: IO TestTree
getTests = do
freezeTests <- Test.Util.discover (Turtle.chars <* "A.dhall") freezeTest (Turtle.lstree freezeDirectory)

let testTree = Tasty.testGroup "freeze tests" [ freezeTests ]
freezeCachedTests <- Test.Util.discover (Turtle.chars <* "A.dhall")
(freezeTest freezeDirectoryCached Cache)
(Turtle.lstree freezeDirectoryCached)
freezeSecureTests <- Test.Util.discover (Turtle.chars <* "A.dhall")
(freezeTest freezeDirectorySecure Secure)
(Turtle.lstree freezeDirectorySecure)

let testTree = Tasty.testGroup "freeze tests"
[ Tasty.testGroup "cached" [freezeCachedTests]
, Tasty.testGroup "secure" [freezeSecureTests]
]

return testTree

freezeTest :: Text -> TestTree
freezeTest prefix =
freezeTest :: FilePath -> Intent -> Text -> TestTree
freezeTest dir intent prefix =
Tasty.HUnit.testCase (Text.unpack prefix) $ do
let inputFile = Text.unpack (prefix <> "A.dhall")
let outputFile = Text.unpack (prefix <> "B.dhall")
Expand All @@ -42,7 +53,7 @@ freezeTest prefix =

parsedInput <- Core.throws (Parser.exprFromText mempty inputText)

actualExpression <- Freeze.freezeExpression (Turtle.encodeString freezeDirectory) AllImports Cache parsedInput
actualExpression <- Freeze.freezeExpression (Turtle.encodeString dir) AllImports intent parsedInput

let actualText = Core.pretty actualExpression <> "\n"

Expand Down
File renamed without changes.
1 change: 1 addition & 0 deletions dhall/tests/freeze/secure/True.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
True
2 changes: 2 additions & 0 deletions dhall/tests/freeze/secure/cachedA.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
./True.dhall sha256:27abdeddfe8503496adeb623466caa47da5f63abd2bc6fa19f6cfcb73ecfed70
? ./True.dhall
2 changes: 2 additions & 0 deletions dhall/tests/freeze/secure/cachedB.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
./True.dhall
sha256:27abdeddfe8503496adeb623466caa47da5f63abd2bc6fa19f6cfcb73ecfed70
1 change: 1 addition & 0 deletions dhall/tests/freeze/secure/incorrectHashA.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
./True.dhall sha256:0000000000000000000000000000000000000000000000000000000000000000
2 changes: 2 additions & 0 deletions dhall/tests/freeze/secure/incorrectHashB.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
./True.dhall
sha256:27abdeddfe8503496adeb623466caa47da5f63abd2bc6fa19f6cfcb73ecfed70
1 change: 1 addition & 0 deletions dhall/tests/freeze/secure/protectedA.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
./True.dhall sha256:27abdeddfe8503496adeb623466caa47da5f63abd2bc6fa19f6cfcb73ecfed70
2 changes: 2 additions & 0 deletions dhall/tests/freeze/secure/protectedB.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
./True.dhall
sha256:27abdeddfe8503496adeb623466caa47da5f63abd2bc6fa19f6cfcb73ecfed70
1 change: 1 addition & 0 deletions dhall/tests/freeze/secure/unprotectedA.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
./True.dhall
2 changes: 2 additions & 0 deletions dhall/tests/freeze/secure/unprotectedB.dhall
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
./True.dhall
sha256:27abdeddfe8503496adeb623466caa47da5f63abd2bc6fa19f6cfcb73ecfed70