SERVER-5020 Mongo shell should check the validity of the db name for use dbname

No automated test for actually using "use" since the shell cannot parse it correctly when "use" is placed inside a function and assert.throws requires you to pass a function to it. Tested ok on SpiderMonkey and V8 by hand.

1 files changed, 5 insertions, 0 deletions

diff --git a/src/mongo/scripting/v8_db.cpp b/src/mongo/scripting/v8_db.cpp
index d84f2c30265..852792eb0a8 100644
--- a/src/mongo/scripting/v8_db.cpp
+++ b/src/mongo/scripting/v8_db.cpp
@@ -23,6 +23,7 @@
 #include "util/text.h"
 #include "../client/syncclusterconnection.h"
 #include "../s/d_logic.h"
+#include "../db/namespacestring.h"
 #include <iostream>
 
 using namespace std;
@@ -566,6 +567,10 @@ namespace mongo {
         for ( int i=0; i<args.Length(); i++ )
             assert( ! args[i]->IsUndefined() );
 
+        if ( !NamespaceString::validDBName( toSTLString( args[1] ))) {
+            return v8::ThrowException( v8::String::New( "invalid database name" ));
+        }
+        
         return v8::Undefined();
     }