From: Mark Huang <mlhuang@cs.princeton.edu>
Date: Wed, 25 Oct 2006 16:58:19 +0000 (+0000)
Subject: - only allow addition of nodes at the same site
X-Git-Tag: pycurl-7_13_1~451
X-Git-Url: http://git.onelab.eu/?a=commitdiff_plain;h=e668028fc37124059fb008f2c4da9da20aae7b2b;p=plcapi.git

- only allow addition of nodes at the same site
---

diff --git a/PLC/Methods/AddNodeToPCU.py b/PLC/Methods/AddNodeToPCU.py
index 53789872..2d06f9f6 100644
--- a/PLC/Methods/AddNodeToPCU.py
+++ b/PLC/Methods/AddNodeToPCU.py
@@ -59,10 +59,14 @@ class AddNodeToPCU(Method):
         if node['node_id'] in pcu['node_ids']:
             raise PLCInvalidArgument, "Node already controlled by PCU"
 
+        if node['site_id'] != pcu['site_id']:
+            raise PLCInvalidArgument, "Node is at a different site than this PCU"
+
         if port in pcu['ports']:
             raise PLCInvalidArgument, "PCU port already in use"
 
         pcu.add_node(node, port)
-	self.object_ids = [pcu['pcu_id']]
+
+	self.object_ids = [node['node_id'], pcu['pcu_id']]
 
         return 1