/usr/lib/python2.7/dist-packages/rekall/io_manager_test.py is in python-rekall-core 1.6.0+dfsg-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | # Rekall Memory Forensics
# Copyright 2013 Google Inc. All Rights Reserved.
#
# Author: Michael Cohen scudette@google.com
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or (at
# your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
import gzip
import os
from rekall import constants
from rekall import io_manager
from rekall import testlib
class IOManagerTest(testlib.RekallBaseUnitTestCase):
"""Test the IO Manager."""
DATA = {
"foo.gz": "hello",
"bar": "goodbye"
}
def setUp(self):
super(IOManagerTest, self).setUp()
# Create a new repository in the temp directory.
self.version = constants.PROFILE_REPOSITORY_VERSION
for filename, data in self.DATA.iteritems():
path = os.path.join(self.temp_directory, self.version,
filename)
if path.endswith("gz"):
opener = gzip.open
else:
opener = open
try:
os.makedirs(os.path.dirname(path))
except (OSError, IOError):
pass
with opener(path, "wb") as fd:
fd.write(data)
def testDirectoryIOManager(self):
manager = io_manager.DirectoryIOManager(
self.temp_directory,
session=self.MakeUserSession())
# Cant decode from json.
self.assertEqual(manager.GetData("foo"), None)
self.assertEqual(manager.GetData("foo", raw=True),
"hello")
# Test ListFiles().
self.assertListEqual(sorted(manager.ListFiles()),
["bar", "foo"])
# Storing a data structure.
data = dict(a=1)
manager.StoreData("baz", data)
self.assertDictEqual(manager.GetData("baz"),
data)
self.assertTrue(
isinstance(manager.GetData("baz", raw=True), basestring))
|