item_func.cc 119 KB
Newer Older
unknown's avatar
unknown committed
1
/* Copyright (C) 2000-2003 MySQL AB
unknown's avatar
unknown committed
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

   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; either version 2 of the License, or
   (at your option) any later version.

   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */


/* This file defines all numerical functions */

20
#ifdef USE_PRAGMA_IMPLEMENTATION
unknown's avatar
unknown committed
21 22 23 24
#pragma implementation				// gcc: Class implementation
#endif

#include "mysql_priv.h"
unknown's avatar
unknown committed
25
#include "slave.h"				// for wait_for_master_pos
unknown's avatar
unknown committed
26 27 28 29 30
#include <m_ctype.h>
#include <hash.h>
#include <time.h>
#include <ft_global.h>

31 32 33
#include "sp_head.h"
#include "sp_rcontext.h"
#include "sp.h"
unknown's avatar
unknown committed
34

unknown's avatar
unknown committed
35 36 37 38 39
#ifdef NO_EMBEDDED_ACCESS_CHECKS
#define sp_restore_security_context(A,B) while (0) {}
#endif


40 41 42 43 44 45 46 47 48 49
bool check_reserved_words(LEX_STRING *name)
{
  if (!my_strcasecmp(system_charset_info, name->str, "GLOBAL") ||
      !my_strcasecmp(system_charset_info, name->str, "LOCAL") ||
      !my_strcasecmp(system_charset_info, name->str, "SESSION"))
    return TRUE;
  return FALSE;
}


unknown's avatar
unknown committed
50 51 52 53 54 55 56 57
/* return TRUE if item is a constant */

bool
eval_const_cond(COND *cond)
{
  return ((Item_func*) cond)->val_int() ? TRUE : FALSE;
}

unknown's avatar
unknown committed
58

59
void Item_func::set_arguments(List<Item> &list)
unknown's avatar
unknown committed
60
{
61
  allowed_arg_cols= 1;
unknown's avatar
unknown committed
62
  arg_count=list.elements;
unknown's avatar
unknown committed
63 64
  args= tmp_arg;                                // If 2 arguments
  if (arg_count <= 2 || (args=(Item**) sql_alloc(sizeof(Item*)*arg_count)))
unknown's avatar
unknown committed
65
  {
unknown's avatar
unknown committed
66
    List_iterator_fast<Item> li(list);
unknown's avatar
unknown committed
67
    Item *item;
unknown's avatar
unknown committed
68
    Item **save_args= args;
unknown's avatar
unknown committed
69 70 71

    while ((item=li++))
    {
unknown's avatar
unknown committed
72
      *(save_args++)= item;
unknown's avatar
unknown committed
73 74 75 76 77 78
      with_sum_func|=item->with_sum_func;
    }
  }
  list.empty();					// Fields are used
}

79 80 81 82 83 84
Item_func::Item_func(List<Item> &list)
  :allowed_arg_cols(1)
{
  set_arguments(list);
}

85
Item_func::Item_func(THD *thd, Item_func *item)
86
  :Item_result_field(thd, item),
87 88 89 90 91
   allowed_arg_cols(item->allowed_arg_cols),
   arg_count(item->arg_count),
   used_tables_cache(item->used_tables_cache),
   not_null_tables_cache(item->not_null_tables_cache),
   const_item_cache(item->const_item_cache)
92 93 94 95 96 97 98 99 100 101
{
  if (arg_count)
  {
    if (arg_count <=2)
      args= tmp_arg;
    else
    {
      if (!(args=(Item**) thd->alloc(sizeof(Item*)*arg_count)))
	return;
    }
102
    memcpy((char*) args, (char*) item->args, sizeof(Item*)*arg_count);
103 104 105
  }
}

unknown's avatar
unknown committed
106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123

/*
  Resolve references to table column for a function and it's argument

  SYNOPSIS:
  fix_fields()
  thd		Thread object
  ref		Pointer to where this object is used.  This reference
		is used if we want to replace this object with another
		one (for example in the summary functions).

  DESCRIPTION
    Call fix_fields() for all arguments to the function.  The main intention
    is to allow all Item_field() objects to setup pointers to the table fields.

    Sets as a side effect the following class variables:
      maybe_null	Set if any argument may return NULL
      with_sum_func	Set if any of the arguments contains a sum function
unknown's avatar
unknown committed
124
      used_tables_cache Set to union of the tables used by arguments
unknown's avatar
unknown committed
125 126 127

      str_value.charset If this is a string function, set this to the
			character set for the first argument.
128
			If any argument is binary, this is set to binary
unknown's avatar
unknown committed
129 130 131 132 133 134 135

   If for any item any of the defaults are wrong, then this can
   be fixed in the fix_length_and_dec() function that is called
   after this one or by writing a specialized fix_fields() for the
   item.

  RETURN VALUES
unknown's avatar
unknown committed
136 137
  FALSE	ok
  TRUE	Got error.  Stored with my_error().
unknown's avatar
unknown committed
138 139
*/

unknown's avatar
unknown committed
140
bool
141
Item_func::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
142
{
143
  DBUG_ASSERT(fixed == 0);
unknown's avatar
unknown committed
144
  Item **arg,**arg_end;
145 146 147
#ifndef EMBEDDED_LIBRARY			// Avoid compiler warning
  char buff[STACK_BUFF_ALLOC];			// Max argument in function
#endif
148

149
  used_tables_cache= not_null_tables_cache= 0;
unknown's avatar
unknown committed
150 151
  const_item_cache=1;

152
  if (check_stack_overrun(thd, STACK_MIN_SIZE, buff))
unknown's avatar
unknown committed
153
    return TRUE;				// Fatal error if flag is set!
unknown's avatar
unknown committed
154 155 156 157
  if (arg_count)
  {						// Print purify happy
    for (arg=args, arg_end=args+arg_count; arg != arg_end ; arg++)
    {
unknown's avatar
unknown committed
158
      Item *item;
unknown's avatar
merge  
unknown committed
159 160 161 162
      /*
	We can't yet set item to *arg as fix_fields may change *arg
	We shouldn't call fix_fields() twice, so check 'fixed' field first
      */
163
      if ((!(*arg)->fixed && (*arg)->fix_fields(thd, arg)))
unknown's avatar
unknown committed
164
	return TRUE;				/* purecov: inspected */
unknown's avatar
unknown committed
165
      item= *arg;
166 167 168 169 170 171 172 173 174 175 176 177 178 179

      if (allowed_arg_cols)
      {
        if (item->check_cols(allowed_arg_cols))
          return 1;
      }
      else
      {
        /*  we have to fetch allowed_arg_cols from first argument */
        DBUG_ASSERT(arg == args); // it is first argument
        allowed_arg_cols= item->cols();
        DBUG_ASSERT(allowed_arg_cols); // Can't be 0 any more
      }

unknown's avatar
unknown committed
180
      if (item->maybe_null)
unknown's avatar
unknown committed
181
	maybe_null=1;
182

unknown's avatar
unknown committed
183
      with_sum_func= with_sum_func || item->with_sum_func;
184 185 186
      used_tables_cache|=     item->used_tables();
      not_null_tables_cache|= item->not_null_tables();
      const_item_cache&=      item->const_item();
unknown's avatar
unknown committed
187 188 189
    }
  }
  fix_length_and_dec();
unknown's avatar
unknown committed
190 191
  if (thd->net.report_error) // An error inside fix_length_and_dec occured
    return TRUE;
192
  fixed= 1;
unknown's avatar
unknown committed
193
  return FALSE;
unknown's avatar
unknown committed
194 195
}

unknown's avatar
unknown committed
196 197 198 199 200 201 202 203 204 205 206 207 208
bool Item_func::walk (Item_processor processor, byte *argument)
{
  if (arg_count)
  {
    Item **arg,**arg_end;
    for (arg= args, arg_end= args+arg_count; arg != arg_end; arg++)
    {
      if ((*arg)->walk(processor, argument))
	return 1;
    }
  }
  return (this->*processor)(argument);
}
209

210 211
void Item_func::traverse_cond(Cond_traverser traverser,
                              void *argument, traverse_order order)
212 213 214 215
{
  if (arg_count)
  {
    Item **arg,**arg_end;
216 217 218

    switch (order) {
    case(PREFIX):
unknown's avatar
unknown committed
219
      (*traverser)(this, argument);
220 221 222 223 224 225 226 227 228 229
      for (arg= args, arg_end= args+arg_count; arg != arg_end; arg++)
      {
	(*arg)->traverse_cond(traverser, argument, order);
      }
      break;
    case (POSTFIX):
      for (arg= args, arg_end= args+arg_count; arg != arg_end; arg++)
      {
	(*arg)->traverse_cond(traverser, argument, order);
      }
unknown's avatar
unknown committed
230
      (*traverser)(this, argument);
231 232 233 234
    }
  }
}

unknown's avatar
unknown committed
235

unknown's avatar
unknown committed
236

unknown's avatar
unknown committed
237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258
/*
  Transform an Item_func object with a transformer callback function
   
  SYNOPSIS
    transform()
    transformer   the transformer callback function to be applied to the nodes
                  of the tree of the object
    argument      parameter to be passed to the transformer
  
  DESCRIPTION
    The function recursively applies the transform method with the
    same transformer to each argument the function.
    If the call of the method for a member item returns a new item
    the old item is substituted for a new one.
    After this the transform method is applied to the root node
    of the Item_func object. 
     
  RETURN VALUES
    Item returned as the result of transformation of the root node 
*/

Item *Item_func::transform(Item_transformer transformer, byte *argument)
259 260 261 262 263 264
{
  if (arg_count)
  {
    Item **arg,**arg_end;
    for (arg= args, arg_end= args+arg_count; arg != arg_end; arg++)
    {
unknown's avatar
unknown committed
265
      Item *new_item= (*arg)->transform(transformer, argument);
266 267
      if (!new_item)
	return 0;
unknown's avatar
unknown committed
268 269
      if (*arg != new_item)
        current_thd->change_item_tree(arg, new_item);
270 271
    }
  }
unknown's avatar
unknown committed
272
  return (this->*transformer)(argument);
273 274 275
}


276 277
/* See comments in Item_cmp_func::split_sum_func() */

278 279
void Item_func::split_sum_func(THD *thd, Item **ref_pointer_array,
                               List<Item> &fields)
unknown's avatar
unknown committed
280
{
281 282
  Item **arg, **arg_end;
  for (arg= args, arg_end= args+arg_count; arg != arg_end ; arg++)
283
    (*arg)->split_sum_func2(thd, ref_pointer_array, fields, arg);
unknown's avatar
unknown committed
284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304
}


void Item_func::update_used_tables()
{
  used_tables_cache=0;
  const_item_cache=1;
  for (uint i=0 ; i < arg_count ; i++)
  {
    args[i]->update_used_tables();
    used_tables_cache|=args[i]->used_tables();
    const_item_cache&=args[i]->const_item();
  }
}


table_map Item_func::used_tables() const
{
  return used_tables_cache;
}

305 306 307 308 309 310 311

table_map Item_func::not_null_tables() const
{
  return not_null_tables_cache;
}


unknown's avatar
unknown committed
312 313 314 315
void Item_func::print(String *str)
{
  str->append(func_name());
  str->append('(');
unknown's avatar
unknown committed
316
  print_args(str, 0);
317 318 319 320
  str->append(')');
}


unknown's avatar
unknown committed
321
void Item_func::print_args(String *str, uint from)
322
{
unknown's avatar
unknown committed
323
  for (uint i=from ; i < arg_count ; i++)
unknown's avatar
unknown committed
324
  {
unknown's avatar
unknown committed
325
    if (i != from)
unknown's avatar
unknown committed
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345
      str->append(',');
    args[i]->print(str);
  }
}


void Item_func::print_op(String *str)
{
  str->append('(');
  for (uint i=0 ; i < arg_count-1 ; i++)
  {
    args[i]->print(str);
    str->append(' ');
    str->append(func_name());
    str->append(' ');
  }
  args[arg_count-1]->print(str);
  str->append(')');
}

unknown's avatar
unknown committed
346

347
bool Item_func::eq(const Item *item, bool binary_cmp) const
unknown's avatar
unknown committed
348 349 350 351 352 353 354 355 356 357 358
{
  /* Assume we don't have rtti */
  if (this == item)
    return 1;
  if (item->type() != FUNC_ITEM)
    return 0;
  Item_func *item_func=(Item_func*) item;
  if (arg_count != item_func->arg_count ||
      func_name() != item_func->func_name())
    return 0;
  for (uint i=0; i < arg_count ; i++)
359
    if (!args[i]->eq(item_func->args[i], binary_cmp))
unknown's avatar
unknown committed
360 361 362 363
      return 0;
  return 1;
}

364

365 366 367 368 369
Field *Item_func::tmp_table_field(TABLE *t_arg)
{
  Field *res;
  LINT_INIT(res);

unknown's avatar
unknown committed
370
  switch (result_type()) {
371 372 373 374 375 376 377 378 379 380 381 382
  case INT_RESULT:
    if (max_length > 11)
      res= new Field_longlong(max_length, maybe_null, name, t_arg,
			      unsigned_flag);
    else
      res= new Field_long(max_length, maybe_null, name, t_arg,
			  unsigned_flag);
    break;
  case REAL_RESULT:
    res= new Field_double(max_length, maybe_null, name, t_arg, decimals);
    break;
  case STRING_RESULT:
383
    res= make_string_field(t_arg);
384
    break;
unknown's avatar
unknown committed
385
  case DECIMAL_RESULT:
unknown's avatar
unknown committed
386 387 388 389
    res= new Field_new_decimal(my_decimal_precision_to_length(decimal_precision(),
                                                              decimals,
                                                              unsigned_flag),
                               maybe_null, name, t_arg, decimals, unsigned_flag);
unknown's avatar
unknown committed
390
    break;
391
  case ROW_RESULT:
unknown's avatar
unknown committed
392
  default:
unknown's avatar
unknown committed
393
    // This case should never be chosen
unknown's avatar
unknown committed
394 395
    DBUG_ASSERT(0);
    break;
396 397 398 399
  }
  return res;
}

unknown's avatar
unknown committed
400 401 402 403 404 405 406
my_decimal *Item_func::val_decimal(my_decimal *decimal_value)
{
  DBUG_ASSERT(fixed);
  int2my_decimal(E_DEC_FATAL_ERROR, val_int(), unsigned_flag, decimal_value);
  return decimal_value;
}

407

unknown's avatar
unknown committed
408 409
String *Item_real_func::val_str(String *str)
{
410
  DBUG_ASSERT(fixed == 1);
411
  double nr= val_real();
unknown's avatar
unknown committed
412 413
  if (null_value)
    return 0; /* purecov: inspected */
unknown's avatar
unknown committed
414
  str->set(nr,decimals, &my_charset_bin);
unknown's avatar
unknown committed
415 416 417 418
  return str;
}


419 420 421 422 423 424 425 426 427 428 429
my_decimal *Item_real_func::val_decimal(my_decimal *decimal_value)
{
  DBUG_ASSERT(fixed);
  double nr= val_real();
  if (null_value)
    return 0; /* purecov: inspected */
  double2my_decimal(E_DEC_FATAL_ERROR, nr, decimal_value);
  return decimal_value;
}


unknown's avatar
unknown committed
430
void Item_func::fix_num_length_and_dec()
unknown's avatar
unknown committed
431
{
unknown's avatar
unknown committed
432 433
  decimals= 0;
  for (uint i=0 ; i < arg_count ; i++)
unknown's avatar
unknown committed
434
  {
unknown's avatar
unknown committed
435
    set_if_bigger(decimals, args[i]->decimals);
unknown's avatar
unknown committed
436
  }
unknown's avatar
unknown committed
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454
  max_length= float_length(decimals);
}


void Item_func_numhybrid::fix_num_length_and_dec()
{}


/*
  Set max_length/decimals of function if function is fixed point and
  result length/precision depends on argument ones

  SYNOPSIS
    Item_func::count_decimal_length()
*/

void Item_func::count_decimal_length()
{
unknown's avatar
unknown committed
455
  int max_int_part= 0;
unknown's avatar
unknown committed
456
  decimals= 0;
unknown's avatar
unknown committed
457
  unsigned_flag= 1;
unknown's avatar
unknown committed
458
  for (uint i=0 ; i < arg_count ; i++)
unknown's avatar
unknown committed
459
  {
unknown's avatar
unknown committed
460
    set_if_bigger(decimals, args[i]->decimals);
unknown's avatar
unknown committed
461 462
    set_if_bigger(max_int_part, args[i]->decimal_int_part());
    set_if_smaller(unsigned_flag, args[i]->unsigned_flag);
unknown's avatar
unknown committed
463
  }
unknown's avatar
unknown committed
464 465 466
  int precision= min(max_int_part + decimals, DECIMAL_MAX_PRECISION);
  max_length= my_decimal_precision_to_length(precision, decimals,
                                             unsigned_flag);
unknown's avatar
unknown committed
467 468 469
}


unknown's avatar
unknown committed
470 471 472 473 474 475 476 477
/*
  Set max_length of if it is maximum length of its arguments

  SYNOPSIS
    Item_func::count_only_length()
*/

void Item_func::count_only_length()
unknown's avatar
unknown committed
478
{
unknown's avatar
unknown committed
479
  max_length= 0;
unknown's avatar
unknown committed
480
  unsigned_flag= 0;
unknown's avatar
unknown committed
481
  for (uint i=0 ; i < arg_count ; i++)
unknown's avatar
unknown committed
482
  {
unknown's avatar
unknown committed
483
    set_if_bigger(max_length, args[i]->max_length);
unknown's avatar
unknown committed
484 485
    set_if_bigger(unsigned_flag, args[i]->unsigned_flag);
  }
unknown's avatar
unknown committed
486 487
}

unknown's avatar
unknown committed
488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523

/*
  Set max_length/decimals of function if function is floating point and
  result length/precision depends on argument ones

  SYNOPSIS
    Item_func::count_real_length()
*/

void Item_func::count_real_length()
{
  uint32 length= 0;
  decimals= 0;
  max_length= 0;
  for (uint i=0 ; i < arg_count ; i++)
  {
    if (decimals != NOT_FIXED_DEC)
    {
      set_if_bigger(decimals, args[i]->decimals);
      set_if_bigger(length, (args[i]->max_length - args[i]->decimals));
    }
    set_if_bigger(max_length, args[i]->max_length);
  }
  if (decimals != NOT_FIXED_DEC)
  {
    max_length= length;
    length+= decimals;
    if (length < max_length)  // If previous operation gave overflow
      max_length= UINT_MAX32;
    else
      max_length= length;
  }
}



unknown's avatar
unknown committed
524 525 526 527 528 529 530 531 532 533
void Item_func::signal_divide_by_null()
{
  THD *thd= current_thd;
  if (thd->variables.sql_mode & MODE_ERROR_FOR_DIVISION_BY_ZERO)
    push_warning(thd, MYSQL_ERROR::WARN_LEVEL_ERROR, ER_DIVISION_BY_ZERO,
                 ER(ER_DIVISION_BY_ZERO));
  null_value= 1;
}


534
Item *Item_func::get_tmp_table_item(THD *thd)
535 536 537
{
  if (!with_sum_func && !const_item())
    return new Item_field(result_field);
538
  return copy_or_same(thd);
539 540
}

unknown's avatar
unknown committed
541 542
String *Item_int_func::val_str(String *str)
{
543
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
544 545 546
  longlong nr=val_int();
  if (null_value)
    return 0;
547
  if (!unsigned_flag)
unknown's avatar
unknown committed
548
    str->set(nr,&my_charset_bin);
549
  else
unknown's avatar
unknown committed
550
    str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
551 552 553
  return str;
}

unknown's avatar
unknown committed
554

555
/*
556 557
  Check arguments here to determine result's type for a numeric
  function of two arguments.
unknown's avatar
unknown committed
558 559 560

  SYNOPSIS
    Item_num_op::find_num_type()
561
*/
unknown's avatar
unknown committed
562 563 564

void Item_num_op::find_num_type(void)
{
unknown's avatar
unknown committed
565 566 567 568 569 570 571 572
  DBUG_ENTER("Item_num_op::find_num_type");
  DBUG_PRINT("info", ("name %s", func_name()));
  DBUG_ASSERT(arg_count == 2);
  Item_result r0= args[0]->result_type();
  Item_result r1= args[1]->result_type();

  if (r0 == REAL_RESULT || r1 == REAL_RESULT ||
      r0 == STRING_RESULT || r1 ==STRING_RESULT)
573
  {
unknown's avatar
unknown committed
574 575 576 577 578 579 580 581 582
    count_real_length();
    max_length= float_length(decimals);
    hybrid_type= REAL_RESULT;
  }
  else if (r0 == DECIMAL_RESULT || r1 == DECIMAL_RESULT)
  {
    hybrid_type= DECIMAL_RESULT;
    result_precision();
  }
583
  else
584
  {
585
    DBUG_ASSERT(r0 == INT_RESULT && r1 == INT_RESULT);
unknown's avatar
unknown committed
586
    decimals= 0;
unknown's avatar
unknown committed
587
    hybrid_type=INT_RESULT;
unknown's avatar
unknown committed
588
    result_precision();
589
  }
unknown's avatar
unknown committed
590 591 592 593 594 595
  DBUG_PRINT("info", ("Type: %s",
             (hybrid_type == REAL_RESULT ? "REAL_RESULT" :
              hybrid_type == DECIMAL_RESULT ? "DECIMAL_RESULT" :
              hybrid_type == INT_RESULT ? "INT_RESULT" :
              "--ILLEGAL!!!--")));
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
596 597
}

unknown's avatar
unknown committed
598 599

/*
600 601 602
  Set result type for a numeric function of one argument
  (can be also used by a numeric function of many arguments, if the result
  type depends only on the first argument)
unknown's avatar
unknown committed
603 604 605 606

  SYNOPSIS
    Item_func_num1::find_num_type()
*/
607

unknown's avatar
unknown committed
608 609 610 611
void Item_func_num1::find_num_type()
{
  DBUG_ENTER("Item_func_num1::find_num_type");
  DBUG_PRINT("info", ("name %s", func_name()));
612
  switch (hybrid_type= args[0]->result_type()) {
unknown's avatar
unknown committed
613
  case INT_RESULT:
614
    unsigned_flag= args[0]->unsigned_flag;
unknown's avatar
unknown committed
615 616 617 618 619 620 621 622 623 624
    break;
  case STRING_RESULT:
  case REAL_RESULT:
    hybrid_type= REAL_RESULT;
    max_length= float_length(decimals);
    break;
  case DECIMAL_RESULT:
    break;
  default:
    DBUG_ASSERT(0);
625
  }
unknown's avatar
unknown committed
626 627 628 629 630 631
  DBUG_PRINT("info", ("Type: %s",
                      (hybrid_type == REAL_RESULT ? "REAL_RESULT" :
                       hybrid_type == DECIMAL_RESULT ? "DECIMAL_RESULT" :
                       hybrid_type == INT_RESULT ? "INT_RESULT" :
                       "--ILLEGAL!!!--")));
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
632 633
}

unknown's avatar
unknown committed
634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649

void Item_func_num1::fix_num_length_and_dec()
{
  decimals= args[0]->decimals;
  max_length= args[0]->max_length;
}


void Item_func_numhybrid::fix_length_and_dec()
{
  fix_num_length_and_dec();
  find_num_type();
}


