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

Unified Diff: syzygy/experimental/protect/protect_lib/protected_basic_block_filter.h

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/protected_basic_block_filter.h
diff --git a/syzygy/experimental/protect/protect_lib/protected_basic_block_filter.h b/syzygy/experimental/protect/protect_lib/protected_basic_block_filter.h
new file mode 100644
index 0000000000000000000000000000000000000000..7a72803380aa71975faff2ab7c27bd746f6625ca
--- /dev/null
+++ b/syzygy/experimental/protect/protect_lib/protected_basic_block_filter.h
@@ -0,0 +1,63 @@
+// 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.
+
+#ifndef SYZYGY_PROTECT_PROTECT_LIB_PROTECTED_BASIC_BLOCK_FILTER_H_
+#define SYZYGY_PROTECT_PROTECT_LIB_PROTECTED_BASIC_BLOCK_FILTER_H_
+
+#include <vector>
+#include "syzygy/block_graph/basic_block_subgraph.h"
+#include "syzygy/block_graph/block_builder.h"
+#include "syzygy/block_graph/block_graph.h"
+
+
+namespace protect {
+
+class ProtectedBBlockFilter {
+ public:
+ // Constructor
+
+ ProtectedBBlockFilter() {}
+
+ ProtectedBBlockFilter(
+ std::vector<std::pair<block_graph::BlockGraph::RelativeAddress,
+ block_graph::BlockGraph::Size> >& filter)
+ : filter(filter) {}
+
+ // Adds another address <-> size pair tot the filter
+ //
+ // @param tuple Address/size pair to be inserted
+ void Add(std::pair<block_graph::BlockGraph::RelativeAddress,
+ block_graph::BlockGraph::Size> &tuple);
+
+ // This is the main function for filtering basic blocks in a subgraph
+ // It checks if the blocks in the subgraph overalp with any of the defined
+ // filter ranges
+ //
+ // @param to_protect Vector through which all the found blocks are returned.
+ // @param subgraph Subgraph that contains all the basic blocks.
+ // @return true if filtering was successfull, false otherwise
+ bool Filter(std::vector<block_graph::BasicBlock *>& to_protect,
+ block_graph::BasicBlockSubGraph *subgraph);
+
+ // Helper function for the current filter
+ // @return String representation of the filter
+ std::string ToString();
+
+ private:
+ // Vector of <Address, Size> pairs used for storing address ranges for filtering.
+ std::vector < std::pair<block_graph::BlockGraph::RelativeAddress,
+ block_graph::BlockGraph::Size> > filter;
+};
+} // namespace protect
+#endif // SYZYGY_PROTECT_PROTECT_LIB_PROTECTED_BASIC_BLOCK_FILTER_H_

Powered by Google App Engine
This is Rietveld 408576698