forked from trdtnguyen/mysql-pmem
-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathauth_utils.h
132 lines (122 loc) · 6.1 KB
/
auth_utils.h
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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
/*
Copyright (c) 2012, 2014, Oracle and/or its affiliates. All rights reserved.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; version 2 of the License.
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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifndef AUTH_UTILS_INCLUDED
#define AUTH_UTILS_INCLUDED
#include <string>
#include <iostream>
#include <map>
#include <auth_acls.h>
#define ERR_FILE 1 // File related error
#define ERR_ENCRYPTION 2 // Encryption related error
#define ERR_SYNTAX 3 // Syntax and parsing related error
#define ERR_OTHER 4 // Unspecified error
#define ERR_NO_SUCH_CATEGORY 5 // The specified category isn't present
#define ALL_OK 0 // Reporting success and good fortune
/**
Trivial parser for the login.cnf file which assumes that first entry
is a [client] header followed by some attribute/value -pairs
@param sin Input stream
@param[out] options Output map
@return success rate
@retval ALL_OK Reporting success and good fortune
@retval ERR_SYNTAX Failed to parse the stream
*/
int parse_cnf_file(std::istream &sin,
std::map<std::string, std::string > *options,
const std::string &header);
/**
Decrypts a file and produces a stringstream.
@param fin Input stream
@param[out] sout Output stream
@return success rate
@retval ALL_OK Reporting success and good fortune
@retval ERR_ENCRYPTION Failed to decrypt the input stream
*/
int decrypt_login_cnf_file(std::istream &fin, std::ostream &sout);
void generate_password(std::string *password, int size);
void trim(std::string *s);
const std::string get_allowed_pwd_chars();
/**
An experimental uniform representation of access privileges in MySQL
*/
class Access_privilege
{
public:
Access_privilege() : m_priv(0) {}
Access_privilege(uint64_t privileges) : m_priv(privileges) {}
Access_privilege(const Access_privilege &priv) : m_priv(priv.m_priv) {}
bool has_select_ac() { return (m_priv & SELECT_ACL) > 0; }
bool has_insert_ac() { return (m_priv & INSERT_ACL) > 0; }
bool has_update_ac() { return (m_priv & UPDATE_ACL) > 0; }
bool has_delete_ac() { return (m_priv & DELETE_ACL) > 0; }
bool has_create_ac() { return (m_priv & CREATE_ACL) > 0; }
bool has_drop_ac() { return (m_priv & DROP_ACL) > 0; }
bool has_relead_ac() { return (m_priv & RELOAD_ACL) > 0; }
bool has_shutdown_ac() { return (m_priv & SHUTDOWN_ACL) > 0; }
bool has_process_ac() { return (m_priv & PROCESS_ACL) > 0; }
bool has_file_ac() { return (m_priv & FILE_ACL) > 0; }
bool has_grant_ac() { return (m_priv & GRANT_ACL) > 0; }
bool has_references_ac() { return (m_priv & REFERENCES_ACL) > 0; }
bool has_index_ac() { return (m_priv & INDEX_ACL) > 0; }
bool has_alter_ac() { return (m_priv & ALTER_ACL) > 0; }
bool has_show_db_ac() { return (m_priv & SHOW_DB_ACL) > 0; }
bool has_super_ac() { return (m_priv & SUPER_ACL) > 0; }
bool has_create_tmp_ac() { return (m_priv & CREATE_TMP_ACL) > 0; }
bool has_lock_tables_ac() { return (m_priv & LOCK_TABLES_ACL) > 0; }
bool has_execute_ac() { return (m_priv & EXECUTE_ACL) > 0; }
bool has_repl_slave_ac() { return (m_priv & REPL_SLAVE_ACL) > 0; }
bool has_repl_client_ac() { return (m_priv & REPL_CLIENT_ACL) > 0; }
bool has_create_view_ac() { return (m_priv & CREATE_VIEW_ACL) > 0; }
bool has_show_view_ac() { return (m_priv & SHOW_VIEW_ACL) > 0; }
bool has_create_proc_ac() { return (m_priv & CREATE_PROC_ACL) > 0; }
bool has_alter_proc_ac() { return (m_priv & ALTER_PROC_ACL) > 0; }
bool has_create_user_ac() { return (m_priv & CREATE_USER_ACL) > 0; }
bool has_event_ac() { return (m_priv & EVENT_ACL) > 0; }
bool has_trigger_ac() { return (m_priv & TRIGGER_ACL) > 0; }
bool has_create_tablespace_ac() { return (m_priv & CREATE_TABLESPACE_ACL) > 0; }
inline static uint64_t select_ac() { return SELECT_ACL; }
inline uint64_t insert_ac() { return INSERT_ACL; }
inline uint64_t update_ac() { return UPDATE_ACL; }
inline uint64_t delete_ac() { return DELETE_ACL; }
inline static uint64_t create_ac() { return CREATE_ACL; }
inline static uint64_t drop_ac() { return DROP_ACL; }
inline static uint64_t relead_ac() { return RELOAD_ACL; }
inline static uint64_t shutdown_ac() { return SHUTDOWN_ACL; }
inline static uint64_t process_ac() { return PROCESS_ACL; }
inline static uint64_t file_ac() { return FILE_ACL; }
inline static uint64_t grant_ac() { return GRANT_ACL; }
inline static uint64_t references_ac() { return REFERENCES_ACL; }
inline static uint64_t index_ac() { return INDEX_ACL; }
inline static uint64_t alter_ac() { return ALTER_ACL; }
inline static uint64_t show_db_ac() { return SHOW_DB_ACL; }
inline static uint64_t super_ac() { return SUPER_ACL; }
inline static uint64_t create_tmp_ac() { return CREATE_TMP_ACL; }
inline static uint64_t lock_tables_ac() { return LOCK_TABLES_ACL; }
inline static uint64_t execute_ac() { return EXECUTE_ACL; }
inline static uint64_t repl_slave_ac() { return REPL_SLAVE_ACL; }
inline static uint64_t repl_client_ac() { return REPL_CLIENT_ACL; }
inline static uint64_t create_view_ac() { return CREATE_VIEW_ACL; }
inline static uint64_t show_view_ac() { return SHOW_VIEW_ACL; }
inline static uint64_t create_proc_ac() { return CREATE_PROC_ACL; }
inline static uint64_t alter_proc_ac() { return ALTER_PROC_ACL; }
inline static uint64_t create_user_ac() { return CREATE_USER_ACL; }
inline static uint64_t event_ac() { return EVENT_ACL; }
inline static uint64_t trigger_ac() { return TRIGGER_ACL; }
inline static uint64_t create_tablespace_ac() { return CREATE_TABLESPACE_ACL; }
inline static uint64_t acl_all() { return ~NO_ACCESS; }
uint64_t to_int() const { return m_priv; };
private:
uint64_t m_priv;
};
#endif