From 1fb820314c4b6f720bfca5ce5bfbc32183e3c3af Mon Sep 17 00:00:00 2001
From: Arthur de Jong <arthur@arthurdejong.org>
Date: Thu, 24 Mar 2011 22:19:42 +0000
Subject: 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
---
 pynslcd/service.py | 113 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 pynslcd/service.py

(limited to 'pynslcd/service.py')

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
-- 
cgit v1.2.3