Commit 76b200cf authored by Nikos Skalkotos's avatar Nikos Skalkotos
Browse files

Populate the USERS metadata for windows images

For windows images check the appropriate registry keys to find out
the system users
parent ef30380d
......@@ -33,9 +33,40 @@
from image_creator.os_type import OSBase
import hivex
import tempfile
import os
class Windows(OSBase):
"""OS class for Windows"""
def __init__(self, rootdev, ghandler, output):
super(Windows, self).__init__(rootdev, ghandler, output)
self.meta["USERS"] = " ".join(self._get_users())
def _get_users(self):
samfd, sam = tempfile.mkstemp()
systemroot = self.g.inspect_get_windows_systemroot(self.root)
path = "%s/system32/config/sam" % systemroot
path = self.g.case_sensitive_path(path), sam)
h = hivex.Hivex(sam)
key = h.root()
# Navigate to /SAM/Domains/Account/Users/Names
for child in ('SAM', 'Domains', 'Account', 'Users', 'Names'):
key = h.node_get_child(key, child)
users = [h.node_name(x) for x in h.node_children(key)]
# Filter out the guest account
return filter(lambda x: x != "Guest", users)
# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment