def get_c2dns(id):
"""Return c2dns artifact associated with the given id
Return: c2dns artifact dictionary"""
entity = c2dns.C2dns.query.get(id)
if not entity:
abort(404)
if not current_user.admin and entity.owner_user_id != current_user.id:
abort(403)
return_dict = entity.to_dict()
return_dict["bookmarked"] = True if is_bookmarked(
ENTITY_MAPPING["DNS"], id, current_user.id) else False
return jsonify(return_dict)
def get_tasks(id):
"""Return task associated with given id
Return: task dictionary"""
entity = tasks.Tasks.query.get(id)
if not entity:
abort(404)
if not current_user.admin and not entity.owner_user_id == current_user.id and not entity.owner_user_id == None:
return jsonify({})
return_dict = entity.to_dict()
return_dict["bookmarked"] = True if is_bookmarked(
ENTITY_MAPPING["TASK"], id, current_user.id) else False
return jsonify(return_dict)
def get_yara_rule(id):
"""Return yara_rule artifact associated with the given id
Return: yara_rule artifact dictionary"""
include_yara_string = request.args.get("include_yara_string", False)
short = distutils.util.strtobool(request.args.get("short", "false"))
if include_yara_string:
include_yara_string = True
entity = yara_rule.Yara_rule.query.get(id)
if not entity:
abort(404)
if not current_user.admin and entity.owner_user_id != current_user.id:
abort(403)
return_dict = entity.to_dict(include_yara_string, short)
return_dict["bookmarked"] = True if is_bookmarked(ENTITY_MAPPING["SIGNATURE"], id, current_user.id) \
else False
return jsonify(return_dict)
def get_tasks(id):
"""Return task associated with given id
Return: task dictionary"""
entity = tasks.Tasks.query.get(id)
if not entity:
abort(404)
show_for_non_admin = cfg_settings.Cfg_settings.get_setting(
"ENABLE_NON_ADMIN_TASK_VISIBILITY")
show_for_non_admin = bool(distutils.util.strtobool(
show_for_non_admin)) if show_for_non_admin else False
if not show_for_non_admin and \
not (current_user.admin or entity.owner_user_id == current_user.id or entity.owner_user_id is None):
return jsonify({})
return_dict = entity.to_dict()
return_dict["bookmarked"] = True if is_bookmarked(
ENTITY_MAPPING["TASK"], id, current_user.id) else False
return jsonify(return_dict)