String *Item_func_numhybrid::val_str(String *str)
unknown's avatar
unknown committed
650
{
651
  DBUG_ASSERT(fixed == 1);
652
  switch (hybrid_type) {
unknown's avatar
unknown committed
653 654 655 656
  case DECIMAL_RESULT:
  {
    my_decimal decimal_value, *val;
    if (!(val= decimal_op(&decimal_value)))
657
      return 0;                                 // null is set
unknown's avatar
unknown committed
658 659 660 661 662 663 664
    my_decimal_round(E_DEC_FATAL_ERROR, val, decimals, FALSE, val);
    my_decimal2string(E_DEC_FATAL_ERROR, val, 0, 0, 0, str);
    break;
  }
  case INT_RESULT:
  {
    longlong nr= int_op();
unknown's avatar
unknown committed
665 666
    if (null_value)
      return 0; /* purecov: inspected */
667
    if (!unsigned_flag)
unknown's avatar
unknown committed
668
      str->set(nr,&my_charset_bin);
669
    else
unknown's avatar
unknown committed
670
      str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
671
    break;
unknown's avatar
unknown committed
672
  }
unknown's avatar
unknown committed
673
  case REAL_RESULT:
unknown's avatar
unknown committed
674
  {
675
    double nr= real_op();
unknown's avatar
unknown committed
676 677
    if (null_value)
      return 0; /* purecov: inspected */
unknown's avatar
unknown committed
678
    str->set(nr,decimals,&my_charset_bin);
unknown's avatar
unknown committed
679 680
    break;
  }
unknown's avatar
unknown committed
681 682
  case STRING_RESULT:
    return str_op(&str_value);
unknown's avatar
unknown committed
683 684
  default:
    DBUG_ASSERT(0);
unknown's avatar
unknown committed
685 686 687 688 689
  }
  return str;
}


unknown's avatar
unknown committed
690 691 692
double Item_func_numhybrid::val_real()
{
  DBUG_ASSERT(fixed == 1);
693
  switch (hybrid_type) {
unknown's avatar
unknown committed
694 695 696 697
  case DECIMAL_RESULT:
  {
    my_decimal decimal_value, *val;
    double result;
698 699
    if (!(val= decimal_op(&decimal_value)))
      return 0.0;                               // null is set
unknown's avatar
unknown committed
700 701 702 703 704 705 706
    my_decimal2double(E_DEC_FATAL_ERROR, val, &result);
    return result;
  }
  case INT_RESULT:
    return (double)int_op();
  case REAL_RESULT:
    return real_op();
unknown's avatar
unknown committed
707 708 709 710 711 712 713 714
  case STRING_RESULT:
  {
    char *end_not_used;
    int err_not_used;
    String *res= str_op(&str_value);
    return (res ? my_strntod(res->charset(), (char*) res->ptr(), res->length(),
			     &end_not_used, &err_not_used) : 0.0);
  }
unknown's avatar
unknown committed
715 716 717 718 719 720 721 722 723 724
  default:
    DBUG_ASSERT(0);
  }
  return 0.0;
}


longlong Item_func_numhybrid::val_int()
{
  DBUG_ASSERT(fixed == 1);
725
  switch (hybrid_type) {
unknown's avatar
unknown committed
726 727 728 729
  case DECIMAL_RESULT:
  {
    my_decimal decimal_value, *val;
    if (!(val= decimal_op(&decimal_value)))
730
      return 0;                                 // null is set
unknown's avatar
unknown committed
731 732 733 734 735 736 737 738
    longlong result;
    my_decimal2int(E_DEC_FATAL_ERROR, val, unsigned_flag, &result);
    return result;
  }
  case INT_RESULT:
    return int_op();
  case REAL_RESULT:
    return (longlong)real_op();
unknown's avatar
unknown committed
739 740 741
  case STRING_RESULT:
  {
    int err_not_used;
unknown's avatar
unknown committed
742 743 744 745
    String *res;
    if (!(res= str_op(&str_value)))
      return 0;

746
    char *end= (char*) res->ptr() + res->length();
unknown's avatar
unknown committed
747
    CHARSET_INFO *cs= str_value.charset();
unknown's avatar
unknown committed
748
    return (*(cs->cset->strtoll10))(cs, res->ptr(), &end, &err_not_used);
unknown's avatar
unknown committed
749
  }
unknown's avatar
unknown committed
750 751 752 753 754 755 756 757 758 759 760
  default:
    DBUG_ASSERT(0);
  }
  return 0;
}


my_decimal *Item_func_numhybrid::val_decimal(my_decimal *decimal_value)
{
  my_decimal *val= decimal_value;
  DBUG_ASSERT(fixed == 1);
761
  switch (hybrid_type) {
unknown's avatar
unknown committed
762 763 764 765 766 767 768 769 770 771 772
  case DECIMAL_RESULT:
    val= decimal_op(decimal_value);
    break;
  case INT_RESULT:
  {
    longlong result= int_op();
    int2my_decimal(E_DEC_FATAL_ERROR, result, unsigned_flag, decimal_value);
    break;
  }
  case REAL_RESULT:
  {
773
    double result= (double)real_op();
unknown's avatar
unknown committed
774 775 776 777
    double2my_decimal(E_DEC_FATAL_ERROR, result, decimal_value);
    break;
  }
  case STRING_RESULT:
unknown's avatar
unknown committed
778
  {
unknown's avatar
unknown committed
779 780 781 782
    String *res;
    if (!(res= str_op(&str_value)))
      return NULL;

unknown's avatar
unknown committed
783 784 785 786
    str2my_decimal(E_DEC_FATAL_ERROR, (char*) res->ptr(),
                   res->length(), res->charset(), decimal_value);
    break;
  }  
unknown's avatar
unknown committed
787 788 789 790 791 792 793 794
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
  }
  return val;
}


795 796
void Item_func_signed::print(String *str)
{
797
  str->append("cast(", 5);
798
  args[0]->print(str);
799
  str->append(" as signed)", 11);
800 801 802 803

}


804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855
longlong Item_func_signed::val_int_from_str(int *error)
{
  char buff[MAX_FIELD_WIDTH], *end;
  String tmp(buff,sizeof(buff), &my_charset_bin), *res;
  longlong value;

  /*
    For a string result, we must first get the string and then convert it
    to a longlong
  */

  if (!(res= args[0]->val_str(&tmp)))
  {
    null_value= 1;
    *error= 0;
    return 0;
  }
  null_value= 0;
  end= (char*) res->ptr()+ res->length();
  value= my_strtoll10(res->ptr(), &end, error);
  if (*error > 0 || end != res->ptr()+ res->length())
    push_warning_printf(current_thd, MYSQL_ERROR::WARN_LEVEL_WARN,
                        ER_TRUNCATED_WRONG_VALUE,
                        ER(ER_TRUNCATED_WRONG_VALUE), "INTEGER",
                        res->c_ptr());
  return value;
}


longlong Item_func_signed::val_int()
{
  longlong value;
  int error;

  if (args[0]->cast_to_int_type() != STRING_RESULT)
  {
    value= args[0]->val_int();
    null_value= args[0]->null_value; 
    return value;
  }

  value= val_int_from_str(&error);
  if (value < 0 && error == 0)
  {
    push_warning(current_thd, MYSQL_ERROR::WARN_LEVEL_WARN, ER_UNKNOWN_ERROR,
                 "Cast to signed converted positive out-of-range integer to "
                 "it's negative complement");
  }
  return value;
}


856 857
void Item_func_unsigned::print(String *str)
{
858
  str->append("cast(", 5);
859
  args[0]->print(str);
860
  str->append(" as unsigned)", 13);
861 862 863 864

}


865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885
longlong Item_func_unsigned::val_int()
{
  longlong value;
  int error;

  if (args[0]->cast_to_int_type() != STRING_RESULT)
  {
    value= args[0]->val_int();
    null_value= args[0]->null_value; 
    return value;
  }

  value= val_int_from_str(&error);
  if (error < 0)
    push_warning(current_thd, MYSQL_ERROR::WARN_LEVEL_WARN, ER_UNKNOWN_ERROR,
                 "Cast to unsigned converted negative integer to it's "
                 "positive complement");
  return value;
}


unknown's avatar
unknown committed
886 887 888
String *Item_decimal_typecast::val_str(String *str)
{
  my_decimal tmp_buf, *tmp= val_decimal(&tmp_buf);
889 890
  if (null_value)
    return NULL;
unknown's avatar
unknown committed
891 892 893 894 895 896 897 898 899
  my_decimal2string(E_DEC_FATAL_ERROR, &tmp_buf, 0, 0, 0, str);
  return str;
}


double Item_decimal_typecast::val_real()
{
  my_decimal tmp_buf, *tmp= val_decimal(&tmp_buf);
  double res;
900 901
  if (null_value)
    return 0.0;
unknown's avatar
unknown committed
902 903 904 905 906 907 908 909 910
  my_decimal2double(E_DEC_FATAL_ERROR, tmp, &res);
  return res;
}


longlong Item_decimal_typecast::val_int()
{
  my_decimal tmp_buf, *tmp= val_decimal(&tmp_buf);
  longlong res;
911 912
  if (null_value)
    return 0;
unknown's avatar
unknown committed
913 914 915 916 917 918 919 920
  my_decimal2int(E_DEC_FATAL_ERROR, tmp, unsigned_flag, &res);
  return res;
}


my_decimal *Item_decimal_typecast::val_decimal(my_decimal *dec)
{
  my_decimal tmp_buf, *tmp= args[0]->val_decimal(&tmp_buf);
921 922
  if ((null_value= args[0]->null_value))
    return NULL;
unknown's avatar
unknown committed
923 924 925 926 927
  my_decimal_round(E_DEC_FATAL_ERROR, tmp, decimals, FALSE, dec);
  return dec;
}


928 929 930 931 932 933 934 935
void Item_decimal_typecast::print(String *str)
{
  str->append("cast(", 5);
  args[0]->print(str);
  str->append(" as decimal)", 12);
}


unknown's avatar
unknown committed
936
double Item_func_plus::real_op()
unknown's avatar
unknown committed
937
{
938
  double value= args[0]->val_real() + args[1]->val_real();
unknown's avatar
unknown committed
939 940 941 942 943
  if ((null_value=args[0]->null_value || args[1]->null_value))
    return 0.0;
  return value;
}

unknown's avatar
unknown committed
944 945

longlong Item_func_plus::int_op()
unknown's avatar
unknown committed
946
{
unknown's avatar
unknown committed
947 948 949 950 951 952 953
  longlong value=args[0]->val_int()+args[1]->val_int();
  if ((null_value=args[0]->null_value || args[1]->null_value))
    return 0;
  return value;
}


954 955 956 957 958 959 960 961 962 963 964 965
/*
  Calculate plus of two decimail's

  SYNOPSIS
    decimal_op()
    decimal_value	Buffer that can be used to store result

  RETURN
   0	Value was NULL;  In this case null_value is set
   #	Value of operation as a decimal
*/

unknown's avatar
unknown committed
966 967
my_decimal *Item_func_plus::decimal_op(my_decimal *decimal_value)
{
968 969 970
  my_decimal value1, *val1;
  my_decimal value2, *val2;
  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
971 972
  if ((null_value= args[0]->null_value))
    return 0;
973
  val2= args[1]->val_decimal(&value2);
974
  if (!(null_value= (args[1]->null_value ||
975 976
                     (my_decimal_add(E_DEC_FATAL_ERROR, decimal_value, val1,
                                     val2) > 3))))
977 978
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
979 980 981 982 983 984 985 986 987 988 989
}

/*
  Set precision of results for additive operations (+ and -)

  SYNOPSIS
    Item_func_additive_op::result_precision()
*/
void Item_func_additive_op::result_precision()
{
  decimals= max(args[0]->decimals, args[1]->decimals);
unknown's avatar
unknown committed
990 991 992 993 994 995 996 997 998 999 1000
  int max_int_part= max(args[0]->decimal_precision() - args[0]->decimals,
                        args[1]->decimal_precision() - args[1]->decimals);
  int precision= min(max_int_part + 1 + decimals, DECIMAL_MAX_PRECISION);

  /* Integer operations keep unsigned_flag if one of arguments is unsigned */
  if (result_type() == INT_RESULT)
    unsigned_flag= args[0]->unsigned_flag | args[1]->unsigned_flag;
  else
    unsigned_flag= args[0]->unsigned_flag & args[1]->unsigned_flag;
  max_length= my_decimal_precision_to_length(precision, decimals,
                                             unsigned_flag);
unknown's avatar
unknown committed
1001 1002
}

1003 1004 1005 1006 1007 1008 1009 1010 1011 1012

/*
  The following function is here to allow the user to force
  subtraction of UNSIGNED BIGINT to return negative values.
*/

void Item_func_minus::fix_length_and_dec()
{
  Item_num_op::fix_length_and_dec();
  if (unsigned_flag &&
1013
      (current_thd->variables.sql_mode & MODE_NO_UNSIGNED_SUBTRACTION))
1014 1015 1016 1017
    unsigned_flag=0;
}


unknown's avatar
unknown committed
1018
double Item_func_minus::real_op()
unknown's avatar
unknown committed
1019
{
1020
  double value= args[0]->val_real() - args[1]->val_real();
unknown's avatar
unknown committed
1021 1022 1023 1024 1025
  if ((null_value=args[0]->null_value || args[1]->null_value))
    return 0.0;
  return value;
}

unknown's avatar
unknown committed
1026 1027

longlong Item_func_minus::int_op()
unknown's avatar
unknown committed
1028
{
unknown's avatar
unknown committed
1029 1030 1031 1032
  longlong value=args[0]->val_int() - args[1]->val_int();
  if ((null_value=args[0]->null_value || args[1]->null_value))
    return 0;
  return value;
unknown's avatar
unknown committed
1033 1034
}

1035

1036 1037
/* See Item_func_plus::decimal_op for comments */

unknown's avatar
unknown committed
1038 1039
my_decimal *Item_func_minus::decimal_op(my_decimal *decimal_value)
{
1040 1041 1042 1043
  my_decimal value1, *val1;
  my_decimal value2, *val2= 

  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1044 1045
  if ((null_value= args[0]->null_value))
    return 0;
1046
  val2= args[1]->val_decimal(&value2);
1047
  if (!(null_value= (args[1]->null_value ||
1048 1049
                     (my_decimal_sub(E_DEC_FATAL_ERROR, decimal_value, val1,
                                     val2) > 3))))
1050 1051
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1052 1053
}

1054

unknown's avatar
unknown committed
1055
double Item_func_mul::real_op()
unknown's avatar
unknown committed
1056
{
1057
  DBUG_ASSERT(fixed == 1);
1058
  double value= args[0]->val_real() * args[1]->val_real();
unknown's avatar
unknown committed
1059
  if ((null_value=args[0]->null_value || args[1]->null_value))
unknown's avatar
unknown committed
1060
    return 0.0;
unknown's avatar
unknown committed
1061 1062 1063
  return value;
}

unknown's avatar
unknown committed
1064 1065

longlong Item_func_mul::int_op()
unknown's avatar
unknown committed
1066
{
1067
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1068 1069 1070 1071
  longlong value=args[0]->val_int()*args[1]->val_int();
  if ((null_value=args[0]->null_value || args[1]->null_value))
    return 0;
  return value;
unknown's avatar
unknown committed
1072 1073 1074
}


1075 1076
/* See Item_func_plus::decimal_op for comments */

unknown's avatar
unknown committed
1077 1078
my_decimal *Item_func_mul::decimal_op(my_decimal *decimal_value)
{
1079 1080 1081
  my_decimal value1, *val1;
  my_decimal value2, *val2;
  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1082 1083
  if ((null_value= args[0]->null_value))
    return 0;
1084
  val2= args[1]->val_decimal(&value2);
1085
  if (!(null_value= (args[1]->null_value ||
1086 1087
                     (my_decimal_mul(E_DEC_FATAL_ERROR, decimal_value, val1,
                                    val2) > 3))))
1088 1089
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1090 1091 1092
}


unknown's avatar
unknown committed
1093 1094
void Item_func_mul::result_precision()
{
unknown's avatar
unknown committed
1095 1096 1097 1098 1099 1100 1101 1102 1103
  /* Integer operations keep unsigned_flag if one of arguments is unsigned */
  if (result_type() == INT_RESULT)
    unsigned_flag= args[0]->unsigned_flag | args[1]->unsigned_flag;
  else
    unsigned_flag= args[0]->unsigned_flag & args[1]->unsigned_flag;
  decimals= min(args[0]->decimals + args[1]->decimals, DECIMAL_MAX_SCALE);
  int precision= min(args[0]->decimal_precision() + args[1]->decimal_precision(),
                     DECIMAL_MAX_PRECISION);
  max_length= my_decimal_precision_to_length(precision, decimals,unsigned_flag);
unknown's avatar
unknown committed
1104 1105 1106 1107
}


double Item_func_div::real_op()
unknown's avatar
unknown committed
1108
{
1109
  DBUG_ASSERT(fixed == 1);
1110 1111
  double value= args[0]->val_real();
  double val2= args[1]->val_real();
unknown's avatar
unknown committed
1112 1113 1114 1115 1116
  if ((null_value= args[0]->null_value || args[1]->null_value))
    return 0.0;
  if (val2 == 0.0)
  {
    signal_divide_by_null();
unknown's avatar
unknown committed
1117
    return 0.0;
unknown's avatar
unknown committed
1118
  }
unknown's avatar
unknown committed
1119 1120 1121
  return value/val2;
}

unknown's avatar
unknown committed
1122

unknown's avatar
unknown committed
1123
my_decimal *Item_func_div::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1124
{
1125 1126
  my_decimal value1, *val1;
  my_decimal value2, *val2;
unknown's avatar
unknown committed
1127
  int err;
1128 1129

  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1130 1131
  if ((null_value= args[0]->null_value))
    return 0;
1132
  val2= args[1]->val_decimal(&value2);
unknown's avatar
unknown committed
1133 1134
  if ((null_value= args[1]->null_value))
    return 0;
unknown's avatar
unknown committed
1135 1136 1137 1138 1139 1140
  if ((err= my_decimal_div(E_DEC_FATAL_ERROR & ~E_DEC_DIV_ZERO, decimal_value,
                           val1, val2, prec_increment)) > 3)
  {
    if (err == E_DEC_DIV_ZERO)
      signal_divide_by_null();
    null_value= 1;
unknown's avatar
unknown committed
1141
    return 0;
1142
  }
unknown's avatar
unknown committed
1143
  return decimal_value;
unknown's avatar
unknown committed
1144 1145 1146 1147 1148
}


void Item_func_div::result_precision()
{
unknown's avatar
unknown committed
1149 1150 1151 1152 1153 1154 1155 1156 1157 1158
  uint precision=min(args[0]->decimal_precision() + prec_increment,
                     DECIMAL_MAX_PRECISION);
  /* Integer operations keep unsigned_flag if one of arguments is unsigned */
  if (result_type() == INT_RESULT)
    unsigned_flag= args[0]->unsigned_flag | args[1]->unsigned_flag;
  else
    unsigned_flag= args[0]->unsigned_flag & args[1]->unsigned_flag;
  decimals= min(args[0]->decimals + prec_increment, DECIMAL_MAX_SCALE);
  max_length= my_decimal_precision_to_length(precision, decimals,
                                             unsigned_flag);
unknown's avatar
unknown committed
1159 1160
}

unknown's avatar
unknown committed
1161

unknown's avatar
unknown committed
1162 1163
void Item_func_div::fix_length_and_dec()
{
unknown's avatar
unknown committed
1164
  DBUG_ENTER("Item_func_div::fix_length_and_dec");
unknown's avatar
unknown committed
1165
  prec_increment= current_thd->variables.div_precincrement;
1166
  Item_num_op::fix_length_and_dec();
1167
  switch(hybrid_type) {
unknown's avatar
unknown committed
1168 1169
  case REAL_RESULT:
  {
unknown's avatar
unknown committed
1170
    decimals=max(args[0]->decimals,args[1]->decimals)+prec_increment;
unknown's avatar
unknown committed
1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189
    set_if_smaller(decimals, NOT_FIXED_DEC);
    max_length=args[0]->max_length - args[0]->decimals + decimals;
    uint tmp=float_length(decimals);
    set_if_smaller(max_length,tmp);
    break;
  }
  case INT_RESULT:
    hybrid_type= DECIMAL_RESULT;
    DBUG_PRINT("info", ("Type changed: DECIMAL_RESULT"));
    result_precision();
    break;
  case DECIMAL_RESULT:
    result_precision();
    break;
  default:
    DBUG_ASSERT(0);
  }
  maybe_null= 1; // devision by zero
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
1190 1191
}

1192 1193 1194 1195

/* Integer division */
longlong Item_func_int_div::val_int()
{
1196
  DBUG_ASSERT(fixed == 1);
1197 1198
  longlong value=args[0]->val_int();
  longlong val2=args[1]->val_int();
1199
  if ((null_value= (args[0]->null_value || args[1]->null_value)))
unknown's avatar
unknown committed
1200 1201 1202 1203
    return 0;
  if (val2 == 0)
  {
    signal_divide_by_null();
1204
    return 0;
unknown's avatar
unknown committed
1205
  }
1206 1207 1208
  return (unsigned_flag ?
	  (ulonglong) value / (ulonglong) val2 :
	  value / val2);
1209 1210 1211 1212 1213 1214 1215
}


void Item_func_int_div::fix_length_and_dec()
{
  max_length=args[0]->max_length - args[0]->decimals;
  maybe_null=1;
unknown's avatar
unknown committed
1216
  unsigned_flag=args[0]->unsigned_flag | args[1]->unsigned_flag;
1217 1218 1219
}


unknown's avatar
unknown committed
1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235
longlong Item_func_mod::int_op()
{
  DBUG_ASSERT(fixed == 1);
  longlong value=  args[0]->val_int();
  longlong val2= args[1]->val_int();
  if ((null_value= args[0]->null_value || args[1]->null_value))
    return 0; /* purecov: inspected */
  if (val2 == 0)
  {
    signal_divide_by_null();
    return 0;
  }
  return value % val2;
}

double Item_func_mod::real_op()
unknown's avatar
unknown committed
1236
{
1237
  DBUG_ASSERT(fixed == 1);
1238 1239
  double value= args[0]->val_real();
  double val2=  args[1]->val_real();
unknown's avatar
unknown committed
1240
  if ((null_value= args[0]->null_value || args[1]->null_value))
unknown's avatar
unknown committed
1241
    return 0.0; /* purecov: inspected */
unknown's avatar
unknown committed
1242 1243 1244 1245 1246
  if (val2 == 0.0)
  {
    signal_divide_by_null();
    return 0.0;
  }
unknown's avatar
unknown committed
1247 1248 1249
  return fmod(value,val2);
}

unknown's avatar
unknown committed
1250 1251

my_decimal *Item_func_mod::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1252
{
1253 1254 1255 1256
  my_decimal value1, *val1;
  my_decimal value2, *val2;

  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1257 1258
  if ((null_value= args[0]->null_value))
    return 0;
1259
  val2= args[1]->val_decimal(&value2);
unknown's avatar
unknown committed
1260 1261 1262
  if ((null_value= args[1]->null_value))
    return 0;
  switch (my_decimal_mod(E_DEC_FATAL_ERROR & ~E_DEC_DIV_ZERO, decimal_value,
1263
                         val1, val2)) {
unknown's avatar
unknown committed
1264 1265 1266 1267
  case E_DEC_TRUNCATED:
  case E_DEC_OK:
    return decimal_value;
  case E_DEC_DIV_ZERO:
unknown's avatar
unknown committed
1268
    signal_divide_by_null();
unknown's avatar
unknown committed
1269
  default:
1270
    null_value= 1;
unknown's avatar
unknown committed
1271 1272
    return 0;
  }
unknown's avatar
unknown committed
1273 1274
}

