1 | /* Copyright (c) 1995, 2011, Oracle and/or its affiliates. All rights reserved. |
2 | |
3 | This program is free software; you can redistribute it and/or modify |
4 | it under the terms of the GNU General Public License as published by |
5 | the Free Software Foundation; version 2 of the License. |
6 | |
7 | This program is distributed in the hope that it will be useful, |
8 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
9 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
10 | GNU General Public License for more details. |
11 | |
12 | You should have received a copy of the GNU General Public License |
13 | along with this program; if not, write to the Free Software |
14 | Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA */ |
15 | |
16 | /********************************************************************** |
17 | This file contains the implementation of error and warnings related |
18 | |
19 | - Whenever an error or warning occurred, it pushes it to a warning list |
20 | that the user can retrieve with SHOW WARNINGS or SHOW ERRORS. |
21 | |
22 | - For each statement, we return the number of warnings generated from this |
23 | command. Note that this can be different from @@warning_count as |
24 | we reset the warning list only for questions that uses a table. |
25 | This is done to allow on to do: |
26 | INSERT ...; |
27 | SELECT @@warning_count; |
28 | SHOW WARNINGS; |
29 | (If we would reset after each command, we could not retrieve the number |
30 | of warnings) |
31 | |
32 | - When client requests the information using SHOW command, then |
33 | server processes from this list and returns back in the form of |
34 | resultset. |
35 | |
36 | Supported syntaxes: |
37 | |
38 | SHOW [COUNT(*)] ERRORS [LIMIT [offset,] rows] |
39 | SHOW [COUNT(*)] WARNINGS [LIMIT [offset,] rows] |
40 | SELECT @@warning_count, @@error_count; |
41 | |
42 | ***********************************************************************/ |
43 | |
44 | #include "mariadb.h" |
45 | #include "sql_priv.h" |
46 | #include "unireg.h" |
47 | #include "sql_error.h" |
48 | #include "sp_rcontext.h" |
49 | |
50 | /* |
51 | Design notes about Sql_condition::m_message_text. |
52 | |
53 | The member Sql_condition::m_message_text contains the text associated with |
54 | an error, warning or note (which are all SQL 'conditions') |
55 | |
56 | Producer of Sql_condition::m_message_text: |
57 | ---------------------------------------- |
58 | |
59 | (#1) the server implementation itself, when invoking functions like |
60 | my_error() or push_warning() |
61 | |
62 | (#2) user code in stored programs, when using the SIGNAL statement. |
63 | |
64 | (#3) user code in stored programs, when using the RESIGNAL statement. |
65 | |
66 | When invoking my_error(), the error number and message is typically |
67 | provided like this: |
68 | - my_error(ER_WRONG_DB_NAME, MYF(0), ...); |
69 | - my_message(ER_SLAVE_IGNORED_TABLE, ER(ER_SLAVE_IGNORED_TABLE), MYF(0)); |
70 | |
71 | In both cases, the message is retrieved from ER(ER_XXX), which in turn |
72 | is read from the resource file errmsg.sys at server startup. |
73 | The strings stored in the errmsg.sys file are expressed in the character set |
74 | that corresponds to the server --language start option |
75 | (see error_message_charset_info). |
76 | |
77 | When executing: |
78 | - a SIGNAL statement, |
79 | - a RESIGNAL statement, |
80 | the message text is provided by the user logic, and is expressed in UTF8. |
81 | |
82 | Storage of Sql_condition::m_message_text: |
83 | --------------------------------------- |
84 | |
85 | (#4) The class Sql_condition is used to hold the message text member. |
86 | This class represents a single SQL condition. |
87 | |
88 | (#5) The class Warning_info represents a SQL condition area, and contains |
89 | a collection of SQL conditions in the Warning_info::m_warn_list |
90 | |
91 | Consumer of Sql_condition::m_message_text: |
92 | ---------------------------------------- |
93 | |
94 | (#6) The statements SHOW WARNINGS and SHOW ERRORS display the content of |
95 | the warning list. |
96 | |
97 | (#7) The GET DIAGNOSTICS statement (planned, not implemented yet) will |
98 | also read the content of: |
99 | - the top level statement condition area (when executed in a query), |
100 | - a sub statement (when executed in a stored program) |
101 | and return the data stored in a Sql_condition. |
102 | |
103 | (#8) The RESIGNAL statement reads the Sql_condition caught by an exception |
104 | handler, to raise a new or modified condition (in #3). |
105 | |
106 | The big picture |
107 | --------------- |
108 | -------------- |
109 | | ^ |
110 | V | |
111 | my_error(#1) SIGNAL(#2) RESIGNAL(#3) | |
112 | |(#A) |(#B) |(#C) | |
113 | | | | | |
114 | ----------------------------|---------------------------- | |
115 | | | |
116 | V | |
117 | Sql_condition(#4) | |
118 | | | |
119 | | | |
120 | V | |
121 | Warning_info(#5) | |
122 | | | |
123 | ----------------------------------------------------- | |
124 | | | | | |
125 | | | | | |
126 | | | | | |
127 | V V V | |
128 | SHOW WARNINGS(#6) GET DIAGNOSTICS(#7) RESIGNAL(#8) | |
129 | | | | | | |
130 | | -------- | V | |
131 | | | | -------------- |
132 | V | | |
133 | Connectors | | |
134 | | | | |
135 | ------------------------- |
136 | | |
137 | V |
138 | Client application |
139 | |
140 | Current implementation status |
141 | ----------------------------- |
142 | |
143 | (#1) (my_error) produces data in the 'error_message_charset_info' CHARSET |
144 | |
145 | (#2) and (#3) (SIGNAL, RESIGNAL) produces data internally in UTF8 |
146 | |
147 | (#6) (SHOW WARNINGS) produces data in the 'error_message_charset_info' CHARSET |
148 | |
149 | (#7) (GET DIAGNOSTICS) is not implemented. |
150 | |
151 | (#8) (RESIGNAL) produces data internally in UTF8 (see #3) |
152 | |
153 | As a result, the design choice for (#4) and (#5) is to store data in |
154 | the 'error_message_charset_info' CHARSET, to minimize impact on the code base. |
155 | This is implemented by using 'String Sql_condition::m_message_text'. |
156 | |
157 | The UTF8 -> error_message_charset_info conversion is implemented in |
158 | Sql_cmd_common_signal::eval_signal_informations() (for path #B and #C). |
159 | |
160 | Future work |
161 | ----------- |
162 | |
163 | - Change (#1) (my_error) to generate errors in UTF8. |
164 | See WL#751 (Recoding of error messages) |
165 | |
166 | - Change (#4 and #5) to store message text in UTF8 natively. |
167 | In practice, this means changing the type of the message text to |
168 | '<UTF8 String 128 class> Sql_condition::m_message_text', and is a direct |
169 | consequence of WL#751. |
170 | |
171 | - Implement (#9) (GET DIAGNOSTICS). |
172 | See WL#2111 (Stored Procedures: Implement GET DIAGNOSTICS) |
173 | */ |
174 | |
175 | |
176 | static void copy_string(MEM_ROOT *mem_root, String* dst, const String* src) |
177 | { |
178 | size_t len= src->length(); |
179 | if (len) |
180 | { |
181 | char* copy= (char*) alloc_root(mem_root, len + 1); |
182 | if (copy) |
183 | { |
184 | memcpy(copy, src->ptr(), len); |
185 | copy[len]= '\0'; |
186 | dst->set(copy, len, src->charset()); |
187 | } |
188 | } |
189 | else |
190 | dst->length(0); |
191 | } |
192 | |
193 | void |
194 | Sql_condition::copy_opt_attributes(const Sql_condition *cond) |
195 | { |
196 | DBUG_ASSERT(this != cond); |
197 | copy_string(m_mem_root, & m_class_origin, & cond->m_class_origin); |
198 | copy_string(m_mem_root, & m_subclass_origin, & cond->m_subclass_origin); |
199 | copy_string(m_mem_root, & m_constraint_catalog, & cond->m_constraint_catalog); |
200 | copy_string(m_mem_root, & m_constraint_schema, & cond->m_constraint_schema); |
201 | copy_string(m_mem_root, & m_constraint_name, & cond->m_constraint_name); |
202 | copy_string(m_mem_root, & m_catalog_name, & cond->m_catalog_name); |
203 | copy_string(m_mem_root, & m_schema_name, & cond->m_schema_name); |
204 | copy_string(m_mem_root, & m_table_name, & cond->m_table_name); |
205 | copy_string(m_mem_root, & m_column_name, & cond->m_column_name); |
206 | copy_string(m_mem_root, & m_cursor_name, & cond->m_cursor_name); |
207 | } |
208 | |
209 | |
210 | void |
211 | Sql_condition::set_builtin_message_text(const char* str) |
212 | { |
213 | /* |
214 | See the comments |
215 | "Design notes about Sql_condition::m_message_text." |
216 | */ |
217 | const char* copy; |
218 | |
219 | copy= strdup_root(m_mem_root, str); |
220 | m_message_text.set(copy, strlen(copy), error_message_charset_info); |
221 | DBUG_ASSERT(! m_message_text.is_alloced()); |
222 | } |
223 | |
224 | const char* |
225 | Sql_condition::get_message_text() const |
226 | { |
227 | return m_message_text.ptr(); |
228 | } |
229 | |
230 | int |
231 | Sql_condition::get_message_octet_length() const |
232 | { |
233 | return m_message_text.length(); |
234 | } |
235 | |
236 | |
237 | void Sql_state_errno_level::assign_defaults(const Sql_state_errno *from) |
238 | { |
239 | DBUG_ASSERT(from); |
240 | int sqlerrno= from->get_sql_errno(); |
241 | /* |
242 | SIGNAL is restricted in sql_yacc.yy to only signal SQLSTATE conditions. |
243 | */ |
244 | DBUG_ASSERT(from->has_sql_state()); |
245 | set_sqlstate(from); |
246 | /* SQLSTATE class "00": illegal, rejected in the parser. */ |
247 | DBUG_ASSERT(m_sqlstate[0] != '0' || get_sqlstate()[1] != '0'); |
248 | |
249 | if (Sql_state::is_warning()) /* SQLSTATE class "01": warning. */ |
250 | { |
251 | m_level= Sql_condition::WARN_LEVEL_WARN; |
252 | m_sql_errno= sqlerrno ? sqlerrno : ER_SIGNAL_WARN; |
253 | } |
254 | else if (Sql_state::is_not_found()) /* SQLSTATE class "02": not found. */ |
255 | { |
256 | m_level= Sql_condition::WARN_LEVEL_ERROR; |
257 | m_sql_errno= sqlerrno ? sqlerrno : ER_SIGNAL_NOT_FOUND; |
258 | } |
259 | else /* other SQLSTATE classes : error. */ |
260 | { |
261 | m_level= Sql_condition::WARN_LEVEL_ERROR; |
262 | m_sql_errno= sqlerrno ? sqlerrno : ER_SIGNAL_EXCEPTION; |
263 | } |
264 | } |
265 | |
266 | |
267 | void Sql_condition::assign_defaults(THD *thd, const Sql_state_errno *from) |
268 | { |
269 | if (from) |
270 | Sql_state_errno_level::assign_defaults(from); |
271 | if (!get_message_text()) |
272 | set_builtin_message_text(ER(get_sql_errno())); |
273 | } |
274 | |
275 | |
276 | Diagnostics_area::Diagnostics_area(bool initialize) |
277 | : is_bulk_execution(0), m_main_wi(0, false, initialize) |
278 | { |
279 | push_warning_info(&m_main_wi); |
280 | |
281 | reset_diagnostics_area(); |
282 | } |
283 | |
284 | Diagnostics_area::Diagnostics_area(ulonglong warning_info_id, |
285 | bool allow_unlimited_warnings, |
286 | bool initialize) |
287 | : is_bulk_execution(0), |
288 | m_main_wi(warning_info_id, allow_unlimited_warnings, initialize) |
289 | { |
290 | push_warning_info(&m_main_wi); |
291 | |
292 | reset_diagnostics_area(); |
293 | } |
294 | |
295 | /** |
296 | Clear this diagnostics area. |
297 | |
298 | Normally called at the end of a statement. |
299 | */ |
300 | |
301 | void |
302 | Diagnostics_area::reset_diagnostics_area() |
303 | { |
304 | DBUG_ENTER("reset_diagnostics_area" ); |
305 | m_skip_flush= FALSE; |
306 | #ifdef DBUG_OFF |
307 | m_can_overwrite_status= FALSE; |
308 | /** Don't take chances in production */ |
309 | m_message[0]= '\0'; |
310 | Sql_state_errno::clear(); |
311 | Sql_user_condition_identity::clear(); |
312 | m_affected_rows= 0; |
313 | m_last_insert_id= 0; |
314 | m_statement_warn_count= 0; |
315 | #endif |
316 | get_warning_info()->clear_error_condition(); |
317 | set_is_sent(false); |
318 | /** Tiny reset in debug mode to see garbage right away */ |
319 | m_status= DA_EMPTY; |
320 | DBUG_VOID_RETURN; |
321 | } |
322 | |
323 | |
324 | /** |
325 | Set OK status -- ends commands that do not return a |
326 | result set, e.g. INSERT/UPDATE/DELETE. |
327 | */ |
328 | |
329 | void |
330 | Diagnostics_area::set_ok_status(ulonglong affected_rows, |
331 | ulonglong last_insert_id, |
332 | const char *message) |
333 | { |
334 | DBUG_ENTER("set_ok_status" ); |
335 | DBUG_ASSERT(!is_set() || (m_status == DA_OK_BULK && is_bulk_op())); |
336 | /* |
337 | In production, refuse to overwrite an error or a custom response |
338 | with an OK packet. |
339 | */ |
340 | if (unlikely(is_error() || is_disabled())) |
341 | return; |
342 | /* |
343 | When running a bulk operation, m_status will be DA_OK for the first |
344 | operation and set to DA_OK_BULK for all following operations. |
345 | */ |
346 | if (m_status == DA_OK_BULK) |
347 | { |
348 | m_statement_warn_count+= current_statement_warn_count(); |
349 | m_affected_rows+= affected_rows; |
350 | } |
351 | else |
352 | { |
353 | m_statement_warn_count= current_statement_warn_count(); |
354 | m_affected_rows= affected_rows; |
355 | m_status= (is_bulk_op() ? DA_OK_BULK : DA_OK); |
356 | } |
357 | m_last_insert_id= last_insert_id; |
358 | if (message) |
359 | strmake_buf(m_message, message); |
360 | else |
361 | m_message[0]= '\0'; |
362 | DBUG_VOID_RETURN; |
363 | } |
364 | |
365 | |
366 | /** |
367 | Set EOF status. |
368 | */ |
369 | |
370 | void |
371 | Diagnostics_area::set_eof_status(THD *thd) |
372 | { |
373 | DBUG_ENTER("set_eof_status" ); |
374 | /* Only allowed to report eof if has not yet reported an error */ |
375 | DBUG_ASSERT(! is_set()); |
376 | /* |
377 | In production, refuse to overwrite an error or a custom response |
378 | with an EOF packet. |
379 | */ |
380 | if (unlikely(is_error() || is_disabled())) |
381 | return; |
382 | |
383 | /* |
384 | If inside a stored procedure, do not return the total |
385 | number of warnings, since they are not available to the client |
386 | anyway. |
387 | */ |
388 | m_statement_warn_count= (thd->spcont ? |
389 | 0 : |
390 | current_statement_warn_count()); |
391 | |
392 | m_status= DA_EOF; |
393 | DBUG_VOID_RETURN; |
394 | } |
395 | |
396 | /** |
397 | Set ERROR status in the Diagnostics Area. This function should be used to |
398 | report fatal errors (such as out-of-memory errors) when no further |
399 | processing is possible. |
400 | |
401 | @param sql_errno SQL-condition error number |
402 | */ |
403 | |
404 | void |
405 | Diagnostics_area::set_error_status(uint sql_errno) |
406 | { |
407 | set_error_status(sql_errno, |
408 | ER(sql_errno), |
409 | mysql_errno_to_sqlstate(sql_errno), |
410 | Sql_user_condition_identity(), |
411 | NULL); |
412 | } |
413 | |
414 | |
415 | /** |
416 | Set ERROR status in the Diagnostics Area. |
417 | |
418 | @note error_condition may be NULL. It happens if a) OOM error is being |
419 | reported; or b) when Warning_info is full. |
420 | |
421 | @param sql_errno SQL-condition error number |
422 | @param message SQL-condition message |
423 | @param sqlstate SQL-condition state |
424 | @param ucid User defined condition identity |
425 | @param error_condition SQL-condition object representing the error state |
426 | |
427 | @note Note, that error_condition may be NULL. It happens if a) OOM error is |
428 | being reported; or b) when Warning_info is full. |
429 | */ |
430 | |
431 | void |
432 | Diagnostics_area::set_error_status(uint sql_errno, |
433 | const char *message, |
434 | const char *sqlstate, |
435 | const Sql_user_condition_identity &ucid, |
436 | const Sql_condition *error_condition) |
437 | { |
438 | DBUG_ENTER("set_error_status" ); |
439 | DBUG_PRINT("enter" , ("error: %d" , sql_errno)); |
440 | /* |
441 | Only allowed to report error if has not yet reported a success |
442 | The only exception is when we flush the message to the client, |
443 | an error can happen during the flush. |
444 | */ |
445 | DBUG_ASSERT(! is_set() || m_can_overwrite_status); |
446 | |
447 | // message must be set properly by the caller. |
448 | DBUG_ASSERT(message); |
449 | |
450 | // sqlstate must be set properly by the caller. |
451 | DBUG_ASSERT(sqlstate); |
452 | |
453 | #ifdef DBUG_OFF |
454 | /* |
455 | In production, refuse to overwrite a custom response with an |
456 | ERROR packet. |
457 | */ |
458 | if (is_disabled()) |
459 | return; |
460 | #endif |
461 | |
462 | Sql_state_errno::set(sql_errno, sqlstate); |
463 | Sql_user_condition_identity::set(ucid); |
464 | strmake_buf(m_message, message); |
465 | |
466 | get_warning_info()->set_error_condition(error_condition); |
467 | |
468 | m_status= DA_ERROR; |
469 | DBUG_VOID_RETURN; |
470 | } |
471 | |
472 | /** |
473 | Mark the diagnostics area as 'DISABLED'. |
474 | |
475 | This is used in rare cases when the COM_ command at hand sends a response |
476 | in a custom format. One example is the query cache, another is |
477 | COM_STMT_PREPARE. |
478 | */ |
479 | |
480 | void |
481 | Diagnostics_area::disable_status() |
482 | { |
483 | DBUG_ENTER("disable_status" ); |
484 | DBUG_ASSERT(! is_set()); |
485 | m_status= DA_DISABLED; |
486 | DBUG_VOID_RETURN; |
487 | } |
488 | |
489 | Warning_info::Warning_info(ulonglong warn_id_arg, |
490 | bool allow_unlimited_warnings, bool initialize) |
491 | :m_current_statement_warn_count(0), |
492 | m_current_row_for_warning(1), |
493 | m_warn_id(warn_id_arg), |
494 | m_error_condition(NULL), |
495 | m_allow_unlimited_warnings(allow_unlimited_warnings), |
496 | initialized(0), |
497 | m_read_only(FALSE) |
498 | { |
499 | m_warn_list.empty(); |
500 | memset(m_warn_count, 0, sizeof(m_warn_count)); |
501 | if (initialize) |
502 | init(); |
503 | } |
504 | |
505 | void Warning_info::init() |
506 | { |
507 | /* Initialize sub structures */ |
508 | DBUG_ASSERT(initialized == 0); |
509 | init_sql_alloc(&m_warn_root, "Warning_info" , WARN_ALLOC_BLOCK_SIZE, |
510 | WARN_ALLOC_PREALLOC_SIZE, MYF(MY_THREAD_SPECIFIC)); |
511 | initialized= 1; |
512 | } |
513 | |
514 | void Warning_info::free_memory() |
515 | { |
516 | if (initialized) |
517 | free_root(&m_warn_root,MYF(0)); |
518 | } |
519 | |
520 | Warning_info::~Warning_info() |
521 | { |
522 | free_memory(); |
523 | } |
524 | |
525 | |
526 | bool Warning_info::has_sql_condition(const char *message_str, size_t message_length) const |
527 | { |
528 | Diagnostics_area::Sql_condition_iterator it(m_warn_list); |
529 | const Sql_condition *err; |
530 | |
531 | while ((err= it++)) |
532 | { |
533 | if (strncmp(message_str, err->get_message_text(), message_length) == 0) |
534 | return true; |
535 | } |
536 | |
537 | return false; |
538 | } |
539 | |
540 | |
541 | void Warning_info::clear(ulonglong new_id) |
542 | { |
543 | id(new_id); |
544 | m_warn_list.empty(); |
545 | m_marked_sql_conditions.empty(); |
546 | free_memory(); |
547 | memset(m_warn_count, 0, sizeof(m_warn_count)); |
548 | m_current_statement_warn_count= 0; |
549 | m_current_row_for_warning= 1; /* Start counting from the first row */ |
550 | clear_error_condition(); |
551 | } |
552 | |
553 | void Warning_info::append_warning_info(THD *thd, const Warning_info *source) |
554 | { |
555 | const Sql_condition *err; |
556 | Diagnostics_area::Sql_condition_iterator it(source->m_warn_list); |
557 | const Sql_condition *src_error_condition = source->get_error_condition(); |
558 | |
559 | while ((err= it++)) |
560 | { |
561 | // Do not use ::push_warning() to avoid invocation of THD-internal-handlers. |
562 | Sql_condition *new_error= Warning_info::push_warning(thd, err); |
563 | |
564 | if (src_error_condition && src_error_condition == err) |
565 | set_error_condition(new_error); |
566 | |
567 | if (source->is_marked_for_removal(err)) |
568 | mark_condition_for_removal(new_error); |
569 | } |
570 | } |
571 | |
572 | |
573 | /** |
574 | Copy Sql_conditions that are not WARN_LEVEL_ERROR from the source |
575 | Warning_info to the current Warning_info. |
576 | |
577 | @param thd Thread context. |
578 | @param sp_wi Stored-program Warning_info |
579 | @param thd Thread context. |
580 | @param src_wi Warning_info to copy from. |
581 | */ |
582 | void Diagnostics_area::copy_non_errors_from_wi(THD *thd, |
583 | const Warning_info *src_wi) |
584 | { |
585 | Sql_condition_iterator it(src_wi->m_warn_list); |
586 | const Sql_condition *cond; |
587 | Warning_info *wi= get_warning_info(); |
588 | |
589 | while ((cond= it++)) |
590 | { |
591 | if (cond->get_level() == Sql_condition::WARN_LEVEL_ERROR) |
592 | continue; |
593 | |
594 | Sql_condition *new_condition= wi->push_warning(thd, cond); |
595 | |
596 | if (src_wi->is_marked_for_removal(cond)) |
597 | wi->mark_condition_for_removal(new_condition); |
598 | } |
599 | } |
600 | |
601 | |
602 | void Warning_info::mark_sql_conditions_for_removal() |
603 | { |
604 | Sql_condition_list::Iterator it(m_warn_list); |
605 | Sql_condition *cond; |
606 | |
607 | while ((cond= it++)) |
608 | mark_condition_for_removal(cond); |
609 | } |
610 | |
611 | |
612 | void Warning_info::remove_marked_sql_conditions() |
613 | { |
614 | List_iterator_fast<Sql_condition> it(m_marked_sql_conditions); |
615 | Sql_condition *cond; |
616 | |
617 | while ((cond= it++)) |
618 | { |
619 | m_warn_list.remove(cond); |
620 | m_warn_count[cond->get_level()]--; |
621 | m_current_statement_warn_count--; |
622 | if (cond == m_error_condition) |
623 | m_error_condition= NULL; |
624 | } |
625 | |
626 | m_marked_sql_conditions.empty(); |
627 | } |
628 | |
629 | |
630 | bool Warning_info::is_marked_for_removal(const Sql_condition *cond) const |
631 | { |
632 | List_iterator_fast<Sql_condition> it( |
633 | const_cast<List<Sql_condition>&> (m_marked_sql_conditions)); |
634 | Sql_condition *c; |
635 | |
636 | while ((c= it++)) |
637 | { |
638 | if (c == cond) |
639 | return true; |
640 | } |
641 | |
642 | return false; |
643 | } |
644 | |
645 | |
646 | void Warning_info::reserve_space(THD *thd, uint count) |
647 | { |
648 | while (m_warn_list.elements() && |
649 | (m_warn_list.elements() + count) > thd->variables.max_error_count) |
650 | m_warn_list.remove(m_warn_list.front()); |
651 | } |
652 | |
653 | Sql_condition *Warning_info::push_warning(THD *thd, |
654 | const Sql_condition_identity *value, |
655 | const char *msg) |
656 | { |
657 | Sql_condition *cond= NULL; |
658 | |
659 | if (! m_read_only) |
660 | { |
661 | if (m_allow_unlimited_warnings || |
662 | m_warn_list.elements() < thd->variables.max_error_count) |
663 | { |
664 | cond= new (& m_warn_root) Sql_condition(& m_warn_root, *value, msg); |
665 | if (cond) |
666 | m_warn_list.push_back(cond); |
667 | } |
668 | m_warn_count[(uint) value->get_level()]++; |
669 | } |
670 | |
671 | m_current_statement_warn_count++; |
672 | return cond; |
673 | } |
674 | |
675 | |
676 | Sql_condition *Warning_info::push_warning(THD *thd, |
677 | const Sql_condition *sql_condition) |
678 | { |
679 | Sql_condition *new_condition= push_warning(thd, sql_condition, |
680 | sql_condition->get_message_text()); |
681 | |
682 | if (new_condition) |
683 | new_condition->copy_opt_attributes(sql_condition); |
684 | |
685 | return new_condition; |
686 | } |
687 | |
688 | /* |
689 | Push the warning to error list if there is still room in the list |
690 | |
691 | SYNOPSIS |
692 | push_warning() |
693 | thd Thread handle |
694 | level Severity of warning (note, warning) |
695 | code Error number |
696 | msg Clear error message |
697 | */ |
698 | |
699 | void push_warning(THD *thd, Sql_condition::enum_warning_level level, |
700 | uint code, const char *msg) |
701 | { |
702 | DBUG_ENTER("push_warning" ); |
703 | DBUG_PRINT("enter" , ("code: %d, msg: %s" , code, msg)); |
704 | |
705 | /* |
706 | Calling push_warning/push_warning_printf with a level of |
707 | WARN_LEVEL_ERROR *is* a bug. Either use my_printf_error(), |
708 | my_error(), or WARN_LEVEL_WARN. |
709 | */ |
710 | DBUG_ASSERT(level != Sql_condition::WARN_LEVEL_ERROR); |
711 | |
712 | if (level == Sql_condition::WARN_LEVEL_ERROR) |
713 | level= Sql_condition::WARN_LEVEL_WARN; |
714 | |
715 | (void) thd->raise_condition(code, NULL, level, msg); |
716 | |
717 | /* Make sure we also count warnings pushed after calling set_ok_status(). */ |
718 | thd->get_stmt_da()->increment_warning(); |
719 | |
720 | DBUG_VOID_RETURN; |
721 | } |
722 | |
723 | |
724 | /* |
725 | Push the warning to error list if there is still room in the list |
726 | |
727 | SYNOPSIS |
728 | push_warning_printf() |
729 | thd Thread handle |
730 | level Severity of warning (note, warning) |
731 | code Error number |
732 | msg Clear error message |
733 | */ |
734 | |
735 | void push_warning_printf(THD *thd, Sql_condition::enum_warning_level level, |
736 | uint code, const char *format, ...) |
737 | { |
738 | va_list args; |
739 | char warning[MYSQL_ERRMSG_SIZE]; |
740 | DBUG_ENTER("push_warning_printf" ); |
741 | DBUG_PRINT("enter" ,("warning: %u" , code)); |
742 | |
743 | DBUG_ASSERT(code != 0); |
744 | DBUG_ASSERT(format != NULL); |
745 | |
746 | va_start(args,format); |
747 | my_vsnprintf_ex(&my_charset_utf8_general_ci, warning, |
748 | sizeof(warning), format, args); |
749 | va_end(args); |
750 | push_warning(thd, level, code, warning); |
751 | DBUG_VOID_RETURN; |
752 | } |
753 | |
754 | |
755 | /* |
756 | Send all notes, errors or warnings to the client in a result set |
757 | |
758 | SYNOPSIS |
759 | mysqld_show_warnings() |
760 | thd Thread handler |
761 | levels_to_show Bitmap for which levels to show |
762 | |
763 | DESCRIPTION |
764 | Takes into account the current LIMIT |
765 | |
766 | RETURN VALUES |
767 | FALSE ok |
768 | TRUE Error sending data to client |
769 | */ |
770 | |
771 | const LEX_CSTRING warning_level_names[]= |
772 | { |
773 | { STRING_WITH_LEN("Note" ) }, |
774 | { STRING_WITH_LEN("Warning" ) }, |
775 | { STRING_WITH_LEN("Error" ) }, |
776 | { STRING_WITH_LEN("?" ) } |
777 | }; |
778 | |
779 | bool mysqld_show_warnings(THD *thd, ulong levels_to_show) |
780 | { |
781 | List<Item> field_list; |
782 | MEM_ROOT *mem_root= thd->mem_root; |
783 | const Sql_condition *err; |
784 | SELECT_LEX *sel= &thd->lex->select_lex; |
785 | SELECT_LEX_UNIT *unit= &thd->lex->unit; |
786 | ulonglong idx= 0; |
787 | Protocol *protocol=thd->protocol; |
788 | DBUG_ENTER("mysqld_show_warnings" ); |
789 | |
790 | DBUG_ASSERT(thd->get_stmt_da()->is_warning_info_read_only()); |
791 | |
792 | field_list.push_back(new (mem_root) |
793 | Item_empty_string(thd, "Level" , 7), |
794 | mem_root); |
795 | field_list.push_back(new (mem_root) |
796 | Item_return_int(thd, "Code" , 4, MYSQL_TYPE_LONG), |
797 | mem_root); |
798 | field_list.push_back(new (mem_root) |
799 | Item_empty_string(thd, "Message" , MYSQL_ERRMSG_SIZE), |
800 | mem_root); |
801 | |
802 | if (protocol->send_result_set_metadata(&field_list, |
803 | Protocol::SEND_NUM_ROWS | |
804 | Protocol::SEND_EOF)) |
805 | DBUG_RETURN(TRUE); |
806 | |
807 | unit->set_limit(sel); |
808 | |
809 | Diagnostics_area::Sql_condition_iterator it= |
810 | thd->get_stmt_da()->sql_conditions(); |
811 | while ((err= it++)) |
812 | { |
813 | /* Skip levels that the user is not interested in */ |
814 | if (!(levels_to_show & ((ulong) 1 << err->get_level()))) |
815 | continue; |
816 | if (++idx <= unit->offset_limit_cnt) |
817 | continue; |
818 | if (idx > unit->select_limit_cnt) |
819 | break; |
820 | protocol->prepare_for_resend(); |
821 | protocol->store(warning_level_names[err->get_level()].str, |
822 | warning_level_names[err->get_level()].length, |
823 | system_charset_info); |
824 | protocol->store((uint32) err->get_sql_errno()); |
825 | protocol->store(err->get_message_text(), |
826 | err->get_message_octet_length(), |
827 | system_charset_info); |
828 | if (protocol->write()) |
829 | DBUG_RETURN(TRUE); |
830 | } |
831 | my_eof(thd); |
832 | |
833 | thd->get_stmt_da()->set_warning_info_read_only(FALSE); |
834 | |
835 | DBUG_RETURN(FALSE); |
836 | } |
837 | |
838 | |
839 | /** |
840 | Convert value for dispatch to error message(see WL#751). |
841 | |
842 | @param to buffer for converted string |
843 | @param to_length size of the buffer |
844 | @param from string which should be converted |
845 | @param from_length string length |
846 | @param from_cs charset from convert |
847 | |
848 | @retval |
849 | result string |
850 | */ |
851 | |
852 | char *err_conv(char *buff, uint to_length, const char *from, |
853 | uint from_length, CHARSET_INFO *from_cs) |
854 | { |
855 | char *to= buff; |
856 | const char *from_start= from; |
857 | size_t res; |
858 | |
859 | DBUG_ASSERT(to_length > 0); |
860 | to_length--; |
861 | if (from_cs == &my_charset_bin) |
862 | { |
863 | uchar char_code; |
864 | res= 0; |
865 | while (1) |
866 | { |
867 | if ((uint)(from - from_start) >= from_length || |
868 | res >= to_length) |
869 | { |
870 | *to= 0; |
871 | break; |
872 | } |
873 | |
874 | char_code= ((uchar) *from); |
875 | if (char_code >= 0x20 && char_code <= 0x7E) |
876 | { |
877 | *to++= char_code; |
878 | from++; |
879 | res++; |
880 | } |
881 | else |
882 | { |
883 | if (res + 4 >= to_length) |
884 | { |
885 | *to= 0; |
886 | break; |
887 | } |
888 | res+= my_snprintf(to, 5, "\\x%02X" , (uint) char_code); |
889 | to+=4; |
890 | from++; |
891 | } |
892 | } |
893 | } |
894 | else |
895 | { |
896 | uint errors; |
897 | res= copy_and_convert(to, to_length, system_charset_info, |
898 | from, from_length, from_cs, &errors); |
899 | to[res]= 0; |
900 | } |
901 | return buff; |
902 | } |
903 | |
904 | |
905 | /** |
906 | Convert string for dispatch to client(see WL#751). |
907 | |
908 | @param to buffer to convert |
909 | @param to_length buffer length |
910 | @param to_cs chraset to convert |
911 | @param from string from convert |
912 | @param from_length string length |
913 | @param from_cs charset from convert |
914 | @param errors count of errors during convertion |
915 | |
916 | @retval |
917 | length of converted string |
918 | */ |
919 | |
920 | size_t convert_error_message(char *to, size_t to_length, CHARSET_INFO *to_cs, |
921 | const char *from, size_t from_length, |
922 | CHARSET_INFO *from_cs, uint *errors) |
923 | { |
924 | int cnvres; |
925 | my_wc_t wc; |
926 | const uchar *from_end= (const uchar*) from+from_length; |
927 | char *to_start= to; |
928 | uchar *to_end; |
929 | my_charset_conv_mb_wc mb_wc= from_cs->cset->mb_wc; |
930 | my_charset_conv_wc_mb wc_mb; |
931 | uint error_count= 0; |
932 | size_t length; |
933 | |
934 | DBUG_ASSERT(to_length > 0); |
935 | /* Make room for the null terminator. */ |
936 | to_length--; |
937 | to_end= (uchar*) (to + to_length); |
938 | |
939 | if (!to_cs || from_cs == to_cs || to_cs == &my_charset_bin) |
940 | { |
941 | length= MY_MIN(to_length, from_length); |
942 | memmove(to, from, length); |
943 | to[length]= 0; |
944 | return length; |
945 | } |
946 | |
947 | wc_mb= to_cs->cset->wc_mb; |
948 | while (1) |
949 | { |
950 | if ((cnvres= (*mb_wc)(from_cs, &wc, (uchar*) from, from_end)) > 0) |
951 | { |
952 | if (!wc) |
953 | break; |
954 | from+= cnvres; |
955 | } |
956 | else if (cnvres == MY_CS_ILSEQ) |
957 | { |
958 | wc= (ulong) (uchar) *from; |
959 | from+=1; |
960 | } |
961 | else |
962 | break; |
963 | |
964 | if ((cnvres= (*wc_mb)(to_cs, wc, (uchar*) to, to_end)) > 0) |
965 | to+= cnvres; |
966 | else if (cnvres == MY_CS_ILUNI) |
967 | { |
968 | length= (wc <= 0xFFFF) ? 6/* '\1234' format*/ : 9 /* '\+123456' format*/; |
969 | if ((uchar*)(to + length) >= to_end) |
970 | break; |
971 | cnvres= (int)my_snprintf(to, 9, |
972 | (wc <= 0xFFFF) ? "\\%04X" : "\\+%06X" , (uint) wc); |
973 | to+= cnvres; |
974 | } |
975 | else |
976 | break; |
977 | } |
978 | |
979 | *to= 0; |
980 | *errors= error_count; |
981 | return (size_t) (to - to_start); |
982 | } |
983 | |
984 | |
985 | /** |
986 | Sanity check for SQLSTATEs. The function does not check if it's really an |
987 | existing SQL-state (there are just too many), it just checks string length and |
988 | looks for bad characters. |
989 | |
990 | @param sqlstate the condition SQLSTATE. |
991 | |
992 | @retval true if it's ok. |
993 | @retval false if it's bad. |
994 | */ |
995 | |
996 | bool is_sqlstate_valid(const LEX_CSTRING *sqlstate) |
997 | { |
998 | if (sqlstate->length != 5) |
999 | return false; |
1000 | |
1001 | for (int i= 0 ; i < 5 ; ++i) |
1002 | { |
1003 | char c = sqlstate->str[i]; |
1004 | |
1005 | if ((c < '0' || '9' < c) && |
1006 | (c < 'A' || 'Z' < c)) |
1007 | return false; |
1008 | } |
1009 | |
1010 | return true; |
1011 | } |
1012 | |