From f624d2e35dab0d87a289a346999c0cb01ed0aa55 Mon Sep 17 00:00:00 2001
From: Brian Waldon <brian.waldon@rackspace.com>
Date: Tue, 29 Mar 2011 11:11:57 -0400
Subject: adding server name validation to create method; adding tests

---
 nova/api/openstack/servers.py | 36 ++++++++++++++++++++++--------------
 1 file changed, 22 insertions(+), 14 deletions(-)

(limited to 'nova/api')

diff --git a/nova/api/openstack/servers.py b/nova/api/openstack/servers.py
index 29c491716..d564b37c1 100644
--- a/nova/api/openstack/servers.py
+++ b/nova/api/openstack/servers.py
@@ -150,6 +150,15 @@ class Controller(wsgi.Controller):
             injected_files = self._get_injected_files(personality)
 
         flavor_id = self._flavor_id_from_req_data(env)
+
+        if not 'name' in env['server']:
+            msg = _("Server name is not defined")
+            return exc.HTTPBadRequest(msg)
+        
+        name = env['server']['name']
+        self._validate_server_name(name)
+        name = name.strip()
+
         try:
             (inst,) = self.compute_api.create(
                 context,
@@ -157,8 +166,8 @@ class Controller(wsgi.Controller):
                 image_id,
                 kernel_id=kernel_id,
                 ramdisk_id=ramdisk_id,
-                display_name=env['server']['name'],
-                display_description=env['server']['name'],
+                display_name=name,
+                display_description=name,
                 key_name=key_name,
                 key_data=key_data,
                 metadata=metadata,
@@ -249,18 +258,8 @@ class Controller(wsgi.Controller):
 
         if 'name' in inst_dict['server']:
             name = inst_dict['server']['name']
-
-            if not isinstance(name, basestring):
-                msg = _("Server name is not a string or unicode")
-                return exc.HTTPBadRequest(msg)
-
-            name = name.strip()
-
-            if name == '':
-                msg = _("Server name is an empty string")
-                return exc.HTTPBadRequest(msg)
-
-            update_dict['display_name'] = name
+            self._validate_server_name(name)
+            update_dict['display_name'] = name.strip()
 
         self._parse_update(ctxt, id, inst_dict, update_dict)
 
@@ -271,6 +270,15 @@ class Controller(wsgi.Controller):
 
         return exc.HTTPNoContent()
 
+    def _validate_server_name(self, value):
+        if not isinstance(value, basestring):
+            msg = _("Server name is not a string or unicode")
+            raise exc.HTTPBadRequest(msg)
+
+        if value.strip() == '':
+            msg = _("Server name is an empty string")
+            raise exc.HTTPBadRequest(msg)
+
     def _parse_update(self, context, id, inst_dict, update_dict):
         pass
 
-- 
cgit