unknown's avatar
unknown committed
1275 1276

void Item_func_mod::result_precision()
unknown's avatar
unknown committed
1277
{
unknown's avatar
unknown committed
1278 1279
  decimals= max(args[0]->decimals, args[1]->decimals);
  max_length= max(args[0]->max_length, args[1]->max_length);
unknown's avatar
unknown committed
1280 1281 1282
}


unknown's avatar
unknown committed
1283
double Item_func_neg::real_op()
unknown's avatar
unknown committed
1284
{
1285
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1286
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1287 1288 1289
  return -value;
}

unknown's avatar
unknown committed
1290

unknown's avatar
unknown committed
1291
longlong Item_func_neg::int_op()
unknown's avatar
unknown committed
1292
{
unknown's avatar
unknown committed
1293 1294
  longlong value= args[0]->val_int();
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1295 1296 1297
  return -value;
}

unknown's avatar
unknown committed
1298

unknown's avatar
unknown committed
1299
my_decimal *Item_func_neg::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1300
{
unknown's avatar
unknown committed
1301 1302 1303 1304 1305
  my_decimal val, *value= args[0]->val_decimal(&val);
  if (!(null_value= args[0]->null_value))
  {
    my_decimal2decimal(value, decimal_value);
    my_decimal_neg(decimal_value);
1306
    return decimal_value;
unknown's avatar
unknown committed
1307
  }
1308
  return 0;
unknown's avatar
unknown committed
1309 1310
}

1311

unknown's avatar
unknown committed
1312
void Item_func_neg::fix_num_length_and_dec()
unknown's avatar
unknown committed
1313
{
unknown's avatar
unknown committed
1314 1315 1316
  decimals= args[0]->decimals;
  /* 1 add because sign can appear */
  max_length= args[0]->max_length + 1;
unknown's avatar
unknown committed
1317 1318
}

unknown's avatar
unknown committed
1319

1320
void Item_func_neg::fix_length_and_dec()
unknown's avatar
unknown committed
1321
{
1322
  DBUG_ENTER("Item_func_neg::fix_length_and_dec");
unknown's avatar
unknown committed
1323
  Item_func_num1::fix_length_and_dec();
1324 1325 1326 1327

  /*
    If this is in integer context keep the context as integer if possible
    (This is how multiplication and other integer functions works)
1328 1329
    Use val() to get value as arg_type doesn't mean that item is
    Item_int or Item_real due to existence of Item_param.
1330
  */
unknown's avatar
unknown committed
1331 1332
  if (hybrid_type == INT_RESULT &&
      args[0]->type() == INT_ITEM &&
unknown's avatar
unknown committed
1333
      ((ulonglong) args[0]->val_int() >= (ulonglong) LONGLONG_MIN))
unknown's avatar
unknown committed
1334 1335
  {
    /*
1336 1337
      Ensure that result is converted to DECIMAL, as longlong can't hold
      the negated number
unknown's avatar
unknown committed
1338
    */
unknown's avatar
unknown committed
1339 1340
    hybrid_type= DECIMAL_RESULT;
    DBUG_PRINT("info", ("Type changed: DECIMAL_RESULT"));
unknown's avatar
unknown committed
1341
  }
unknown's avatar
unknown committed
1342
  unsigned_flag= 0;
unknown's avatar
unknown committed
1343
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
1344 1345
}

unknown's avatar
unknown committed
1346

unknown's avatar
unknown committed
1347
double Item_func_abs::real_op()
unknown's avatar
unknown committed
1348
{
1349
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1350
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1351 1352 1353
  return fabs(value);
}

unknown's avatar
unknown committed
1354

unknown's avatar
unknown committed
1355
longlong Item_func_abs::int_op()
unknown's avatar
unknown committed
1356
{
unknown's avatar
unknown committed
1357 1358
  longlong value= args[0]->val_int();
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1359 1360 1361
  return value >= 0 ? value : -value;
}

unknown's avatar
unknown committed
1362

unknown's avatar
unknown committed
1363
my_decimal *Item_func_abs::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1364
{
unknown's avatar
unknown committed
1365 1366
  my_decimal val, *value= args[0]->val_decimal(&val);
  if (!(null_value= args[0]->null_value))
unknown's avatar
unknown committed
1367
  {
unknown's avatar
unknown committed
1368 1369 1370
    my_decimal2decimal(value, decimal_value);
    if (decimal_value->sign())
      my_decimal_neg(decimal_value);
1371
    return decimal_value;
unknown's avatar
unknown committed
1372
  }
1373
  return 0;
unknown's avatar
unknown committed
1374 1375 1376 1377 1378 1379
}


void Item_func_abs::fix_length_and_dec()
{
  Item_func_num1::fix_length_and_dec();
unknown's avatar
unknown committed
1380
  maybe_null= 1;
unknown's avatar
unknown committed
1381 1382
}

unknown's avatar
unknown committed
1383

1384
/* Gateway to natural LOG function */
1385
double Item_func_ln::val_real()
1386
{
1387
  DBUG_ASSERT(fixed == 1);
1388
  double value= args[0]->val_real();
1389
  if ((null_value= args[0]->null_value))
1390
    return 0.0;
1391
  if (value <= 0.0)
1392 1393
  {
    signal_divide_by_null();
1394
    return 0.0;
1395
  }
1396 1397 1398 1399 1400 1401 1402 1403
  return log(value);
}

/* 
 Extended but so slower LOG function
 We have to check if all values are > zero and first one is not one
 as these are the cases then result is not a number.
*/ 
1404
double Item_func_log::val_real()
unknown's avatar
unknown committed
1405
{
1406
  DBUG_ASSERT(fixed == 1);
1407
  double value= args[0]->val_real();
1408
  if ((null_value= args[0]->null_value))
1409
    return 0.0;
1410
  if (value <= 0.0)
1411 1412 1413 1414
  {
    signal_divide_by_null();
    return 0.0;
  }
1415 1416
  if (arg_count == 2)
  {
1417
    double value2= args[1]->val_real();
1418
    if ((null_value= args[1]->null_value))
1419
      return 0.0;
1420
    if (value2 <= 0.0 || value == 1.0)
1421 1422 1423 1424
    {
      signal_divide_by_null();
      return 0.0;
    }
1425 1426
    return log(value2) / log(value);
  }
unknown's avatar
unknown committed
1427 1428 1429
  return log(value);
}

1430
double Item_func_log2::val_real()
1431
{
1432
  DBUG_ASSERT(fixed == 1);
1433
  double value= args[0]->val_real();
1434 1435

  if ((null_value=args[0]->null_value))
1436
    return 0.0;
1437
  if (value <= 0.0)
1438 1439 1440 1441
  {
    signal_divide_by_null();
    return 0.0;
  }
1442
  return log(value) / M_LN2;
1443 1444
}

1445
double Item_func_log10::val_real()
unknown's avatar
unknown committed
1446
{
1447
  DBUG_ASSERT(fixed == 1);
1448
  double value= args[0]->val_real();
1449
  if ((null_value= args[0]->null_value))
1450
    return 0.0;
1451
  if (value <= 0.0)
1452 1453 1454 1455
  {
    signal_divide_by_null();
    return 0.0;
  }
unknown's avatar
unknown committed
1456 1457 1458
  return log10(value);
}

1459
double Item_func_exp::val_real()
unknown's avatar
unknown committed
1460
{
1461
  DBUG_ASSERT(fixed == 1);
1462
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1463 1464 1465 1466 1467
  if ((null_value=args[0]->null_value))
    return 0.0; /* purecov: inspected */
  return exp(value);
}

1468
double Item_func_sqrt::val_real()
unknown's avatar
unknown committed
1469
{
1470
  DBUG_ASSERT(fixed == 1);
1471
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1472 1473 1474 1475 1476
  if ((null_value=(args[0]->null_value || value < 0)))
    return 0.0; /* purecov: inspected */
  return sqrt(value);
}

1477
double Item_func_pow::val_real()
unknown's avatar
unknown committed
1478
{
1479
  DBUG_ASSERT(fixed == 1);
1480 1481
  double value= args[0]->val_real();
  double val2= args[1]->val_real();
unknown's avatar
unknown committed
1482 1483 1484 1485 1486 1487 1488
  if ((null_value=(args[0]->null_value || args[1]->null_value)))
    return 0.0; /* purecov: inspected */
  return pow(value,val2);
}

// Trigonometric functions

1489
double Item_func_acos::val_real()
unknown's avatar
unknown committed
1490
{
1491
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1492
  // the volatile's for BUG #2338 to calm optimizer down (because of gcc's bug)
1493
  volatile double value= args[0]->val_real();
unknown's avatar
unknown committed
1494 1495 1496 1497 1498
  if ((null_value=(args[0]->null_value || (value < -1.0 || value > 1.0))))
    return 0.0;
  return fix_result(acos(value));
}

1499
double Item_func_asin::val_real()
unknown's avatar
unknown committed
1500
{
1501
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1502
  // the volatile's for BUG #2338 to calm optimizer down (because of gcc's bug)
1503
  volatile double value= args[0]->val_real();
unknown's avatar
unknown committed
1504 1505 1506 1507 1508
  if ((null_value=(args[0]->null_value || (value < -1.0 || value > 1.0))))
    return 0.0;
  return fix_result(asin(value));
}

1509
double Item_func_atan::val_real()
unknown's avatar
unknown committed
1510
{
1511
  DBUG_ASSERT(fixed == 1);
1512
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1513 1514 1515 1516
  if ((null_value=args[0]->null_value))
    return 0.0;
  if (arg_count == 2)
  {
1517
    double val2= args[1]->val_real();
unknown's avatar
unknown committed
1518 1519 1520 1521 1522 1523 1524
    if ((null_value=args[1]->null_value))
      return 0.0;
    return fix_result(atan2(value,val2));
  }
  return fix_result(atan(value));
}

1525
double Item_func_cos::val_real()
unknown's avatar
unknown committed
1526
{
1527
  DBUG_ASSERT(fixed == 1);
1528
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1529 1530 1531 1532 1533
  if ((null_value=args[0]->null_value))
    return 0.0;
  return fix_result(cos(value));
}

1534
double Item_func_sin::val_real()
unknown's avatar
unknown committed
1535
{
1536
  DBUG_ASSERT(fixed == 1);
1537
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1538 1539 1540 1541 1542
  if ((null_value=args[0]->null_value))
    return 0.0;
  return fix_result(sin(value));
}

1543
double Item_func_tan::val_real()
unknown's avatar
unknown committed
1544
{
1545
  DBUG_ASSERT(fixed == 1);
1546
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
  if ((null_value=args[0]->null_value))
    return 0.0;
  return fix_result(tan(value));
}


// Shift-functions, same as << and >> in C/C++


longlong Item_func_shift_left::val_int()
{
1558
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572
  uint shift;
  ulonglong res= ((ulonglong) args[0]->val_int() <<
		  (shift=(uint) args[1]->val_int()));
  if (args[0]->null_value || args[1]->null_value)
  {
    null_value=1;
    return 0;
  }
  null_value=0;
  return (shift < sizeof(longlong)*8 ? (longlong) res : LL(0));
}

longlong Item_func_shift_right::val_int()
{
1573
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
  uint shift;
  ulonglong res= (ulonglong) args[0]->val_int() >>
    (shift=(uint) args[1]->val_int());
  if (args[0]->null_value || args[1]->null_value)
  {
    null_value=1;
    return 0;
  }
  null_value=0;
  return (shift < sizeof(longlong)*8 ? (longlong) res : LL(0));
}


longlong Item_func_bit_neg::val_int()
{
1589
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606
  ulonglong res= (ulonglong) args[0]->val_int();
  if ((null_value=args[0]->null_value))
    return 0;
  return ~res;
}


// Conversion functions

void Item_func_integer::fix_length_and_dec()
{
  max_length=args[0]->max_length - args[0]->decimals+1;
  uint tmp=float_length(decimals);
  set_if_smaller(max_length,tmp);
  decimals=0;
}

unknown's avatar
unknown committed
1607
void Item_func_int_val::fix_num_length_and_dec()
unknown's avatar
unknown committed
1608
{
unknown's avatar
unknown committed
1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660
  max_length= args[0]->max_length - (args[0]->decimals ?
                                     args[0]->decimals + 1 :
                                     0) + 2;
  uint tmp= float_length(decimals);
  set_if_smaller(max_length,tmp);
  decimals= 0;
}


void Item_func_int_val::find_num_type()
{
  DBUG_ENTER("Item_func_int_val::find_num_type");
  DBUG_PRINT("info", ("name %s", func_name()));
  switch(hybrid_type= args[0]->result_type())
  {
  case STRING_RESULT:
  case REAL_RESULT:
    hybrid_type= REAL_RESULT;
    max_length= float_length(decimals);
    break;
  case INT_RESULT:
  case DECIMAL_RESULT:
    /*
      -2 because in most high position can't be used any digit for longlong
      and one position for increasing value during operation
    */
    if ((args[0]->max_length - args[0]->decimals) >=
        (DECIMAL_LONGLONG_DIGITS - 2))
    {
      hybrid_type= DECIMAL_RESULT;
    }
    else
    {
      unsigned_flag= args[0]->unsigned_flag;
      hybrid_type= INT_RESULT;
    }
    break;
  default:
    DBUG_ASSERT(0);
  }
  DBUG_PRINT("info", ("Type: %s",
                      (hybrid_type == REAL_RESULT ? "REAL_RESULT" :
                       hybrid_type == DECIMAL_RESULT ? "DECIMAL_RESULT" :
                       hybrid_type == INT_RESULT ? "INT_RESULT" :
                       "--ILLEGAL!!!--")));

  DBUG_VOID_RETURN;
}


longlong Item_func_ceiling::int_op()
{
1661 1662 1663 1664 1665 1666 1667 1668 1669
  longlong result;
  switch (args[0]->result_type()) {
  case INT_RESULT:
    result= args[0]->val_int();
    null_value= args[0]->null_value;
    break;
  case DECIMAL_RESULT:
  {
    my_decimal dec_buf, *dec;
1670
    if ((dec= Item_func_ceiling::decimal_op(&dec_buf)))
1671 1672 1673 1674 1675 1676
      my_decimal2int(E_DEC_FATAL_ERROR, dec, unsigned_flag, &result);
    else
      result= 0;
    break;
  }
  default:
1677
    result= (longlong)Item_func_ceiling::real_op();
1678 1679
  };
  return result;
unknown's avatar
unknown committed
1680 1681
}

unknown's avatar
unknown committed
1682 1683

double Item_func_ceiling::real_op()
unknown's avatar
unknown committed
1684
{
unknown's avatar
unknown committed
1685 1686 1687 1688
  /*
    the volatile's for BUG #3051 to calm optimizer down (because of gcc's
    bug)
  */
1689
  volatile double value= args[0]->val_real();
unknown's avatar
unknown committed
1690 1691 1692 1693 1694 1695
  null_value= args[0]->null_value;
  return ceil(value);
}


my_decimal *Item_func_ceiling::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1696
{
unknown's avatar
unknown committed
1697
  my_decimal val, *value= args[0]->val_decimal(&val);
1698 1699 1700 1701 1702
  if (!(null_value= (args[0]->null_value ||
                     my_decimal_ceiling(E_DEC_FATAL_ERROR, value,
                                        decimal_value) > 1)))
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1703 1704 1705 1706 1707
}


longlong Item_func_floor::int_op()
{
1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726
  longlong result;
  switch (args[0]->result_type()) {
  case INT_RESULT:
    result= args[0]->val_int();
    null_value= args[0]->null_value;
    break;
  case DECIMAL_RESULT:
  {
    my_decimal dec_buf, *dec;
    if ((dec= Item_func_floor::decimal_op(&dec_buf)))
      my_decimal2int(E_DEC_FATAL_ERROR, dec, unsigned_flag, &result);
    else
      result= 0;
    break;
  }
  default:
    result= (longlong)Item_func_floor::real_op();
  };
  return result;
unknown's avatar
unknown committed
1727 1728
}

unknown's avatar
unknown committed
1729 1730

double Item_func_floor::real_op()
unknown's avatar
unknown committed
1731
{
unknown's avatar
unknown committed
1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744
  /*
    the volatile's for BUG #3051 to calm optimizer down (because of gcc's
    bug)
  */
  volatile double value= args[0]->val_real();
  null_value= args[0]->null_value;
  return floor(value);
}


my_decimal *Item_func_floor::decimal_op(my_decimal *decimal_value)
{
  my_decimal val, *value= args[0]->val_decimal(&val);
1745 1746 1747 1748 1749
  if (!(null_value= (args[0]->null_value ||
                     my_decimal_floor(E_DEC_FATAL_ERROR, value,
                                      decimal_value) > 1)))
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1750 1751 1752
}


unknown's avatar
unknown committed
1753
void Item_func_round::fix_length_and_dec()
unknown's avatar
unknown committed
1754
{
unknown's avatar
unknown committed
1755 1756
  unsigned_flag= args[0]->unsigned_flag;
  if (!args[1]->const_item())
unknown's avatar
unknown committed
1757
  {
unknown's avatar
unknown committed
1758 1759 1760 1761 1762 1763
    max_length= args[0]->max_length;
    decimals= args[0]->decimals;
    hybrid_type= REAL_RESULT;
    return;
  }
  
1764
  int decimals_to_set= max((int)args[1]->val_int(), 0);
unknown's avatar
unknown committed
1765 1766 1767 1768 1769 1770 1771 1772
  if (args[0]->decimals == NOT_FIXED_DEC)
  {
    max_length= args[0]->max_length;
    decimals= min(decimals_to_set, NOT_FIXED_DEC);
    hybrid_type= REAL_RESULT;
    return;
  }
  
unknown's avatar
unknown committed
1773
  switch (args[0]->result_type()) {
unknown's avatar
unknown committed
1774 1775 1776 1777 1778 1779 1780
  case REAL_RESULT:
  case STRING_RESULT:
    hybrid_type= REAL_RESULT;
    decimals= min(decimals_to_set, NOT_FIXED_DEC);
    max_length= float_length(decimals);
    break;
  case INT_RESULT:
unknown's avatar
unknown committed
1781
    if (!decimals_to_set &&
1782
        (truncate || (args[0]->decimal_precision() < DECIMAL_LONGLONG_DIGITS)))
unknown's avatar
unknown committed
1783
    {
unknown's avatar
unknown committed
1784 1785
      int length_can_increase= test(!truncate && (args[1]->val_int() < 0));
      max_length= args[0]->max_length + length_can_increase;
unknown's avatar
unknown committed
1786 1787 1788 1789 1790
      /* Here we can keep INT_RESULT */
      hybrid_type= INT_RESULT;
      decimals= 0;
      break;
    }
unknown's avatar
unknown committed
1791
    /* fall through */
unknown's avatar
unknown committed
1792 1793 1794 1795 1796
  case DECIMAL_RESULT:
  {
    hybrid_type= DECIMAL_RESULT;
    int decimals_delta= args[0]->decimals - decimals_to_set;
    int precision= args[0]->decimal_precision();
1797 1798 1799 1800
    int length_increase= ((decimals_delta <= 0) || truncate) ? 0:1;

    precision-= decimals_delta - length_increase;
    decimals= decimals_to_set;
unknown's avatar
unknown committed
1801
    max_length= my_decimal_precision_to_length(precision, decimals,
1802
                                               unsigned_flag);
unknown's avatar
unknown committed
1803 1804 1805 1806
    break;
  }
  default:
    DBUG_ASSERT(0); /* This result type isn't handled */
unknown's avatar
unknown committed
1807 1808 1809
  }
}

1810
double my_double_round(double value, int dec, bool truncate)
unknown's avatar
unknown committed
1811
{
1812
  double tmp;
1813
  uint abs_dec= abs(dec);
1814 1815 1816 1817 1818
  /*
    tmp2 is here to avoid return the value with 80 bit precision
    This will fix that the test round(0.1,1) = round(0.1,1) is true
  */
  volatile double tmp2;
unknown's avatar
unknown committed
1819

1820 1821
  tmp=(abs_dec < array_elements(log_10) ?
       log_10[abs_dec] : pow(10.0,(double) abs_dec));
unknown's avatar
unknown committed
1822 1823

  if (truncate)
unknown's avatar
unknown committed
1824 1825
  {
    if (value >= 0)
1826
      tmp2= dec < 0 ? floor(value/tmp)*tmp : floor(value*tmp)/tmp;
unknown's avatar
unknown committed
1827
    else
1828
      tmp2= dec < 0 ? ceil(value/tmp)*tmp : ceil(value*tmp)/tmp;
unknown's avatar
unknown committed
1829
  }
1830 1831 1832
  else
    tmp2=dec < 0 ? rint(value/tmp)*tmp : rint(value*tmp)/tmp;
  return tmp2;
unknown's avatar
unknown committed
1833 1834 1835
}


1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847
double Item_func_round::real_op()
{
  double value= args[0]->val_real();
  int dec= (int) args[1]->val_int();

  if (!(null_value= args[0]->null_value || args[1]->null_value))
    return my_double_round(value, dec, truncate);

  return 0.0;
}


unknown's avatar
unknown committed
1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872
longlong Item_func_round::int_op()
{
  longlong value= args[0]->val_int();
  int dec=(int) args[1]->val_int();
  decimals= 0;
  uint abs_dec;
  if ((null_value= args[0]->null_value || args[1]->null_value))
    return 0;
  if (dec >= 0)
    return value; // integer have not digits after point

  abs_dec= -dec;
  double tmp;
  /*
    tmp2 is here to avoid return the value with 80 bit precision
    This will fix that the test round(0.1,1) = round(0.1,1) is true
  */
  volatile double tmp2;

  tmp= (abs_dec < array_elements(log_10) ?
        log_10[abs_dec] : pow(10.0, (double) abs_dec));

  if (truncate)
  {
    if (unsigned_flag)
1873
      tmp2= floor(ulonglong2double(value)/tmp)*tmp;
unknown's avatar
unknown committed
1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889
    else if (value >= 0)
      tmp2= floor(((double)value)/tmp)*tmp;
    else
      tmp2= ceil(((double)value)/tmp)*tmp;
  }
  else
    tmp2= rint(((double)value)/tmp)*tmp;
  return (longlong)tmp2;
}


my_decimal *Item_func_round::decimal_op(my_decimal *decimal_value)
{
  my_decimal val, *value= args[0]->val_decimal(&val);
  int dec=(int) args[1]->val_int();
  if (dec > 0)
unknown's avatar
unknown committed
1890 1891 1892
  {
    decimals= min(dec, DECIMAL_MAX_SCALE); // to get correct output
  }
1893 1894 1895 1896 1897
  if (!(null_value= (args[0]->null_value || args[1]->null_value ||
                     my_decimal_round(E_DEC_FATAL_ERROR, value, dec, truncate,
                                      decimal_value) > 1)))
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1898 1899 1900
}


