305 lines
12 KiB
ReStructuredText
305 lines
12 KiB
ReStructuredText
Writing a Plugin
|
|
================
|
|
|
|
Several interfaces exist for extending Lemur:
|
|
|
|
* Issuer (lemur.plugins.base.issuer)
|
|
* Destination (lemur.plugins.base.destination)
|
|
* Source (lemur.plugins.base.source)
|
|
* Notification (lemur.plugins.base.notification)
|
|
|
|
Each interface has its own functions that will need to be defined in order for
|
|
your plugin to work correctly. See :ref:`Plugin Interfaces <PluginInterfaces>` for details.
|
|
|
|
|
|
Structure
|
|
---------
|
|
|
|
A plugins layout generally looks like the following::
|
|
|
|
setup.py
|
|
lemur_pluginname/
|
|
lemur_pluginname/__init__.py
|
|
lemur_pluginname/plugin.py
|
|
|
|
The ``__init__.py`` file should contain no plugin logic, and at most, a VERSION = 'x.x.x' line. For example,
|
|
if you want to pull the version using pkg_resources (which is what we recommend), your file might contain::
|
|
|
|
try:
|
|
VERSION = __import__('pkg_resources') \
|
|
.get_distribution(__name__).version
|
|
except Exception, e:
|
|
VERSION = 'unknown'
|
|
|
|
Inside of ``plugin.py``, you'll declare your Plugin class::
|
|
|
|
import lemur_pluginname
|
|
from lemur.plugins.base.issuer import IssuerPlugin
|
|
|
|
class PluginName(IssuerPlugin):
|
|
title = 'Plugin Name'
|
|
slug = 'pluginname'
|
|
description = 'My awesome plugin!'
|
|
version = lemur_pluginname.VERSION
|
|
|
|
author = 'Your Name'
|
|
author_url = 'https://github.com/yourname/lemur_pluginname'
|
|
|
|
def widget(self, request, group, **kwargs):
|
|
return "<p>Absolutely useless widget</p>"
|
|
|
|
And you'll register it via ``entry_points`` in your ``setup.py``::
|
|
|
|
setup(
|
|
# ...
|
|
entry_points={
|
|
'lemur.plugins': [
|
|
'pluginname = lemur_pluginname.issuers:PluginName'
|
|
],
|
|
},
|
|
)
|
|
|
|
You can potentially package multiple plugin types in one package, say you want to create a source and
|
|
destination plugins for the same third-party. To accomplish this simply alias the plugin in entry points to point
|
|
at multiple plugins within your package::
|
|
|
|
setup(
|
|
# ...
|
|
entry_points={
|
|
'lemur.plugins': [
|
|
'pluginnamesource = lemur_pluginname.plugin:PluginNameSource',
|
|
'pluginnamedestination = lemur_pluginname.plugin:PluginNameDestination'
|
|
],
|
|
},
|
|
)
|
|
|
|
That's it! Users will be able to install your plugin via ``pip install <package name>``.
|
|
|
|
.. SeeAlso:: For more information about python packages see `Python Packaging <https://packaging.python.org/en/latest/distributing.html>`_
|
|
|
|
.. _PluginInterfaces:
|
|
|
|
Plugin Interfaces
|
|
=================
|
|
|
|
In order to use the interfaces all plugins are required to inherit and override unimplemented functions
|
|
of the parent object.
|
|
|
|
Issuer
|
|
------
|
|
|
|
Issuer plugins are used when you have an external service that creates certificates or authorities.
|
|
In the simple case the third party only issues certificates (Verisign, DigiCert, etc.).
|
|
|
|
If you have a third party or internal service that creates authorities (EJBCA, etc.), Lemur has you covered,
|
|
it can treat any issuer plugin as both a source of creating new certificates as well as new authorities.
|
|
|
|
|
|
The `IssuerPlugin` exposes two functions::
|
|
|
|
def create_certificate(self, options):
|
|
# requests.get('a third party')
|
|
|
|
Lemur will pass a dictionary of all possible options for certificate creation. Including a valid CSR, and the raw options associated with the request.
|
|
|
|
If you wish to be able to create new authorities implement the following function and ensure that the ROOT_CERTIFICATE and the INTERMEDIATE_CERTIFICATE (if any) for the new authority is returned::
|
|
|
|
def create_authority(self, options):
|
|
root_cert, intermediate_cert, username, password = request.get('a third party')
|
|
|
|
# if your provider creates specific credentials for each authority you can associated them with the role associated with the authority
|
|
# these credentials will be provided along with any other options when a certificate is created
|
|
role = dict(username=username, password=password, name='generatedAuthority')
|
|
return root_cert, intermediate_cert, [role]
|
|
|
|
|
|
.. Note::
|
|
Lemur uses PEM formatted certificates as it's internal standard, if you receive certificates in other formats convert them to PEM before returning.
|
|
|
|
|
|
If instead you do not need need to generate authorities but instead use a static authority (Verisign, DigiCert), you can use publicly available constants::
|
|
|
|
|
|
def create_authority(self, options):
|
|
# optionally associate a role with authority to control who can use it
|
|
role = dict(username='', password='', name='exampleAuthority')
|
|
# username and password don't really matter here because we do no need to authenticate our authority against a third party
|
|
return EXAMPLE_ROOT_CERTIFICATE, EXAMPLE_INTERMEDIATE_CERTIFICATE, [role]
|
|
|
|
|
|
.. Note:: You do not need to associate roles to the authority at creation time as they can always be associated after the fact.
|
|
|
|
|
|
The `IssuerPlugin` doesn't have any options like Destination, Source, and Notification plugins. Essentially Lemur **should** already have
|
|
any fields you might need to submit a request to a third party. If there are additional options you need
|
|
in your plugin feel free to open an issue, or look into adding additional options to issuers yourself.
|
|
|
|
Destination
|
|
-----------
|
|
|
|
Destination plugins allow you to propagate certificates managed by Lemur to additional third parties. This provides flexibility when
|
|
different orchestration systems have their own way of manage certificates or there is an existing system you wish to integrate with Lemur.
|
|
|
|
The DestinationPlugin requires only one function to be implemented::
|
|
|
|
def upload(self, cert, private_key, cert_chain, options, **kwargs):
|
|
# request.post('a third party')
|
|
|
|
Additionally the DestinationPlugin allows the plugin author to add additional options
|
|
that can be used to help define sub-destinations.
|
|
|
|
For example, if we look at the aws-destination plugin we can see that it defines an `accountNumber` option::
|
|
|
|
options = [
|
|
{
|
|
'name': 'accountNumber',
|
|
'type': 'int',
|
|
'required': True,
|
|
'validation': '/^[0-9]{12,12}$/',
|
|
'helpMessage': 'Must be a valid AWS account number!',
|
|
}
|
|
]
|
|
|
|
By defining an `accountNumber` we can make this plugin handle many N number of AWS accounts instead of just one.
|
|
|
|
The schema for defining plugin options are pretty straightforward:
|
|
|
|
- **Name**: name of the variable you wish to present the user, snake case (snakeCase) is preferrred as Lemur
|
|
will parse these and create pretty variable titles
|
|
- **Type** there are currently four supported variable types
|
|
- **Int** creates an html integer box for the user to enter integers into
|
|
- **Str** creates a html text input box
|
|
- **Boolean** creates a checkbox for the user to signify truithyness
|
|
- **Select** creates a select box that gives the user a list of options
|
|
- When used a `available` key must be provided with a list of selectable options
|
|
- **Required** determines if this option is required, this **must be a boolean value**
|
|
- **Validation** simple JavaScript regular expression used to give the user an indication if the input value is valid
|
|
- **HelpMessage** simple string that provides more detail about the option
|
|
|
|
.. Note::
|
|
DestinationPlugin, NotificationPlugin and SourcePlugin all support the option
|
|
schema outlined above.
|
|
|
|
|
|
Notification
|
|
------------
|
|
|
|
Lemur includes the ability to create Email notifications by **default**. These notifications
|
|
currently come in the form of expiration noticies. Lemur periodically checks certifications expiration dates and
|
|
determines if a given certificate is eligible for notification. There are currently only two parameters used to
|
|
determine if a certificate is eligible; validity expiration (date the certificate is no longer valid) and the number
|
|
of days the current date (UTC) is from that expiration date.
|
|
|
|
There are currently two objects that available for notification plugins the first is `NotficationPlugin`. This is the base object for
|
|
any notification within Lemur. Currently the only support notification type is an certificate expiration notification. If you
|
|
are trying to create a new notification type (audit, failed logins, etc.) this would be the object to base your plugin on.
|
|
You would also then need to build additional code to trigger the new notification type.
|
|
|
|
The second is `ExpirationNotificationPlugin`, this object inherits from `NotificationPlugin` object.
|
|
You will most likely want to base your plugin on, if you want to add new channels for expiration notices (Slack, Hipcat, Jira, etc.). It adds default options that are required by
|
|
by all expiration notifications (interval, unit). This interface expects for the child to define the following function::
|
|
|
|
def send(self):
|
|
# request.post("some alerting infrastructure")
|
|
|
|
|
|
Source
|
|
------
|
|
|
|
When building Lemur we realized that although it would be nice if every certificate went through Lemur to get issued, but this is not
|
|
always be the case. Often times there are third parties that will issue certificates on your behalf and these can get deployed
|
|
to infrastructure without any interaction with Lemur. In an attempt to combat this and try to track every certificate, Lemur has a notion of
|
|
certificate **Sources**. Lemur will contact the source at periodic intervals and attempt to **sync** against the source. This means downloading or discovering any
|
|
certificate Lemur does not know about and adding the certificate to it's inventory to be tracked and alerted on.
|
|
|
|
The `SourcePlugin` object has one default option of `pollRate`. This controls the number of seconds which to get new certificates.
|
|
|
|
.. warning::
|
|
Lemur currently has a very basic polling system of running a cron job every 15min to see which source plugins need to be run. A lock file is generated to guarantee that
|
|
only one sync is running at a time. It also means that the minimum resolution of a source plugin poll rate is effectively 15min. You can always specify a faster cron
|
|
job if you need a higher resolution sync job.
|
|
|
|
|
|
The `SourcePlugin` object requires implementation of one function::
|
|
|
|
def get_certificates(self, **kwargs):
|
|
# request.get("some source of certificates")
|
|
|
|
|
|
.. Note::
|
|
Often times to facilitate code re-use it makes sense put source and destination plugins into one package.
|
|
|
|
|
|
Testing
|
|
=======
|
|
|
|
Lemur provides a basic py.test-based testing framework for extensions.
|
|
|
|
In a simple project, you'll need to do a few things to get it working:
|
|
|
|
setup.py
|
|
--------
|
|
|
|
Augment your setup.py to ensure at least the following:
|
|
|
|
.. code-block:: python
|
|
|
|
setup(
|
|
# ...
|
|
install_requires=[
|
|
'lemur',
|
|
]
|
|
)
|
|
|
|
|
|
conftest.py
|
|
-----------
|
|
|
|
The ``conftest.py`` file is our main entry-point for py.test. We need to configure it to load the Lemur pytest configuration:
|
|
|
|
.. code-block:: python
|
|
|
|
from __future__ import absolute_import
|
|
|
|
pytest_plugins = [
|
|
'lemur.utils.pytest'
|
|
]
|
|
|
|
|
|
Test Cases
|
|
----------
|
|
|
|
You can now inherit from Lemur's core test classes. These are Django-based and ensure the database and other basic utilities are in a clean state:
|
|
|
|
.. code-block:: python
|
|
|
|
# test_myextension.py
|
|
from __future__ import absolute_import
|
|
|
|
from lemur.testutils import TestCase
|
|
|
|
class MyExtensionTest(TestCase):
|
|
def test_simple(self):
|
|
assert 1 != 2
|
|
|
|
|
|
Running Tests
|
|
-------------
|
|
|
|
Running tests follows the py.test standard. As long as your test files and methods are named appropriately (``test_filename.py`` and ``test_function()``) you can simply call out to py.test:
|
|
|
|
::
|
|
|
|
$ py.test -v
|
|
============================== test session starts ==============================
|
|
platform darwin -- Python 2.7.9 -- py-1.4.26 -- pytest-2.6.4/python2.7
|
|
plugins: django
|
|
collected 1 items
|
|
|
|
tests/test_myextension.py::MyExtensionTest::test_simple PASSED
|
|
|
|
=========================== 1 passed in 0.35 seconds ============================
|
|
|
|
|
|
.. SeeAlso:: Lemur bundles several plugins that use the same interfaces mentioned above. View the source: # TODO
|