def _StreamBody(
self,
args: ApiGetCollectedTimelineArgs,
) -> api_call_handler_base.ApiBinaryStream:
client_id = str(args.client_id)
flow_id = str(args.flow_id)
opts = body.Opts()
opts.timestamp_subsecond_precision = args.body_opts.timestamp_subsecond_precision
opts.backslash_escape = args.body_opts.backslash_escape
opts.carriage_return_escape = args.body_opts.carriage_return_escape
opts.non_printable_escape = args.body_opts.non_printable_escape
if args.body_opts.HasField("inode_ntfs_file_reference_format"):
# If the field is set explicitly, we respect the choice no matter what
# filesystem we detected.
if args.body_opts.inode_ntfs_file_reference_format:
opts.inode_format = body.Opts.InodeFormat.NTFS_FILE_REFERENCE
else:
fstype = timeline.FilesystemType(client_id=client_id,
flow_id=flow_id)
if fstype is not None and fstype.lower() == "ntfs":
opts.inode_format = body.Opts.InodeFormat.NTFS_FILE_REFERENCE
entries = timeline.ProtoEntries(client_id=client_id, flow_id=flow_id)
content = body.Stream(entries, opts=opts)
filename = "timeline_{}.body".format(flow_id)
return api_call_handler_base.ApiBinaryStream(filename, content)
def testPathWithNewline(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "/foo bar\nbaz\n\nquux/thud".encode("utf-8")
stream = body.Stream(iter([entry]))
content = b"".join(stream).decode("utf-8")
self.assertIn("|/foo bar\\nbaz\\n\\nquux/thud|", content)
def testPathWithWhitespace(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "/foo bar\tbaz\rquux norf/thud".encode("utf-8")
stream = body.Stream(iter([entry]))
content = b"".join(stream).decode("utf-8")
self.assertIn("|/foo bar\tbaz\rquux norf/thud|", content)
def testBackslashEscape(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "C:\\Windows\\system32\\notepad.exe".encode("utf-8")
stream = body.Stream(iter([entry]), backslash_escape=True)
content = b"".join(stream).decode("utf-8")
self.assertIn("|C:\\\\Windows\\\\system32\\\\notepad.exe|", content)
def testPathWithBackslash(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "C:\\Windows\\system32".encode("utf-8")
stream = body.Stream(iter([entry]))
content = b"".join(stream).decode("utf-8")
self.assertIn("|C:\\Windows\\system32|", content)
def testPathWithCarriageReturn(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "/foo/bar\rbaz/quux\r\rnorf".encode("utf-8")
stream = body.Stream(iter([entry]))
content = b"".join(stream).decode("utf-8")
self.assertIn("|/foo/bar\rbaz/quux\r\rnorf|", content)
def testPathWithQuote(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "/foo\"bar".encode("utf-8")
stream = body.Stream(iter([entry]))
content = b"".join(stream).decode("utf-8")
self.assertIn("|/foo\"bar|", content)
def testSubsecondPrecision(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "/foo/bar".encode("utf-8")
entry.atime_ns = 123_456_789_000
stream = body.Stream(iter([entry]), timestamp_subsecond_precision=True)
content = b"".join(stream).decode("utf-8")
self.assertIn("/foo/bar", content)
self.assertIn("123.456789", content)
def _StreamBody(
self,
client_id: Text,
flow_id: Text,
) -> api_call_handler_base.ApiBinaryStream:
entries = timeline.Entries(client_id=client_id, flow_id=flow_id)
content = body.Stream(entries)
filename = "timeline_{}.body".format(flow_id)
return api_call_handler_base.ApiBinaryStream(filename, content)
def testUnicode(self):
entry = rdf_timeline.TimelineEntry()
entry.path = "/zażółć/gęślą/jaźń/💪".encode("utf-8")
content = b"".join(body.Stream(iter([entry]))).decode("utf-8")
reader = csv.reader(io.StringIO(content), delimiter="|")
rows = list(reader)
self.assertLen(rows, 1)
self.assertEqual(rows[0][1].encode("utf-8"), entry.path)
def testNtfsFileReference(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "/foo/bar".encode("utf-8")
entry.ino = 1688849860339456
stream = body.Stream(iter([entry]), inode_ntfs_file_reference_format=True)
content = b"".join(stream).decode("utf-8")
self.assertIn("/foo/bar", content)
self.assertIn("75520-6", content)
def _Test(c):
entry = timeline_pb2.TimelineEntry()
entry.path = f"/foo{c}bar".encode("utf-8")
content = b"".join(body.Stream(iter([entry]))).decode("utf-8")
reader = csv.reader(io.StringIO(content), delimiter="|")
rows = list(reader)
self.assertLen(rows, 1)
self.assertEqual(rows[0][1].encode("utf-8"), entry.path)
def testCarriageReturnEscape(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "C:\\Foo\rBar\\Baz\r\rQuux".encode("utf-8")
opts = body.Opts()
opts.carriage_return_escape = True
stream = body.Stream(iter([entry]), opts=opts)
content = b"".join(stream).decode("utf-8")
self.assertIn("|C:\\Foo\\rBar\\Baz\\r\\rQuux|", content)
def testNonPrintableEscape(self):
entry = timeline_pb2.TimelineEntry()
entry.path = b"/f\x00b\x0ar\x1baz"
opts = body.Opts()
opts.non_printable_escape = True
stream = body.Stream(iter([entry]), opts=opts)
content = b"".join(stream).decode("utf-8")
self.assertIn(r"|/f\x00b\x0ar\x1baz|", content)
def testNtfsFileReference(self):
entry = timeline_pb2.TimelineEntry()
entry.path = "/foo/bar".encode("utf-8")
entry.ino = 1688849860339456
opts = body.Opts()
opts.inode_format = body.Opts.InodeFormat.NTFS_FILE_REFERENCE
stream = body.Stream(iter([entry]), opts=opts)
content = b"".join(stream).decode("utf-8")
self.assertIn("/foo/bar", content)
self.assertIn("75520-6", content)
def testSingle(self):
entry = rdf_timeline.TimelineEntry()
entry.path = "/foo/bar/baz".encode("utf-8")
entry.mode = 0o100644
entry.size = 42
entry.dev = 101
entry.ino = 404
entry.uid = 13
entry.gid = 7
entry.atime_ns = 123 * 1_000_000_000
entry.mtime_ns = 456 * 1_000_000_000
entry.ctime_ns = 789 * 1_000_000_000
content = b"".join(body.Stream(iter([entry]))).decode("utf-8")
expected = "0|/foo/bar/baz|404|-rw-r--r--|13|7|42|123|456|789|0\n"
self.assertEqual(content, expected)
def _StreamBody(
self,
args: ApiGetCollectedTimelineArgs,
) -> api_call_handler_base.ApiBinaryStream:
client_id = str(args.client_id)
flow_id = str(args.flow_id)
opts = body.Opts()
opts.timestamp_subsecond_precision = args.body_opts.timestamp_subsecond_precision
opts.inode_ntfs_file_reference_format = args.body_opts.inode_ntfs_file_reference_format
opts.backslash_escape = args.body_opts.backslash_escape
entries = timeline.ProtoEntries(client_id=client_id, flow_id=flow_id)
content = body.Stream(entries, opts=opts)
filename = "timeline_{}.body".format(flow_id)
return api_call_handler_base.ApiBinaryStream(filename, content)
def testChunks(self):
entries = []
for idx in range(1024):
entry = rdf_timeline.TimelineEntry()
entry.path = "/foo/bar{}".format(idx).encode("utf-8")
entries.append(entry)
chunks = list(body.Stream(iter(entries), chunk_size=6))
self.assertLen(chunks, len(entries))
content = b"".join(chunks).decode("utf-8")
reader = csv.reader(io.StringIO(content), delimiter="|")
rows = list(reader)
self.assertLen(rows, len(entries))
for idx, row in enumerate(rows):
self.assertEqual(row[1].encode("utf-8"), entries[idx].path)
def testMultiple(self):
entries = []
for idx in range(100):
entry = rdf_timeline.TimelineEntry()
entry.path = "/foo/bar/baz{}".format(idx).encode("utf-8")
entry.size = idx
entries.append(entry)
content = b"".join(body.Stream(iter(entries))).decode("utf-8")
reader = csv.reader(io.StringIO(content), delimiter="|")
rows = list(reader)
self.assertLen(rows, len(entries))
for idx, row in enumerate(rows):
self.assertEqual(row[1].encode("utf-8"), entries[idx].path)
self.assertEqual(int(row[6]), entries[idx].size)
def testNonUnicode(self):
entry = timeline_pb2.TimelineEntry()
entry.path = b"/\xff\x00\x00/\xfb\xfa\xf5"
content = b"".join(body.Stream(iter([entry])))
self.assertIn(b"/\xff\x00\x00/\xfb\xfa\xf5", content)