1901
bool Item_func_rand::fix_fields(THD *thd,Item **ref)
unknown's avatar
unknown committed
1902
{
1903
  if (Item_real_func::fix_fields(thd, ref))
1904
    return TRUE;
1905
  used_tables_cache|= RAND_TABLE_BIT;
unknown's avatar
unknown committed
1906 1907
  if (arg_count)
  {					// Only use argument once in query
1908 1909 1910 1911 1912
    if (!args[0]->const_during_execution())
    {
      my_error(ER_WRONG_ARGUMENTS, MYF(0), "RAND");
      return TRUE;
    }
1913
    /*
unknown's avatar
unknown committed
1914
      Allocate rand structure once: we must use thd->stmt_arena
1915 1916
      to create rand in proper mem_root if it's a prepared statement or
      stored procedure.
1917 1918 1919

      No need to send a Rand log event if seed was given eg: RAND(seed),
      as it will be replicated in the query as such.
1920 1921
    */
    if (!rand && !(rand= (struct rand_struct*)
unknown's avatar
unknown committed
1922
                   thd->stmt_arena->alloc(sizeof(*rand))))
1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937
      return TRUE;
    /*
      PARAM_ITEM is returned if we're in statement prepare and consequently
      no placeholder value is set yet.
    */
    if (args[0]->type() != PARAM_ITEM)
    {
      /*
        TODO: do not do reinit 'rand' for every execute of PS/SP if
        args[0] is a constant.
      */
      uint32 tmp= (uint32) args[0]->val_int();
      randominit(rand, (uint32) (tmp*0x10001L+55555555L),
                 (uint32) (tmp*0x10000001L));
    }
unknown's avatar
unknown committed
1938
  }
1939
  else
unknown's avatar
unknown committed
1940
  {
1941 1942 1943 1944 1945
    /*
      Save the seed only the first time RAND() is used in the query
      Once events are forwarded rather than recreated,
      the following can be skipped if inside the slave thread
    */
1946 1947 1948 1949 1950 1951
    if (!thd->rand_used)
    {
      thd->rand_used= 1;
      thd->rand_saved_seed1= thd->rand.seed1;
      thd->rand_saved_seed2= thd->rand.seed2;
    }
1952
    rand= &thd->rand;
unknown's avatar
unknown committed
1953
  }
1954
  return FALSE;
1955 1956
}

unknown's avatar
unknown committed
1957 1958 1959 1960 1961 1962
void Item_func_rand::update_used_tables()
{
  Item_real_func::update_used_tables();
  used_tables_cache|= RAND_TABLE_BIT;
}

1963

1964
double Item_func_rand::val_real()
1965
{
1966
  DBUG_ASSERT(fixed == 1);
1967
  return my_rnd(rand);
unknown's avatar
unknown committed
1968 1969 1970 1971
}

longlong Item_func_sign::val_int()
{
1972
  DBUG_ASSERT(fixed == 1);
1973
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1974 1975 1976 1977 1978
  null_value=args[0]->null_value;
  return value < 0.0 ? -1 : (value > 0 ? 1 : 0);
}


1979
double Item_func_units::val_real()
unknown's avatar
unknown committed
1980
{
1981
  DBUG_ASSERT(fixed == 1);
1982
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1983 1984 1985 1986 1987 1988 1989 1990
  if ((null_value=args[0]->null_value))
    return 0;
  return value*mul+add;
}


void Item_func_min_max::fix_length_and_dec()
{
unknown's avatar
unknown committed
1991
  int max_int_part=0;
unknown's avatar
unknown committed
1992 1993
  decimals=0;
  max_length=0;
unknown's avatar
unknown committed
1994
  maybe_null=0;
unknown's avatar
unknown committed
1995
  cmp_type=args[0]->result_type();
1996

unknown's avatar
unknown committed
1997 1998
  for (uint i=0 ; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
1999 2000
    set_if_bigger(max_length, args[i]->max_length);
    set_if_bigger(decimals, args[i]->decimals);
unknown's avatar
unknown committed
2001
    set_if_bigger(max_int_part, args[i]->decimal_int_part());
unknown's avatar
unknown committed
2002 2003
    if (args[i]->maybe_null)
      maybe_null=1;
unknown's avatar
unknown committed
2004 2005
    cmp_type=item_cmp_type(cmp_type,args[i]->result_type());
  }
unknown's avatar
unknown committed
2006
  if (cmp_type == STRING_RESULT)
unknown's avatar
unknown committed
2007
    agg_arg_charsets(collation, args, arg_count, MY_COLL_CMP_CONV);
unknown's avatar
unknown committed
2008 2009 2010
  else if ((cmp_type == DECIMAL_RESULT) || (cmp_type == INT_RESULT))
    max_length= my_decimal_precision_to_length(max_int_part+decimals, decimals,
                                            unsigned_flag);
unknown's avatar
unknown committed
2011 2012 2013 2014 2015
}


String *Item_func_min_max::val_str(String *str)
{
2016
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2017 2018 2019 2020 2021 2022
  switch (cmp_type) {
  case INT_RESULT:
  {
    longlong nr=val_int();
    if (null_value)
      return 0;
2023
    if (!unsigned_flag)
unknown's avatar
unknown committed
2024
      str->set(nr,&my_charset_bin);
2025
    else
unknown's avatar
unknown committed
2026
      str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
2027 2028
    return str;
  }
unknown's avatar
unknown committed
2029 2030 2031 2032 2033 2034 2035 2036
  case DECIMAL_RESULT:
  {
    my_decimal dec_buf, *dec_val= val_decimal(&dec_buf);
    if (null_value)
      return 0;
    my_decimal2string(E_DEC_FATAL_ERROR, dec_val, 0, 0, 0, str);
    return str;
  }
unknown's avatar
unknown committed
2037 2038
  case REAL_RESULT:
  {
2039
    double nr= val_real();
unknown's avatar
unknown committed
2040 2041
    if (null_value)
      return 0; /* purecov: inspected */
unknown's avatar
unknown committed
2042
    str->set(nr,decimals,&my_charset_bin);
unknown's avatar
unknown committed
2043 2044 2045 2046 2047 2048 2049 2050
    return str;
  }
  case STRING_RESULT:
  {
    String *res;
    LINT_INIT(res);
    for (uint i=0; i < arg_count ; i++)
    {
unknown's avatar
unknown committed
2051
      if (i == 0)
unknown's avatar
unknown committed
2052 2053 2054 2055 2056 2057 2058
	res=args[i]->val_str(str);
      else
      {
	String *res2;
	res2= args[i]->val_str(res == str ? &tmp_value : str);
	if (res2)
	{
2059
	  int cmp= sortcmp(res,res2,collation.collation);
unknown's avatar
unknown committed
2060 2061 2062 2063
	  if ((cmp_sign < 0 ? cmp : -cmp) < 0)
	    res=res2;
	}
      }
unknown's avatar
unknown committed
2064
      if ((null_value= args[i]->null_value))
2065
        return 0;
unknown's avatar
unknown committed
2066
    }
2067
    res->set_charset(collation.collation);
unknown's avatar
unknown committed
2068 2069
    return res;
  }
2070
  case ROW_RESULT:
unknown's avatar
unknown committed
2071
  default:
unknown's avatar
unknown committed
2072
    // This case should never be chosen
unknown's avatar
unknown committed
2073 2074
    DBUG_ASSERT(0);
    return 0;
unknown's avatar
unknown committed
2075 2076 2077 2078 2079
  }
  return 0;					// Keep compiler happy
}


2080
double Item_func_min_max::val_real()
unknown's avatar
unknown committed
2081
{
2082
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2083 2084 2085
  double value=0.0;
  for (uint i=0; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
2086
    if (i == 0)
2087
      value= args[i]->val_real();
unknown's avatar
unknown committed
2088 2089
    else
    {
2090
      double tmp= args[i]->val_real();
unknown's avatar
unknown committed
2091 2092 2093
      if (!args[i]->null_value && (tmp < value ? cmp_sign : -cmp_sign) > 0)
	value=tmp;
    }
unknown's avatar
unknown committed
2094 2095
    if ((null_value= args[i]->null_value))
      break;
unknown's avatar
unknown committed
2096 2097 2098 2099 2100 2101 2102
  }
  return value;
}


longlong Item_func_min_max::val_int()
{
2103
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2104 2105 2106
  longlong value=0;
  for (uint i=0; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
2107
    if (i == 0)
unknown's avatar
unknown committed
2108 2109 2110 2111 2112 2113 2114
      value=args[i]->val_int();
    else
    {
      longlong tmp=args[i]->val_int();
      if (!args[i]->null_value && (tmp < value ? cmp_sign : -cmp_sign) > 0)
	value=tmp;
    }
unknown's avatar
unknown committed
2115 2116
    if ((null_value= args[i]->null_value))
      break;
unknown's avatar
unknown committed
2117 2118 2119 2120
  }
  return value;
}

unknown's avatar
unknown committed
2121 2122 2123 2124

my_decimal *Item_func_min_max::val_decimal(my_decimal *dec)
{
  DBUG_ASSERT(fixed == 1);
2125 2126 2127
  my_decimal tmp_buf, *tmp, *res;
  LINT_INIT(res);

unknown's avatar
unknown committed
2128 2129
  for (uint i=0; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
2130
    if (i == 0)
unknown's avatar
unknown committed
2131 2132 2133
      res= args[i]->val_decimal(dec);
    else
    {
2134 2135
      tmp= args[i]->val_decimal(&tmp_buf);      // Zero if NULL
      if (tmp && (my_decimal_cmp(tmp, res) * cmp_sign) < 0)
unknown's avatar
unknown committed
2136 2137 2138
      {
        if (tmp == &tmp_buf)
        {
2139
          /* Move value out of tmp_buf as this will be reused on next loop */
unknown's avatar
unknown committed
2140 2141 2142 2143 2144 2145 2146
          my_decimal2decimal(tmp, dec);
          res= dec;
        }
        else
          res= tmp;
      }
    }
unknown's avatar
unknown committed
2147
    if ((null_value= args[i]->null_value))
2148 2149
    {
      res= 0;
unknown's avatar
unknown committed
2150
      break;
2151
    }
unknown's avatar
unknown committed
2152 2153 2154 2155 2156
  }
  return res;
}


unknown's avatar
unknown committed
2157 2158
longlong Item_func_length::val_int()
{
2159
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2160 2161 2162 2163 2164 2165 2166 2167 2168 2169
  String *res=args[0]->val_str(&value);
  if (!res)
  {
    null_value=1;
    return 0; /* purecov: inspected */
  }
  null_value=0;
  return (longlong) res->length();
}

unknown's avatar
unknown committed
2170

unknown's avatar
unknown committed
2171 2172
longlong Item_func_char_length::val_int()
{
2173
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2174 2175 2176 2177 2178 2179 2180
  String *res=args[0]->val_str(&value);
  if (!res)
  {
    null_value=1;
    return 0; /* purecov: inspected */
  }
  null_value=0;
2181
  return (longlong) res->numchars();
unknown's avatar
unknown committed
2182 2183
}

unknown's avatar
unknown committed
2184

unknown's avatar
unknown committed
2185 2186
longlong Item_func_coercibility::val_int()
{
2187
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2188
  null_value= 0;
2189
  return (longlong) args[0]->collation.derivation;
unknown's avatar
unknown committed
2190
}
unknown's avatar
unknown committed
2191

unknown's avatar
unknown committed
2192

2193 2194 2195
void Item_func_locate::fix_length_and_dec()
{
  maybe_null=0; max_length=11;
unknown's avatar
unknown committed
2196
  agg_arg_charsets(cmp_collation, args, 2, MY_COLL_CMP_CONV);
2197 2198
}

unknown's avatar
unknown committed
2199

unknown's avatar
unknown committed
2200 2201
longlong Item_func_locate::val_int()
{
2202
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2203 2204 2205 2206 2207 2208 2209 2210
  String *a=args[0]->val_str(&value1);
  String *b=args[1]->val_str(&value2);
  if (!a || !b)
  {
    null_value=1;
    return 0; /* purecov: inspected */
  }
  null_value=0;
2211 2212
  uint start=0;
  uint start0=0;
2213
  my_match_t match;
2214

unknown's avatar
unknown committed
2215 2216
  if (arg_count == 3)
  {
2217 2218 2219
    start0= start =(uint) args[2]->val_int()-1;
    start=a->charpos(start);
    
unknown's avatar
unknown committed
2220 2221 2222
    if (start > a->length() || start+b->length() > a->length())
      return 0;
  }
2223

unknown's avatar
unknown committed
2224 2225
  if (!b->length())				// Found empty string at start
    return (longlong) (start+1);
2226
  
2227
  if (!cmp_collation.collation->coll->instr(cmp_collation.collation,
2228
                                            a->ptr()+start, a->length()-start,
2229 2230 2231 2232
                                            b->ptr(), b->length(),
                                            &match, 1))
    return 0;
  return (longlong) match.mblen + start0 + 1;
unknown's avatar
unknown committed
2233 2234 2235
}


2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250
void Item_func_locate::print(String *str)
{
  str->append("locate(", 7);
  args[1]->print(str);
  str->append(',');
  args[0]->print(str);
  if (arg_count == 3)
  {
    str->append(',');
    args[2]->print(str);
  }
  str->append(')');
}


unknown's avatar
unknown committed
2251 2252
longlong Item_func_field::val_int()
{
2253
  DBUG_ASSERT(fixed == 1);
2254

2255 2256 2257
  if (cmp_type == STRING_RESULT)
  {
    String *field;
2258 2259
    if (!(field= args[0]->val_str(&value)))
      return 0;
2260
    for (uint i=1 ; i < arg_count ; i++)
2261 2262
    {
      String *tmp_value=args[i]->val_str(&tmp);
unknown's avatar
Fix:  
unknown committed
2263
      if (tmp_value && !sortcmp(field,tmp_value,cmp_collation.collation))
2264
        return (longlong) (i);
2265 2266 2267 2268
    }
  }
  else if (cmp_type == INT_RESULT)
  {
2269
    longlong val= args[0]->val_int();
unknown's avatar
unknown committed
2270 2271
    if (args[0]->null_value)
      return 0;
2272
    for (uint i=1; i < arg_count ; i++)
2273
    {
2274
      if (val == args[i]->val_int() && !args[i]->null_value)
2275
        return (longlong) (i);
2276 2277
    }
  }
unknown's avatar
unknown committed
2278 2279 2280 2281
  else if (cmp_type == DECIMAL_RESULT)
  {
    my_decimal dec_arg_buf, *dec_arg,
               dec_buf, *dec= args[0]->val_decimal(&dec_buf);
unknown's avatar
unknown committed
2282 2283
    if (args[0]->null_value)
      return 0;
unknown's avatar
unknown committed
2284 2285 2286
    for (uint i=1; i < arg_count; i++)
    {
      dec_arg= args[i]->val_decimal(&dec_arg_buf);
unknown's avatar
unknown committed
2287
      if (!args[i]->null_value && !my_decimal_cmp(dec_arg, dec))
unknown's avatar
unknown committed
2288 2289 2290
        return (longlong) (i);
    }
  }
2291
  else
unknown's avatar
unknown committed
2292
  {
2293
    double val= args[0]->val_real();
unknown's avatar
unknown committed
2294 2295
    if (args[0]->null_value)
      return 0;
2296
    for (uint i=1; i < arg_count ; i++)
2297
    {
unknown's avatar
unknown committed
2298
      if (val == args[i]->val_real() && !args[i]->null_value)
2299
        return (longlong) (i);
2300
    }
unknown's avatar
unknown committed
2301 2302 2303 2304
  }
  return 0;
}

unknown's avatar
unknown committed
2305

2306 2307 2308
void Item_func_field::fix_length_and_dec()
{
  maybe_null=0; max_length=3;
2309 2310
  cmp_type= args[0]->result_type();
  for (uint i=1; i < arg_count ; i++)
2311 2312
    cmp_type= item_cmp_type(cmp_type, args[i]->result_type());
  if (cmp_type == STRING_RESULT)
unknown's avatar
unknown committed
2313
    agg_arg_charsets(cmp_collation, args, arg_count, MY_COLL_CMP_CONV);
2314
}
unknown's avatar
unknown committed
2315

2316

unknown's avatar
unknown committed
2317 2318
longlong Item_func_ascii::val_int()
{
2319
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331
  String *res=args[0]->val_str(&value);
  if (!res)
  {
    null_value=1;
    return 0;
  }
  null_value=0;
  return (longlong) (res->length() ? (uchar) (*res)[0] : (uchar) 0);
}

longlong Item_func_ord::val_int()
{
2332
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2333 2334 2335 2336 2337 2338 2339 2340 2341
  String *res=args[0]->val_str(&value);
  if (!res)
  {
    null_value=1;
    return 0;
  }
  null_value=0;
  if (!res->length()) return 0;
#ifdef USE_MB
2342
  if (use_mb(res->charset()))
unknown's avatar
unknown committed
2343 2344
  {
    register const char *str=res->ptr();
2345
    register uint32 n=0, l=my_ismbchar(res->charset(),str,str+res->length());
unknown's avatar
unknown committed
2346 2347
    if (!l)
      return (longlong)((uchar) *str);
unknown's avatar
unknown committed
2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371
    while (l--)
      n=(n<<8)|(uint32)((uchar) *str++);
    return (longlong) n;
  }
#endif
  return (longlong) ((uchar) (*res)[0]);
}

	/* Search after a string in a string of strings separated by ',' */
	/* Returns number of found type >= 1 or 0 if not found */
	/* This optimizes searching in enums to bit testing! */

void Item_func_find_in_set::fix_length_and_dec()
{
  decimals=0;
  max_length=3;					// 1-999
  if (args[0]->const_item() && args[1]->type() == FIELD_ITEM)
  {
    Field *field= ((Item_field*) args[1])->field;
    if (field->real_type() == FIELD_TYPE_SET)
    {
      String *find=args[0]->val_str(&value);
      if (find)
      {
2372 2373
	enum_value= find_type(((Field_enum*) field)->typelib,find->ptr(),
			      find->length(), 0);
unknown's avatar
unknown committed
2374 2375 2376 2377 2378 2379
	enum_bit=0;
	if (enum_value)
	  enum_bit=LL(1) << (enum_value-1);
      }
    }
  }
2380
  agg_arg_charsets(cmp_collation, args, 2, MY_COLL_CMP_CONV);
unknown's avatar
unknown committed
2381 2382 2383 2384 2385 2386
}

static const char separator=',';

longlong Item_func_find_in_set::val_int()
{
2387
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410
  if (enum_value)
  {
    ulonglong tmp=(ulonglong) args[1]->val_int();
    if (!(null_value=args[1]->null_value || args[0]->null_value))
    {
      if (tmp & enum_bit)
	return enum_value;
    }
    return 0L;
  }

  String *find=args[0]->val_str(&value);
  String *buffer=args[1]->val_str(&value2);
  if (!find || !buffer)
  {
    null_value=1;
    return 0; /* purecov: inspected */
  }
  null_value=0;

  int diff;
  if ((diff=buffer->length() - find->length()) >= 0)
  {
2411 2412 2413 2414 2415 2416 2417 2418 2419
    my_wc_t wc;
    CHARSET_INFO *cs= cmp_collation.collation;
    const char *str_begin= buffer->ptr();
    const char *str_end= buffer->ptr();
    const char *real_end= str_end+buffer->length();
    const uchar *find_str= (const uchar *) find->ptr();
    uint find_str_len= find->length();
    int position= 0;
    while (1)
unknown's avatar
unknown committed
2420
    {
2421 2422 2423
      int symbol_len;
      if ((symbol_len= cs->cset->mb_wc(cs, &wc, (uchar*) str_end, 
                                       (uchar*) real_end)) > 0)
unknown's avatar
unknown committed
2424
      {
2425 2426
        const char *substr_end= str_end + symbol_len;
        bool is_last_item= (substr_end == real_end);
2427 2428
        bool is_separator= (wc == (my_wc_t) separator);
        if (is_separator || is_last_item)
2429 2430
        {
          position++;
2431
          if (is_last_item && !is_separator)
2432 2433 2434 2435 2436 2437 2438 2439 2440
            str_end= substr_end;
          if (!my_strnncoll(cs, (const uchar *) str_begin,
                            str_end - str_begin,
                            find_str, find_str_len))
            return (longlong) position;
          else
            str_begin= substr_end;
        }
        str_end= substr_end;
unknown's avatar
unknown committed
2441
      }
unknown's avatar
unknown committed
2442 2443
      else if (str_end - str_begin == 0 &&
               find_str_len == 0 &&
2444 2445 2446
               wc == (my_wc_t) separator)
        return (longlong) ++position;
      else
unknown's avatar
unknown committed
2447
        return LL(0);
2448
    }
unknown's avatar
unknown committed
2449 2450 2451 2452 2453 2454
  }
  return 0;
}

longlong Item_func_bit_count::val_int()
{
2455
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2456 2457 2458 2459 2460 2461
  ulonglong value= (ulonglong) args[0]->val_int();
  if (args[0]->null_value)
  {
    null_value=1; /* purecov: inspected */
    return 0; /* purecov: inspected */
  }
unknown's avatar
unknown committed
2462
  return (longlong) my_count_bits(value);
unknown's avatar
unknown committed
2463 2464 2465 2466 2467 2468
}


/****************************************************************************
** Functions to handle dynamic loadable functions
** Original source by: Alexis Mikhailov <root@medinf.chuvashia.su>
2469
** Rewritten by monty.
unknown's avatar
unknown committed
2470 2471 2472 2473
****************************************************************************/

#ifdef HAVE_DLOPEN

2474
void udf_handler::cleanup()
unknown's avatar
unknown committed
2475
{
2476
  if (!not_original)
unknown's avatar
unknown committed
2477
  {
2478
    if (initialized)
unknown's avatar
unknown committed
2479
    {
2480 2481 2482 2483 2484 2485 2486
      if (u_d->func_deinit != NULL)
      {
        void (*deinit)(UDF_INIT *) = (void (*)(UDF_INIT*))
        u_d->func_deinit;
        (*deinit)(&initid);
      }
      free_udf(u_d);
2487
      initialized= FALSE;
unknown's avatar
unknown committed
2488
    }
2489 2490
    if (buffers)				// Because of bug in ecc
      delete [] buffers;
2491
    buffers= 0;
unknown's avatar
unknown committed
2492 2493 2494 2495 2496
  }
}


