def collect(self):
apis = []
for plugin_api in self.generate_api():
apis.append(plugin_api)
yield dict(plugin=plugin_api["plugin"],
api=yaml_utils.safe_dump(plugin_api),
raw_api=plugin_api)
if self.plugin_args.output_file:
with open(self.plugin_args.output_file, "wb") as fd:
fd.write(yaml_utils.safe_dump(apis))
def collect(self):
apis = []
for plugin_api in self.generate_api():
apis.append(plugin_api)
yield dict(plugin=plugin_api["plugin"],
api=yaml_utils.safe_dump(plugin_api),
raw_api=plugin_api)
if self.plugin_args.output_file:
with open(self.plugin_args.output_file, "wb") as fd:
fd.write(yaml_utils.safe_dump(apis))
def write_manifest(self):
yield dict(Message="Writing manifest file.")
# Now upload the signed manifest to the bucket. Manifest must be
# publicly accessible.
upload_location = self._config.server.manifest_for_server()
yield dict(Message="Writing manifest file to %s" % (
upload_location.to_path()))
upload_location.write_file(self._config.signed_manifest.to_json())
print yaml_utils.safe_dump(self._config.manifest.to_primitive())
def write_manifest(self):
yield dict(Message="Writing manifest file.")
# Now upload the signed manifest to the bucket. Manifest must be
# publicly accessible.
upload_location = self._config.server.manifest_for_server()
yield dict(Message="Writing manifest file to %s" %
(upload_location.to_path()))
upload_location.write_file(self._config.signed_manifest.to_json())
print yaml_utils.safe_dump(self._config.manifest.to_primitive())
def write_config(self):
server_config_filename = os.path.join(
self.config_dir, self.server_config_filename)
if os.access(server_config_filename, os.R_OK):
yield dict(
Message="Server config at %s exists. Remove to regenerate." % (
server_config_filename))
# Load existing server config.
server_config_data = open(server_config_filename, "rb").read()
config = agent.Configuration.from_primitive(
yaml.safe_load(server_config_data), session=self.session)
else:
# Make a new configuration
config = agent.Configuration(session=self.session)
self.session.SetParameter("agent_config_obj", config)
self._build_config(config)
yield dict(Message="Writing server config file %s" %
server_config_filename)
with open(server_config_filename, "wb") as fd:
fd.write(yaml_utils.safe_dump(config.to_primitive()))
# The client gets just the client part of the configuration.
client_config = agent.Configuration(session=self.session)
client_config.client = config.client
client_config.ca_certificate = config.ca_certificate
client_config_filename = os.path.join(
self.config_dir, self.client_config_filename)
yield dict(
Message="Writing client config file %s" % (
client_config_filename))
with open(client_config_filename, "wb") as fd:
fd.write(self.client_config_warning +
yaml_utils.safe_dump(client_config.to_primitive()))
# Now load the server config file to make sure it is validly written.
self.session.SetParameter("agent_configuration", server_config_filename)
self._config = self.session.GetParameter(
"agent_config_obj", cached=False)
if self._config == None:
raise RuntimeError("Unable to parse provided configuration.")
def write_config(self):
server_config_filename = os.path.join(self.config_dir,
self.server_config_filename)
if os.access(server_config_filename, os.R_OK):
yield dict(
Message="Server config at %s exists. Remove to regenerate." %
(server_config_filename))
# Load existing server config.
server_config_data = open(server_config_filename, "rb").read()
config = agent.Configuration.from_primitive(
yaml.safe_load(server_config_data), session=self.session)
else:
# Make a new configuration
config = agent.Configuration(session=self.session)
self.session.SetParameter("agent_config_obj", config)
self._build_config(config)
yield dict(Message="Writing server config file %s" %
server_config_filename)
with open(server_config_filename, "wb") as fd:
fd.write(yaml_utils.safe_dump(config.to_primitive()))
# The client gets just the client part of the configuration.
client_config = agent.Configuration(session=self.session)
client_config.client = config.client
client_config.ca_certificate = config.ca_certificate
client_config_filename = os.path.join(self.config_dir,
self.client_config_filename)
yield dict(Message="Writing client config file %s" %
(client_config_filename))
with open(client_config_filename, "wb") as fd:
fd.write(self.client_config_warning +
yaml_utils.safe_dump(client_config.to_primitive()))
# Now load the server config file to make sure it is validly written.
self.session.SetParameter("agent_configuration",
server_config_filename)
self._config = self.session.GetParameter("agent_config_obj",
cached=False)
if self._config == None:
raise RuntimeError("Unable to parse provided configuration.")
def collect(self):
apis = []
for option, api in six.iteritems(config.OPTIONS.args):
for k, v in list(api.items()):
if callable(v):
api[k] = v()
apis.append(api)
yield dict(option=option,
api=yaml_utils.safe_dump(api),
raw_api=api)
if self.plugin_args.output_file:
with open(self.plugin_args.output_file, "wb") as fd:
fd.write(yaml_utils.safe_dump(apis))
def collect(self):
apis = []
for option, api in config.OPTIONS.args.items():
for k, v in api.items():
if callable(v):
api[k] = v()
apis.append(api)
yield dict(option=option,
api=yaml_utils.safe_dump(api),
raw_api=api)
if self.plugin_args.output_file:
with open(self.plugin_args.output_file, "wb") as fd:
fd.write(yaml_utils.safe_dump(apis))
def collect(self):
artifact_profile = self.session.LoadProfile("artifacts")
for artifact in self.plugin_args.artifacts:
definition = artifact_profile.definitions_by_name.get(artifact)
if definition:
yield dict(divider=artifact)
yield dict(Message=yaml_utils.safe_dump(definition))
def collect(self):
artifact_profile = self.session.LoadProfile("artifacts")
for artifact in self.plugin_args.artifacts:
definition = artifact_profile.definitions_by_name.get(artifact)
if definition:
yield dict(divider=artifact)
yield dict(Message=yaml_utils.safe_dump(definition))
def CreateDefaultConfigFile(session):
"""Creates a default config file."""
homedir = GetHomeDir(session)
if homedir:
try:
filename = "%s/.rekallrc" % homedir
with open(filename, "wt") as fd:
fd.write(yaml_utils.safe_dump(DEFAULT_CONFIGURATION))
logging.info("Created new configuration file %s", filename)
cache_dir = os.path.join(
homedir, DEFAULT_CONFIGURATION["cache_dir"])
os.makedirs(cache_dir)
logging.info("Created new cache directory %s", cache_dir)
return DEFAULT_CONFIGURATION
except (IOError, OSError):
pass
# Can not write it anywhere but at least we start with something sensible.
return DEFAULT_CONFIGURATION