SERVER-16741 Add escape sequence to directoryperdb paths in KV Store

1 files changed, 44 insertions, 0 deletions

diff --git a/src/mongo/db/namespace_string.cpp b/src/mongo/db/namespace_string.cpp
index d4637695ee7..fe50e2039be 100644
--- a/src/mongo/db/namespace_string.cpp
+++ b/src/mongo/db/namespace_string.cpp
@@ -34,6 +34,38 @@ namespace mongo {
 
     using std::string;
 
+    /*  A map of characters to escape. Instead of printing certain characters we output
+     *  based on the following table.
+     */
+    static const std::string escapeTable[256] = {
+        ".00", ".01", ".02", ".03", ".04", ".05", ".06", ".07", ".08", ".09",
+        ".10", ".11", ".12", ".13", ".14", ".15", ".16", ".17", ".18", ".19",
+        ".20", ".21", ".22", ".23", ".24", ".25", ".26", ".27", ".28", ".29",
+        ".30", ".31", ".32", ".33", ".34", ".35", ".36", ".37", ".38", ".39",
+        ".40", ".41", ".42", ".43", ".44", ".45", ".", ".47", "0", "1",
+        "2", "3", "4", "5", "6", "7", "8", "9", ".58", ".59",
+        ".60", ".61", ".62", ".63", ".64", "A", "B", "C", "D", "E",
+        "F", "G", "H", "I", "J", "K", "L", "M", "N", "O",
+        "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y",
+        "Z", ".91", ".92", ".93", ".94", "_", ".96", "a", "b", "c",
+        "d", "e", "f", "g", "h", "i", "j", "k", "l", "m",
+        "n", "o", "p", "q", "r", "s", "t", "u", "v", "w",
+        "x", "y", "z", ".123", ".124", ".125", ".126", ".127", ".128", ".129",
+        ".130", ".131", ".132", ".133", ".134", ".135", ".136", ".137", ".138", ".139",
+        ".140", ".141", ".142", ".143", ".144", ".145", ".146", ".147", ".148", ".149",
+        ".150", ".151", ".152", ".153", ".154", ".155", ".156", ".157", ".158", ".159",
+        ".160", ".161", ".162", ".163", ".164", ".165", ".166", ".167", ".168", ".169",
+        ".170", ".171", ".172", ".173", ".174", ".175", ".176", ".177", ".178", ".179",
+        ".180", ".181", ".182", ".183", ".184", ".185", ".186", ".187", ".188", ".189",
+        ".190", ".191", ".192", ".193", ".194", ".195", ".196", ".197", ".198", ".199",
+        ".200", ".201", ".202", ".203", ".204", ".205", ".206", ".207", ".208", ".209",
+        ".210", ".211", ".212", ".213", ".214", ".215", ".216", ".217", ".218", ".219",
+        ".220", ".221", ".222", ".223", ".224", ".225", ".226", ".227", ".228", ".229",
+        ".230", ".231", ".232", ".233", ".234", ".235", ".236", ".237", ".238", ".239",
+        ".240", ".241", ".242", ".243", ".244", ".245", ".246", ".247", ".248", ".249",
+        ".250", ".251", ".252", ".253", ".254", ".255"
+    };
+
     bool legalClientSystemNS( StringData ns , bool write ) {
         if( ns == "local.system.replset" ) return true;
 
@@ -68,4 +100,16 @@ namespace mongo {
         return NamespaceString(db(), coll().substr(listIndexesGetMoreNSPrefix.size()));
     }
 
+    string NamespaceString::escapeDbName( const StringData dbname ) {
+        std::string escapedDbName;
+
+        // pre-alloc the return string as it will always be the same as dbname at a minimum.
+        escapedDbName.reserve(dbname.size());
+
+        for (unsigned char c : dbname) {
+            escapedDbName += escapeTable[c];
+        }
+        return escapedDbName;
+    }
+
 }