bool
2497
udf_handler::fix_fields(THD *thd, Item_result_field *func,
unknown's avatar
unknown committed
2498 2499
			uint arg_count, Item **arguments)
{
2500 2501 2502
#ifndef EMBEDDED_LIBRARY			// Avoid compiler warning
  char buff[STACK_BUFF_ALLOC];			// Max argument in function
#endif
unknown's avatar
unknown committed
2503 2504
  DBUG_ENTER("Item_udf_func::fix_fields");

2505
  if (check_stack_overrun(thd, STACK_MIN_SIZE, buff))
unknown's avatar
unknown committed
2506
    DBUG_RETURN(TRUE);				// Fatal error flag is set!
unknown's avatar
unknown committed
2507

2508
  udf_func *tmp_udf=find_udf(u_d->name.str,(uint) u_d->name.length,1);
unknown's avatar
unknown committed
2509 2510 2511

  if (!tmp_udf)
  {
2512
    my_error(ER_CANT_FIND_UDF, MYF(0), u_d->name.str, errno);
unknown's avatar
unknown committed
2513
    DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2514 2515 2516 2517 2518
  }
  u_d=tmp_udf;
  args=arguments;

  /* Fix all arguments */
2519
  func->maybe_null=0;
unknown's avatar
unknown committed
2520 2521 2522 2523 2524 2525 2526 2527 2528 2529
  used_tables_cache=0;
  const_item_cache=1;

  if ((f_args.arg_count=arg_count))
  {
    if (!(f_args.arg_type= (Item_result*)
	  sql_alloc(f_args.arg_count*sizeof(Item_result))))

    {
      free_udf(u_d);
unknown's avatar
unknown committed
2530
      DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2531 2532 2533 2534 2535 2536 2537
    }
    uint i;
    Item **arg,**arg_end;
    for (i=0, arg=arguments, arg_end=arguments+arg_count;
	 arg != arg_end ;
	 arg++,i++)
    {
unknown's avatar
unknown committed
2538
      if (!(*arg)->fixed &&
2539
          (*arg)->fix_fields(thd, arg))
unknown's avatar
unknown committed
2540
	DBUG_RETURN(1);
2541
      // we can't assign 'item' before, because fix_fields() can change arg
unknown's avatar
unknown committed
2542
      Item *item= *arg;
2543
      if (item->check_cols(1))
unknown's avatar
unknown committed
2544
	DBUG_RETURN(TRUE);
2545 2546 2547 2548 2549 2550 2551 2552 2553
      /*
	TODO: We should think about this. It is not always
	right way just to set an UDF result to return my_charset_bin
	if one argument has binary sorting order.
	The result collation should be calculated according to arguments
	derivations in some cases and should not in other cases.
	Moreover, some arguments can represent a numeric input
	which doesn't effect the result character set and collation.
	There is no a general rule for UDF. Everything depends on
unknown's avatar
unknown committed
2554
        the particular user defined function.
2555
      */
2556 2557
      if (item->collation.collation->state & MY_CS_BINSORT)
	func->collation.set(&my_charset_bin);
unknown's avatar
unknown committed
2558
      if (item->maybe_null)
unknown's avatar
unknown committed
2559
	func->maybe_null=1;
unknown's avatar
unknown committed
2560 2561 2562 2563
      func->with_sum_func= func->with_sum_func || item->with_sum_func;
      used_tables_cache|=item->used_tables();
      const_item_cache&=item->const_item();
      f_args.arg_type[i]=item->result_type();
unknown's avatar
unknown committed
2564
    }
unknown's avatar
unknown committed
2565
    //TODO: why all following memory is not allocated with 1 call of sql_alloc?
unknown's avatar
unknown committed
2566 2567
    if (!(buffers=new String[arg_count]) ||
	!(f_args.args= (char**) sql_alloc(arg_count * sizeof(char *))) ||
2568 2569 2570 2571 2572 2573 2574
	!(f_args.lengths= (ulong*) sql_alloc(arg_count * sizeof(long))) ||
	!(f_args.maybe_null= (char*) sql_alloc(arg_count * sizeof(char))) ||
	!(num_buffer= (char*) sql_alloc(arg_count *
					ALIGN_SIZE(sizeof(double)))) ||
	!(f_args.attributes= (char**) sql_alloc(arg_count * sizeof(char *))) ||
	!(f_args.attribute_lengths= (ulong*) sql_alloc(arg_count *
						       sizeof(long))))
unknown's avatar
unknown committed
2575 2576
    {
      free_udf(u_d);
unknown's avatar
unknown committed
2577
      DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592
    }
  }
  func->fix_length_and_dec();
  initid.max_length=func->max_length;
  initid.maybe_null=func->maybe_null;
  initid.const_item=const_item_cache;
  initid.decimals=func->decimals;
  initid.ptr=0;

  if (u_d->func_init)
  {
    char *to=num_buffer;
    for (uint i=0; i < arg_count; i++)
    {
      f_args.args[i]=0;
2593 2594 2595 2596
      f_args.lengths[i]= arguments[i]->max_length;
      f_args.maybe_null[i]= (char) arguments[i]->maybe_null;
      f_args.attributes[i]= arguments[i]->name;
      f_args.attribute_lengths[i]= arguments[i]->name_length;
unknown's avatar
unknown committed
2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615

      switch(arguments[i]->type()) {
      case Item::STRING_ITEM:			// Constant string !
      {
	String *res=arguments[i]->val_str((String *) 0);
	if (arguments[i]->null_value)
	  continue;
	f_args.args[i]=    (char*) res->ptr();
	break;
      }
      case Item::INT_ITEM:
	*((longlong*) to) = arguments[i]->val_int();
	if (!arguments[i]->null_value)
	{
	  f_args.args[i]=to;
	  to+= ALIGN_SIZE(sizeof(longlong));
	}
	break;
      case Item::REAL_ITEM:
2616
	*((double*) to)= arguments[i]->val_real();
unknown's avatar
unknown committed
2617 2618 2619 2620 2621 2622 2623 2624 2625 2626
	if (!arguments[i]->null_value)
	{
	  f_args.args[i]=to;
	  to+= ALIGN_SIZE(sizeof(double));
	}
	break;
      default:					// Skip these
	break;
      }
    }
2627
    thd->net.last_error[0]=0;
unknown's avatar
unknown committed
2628 2629 2630 2631 2632
    my_bool (*init)(UDF_INIT *, UDF_ARGS *, char *)=
      (my_bool (*)(UDF_INIT *, UDF_ARGS *,  char *))
      u_d->func_init;
    if ((error=(uchar) init(&initid, &f_args, thd->net.last_error)))
    {
2633 2634
      my_error(ER_CANT_INITIALIZE_UDF, MYF(0),
               u_d->name.str, thd->net.last_error);
unknown's avatar
unknown committed
2635
      free_udf(u_d);
unknown's avatar
unknown committed
2636
      DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2637 2638 2639 2640
    }
    func->max_length=min(initid.max_length,MAX_BLOB_WIDTH);
    func->maybe_null=initid.maybe_null;
    const_item_cache=initid.const_item;
2641
    func->decimals=min(initid.decimals,NOT_FIXED_DEC);
unknown's avatar
unknown committed
2642 2643 2644 2645
  }
  initialized=1;
  if (error)
  {
2646 2647
    my_error(ER_CANT_INITIALIZE_UDF, MYF(0),
             u_d->name.str, ER(ER_UNKNOWN_ERROR));
unknown's avatar
unknown committed
2648
    DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2649
  }
unknown's avatar
unknown committed
2650
  DBUG_RETURN(FALSE);
unknown's avatar
unknown committed
2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664
}


bool udf_handler::get_arguments()
{
  if (error)
    return 1;					// Got an error earlier
  char *to= num_buffer;
  uint str_count=0;
  for (uint i=0; i < f_args.arg_count; i++)
  {
    f_args.args[i]=0;
    switch (f_args.arg_type[i]) {
    case STRING_RESULT:
unknown's avatar
unknown committed
2665
    case DECIMAL_RESULT:
unknown's avatar
unknown committed
2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683
      {
	String *res=args[i]->val_str(&buffers[str_count++]);
	if (!(args[i]->null_value))
	{
	  f_args.args[i]=    (char*) res->ptr();
	  f_args.lengths[i]= res->length();
	  break;
	}
      }
    case INT_RESULT:
      *((longlong*) to) = args[i]->val_int();
      if (!args[i]->null_value)
      {
	f_args.args[i]=to;
	to+= ALIGN_SIZE(sizeof(longlong));
      }
      break;
    case REAL_RESULT:
2684
      *((double*) to)= args[i]->val_real();
unknown's avatar
unknown committed
2685 2686 2687 2688 2689 2690
      if (!args[i]->null_value)
      {
	f_args.args[i]=to;
	to+= ALIGN_SIZE(sizeof(double));
      }
      break;
2691
    case ROW_RESULT:
unknown's avatar
unknown committed
2692
    default:
unknown's avatar
unknown committed
2693
      // This case should never be chosen
unknown's avatar
unknown committed
2694
      DBUG_ASSERT(0);
2695
      break;
unknown's avatar
unknown committed
2696 2697 2698 2699 2700 2701 2702 2703 2704
    }
  }
  return 0;
}

/* This returns (String*) 0 in case of NULL values */

String *udf_handler::val_str(String *str,String *save_str)
{
2705
  uchar is_null_tmp=0;
unknown's avatar
unknown committed
2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721
  ulong res_length;

  if (get_arguments())
    return 0;
  char * (*func)(UDF_INIT *, UDF_ARGS *, char *, ulong *, uchar *, uchar *)=
    (char* (*)(UDF_INIT *, UDF_ARGS *, char *, ulong *, uchar *, uchar *))
    u_d->func;

  if ((res_length=str->alloced_length()) < MAX_FIELD_WIDTH)
  {						// This happens VERY seldom
    if (str->alloc(MAX_FIELD_WIDTH))
    {
      error=1;
      return 0;
    }
  }
2722 2723 2724
  char *res=func(&initid, &f_args, (char*) str->ptr(), &res_length,
		 &is_null_tmp, &error);
  if (is_null_tmp || !res || error)		// The !res is for safety
unknown's avatar
unknown committed
2725 2726 2727 2728 2729 2730 2731 2732
  {
    return 0;
  }
  if (res == str->ptr())
  {
    str->length(res_length);
    return str;
  }
2733
  save_str->set(res, res_length, str->charset());
unknown's avatar
unknown committed
2734 2735 2736 2737
  return save_str;
}


2738 2739 2740 2741
/*
  For the moment, UDF functions are returning DECIMAL values as strings
*/

unknown's avatar
unknown committed
2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761
my_decimal *udf_handler::val_decimal(my_bool *null_value, my_decimal *dec_buf)
{
  char buf[DECIMAL_MAX_STR_LENGTH+1], *end;
  ulong res_length= DECIMAL_MAX_STR_LENGTH;

  if (get_arguments())
  {
    *null_value=1;
    return 0;
  }
  char *(*func)(UDF_INIT *, UDF_ARGS *, char *, ulong *, uchar *, uchar *)=
    (char* (*)(UDF_INIT *, UDF_ARGS *, char *, ulong *, uchar *, uchar *))
    u_d->func;

  char *res= func(&initid, &f_args, buf, &res_length, &is_null, &error);
  if (is_null || error)
  {
    *null_value= 1;
    return 0;
  }
2762 2763
  end= res+ res_length;
  str2my_decimal(E_DEC_FATAL_ERROR, res, dec_buf, &end);
unknown's avatar
unknown committed
2764 2765 2766
  return dec_buf;
}

unknown's avatar
unknown committed
2767

2768 2769 2770 2771 2772 2773
void Item_udf_func::cleanup()
{
  udf.cleanup();
  Item_func::cleanup();
}

unknown's avatar
unknown committed
2774

2775
double Item_func_udf_float::val_real()
unknown's avatar
unknown committed
2776
{
2777
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2778 2779 2780 2781 2782 2783 2784 2785 2786
  DBUG_ENTER("Item_func_udf_float::val");
  DBUG_PRINT("info",("result_type: %d  arg_count: %d",
		     args[0]->result_type(), arg_count));
  DBUG_RETURN(udf.val(&null_value));
}


String *Item_func_udf_float::val_str(String *str)
{
2787
  DBUG_ASSERT(fixed == 1);
2788
  double nr= val_real();
unknown's avatar
unknown committed
2789 2790
  if (null_value)
    return 0;					/* purecov: inspected */
unknown's avatar
unknown committed
2791
  str->set(nr,decimals,&my_charset_bin);
unknown's avatar
unknown committed
2792 2793 2794 2795 2796 2797
  return str;
}


longlong Item_func_udf_int::val_int()
{
2798
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2799 2800 2801 2802 2803 2804 2805 2806 2807 2808
  DBUG_ENTER("Item_func_udf_int::val_int");
  DBUG_PRINT("info",("result_type: %d  arg_count: %d",
		     args[0]->result_type(), arg_count));

  DBUG_RETURN(udf.val_int(&null_value));
}


String *Item_func_udf_int::val_str(String *str)
{
2809
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2810 2811 2812
  longlong nr=val_int();
  if (null_value)
    return 0;
2813
  if (!unsigned_flag)
unknown's avatar
unknown committed
2814
    str->set(nr,&my_charset_bin);
2815
  else
unknown's avatar
unknown committed
2816
    str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
2817 2818 2819
  return str;
}

unknown's avatar
unknown committed
2820 2821 2822 2823

longlong Item_func_udf_decimal::val_int()
{
  my_decimal dec_buf, *dec= udf.val_decimal(&null_value, &dec_buf);
2824
  longlong result;
unknown's avatar
unknown committed
2825 2826 2827 2828 2829 2830 2831 2832 2833 2834
  if (null_value)
    return 0;
  my_decimal2int(E_DEC_FATAL_ERROR, dec, unsigned_flag, &result);
  return result;
}


double Item_func_udf_decimal::val_real()
{
  my_decimal dec_buf, *dec= udf.val_decimal(&null_value, &dec_buf);
2835
  double result;
unknown's avatar
unknown committed
2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872
  if (null_value)
    return 0.0;
  my_decimal2double(E_DEC_FATAL_ERROR, dec, &result);
  return result;
}


my_decimal *Item_func_udf_decimal::val_decimal(my_decimal *dec_buf)
{
  DBUG_ASSERT(fixed == 1);
  DBUG_ENTER("Item_func_udf_decimal::val_decimal");
  DBUG_PRINT("info",("result_type: %d  arg_count: %d",
                     args[0]->result_type(), arg_count));

  DBUG_RETURN(udf.val_decimal(&null_value, dec_buf));
}


String *Item_func_udf_decimal::val_str(String *str)
{
  my_decimal dec_buf, *dec= udf.val_decimal(&null_value, &dec_buf);
  if (null_value)
    return 0;
  if (str->length() < DECIMAL_MAX_STR_LENGTH)
    str->length(DECIMAL_MAX_STR_LENGTH);
  my_decimal_round(E_DEC_FATAL_ERROR, dec, decimals, FALSE, &dec_buf);
  my_decimal2string(E_DEC_FATAL_ERROR, &dec_buf, 0, 0, '0', str);
  return str;
}


void Item_func_udf_decimal::fix_length_and_dec()
{
  fix_num_length_and_dec();
}


unknown's avatar
unknown committed
2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885
/* Default max_length is max argument length */

void Item_func_udf_str::fix_length_and_dec()
{
  DBUG_ENTER("Item_func_udf_str::fix_length_and_dec");
  max_length=0;
  for (uint i = 0; i < arg_count; i++)
    set_if_bigger(max_length,args[i]->max_length);
  DBUG_VOID_RETURN;
}

String *Item_func_udf_str::val_str(String *str)
{
2886
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2887 2888 2889 2890 2891
  String *res=udf.val_str(str,&str_value);
  null_value = !res;
  return res;
}

2892 2893

/*
2894 2895
   This has to come last in the udf_handler methods, or C for AIX
   version 6.0.0.0 fails to compile with debugging enabled. (Yes, really.)
2896 2897 2898 2899
 */

udf_handler::~udf_handler()
{
2900 2901
  /* Everything should be properly cleaned up by this moment. */
  DBUG_ASSERT(not_original || !(initialized || buffers));
2902 2903
}

unknown's avatar
unknown committed
2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914
#else
bool udf_handler::get_arguments() { return 0; }
#endif /* HAVE_DLOPEN */

/*
** User level locks
*/

pthread_mutex_t LOCK_user_locks;
static HASH hash_user_locks;

2915
class User_level_lock
unknown's avatar
unknown committed
2916 2917 2918 2919 2920 2921 2922 2923 2924
{
  char *key;
  uint key_length;

public:
  int count;
  bool locked;
  pthread_cond_t cond;
  pthread_t thread;
unknown's avatar
SCRUM  
unknown committed
2925
  ulong thread_id;
unknown's avatar
unknown committed
2926

2927
  User_level_lock(const char *key_arg,uint length, ulong id) 
unknown's avatar
SCRUM  
unknown committed
2928
    :key_length(length),count(1),locked(1), thread_id(id)
unknown's avatar
unknown committed
2929 2930 2931 2932 2933
  {
    key=(char*) my_memdup((byte*) key_arg,length,MYF(0));
    pthread_cond_init(&cond,NULL);
    if (key)
    {
unknown's avatar
SCRUM  
unknown committed
2934
      if (my_hash_insert(&hash_user_locks,(byte*) this))
unknown's avatar
unknown committed
2935 2936 2937 2938 2939 2940
      {
	my_free((gptr) key,MYF(0));
	key=0;
      }
    }
  }
2941
  ~User_level_lock()
unknown's avatar
unknown committed
2942 2943 2944 2945 2946 2947 2948 2949 2950
  {
    if (key)
    {
      hash_delete(&hash_user_locks,(byte*) this);
      my_free((gptr) key,MYF(0));
    }
    pthread_cond_destroy(&cond);
  }
  inline bool initialized() { return key != 0; }
2951 2952 2953
  friend void item_user_lock_release(User_level_lock *ull);
  friend char *ull_get_key(const User_level_lock *ull, uint *length,
                           my_bool not_used);
unknown's avatar
unknown committed
2954 2955
};

2956
char *ull_get_key(const User_level_lock *ull, uint *length,
unknown's avatar
unknown committed
2957 2958 2959 2960 2961 2962
		  my_bool not_used __attribute__((unused)))
{
  *length=(uint) ull->key_length;
  return (char*) ull->key;
}

2963 2964 2965

static bool item_user_lock_inited= 0;

unknown's avatar
unknown committed
2966 2967
void item_user_lock_init(void)
{
2968
  pthread_mutex_init(&LOCK_user_locks,MY_MUTEX_INIT_SLOW);
unknown's avatar
unknown committed
2969 2970
  hash_init(&hash_user_locks,system_charset_info,
	    16,0,0,(hash_get_key) ull_get_key,NULL,0);
2971
  item_user_lock_inited= 1;
unknown's avatar
unknown committed
2972 2973 2974 2975
}

void item_user_lock_free(void)
{
2976 2977 2978 2979 2980 2981
  if (item_user_lock_inited)
  {
    item_user_lock_inited= 0;
    hash_free(&hash_user_locks);
    pthread_mutex_destroy(&LOCK_user_locks);
  }
unknown's avatar
unknown committed
2982 2983
}

2984
void item_user_lock_release(User_level_lock *ull)
unknown's avatar
unknown committed
2985 2986 2987 2988 2989 2990 2991 2992
{
  ull->locked=0;
  if (--ull->count)
    pthread_cond_signal(&ull->cond);
  else
    delete ull;
}

unknown's avatar
unknown committed
2993 2994 2995 2996 2997 2998 2999
/*
   Wait until we are at or past the given position in the master binlog
   on the slave
 */

longlong Item_master_pos_wait::val_int()
{
3000
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3001 3002
  THD* thd = current_thd;
  String *log_name = args[0]->val_str(&value);
3003
  int event_count= 0;
unknown's avatar
unknown committed
3004

unknown's avatar
unknown committed
3005 3006 3007 3008 3009 3010
  null_value=0;
  if (thd->slave_thread || !log_name || !log_name->length())
  {
    null_value = 1;
    return 0;
  }
unknown's avatar
unknown committed
3011
  longlong pos = (ulong)args[1]->val_int();
3012
  longlong timeout = (arg_count==3) ? args[2]->val_int() : 0 ;
unknown's avatar
SCRUM  
unknown committed
3013
#ifdef HAVE_REPLICATION
3014
  if ((event_count = active_mi->rli.wait_for_pos(thd, log_name, pos, timeout)) == -2)
unknown's avatar
unknown committed
3015 3016 3017 3018
  {
    null_value = 1;
    event_count=0;
  }
3019
#endif
unknown's avatar
unknown committed
3020 3021 3022
  return event_count;
}

3023 3024 3025 3026
#ifdef EXTRA_DEBUG
void debug_sync_point(const char* lock_name, uint lock_timeout)
{
  THD* thd=current_thd;
3027
  User_level_lock* ull;
3028
  struct timespec abstime;
3029
  int lock_name_len;
3030 3031 3032 3033 3034 3035 3036 3037 3038
  lock_name_len=strlen(lock_name);
  pthread_mutex_lock(&LOCK_user_locks);

  if (thd->ull)
  {
    item_user_lock_release(thd->ull);
    thd->ull=0;
  }

3039 3040 3041 3042 3043
  /*
    If the lock has not been aquired by some client, we do not want to
    create an entry for it, since we immediately release the lock. In
    this case, we will not be waiting, but rather, just waste CPU and
    memory on the whole deal
3044
  */
3045
  if (!(ull= ((User_level_lock*) hash_search(&hash_user_locks, lock_name,
3046 3047 3048 3049 3050 3051 3052
				 lock_name_len))))
  {
    pthread_mutex_unlock(&LOCK_user_locks);
    return;
  }
  ull->count++;

3053 3054 3055 3056
  /*
    Structure is now initialized.  Try to get the lock.
    Set up control struct to allow others to abort locks
  */
3057 3058 3059 3060
  thd->proc_info="User lock";
  thd->mysys_var->current_mutex= &LOCK_user_locks;
  thd->mysys_var->current_cond=  &ull->cond;

3061
  set_timespec(abstime,lock_timeout);
unknown's avatar
unknown committed
3062 3063 3064 3065 3066 3067 3068
  while (ull->locked && !thd->killed)
  {
    int error= pthread_cond_timedwait(&ull->cond, &LOCK_user_locks, &abstime);
    if (error == ETIMEDOUT || error == ETIME)
      break;
  }

3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096
  if (ull->locked)
  {
    if (!--ull->count)
      delete ull;				// Should never happen
  }
  else
  {
    ull->locked=1;
    ull->thread=thd->real_id;
    thd->ull=ull;
  }
  pthread_mutex_unlock(&LOCK_user_locks);
  pthread_mutex_lock(&thd->mysys_var->mutex);
  thd->proc_info=0;
  thd->mysys_var->current_mutex= 0;
  thd->mysys_var->current_cond=  0;
  pthread_mutex_unlock(&thd->mysys_var->mutex);
  pthread_mutex_lock(&LOCK_user_locks);
  if (thd->ull)
  {
    item_user_lock_release(thd->ull);
    thd->ull=0;
  }
  pthread_mutex_unlock(&LOCK_user_locks);
}

#endif

unknown's avatar
unknown committed
3097 3098 3099 3100 3101 3102 3103 3104 3105
/*
  Get a user level lock. If the thread has an old lock this is first released.
  Returns 1:  Got lock
  Returns 0:  Timeout
  Returns NULL: Error
*/

longlong Item_func_get_lock::val_int()
{
3106
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3107 3108 3109 3110
  String *res=args[0]->val_str(&value);
  longlong timeout=args[1]->val_int();
  struct timespec abstime;
  THD *thd=current_thd;
3111
  User_level_lock *ull;
unknown's avatar
unknown committed
3112
  int error;
unknown's avatar
unknown committed
3113

3114 3115 3116 3117 3118 3119 3120 3121 3122 3123
  /*
    In slave thread no need to get locks, everything is serialized. Anyway
    there is no way to make GET_LOCK() work on slave like it did on master
    (i.e. make it return exactly the same value) because we don't have the
    same other concurrent threads environment. No matter what we return here,
    it's not guaranteed to be same as on master.
  */
  if (thd->slave_thread)
    return 1;

unknown's avatar
unknown committed
3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139
  pthread_mutex_lock(&LOCK_user_locks);

  if (!res || !res->length())
  {
    pthread_mutex_unlock(&LOCK_user_locks);
    null_value=1;
    return 0;
  }
  null_value=0;

  if (thd->ull)
  {
    item_user_lock_release(thd->ull);
    thd->ull=0;
  }

3140 3141 3142
  if (!(ull= ((User_level_lock *) hash_search(&hash_user_locks,
                                              (byte*) res->ptr(),
                                              res->length()))))
unknown's avatar
unknown committed
3143
  {
3144
    ull=new User_level_lock(res->ptr(),res->length(), thd->thread_id);
unknown's avatar
unknown committed
3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158
    if (!ull || !ull->initialized())
    {
      delete ull;
      pthread_mutex_unlock(&LOCK_user_locks);
      null_value=1;				// Probably out of memory
      return 0;
    }
    ull->thread=thd->real_id;
    thd->ull=ull;
    pthread_mutex_unlock(&LOCK_user_locks);
    return 1;					// Got new lock
  }
  ull->count++;

3159 3160 3161 3162
  /*
    Structure is now initialized.  Try to get the lock.
    Set up control struct to allow others to abort locks.
  */
unknown's avatar
unknown committed
3163 3164 3165 3166
  thd->proc_info="User lock";
  thd->mysys_var->current_mutex= &LOCK_user_locks;
  thd->mysys_var->current_cond=  &ull->cond;

3167
  set_timespec(abstime,timeout);
unknown's avatar
unknown committed
3168 3169 3170 3171 3172 3173 3174 3175 3176
  error= 0;
  while (ull->locked && !thd->killed)
  {
    error= pthread_cond_timedwait(&ull->cond,&LOCK_user_locks,&abstime);
    if (error == ETIMEDOUT || error == ETIME)
      break;
    error= 0;
  }

unknown's avatar
unknown committed
3177 3178 3179
  if (ull->locked)
  {
    if (!--ull->count)
unknown's avatar
unknown committed
3180 3181
    {
      DBUG_ASSERT(0);
unknown's avatar
unknown committed
3182
      delete ull;				// Should never happen
unknown's avatar
unknown committed
3183 3184
    }
    if (!error)                                 // Killed (thd->killed != 0)
unknown's avatar
unknown committed
3185 3186 3187 3188 3189
    {
      error=1;
      null_value=1;				// Return NULL
    }
  }
unknown's avatar
unknown committed
3190
  else                                          // We got the lock
unknown's avatar
unknown committed
3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209
  {
    ull->locked=1;
    ull->thread=thd->real_id;
    thd->ull=ull;
    error=0;
  }
  pthread_mutex_unlock(&LOCK_user_locks);

  pthread_mutex_lock(&thd->mysys_var->mutex);
  thd->proc_info=0;
  thd->mysys_var->current_mutex= 0;
  thd->mysys_var->current_cond=  0;
  pthread_mutex_unlock(&thd->mysys_var->mutex);

  return !error ? 1 : 0;
}


