Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(306)

Unified Diff: syzygy/experimental/protect/protect_lib/protect_flummox.cc

Issue 2535563002: Added all code for integrity check transform (Closed)
Patch Set: Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: syzygy/experimental/protect/protect_lib/protect_flummox.cc
diff --git a/syzygy/experimental/protect/protect_lib/protect_flummox.cc b/syzygy/experimental/protect/protect_lib/protect_flummox.cc
new file mode 100644
index 0000000000000000000000000000000000000000..1829636e0992365889018e5ca58b9419d6e2e81c
--- /dev/null
+++ b/syzygy/experimental/protect/protect_lib/protect_flummox.cc
@@ -0,0 +1,67 @@
+// Copyright 2015 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_flummox.h"
+
+#include <algorithm>
+#include <sstream>
+
+#include "base/files/file_util.h"
+#include "base/strings/string_util.h"
+#include "syzygy/application/application.h"
+
+namespace protect {
+
+bool CustomFlummoxInstrumenter::InstrumentPrepare() {
+ return config_.ReadFromJSONPath(flummox_config_path_);
+}
+
+bool CustomFlummoxInstrumenter::InstrumentImpl() {
+ flummox_transform_.reset(
+ new protect::IntegrityCheckTransform(&config_));
+
+ if (!relinker_->AppendTransform(flummox_transform_.get())) {
+ LOG(ERROR) << "Failed to apply transform.";
+ return false;
+ }
+
+ layout_transform_.reset(
+ new protect::IntegrityCheckLayoutTransform(&config_));
+
+ if (!relinker_->AppendLayoutTransform(layout_transform_.get())) {
+ LOG(ERROR) << "Failed to apply layout transform.";
+ return false;
+ }
+ return true;
+}
+
+bool CustomFlummoxInstrumenter::DoCommandLineParse(
+ const base::CommandLine* command_line) {
+ DCHECK(command_line != nullptr);
+
+ if (!Super::DoCommandLineParse(command_line))
+ return false;
+
+ // Parse the target list filename.
+ flummox_config_path_ = application::AppImplBase::AbsolutePath(
+ command_line->GetSwitchValuePath("flummox-config-path"));
+ if (flummox_config_path_.empty()) {
+ LOG(ERROR) << "You must specify --flummox-config-path.";
+ return false;
+ }
+
+ return true;
+}
+
+} // namespace protect
« no previous file with comments | « syzygy/experimental/protect/protect_lib/protect_flummox.h ('k') | syzygy/experimental/protect/protect_lib/protect_util.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698