← Back to branch summary

~clint-fewbar/drizzle/authorize-default-deny

« back to all changes in this revision

Viewing changes to client/server_detect.h

  • Committer: Clint Byrum
  • Date: 2011-02-22 22:25:07 UTC
  • mfrom: (2177.1.15 build)
  • Revision ID: clint@ubuntu.com-20110222222507-azaofg4p8w55gdvr
merging trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
client/server_detect.h
 
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
 
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
3
 *
 
4
 *  Copyright (C) 2011 Andrew Hutchings
 
5
 *
 
6
 *  This program is free software; you can redistribute it and/or modify
 
7
 *  it under the terms of the GNU General Public License as published by
 
8
 *  the Free Software Foundation; version 2 of the License.
 
9
 *
 
10
 *  This program is distributed in the hope that it will be useful,
 
11
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
 *  GNU General Public License for more details.
 
14
 *
 
15
 *  You should have received a copy of the GNU General Public License
 
16
 *  along with this program; if not, write to the Free Software
 
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
18
 */
 
19
 
 
20
#ifndef CLIENT_SERVER_DETECT_H
 
21
#define CLIENT_SERVER_DETECT_H
 
22
 
 
23
#include <boost/regex.hpp>
 
24
 
 
25
class ServerDetect
 
26
{
 
27
  public:
 
28
    enum server_type {
 
29
      SERVER_MYSQL_FOUND,
 
30
      SERVER_DRIZZLE_FOUND,
 
31
      SERVER_UNKNOWN_FOUND
 
32
    };
 
33
 
 
34
    server_type getServerType() { return type; }
 
35
    std::string& getServerVersion() { return version; }
 
36
 
 
37
    ServerDetect(drizzle_con_st *connection) :
 
38
      type(SERVER_UNKNOWN_FOUND),
 
39
      version("")
 
40
    {
 
41
      boost::match_flag_type flags = boost::match_default;
 
42
 
 
43
      boost::regex mysql_regex("(5\\.[0-9]+\\.[0-9]+)");
 
44
      boost::regex drizzle_regex("(20[0-9]{2}\\.(0[1-9]|1[012])\\.[0-9]+)");
 
45
 
 
46
      version= drizzle_con_server_version(connection);
 
47
 
 
48
      if (regex_search(version, mysql_regex, flags))
 
49
        type= SERVER_MYSQL_FOUND;
 
50
      else if (regex_search(version, drizzle_regex, flags))
 
51
        type= SERVER_DRIZZLE_FOUND;
 
52
      else
 
53
        type= SERVER_UNKNOWN_FOUND;
 
54
    }
 
55
 
 
56
  private:
 
57
    server_type type;
 
58
    std::string version;
 
59
};
 
60
 
 
61
#endif /* CLIENT_SERVER_DETECT_H */