/*
3210 3211 3212 3213 3214
  Release a user level lock.
  Return:
    1 if lock released
    0 if lock wasn't held
    (SQL) NULL if no such lock
unknown's avatar
unknown committed
3215 3216 3217 3218
*/

longlong Item_func_release_lock::val_int()
{
3219
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3220
  String *res=args[0]->val_str(&value);
3221
  User_level_lock *ull;
unknown's avatar
unknown committed
3222 3223 3224 3225 3226 3227 3228 3229 3230 3231
  longlong result;
  if (!res || !res->length())
  {
    null_value=1;
    return 0;
  }
  null_value=0;

  result=0;
  pthread_mutex_lock(&LOCK_user_locks);
3232 3233 3234
  if (!(ull= ((User_level_lock*) hash_search(&hash_user_locks,
                                             (const byte*) res->ptr(),
                                             res->length()))))
unknown's avatar
unknown committed
3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251
  {
    null_value=1;
  }
  else
  {
    if (ull->locked && pthread_equal(pthread_self(),ull->thread))
    {
      result=1;					// Release is ok
      item_user_lock_release(ull);
      current_thd->ull=0;
    }
  }
  pthread_mutex_unlock(&LOCK_user_locks);
  return result;
}


3252
longlong Item_func_last_insert_id::val_int()
unknown's avatar
unknown committed
3253
{
unknown's avatar
unknown committed
3254
  THD *thd= current_thd;
3255
  DBUG_ASSERT(fixed == 1);
3256 3257
  if (arg_count)
  {
unknown's avatar
unknown committed
3258 3259 3260 3261
    longlong value= args[0]->val_int();
    thd->insert_id(value);
    null_value= args[0]->null_value;
    return value;                       // Avoid side effect of insert_id()
3262
  }
unknown's avatar
unknown committed
3263 3264
  thd->lex->uncacheable(UNCACHEABLE_SIDEEFFECT);
  return thd->insert_id();
unknown's avatar
unknown committed
3265 3266 3267 3268 3269 3270
}

/* This function is just used to test speed of different functions */

longlong Item_func_benchmark::val_int()
{
3271
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3272
  char buff[MAX_FIELD_WIDTH];
unknown's avatar
unknown committed
3273
  String tmp(buff,sizeof(buff), &my_charset_bin);
unknown's avatar
unknown committed
3274 3275 3276 3277 3278 3279
  THD *thd=current_thd;

  for (ulong loop=0 ; loop < loop_count && !thd->killed; loop++)
  {
    switch (args[0]->result_type()) {
    case REAL_RESULT:
3280
      (void) args[0]->val_real();
unknown's avatar
unknown committed
3281 3282 3283 3284 3285 3286 3287
      break;
    case INT_RESULT:
      (void) args[0]->val_int();
      break;
    case STRING_RESULT:
      (void) args[0]->val_str(&tmp);
      break;
3288
    case ROW_RESULT:
unknown's avatar
unknown committed
3289
    default:
unknown's avatar
unknown committed
3290
      // This case should never be chosen
unknown's avatar
unknown committed
3291 3292
      DBUG_ASSERT(0);
      return 0;
unknown's avatar
unknown committed
3293 3294 3295 3296 3297
    }
  }
  return 0;
}

3298

3299 3300 3301 3302
void Item_func_benchmark::print(String *str)
{
  str->append("benchmark(", 10);
  char buffer[20];
unknown's avatar
unknown committed
3303 3304 3305
  // my_charset_bin is good enough for numbers
  String st(buffer, sizeof(buffer), &my_charset_bin);
  st.set((ulonglong)loop_count, &my_charset_bin);
3306 3307 3308 3309 3310 3311
  str->append(st);
  str->append(',');
  args[0]->print(str);
  str->append(')');
}

unknown's avatar
unknown committed
3312

3313 3314 3315 3316
/* This function is just used to create tests with time gaps */

longlong Item_func_sleep::val_int()
{
3317 3318 3319
  THD *thd= current_thd;
  struct timespec abstime;
  pthread_cond_t cond;
3320
  int error;
3321

3322
  DBUG_ASSERT(fixed == 1);
3323

3324
  double time= args[0]->val_real();
3325 3326 3327 3328 3329 3330 3331 3332
  set_timespec_nsec(abstime, (ulonglong)(time * ULL(1000000000)));

  pthread_cond_init(&cond, NULL);
  pthread_mutex_lock(&LOCK_user_locks);

  thd->mysys_var->current_mutex= &LOCK_user_locks;
  thd->mysys_var->current_cond=  &cond;

unknown's avatar
unknown committed
3333 3334 3335 3336 3337 3338 3339 3340
  error= 0;
  while (!thd->killed)
  {
    error= pthread_cond_timedwait(&cond, &LOCK_user_locks, &abstime);
    if (error == ETIMEDOUT || error == ETIME)
      break;
    error= 0;
  }
3341 3342 3343 3344 3345 3346 3347 3348 3349

  pthread_mutex_lock(&thd->mysys_var->mutex);
  thd->mysys_var->current_mutex= 0;
  thd->mysys_var->current_cond=  0;
  pthread_mutex_unlock(&thd->mysys_var->mutex);

  pthread_mutex_unlock(&LOCK_user_locks);
  pthread_cond_destroy(&cond);

unknown's avatar
unknown committed
3350
  return test(!error); 		// Return 1 killed
3351 3352 3353
}


unknown's avatar
unknown committed
3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374
#define extra_size sizeof(double)

static user_var_entry *get_variable(HASH *hash, LEX_STRING &name,
				    bool create_if_not_exists)
{
  user_var_entry *entry;

  if (!(entry = (user_var_entry*) hash_search(hash, (byte*) name.str,
					      name.length)) &&
      create_if_not_exists)
  {
    uint size=ALIGN_SIZE(sizeof(user_var_entry))+name.length+1+extra_size;
    if (!hash_inited(hash))
      return 0;
    if (!(entry = (user_var_entry*) my_malloc(size,MYF(MY_WME))))
      return 0;
    entry->name.str=(char*) entry+ ALIGN_SIZE(sizeof(user_var_entry))+
      extra_size;
    entry->name.length=name.length;
    entry->value=0;
    entry->length=0;
3375
    entry->update_query_id=0;
3376
    entry->collation.set(NULL, DERIVATION_IMPLICIT);
3377 3378 3379 3380 3381 3382 3383 3384 3385 3386
    /*
      If we are here, we were called from a SET or a query which sets a
      variable. Imagine it is this:
      INSERT INTO t SELECT @a:=10, @a:=@a+1.
      Then when we have a Item_func_get_user_var (because of the @a+1) so we
      think we have to write the value of @a to the binlog. But before that,
      we have a Item_func_set_user_var to create @a (@a:=10), in this we mark
      the variable as "already logged" (line below) so that it won't be logged
      by Item_func_get_user_var (because that's not necessary).
    */
unknown's avatar
unknown committed
3387
    entry->used_query_id=current_thd->query_id;
unknown's avatar
unknown committed
3388 3389
    entry->type=STRING_RESULT;
    memcpy(entry->name.str, name.str, name.length+1);
unknown's avatar
SCRUM  
unknown committed
3390
    if (my_hash_insert(hash,(byte*) entry))
unknown's avatar
unknown committed
3391 3392 3393 3394 3395 3396 3397 3398
    {
      my_free((char*) entry,MYF(0));
      return 0;
    }
  }
  return entry;
}

3399
/*
3400 3401
  When a user variable is updated (in a SET command or a query like
  SELECT @a:= ).
3402
*/
3403

3404
bool Item_func_set_user_var::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
3405
{
3406
  DBUG_ASSERT(fixed == 0);
3407
  /* fix_fields will call Item_func_set_user_var::fix_length_and_dec */
3408
  if (Item_func::fix_fields(thd, ref) ||
unknown's avatar
unknown committed
3409
      !(entry= get_variable(&thd->user_vars, name, 1)))
unknown's avatar
unknown committed
3410
    return TRUE;
3411 3412 3413 3414 3415
  /* 
     Remember the last query which updated it, this way a query can later know
     if this variable is a constant item in the query (it is if update_query_id
     is different from query_id).
  */
3416
  entry->update_query_id= thd->query_id;
3417 3418 3419 3420 3421 3422 3423 3424 3425 3426 3427 3428 3429 3430 3431
  /*
    As it is wrong and confusing to associate any 
    character set with NULL, @a should be latin2
    after this query sequence:

      SET @a=_latin2'string';
      SET @a=NULL;

    I.e. the second query should not change the charset
    to the current default value, but should keep the 
    original value assigned during the first query.
    In order to do it, we don't copy charset
    from the argument if the argument is NULL
    and the variable has previously been initialized.
  */
3432 3433
  null_item= (args[0]->type() == NULL_ITEM);
  if (!entry->collation.collation || !null_item)
3434 3435
    entry->collation.set(args[0]->collation.collation, DERIVATION_IMPLICIT);
  collation.set(entry->collation.collation, DERIVATION_IMPLICIT);
3436
  cached_result_type= args[0]->result_type();
unknown's avatar
unknown committed
3437
  return FALSE;
unknown's avatar
unknown committed
3438 3439 3440 3441 3442 3443 3444 3445 3446
}


