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)