Files
volatility3/volatility/framework/layers/scanners/multistring_testrig.py
T

41 lines
2.0 KiB
Python

# This file was contributed to the Volatility Framework Version 3.
# Copyright (C) 2018 Volatility Foundation.
#
# THE LICENSED WORK IS PROVIDED UNDER THE TERMS OF THE Volatility Contributors
# Public License V1.0("LICENSE") AS FIRST COMPLETED BY: Volatility Foundation,
# Inc. ANY USE, PUBLIC DISPLAY, PUBLIC PERFORMANCE, REPRODUCTION OR DISTRIBUTION
# OF, OR PREPARATION OF SUBSEQUENT WORKS, DERIVATIVE WORKS OR DERIVED WORKS BASED
# ON, THE LICENSED WORK CONSTITUTES RECIPIENT'S ACCEPTANCE OF THIS LICENSE AND ITS
# TERMS, WHETHER OR NOT SUCH RECIPIENT READS THE TERMS OF THE LICENSE. "LICENSED
# WORK,” “RECIPIENT" AND “DISTRIBUTOR" ARE DEFINED IN THE LICENSE. A COPY OF THE
# LICENSE IS LOCATED IN THE TEXT FILE ENTITLED "LICENSE.txt" ACCOMPANYING THE
# CONTENTS OF THIS FILE. IF A COPY OF THE LICENSE DOES NOT ACCOMPANY THIS FILE, A
# COPY OF THE LICENSE MAY ALSO BE OBTAINED AT THE FOLLOWING WEB SITE:
# https://www.volatilityfoundation.org/license/vcpl_v1.0
#
# Software distributed under the License is distributed on an "AS IS" basis,
# WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for the
# specific language governing rights and limitations under the License.
#
def tester(searcher):
import argparse
parser = argparse.ArgumentParser(usage = "Searches through a haystack for a set of needles")
parser.add_argument(
"-n", "--needles", help = "The filename of the file containing newline separated needles", required = True)
parser.add_argument("haystack", help = "The filename of the binary haystack file to search")
args = parser.parse_args()
with open(args.needles, "rb") as needles_fp:
needles = needles_fp.read().split(b"\n")
with open(args.haystack, "rb") as haystack_fp:
haystack = haystack_fp.read()
for needle in needles:
if len(needle):
searcher.add_pattern(needle)
searcher.preprocess()
for result in searcher.search(haystack):
print("0x{:x} - {}".format(*result))