mirror of
https://salsa.debian.org/freeipa-team/freeipa.git
synced 2024-12-28 01:41:14 -06:00
8de13bd7dd
In Python 3, `print` is no longer a statement. Call it as a function everywhere, and include the future import to remove the statement in Python 2 code as well. Reviewed-By: Christian Heimes <cheimes@redhat.com> Reviewed-By: Jan Cholasta <jcholast@redhat.com>
211 lines
7.6 KiB
Python
Executable File
211 lines
7.6 KiB
Python
Executable File
#!/usr/bin/python2
|
|
# Authors: Jr Aquino <jr.aquino@citrix.com>
|
|
#
|
|
# Copyright (C) 2011 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, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# 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, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
from __future__ import print_function
|
|
|
|
import re
|
|
import sys
|
|
from optparse import OptionParser
|
|
|
|
from ipapython import ipautil, config, ipaldap
|
|
from ipaserver.install import installutils
|
|
from ipalib import api, errors
|
|
from ipalib.constants import CACERT
|
|
from ipapython.ipa_log_manager import *
|
|
from ipapython.dn import DN
|
|
|
|
def parse_options():
|
|
usage = "%prog [options] <status|enable|disable>\n"
|
|
usage += "%prog [options]\n"
|
|
parser = OptionParser(usage=usage, formatter=config.IPAFormatter())
|
|
|
|
parser.add_option("-d", "--debug", action="store_true", dest="debug",
|
|
help="Display debugging information about the update(s)")
|
|
parser.add_option("-e", "--entry", dest="managed_entry",
|
|
default=None, type="string",
|
|
help="DN for the Managed Entry Definition")
|
|
parser.add_option("-l", "--list", dest="list_managed_entries",
|
|
action="store_true",
|
|
help="DN for the Managed Entry Definition")
|
|
parser.add_option("-p", "--password", dest="dirman_password",
|
|
help="Directory Manager password")
|
|
|
|
options, args = parser.parse_args()
|
|
|
|
return options, args
|
|
|
|
def get_dirman_password():
|
|
"""Prompt the user for the Directory Manager password and verify its
|
|
correctness.
|
|
"""
|
|
password = installutils.read_password("Directory Manager", confirm=False,
|
|
validate=True)
|
|
|
|
return password
|
|
|
|
def main():
|
|
retval = 0
|
|
def_dn = None
|
|
options, args = parse_options()
|
|
|
|
if options.list_managed_entries:
|
|
pass
|
|
elif len(args) != 1:
|
|
sys.exit("You must specify an action, either status, enable or disable")
|
|
elif args[0] != "enable" and args[0] != "disable" and args[0] != "status":
|
|
sys.exit("Unrecognized action [" + args[0] + "]")
|
|
standard_logging_setup(None, debug=options.debug)
|
|
|
|
host = installutils.get_fqdn()
|
|
api.bootstrap(context='cli', debug=options.debug)
|
|
api.finalize()
|
|
|
|
managed_entry_definitions_dn = DN(
|
|
('cn', 'Definitions'),
|
|
('cn', 'Managed Entries'),
|
|
('cn', 'etc'),
|
|
api.env.basedn
|
|
)
|
|
|
|
conn = None
|
|
try:
|
|
filter = '(objectClass=extensibleObject)'
|
|
conn = ipaldap.IPAdmin(host, 636, cacert=CACERT)
|
|
|
|
if options.dirman_password:
|
|
try:
|
|
conn.do_simple_bind(bindpw=options.dirman_password)
|
|
except errors.ACIError:
|
|
sys.exit("Invalid credentials")
|
|
else:
|
|
conn.do_sasl_gssapi_bind()
|
|
except errors.ACIError:
|
|
dirman_password = get_dirman_password()
|
|
if dirman_password is None:
|
|
sys.exit("Directory Manager password required")
|
|
try:
|
|
conn.do_simple_bind(bindpw=dirman_password)
|
|
except errors.ACIError:
|
|
sys.exit("Invalid credentials")
|
|
except errors.ExecutionError as lde:
|
|
sys.exit("An error occurred while connecting to the server.\n%s\n" %
|
|
str(lde))
|
|
|
|
if options.list_managed_entries:
|
|
# List available Managed Entry Plugins
|
|
managed_entries = None
|
|
try:
|
|
entries = conn.get_entries(
|
|
managed_entry_definitions_dn, conn.SCOPE_SUBTREE, filter)
|
|
except Exception as e:
|
|
root_logger.debug("Search for managed entries failed: %s" % str(e))
|
|
sys.exit("Unable to find managed entries at %s" % managed_entry_definitions_dn)
|
|
managed_entries = [entry.single_value['cn'] for entry in entries]
|
|
if managed_entries:
|
|
print("Available Managed Entry Definitions:")
|
|
for managed_entry in managed_entries:
|
|
print(managed_entry)
|
|
retval = 0
|
|
sys.exit()
|
|
|
|
if not options.managed_entry:
|
|
sys.exit("\nYou must specify a managed entry definition")
|
|
else:
|
|
def_dn = DN(('cn', options.managed_entry), managed_entry_definitions_dn)
|
|
|
|
disabled = True
|
|
try:
|
|
[entry] = conn.get_entries(def_dn, conn.SCOPE_BASE,
|
|
filter, ['originfilter'])
|
|
disable_attr = '(objectclass=disable)'
|
|
try:
|
|
org_filter = entry.single_value.get('originfilter')
|
|
disabled = re.search(r'%s' % disable_attr, org_filter)
|
|
except KeyError:
|
|
sys.exit("%s is not a valid Managed Entry" % def_dn)
|
|
except errors.NotFound:
|
|
sys.exit("%s is not a valid Managed Entry" % def_dn)
|
|
except errors.ExecutionError as lde:
|
|
print("An error occurred while talking to the server.")
|
|
print(lde)
|
|
|
|
if args[0] == "status":
|
|
if not disabled:
|
|
print("Plugin Enabled")
|
|
else:
|
|
print("Plugin Disabled")
|
|
return 0
|
|
|
|
if args[0] == "enable":
|
|
try:
|
|
if not disabled:
|
|
print("Plugin already Enabled")
|
|
retval = 2
|
|
else:
|
|
# Remove disable_attr from filter
|
|
enable_attr = org_filter.replace(disable_attr, '')
|
|
#enable_attr = {'originfilter': enable_attr}
|
|
entry['originfilter'] = [enable_attr]
|
|
conn.update_entry(entry)
|
|
print("Enabling Plugin")
|
|
retval = 0
|
|
except errors.NotFound:
|
|
print("Enabling Plugin")
|
|
except errors.ExecutionError as lde:
|
|
print("An error occurred while talking to the server.")
|
|
print(lde)
|
|
retval = 1
|
|
|
|
elif args[0] == "disable":
|
|
# Set originFilter to objectclass=disabled
|
|
# In future we should we should dedicate an attribute for enabling/
|
|
# disabling.
|
|
try:
|
|
if disabled:
|
|
print("Plugin already disabled")
|
|
retval = 2
|
|
else:
|
|
if org_filter[:2] == '(&' and org_filter[-1] == ')':
|
|
disable_attr = org_filter[:2] + disable_attr + org_filter[2:]
|
|
else:
|
|
disable_attr = '(&%s(%s))' % (disable_attr, org_filter)
|
|
entry['originfilter'] = [disable_attr]
|
|
conn.update_entry(entry)
|
|
print("Disabling Plugin")
|
|
except errors.NotFound:
|
|
print("Plugin is already disabled")
|
|
retval = 2
|
|
except errors.DatabaseError as dbe:
|
|
print("An error occurred while talking to the server.")
|
|
print(dbe)
|
|
retval = 1
|
|
except errors.ExecutionError as lde:
|
|
print("An error occurred while talking to the server.")
|
|
print(lde)
|
|
retval = 1
|
|
|
|
else:
|
|
retval = 1
|
|
|
|
return retval
|
|
|
|
if __name__ == '__main__':
|
|
installutils.run_script(main, operation_name='ipa-managed-entries')
|