| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 1 | # -*- coding: utf-8 -*- |
| 2 | |
| Eduardo Sousa | d795f87 | 2019-02-05 16:05:53 +0000 | [diff] [blame] | 3 | # Copyright 2018 Whitestack, LLC |
| 4 | # Copyright 2018 Telefonica S.A. |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 5 | # |
| Eduardo Sousa | d795f87 | 2019-02-05 16:05:53 +0000 | [diff] [blame] | 6 | # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| 7 | # not use this file except in compliance with the License. You may obtain |
| 8 | # a copy of the License at |
| 9 | # |
| 10 | # http://www.apache.org/licenses/LICENSE-2.0 |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 11 | # |
| 12 | # Unless required by applicable law or agreed to in writing, software |
| Eduardo Sousa | d795f87 | 2019-02-05 16:05:53 +0000 | [diff] [blame] | 13 | # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 14 | # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 15 | # License for the specific language governing permissions and limitations |
| 16 | # under the License. |
| 17 | # |
| 18 | # For those usages not covered by the Apache License, Version 2.0 please |
| 19 | # contact: esousa@whitestack.com or alfonso.tiernosepulveda@telefonica.com |
| 20 | ## |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 21 | |
| 22 | |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 23 | """ |
| 24 | Authenticator is responsible for authenticating the users, |
| 25 | create the tokens unscoped and scoped, retrieve the role |
| 26 | list inside the projects that they are inserted |
| 27 | """ |
| 28 | |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 29 | __author__ = "Eduardo Sousa <esousa@whitestack.com>; Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>" |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 30 | __date__ = "$27-jul-2018 23:59:59$" |
| 31 | |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 32 | import cherrypy |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 33 | import logging |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 34 | import yaml |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 35 | from base64 import standard_b64decode |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 36 | from copy import deepcopy |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 37 | |
| tierno | 38dcfeb | 2019-06-10 16:44:00 +0000 | [diff] [blame] | 38 | # from functools import reduce |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 39 | from http import HTTPStatus |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 40 | from time import time |
| Eduardo Sousa | 5c01e19 | 2019-05-08 02:35:47 +0100 | [diff] [blame] | 41 | from os import path |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 42 | |
| tierno | c23a9bb | 2020-06-24 10:54:11 +0000 | [diff] [blame] | 43 | from osm_nbi.authconn import AuthException, AuthconnException, AuthExceptionUnauthorized |
| tierno | 9c63011 | 2019-08-29 14:21:41 +0000 | [diff] [blame] | 44 | from osm_nbi.authconn_keystone import AuthconnKeystone |
| delacruzramo | ad682a5 | 2019-12-10 16:26:34 +0100 | [diff] [blame] | 45 | from osm_nbi.authconn_internal import AuthconnInternal |
| K Sai Kiran | 7ddb073 | 2020-10-30 11:14:44 +0530 | [diff] [blame] | 46 | from osm_nbi.authconn_tacacs import AuthconnTacacs |
| elumalai | 7802ff8 | 2023-04-24 20:38:32 +0530 | [diff] [blame] | 47 | from osm_nbi.utils import cef_event, cef_event_builder |
| delacruzramo | ad682a5 | 2019-12-10 16:26:34 +0100 | [diff] [blame] | 48 | from osm_common import dbmemory, dbmongo, msglocal, msgkafka |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 49 | from osm_common.dbbase import DbException |
| delacruzramo | 029405d | 2019-09-26 10:52:56 +0200 | [diff] [blame] | 50 | from osm_nbi.validation import is_valid_uuid |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 51 | from itertools import chain |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 52 | from uuid import uuid4 |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 53 | |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 54 | |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 55 | class Authenticator: |
| 56 | """ |
| 57 | This class should hold all the mechanisms for User Authentication and |
| 58 | Authorization. Initially it should support Openstack Keystone as a |
| 59 | backend through a plugin model where more backends can be added and a |
| 60 | RBAC model to manage permissions on operations. |
| tierno | 65ca36d | 2019-02-12 19:27:52 +0100 | [diff] [blame] | 61 | This class must be threading safe |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 62 | """ |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 63 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 64 | periodin_db_pruning = ( |
| 65 | 60 * 30 |
| 66 | ) # for the internal backend only. every 30 minutes expired tokens will be pruned |
| 67 | token_limit = 500 # when reached, the token cache will be cleared |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 68 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 69 | def __init__(self, valid_methods, valid_query_string): |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 70 | """ |
| 71 | Authenticator initializer. Setup the initial state of the object, |
| 72 | while it waits for the config dictionary and database initialization. |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 73 | """ |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 74 | self.backend = None |
| 75 | self.config = None |
| 76 | self.db = None |
| delacruzramo | ad682a5 | 2019-12-10 16:26:34 +0100 | [diff] [blame] | 77 | self.msg = None |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 78 | self.tokens_cache = dict() |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 79 | self.next_db_prune_time = ( |
| 80 | 0 # time when next cleaning of expired tokens must be done |
| 81 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 82 | self.roles_to_operations_file = None |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 83 | # self.roles_to_operations_table = None |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 84 | self.resources_to_operations_mapping = {} |
| 85 | self.operation_to_allowed_roles = {} |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 86 | self.logger = logging.getLogger("nbi.authenticator") |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 87 | self.role_permissions = [] |
| 88 | self.valid_methods = valid_methods |
| 89 | self.valid_query_string = valid_query_string |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 90 | self.system_admin_role_id = None # system_role id |
| tierno | d4a705a | 2020-06-22 10:58:26 +0000 | [diff] [blame] | 91 | self.test_project_id = None # test_project_id |
| elumalai | 7802ff8 | 2023-04-24 20:38:32 +0530 | [diff] [blame] | 92 | self.cef_logger = None |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 93 | |
| 94 | def start(self, config): |
| 95 | """ |
| 96 | Method to configure the Authenticator object. This method should be called |
| 97 | after object creation. It is responsible by initializing the selected backend, |
| 98 | as well as the initialization of the database connection. |
| 99 | |
| 100 | :param config: dictionary containing the relevant parameters for this object. |
| 101 | """ |
| 102 | self.config = config |
| elumalai | 7802ff8 | 2023-04-24 20:38:32 +0530 | [diff] [blame] | 103 | self.cef_logger = cef_event_builder(config["authentication"]) |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 104 | |
| 105 | try: |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 106 | if not self.db: |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 107 | if config["database"]["driver"] == "mongo": |
| 108 | self.db = dbmongo.DbMongo() |
| 109 | self.db.db_connect(config["database"]) |
| 110 | elif config["database"]["driver"] == "memory": |
| 111 | self.db = dbmemory.DbMemory() |
| 112 | self.db.db_connect(config["database"]) |
| 113 | else: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 114 | raise AuthException( |
| 115 | "Invalid configuration param '{}' at '[database]':'driver'".format( |
| 116 | config["database"]["driver"] |
| 117 | ) |
| 118 | ) |
| delacruzramo | ad682a5 | 2019-12-10 16:26:34 +0100 | [diff] [blame] | 119 | if not self.msg: |
| 120 | if config["message"]["driver"] == "local": |
| 121 | self.msg = msglocal.MsgLocal() |
| 122 | self.msg.connect(config["message"]) |
| 123 | elif config["message"]["driver"] == "kafka": |
| 124 | self.msg = msgkafka.MsgKafka() |
| 125 | self.msg.connect(config["message"]) |
| 126 | else: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 127 | raise AuthException( |
| 128 | "Invalid configuration param '{}' at '[message]':'driver'".format( |
| 129 | config["message"]["driver"] |
| 130 | ) |
| 131 | ) |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 132 | if not self.backend: |
| 133 | if config["authentication"]["backend"] == "keystone": |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 134 | self.backend = AuthconnKeystone( |
| 135 | self.config["authentication"], self.db, self.role_permissions |
| 136 | ) |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 137 | elif config["authentication"]["backend"] == "internal": |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 138 | self.backend = AuthconnInternal( |
| 139 | self.config["authentication"], self.db, self.role_permissions |
| 140 | ) |
| K Sai Kiran | 7ddb073 | 2020-10-30 11:14:44 +0530 | [diff] [blame] | 141 | self._internal_tokens_prune("tokens") |
| 142 | elif config["authentication"]["backend"] == "tacacs": |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 143 | self.backend = AuthconnTacacs( |
| 144 | self.config["authentication"], self.db, self.role_permissions |
| 145 | ) |
| K Sai Kiran | 7ddb073 | 2020-10-30 11:14:44 +0530 | [diff] [blame] | 146 | self._internal_tokens_prune("tokens_tacacs") |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 147 | else: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 148 | raise AuthException( |
| 149 | "Unknown authentication backend: {}".format( |
| 150 | config["authentication"]["backend"] |
| 151 | ) |
| 152 | ) |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 153 | |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 154 | if not self.roles_to_operations_file: |
| 155 | if "roles_to_operations" in config["rbac"]: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 156 | self.roles_to_operations_file = config["rbac"][ |
| 157 | "roles_to_operations" |
| 158 | ] |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 159 | else: |
| Eduardo Sousa | 5c01e19 | 2019-05-08 02:35:47 +0100 | [diff] [blame] | 160 | possible_paths = ( |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 161 | __file__[: __file__.rfind("auth.py")] |
| 162 | + "roles_to_operations.yml", |
| 163 | "./roles_to_operations.yml", |
| Eduardo Sousa | 5c01e19 | 2019-05-08 02:35:47 +0100 | [diff] [blame] | 164 | ) |
| 165 | for config_file in possible_paths: |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 166 | if path.isfile(config_file): |
| 167 | self.roles_to_operations_file = config_file |
| 168 | break |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 169 | if not self.roles_to_operations_file: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 170 | raise AuthException( |
| 171 | "Invalid permission configuration: roles_to_operations file missing" |
| 172 | ) |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 173 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 174 | # load role_permissions |
| 175 | def load_role_permissions(method_dict): |
| 176 | for k in method_dict: |
| 177 | if k == "ROLE_PERMISSION": |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 178 | for method in chain( |
| 179 | method_dict.get("METHODS", ()), method_dict.get("TODO", ()) |
| 180 | ): |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 181 | permission = method_dict["ROLE_PERMISSION"] + method.lower() |
| 182 | if permission not in self.role_permissions: |
| 183 | self.role_permissions.append(permission) |
| 184 | elif k in ("TODO", "METHODS"): |
| 185 | continue |
| tierno | 74b5358 | 2020-06-18 10:52:37 +0000 | [diff] [blame] | 186 | elif method_dict[k]: |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 187 | load_role_permissions(method_dict[k]) |
| 188 | |
| 189 | load_role_permissions(self.valid_methods) |
| 190 | for query_string in self.valid_query_string: |
| 191 | for method in ("get", "put", "patch", "post", "delete"): |
| 192 | permission = query_string.lower() + ":" + method |
| 193 | if permission not in self.role_permissions: |
| 194 | self.role_permissions.append(permission) |
| 195 | |
| tierno | d4a705a | 2020-06-22 10:58:26 +0000 | [diff] [blame] | 196 | # get ids of role system_admin and test project |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 197 | role_system_admin = self.db.get_one( |
| 198 | "roles", {"name": "system_admin"}, fail_on_empty=False |
| 199 | ) |
| tierno | d4a705a | 2020-06-22 10:58:26 +0000 | [diff] [blame] | 200 | if role_system_admin: |
| 201 | self.system_admin_role_id = role_system_admin["_id"] |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 202 | test_project_name = self.config["authentication"].get( |
| 203 | "project_not_authorized", "admin" |
| 204 | ) |
| 205 | test_project = self.db.get_one( |
| 206 | "projects", {"name": test_project_name}, fail_on_empty=False |
| 207 | ) |
| tierno | d4a705a | 2020-06-22 10:58:26 +0000 | [diff] [blame] | 208 | if test_project: |
| 209 | self.test_project_id = test_project["_id"] |
| 210 | |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 211 | except Exception as e: |
| 212 | raise AuthException(str(e)) |
| 213 | |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 214 | def stop(self): |
| 215 | try: |
| 216 | if self.db: |
| 217 | self.db.db_disconnect() |
| 218 | except DbException as e: |
| 219 | raise AuthException(str(e), http_code=e.http_code) |
| 220 | |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 221 | def create_admin_project(self): |
| 222 | """ |
| 223 | Creates a new project 'admin' into database if it doesn't exist. Useful for initialization. |
| 224 | :return: _id identity of the 'admin' project |
| 225 | """ |
| 226 | |
| 227 | # projects = self.db.get_one("projects", fail_on_empty=False, fail_on_more=False) |
| 228 | project_desc = {"name": "admin"} |
| 229 | projects = self.backend.get_project_list(project_desc) |
| 230 | if projects: |
| 231 | return projects[0]["_id"] |
| 232 | now = time() |
| 233 | project_desc["_id"] = str(uuid4()) |
| 234 | project_desc["_admin"] = {"created": now, "modified": now} |
| 235 | pid = self.backend.create_project(project_desc) |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 236 | self.logger.info( |
| 237 | "Project '{}' created at database".format(project_desc["name"]) |
| 238 | ) |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 239 | return pid |
| 240 | |
| 241 | def create_admin_user(self, project_id): |
| 242 | """ |
| 243 | Creates a new user admin/admin into database if database is empty. Useful for initialization |
| 244 | :return: _id identity of the inserted data, or None |
| 245 | """ |
| 246 | # users = self.db.get_one("users", fail_on_empty=False, fail_on_more=False) |
| 247 | users = self.backend.get_user_list() |
| 248 | if users: |
| 249 | return None |
| 250 | # user_desc = {"username": "admin", "password": "admin", "projects": [project_id]} |
| 251 | now = time() |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 252 | user_desc = { |
| 253 | "username": "admin", |
| 254 | "password": "admin", |
| garciadeblas | 6d83f8f | 2023-06-19 22:34:49 +0200 | [diff] [blame] | 255 | "_admin": {"created": now, "modified": now, "user_status": "always-active"}, |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 256 | } |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 257 | if project_id: |
| 258 | pid = project_id |
| 259 | else: |
| 260 | # proj = self.db.get_one("projects", {"name": "admin"}, fail_on_empty=False, fail_on_more=False) |
| 261 | proj = self.backend.get_project_list({"name": "admin"}) |
| 262 | pid = proj[0]["_id"] if proj else None |
| 263 | # role = self.db.get_one("roles", {"name": "system_admin"}, fail_on_empty=False, fail_on_more=False) |
| 264 | roles = self.backend.get_role_list({"name": "system_admin"}) |
| 265 | if pid and roles: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 266 | user_desc["project_role_mappings"] = [ |
| 267 | {"project": pid, "role": roles[0]["_id"]} |
| 268 | ] |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 269 | uid = self.backend.create_user(user_desc) |
| 270 | self.logger.info("User '{}' created at database".format(user_desc["username"])) |
| 271 | return uid |
| 272 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 273 | def init_db(self, target_version="1.0"): |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 274 | """ |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 275 | Check if the database has been initialized, with at least one user. If not, create the required tables |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 276 | and insert the predefined mappings between roles and permissions. |
| 277 | |
| 278 | :param target_version: schema version that should be present in the database. |
| 279 | :return: None if OK, exception if error or version is different. |
| 280 | """ |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 281 | |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 282 | records = self.backend.get_role_list() |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 283 | |
| tierno | c23a9bb | 2020-06-24 10:54:11 +0000 | [diff] [blame] | 284 | # Loading permissions to AUTH. At lease system_admin must be present. |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 285 | if not records or not next( |
| 286 | (r for r in records if r["name"] == "system_admin"), None |
| 287 | ): |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 288 | with open(self.roles_to_operations_file, "r") as stream: |
| garciadeblas | 4cd875d | 2023-02-14 19:05:34 +0100 | [diff] [blame] | 289 | roles_to_operations_yaml = yaml.safe_load(stream) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 290 | |
| tierno | 1f029d8 | 2019-06-13 22:37:04 +0000 | [diff] [blame] | 291 | role_names = [] |
| 292 | for role_with_operations in roles_to_operations_yaml["roles"]: |
| 293 | # Verifying if role already exists. If it does, raise exception |
| 294 | if role_with_operations["name"] not in role_names: |
| 295 | role_names.append(role_with_operations["name"]) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 296 | else: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 297 | raise AuthException( |
| 298 | "Duplicated role name '{}' at file '{}''".format( |
| 299 | role_with_operations["name"], self.roles_to_operations_file |
| 300 | ) |
| 301 | ) |
| tierno | 1f029d8 | 2019-06-13 22:37:04 +0000 | [diff] [blame] | 302 | |
| 303 | if not role_with_operations["permissions"]: |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 304 | continue |
| 305 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 306 | for permission, is_allowed in role_with_operations[ |
| 307 | "permissions" |
| 308 | ].items(): |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 309 | if not isinstance(is_allowed, bool): |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 310 | raise AuthException( |
| 311 | "Invalid value for permission '{}' at role '{}'; at file '{}'".format( |
| 312 | permission, |
| 313 | role_with_operations["name"], |
| 314 | self.roles_to_operations_file, |
| 315 | ) |
| 316 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 317 | |
| tierno | c23a9bb | 2020-06-24 10:54:11 +0000 | [diff] [blame] | 318 | # TODO check permission is ok |
| tierno | 1f029d8 | 2019-06-13 22:37:04 +0000 | [diff] [blame] | 319 | if permission[-1] == ":": |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 320 | raise AuthException( |
| 321 | "Invalid permission '{}' terminated in ':' for role '{}'; at file {}".format( |
| 322 | permission, |
| 323 | role_with_operations["name"], |
| 324 | self.roles_to_operations_file, |
| 325 | ) |
| 326 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 327 | |
| tierno | 1f029d8 | 2019-06-13 22:37:04 +0000 | [diff] [blame] | 328 | if "default" not in role_with_operations["permissions"]: |
| 329 | role_with_operations["permissions"]["default"] = False |
| 330 | if "admin" not in role_with_operations["permissions"]: |
| 331 | role_with_operations["permissions"]["admin"] = False |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 332 | |
| 333 | now = time() |
| tierno | 1f029d8 | 2019-06-13 22:37:04 +0000 | [diff] [blame] | 334 | role_with_operations["_admin"] = { |
| 335 | "created": now, |
| 336 | "modified": now, |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 337 | } |
| 338 | |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 339 | # self.db.create(self.roles_to_operations_table, role_with_operations) |
| tierno | c23a9bb | 2020-06-24 10:54:11 +0000 | [diff] [blame] | 340 | try: |
| 341 | self.backend.create_role(role_with_operations) |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 342 | self.logger.info( |
| 343 | "Role '{}' created".format(role_with_operations["name"]) |
| 344 | ) |
| tierno | c23a9bb | 2020-06-24 10:54:11 +0000 | [diff] [blame] | 345 | except (AuthException, AuthconnException) as e: |
| 346 | if role_with_operations["name"] == "system_admin": |
| 347 | raise |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 348 | self.logger.error( |
| 349 | "Role '{}' cannot be created: {}".format( |
| 350 | role_with_operations["name"], e |
| 351 | ) |
| 352 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 353 | |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 354 | # Create admin project&user if required |
| 355 | pid = self.create_admin_project() |
| tierno | 9ebbf85 | 2019-09-03 14:58:09 +0000 | [diff] [blame] | 356 | user_id = self.create_admin_user(pid) |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 357 | |
| tierno | 9ebbf85 | 2019-09-03 14:58:09 +0000 | [diff] [blame] | 358 | # try to assign system_admin role to user admin if not any user has this role |
| 359 | if not user_id: |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 360 | try: |
| tierno | 9ebbf85 | 2019-09-03 14:58:09 +0000 | [diff] [blame] | 361 | users = self.backend.get_user_list() |
| 362 | roles = self.backend.get_role_list({"name": "system_admin"}) |
| 363 | role_id = roles[0]["_id"] |
| 364 | user_with_system_admin = False |
| 365 | user_admin_id = None |
| 366 | for user in users: |
| 367 | if not user_admin_id: |
| 368 | user_admin_id = user["_id"] |
| 369 | if user["username"] == "admin": |
| 370 | user_admin_id = user["_id"] |
| 371 | for prm in user.get("project_role_mappings", ()): |
| 372 | if prm["role"] == role_id: |
| 373 | user_with_system_admin = True |
| 374 | break |
| 375 | if user_with_system_admin: |
| 376 | break |
| 377 | if not user_with_system_admin: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 378 | self.backend.update_user( |
| 379 | { |
| 380 | "_id": user_admin_id, |
| 381 | "add_project_role_mappings": [ |
| 382 | {"project": pid, "role": role_id} |
| 383 | ], |
| 384 | } |
| 385 | ) |
| 386 | self.logger.info( |
| 387 | "Added role system admin to user='{}' project=admin".format( |
| 388 | user_admin_id |
| 389 | ) |
| 390 | ) |
| delacruzramo | 15ec706 | 2019-12-26 10:09:04 +0000 | [diff] [blame] | 391 | except Exception as e: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 392 | self.logger.error( |
| 393 | "Error in Authorization DataBase initialization: {}: {}".format( |
| 394 | type(e).__name__, e |
| 395 | ) |
| 396 | ) |
| tierno | 1f029d8 | 2019-06-13 22:37:04 +0000 | [diff] [blame] | 397 | |
| 398 | self.load_operation_to_allowed_roles() |
| 399 | |
| 400 | def load_operation_to_allowed_roles(self): |
| 401 | """ |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 402 | Fills the internal self.operation_to_allowed_roles based on database role content and self.role_permissions |
| tierno | a6bb45d | 2019-06-14 09:45:39 +0000 | [diff] [blame] | 403 | It works in a shadow copy and replace at the end to allow other threads working with the old copy |
| tierno | 1f029d8 | 2019-06-13 22:37:04 +0000 | [diff] [blame] | 404 | :return: None |
| 405 | """ |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 406 | permissions = {oper: [] for oper in self.role_permissions} |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 407 | # records = self.db.get_list(self.roles_to_operations_table) |
| 408 | records = self.backend.get_role_list() |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 409 | |
| tierno | a6bb45d | 2019-06-14 09:45:39 +0000 | [diff] [blame] | 410 | ignore_fields = ["_id", "_admin", "name", "default"] |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 411 | for record in records: |
| delacruzramo | 01b15d3 | 2019-07-02 14:37:47 +0200 | [diff] [blame] | 412 | if not record.get("permissions"): |
| 413 | continue |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 414 | record_permissions = { |
| 415 | oper: record["permissions"].get("default", False) |
| 416 | for oper in self.role_permissions |
| 417 | } |
| 418 | operations_joined = [ |
| 419 | (oper, value) |
| 420 | for oper, value in record["permissions"].items() |
| 421 | if oper not in ignore_fields |
| 422 | ] |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 423 | operations_joined.sort(key=lambda x: x[0].count(":")) |
| 424 | |
| 425 | for oper in operations_joined: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 426 | match = list( |
| 427 | filter(lambda x: x.find(oper[0]) == 0, record_permissions.keys()) |
| 428 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 429 | |
| 430 | for m in match: |
| 431 | record_permissions[m] = oper[1] |
| 432 | |
| 433 | allowed_operations = [k for k, v in record_permissions.items() if v is True] |
| 434 | |
| 435 | for allowed_op in allowed_operations: |
| Eduardo Sousa | 5c01e19 | 2019-05-08 02:35:47 +0100 | [diff] [blame] | 436 | permissions[allowed_op].append(record["name"]) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 437 | |
| tierno | a6bb45d | 2019-06-14 09:45:39 +0000 | [diff] [blame] | 438 | self.operation_to_allowed_roles = permissions |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 439 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 440 | def authorize( |
| 441 | self, role_permission=None, query_string_operations=None, item_id=None |
| 442 | ): |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 443 | token = None |
| 444 | user_passwd64 = None |
| 445 | try: |
| 446 | # 1. Get token Authorization bearer |
| 447 | auth = cherrypy.request.headers.get("Authorization") |
| 448 | if auth: |
| 449 | auth_list = auth.split(" ") |
| 450 | if auth_list[0].lower() == "bearer": |
| 451 | token = auth_list[-1] |
| 452 | elif auth_list[0].lower() == "basic": |
| 453 | user_passwd64 = auth_list[-1] |
| 454 | if not token: |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 455 | if cherrypy.session.get("Authorization"): # pylint: disable=E1101 |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 456 | # 2. Try using session before request a new token. If not, basic authentication will generate |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 457 | token = cherrypy.session.get( # pylint: disable=E1101 |
| 458 | "Authorization" |
| 459 | ) |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 460 | if token == "logout": |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 461 | token = None # force Unauthorized response to insert user password again |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 462 | elif user_passwd64 and cherrypy.request.config.get( |
| 463 | "auth.allow_basic_authentication" |
| 464 | ): |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 465 | # 3. Get new token from user password |
| 466 | user = None |
| 467 | passwd = None |
| 468 | try: |
| 469 | user_passwd = standard_b64decode(user_passwd64).decode() |
| 470 | user, _, passwd = user_passwd.partition(":") |
| 471 | except Exception: |
| 472 | pass |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 473 | outdata = self.new_token( |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 474 | None, {"username": user, "password": passwd}, None |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 475 | ) |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 476 | token = outdata["_id"] |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 477 | cherrypy.session["Authorization"] = token # pylint: disable=E1101 |
| tierno | a6bb45d | 2019-06-14 09:45:39 +0000 | [diff] [blame] | 478 | |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 479 | if not token: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 480 | raise AuthException( |
| 481 | "Needed a token or Authorization http header", |
| 482 | http_code=HTTPStatus.UNAUTHORIZED, |
| 483 | ) |
| delacruzramo | ad682a5 | 2019-12-10 16:26:34 +0100 | [diff] [blame] | 484 | |
| 485 | # try to get from cache first |
| 486 | now = time() |
| 487 | token_info = self.tokens_cache.get(token) |
| 488 | if token_info and token_info["expires"] < now: |
| 489 | # delete token. MUST be done with care, as another thread maybe already delete it. Do not use del |
| 490 | self.tokens_cache.pop(token, None) |
| 491 | token_info = None |
| 492 | |
| 493 | # get from database if not in cache |
| 494 | if not token_info: |
| 495 | token_info = self.backend.validate_token(token) |
| 496 | # Clear cache if token limit reached |
| 497 | if len(self.tokens_cache) > self.token_limit: |
| 498 | self.tokens_cache.clear() |
| 499 | self.tokens_cache[token] = token_info |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 500 | # TODO add to token info remote host, port |
| 501 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 502 | if role_permission: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 503 | RBAC_auth = self.check_permissions( |
| 504 | token_info, |
| 505 | cherrypy.request.method, |
| 506 | role_permission, |
| 507 | query_string_operations, |
| 508 | item_id, |
| 509 | ) |
| sousaedu | 60bf895 | 2021-07-08 17:17:23 +0200 | [diff] [blame] | 510 | self.logger.info("RBAC_auth: {}".format(RBAC_auth)) |
| elumalai | 7802ff8 | 2023-04-24 20:38:32 +0530 | [diff] [blame] | 511 | if RBAC_auth: |
| 512 | cef_event( |
| 513 | self.cef_logger, |
| 514 | { |
| 515 | "name": "System Access", |
| 516 | "sourceUserName": token_info.get("username"), |
| 517 | "message": "Accessing account with system privileges, Project={}".format( |
| 518 | token_info.get("project_name") |
| 519 | ), |
| 520 | }, |
| 521 | ) |
| 522 | self.logger.info("{}".format(self.cef_logger)) |
| delacruzramo | 029405d | 2019-09-26 10:52:56 +0200 | [diff] [blame] | 523 | token_info["allow_show_user_project_role"] = RBAC_auth |
| 524 | |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 525 | return token_info |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 526 | except AuthException as e: |
| tierno | c844536 | 2019-06-14 12:07:15 +0000 | [diff] [blame] | 527 | if not isinstance(e, AuthExceptionUnauthorized): |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 528 | if cherrypy.session.get("Authorization"): # pylint: disable=E1101 |
| 529 | del cherrypy.session["Authorization"] # pylint: disable=E1101 |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 530 | cherrypy.response.headers[ |
| 531 | "WWW-Authenticate" |
| 532 | ] = 'Bearer realm="{}"'.format(e) |
| tierno | d4a705a | 2020-06-22 10:58:26 +0000 | [diff] [blame] | 533 | if self.config["authentication"].get("user_not_authorized"): |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 534 | return { |
| 535 | "id": "testing-token", |
| 536 | "_id": "testing-token", |
| 537 | "project_id": self.test_project_id, |
| 538 | "username": self.config["authentication"]["user_not_authorized"], |
| 539 | "roles": [self.system_admin_role_id], |
| 540 | "admin": True, |
| 541 | "allow_show_user_project_role": True, |
| 542 | } |
| tierno | c844536 | 2019-06-14 12:07:15 +0000 | [diff] [blame] | 543 | raise |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 544 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 545 | def new_token(self, token_info, indata, remote): |
| 546 | new_token_info = self.backend.authenticate( |
| tierno | 6486f74 | 2020-02-13 16:30:14 +0000 | [diff] [blame] | 547 | credentials=indata, |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 548 | token_info=token_info, |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 549 | ) |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 550 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 551 | new_token_info["remote_port"] = remote.port |
| 552 | if not new_token_info.get("expires"): |
| 553 | new_token_info["expires"] = time() + 3600 |
| 554 | if not new_token_info.get("admin"): |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 555 | new_token_info["admin"] = ( |
| 556 | True if new_token_info.get("project_name") == "admin" else False |
| 557 | ) |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 558 | # TODO put admin in RBAC |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 559 | |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 560 | if remote.name: |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 561 | new_token_info["remote_host"] = remote.name |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 562 | elif remote.ip: |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 563 | new_token_info["remote_host"] = remote.ip |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 564 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 565 | # TODO call self._internal_tokens_prune(now) ? |
| 566 | return deepcopy(new_token_info) |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 567 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 568 | def get_token_list(self, token_info): |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 569 | if self.config["authentication"]["backend"] == "internal": |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 570 | return self._internal_get_token_list(token_info) |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 571 | else: |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 572 | # TODO: check if this can be avoided. Backend may provide enough information |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 573 | return [ |
| 574 | deepcopy(token) |
| 575 | for token in self.tokens_cache.values() |
| 576 | if token["username"] == token_info["username"] |
| 577 | ] |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 578 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 579 | def get_token(self, token_info, token): |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 580 | if self.config["authentication"]["backend"] == "internal": |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 581 | return self._internal_get_token(token_info, token) |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 582 | else: |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 583 | # TODO: check if this can be avoided. Backend may provide enough information |
| 584 | token_value = self.tokens_cache.get(token) |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 585 | if not token_value: |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 586 | raise AuthException("token not found", http_code=HTTPStatus.NOT_FOUND) |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 587 | if ( |
| 588 | token_value["username"] != token_info["username"] |
| 589 | and not token_info["admin"] |
| 590 | ): |
| 591 | raise AuthException( |
| 592 | "needed admin privileges", http_code=HTTPStatus.UNAUTHORIZED |
| 593 | ) |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 594 | return token_value |
| Eduardo Sousa | 2f98821 | 2018-07-26 01:04:11 +0100 | [diff] [blame] | 595 | |
| Eduardo Sousa | 819d34c | 2018-07-31 01:20:02 +0100 | [diff] [blame] | 596 | def del_token(self, token): |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 597 | try: |
| 598 | self.backend.revoke_token(token) |
| delacruzramo | ad682a5 | 2019-12-10 16:26:34 +0100 | [diff] [blame] | 599 | # self.tokens_cache.pop(token, None) |
| 600 | self.remove_token_from_cache(token) |
| delacruzramo | ceb8baf | 2019-06-21 14:25:38 +0200 | [diff] [blame] | 601 | return "token '{}' deleted".format(token) |
| 602 | except KeyError: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 603 | raise AuthException( |
| 604 | "Token '{}' not found".format(token), http_code=HTTPStatus.NOT_FOUND |
| 605 | ) |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 606 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 607 | def check_permissions( |
| 608 | self, |
| 609 | token_info, |
| 610 | method, |
| 611 | role_permission=None, |
| 612 | query_string_operations=None, |
| 613 | item_id=None, |
| 614 | ): |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 615 | """ |
| 616 | Checks that operation has permissions to be done, base on the assigned roles to this user project |
| 617 | :param token_info: Dictionary that contains "roles" with a list of assigned roles. |
| 618 | This method fills the token_info["admin"] with True or False based on assigned tokens, if any allows admin |
| 619 | This will be used among others to hide or not the _admin content of topics |
| 620 | :param method: GET,PUT, POST, ... |
| 621 | :param role_permission: role permission name of the operation required |
| 622 | :param query_string_operations: list of possible admin query strings provided by user. It is checked that the |
| 623 | assigned role allows this query string for this method |
| delacruzramo | 029405d | 2019-09-26 10:52:56 +0200 | [diff] [blame] | 624 | :param item_id: item identifier if included in the URL, None otherwise |
| 625 | :return: True if access granted by permission rules, False if access granted by default rules (Bug 853) |
| 626 | :raises: AuthExceptionUnauthorized if access denied |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 627 | """ |
| sousaedu | 490d019 | 2021-05-05 12:48:16 +0200 | [diff] [blame] | 628 | self.load_operation_to_allowed_roles() |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 629 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 630 | roles_required = self.operation_to_allowed_roles[role_permission] |
| 631 | roles_allowed = [role["name"] for role in token_info["roles"]] |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 632 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 633 | # fills token_info["admin"] if some roles allows it |
| 634 | token_info["admin"] = False |
| tierno | a6bb45d | 2019-06-14 09:45:39 +0000 | [diff] [blame] | 635 | for role in roles_allowed: |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 636 | if role in self.operation_to_allowed_roles["admin:" + method.lower()]: |
| 637 | token_info["admin"] = True |
| tierno | a6bb45d | 2019-06-14 09:45:39 +0000 | [diff] [blame] | 638 | break |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 639 | |
| 640 | if "anonymous" in roles_required: |
| delacruzramo | 029405d | 2019-09-26 10:52:56 +0200 | [diff] [blame] | 641 | return True |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 642 | operation_allowed = False |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 643 | for role in roles_allowed: |
| 644 | if role in roles_required: |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 645 | operation_allowed = True |
| 646 | # if query_string operations, check if this role allows it |
| 647 | if not query_string_operations: |
| delacruzramo | 029405d | 2019-09-26 10:52:56 +0200 | [diff] [blame] | 648 | return True |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 649 | for query_string_operation in query_string_operations: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 650 | if ( |
| 651 | role |
| 652 | not in self.operation_to_allowed_roles[query_string_operation] |
| 653 | ): |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 654 | break |
| 655 | else: |
| delacruzramo | 029405d | 2019-09-26 10:52:56 +0200 | [diff] [blame] | 656 | return True |
| 657 | |
| 658 | # Bug 853 - Final Solution |
| 659 | # User/Project/Role whole listings are filtered elsewhere |
| 660 | # uid, pid, rid = ("user_id", "project_id", "id") if is_valid_uuid(id) else ("username", "project_name", "name") |
| 661 | uid = "user_id" if is_valid_uuid(item_id) else "username" |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 662 | if ( |
| 663 | role_permission |
| 664 | in [ |
| 665 | "projects:get", |
| 666 | "projects:id:get", |
| 667 | "roles:get", |
| 668 | "roles:id:get", |
| 669 | "users:get", |
| 670 | ] |
| 671 | ) or (role_permission == "users:id:get" and item_id == token_info[uid]): |
| delacruzramo | 029405d | 2019-09-26 10:52:56 +0200 | [diff] [blame] | 672 | # or (role_permission == "projects:id:get" and item_id == token_info[pid]) \ |
| 673 | # or (role_permission == "roles:id:get" and item_id in [role[rid] for role in token_info["roles"]]): |
| 674 | return False |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 675 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 676 | if not operation_allowed: |
| 677 | raise AuthExceptionUnauthorized("Access denied: lack of permissions.") |
| 678 | else: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 679 | raise AuthExceptionUnauthorized( |
| 680 | "Access denied: You have not permissions to use these admin query string" |
| 681 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 682 | |
| Eduardo Sousa | 5c01e19 | 2019-05-08 02:35:47 +0100 | [diff] [blame] | 683 | def get_user_list(self): |
| 684 | return self.backend.get_user_list() |
| 685 | |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 686 | def _normalize_url(self, url, method): |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 687 | # DEPRECATED !!! |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 688 | # Removing query strings |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 689 | normalized_url = url if "?" not in url else url[: url.find("?")] |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 690 | normalized_url_splitted = normalized_url.split("/") |
| 691 | parameters = {} |
| 692 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 693 | filtered_keys = [ |
| 694 | key |
| 695 | for key in self.resources_to_operations_mapping.keys() |
| 696 | if method in key.split()[0] |
| 697 | ] |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 698 | |
| 699 | for idx, path_part in enumerate(normalized_url_splitted): |
| 700 | tmp_keys = [] |
| 701 | for tmp_key in filtered_keys: |
| 702 | splitted = tmp_key.split()[1].split("/") |
| Eduardo Sousa | cc02e9a | 2019-03-20 17:32:36 +0000 | [diff] [blame] | 703 | if idx >= len(splitted): |
| 704 | continue |
| 705 | elif "<" in splitted[idx] and ">" in splitted[idx]: |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 706 | if splitted[idx] == "<artifactPath>": |
| 707 | tmp_keys.append(tmp_key) |
| 708 | continue |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 709 | elif idx == len(normalized_url_splitted) - 1 and len( |
| 710 | normalized_url_splitted |
| 711 | ) != len(splitted): |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 712 | continue |
| 713 | else: |
| 714 | tmp_keys.append(tmp_key) |
| 715 | elif splitted[idx] == path_part: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 716 | if idx == len(normalized_url_splitted) - 1 and len( |
| 717 | normalized_url_splitted |
| 718 | ) != len(splitted): |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 719 | continue |
| 720 | else: |
| 721 | tmp_keys.append(tmp_key) |
| 722 | filtered_keys = tmp_keys |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 723 | if ( |
| 724 | len(filtered_keys) == 1 |
| 725 | and filtered_keys[0].split("/")[-1] == "<artifactPath>" |
| 726 | ): |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 727 | break |
| 728 | |
| 729 | if len(filtered_keys) == 0: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 730 | raise AuthException( |
| 731 | "Cannot make an authorization decision. URL not found. URL: {0}".format( |
| 732 | url |
| 733 | ) |
| 734 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 735 | elif len(filtered_keys) > 1: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 736 | raise AuthException( |
| 737 | "Cannot make an authorization decision. Multiple URLs found. URL: {0}".format( |
| 738 | url |
| 739 | ) |
| 740 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 741 | |
| 742 | filtered_key = filtered_keys[0] |
| 743 | |
| 744 | for idx, path_part in enumerate(filtered_key.split()[1].split("/")): |
| 745 | if "<" in path_part and ">" in path_part: |
| 746 | if path_part == "<artifactPath>": |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 747 | parameters[path_part[1:-1]] = "/".join( |
| 748 | normalized_url_splitted[idx:] |
| 749 | ) |
| Eduardo Sousa | 29933fc | 2018-11-14 06:36:35 +0000 | [diff] [blame] | 750 | else: |
| 751 | parameters[path_part[1:-1]] = normalized_url_splitted[idx] |
| 752 | |
| 753 | return filtered_key, parameters |
| 754 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 755 | def _internal_get_token_list(self, token_info): |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 756 | now = time() |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 757 | token_list = self.db.get_list( |
| 758 | "tokens", {"username": token_info["username"], "expires.gt": now} |
| 759 | ) |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 760 | return token_list |
| 761 | |
| tierno | 701018c | 2019-06-25 11:13:14 +0000 | [diff] [blame] | 762 | def _internal_get_token(self, token_info, token_id): |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 763 | token_value = self.db.get_one("tokens", {"_id": token_id}, fail_on_empty=False) |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 764 | if not token_value: |
| 765 | raise AuthException("token not found", http_code=HTTPStatus.NOT_FOUND) |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 766 | if ( |
| 767 | token_value["username"] != token_info["username"] |
| 768 | and not token_info["admin"] |
| 769 | ): |
| 770 | raise AuthException( |
| 771 | "needed admin privileges", http_code=HTTPStatus.UNAUTHORIZED |
| 772 | ) |
| Eduardo Sousa | d1b525d | 2018-10-04 04:24:18 +0100 | [diff] [blame] | 773 | return token_value |
| 774 | |
| K Sai Kiran | 7ddb073 | 2020-10-30 11:14:44 +0530 | [diff] [blame] | 775 | def _internal_tokens_prune(self, token_collection, now=None): |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 776 | now = now or time() |
| 777 | if not self.next_db_prune_time or self.next_db_prune_time >= now: |
| K Sai Kiran | 7ddb073 | 2020-10-30 11:14:44 +0530 | [diff] [blame] | 778 | self.db.del_list(token_collection, {"expires.lt": now}) |
| tierno | 0ea204e | 2019-01-25 14:16:24 +0000 | [diff] [blame] | 779 | self.next_db_prune_time = self.periodin_db_pruning + now |
| delacruzramo | ad682a5 | 2019-12-10 16:26:34 +0100 | [diff] [blame] | 780 | # self.tokens_cache.clear() # not required any more |
| 781 | |
| 782 | def remove_token_from_cache(self, token=None): |
| 783 | if token: |
| 784 | self.tokens_cache.pop(token, None) |
| 785 | else: |
| 786 | self.tokens_cache.clear() |
| 787 | self.msg.write("admin", "revoke_token", {"_id": token} if token else None) |
| selvi.j | a9a1fc8 | 2022-04-04 06:54:30 +0000 | [diff] [blame] | 788 | |
| 789 | def check_password_expiry(self, outdata): |
| 790 | """ |
| 791 | This method will check for password expiry of the user |
| 792 | :param outdata: user token information |
| 793 | """ |
| garciadeblas | 6d83f8f | 2023-06-19 22:34:49 +0200 | [diff] [blame] | 794 | user_list = None |
| selvi.j | a9a1fc8 | 2022-04-04 06:54:30 +0000 | [diff] [blame] | 795 | present_time = time() |
| 796 | user = outdata["username"] |
| garciadeblas | 6d83f8f | 2023-06-19 22:34:49 +0200 | [diff] [blame] | 797 | if self.config["authentication"].get("user_management"): |
| 798 | user_list = self.db.get_list("users", {"username": user}) |
| 799 | if user_list: |
| 800 | user_content = user_list[0] |
| 801 | if not user_content.get("username") == "admin": |
| 802 | user_content["_admin"]["modified"] = present_time |
| 803 | if user_content.get("_admin").get("password_expire_time"): |
| 804 | password_expire_time = user_content["_admin"][ |
| 805 | "password_expire_time" |
| 806 | ] |
| 807 | else: |
| 808 | password_expire_time = present_time |
| 809 | uid = user_content["_id"] |
| 810 | self.db.set_one("users", {"_id": uid}, user_content) |
| 811 | if not present_time < password_expire_time: |
| 812 | return True |
| selvi.j | a9a1fc8 | 2022-04-04 06:54:30 +0000 | [diff] [blame] | 813 | else: |
| 814 | pass |