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
|
/* Copyright (c) 2019, 2025, Oracle and/or its affiliates.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License, version 2.0,
as published by the Free Software Foundation.
This program is designed to work with certain software (including
but not limited to OpenSSL) that is licensed under separate terms,
as designated in a particular file or component or in included license
documentation. The authors of MySQL hereby grant you an additional
permission to link the program and your derivative works with the
separately licensed software that they have either included with
the program or referenced in the documentation.
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, version 2.0, 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 TEST_UDF_EXTENSION_INCLUDED
#define TEST_UDF_EXTENSION_INCLUDED
#include <mysql/udf_registration_types.h>
#include <sstream>
#include <string>
namespace udf_ext {
enum class Type { charset = 0, collation = 1 };
/*
A stateless base class that implements the methods needed to test the
character set/collation of UDF's arguments and return value. It implements
the methods which are shared by the derived classes.
*/
class Test_udf_charset_base {
public:
static std::string get_last_error();
protected:
static bool validate_inputs(UDF_ARGS *args, const size_t expected_arg_count);
static bool set_udf_init(UDF_INIT *initid, UDF_ARGS *args);
static bool set_args_init(UDF_ARGS *args, const std::string &name);
static bool set_return_value_charset_or_collation(UDF_INIT *initid,
const std::string &name);
static void set_ext_type(Type ext_type);
static bool run_return_udf(UDF_INIT *initd, UDF_ARGS *args, char **result,
unsigned long &result_len);
static bool run_args_udf(UDF_INIT *initd, UDF_ARGS *args, char **result,
unsigned long &result_len);
static bool convert(const std::string &out_charset_name,
const std::string &in_charset_name,
const std::string &in_buffer, size_t out_buffer_length,
char *out_buffer);
static void deinit(UDF_INIT *initd);
static std::stringstream s_message;
static std::string s_ext_type;
};
/**
A stateless class that provides the implementation of the UDFs which test the
Character set extension argument. It determines the charset or collation names
of the second UDF argument and use that to convert either return value or
first UDF argument.
*/
class Test_udf_charset : public Test_udf_charset_base {
public:
static bool prepare_return_udf(UDF_INIT *initid, UDF_ARGS *args,
const size_t expected_arg_count, Type type);
static bool prepare_args_udf(UDF_INIT *initid, UDF_ARGS *args,
const size_t expected_arg_count, Type type);
static bool run_return_udf(UDF_INIT *initd, UDF_ARGS *args, char **result,
unsigned long &result_len);
static bool run_args_udf(UDF_INIT *initd, UDF_ARGS *args, char **result,
unsigned long &result_len);
static void deinit(UDF_INIT *initd);
private:
static bool fetch_charset_or_collation_from_arg(UDF_ARGS *args,
const int index,
std::string &name);
};
/**
A stateless class that provides the implementation of the UDFs which test the
Character set extension argument. It determines the charset or collation names
from the second UDF argument, and use that to convert either return value or
first UDF argument.
*/
class Test_udf_charset_const_value : public Test_udf_charset_base {
public:
static bool prepare_return_udf(UDF_INIT *initid, UDF_ARGS *args,
const size_t expected_arg_count, Type type);
static bool prepare_args_udf(UDF_INIT *initid, UDF_ARGS *args,
const size_t expected_arg_count, Type type);
static bool run_return_udf(UDF_INIT *initd, UDF_ARGS *args, char **result,
unsigned long &result_len);
static bool run_args_udf(UDF_INIT *initd, UDF_ARGS *args, char **result,
unsigned long &result_len);
static void deinit(UDF_INIT *initd);
private:
static bool fetch_charset_or_collation_from_arg(UDF_ARGS *args,
const int index,
std::string &name);
};
} // namespace udf_ext
#endif // !TEST_UDF_EXTENSION_INCLUDED
|
