4 files changed, 292 insertions, 0 deletions
diff --git a/pynslcd/protocol.py b/pynslcd/protocol.py
new file mode 100644
index 0000000..32334ff
--- /dev/null
+++ b/pynslcd/protocol.py
@@ -0,0 +1,88 @@
+
+# protocol.py - protocol name and number 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 ProtocolRequest(common.Request):
+
+    filter = '(objectClass=ipProtocol)'
+
+    attmap_cn               = 'cn'
+    attmap_ipProtocolNumber = 'ipProtocolNumber'
+
+    attributes = ( 'cn', 'ipProtocolNumber' )
+
+    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:
+            return # case of result entry did not match
+        if not name:
+            name = names.pop(0)
+        elif name in names:
+            names.remove(name)
+        # get number
+        ( number, ) = attributes.get(self.attmap_ipProtocolNumber, [])
+        if not number:
+            print 'Error: entry %s does not contain %s value' % ( dn, self.attmap_ipProtocolNumber)
+        number = int(number)
+        # write result
+        self.fp.write_int32(constants.NSLCD_RESULT_BEGIN)
+        self.fp.write_string(name)
+        self.fp.write_stringlist(names)
+        self.fp.write_int32(number)
+
+
+class ProtocolByNameRequest(ProtocolRequest):
+
+    action = constants.NSLCD_ACTION_PROTOCOL_BYNAME
+
+    def read_parameters(self):
+        self.name = self.fp.read_string()
+
+    def mk_filter(self):
+        return '(&%s(%s=%s))' % ( self.filter,
+                  self.attmap_cn, ldap.filter.escape_filter_chars(self.name) )
+
+
+class ProtocolByNumberRequest(ProtocolRequest):
+
+    action = constants.NSLCD_ACTION_PROTOCOL_BYNUMBER
+
+    def read_parameters(self):
+        self.number = self.fp.read_int32()
+
+    def mk_filter(self):
+        return '(&%s(%s=%d))' % ( self.filter,
+                  self.attmap_ipProtocolNumber, self.number )
+
+
+class ProtocolAllRequest(ProtocolRequest):
+
+    action = constants.NSLCD_ACTION_PROTOCOL_ALL
diff --git a/pynslcd/pynslcd.py b/pynslcd/pynslcd.py
index f4eb5f1..2deab07 100755
--- a/pynslcd/pynslcd.py
+++ b/pynslcd/pynslcd.py
@@ -146,6 +146,9 @@ handlers.update(common.get_handlers('netgroup'))
 handlers.update(common.get_handlers('network'))
 handlers.update(common.get_handlers('pam'))
 handlers.update(common.get_handlers('passwd'))
+handlers.update(common.get_handlers('protocol'))
+handlers.update(common.get_handlers('rpc'))
+handlers.update(common.get_handlers('service'))
 handlers.update(common.get_handlers('shadow'))
 
 def acceptconnection(session):
diff --git a/pynslcd/rpc.py b/pynslcd/rpc.py
new file mode 100644
index 0000000..5682d14
--- /dev/null
+++ b/pynslcd/rpc.py
@@ -0,0 +1,88 @@
+
+# rpc.py - rpc name 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 RpcRequest(common.Request):
+
+    filter = '(objectClass=oncRpc)'
+
+    attmap_cn           = 'cn'
+    attmap_oncRpcNumber = 'oncRpcNumber'
+
+    attributes = ( 'cn', 'oncRpcNumber' )
+
+    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:
+            return # case of result entry did not match
+        if not name:
+            name = names.pop(0)
+        elif name in names:
+            names.remove(name)
+        # get number
+        ( number, ) = attributes.get(self.attmap_oncRpcNumber, [])
+        if not number:
+            print 'Error: entry %s does not contain %s value' % ( dn, self.attmap_oncRpcNumber)
+        number = int(number)
+        # write result
+        self.fp.write_int32(constants.NSLCD_RESULT_BEGIN)
+        self.fp.write_string(name)
+        self.fp.write_stringlist(names)
+        self.fp.write_int32(number)
+
+
+class RpcByNameRequest(RpcRequest):
+
+    action = constants.NSLCD_ACTION_RPC_BYNAME
+
+    def read_parameters(self):
+        self.name = self.fp.read_string()
+
+    def mk_filter(self):
+        return '(&%s(%s=%s))' % ( self.filter,
+                  self.attmap_cn, ldap.filter.escape_filter_chars(self.name) )
+
+
+class RpcByNumberRequest(RpcRequest):
+
+    action = constants.NSLCD_ACTION_RPC_BYNUMBER
+
+    def read_parameters(self):
+        self.number = self.fp.read_int32()
+
+    def mk_filter(self):
+        return '(&%s(%s=%d))' % ( self.filter,
+                  self.attmap_oncRpcNumber, self.number )
+
+
+class RpcAllRequest(RpcRequest):
+
+    action = constants.NSLCD_ACTION_RPC_ALL
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