diff options
| author | Arthur de Jong <arthur@arthurdejong.org> | 2011-03-24 23:19:42 +0100 |
|---|---|---|
| committer | Arthur de Jong <arthur@arthurdejong.org> | 2011-03-24 23:19:42 +0100 |
| commit | 1fb820314c4b6f720bfca5ce5bfbc32183e3c3af (patch) | |
| tree | 5eedee2276a7763a626b7f82c481afb420349129 /pynslcd/service.py | |
| parent | 4578bac38dd1ef63d0850a5139a7e4dbca3127f3 (diff) | |
implement service, protocol and rpc lookups
git-svn-id: http://arthurdejong.org/svn/nss-pam-ldapd/nss-pam-ldapd@1410 ef36b2f9-881f-0410-afb5-c4e39611909c
Diffstat (limited to 'pynslcd/service.py')
| -rw-r--r-- | pynslcd/service.py | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/pynslcd/service.py b/pynslcd/service.py new file mode 100644 index 0000000..959344c --- /dev/null +++ b/pynslcd/service.py @@ -0,0 +1,113 @@ + +# service.py - service entry lookup routines +# +# Copyright (C) 2011 Arthur de Jong +# +# This library is free software; you can redistribute it and/or +# modify it under the terms of the GNU Lesser General Public +# License as published by the Free Software Foundation; either +# version 2.1 of the License, or (at your option) any later version. +# +# This library 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 +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this library; if not, write to the Free Software +# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA +# 02110-1301 USA + +import constants +import common + +import struct +import ldap.filter + + +class ServiceRequest(common.Request): + + filter = '(objectClass=ipService)' + + attmap_cn = 'cn' + attmap_ipServicePort = 'ipServicePort' + attmap_ipServiceProtocol = 'ipServiceProtocol' + + attributes = ( 'cn', 'ipServicePort', 'ipServiceProtocol' ) + + def __init__(self, *args): + super(ServiceRequest, self).__init__(*args) + self.protocol = None + + def write(self, entry): + dn, attributes = entry + # get name + name = common.get_rdn_value(entry, self.attmap_cn) + names = attributes.get(self.attmap_cn, []) + if not names: + print 'Error: entry %s does not contain %s value' % ( dn, self.attmap_cn ) + if self.name and self.name not in names + [ name, ]: + return # case of result entry did not match + if not name: + name = names.pop(0) + elif name in names: + names.remove(name) + # get port number + ( port, ) = attributes.get(self.attmap_ipServicePort, []) + if not port: + print 'Error: entry %s does not contain %s value' % ( dn, self.attmap_ipServicePort) + port = int(port) + # get protocol + protocols = attributes.get(self.attmap_ipServiceProtocol, []) + if self.protocol: + if self.protocol not in protocols: + return + protocols = ( self.protocol, ) + # write result + for protocol in protocols: + self.fp.write_int32(constants.NSLCD_RESULT_BEGIN) + self.fp.write_string(name) + self.fp.write_stringlist(names) + self.fp.write_int32(port) + self.fp.write_string(protocol) + + +class ServiceByNameRequest(ServiceRequest): + + action = constants.NSLCD_ACTION_SERVICE_BYNAME + + def read_parameters(self): + self.name = self.fp.read_string() + self.protocol = self.fp.read_string() + + def mk_filter(self): + if self.protocol: + return '(&%s(%s=%s)(%s=%s))' % ( self.filter, + self.attmap_cn, ldap.filter.escape_filter_chars(self.name), + self.attmap_ipServiceProtocol, ldap.filter.escape_filter_chars(self.protocol) ) + else: + return '(&%s(%s=%s))' % ( self.filter, + self.attmap_cn, ldap.filter.escape_filter_chars(self.name) ) + + +class ServiceByNumberRequest(ServiceRequest): + + action = constants.NSLCD_ACTION_SERVICE_BYNUMBER + + def read_parameters(self): + self.number = self.fp.read_int32() + self.protocol = self.fp.read_string() + + def mk_filter(self): + if self.protocol: + return '(&%s(%s=%d)(%s=%s))' % ( self.filter, + self.attmap_ipServicePort, self.number, + self.attmap_ipServiceProtocol, ldap.filter.escape_filter_chars(self.protocol) ) + else: + return '(&%s(%s=%d))' % ( self.filter, + self.attmap_ipServicePort, self.number ) + + +class ServiceAllRequest(ServiceRequest): + + action = constants.NSLCD_ACTION_SERVICE_ALL |