def main(argv=None):
del argv # Unused.
config.CONFIG.AddContext(contexts.COMMAND_LINE_CONTEXT)
config.CONFIG.AddContext(
contexts.CONSOLE_CONTEXT,
"Context applied when running the console binary.")
server_startup.Init()
fleetspeak_connector.Init()
username = flags.FLAGS.username
if not username:
username = os.environ["USER"]
if not username:
print(
"Username has to be specified with either --username flag or "
"USER environment variable.")
sys.exit(1)
grrapi = api.GrrApi(connector=api_shell_raw_access_lib.RawConnector(
token=access_control.ACLToken(username=username),
page_size=flags.FLAGS.page_size))
if flags.FLAGS.exec_code and flags.FLAGS.exec_file:
print "--exec_code --exec_file flags can't be supplied together."
sys.exit(1)
elif flags.FLAGS.exec_code:
# pylint: disable=exec-used
exec(flags.FLAGS.exec_code, dict(grrapi=grrapi))
# pylint: enable=exec-used
elif flags.FLAGS.exec_file:
execfile(flags.FLAGS.exec_file, dict(grrapi=grrapi))
else:
api_shell_lib.IPShell([sys.argv[0]], user_ns=dict(grrapi=grrapi))
def main(argv):
"""Main."""
del argv # Unused.
server_startup.Init()
filename = flags.FLAGS.filename
if not os.path.exists(filename):
print "File %s does not exist" % filename
return
with aff4.FACTORY.Create(filestore.NSRLFileStore.PATH,
filestore.NSRLFileStore,
mode="rw",
token=aff4.FACTORY.root_token) as store:
imported = ImportFile(store, filename, flags.FLAGS.start)
data_store.DB.Flush()
print "Imported %d hashes" % imported
def AddUsers(token=None):
# Now initialize with our modified config.
server_startup.Init()
print "\nStep 3: Adding Admin User"
try:
maintenance_utils.AddUser(
"admin",
labels=["admin"],
token=token,
password=flags.FLAGS.admin_password)
except maintenance_utils.UserError:
if flags.FLAGS.noprompt:
maintenance_utils.UpdateUser(
"admin",
password=flags.FLAGS.admin_password,
add_labels=["admin"],
token=token)
else:
if ((raw_input("User 'admin' already exists, do you want to "
"reset the password? [yN]: ").upper() or "N") == "Y"):
maintenance_utils.UpdateUser(
"admin", password=True, add_labels=["admin"], token=token)
