def Run(self, args):
hashers = {}
for t in args.tuples:
for hash_name in t.hashers:
hashers[str(hash_name).lower()] = self._hash_types[str(
hash_name)]()
with vfs.VFSOpen(args.pathspec,
progress_callback=self.Progress) as file_obj:
# Only read as many bytes as we were told.
bytes_read = 0
while bytes_read < args.max_filesize:
self.Progress()
data = file_obj.Read(MAX_BUFFER_SIZE)
if not data:
break
for hasher in hashers.values():
hasher.update(data)
bytes_read += len(data)
response = rdf_client.FingerprintResponse(
pathspec=file_obj.pathspec,
bytes_read=bytes_read,
hash=rdf_crypto.Hash(**dict(
(k, v.digest()) for k, v in hashers.iteritems())))
self.SendReply(response)
def Run(self, args):
"""Fingerprint a file."""
with vfs.VFSOpen(args.pathspec,
progress_callback=self.Progress) as file_obj:
fingerprinter = Fingerprinter(self.Progress, file_obj)
response = rdf_client.FingerprintResponse()
response.pathspec = file_obj.pathspec
if args.tuples:
tuples = args.tuples
else:
# There are none selected -- we will cover everything
tuples = list()
for k in self._fingerprint_types.iterkeys():
tuples.append(rdf_client.FingerprintTuple(fp_type=k))
for finger in tuples:
hashers = [self._hash_types[h] for h in finger.hashers] or None
if finger.fp_type in self._fingerprint_types:
invoke = self._fingerprint_types[finger.fp_type]
res = invoke(fingerprinter, hashers)
if res:
response.matching_types.append(finger.fp_type)
else:
raise RuntimeError(
"Encountered unknown fingerprint type. %s" %
finger.fp_type)
# Structure of the results is a list of dicts, each containing the
# name of the hashing method, hashes for enabled hash algorithms,
# and auxilliary data where present (e.g. signature blobs).
# Also see Fingerprint:HashIt()
response.results = fingerprinter.HashIt()
# We now return data in a more structured form.
for result in response.results:
if result.GetItem("name") == "generic":
for hash_type in ["md5", "sha1", "sha256"]:
value = result.GetItem(hash_type)
if value is not None:
setattr(response.hash, hash_type, value)
if result["name"] == "pecoff":
for hash_type in ["md5", "sha1", "sha256"]:
value = result.GetItem(hash_type)
if value:
setattr(response.hash, "pecoff_" + hash_type,
value)
signed_data = result.GetItem("SignedData", [])
for data in signed_data:
response.hash.signed_data.Append(revision=data[0],
cert_type=data[1],
certificate=data[2])
self.SendReply(response)
def Run(self, args):
hash_types = set()
for t in args.tuples:
for hash_name in t.hashers:
hash_types.add(str(hash_name).lower())
hasher = client_utils_common.MultiHasher(hash_types, progress=self.Progress)
with vfs.VFSOpen(args.pathspec, progress_callback=self.Progress) as fd:
hasher.HashFile(fd, args.max_filesize)
hash_object = hasher.GetHashObject()
response = rdf_client.FingerprintResponse(
pathspec=fd.pathspec,
bytes_read=hash_object.num_bytes,
hash=hash_object)
self.SendReply(response)
def Run(self, args):
hash_types = set()
for t in args.tuples:
for hash_name in t.hashers:
hash_types.add(str(hash_name).lower())
with vfs.VFSOpen(args.pathspec,
progress_callback=self.Progress) as file_obj:
hashers, bytes_read = self.HashFile(hash_types, file_obj,
args.max_filesize)
self.SendReply(
rdf_client.FingerprintResponse(
pathspec=file_obj.pathspec,
bytes_read=bytes_read,
hash=rdf_crypto.Hash(**dict(
(k, v.digest()) for k, v in hashers.iteritems()))))
