summaryrefslogtreecommitdiff
path: root/src/mongo/db/namespace_string.cpp
blob: fe50e2039befc9ebb996ed951680233fabb21104 (plain)
1
2
3
4
5
6
7
8
9
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
// namespace_string.cpp

/**
*    Copyright (C) 2014 MongoDB Inc.
*
*    This program is free software: you can redistribute it and/or  modify
*    it under the terms of the GNU Affero General Public License, version 3,
*    as published by the Free Software Foundation.
*
*    This program is distributed in the hope that it will be useful,
*    but WITHOUT ANY WARRANTY; without even the implied warranty of
*    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
*    GNU Affero General Public License for more details.
*
*    You should have received a copy of the GNU Affero General Public License
*    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*
*    As a special exception, the copyright holders give permission to link the
*    code of portions of this program with the OpenSSL library under certain
*    conditions as described in each individual source file and distribute
*    linked combinations including the program with the OpenSSL library. You
*    must comply with the GNU Affero General Public License in all respects for
*    all of the code used other than as permitted herein. If you modify file(s)
*    with this exception, you may extend this exception to your version of the
*    file(s), but you are not obligated to do so. If you do not wish to do so,
*    delete this exception statement from your version. If you delete this
*    exception statement from all source files in the program, then also delete
*    it in the license file.
*/

#include "mongo/db/namespace_string.h"

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;

        if ( ns.find( ".system.users" ) != string::npos )
            return true;

        if ( ns == "admin.system.roles" ) return true;
        if ( ns == "admin.system.version" ) return true;
        if ( ns == "admin.system.new_users" ) return true;
        if ( ns == "admin.system.backup_users" ) return true;

        if ( ns.find( ".system.js" ) != string::npos ) return true;

        return false;
    }

    bool NamespaceString::isListCollectionsGetMore() const {
        return coll() == StringData("$cmd.listCollections", StringData::LiteralTag());
    }

    namespace {
        const StringData listIndexesGetMoreNSPrefix("$cmd.listIndexes.", StringData::LiteralTag());
    }  // namespace

    bool NamespaceString::isListIndexesGetMore() const {
        return coll().size() > listIndexesGetMoreNSPrefix.size() &&
               coll().startsWith(listIndexesGetMoreNSPrefix);
    }

    NamespaceString NamespaceString::getTargetNSForListIndexesGetMore() const {
        dassert(isListIndexesGetMore());
        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;
    }

}