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)