freeipa/ipalib/plugins/join.py

# Authors:
#   Rob Crittenden <rcritten@redhat.com>
#
# Copyright (C) 2009  Red Hat
# see file 'COPYING' for use and warranty information
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; version 2 only
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

"""
join a machine to the IPA domain
"""

from ipalib import api, util
from ipalib import Command, Str, Int
from ipalib import errors
import krbV
import os, subprocess
from ipapython import ipautil
from ipapython import certdb
from ipapython import dogtag
import tempfile
import sha
import httplib
import xml.dom.minidom
import stat
import shutil

def get_realm():
    krbctx = krbV.default_context()

    return unicode(krbctx.default_realm)

def validate_host(ugettext, cn):
    """
    Require at least one dot in the hostname (to support localhost.localdomain)
    """
    dots = len(cn.split('.'))
    if dots < 2:
        return 'Fully-qualified hostname required'
    return None

class join(Command):
    """Join an IPA domain"""

    requires_root = True

    takes_args = (
        Str('cn',
            validate_host,
            cli_name='hostname',
            doc="The hostname to register as",
            create_default=lambda **kw: unicode(util.get_fqdn()),
            autofill=True,
            #normalizer=lamda value: value.lower(),
        ),
    )
    takes_options= (
        Str('realm',
            doc="The IPA realm",
            create_default=lambda **kw: get_realm(),
            autofill=True,
        ),
    )

    def execute(self, hostname, **kw):
        """
        Execute the machine join operation.

        Returns the entry as it will be created in LDAP.

        :param hostname: The name of the host joined
        :param kw: Keyword arguments for the other attributes.
        """
        assert 'cn' not in kw
        ldap = self.api.Backend.ldap

        try:
            host = api.Command['host_show'](hostname)
        except errors.NotFound:
            pass
        else:
            raise errors.DuplicateEntry

        return api.Command['host_add'](hostname)

    def output_for_cli(self, textui, result, args, **options):
        textui.print_plain("Welcome to the %s realm" % options['realm'])
        textui.print_plain("Your keytab is in %s" % result.get('keytab'))
        if result.get('pkcs12'):
            textui.print_plain("An X.509 server certificate is in %s" % result.get('pkcs12'))

    def run(self, *args, **options):
        """
        Dispatch to forward() and execute() to do work locally and on the
        server.
        """
        if self.env.in_server:
            return self.execute(*args, **options)

        # This forward will call the server-side portion of join
        result = self.forward(*args, **options)

        self._get_keytab(result['krbprincipalname'])
        self._generate_server_cert(args)
        result['keytab'] = '/tmp/kt'
        self._set_perms('/tmp/kt')
        if ipautil.file_exists('/tmp/server.p12'):
            self._set_perms('/tmp/server.p12')
        result['pkcs12'] = '/tmp/server.p12'
        return result

    def _set_perms(self, filename):
        os.chown(filename, 0, 0)
        os.chmod(filename,  stat.S_IRUSR)

    def _get_keytab(self, principal, stdin=None):
        args = ["/usr/sbin/ipa-getkeytab", "-s", self.env.host, "-p", principal,"-k", "/tmp/kt"]
        return ipautil.run(args, stdin)
    def _generate_server_cert(self, hostname):
        subject = "CN=%s,OU=pki-ipa,O=IPA" % hostname
        cdb = certdb.CertDB(secdir=None, temporary=True)

        csr = cdb.generate_csr(subject, keysize=1024)

        # Request a cert
        try:
            result = api.Command['cert_request'](unicode(csr), **{})
        except KeyError:
            return "Certificates are not supported"

        # Load the cert into our temporary database
        if result.get('certificate', False):
            cert_file = cdb.secdir + "/cert.txt"
            f = open(cert_file, "w")
            f.write(result.get('certificate'))
            f.close()

            cdb.add_certificate(cert_file, "Server-Cert", is_ca=False)

            ca_chain = dogtag.get_ca_certchain()

            ca_file = cdb.secdir + "/ca.txt"
            f = open(ca_file, "w")
            f.write(ca_chain)
            f.close()

            cdb.add_certificate(ca_file, "caCert", is_ca=True)

            result = cdb.create_pkcs12("/tmp/server.p12", "Server-Cert")
        else:
            # Raise some error?
            pass

        return result

api.register(join)