summaryrefslogtreecommitdiff
path: root/src/mongo/util/base64.h
blob: 505b5d78ccac7fbea0b738a5505f6bc36ef349dc (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
// util/base64.h

/*    Copyright 2009 10gen Inc.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

#pragma once

namespace mongo {
    namespace base64 {

        class Alphabet {
        public:
            Alphabet()
                : encode((unsigned char*)
                         "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
                         "abcdefghijklmnopqrstuvwxyz"
                         "0123456789"
                         "+/")
                , decode(new unsigned char[257]) {
                memset( decode.get() , 0 , 256 );
                for ( int i=0; i<64; i++ ) {
                    decode[ encode[i] ] = i;
                }

                test();
            }
            void test() {
                assert( strlen( (char*)encode ) == 64 );
                for ( int i=0; i<26; i++ )
                    assert( encode[i] == toupper( encode[i+26] ) );
            }

            char e( int x ) {
                return encode[x&0x3f];
            }

        private:
            const unsigned char * encode;
        public:
            boost::scoped_array<unsigned char> decode;
        };

        extern Alphabet alphabet;


        void encode( stringstream& ss , const char * data , int size );
        string encode( const char * data , int size );
        string encode( const string& s );

        void decode( stringstream& ss , const string& s );
        string decode( const string& s );


        void testAlphabet();
    }
}