Changeset 5490 for trunk/Nebulous/lib

Timestamp:

Nov 8, 2005, 3:31:07 PM (21 years ago)

Author:

jhoblitt

Message:

change create_object to verify that it's class param is valid

Location:

trunk/Nebulous/lib/Nebulous

Files:

• Server.pm (modified) (3 diffs)
• Server/SQL.pm (modified) (2 diffs)

trunk/Nebulous/lib/Nebulous/Server.pm

@@ -1 +1 @@
 # Copyright (c) 2004  Joshua Hoblitt
 #
-# $Id: Server.pm,v 1.12 2005-11-09 00:48:24 jhoblitt Exp $
+# $Id: Server.pm,v 1.13 2005-11-09 01:31:07 jhoblitt Exp $
 
 package Nebulous::Server;
@@ -17 +17 @@
 use Log::Log4perl;
 use Params::Validate qw( validate_pos SCALAR );
-
-my %class = (
-    raw => 1,
-);
 
 my $log;
@@ -99 +95 @@
 
     # check that the comment isn't too long
+    # check that the volume requested is valid
+
     # check that the class requested is valid
-    # check that the volume requested is valid
+    if ( $class ) {
+        my $class_id;
+        eval {
+            my $query = $db->prepare_cached( $sql->get_class ); 
+            $query->execute( $class );
+            ( $class_id ) = $query->fetchrow_array;
+
+            $query->finish;
+        };
+        if ( $@ ) {
+            $db->rollback;
+            $log->logdie( "database error: $@" );
+        }
+
+        unless ( $class_id ) {
+            $log->logdie( "Invalid class ID: $class" );
+        }
+    }
 
     my $uri;

trunk/Nebulous/lib/Nebulous/Server/SQL.pm

@@ -1 +1 @@
 # Copyright (c) 2004  Joshua Hoblitt
 #
-# $Id: SQL.pm,v 1.13 2005-11-09 00:59:20 jhoblitt Exp $
+# $Id: SQL.pm,v 1.14 2005-11-09 01:31:07 jhoblitt Exp $
 
 package Nebulous::Server::SQL;
@@ -123 +123 @@
         INSERT INTO volume (name, uri)
         VALUES (?, ?)
+    },
+    get_class => qq{
+        SELECT class_id, priority, comment
+        FROM class
+        WHERE class_id = ?
     },
 );