216 lines
8.3 KiB
Python
216 lines
8.3 KiB
Python
# encoding: utf-8
|
|
|
|
import falcon
|
|
import mimetypes
|
|
import logging
|
|
import os
|
|
import click
|
|
from datetime import datetime
|
|
from time import sleep
|
|
from certidude import authority, mailer
|
|
from certidude.auth import login_required, authorize_admin
|
|
from certidude.decorators import serialize, event_source, csrf_protection
|
|
from certidude.wrappers import Request, Certificate
|
|
from certidude import constants, config
|
|
|
|
logger = logging.getLogger("api")
|
|
|
|
class CertificateStatusResource(object):
|
|
"""
|
|
openssl ocsp -issuer CAcert_class1.pem -serial 0x<serial no in hex> -url http://localhost -CAfile cacert_both.pem
|
|
"""
|
|
def on_post(self, req, resp):
|
|
ocsp_request = req.stream.read(req.content_length)
|
|
for component in decoder.decode(ocsp_request):
|
|
click.echo(component)
|
|
resp.append_header("Content-Type", "application/ocsp-response")
|
|
resp.status = falcon.HTTP_200
|
|
raise NotImplementedError()
|
|
|
|
|
|
class CertificateAuthorityResource(object):
|
|
def on_get(self, req, resp):
|
|
logger.info("Served CA certificate to %s", req.context.get("remote_addr"))
|
|
resp.stream = open(config.AUTHORITY_CERTIFICATE_PATH, "rb")
|
|
resp.append_header("Content-Type", "application/x-x509-ca-cert")
|
|
resp.append_header("Content-Disposition", "attachment; filename=ca.crt")
|
|
|
|
|
|
class SessionResource(object):
|
|
@serialize
|
|
@login_required
|
|
@authorize_admin
|
|
@event_source
|
|
def on_get(self, req, resp):
|
|
if config.ACCOUNTS_BACKEND == "ldap":
|
|
import ldap
|
|
ft = config.LDAP_MEMBERS_FILTER % (config.ADMINS_GROUP, "*")
|
|
r = req.context.get("ldap_conn").search_s(config.LDAP_BASE,
|
|
ldap.SCOPE_SUBTREE, ft.encode("utf-8"), ["cn", "member"])
|
|
|
|
for dn,entry in r:
|
|
cn, = entry.get("cn")
|
|
break
|
|
else:
|
|
raise ValueError("Failed to look up group %s in LDAP" % repr(group_name))
|
|
|
|
admins = dict([(j, j.split(",")[0].split("=")[1]) for j in entry.get("member")])
|
|
elif config.ACCOUNTS_BACKEND == "posix":
|
|
import grp
|
|
_, _, gid, members = grp.getgrnam(config.ADMINS_GROUP)
|
|
admins = dict([(j, j) for j in members])
|
|
else:
|
|
raise NotImplementedError("Authorization backend %s not supported" % config.AUTHORIZATION_BACKEND)
|
|
|
|
return dict(
|
|
user = dict(
|
|
name=req.context.get("user").name,
|
|
gn=req.context.get("user").given_name,
|
|
sn=req.context.get("user").surname,
|
|
mail=req.context.get("user").mail
|
|
),
|
|
request_submission_allowed = sum( # Dirty hack!
|
|
[req.context.get("remote_addr") in j
|
|
for j in config.REQUEST_SUBNETS]),
|
|
user_subnets = config.USER_SUBNETS,
|
|
autosign_subnets = config.AUTOSIGN_SUBNETS,
|
|
request_subnets = config.REQUEST_SUBNETS,
|
|
admin_subnets=config.ADMIN_SUBNETS,
|
|
admin_users = admins,
|
|
#admin_users=config.ADMIN_USERS,
|
|
authority = dict(
|
|
outbox = config.OUTBOX,
|
|
certificate = authority.certificate,
|
|
events = config.PUSH_EVENT_SOURCE % config.PUSH_TOKEN,
|
|
requests=authority.list_requests(),
|
|
signed=authority.list_signed(),
|
|
revoked=authority.list_revoked(),
|
|
) if config.ADMINS_GROUP in req.context.get("groups") else None,
|
|
features=dict(
|
|
tagging=config.TAGGING_BACKEND,
|
|
leases=False, #config.LEASES_BACKEND,
|
|
logging=config.LOGGING_BACKEND))
|
|
|
|
|
|
class StaticResource(object):
|
|
def __init__(self, root):
|
|
self.root = os.path.realpath(root)
|
|
|
|
def __call__(self, req, resp):
|
|
|
|
path = os.path.realpath(os.path.join(self.root, req.path[1:]))
|
|
if not path.startswith(self.root):
|
|
raise falcon.HTTPForbidden
|
|
|
|
if os.path.isdir(path):
|
|
path = os.path.join(path, "index.html")
|
|
click.echo("Serving: %s" % path)
|
|
|
|
if os.path.exists(path):
|
|
content_type, content_encoding = mimetypes.guess_type(path)
|
|
if content_type:
|
|
resp.append_header("Content-Type", content_type)
|
|
if content_encoding:
|
|
resp.append_header("Content-Encoding", content_encoding)
|
|
resp.stream = open(path, "rb")
|
|
else:
|
|
resp.status = falcon.HTTP_404
|
|
resp.body = "File '%s' not found" % req.path
|
|
|
|
|
|
class BundleResource(object):
|
|
@login_required
|
|
def on_get(self, req, resp):
|
|
common_name = req.context["user"].mail
|
|
logger.info("Signing bundle %s for %s", common_name, req.context.get("user"))
|
|
resp.set_header("Content-Type", "application/x-pkcs12")
|
|
resp.set_header("Content-Disposition", "attachment; filename=%s.p12" % common_name)
|
|
resp.body, cert = authority.generate_pkcs12_bundle(common_name,
|
|
owner=req.context.get("user"))
|
|
|
|
|
|
import ipaddress
|
|
|
|
class NormalizeMiddleware(object):
|
|
@csrf_protection
|
|
def process_request(self, req, resp, *args):
|
|
assert not req.get_param("unicode") or req.get_param("unicode") == u"✓", "Unicode sanity check failed"
|
|
req.context["remote_addr"] = ipaddress.ip_address(req.env["REMOTE_ADDR"].decode("utf-8"))
|
|
|
|
def process_response(self, req, resp, resource):
|
|
# wtf falcon?!
|
|
if isinstance(resp.location, unicode):
|
|
resp.location = resp.location.encode("ascii")
|
|
|
|
def certidude_app():
|
|
from certidude import config
|
|
|
|
from .revoked import RevocationListResource
|
|
from .signed import SignedCertificateListResource, SignedCertificateDetailResource
|
|
from .request import RequestListResource, RequestDetailResource
|
|
from .lease import LeaseResource
|
|
from .whois import WhoisResource
|
|
from .log import LogResource
|
|
from .tag import TagResource, TagDetailResource
|
|
from .cfg import ConfigResource, ScriptResource
|
|
|
|
app = falcon.API(middleware=NormalizeMiddleware())
|
|
|
|
# Certificate authority API calls
|
|
app.add_route("/api/ocsp/", CertificateStatusResource())
|
|
app.add_route("/api/bundle/", BundleResource())
|
|
app.add_route("/api/certificate/", CertificateAuthorityResource())
|
|
app.add_route("/api/revoked/", RevocationListResource())
|
|
app.add_route("/api/signed/{cn}/", SignedCertificateDetailResource())
|
|
app.add_route("/api/signed/", SignedCertificateListResource())
|
|
app.add_route("/api/request/{cn}/", RequestDetailResource())
|
|
app.add_route("/api/request/", RequestListResource())
|
|
app.add_route("/api/", SessionResource())
|
|
|
|
# Gateway API calls, should this be moved to separate project?
|
|
app.add_route("/api/lease/", LeaseResource())
|
|
app.add_route("/api/whois/", WhoisResource())
|
|
|
|
log_handlers = []
|
|
if config.LOGGING_BACKEND == "sql":
|
|
from certidude.mysqllog import LogHandler
|
|
uri = config.cp.get("logging", "database")
|
|
log_handlers.append(LogHandler(uri))
|
|
app.add_route("/api/log/", LogResource(uri))
|
|
elif config.LOGGING_BACKEND == "syslog":
|
|
from logging.handlers import SyslogHandler
|
|
log_handlers.append(SysLogHandler())
|
|
# Browsing syslog via HTTP is obviously not possible out of the box
|
|
elif config.LOGGING_BACKEND:
|
|
raise ValueError("Invalid logging.backend = %s" % config.LOGGING_BACKEND)
|
|
|
|
if config.TAGGING_BACKEND == "sql":
|
|
uri = config.cp.get("tagging", "database")
|
|
app.add_route("/api/tag/", TagResource(uri))
|
|
app.add_route("/api/tag/{identifier}/", TagDetailResource(uri))
|
|
app.add_route("/api/config/", ConfigResource(uri))
|
|
app.add_route("/api/script/", ScriptResource(uri))
|
|
elif config.TAGGING_BACKEND:
|
|
raise ValueError("Invalid tagging.backend = %s" % config.TAGGING_BACKEND)
|
|
|
|
if config.PUSH_PUBLISH:
|
|
from certidude.push import PushLogHandler
|
|
log_handlers.append(PushLogHandler())
|
|
|
|
for facility in "api", "cli":
|
|
logger = logging.getLogger(facility)
|
|
logger.setLevel(logging.DEBUG)
|
|
for handler in log_handlers:
|
|
logger.addHandler(handler)
|
|
|
|
logging.getLogger("cli").debug("Started Certidude at %s", constants.FQDN)
|
|
|
|
import atexit
|
|
|
|
def exit_handler():
|
|
logging.getLogger("cli").debug("Shutting down Certidude")
|
|
|
|
atexit.register(exit_handler)
|
|
|
|
return app
|