# 2017-02-23 Cornelius Kölbel <cornelius.koelbel@netknights.it>
# Add certificate templates to the list of CA connector
# 2015-04-23 Cornelius Kölbel <cornelius.koelbel@netknights.it>
# Initial writup
# This code is inspired by the resolver.py which was forked from
# LinOTP, which was originally (c) by LSE Leading Security Experts.
#
# License: AGPLv3
#
# This code is free software; you can redistribute it and/or
# modify it under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE
# License as published by the Free Software Foundation; either
# version 3 of the License, or any later version.
#
# This code is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU AFFERO GENERAL PUBLIC LICENSE for more details.
#
# You should have received a copy of the GNU Affero General Public
# License along with this program. If not, see <http://www.gnu.org/licenses/>.
#
#
__doc__ = """This the library for handling CA connector definitions which are
stored in the database table "caconnector".
The code is tested in tests/test_lib_caconnector.py.
"""
import logging
from sqlalchemy import func, select, update
from privacyidea.lib.config import get_config_object
from privacyidea.lib.utils import (sanity_name_check, get_data_from_params, fetch_one_resource)
from privacyidea.lib.utils.export import (register_import, register_export)
from .config import (get_caconnector_types,
get_caconnector_class_dict)
from .crypto import encryptPassword, decryptPassword
from .log import log_with
from ..api.lib.utils import get_required
from ..models import (CAConnector,
CAConnectorConfig, db, save_config_timestamp)
log = logging.getLogger(__name__)
[docs]
@log_with(log)
def save_caconnector(params: dict) -> int:
"""
Create a new CA connector from the given parameters and save it to the
database.
If the CA Connector already exists, it is updated.
For updating some attributes of an existing CA connector you do not need
to pass all attributes again, but only those, which should be changed.
When updating the CA connector the type must not be changed,
since another type might require different attributes.
:param params: request parameters like "caconnector" (name) and "type"
and the specific attributes of the ca connector.
:return: the database ID of the CA connector
"""
# before we create the resolver in the database, we need to check
# for the name and type thing...
connector_name = get_required(params, 'caconnector')
connector_type = get_required(params, 'type')
update_connector = False
existing_connector_name = None
sanity_name_check(connector_name)
# check the type
if connector_type not in get_caconnector_types():
raise Exception("connector type : {0!s} not in {1!s}".format(connector_type, get_caconnector_types()))
# check the name
connectors = get_caconnector_list(filter_caconnector_name=connector_name)
for connector in connectors:
if connector.get("type") == connector_type:
# There is a CA connector with the same name, so we will update
# the CA Connector config
update_connector = True
existing_connector_name = connector.get("connectorname")
else: # pragma: no cover
raise Exception("CA Connector with similar name and other type "
"already exists: %s" % connector_name)
# create a dictionary for the ResolverConfig
connector_config = get_caconnector_config_description(connector_type)
config_description = connector_config.get(connector_type, {})
data, types, desc = get_data_from_params(params,
["caconnector", "type"],
config_description,
"CA connector",
connector_type)
# Everything passed. So lets actually create the CA Connector in the DB
if update_connector:
stmt = select(CAConnector).where(CAConnector.name == existing_connector_name)
connector_id = db.session.scalars(stmt).first().id
else:
db_connector = CAConnector(params.get("caconnector"),
params.get("type"))
db.session.add(db_connector)
db.session.flush() # flush to assign id
connector_id = db_connector.id
# create the config
for key, value in data.items():
if types.get(key) == "password":
value = encryptPassword(value)
stmt = select(CAConnectorConfig).filter_by(caconnector_id=connector_id, Key=key)
config = db.session.execute(stmt).scalar_one_or_none()
if config:
config.Value = value
config.Type = types.get(key, "")
config.Description = desc.get(key, "")
else:
config = CAConnectorConfig(caconnector_id=connector_id,
Key=key,
Value=value,
Type=types.get(key, ""),
Description=desc.get(key, ""))
db.session.add(config)
db.session.commit()
save_config_timestamp()
return connector_id
[docs]
def get_caconnector_list(filter_caconnector_type=None,
filter_caconnector_name=None,
return_config=True):
"""
Gets the list of configured CA Connectors from the database
:param filter_caconnector_type: Only CA connectors of the given type are
returned
:type filter_caconnector_type: string
:param return_config: Whether the configuration should be returned. If False
only the list of the CAconncetor names is returned
:rtype: list of the connectors and their configuration
"""
all_connector_objs = get_all_caconnectors()
Connectors = []
for c in all_connector_objs:
if filter_caconnector_name:
if filter_caconnector_name == c.get("connectorname"):
Connectors.append(c)
elif filter_caconnector_type:
if filter_caconnector_type == c.get("type"):
Connectors.append(c)
else:
Connectors.append(c)
if not return_config:
# We need to copy the dict to avoid destroying the dict-by-reference
reduced_connectors = []
# we strip all config data
for conn in Connectors:
reduced_connector = {}
for key in conn.keys():
if key in ["data"]:
reduced_connector[key] = {}
else:
reduced_connector[key] = conn.get(key)
reduced_connectors.append(reduced_connector)
return reduced_connectors
return Connectors
[docs]
@log_with(log)
def get_caconnector_specific_options(catype, data):
"""
Given the raw data of a CA connector configuration this function returns
a dict of all available specific options to this instance of a CA connector.
:param catype:
:param data:
:return:
"""
c_obj_class = get_caconnector_class(catype)
c_obj = c_obj_class("dummy", data)
return c_obj.get_specific_options()
[docs]
@log_with(log)
def delete_caconnector(connector_name):
"""
delete a CA connector and all related config entries.
If there was no CA connector, that could be deleted, a ResourceNotFoundError is raised.
:param connector_name: The name of the CA connector that is to be deleted
:type connector_name: basestring
:return: The Id of the resolver
:rtype: int
"""
return fetch_one_resource(CAConnector, name=connector_name).delete()
[docs]
@log_with(log)
# @cache.memoize(10)
def get_caconnector_config(connector_name):
"""
return the complete config of a given CA connector from the database
:param connector_name: the name of the CA connector
:type connector_name: basestring
:return: the config of the CA connector
:rtype: dict
"""
conn = get_caconnector_list(filter_caconnector_name=connector_name, return_config=True)
return conn[0].get("data", {})
# @log_with(log)
# @cache.memoize(10)
[docs]
def get_caconnector_config_description(caconnector_type):
"""
Get the description of the configuration of a CA connector
:param caconnector_type: the type of the CA connector like "local"
:type caconnector_type: basestring
:return: configuration description dict
that looks like this::
{'local': {'attribute1': 'string',
'attribute2': 'int'}}
"""
descriptor = None
connector_class = get_caconnector_class(caconnector_type)
if connector_class is not None:
descriptor = connector_class.get_caconnector_description()
return descriptor
# @cache.memoize(10)
[docs]
def get_caconnector_class(connector_type):
"""
Return the class for a given CA connector type.
:param connector_type: The type of the connector
:type connector_type: basestring
:return: CA Connector Class
"""
ret = None
(connector_classes, connector_types) = get_caconnector_class_dict()
if connector_type in connector_types.values():
for k, v in connector_types.items():
if v == connector_type:
ret = connector_classes.get(k)
break
return ret
# @cache.memoize(10)
[docs]
def get_caconnector_type(connector_name):
"""
return the type of CA connector
:param connector_name: The name of the CA connector
:return: The type of the CA connector
:rtype: string
"""
c_type = None
connector_list = get_caconnector_list(filter_caconnector_name=connector_name)
if connector_list:
c_type = connector_list[0].get("type")
return c_type
[docs]
def get_caconnector_object(connector_name):
"""
create a CA Connector object from a connector_name
:param connector_name: the name of the CA connector
:return: instance of the CA Connector with the loaded config
"""
c_obj = None
stmt = select(CAConnector).where(func.lower(CAConnector.name) == connector_name.lower())
connectors = db.session.scalars(stmt).all()
for conn in connectors:
c_obj_class = get_caconnector_class(conn.catype)
if c_obj_class is None:
log.error("unknown CA connector class {0!s} ".format(connector_name))
else:
# create the resolver instance and load the config
c_obj = c_obj_class(connector_name)
if c_obj is not None:
connector_config = {}
for conf in conn.caconfig:
# Handle passwords
value = conf.Value
if conf.Type == "password":
value = decryptPassword(value)
connector_config[conf.Key] = value
c_obj.set_config(connector_config)
if not c_obj:
log.warning("A CA connector with the name {0!s} could not be found!".format(connector_name))
return c_obj
[docs]
@register_export('caconnector')
def export_caconnector(name=None):
""" Export given or all caconnector configuration """
return get_caconnector_list(filter_caconnector_name=name)
[docs]
@register_import('caconnector')
def import_caconnector(data, name=None):
"""Import caconnector configuration"""
log.debug('Import caconnector config: {0!s}'.format(data))
for res_data in data:
if name and name != res_data.get('connectorname'):
continue
# Todo: unfortunately privacyidea delivers 'connectorname' on reading,
# but requires 'caconnector' in save_caconnector.
res_data['caconnector'] = res_data.pop('connectorname')
# also the export saves the caconnector configuration in a data dict
res_data.update(res_data.pop('data'))
rid = save_caconnector(res_data)
log.info('Import of caconnector "{0!s}" finished,'
' id: {1!s}'.format(res_data['caconnector'], rid))
[docs]
def get_all_caconnectors():
"""
Shorthand to retrieve all caconnectors of the request-local config object
"""
conf = get_config_object()
return conf.caconnectors
