1 | #ifndef SQL_TRIGGER_INCLUDED |
2 | #define SQL_TRIGGER_INCLUDED |
3 | |
4 | /* |
5 | Copyright (c) 2004, 2011, Oracle and/or its affiliates. |
6 | Copyright (c) 2017, MariaDB Corporation. |
7 | |
8 | This program is free software; you can redistribute it and/or modify |
9 | it under the terms of the GNU General Public License as published by |
10 | the Free Software Foundation; version 2 of the License. |
11 | |
12 | This program is distributed in the hope that it will be useful, |
13 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
15 | GNU General Public License for more details. |
16 | |
17 | You should have received a copy of the GNU General Public License |
18 | along with this program; if not, write to the Free Software |
19 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */ |
20 | |
21 | #include <mysqld_error.h> |
22 | |
23 | /* Forward declarations */ |
24 | |
25 | class Item_trigger_field; |
26 | class sp_head; |
27 | class sp_name; |
28 | class Query_tables_list; |
29 | struct TABLE_LIST; |
30 | class Query_tables_list; |
31 | |
32 | /** Event on which trigger is invoked. */ |
33 | enum trg_event_type |
34 | { |
35 | TRG_EVENT_INSERT= 0, |
36 | TRG_EVENT_UPDATE= 1, |
37 | TRG_EVENT_DELETE= 2, |
38 | TRG_EVENT_MAX |
39 | }; |
40 | |
41 | #include "table.h" /* GRANT_INFO */ |
42 | |
43 | /* |
44 | We need this two enums here instead of sql_lex.h because |
45 | at least one of them is used by Item_trigger_field interface. |
46 | |
47 | Time when trigger is invoked (i.e. before or after row actually |
48 | inserted/updated/deleted). |
49 | */ |
50 | enum trg_action_time_type |
51 | { |
52 | TRG_ACTION_BEFORE= 0, TRG_ACTION_AFTER= 1, TRG_ACTION_MAX |
53 | }; |
54 | |
55 | enum trigger_order_type |
56 | { |
57 | TRG_ORDER_NONE= 0, |
58 | TRG_ORDER_FOLLOWS= 1, |
59 | TRG_ORDER_PRECEDES= 2 |
60 | }; |
61 | |
62 | |
63 | struct st_trg_execution_order |
64 | { |
65 | /** |
66 | FOLLOWS or PRECEDES as specified in the CREATE TRIGGER statement. |
67 | */ |
68 | enum trigger_order_type ordering_clause; |
69 | |
70 | /** |
71 | Trigger name referenced in the FOLLOWS/PRECEDES clause of the |
72 | CREATE TRIGGER statement. |
73 | */ |
74 | LEX_CSTRING anchor_trigger_name; |
75 | }; |
76 | |
77 | |
78 | class Table_triggers_list; |
79 | |
80 | /** |
81 | The trigger object |
82 | */ |
83 | |
84 | class Trigger :public Sql_alloc |
85 | { |
86 | public: |
87 | Trigger(Table_triggers_list *base_arg, sp_head *code): |
88 | base(base_arg), body(code), next(0), trigger_fields(0), action_order(0) |
89 | { |
90 | bzero((char *)&subject_table_grants, sizeof(subject_table_grants)); |
91 | } |
92 | ~Trigger(); |
93 | Table_triggers_list *base; |
94 | sp_head *body; |
95 | Trigger *next; /* Next trigger of same type */ |
96 | |
97 | /** |
98 | Heads of the lists linking items for all fields used in triggers |
99 | grouped by event and action_time. |
100 | */ |
101 | Item_trigger_field *trigger_fields; |
102 | LEX_CSTRING name; |
103 | LEX_CSTRING on_table_name; /* Raw table name */ |
104 | LEX_CSTRING definition; |
105 | LEX_CSTRING definer; |
106 | |
107 | /* Character sets used */ |
108 | LEX_CSTRING client_cs_name; |
109 | LEX_CSTRING connection_cl_name; |
110 | LEX_CSTRING db_cl_name; |
111 | |
112 | GRANT_INFO subject_table_grants; |
113 | sql_mode_t sql_mode; |
114 | /* Store create time. Can't be mysql_time_t as this holds also sub seconds */ |
115 | ulonglong create_time; |
116 | trg_event_type event; |
117 | trg_action_time_type action_time; |
118 | uint action_order; |
119 | |
120 | bool is_fields_updated_in_trigger(MY_BITMAP *used_fields); |
121 | void get_trigger_info(LEX_CSTRING *stmt, LEX_CSTRING *body, |
122 | LEX_STRING *definer); |
123 | /* Functions executed over each active trigger */ |
124 | bool change_on_table_name(void* param_arg); |
125 | bool change_table_name(void* param_arg); |
126 | bool add_to_file_list(void* param_arg); |
127 | }; |
128 | |
129 | typedef bool (Trigger::*Triggers_processor)(void *arg); |
130 | |
131 | /** |
132 | This class holds all information about triggers of table. |
133 | */ |
134 | |
135 | class Table_triggers_list: public Sql_alloc |
136 | { |
137 | friend class Trigger; |
138 | |
139 | /* Points to first trigger for a certain type */ |
140 | Trigger *triggers[TRG_EVENT_MAX][TRG_ACTION_MAX]; |
141 | /** |
142 | Copy of TABLE::Field array which all fields made nullable |
143 | (using extra_null_bitmap, if needed). Used for NEW values in |
144 | BEFORE INSERT/UPDATE triggers. |
145 | */ |
146 | Field **record0_field; |
147 | uchar *; |
148 | /** |
149 | Copy of TABLE::Field array with field pointers set to TABLE::record[1] |
150 | buffer instead of TABLE::record[0] (used for OLD values in on UPDATE |
151 | trigger and DELETE trigger when it is called for REPLACE). |
152 | */ |
153 | Field **record1_field; |
154 | /** |
155 | During execution of trigger new_field and old_field should point to the |
156 | array of fields representing new or old version of row correspondingly |
157 | (so it can point to TABLE::field or to Tale_triggers_list::record1_field) |
158 | */ |
159 | Field **new_field; |
160 | Field **old_field; |
161 | |
162 | /* TABLE instance for which this triggers list object was created */ |
163 | TABLE *trigger_table; |
164 | |
165 | /** |
166 | This flag indicates that one of the triggers was not parsed successfully, |
167 | and as a precaution the object has entered a state where all trigger |
168 | access results in errors until all such triggers are dropped. It is not |
169 | safe to add triggers since we don't know if the broken trigger has the |
170 | same name or event type. Nor is it safe to invoke any trigger for the |
171 | aforementioned reasons. The only safe operations are drop_trigger and |
172 | drop_all_triggers. |
173 | |
174 | @see Table_triggers_list::set_parse_error |
175 | */ |
176 | bool m_has_unparseable_trigger; |
177 | |
178 | /** |
179 | This error will be displayed when the user tries to manipulate or invoke |
180 | triggers on a table that has broken triggers. It will get set only once |
181 | per statement and thus will contain the first parse error encountered in |
182 | the trigger file. |
183 | */ |
184 | char m_parse_error_message[MYSQL_ERRMSG_SIZE]; |
185 | uint count; /* Number of triggers */ |
186 | |
187 | public: |
188 | /** |
189 | Field responsible for storing triggers definitions in file. |
190 | It have to be public because we are using it directly from parser. |
191 | */ |
192 | List<LEX_CSTRING> definitions_list; |
193 | /** |
194 | List of sql modes for triggers |
195 | */ |
196 | List<ulonglong> definition_modes_list; |
197 | /** Create times for triggers */ |
198 | List<ulonglong> create_times; |
199 | |
200 | List<LEX_CSTRING> definers_list; |
201 | |
202 | /* Character set context, used for parsing and executing triggers. */ |
203 | |
204 | List<LEX_CSTRING> client_cs_names; |
205 | List<LEX_CSTRING> connection_cl_names; |
206 | List<LEX_CSTRING> db_cl_names; |
207 | |
208 | /* End of character ser context. */ |
209 | |
210 | Table_triggers_list(TABLE *table_arg) |
211 | :record0_field(0), extra_null_bitmap(0), record1_field(0), |
212 | trigger_table(table_arg), |
213 | m_has_unparseable_trigger(false), count(0) |
214 | { |
215 | bzero((char *) triggers, sizeof(triggers)); |
216 | } |
217 | ~Table_triggers_list(); |
218 | |
219 | bool create_trigger(THD *thd, TABLE_LIST *table, String *stmt_query); |
220 | bool drop_trigger(THD *thd, TABLE_LIST *table, String *stmt_query); |
221 | bool process_triggers(THD *thd, trg_event_type event, |
222 | trg_action_time_type time_type, |
223 | bool old_row_is_record1); |
224 | void empty_lists(); |
225 | bool create_lists_needed_for_files(MEM_ROOT *root); |
226 | bool save_trigger_file(THD *thd, const LEX_CSTRING *db, const LEX_CSTRING *table_name); |
227 | |
228 | static bool check_n_load(THD *thd, const LEX_CSTRING *db, const LEX_CSTRING *table_name, |
229 | TABLE *table, bool names_only); |
230 | static bool drop_all_triggers(THD *thd, const LEX_CSTRING *db, |
231 | const LEX_CSTRING *table_name); |
232 | static bool change_table_name(THD *thd, const LEX_CSTRING *db, |
233 | const LEX_CSTRING *old_alias, |
234 | const LEX_CSTRING *old_table, |
235 | const LEX_CSTRING *new_db, |
236 | const LEX_CSTRING *new_table); |
237 | void add_trigger(trg_event_type event_type, |
238 | trg_action_time_type action_time, |
239 | trigger_order_type ordering_clause, |
240 | LEX_CSTRING *anchor_trigger_name, |
241 | Trigger *trigger); |
242 | Trigger *get_trigger(trg_event_type event_type, |
243 | trg_action_time_type action_time) |
244 | { |
245 | return triggers[event_type][action_time]; |
246 | } |
247 | /* Simpler version of the above, to avoid casts in the code */ |
248 | Trigger *get_trigger(uint event_type, uint action_time) |
249 | { |
250 | return get_trigger((trg_event_type) event_type, |
251 | (trg_action_time_type) action_time); |
252 | } |
253 | |
254 | bool has_triggers(trg_event_type event_type, |
255 | trg_action_time_type action_time) |
256 | { |
257 | return get_trigger(event_type,action_time) != 0; |
258 | } |
259 | bool has_delete_triggers() |
260 | { |
261 | return (has_triggers(TRG_EVENT_DELETE,TRG_ACTION_BEFORE) || |
262 | has_triggers(TRG_EVENT_DELETE,TRG_ACTION_AFTER)); |
263 | } |
264 | |
265 | void mark_fields_used(trg_event_type event); |
266 | |
267 | void set_parse_error_message(char *error_message); |
268 | |
269 | friend class Item_trigger_field; |
270 | |
271 | bool add_tables_and_routines_for_triggers(THD *thd, |
272 | Query_tables_list *prelocking_ctx, |
273 | TABLE_LIST *table_list); |
274 | |
275 | Field **nullable_fields() { return record0_field; } |
276 | void () |
277 | { |
278 | size_t null_bytes= (trigger_table->s->stored_fields - |
279 | trigger_table->s->null_fields + 7)/8; |
280 | bzero(extra_null_bitmap, null_bytes); |
281 | } |
282 | |
283 | Trigger *find_trigger(const LEX_CSTRING *name, bool remove_from_list); |
284 | |
285 | Trigger* for_all_triggers(Triggers_processor func, void *arg); |
286 | |
287 | private: |
288 | bool prepare_record_accessors(TABLE *table); |
289 | Trigger *change_table_name_in_trignames(const LEX_CSTRING *old_db_name, |
290 | const LEX_CSTRING *new_db_name, |
291 | const LEX_CSTRING *new_table_name, |
292 | Trigger *trigger); |
293 | bool change_table_name_in_triggers(THD *thd, |
294 | const LEX_CSTRING *old_db_name, |
295 | const LEX_CSTRING *new_db_name, |
296 | const LEX_CSTRING *old_table_name, |
297 | const LEX_CSTRING *new_table_name); |
298 | |
299 | bool check_for_broken_triggers() |
300 | { |
301 | if (m_has_unparseable_trigger) |
302 | { |
303 | my_message(ER_PARSE_ERROR, m_parse_error_message, MYF(0)); |
304 | return true; |
305 | } |
306 | return false; |
307 | } |
308 | }; |
309 | |
310 | bool add_table_for_trigger(THD *thd, |
311 | const sp_name *trg_name, |
312 | bool continue_if_not_exist, |
313 | TABLE_LIST **table); |
314 | |
315 | void build_trn_path(THD *thd, const sp_name *trg_name, LEX_STRING *trn_path); |
316 | |
317 | bool check_trn_exists(const LEX_CSTRING *trn_path); |
318 | |
319 | bool load_table_name_for_trigger(THD *thd, |
320 | const sp_name *trg_name, |
321 | const LEX_CSTRING *trn_path, |
322 | LEX_CSTRING *tbl_name); |
323 | bool mysql_create_or_drop_trigger(THD *thd, TABLE_LIST *tables, bool create); |
324 | |
325 | extern const char * const TRG_EXT; |
326 | extern const char * const TRN_EXT; |
327 | |
328 | #endif /* SQL_TRIGGER_INCLUDED */ |
329 | |