void
Item_func_set_user_var::fix_length_and_dec()
{
  maybe_null=args[0]->maybe_null;
  max_length=args[0]->max_length;
  decimals=args[0]->decimals;
3447
  collation.set(args[0]->collation.collation, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3448 3449
}

3450

unknown's avatar
unknown committed
3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470
/*
  Set value to user variable.

  SYNOPSYS
    update_hash()
    entry    - pointer to structure representing variable
    set_null - should we set NULL value ?
    ptr      - pointer to buffer with new value
    length   - length of new value
    type     - type of new value
    cs       - charset info for new value
    dv       - derivation for new value

  RETURN VALUE
    False - success, True - failure
*/

static bool
update_hash(user_var_entry *entry, bool set_null, void *ptr, uint length,
            Item_result type, CHARSET_INFO *cs, Derivation dv)
unknown's avatar
unknown committed
3471
{
unknown's avatar
unknown committed
3472
  if (set_null)
unknown's avatar
unknown committed
3473 3474 3475 3476
  {
    char *pos= (char*) entry+ ALIGN_SIZE(sizeof(user_var_entry));
    if (entry->value && entry->value != pos)
      my_free(entry->value,MYF(0));
3477 3478
    entry->value= 0;
    entry->length= 0;
unknown's avatar
unknown committed
3479 3480 3481
  }
  else
  {
3482 3483
    if (type == STRING_RESULT)
      length++;					// Store strings with end \0
unknown's avatar
unknown committed
3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504
    if (length <= extra_size)
    {
      /* Save value in value struct */
      char *pos= (char*) entry+ ALIGN_SIZE(sizeof(user_var_entry));
      if (entry->value != pos)
      {
	if (entry->value)
	  my_free(entry->value,MYF(0));
	entry->value=pos;
      }
    }
    else
    {
      /* Allocate variable */
      if (entry->length != length)
      {
	char *pos= (char*) entry+ ALIGN_SIZE(sizeof(user_var_entry));
	if (entry->value == pos)
	  entry->value=0;
	if (!(entry->value=(char*) my_realloc(entry->value, length,
					      MYF(MY_ALLOW_ZERO_PTR))))
unknown's avatar
unknown committed
3505
	  return 1;
unknown's avatar
unknown committed
3506 3507
      }
    }
3508 3509 3510 3511 3512
    if (type == STRING_RESULT)
    {
      length--;					// Fix length change above
      entry->value[length]= 0;			// Store end \0
    }
unknown's avatar
unknown committed
3513
    memcpy(entry->value,ptr,length);
unknown's avatar
unknown committed
3514 3515
    if (type == DECIMAL_RESULT)
      ((my_decimal*)entry->value)->fix_buffer_pointer();
unknown's avatar
unknown committed
3516
    entry->length= length;
3517
    entry->collation.set(cs, dv);
unknown's avatar
unknown committed
3518
  }
3519
  entry->type=type;
3520
  return 0;
unknown's avatar
unknown committed
3521
}
unknown's avatar
unknown committed
3522

unknown's avatar
unknown committed
3523 3524 3525 3526 3527

bool
Item_func_set_user_var::update_hash(void *ptr, uint length, Item_result type,
                                    CHARSET_INFO *cs, Derivation dv)
{
3528 3529 3530 3531 3532 3533
  /*
    If we set a variable explicitely to NULL then keep the old
    result type of the variable
  */
  if ((null_value= args[0]->null_value) && null_item)
    type= entry->type;                          // Don't change type of item
unknown's avatar
unknown committed
3534 3535 3536 3537 3538 3539 3540 3541
  if (::update_hash(entry, (null_value= args[0]->null_value),
                    ptr, length, type, cs, dv))
  {
    current_thd->fatal_error();     // Probably end of memory
    null_value= 1;
    return 1;
  }
  return 0;
unknown's avatar
unknown committed
3542 3543 3544
}


3545 3546
/* Get the value of a variable as a double */

unknown's avatar
unknown committed
3547
double user_var_entry::val_real(my_bool *null_value)
unknown's avatar
unknown committed
3548
{
3549 3550 3551 3552 3553 3554 3555 3556
  if ((*null_value= (value == 0)))
    return 0.0;

  switch (type) {
  case REAL_RESULT:
    return *(double*) value;
  case INT_RESULT:
    return (double) *(longlong*) value;
unknown's avatar
unknown committed
3557 3558 3559 3560 3561 3562
  case DECIMAL_RESULT:
  {
    double result;
    my_decimal2double(E_DEC_FATAL_ERROR, (my_decimal *)value, &result);
    return result;
  }
3563
  case STRING_RESULT:
unknown's avatar
unknown committed
3564
    return my_atof(value);                      // This is null terminated
3565 3566 3567
  case ROW_RESULT:
    DBUG_ASSERT(1);				// Impossible
    break;
unknown's avatar
unknown committed
3568
  }
3569
  return 0.0;					// Impossible
unknown's avatar
unknown committed
3570 3571 3572
}


3573 3574 3575
/* Get the value of a variable as an integer */

longlong user_var_entry::val_int(my_bool *null_value)
unknown's avatar
unknown committed
3576
{
3577 3578 3579 3580 3581 3582 3583 3584
  if ((*null_value= (value == 0)))
    return LL(0);

  switch (type) {
  case REAL_RESULT:
    return (longlong) *(double*) value;
  case INT_RESULT:
    return *(longlong*) value;
unknown's avatar
unknown committed
3585 3586 3587
  case DECIMAL_RESULT:
  {
    longlong result;
3588
    my_decimal2int(E_DEC_FATAL_ERROR, (my_decimal *)value, 0, &result);
unknown's avatar
unknown committed
3589 3590
    return result;
  }
unknown's avatar
unknown committed
3591
  case STRING_RESULT:
unknown's avatar
unknown committed
3592 3593 3594 3595
  {
    int error;
    return my_strtoll10(value, (char**) 0, &error);// String is null terminated
  }
3596 3597 3598
  case ROW_RESULT:
    DBUG_ASSERT(1);				// Impossible
    break;
unknown's avatar
unknown committed
3599
  }
3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613
  return LL(0);					// Impossible
}


/* Get the value of a variable as a string */

String *user_var_entry::val_str(my_bool *null_value, String *str,
				uint decimals)
{
  if ((*null_value= (value == 0)))
    return (String*) 0;

  switch (type) {
  case REAL_RESULT:
unknown's avatar
unknown committed
3614
    str->set(*(double*) value, decimals, &my_charset_bin);
3615 3616
    break;
  case INT_RESULT:
unknown's avatar
unknown committed
3617
    str->set(*(longlong*) value, &my_charset_bin);
3618
    break;
unknown's avatar
unknown committed
3619 3620 3621
  case DECIMAL_RESULT:
    my_decimal2string(E_DEC_FATAL_ERROR, (my_decimal *)value, 0, 0, 0, str);
    break;
3622
  case STRING_RESULT:
unknown's avatar
unknown committed
3623
    if (str->copy(value, length, collation.collation))
3624
      str= 0;					// EOM error
3625 3626 3627
  case ROW_RESULT:
    DBUG_ASSERT(1);				// Impossible
    break;
unknown's avatar
unknown committed
3628
  }
3629
  return(str);
unknown's avatar
unknown committed
3630 3631
}

unknown's avatar
unknown committed
3632 3633 3634 3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658
/* Get the value of a variable as a decimal */

my_decimal *user_var_entry::val_decimal(my_bool *null_value, my_decimal *val)
{
  if ((*null_value= (value == 0)))
    return 0;

  switch (type) {
  case REAL_RESULT:
    double2my_decimal(E_DEC_FATAL_ERROR, *(double*) value, val);
    break;
  case INT_RESULT:
    int2my_decimal(E_DEC_FATAL_ERROR, *(longlong*) value, 0, val);
    break;
  case DECIMAL_RESULT:
    val= (my_decimal *)value;
    break;
  case STRING_RESULT:
    str2my_decimal(E_DEC_FATAL_ERROR, value, length, collation.collation, val);
    break;
  case ROW_RESULT:
    DBUG_ASSERT(1);				// Impossible
    break;
  }
  return(val);
}

3659 3660
/*
  This functions is invoked on SET @variable or @variable:= expression.
unknown's avatar
unknown committed
3661
  Evaluate (and check expression), store results.
3662 3663 3664 3665 3666

  SYNOPSYS
    Item_func_set_user_var::check()

  NOTES
unknown's avatar
unknown committed
3667 3668
    For now it always return OK. All problem with value evaluating
    will be caught by thd->net.report_error check in sql_set_variables().
3669 3670

  RETURN
unknown's avatar
unknown committed
3671
    FALSE OK.
3672 3673 3674 3675 3676 3677 3678 3679 3680 3681
*/

bool
Item_func_set_user_var::check()
{
  DBUG_ENTER("Item_func_set_user_var::check");

  switch (cached_result_type) {
  case REAL_RESULT:
  {
3682
    save_result.vreal= args[0]->val_real();
3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693 3694
    break;
  }
  case INT_RESULT:
  {
    save_result.vint= args[0]->val_int();
    break;
  }
  case STRING_RESULT:
  {
    save_result.vstr= args[0]->val_str(&value);
    break;
  }
unknown's avatar
unknown committed
3695 3696 3697 3698 3699
  case DECIMAL_RESULT:
  {
    save_result.vdec= args[0]->val_decimal(&decimal_buff);
    break;
  }
3700 3701
  case ROW_RESULT:
  default:
unknown's avatar
unknown committed
3702
    // This case should never be chosen
3703 3704 3705
    DBUG_ASSERT(0);
    break;
  }
unknown's avatar
unknown committed
3706
  DBUG_RETURN(FALSE);
3707 3708
}

3709 3710 3711 3712 3713 3714 3715 3716 3717 3718 3719 3720

/*
  This functions is invoked on SET @variable or @variable:= expression.

  SYNOPSIS
    Item_func_set_user_var::update()

  NOTES
    We have to store the expression as such in the variable, independent of
    the value method used by the user

  RETURN
unknown's avatar
unknown committed
3721
    0	OK
3722 3723 3724 3725
    1	EOM Error

*/

unknown's avatar
unknown committed
3726 3727 3728
bool
Item_func_set_user_var::update()
{
3729 3730 3731 3732
  bool res;
  DBUG_ENTER("Item_func_set_user_var::update");
  LINT_INIT(res);

unknown's avatar
unknown committed
3733 3734
  switch (cached_result_type) {
  case REAL_RESULT:
unknown's avatar
unknown committed
3735
  {
3736
    res= update_hash((void*) &save_result.vreal,sizeof(save_result.vreal),
3737
		     REAL_RESULT, &my_charset_bin, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3738
    break;
unknown's avatar
unknown committed
3739
  }
unknown's avatar
unknown committed
3740
  case INT_RESULT:
3741
  {
3742
    res= update_hash((void*) &save_result.vint, sizeof(save_result.vint),
3743
		     INT_RESULT, &my_charset_bin, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3744
    break;
unknown's avatar
unknown committed
3745
  }
unknown's avatar
unknown committed
3746
  case STRING_RESULT:
unknown's avatar
unknown committed
3747
  {
3748
    if (!save_result.vstr)					// Null value
unknown's avatar
unknown committed
3749
      res= update_hash((void*) 0, 0, STRING_RESULT, &my_charset_bin,
3750
		       DERIVATION_IMPLICIT);
3751
    else
3752 3753 3754
      res= update_hash((void*) save_result.vstr->ptr(),
		       save_result.vstr->length(), STRING_RESULT,
		       save_result.vstr->charset(),
3755
		       DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3756 3757
    break;
  }
unknown's avatar
unknown committed
3758 3759 3760 3761
  case DECIMAL_RESULT:
  {
    if (!save_result.vdec)					// Null value
      res= update_hash((void*) 0, 0, DECIMAL_RESULT, &my_charset_bin,
unknown's avatar
unknown committed
3762
                       DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3763 3764 3765
    else
      res= update_hash((void*) save_result.vdec,
                       sizeof(my_decimal), DECIMAL_RESULT,
unknown's avatar
unknown committed
3766
                       &my_charset_bin, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3767 3768
    break;
  }
3769
  case ROW_RESULT:
unknown's avatar
unknown committed
3770
  default:
unknown's avatar
unknown committed
3771
    // This case should never be chosen
unknown's avatar
unknown committed
3772 3773 3774
    DBUG_ASSERT(0);
    break;
  }
3775
  DBUG_RETURN(res);
unknown's avatar
unknown committed
3776 3777 3778
}


3779
double Item_func_set_user_var::val_real()
unknown's avatar
unknown committed
3780
{
3781
  DBUG_ASSERT(fixed == 1);
3782
  check();
3783
  update();					// Store expression
unknown's avatar
unknown committed
3784
  return entry->val_real(&null_value);
unknown's avatar
unknown committed
3785 3786
}

3787
longlong Item_func_set_user_var::val_int()
unknown's avatar
unknown committed
3788
{
3789
  DBUG_ASSERT(fixed == 1);
3790
  check();
3791 3792
  update();					// Store expression
  return entry->val_int(&null_value);
unknown's avatar
unknown committed
3793 3794
}

3795
String *Item_func_set_user_var::val_str(String *str)
unknown's avatar
unknown committed
3796
{
3797
  DBUG_ASSERT(fixed == 1);
3798
  check();
3799 3800
  update();					// Store expression
  return entry->val_str(&null_value, str, decimals);
unknown's avatar
unknown committed
3801 3802 3803
}


unknown's avatar
unknown committed
3804 3805 3806 3807 3808 3809 3810 3811 3812
my_decimal *Item_func_set_user_var::val_decimal(my_decimal *val)
{
  DBUG_ASSERT(fixed == 1);
  check();
  update();					// Store expression
  return entry->val_decimal(&null_value, val);
}


unknown's avatar
unknown committed
3813 3814
void Item_func_set_user_var::print(String *str)
{
3815 3816 3817
  str->append("(@", 2);
  str->append(name.str, name.length);
  str->append(":=", 2);
unknown's avatar
unknown committed
3818 3819 3820 3821 3822
  args[0]->print(str);
  str->append(')');
}


3823 3824 3825 3826 3827 3828 3829 3830 3831 3832
void Item_func_set_user_var::print_as_stmt(String *str)
{
  str->append("set @", 5);
  str->append(name.str, name.length);
  str->append(":=", 2);
  args[0]->print(str);
  str->append(')');
}


unknown's avatar
unknown committed
3833 3834 3835
String *
Item_func_get_user_var::val_str(String *str)
{
3836
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3837
  DBUG_ENTER("Item_func_get_user_var::val_str");
3838
  if (!var_entry)
3839
    DBUG_RETURN((String*) 0);			// No such variable
3840
  DBUG_RETURN(var_entry->val_str(&null_value, str, decimals));
unknown's avatar
unknown committed
3841 3842 3843
}


3844
double Item_func_get_user_var::val_real()
unknown's avatar
unknown committed
3845
{
3846
  DBUG_ASSERT(fixed == 1);
3847 3848
  if (!var_entry)
    return 0.0;					// No such variable
unknown's avatar
unknown committed
3849 3850 3851 3852 3853 3854 3855 3856 3857 3858
  return (var_entry->val_real(&null_value));
}


my_decimal *Item_func_get_user_var::val_decimal(my_decimal *dec)
{
  DBUG_ASSERT(fixed == 1);
  if (!var_entry)
    return 0;
  return var_entry->val_decimal(&null_value, dec);
unknown's avatar
unknown committed
3859 3860 3861 3862 3863
}


longlong Item_func_get_user_var::val_int()
{
3864
  DBUG_ASSERT(fixed == 1);
3865 3866 3867
  if (!var_entry)
    return LL(0);				// No such variable
  return (var_entry->val_int(&null_value));
unknown's avatar
unknown committed
3868 3869 3870
}


3871
/*
3872 3873 3874 3875 3876 3877 3878 3879 3880 3881
  Get variable by name and, if necessary, put the record of variable 
  use into the binary log.
  
  SYNOPSIS
    get_var_with_binlog()
      thd        Current thread
      name       Variable name 
      out_entry  [out] variable structure or NULL. The pointer is set 
                 regardless of whether function succeeded or not.

3882 3883 3884 3885
  When a user variable is invoked from an update query (INSERT, UPDATE etc),
  stores this variable and its value in thd->user_var_events, so that it can be
  written to the binlog (will be written just before the query is written, see
  log.cc).
3886

3887
  RETURN
3888
    0  OK
unknown's avatar
unknown committed
3889
    1  Failed to put appropriate record into binary log
3890

3891 3892
*/

3893 3894
int get_var_with_binlog(THD *thd, enum_sql_command sql_command,
                        LEX_STRING &name, user_var_entry **out_entry)
unknown's avatar
unknown committed
3895
{
3896
  BINLOG_USER_VAR_EVENT *user_var_event;
3897 3898
  user_var_entry *var_entry;
  var_entry= get_variable(&thd->user_vars, name, 0);
3899 3900

  if (!(opt_bin_log && is_update_query(sql_command)))
3901 3902 3903 3904
  {
    *out_entry= var_entry;
    return 0;
  }
3905 3906

  if (!var_entry)
unknown's avatar
unknown committed
3907
  {
3908
    /*
3909 3910 3911 3912
      If the variable does not exist, it's NULL, but we want to create it so
      that it gets into the binlog (if it didn't, the slave could be
      influenced by a variable of the same name previously set by another
      thread).
unknown's avatar
unknown committed
3913 3914
      We create it like if it had been explicitly set with SET before.
      The 'new' mimics what sql_yacc.yy does when 'SET @a=10;'.
3915 3916 3917
      sql_set_variables() is what is called from 'case SQLCOM_SET_OPTION'
      in dispatch_command()). Instead of building a one-element list to pass to
      sql_set_variables(), we could instead manually call check() and update();
3918 3919
      this would save memory and time; but calling sql_set_variables() makes
      one unique place to maintain (sql_set_variables()). 
3920 3921 3922 3923 3924
    */

    List<set_var_base> tmp_var_list;
    tmp_var_list.push_back(new set_var_user(new Item_func_set_user_var(name,
                                                                       new Item_null())));
3925 3926
    /* Create the variable */
    if (sql_set_variables(thd, &tmp_var_list))
3927 3928 3929 3930
      goto err;
    if (!(var_entry= get_variable(&thd->user_vars, name, 0)))
      goto err;
  }
3931 3932
  else if (var_entry->used_query_id == thd->query_id ||
           mysql_bin_log.is_query_in_union(thd, var_entry->used_query_id))
3933 3934 3935 3936 3937 3938 3939 3940 3941
  {
    /* 
       If this variable was already stored in user_var_events by this query
       (because it's used in more than one place in the query), don't store
       it.
    */
    *out_entry= var_entry;
    return 0;
  }
3942 3943 3944 3945

  uint size;
  /*
    First we need to store value of var_entry, when the next situation
unknown's avatar
unknown committed
3946
    appears:
3947 3948
    > set @a:=1;
    > insert into t1 values (@a), (@a:=@a+1), (@a:=@a+1);
3949 3950 3951 3952 3953 3954
    We have to write to binlog value @a= 1.
    
    We allocate the user_var_event on user_var_events_alloc pool, not on
    the this-statement-execution pool because in SPs user_var_event objects 
    may need to be valid after current [SP] statement execution pool is
    destroyed.
3955
  */
3956 3957 3958
  size= ALIGN_SIZE(sizeof(BINLOG_USER_VAR_EVENT)) + var_entry->length;
  if (!(user_var_event= (BINLOG_USER_VAR_EVENT *)
        alloc_root(thd->user_var_events_alloc, size)))
3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970
    goto err;
  
  user_var_event->value= (char*) user_var_event +
    ALIGN_SIZE(sizeof(BINLOG_USER_VAR_EVENT));
  user_var_event->user_var_event= var_entry;
  user_var_event->type= var_entry->type;
  user_var_event->charset_number= var_entry->collation.collation->number;
  if (!var_entry->value)
  {
    /* NULL value*/
    user_var_event->length= 0;
    user_var_event->value= 0;
unknown's avatar
unknown committed
3971
  }
3972 3973 3974 3975 3976 3977 3978 3979 3980 3981
  else
  {
    user_var_event->length= var_entry->length;
    memcpy(user_var_event->value, var_entry->value,
           var_entry->length);
  }
  /* Mark that this variable has been used by this query */
  var_entry->used_query_id= thd->query_id;
  if (insert_dynamic(&thd->user_var_events, (gptr) &user_var_event))
    goto err;
3982 3983 3984
  
  *out_entry= var_entry;
  return 0;
3985

unknown's avatar
unknown committed
3986
err:
3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998
  *out_entry= var_entry;
  return 1;
}


void Item_func_get_user_var::fix_length_and_dec()
{
  THD *thd=current_thd;
  int error;
  maybe_null=1;
  decimals=NOT_FIXED_DEC;
  max_length=MAX_BLOB_WIDTH;
3999

4000
  error= get_var_with_binlog(thd, thd->lex->sql_command, name, &var_entry);
4001 4002

  if (var_entry)
4003
  {
4004
    collation.set(var_entry->collation);
4005 4006 4007
    switch (var_entry->type) {
    case REAL_RESULT:
      max_length= DBL_DIG + 8;
unknown's avatar
unknown committed
4008
      break;
4009 4010
    case INT_RESULT:
      max_length= MAX_BIGINT_WIDTH;
unknown's avatar
unknown committed
4011
      decimals=0;
4012 4013 4014 4015
      break;
    case STRING_RESULT:
      max_length= MAX_BLOB_WIDTH;
      break;
unknown's avatar
unknown committed
4016
    case DECIMAL_RESULT:
unknown's avatar
unknown committed
4017 4018
      max_length= DECIMAL_MAX_STR_LENGTH;
      decimals= DECIMAL_MAX_SCALE;
unknown's avatar
unknown committed
4019
      break;
unknown's avatar
unknown committed
4020
    case ROW_RESULT:                            // Keep compiler happy
unknown's avatar
unknown committed
4021 4022
    default:
      DBUG_ASSERT(0);
unknown's avatar
unknown committed
4023
      break;
4024 4025
    }
  }
4026
  else
4027 4028
  {
    collation.set(&my_charset_bin, DERIVATION_IMPLICIT);
4029
    null_value= 1;
4030
  }
4031

4032 4033
  if (error)
    thd->fatal_error();
4034

unknown's avatar
unknown committed
4035
  return;
unknown's avatar
unknown committed
4036 4037 4038
}


4039
bool Item_func_get_user_var::const_item() const
4040
{
4041
  return (!var_entry || current_thd->query_id != var_entry->update_query_id);
4042
}
unknown's avatar
unknown committed
4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054


enum Item_result Item_func_get_user_var::result_type() const
{
  user_var_entry *entry;
  if (!(entry = (user_var_entry*) hash_search(&current_thd->user_vars,
					      (byte*) name.str,
					      name.length)))
    return STRING_RESULT;
  return entry->type;
}

unknown's avatar
unknown committed
4055 4056 4057

void Item_func_get_user_var::print(String *str)
{
4058
  str->append("(@", 2);
unknown's avatar
unknown committed
4059 4060 4061 4062
  str->append(name.str,name.length);
  str->append(')');
}

4063

4064
bool Item_func_get_user_var::eq(const Item *item, bool binary_cmp) const
unknown's avatar
unknown committed
4065 4066 4067 4068 4069 4070
{
  /* Assume we don't have rtti */
  if (this == item)
    return 1;					// Same item is same.
  /* Check if other type is also a get_user_var() object */
  if (item->type() != FUNC_ITEM ||
4071
      ((Item_func*) item)->functype() != functype())
unknown's avatar
unknown committed
4072 4073 4074 4075 4076 4077 4078
    return 0;
  Item_func_get_user_var *other=(Item_func_get_user_var*) item;
  return (name.length == other->name.length &&
	  !memcmp(name.str, other->name.str, name.length));
}


4079
bool Item_user_var_as_out_param::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
4080 4081
{
  DBUG_ASSERT(fixed == 0);
4082
  if (Item::fix_fields(thd, ref) ||
unknown's avatar
unknown committed
4083 4084 4085 4086 4087 4088 4089 4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148
      !(entry= get_variable(&thd->user_vars, name, 1)))
    return TRUE;
  entry->type= STRING_RESULT;
  /*
    Let us set the same collation which is used for loading
    of fields in LOAD DATA INFILE.
    (Since Item_user_var_as_out_param is used only there).
  */
  entry->collation.set(thd->variables.collation_database);
  entry->update_query_id= thd->query_id;
  return FALSE;
}


void Item_user_var_as_out_param::set_null_value(CHARSET_INFO* cs)
{
  if (::update_hash(entry, TRUE, 0, 0, STRING_RESULT, cs,
                    DERIVATION_IMPLICIT))
    current_thd->fatal_error();			// Probably end of memory
}


void Item_user_var_as_out_param::set_value(const char *str, uint length,
                                           CHARSET_INFO* cs)
{
  if (::update_hash(entry, FALSE, (void*)str, length, STRING_RESULT, cs,
                    DERIVATION_IMPLICIT))
    current_thd->fatal_error();			// Probably end of memory
}


double Item_user_var_as_out_param::val_real()
{
  DBUG_ASSERT(0);
  return 0.0;
}


longlong Item_user_var_as_out_param::val_int()
{
  DBUG_ASSERT(0);
  return 0;
}


String* Item_user_var_as_out_param::val_str(String *str)
{
  DBUG_ASSERT(0);
  return 0;
}


my_decimal* Item_user_var_as_out_param::val_decimal(my_decimal *decimal_buffer)
{
  DBUG_ASSERT(0);
  return 0;
}


void Item_user_var_as_out_param::print(String *str)
{
  str->append('@');
  str->append(name.str,name.length);
}


4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159 4160
Item_func_get_system_var::
Item_func_get_system_var(sys_var *var_arg, enum_var_type var_type_arg,
                       LEX_STRING *component_arg, const char *name_arg,
                       size_t name_len_arg)
  :var(var_arg), var_type(var_type_arg), component(*component_arg)
{
  /* set_name() will allocate the name */
  set_name(name_arg, name_len_arg, system_charset_info);
}


bool
unknown's avatar
unknown committed
4161
Item_func_get_system_var::fix_fields(THD *thd, Item **ref)
4162
{
4163
  Item *item;
4164
  DBUG_ENTER("Item_func_get_system_var::fix_fields");
4165

4166 4167 4168 4169 4170
  /*
    Evaluate the system variable and substitute the result (a basic constant)
    instead of this item. If the variable can not be evaluated,
    the error is reported in sys_var::item().
  */
4171
  if (!(item= var->item(thd, var_type, &component)))
4172 4173 4174 4175 4176 4177 4178 4179
    DBUG_RETURN(1);                             // Impossible
  item->set_name(name, 0, system_charset_info); // don't allocate a new name
  thd->change_item_tree(ref, item);

  DBUG_RETURN(0);
}


unknown's avatar
unknown committed
4180 4181
longlong Item_func_inet_aton::val_int()
{
4182
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4183 4184 4185 4186 4187
  uint byte_result = 0;
  ulonglong result = 0;			// We are ready for 64 bit addresses
  const char *p,* end;
  char c = '.'; // we mark c to indicate invalid IP in case length is 0
  char buff[36];
4188
  int dot_count= 0;
unknown's avatar
unknown committed
4189

unknown's avatar
unknown committed
4190
  String *s,tmp(buff,sizeof(buff),&my_charset_bin);
unknown's avatar
unknown committed
4191 4192 4193 4194 4195 4196 4197 4198 4199 4200 4201 4202 4203 4204 4205 4206
  if (!(s = args[0]->val_str(&tmp)))		// If null value
    goto err;
  null_value=0;

  end= (p = s->ptr()) + s->length();
  while (p < end)
  {
    c = *p++;
    int digit = (int) (c - '0');		// Assume ascii
    if (digit >= 0 && digit <= 9)
    {
      if ((byte_result = byte_result * 10 + digit) > 255)
	goto err;				// Wrong address
    }
    else if (c == '.')
    {
4207
      dot_count++;
unknown's avatar
unknown committed
4208 4209 4210 4211 4212 4213 4214
      result= (result << 8) + (ulonglong) byte_result;
      byte_result = 0;
    }
    else
      goto err;					// Invalid character
  }
  if (c != '.')					// IP number can't end on '.'
4215
  {
4216 4217 4218 4219 4220 4221
    /*
      Handle short-forms addresses according to standard. Examples:
      127		-> 0.0.0.127
      127.1		-> 127.0.0.1
      127.2.1		-> 127.2.0.1
    */
unknown's avatar
unknown committed
4222
    switch (dot_count) {
4223 4224
    case 1: result<<= 8; /* Fall through */
    case 2: result<<= 8; /* Fall through */
4225
    }
unknown's avatar
unknown committed
4226
    return (result << 8) + (ulonglong) byte_result;
4227
  }
unknown's avatar
unknown committed
4228 4229 4230 4231 4232 4233

err:
  null_value=1;
  return 0;
}

unknown's avatar
unknown committed
4234

4235
void Item_func_match::init_search(bool no_order)
4236
{
unknown's avatar
unknown committed
4237
  DBUG_ENTER("Item_func_match::init_search");
4238 4239

  /* Check if init_search() has been called before */
4240
  if (ft_handler)
unknown's avatar
unknown committed
4241
    DBUG_VOID_RETURN;
4242

4243
  if (key == NO_SUCH_KEY)
4244 4245
  {
    List<Item> fields;
4246
    fields.push_back(new Item_string(" ",1, cmp_collation.collation));
4247 4248
    for (uint i=1; i < arg_count; i++)
      fields.push_back(args[i]);
4249
    concat=new Item_func_concat_ws(fields);
unknown's avatar
unknown committed
4250 4251 4252
    /*
      Above function used only to get value and do not need fix_fields for it:
      Item_string - basic constant
unknown's avatar
unknown committed
4253 4254
      fields - fix_fields() was already called for this arguments
      Item_func_concat_ws - do not need fix_fields() to produce value
unknown's avatar
unknown committed
4255 4256
    */
    concat->quick_fix_field();
4257
  }
4258

4259 4260
  if (master)
  {
4261 4262
    join_key=master->join_key=join_key|master->join_key;
    master->init_search(no_order);
4263 4264
    ft_handler=master->ft_handler;
    join_key=master->join_key;
unknown's avatar
unknown committed
4265
    DBUG_VOID_RETURN;
4266 4267
  }

unknown's avatar
unknown committed
4268
  String *ft_tmp= 0;
4269

4270
  // MATCH ... AGAINST (NULL) is meaningless, but possible
4271
  if (!(ft_tmp=key_item()->val_str(&value)))
4272
  {
4273 4274
    ft_tmp= &value;
    value.set("",0,cmp_collation.collation);
4275 4276
  }

4277 4278
  if (ft_tmp->charset() != cmp_collation.collation)
  {
4279
    uint dummy_errors;
4280
    search_value.copy(ft_tmp->ptr(), ft_tmp->length(), ft_tmp->charset(),
4281
                      cmp_collation.collation, &dummy_errors);
4282
    ft_tmp= &search_value;
4283 4284
  }

4285 4286
  if (join_key && !no_order)
    flags|=FT_SORTED;
4287
  ft_handler=table->file->ft_init_ext(flags, key, ft_tmp);
4288 4289 4290

  if (join_key)
    table->file->ft_handler=ft_handler;
unknown's avatar
unknown committed
4291 4292

  DBUG_VOID_RETURN;
4293 4294
}

unknown's avatar
unknown committed
4295

4296
bool Item_func_match::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
4297
{
4298
  DBUG_ASSERT(fixed == 0);
unknown's avatar
unknown committed
4299
  Item *item;
4300
  LINT_INIT(item);				// Safe as arg_count is > 1
unknown's avatar
unknown committed
4301

4302 4303 4304
  maybe_null=1;
  join_key=0;

4305 4306 4307 4308 4309
  /*
    const_item is assumed in quite a bit of places, so it would be difficult
    to remove;  If it would ever to be removed, this should include
    modifications to find_best and auto_close as complement to auto_init code
    above.
4310
   */
4311
  if (Item_func::fix_fields(thd, ref) ||
4312
      !args[0]->const_during_execution())
4313 4314
  {
    my_error(ER_WRONG_ARGUMENTS,MYF(0),"AGAINST");
unknown's avatar
unknown committed
4315
    return TRUE;
4316
  }
unknown's avatar
unknown committed
4317

4318 4319
  const_item_cache=0;
  for (uint i=1 ; i < arg_count ; i++)
unknown's avatar
unknown committed
4320
  {
4321
    item=args[i];
unknown's avatar
unknown committed
4322
    if (item->type() == Item::REF_ITEM)
4323 4324
      args[i]= item= *((Item_ref *)item)->ref;
    if (item->type() != Item::FIELD_ITEM)
4325
      key=NO_SUCH_KEY;
unknown's avatar
unknown committed
4326
  }
4327 4328
  /*
    Check that all columns come from the same table.
4329
    We've already checked that columns in MATCH are fields so
4330 4331 4332
    PARAM_TABLE_BIT can only appear from AGAINST argument.
  */
  if ((used_tables_cache & ~PARAM_TABLE_BIT) != item->used_tables())
4333
    key=NO_SUCH_KEY;
4334

4335
  if (key == NO_SUCH_KEY && !(flags & FT_BOOL))
4336 4337
  {
    my_error(ER_WRONG_ARGUMENTS,MYF(0),"MATCH");
unknown's avatar
unknown committed
4338
    return TRUE;
4339
  }
4340
  table=((Item_field *)item)->field->table;
unknown's avatar
unknown committed
4341
  if (!(table->file->table_flags() & HA_CAN_FULLTEXT))
4342
  {
unknown's avatar
unknown committed
4343
    my_error(ER_TABLE_CANT_HANDLE_FT, MYF(0));
unknown's avatar
unknown committed
4344
    return 1;
4345
  }
4346 4347
  table->fulltext_searched=1;
  return agg_arg_collations_for_comparison(cmp_collation, args+1, arg_count-1);
unknown's avatar
unknown committed
4348
}
4349

unknown's avatar
unknown committed
4350 4351 4352
bool Item_func_match::fix_index()
{
  Item_field *item;
4353
  uint ft_to_key[MAX_KEY], ft_cnt[MAX_KEY], fts=0, keynr;
4354
  uint max_cnt=0, mkeys=0, i;
4355

4356
  if (key == NO_SUCH_KEY)
4357
    return 0;
4358 4359 4360
  
  if (!table) 
    goto err;
unknown's avatar
unknown committed
4361

4362
  for (keynr=0 ; keynr < table->s->keys ; keynr++)
unknown's avatar
unknown committed
4363
  {
4364
    if ((table->key_info[keynr].flags & HA_FULLTEXT) &&
4365
        (table->keys_in_use_for_query.is_set(keynr)))
unknown's avatar
unknown committed
4366
    {
4367
      ft_to_key[fts]=keynr;
unknown's avatar
unknown committed
4368 4369 4370 4371 4372 4373
      ft_cnt[fts]=0;
      fts++;
    }
  }

  if (!fts)
4374
    goto err;
unknown's avatar
unknown committed
4375

4376
  for (i=1; i < arg_count; i++)
unknown's avatar
unknown committed
4377
  {
4378
    item=(Item_field*)args[i];
4379
    for (keynr=0 ; keynr < fts ; keynr++)
unknown's avatar
unknown committed
4380
    {
4381
      KEY *ft_key=&table->key_info[ft_to_key[keynr]];
unknown's avatar
unknown committed
4382 4383 4384 4385 4386
      uint key_parts=ft_key->key_parts;

      for (uint part=0 ; part < key_parts ; part++)
      {
	if (item->field->eq(ft_key->key_part[part].field))
4387
	  ft_cnt[keynr]++;
unknown's avatar
unknown committed
4388 4389 4390 4391
      }
    }
  }

4392
  for (keynr=0 ; keynr < fts ; keynr++)
unknown's avatar
unknown committed
4393
  {
4394
    if (ft_cnt[keynr] > max_cnt)
unknown's avatar
unknown committed
4395
    {
4396
      mkeys=0;
4397 4398
      max_cnt=ft_cnt[mkeys]=ft_cnt[keynr];
      ft_to_key[mkeys]=ft_to_key[keynr];
4399 4400
      continue;
    }
4401
    if (max_cnt && ft_cnt[keynr] == max_cnt)
4402 4403
    {
      mkeys++;
4404 4405
      ft_cnt[mkeys]=ft_cnt[keynr];
      ft_to_key[mkeys]=ft_to_key[keynr];
4406
      continue;
unknown's avatar
unknown committed
4407 4408 4409
    }
  }

4410
  for (keynr=0 ; keynr <= mkeys ; keynr++)
unknown's avatar
unknown committed
4411
  {
4412 4413
    // partial keys doesn't work
    if (max_cnt < arg_count-1 ||
4414
        max_cnt < table->key_info[ft_to_key[keynr]].key_parts)
4415
      continue;
unknown's avatar
unknown committed
4416

4417
    key=ft_to_key[keynr];
4418

4419 4420 4421
    return 0;
  }

4422
err:
4423
  if (flags & FT_BOOL)
4424
  {
4425
    key=NO_SUCH_KEY;
4426 4427
    return 0;
  }
unknown's avatar
unknown committed
4428 4429
  my_message(ER_FT_MATCHING_KEY_NOT_FOUND,
             ER(ER_FT_MATCHING_KEY_NOT_FOUND), MYF(0));
4430
  return 1;
4431 4432
}

4433

4434
bool Item_func_match::eq(const Item *item, bool binary_cmp) const
4435
{
unknown's avatar
unknown committed
4436 4437
  if (item->type() != FUNC_ITEM ||
      ((Item_func*)item)->functype() != FT_FUNC ||
4438
      flags != ((Item_func_match*)item)->flags)
4439 4440 4441 4442 4443
    return 0;

  Item_func_match *ifm=(Item_func_match*) item;

  if (key == ifm->key && table == ifm->table &&
4444
      key_item()->eq(ifm->key_item(), binary_cmp))
4445
    return 1;
unknown's avatar
unknown committed
4446 4447 4448 4449

  return 0;
}

4450

4451
double Item_func_match::val_real()
unknown's avatar
unknown committed
4452
{
4453
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4454
  DBUG_ENTER("Item_func_match::val");
unknown's avatar
unknown committed
4455
  if (ft_handler == NULL)
unknown's avatar
unknown committed
4456
    DBUG_RETURN(-1.0);
unknown's avatar
unknown committed
4457

unknown's avatar
unknown committed
4458 4459 4460
  if (table->null_row) /* NULL row from an outer join */
    return 0.0;

unknown's avatar
unknown committed
4461 4462 4463
  if (join_key)
  {
    if (table->file->ft_handler)
unknown's avatar
unknown committed
4464
      DBUG_RETURN(ft_handler->please->get_relevance(ft_handler));
unknown's avatar
unknown committed
4465 4466 4467
    join_key=0;
  }

4468
  if (key == NO_SUCH_KEY)
unknown's avatar
unknown committed
4469
  {
4470 4471
    String *a= concat->val_str(&value);
    if ((null_value= (a == 0)))
unknown's avatar
unknown committed
4472 4473 4474
      DBUG_RETURN(0);
    DBUG_RETURN(ft_handler->please->find_relevance(ft_handler,
				      (byte *)a->ptr(), a->length()));
unknown's avatar
unknown committed
4475 4476
  }
  else
4477 4478
    DBUG_RETURN(ft_handler->please->find_relevance(ft_handler,
                                                   table->record[0], 0));
unknown's avatar
unknown committed
4479 4480
}

4481 4482
void Item_func_match::print(String *str)
{
4483
  str->append("(match ", 7);
unknown's avatar
unknown committed
4484
  print_args(str, 1);
4485
  str->append(" against (", 10);
4486
  args[0]->print(str);
unknown's avatar
unknown committed
4487
  if (flags & FT_BOOL)
4488
    str->append(" in boolean mode", 16);
unknown's avatar
unknown committed
4489 4490
  else if (flags & FT_EXPAND)
    str->append(" with query expansion", 21);
4491
  str->append("))", 2);
4492
}
unknown's avatar
unknown committed
4493

unknown's avatar
unknown committed
4494 4495
longlong Item_func_bit_xor::val_int()
{
4496
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4497 4498
  ulonglong arg1= (ulonglong) args[0]->val_int();
  ulonglong arg2= (ulonglong) args[1]->val_int();
4499
  if ((null_value= (args[0]->null_value || args[1]->null_value)))
unknown's avatar
unknown committed
4500 4501 4502 4503
    return 0;
  return (longlong) (arg1 ^ arg2);
}

4504

4505 4506 4507 4508
/***************************************************************************
  System variables
****************************************************************************/

4509 4510 4511 4512 4513 4514 4515 4516 4517 4518 4519 4520 4521 4522 4523 4524 4525
/*
  Return value of an system variable base[.name] as a constant item

  SYNOPSIS
    get_system_var()
    thd			Thread handler
    var_type		global / session
    name		Name of base or system variable
    component		Component.

  NOTES
    If component.str = 0 then the variable name is in 'name'

  RETURN
    0	error
    #	constant item
*/
4526

4527 4528 4529

Item *get_system_var(THD *thd, enum_var_type var_type, LEX_STRING name,
		     LEX_STRING component)
4530
{
4531 4532 4533
  sys_var *var;
  LEX_STRING *base_name, *component_name;

4534 4535
  if (component.str == 0 &&
      !my_strcasecmp(system_charset_info, name.str, "VERSION"))
4536
    return new Item_string(NULL, server_version,
4537
			   (uint) strlen(server_version),
unknown's avatar
unknown committed
4538
			   system_charset_info, DERIVATION_SYSCONST);
unknown's avatar
unknown committed
4539

4540 4541 4542 4543 4544 4545 4546 4547 4548 4549
  if (component.str)
  {
    base_name= &component;
    component_name= &name;
  }
  else
  {
    base_name= &name;
    component_name= &component;			// Empty string
  }
unknown's avatar
unknown committed
4550

4551
  if (!(var= find_sys_var(base_name->str, base_name->length)))
unknown's avatar
unknown committed
4552
    return 0;
4553 4554 4555 4556
  if (component.str)
  {
    if (!var->is_struct())
    {
4557
      my_error(ER_VARIABLE_IS_NOT_STRUCT, MYF(0), base_name->str);
4558 4559 4560
      return 0;
    }
  }
unknown's avatar
unknown committed
4561
  thd->lex->uncacheable(UNCACHEABLE_SIDEEFFECT);
4562

4563 4564
  set_if_smaller(component_name->length, MAX_SYS_VAR_LENGTH);

4565
  return new Item_func_get_system_var(var, var_type, component_name,
4566
                                      NULL, 0);
4567 4568 4569
}


unknown's avatar
unknown committed
4570 4571
/*
  Check a user level lock.
4572 4573 4574 4575 4576 4577 4578 4579

  SYNOPSIS:
    val_int()

  RETURN VALUES
    1		Available
    0		Already taken
    NULL	Error
unknown's avatar
unknown committed
4580 4581
*/

4582
longlong Item_func_is_free_lock::val_int()
unknown's avatar
unknown committed
4583
{
4584
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4585
  String *res=args[0]->val_str(&value);
4586
  User_level_lock *ull;
unknown's avatar
unknown committed
4587 4588

  null_value=0;
4589
  if (!res || !res->length())
unknown's avatar
unknown committed
4590 4591 4592 4593 4594 4595
  {
    null_value=1;
    return 0;
  }
  
  pthread_mutex_lock(&LOCK_user_locks);
4596
  ull= (User_level_lock *) hash_search(&hash_user_locks, (byte*) res->ptr(),
unknown's avatar
unknown committed
4597 4598 4599 4600 4601 4602
			  res->length());
  pthread_mutex_unlock(&LOCK_user_locks);
  if (!ull || !ull->locked)
    return 1;
  return 0;
}
unknown's avatar
unknown committed
4603

unknown's avatar
SCRUM  
unknown committed
4604 4605
longlong Item_func_is_used_lock::val_int()
{
4606
  DBUG_ASSERT(fixed == 1);
unknown's avatar
SCRUM  
unknown committed
4607
  String *res=args[0]->val_str(&value);
4608
  User_level_lock *ull;
unknown's avatar
SCRUM  
unknown committed
4609 4610 4611 4612 4613 4614

  null_value=1;
  if (!res || !res->length())
    return 0;
  
  pthread_mutex_lock(&LOCK_user_locks);
4615
  ull= (User_level_lock *) hash_search(&hash_user_locks, (byte*) res->ptr(),
unknown's avatar
SCRUM  
unknown committed
4616 4617 4618 4619 4620 4621 4622 4623 4624
			  res->length());
  pthread_mutex_unlock(&LOCK_user_locks);
  if (!ull || !ull->locked)
    return 0;

  null_value=0;
  return ull->thread_id;
}

4625

4626 4627 4628 4629 4630 4631 4632 4633 4634
longlong Item_func_row_count::val_int()
{
  DBUG_ASSERT(fixed == 1);
  THD *thd= current_thd;

  return thd->row_count_func;
}


4635 4636 4637
Item_func_sp::Item_func_sp(Name_resolution_context *context_arg, sp_name *name)
  :Item_func(), context(context_arg), m_name(name), m_sp(NULL),
   result_field(NULL)
4638
{
unknown's avatar
unknown committed
4639
  maybe_null= 1;
4640
  m_name->init_qname(current_thd);
unknown's avatar
unknown committed
4641
  dummy_table= (TABLE*) sql_calloc(sizeof(TABLE));
4642 4643
}

unknown's avatar
unknown committed
4644

4645 4646 4647 4648
Item_func_sp::Item_func_sp(Name_resolution_context *context_arg,
                           sp_name *name, List<Item> &list)
  :Item_func(list), context(context_arg), m_name(name), m_sp(NULL),
   result_field(NULL)
4649
{
unknown's avatar
unknown committed
4650
  maybe_null= 1;
4651
  m_name->init_qname(current_thd);
unknown's avatar
unknown committed
4652
  dummy_table= (TABLE*) sql_calloc(sizeof(TABLE));
4653 4654
}

4655 4656 4657 4658 4659 4660 4661 4662
void
Item_func_sp::cleanup()
{
  if (result_field)
  {
    delete result_field;
    result_field= NULL;
  }
4663
  m_sp= NULL;
4664 4665
  Item_func::cleanup();
}
unknown's avatar
unknown committed
4666

4667 4668 4669
const char *
Item_func_sp::func_name() const
{
4670
  THD *thd= current_thd;
unknown's avatar
unknown committed
4671
  /* Calculate length to avoid reallocation of string for sure */
4672 4673 4674 4675
  uint len= ((m_name->m_db.length +
              m_name->m_name.length)*2 + //characters*quoting
             2 +                         // ` and `
             1 +                         // .
4676 4677
             1 +                         // end of string
             ALIGN_SIZE(1));             // to avoid String reallocation
unknown's avatar
unknown committed
4678
  String qname((char *)alloc_root(thd->mem_root, len), len,
4679
               system_charset_info);
4680

4681 4682 4683 4684 4685
  qname.length(0);
  append_identifier(thd, &qname, m_name->m_db.str, m_name->m_db.length);
  qname.append('.');
  append_identifier(thd, &qname, m_name->m_name.str, m_name->m_name.length);
  return qname.ptr();
4686 4687
}

4688

unknown's avatar
unknown committed
4689 4690 4691
Field *
Item_func_sp::sp_result_field(void) const
{
unknown's avatar
unknown committed
4692
  Field *field;
unknown's avatar
unknown committed
4693
  DBUG_ENTER("Item_func_sp::sp_result_field");
unknown's avatar
unknown committed
4694 4695

  if (!m_sp)
unknown's avatar
unknown committed
4696
  {
unknown's avatar
unknown committed
4697
    if (!(m_sp= sp_find_function(current_thd, m_name, TRUE)))
unknown's avatar
unknown committed
4698
    {
unknown's avatar
unknown committed
4699 4700
      my_error(ER_SP_DOES_NOT_EXIST, MYF(0), "FUNCTION", m_name->m_qname.str);
      DBUG_RETURN(0);
unknown's avatar
unknown committed
4701 4702
    }
  }
unknown's avatar
unknown committed
4703 4704 4705 4706 4707 4708 4709 4710 4711 4712 4713
  if (!dummy_table->s)
  {
    char *empty_name= (char *) "";
    TABLE_SHARE *share;
    dummy_table->s= share= &dummy_table->share_not_to_be_used;      
    dummy_table->alias = empty_name;
    dummy_table->maybe_null = maybe_null;
    dummy_table->in_use= current_thd;
    share->table_cache_key = empty_name;
    share->table_name = empty_name;
  }
4714 4715
  field= m_sp->make_field(max_length, name, dummy_table);
  DBUG_RETURN(field);
unknown's avatar
unknown committed
4716 4717 4718
}


unknown's avatar
unknown committed
4719 4720 4721 4722 4723 4724 4725 4726
/*
  Execute function & store value in field

  RETURN
   0  value <> NULL
   1  value =  NULL  or error
*/

4727 4728 4729 4730 4731 4732 4733 4734
int
Item_func_sp::execute(Field **flp)
{
  Item *it;
  Field *f;
  if (execute(&it))
  {
    null_value= 1;
4735
    context->process_error(current_thd);
4736 4737 4738 4739 4740 4741 4742 4743 4744 4745 4746
    return 1;
  }
  if (!(f= *flp))
  {
    *flp= f= sp_result_field();
    f->move_field((f->pack_length() > sizeof(result_buf)) ? 
                  sql_alloc(f->pack_length()) : result_buf);
    f->null_ptr= (uchar *)&null_value;
    f->null_bit= 1;
  }
  it->save_in_field(f, 1);
unknown's avatar
unknown committed
4747
  return null_value= f->is_null();
4748 4749 4750
}


4751 4752 4753 4754 4755
int
Item_func_sp::execute(Item **itp)
{
  DBUG_ENTER("Item_func_sp::execute");
  THD *thd= current_thd;
4756
  int res= -1;
4757
  Sub_statement_state statement_state;
4758
  Security_context *save_security_ctx= 0, *save_ctx_func;
4759

4760 4761 4762 4763 4764 4765 4766 4767 4768
#ifndef NO_EMBEDDED_ACCESS_CHECKS
  if (context->security_ctx)
  {
    /* Set view definer security context */
    save_security_ctx= thd->security_ctx;
    thd->security_ctx= context->security_ctx;
  }
#endif
  if (find_and_check_access(thd, EXECUTE_ACL, &save_ctx_func))
4769
    goto error;
4770

4771 4772 4773 4774 4775
  /*
    Disable the binlogging if this is not a SELECT statement. If this is a
    SELECT, leave binlogging on, so execute_function() code writes the
    function call into binlog.
  */
4776
  thd->reset_sub_statement_state(&statement_state, SUB_STMT_FUNCTION);
4777
  res= m_sp->execute_function(thd, args, arg_count, itp);
4778
  thd->restore_sub_statement_state(&statement_state);
unknown's avatar
unknown committed
4779

4780 4781 4782 4783 4784 4785
  if (res && mysql_bin_log.is_open() &&
      (m_sp->m_chistics->daccess == SP_CONTAINS_SQL ||
       m_sp->m_chistics->daccess == SP_MODIFIES_SQL_DATA))
    push_warning(thd, MYSQL_ERROR::WARN_LEVEL_WARN,
                 ER_FAILED_ROUTINE_BREAK_BINLOG,
		 ER(ER_FAILED_ROUTINE_BREAK_BINLOG));
4786 4787
#ifndef NO_EMBEDDED_ACCESS_CHECKS
  sp_restore_security_context(thd, save_ctx_func);
4788
error:
4789 4790 4791 4792 4793
  if (save_security_ctx)
    thd->security_ctx= save_security_ctx;
#else
error:
#endif
4794
  DBUG_RETURN(res);
4795 4796
}

4797

unknown's avatar
unknown committed
4798 4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812 4813
void
Item_func_sp::make_field(Send_field *tmp_field)
{
  Field *field;
  DBUG_ENTER("Item_func_sp::make_field");
  if ((field= sp_result_field()))
  {
    field->make_field(tmp_field);
    delete field;
    DBUG_VOID_RETURN;
  }
  init_make_field(tmp_field, MYSQL_TYPE_VARCHAR);  
  DBUG_VOID_RETURN;
}


4814 4815
enum enum_field_types
Item_func_sp::field_type() const
4816
{
unknown's avatar
unknown committed
4817
  Field *field;
4818
  DBUG_ENTER("Item_func_sp::field_type");
4819

4820 4821
  if (result_field)
    DBUG_RETURN(result_field->type());
unknown's avatar
unknown committed
4822
  if ((field= sp_result_field()))
4823
  {
unknown's avatar
unknown committed
4824 4825 4826
    enum_field_types result= field->type();
    delete field;
    DBUG_RETURN(result);
4827
  }
4828
  DBUG_RETURN(MYSQL_TYPE_VARCHAR);
4829 4830
}

4831

4832 4833
Item_result
Item_func_sp::result_type() const
4834
{
unknown's avatar
unknown committed
4835
  Field *field;
4836 4837
  DBUG_ENTER("Item_func_sp::result_type");
  DBUG_PRINT("info", ("m_sp = %p", m_sp));
4838

4839 4840
  if (result_field)
    DBUG_RETURN(result_field->result_type());
unknown's avatar
unknown committed
4841
  if ((field= sp_result_field()))
4842
  {
unknown's avatar
unknown committed
4843 4844 4845
    Item_result result= field->result_type();
    delete field;
    DBUG_RETURN(result);
4846
  }
4847
  DBUG_RETURN(STRING_RESULT);
4848 4849 4850 4851 4852
}

void
Item_func_sp::fix_length_and_dec()
{
unknown's avatar
unknown committed
4853
  Field *field;
4854 4855
  DBUG_ENTER("Item_func_sp::fix_length_and_dec");

4856 4857 4858
  if (result_field)
  {
    decimals= result_field->decimals();
unknown's avatar
unknown committed
4859
    max_length= result_field->field_length;
4860 4861 4862
    DBUG_VOID_RETURN;
  }

unknown's avatar
unknown committed
4863
  if (!(field= sp_result_field()))
4864 4865
  {
    context->process_error(current_thd);
unknown's avatar
unknown committed
4866
    DBUG_VOID_RETURN;
4867
  }
unknown's avatar
unknown committed
4868
  decimals= field->decimals();
unknown's avatar
merge  
unknown committed
4869 4870
  max_length= field->field_length;
  maybe_null= 1;
unknown's avatar
unknown committed
4871
  delete field;
4872 4873
  DBUG_VOID_RETURN;
}
unknown's avatar
unknown committed
4874 4875


4876 4877 4878
longlong Item_func_found_rows::val_int()
{
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4879
  return current_thd->found_rows();
4880
}
unknown's avatar
unknown committed
4881

unknown's avatar
unknown committed
4882

unknown's avatar
unknown committed
4883 4884 4885 4886 4887 4888 4889 4890 4891 4892 4893 4894 4895 4896
Field *
Item_func_sp::tmp_table_field(TABLE *t_arg)
{
  Field *res= 0;
  DBUG_ENTER("Item_func_sp::tmp_table_field");

  if (m_sp)
    res= m_sp->make_field(max_length, (const char *)name, t_arg);
  
  if (!res) 
    res= Item_func::tmp_table_field(t_arg);

  DBUG_RETURN(res);
}
4897

unknown's avatar
unknown committed
4898

4899
/*
unknown's avatar
unknown committed
4900
  Find the function and chack access rigths to the function
4901 4902

  SYNOPSIS
unknown's avatar
unknown committed
4903 4904
    find_and_check_access()
    thd           thread handler
4905
    want_access   requested access
unknown's avatar
unknown committed
4906
    save          backup of security context
4907 4908

  RETURN
unknown's avatar
unknown committed
4909 4910
    FALSE    Access granted
    TRUE     Requested access can't be granted or function doesn't exists
unknown's avatar
unknown committed
4911
	     In this case security context is not changed and *save = 0
4912 4913 4914 4915 4916

  NOTES
    Checks if requested access to function can be granted to user.
    If function isn't found yet, it searches function first.
    If function can't be found or user don't have requested access
unknown's avatar
unknown committed
4917
    error is raised.
4918 4919
    If security context sp_ctx is provided and access can be granted then
    switch back to previous context isn't performed.
unknown's avatar
unknown committed
4920 4921
    In case of access error or if context is not provided then
    find_and_check_access() switches back to previous security context.
4922
*/
unknown's avatar
unknown committed
4923

4924
bool
unknown's avatar
unknown committed
4925
Item_func_sp::find_and_check_access(THD *thd, ulong want_access,
unknown's avatar
unknown committed
4926
                                    Security_context **save)
4927
{
unknown's avatar
unknown committed
4928 4929 4930
  bool res= TRUE;

  *save= 0;                                     // Safety if error
4931 4932 4933 4934 4935 4936
  if (! m_sp && ! (m_sp= sp_find_function(thd, m_name, TRUE)))
  {
    my_error(ER_SP_DOES_NOT_EXIST, MYF(0), "FUNCTION", m_name->m_qname.str);
    goto error;
  }

unknown's avatar
unknown committed
4937
#ifndef NO_EMBEDDED_ACCESS_CHECKS
4938
  if (check_routine_access(thd, want_access,
unknown's avatar
unknown committed
4939
			   m_sp->m_db.str, m_sp->m_name.str, 0, FALSE))
4940 4941
    goto error;

unknown's avatar
unknown committed
4942
  sp_change_security_context(thd, m_sp, save);
unknown's avatar
unknown committed
4943 4944 4945 4946 4947 4948 4949 4950 4951 4952
  /*
    If we changed context to run as another user, we need to check the
    access right for the new context again as someone may have deleted
    this person the right to use the procedure

    TODO:
      Cache if the definer has the right to use the object on the first
      usage and only reset the cache if someone does a GRANT statement
      that 'may' affect this.
  */
unknown's avatar
unknown committed
4953
  if (*save &&
4954
      check_routine_access(thd, want_access,
unknown's avatar
unknown committed
4955
			   m_sp->m_db.str, m_sp->m_name.str, 0, FALSE))
4956
  {
unknown's avatar
unknown committed
4957 4958 4959
    sp_restore_security_context(thd, *save);
    *save= 0;                                   // Safety
    goto error;
4960 4961
  }
#endif
unknown's avatar
unknown committed
4962 4963 4964
  res= FALSE;                                   // no error

error:
4965
  return res;
unknown's avatar
unknown committed
4966
}
4967 4968 4969 4970 4971 4972 4973

bool
Item_func_sp::fix_fields(THD *thd, Item **ref)
{
  bool res;
  DBUG_ASSERT(fixed == 0);
  res= Item_func::fix_fields(thd, ref);
4974
  if (!res && thd->lex->view_prepare_mode)
unknown's avatar
unknown committed
4975
  {
4976 4977 4978 4979 4980 4981 4982 4983 4984 4985
    /*
      Here we check privileges of the stored routine only during view
      creation, in order to validate the view. A runtime check is perfomed
      in Item_func_sp::execute(), and this method is not called during
      context analysis. We do not need to restore the security context
      changed in find_and_check_access because all view structures created
      in CREATE VIEW are not used for execution.  Notice, that during view
      creation we do not infer into stored routine bodies and do not check
      privileges of its statements, which would probably be a good idea
      especially if the view has SQL SECURITY DEFINER and the used stored
unknown's avatar
unknown committed
4986
      procedure has SQL SECURITY DEFINER
4987
    */
unknown's avatar
unknown committed
4988 4989 4990 4991
    Security_context *save_ctx;
    if (!(res= find_and_check_access(thd, EXECUTE_ACL, &save_ctx)))
      sp_restore_security_context(thd, save_ctx);
  }
4992 4993
  return res;
}