X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Fopenstack%2Fsecurity_group.py;h=4af0e581c85fa59a5213a8f813a99923a42d74a3;hb=1db1a879ffde8991aa95dd80142d555551655e88;hp=63f86767e34f0613830b579bcac73a669ab826d2;hpb=0dc62ed53820a645f6c1cf752bbcdba0badebc82;p=sfa.git diff --git a/sfa/openstack/security_group.py b/sfa/openstack/security_group.py index 63f86767..4af0e581 100644 --- a/sfa/openstack/security_group.py +++ b/sfa/openstack/security_group.py @@ -9,14 +9,14 @@ class SecurityGroup: def create_security_group(self, name): conn = self.driver.euca_shell.get_euca_connection() try: - conn.create_security_group(name=group_name) + conn.create_security_group(name=name, description="") except Exception, ex: logger.log_exc("Failed to add security group") def delete_security_group(self, name): conn = self.driver.euca_shell.get_euca_connection() try: - conn.delete_security_group(name=group_name) + conn.delete_security_group(name=name) except Exception, ex: logger.log_exc("Failed to delete security group") @@ -24,14 +24,12 @@ class SecurityGroup: def _validate_port_range(self, port_range): from_port = to_port = None if isinstance(port_range, str): - ports = port_range.split('-') + ports = port_range.split(':') if len(ports) > 1: from_port = int(ports[0]) to_port = int(ports[1]) else: from_port = to_port = int(ports[0]) - else: - from_port = to_port = None return (from_port, to_port) def _validate_icmp_type_code(self, icmp_type_code): @@ -51,17 +49,12 @@ class SecurityGroup: port_range=None, icmp_type_code=None, source_group_name=None, source_group_owner_id=None): - from_port, to_port = self._validate_port_range(port_range) icmp_type = self._validate_icmp_type_code(icmp_type_code) - if icmp_type: + if icmp_type and icmp_type[0] and icmp_type[1]: from_port, to_port = icmp_type[0], icmp_type[1] if group_name: - if cidr_ip: - self.driver.euca_shell.euca2ool.validate_address(cidr_ip) - if protocol: - self.driver.euca_shell.euca2ool.validate_protocol(protocol) conn = self.driver.euca_shell.get_euca_connection() try: conn.authorize_security_group( @@ -71,7 +64,7 @@ class SecurityGroup: ip_protocol=protocol, from_port=from_port, to_port=to_port, - cidr_ip=ip, + cidr_ip=cidr_ip, ) except Exception, ex: logger.log_exc("Failed to add rule to group %s" % group_name) @@ -87,10 +80,6 @@ class SecurityGroup: from_port, to_port = icmp_type[0], icmp_type[1] if group_name: - if cidr_ip: - self.driver.euca_shell.euca2ool.validate_address(cidr_ip) - if protocol: - self.driver.euca_shell.euca2ool.validate_protocol(protocol) conn = self.driver.euca_shell.get_euca_connection() try: conn.revoke_security_group(