Added all code for integrity check transform

syzygy/experimental/protect/protect_unittest/protect_app_unittests.cc

+// Copyright 2013 Google Inc. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "syzygy/experimental/protect/protect_lib/protect_app.h"
+
+#include "base/strings/utf_string_conversions.h"
+#include "gmock/gmock.h"
+#include "gtest/gtest.h"
+#include "syzygy/core/unittest_util.h"
+#include "syzygy/pe/unittest_util.h"
+
+namespace protect {
+
+namespace {
+static wchar_t kConfigGoodExistingOutput[] =
+  L"syzygy/experimental/protect/test_data/config-good-existing-output.txt";
+
+class TestProtectApp : public ProtectApp {
+public:
+  using ProtectApp::overwrite_;
+};
+
+typedef application::Application<TestProtectApp> TestApp;
+
+class ProtectAppTest : public testing::PELibUnitTest {
+public:
+  typedef testing::PELibUnitTest Super;
+
+  ProtectAppTest()
+    : cmd_line_(base::FilePath(L"protect.exe")),
+    test_impl_(test_app_.implementation()) {
+  }
+
+  void SetUp() {
+    Super::SetUp();
+
+    logging::SetMinLogLevel(logging::LOG_ERROR);
+
+    // Setup the IO streams.
+    CreateTemporaryDir(&temp_dir_);
+    stdin_path_ = temp_dir_.Append(L"NUL");
+    stdout_path_ = temp_dir_.Append(L"stdout.txt");
+    stderr_path_ = temp_dir_.Append(L"stderr.txt");
+    InitStreams(stdin_path_, stdout_path_, stderr_path_);
+
+    ASSERT_NO_FATAL_FAILURE(ConfigureTestApp(&test_app_));
+
+    config_file_ = temp_dir_.Append(L"config.txt");
+  }
+
+  // Points the application at the fixture's command-line and IO streams.
+  template<typename TestAppType>
+  void ConfigureTestApp(TestAppType* test_app) {
+    test_app->set_command_line(&cmd_line_);
+    test_app->set_in(in());
+    test_app->set_out(out());
+    test_app->set_err(err());
+  }
+
+  // Stashes the current log-level before each test instance and restores it
+  // after each test completes.
+  testing::ScopedLogLevelSaver log_level_saver;
+
+  // @name The application under test.
+  // @{
+  TestApp test_app_;
+  TestApp::Implementation& test_impl_;
+  base::FilePath temp_dir_;
+  base::FilePath stdin_path_;
+  base::FilePath stdout_path_;
+  base::FilePath stderr_path_;
+  // @}
+
+  base::CommandLine cmd_line_;
+  base::FilePath config_file_;
+};
+
+}  // namespace
+
+TEST_F(ProtectAppTest, ParseEmptyCommandLineFails) {
+  ASSERT_FALSE(test_impl_.ParseCommandLine(&cmd_line_));
+}
+
+TEST_F(ProtectAppTest, ConfigurationFailsExistingOutput) {
+  base::FilePath input_module = testing::GetOutputRelativePath(
+    testing::kTestDllName);
+  base::FilePath output_module = temp_dir_.Append(testing::kTestDllName);
+
+  config_file_ = testing::GetSrcRelativePath(kConfigGoodExistingOutput);
+  cmd_line_.AppendSwitchPath("input-image", input_module);
+  cmd_line_.AppendSwitchPath("output-image", output_module);
+  cmd_line_.AppendSwitchPath("flummox-config-path", config_file_);
+  ASSERT_TRUE(test_impl_.ParseCommandLine(&cmd_line_));
+}
+
+TEST_F(ProtectAppTest, ConfigurationLoadsExistingOutput) {
+  base::FilePath input_module = testing::GetOutputRelativePath(
+    testing::kTestDllName);
+  base::FilePath output_module = temp_dir_.Append(testing::kTestDllName);
+
+  config_file_ = testing::GetSrcRelativePath(kConfigGoodExistingOutput);
+  cmd_line_.AppendSwitchPath("input-image", input_module);
+  cmd_line_.AppendSwitchPath("output-image", output_module);
+  cmd_line_.AppendSwitchPath("flummox-config-path", config_file_);
+  cmd_line_.AppendSwitch("overwrite");
+  ASSERT_TRUE(test_impl_.ParseCommandLine(&cmd_line_));
+}
+
+}  // namespace protect