item_func.cc 121 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();
187
      with_subselect|=        item->with_subselect;
unknown's avatar
unknown committed
188 189 190
    }
  }
  fix_length_and_dec();
unknown's avatar
unknown committed
191 192
  if (thd->net.report_error) // An error inside fix_length_and_dec occured
    return TRUE;
193
  fixed= 1;
unknown's avatar
unknown committed
194
  return FALSE;
unknown's avatar
unknown committed
195 196
}

unknown's avatar
unknown committed
197 198 199 200 201 202 203 204 205 206 207 208 209
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);
}
210

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

    switch (order) {
    case(PREFIX):
unknown's avatar
unknown committed
220
      (*traverser)(this, argument);
221 222 223 224 225 226 227 228 229 230
      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
231
      (*traverser)(this, argument);
232 233 234 235
    }
  }
}

unknown's avatar
unknown committed
236

unknown's avatar
unknown committed
237

unknown's avatar
unknown committed
238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
/*
  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)
260 261 262 263 264 265
{
  if (arg_count)
  {
    Item **arg,**arg_end;
    for (arg= args, arg_end= args+arg_count; arg != arg_end; arg++)
    {
unknown's avatar
unknown committed
266
      Item *new_item= (*arg)->transform(transformer, argument);
267 268
      if (!new_item)
	return 0;
unknown's avatar
unknown committed
269 270
      if (*arg != new_item)
        current_thd->change_item_tree(arg, new_item);
271 272
    }
  }
unknown's avatar
unknown committed
273
  return (this->*transformer)(argument);
274 275 276
}


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

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


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;
}

306 307 308 309 310 311 312

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


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


unknown's avatar
unknown committed
322
void Item_func::print_args(String *str, uint from)
323
{
unknown's avatar
unknown committed
324
  for (uint i=from ; i < arg_count ; i++)
unknown's avatar
unknown committed
325
  {
unknown's avatar
unknown committed
326
    if (i != from)
unknown's avatar
unknown committed
327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346
      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
347

348
bool Item_func::eq(const Item *item, bool binary_cmp) const
unknown's avatar
unknown committed
349 350 351 352 353 354 355 356 357 358 359
{
  /* 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++)
360
    if (!args[i]->eq(item_func->args[i], binary_cmp))
unknown's avatar
unknown committed
361 362 363 364
      return 0;
  return 1;
}

365

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

unknown's avatar
unknown committed
371
  switch (result_type()) {
372 373 374 375 376 377 378 379 380 381 382 383
  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:
384
    res= make_string_field(t_arg);
385
    break;
unknown's avatar
unknown committed
386
  case DECIMAL_RESULT:
unknown's avatar
unknown committed
387 388 389 390
    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
391
    break;
392
  case ROW_RESULT:
unknown's avatar
unknown committed
393
  default:
unknown's avatar
unknown committed
394
    // This case should never be chosen
unknown's avatar
unknown committed
395 396
    DBUG_ASSERT(0);
    break;
397 398 399 400
  }
  return res;
}

unknown's avatar
unknown committed
401 402 403 404 405 406 407
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;
}

408

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


420 421 422 423 424 425 426 427 428 429 430
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
431
void Item_func::fix_num_length_and_dec()
unknown's avatar
unknown committed
432
{
433
  uint fl_length= 0;
unknown's avatar
unknown committed
434
  decimals=0;
unknown's avatar
unknown committed
435
  for (uint i=0 ; i < arg_count ; i++)
unknown's avatar
unknown committed
436 437
  {
    set_if_bigger(decimals,args[i]->decimals);
438 439
    set_if_bigger(fl_length, args[i]->max_length);
  }
unknown's avatar
unknown committed
440
  max_length=float_length(decimals);
441 442 443 444
  if (fl_length > max_length)
  {
    decimals= NOT_FIXED_DEC;
    max_length= float_length(NOT_FIXED_DEC);
unknown's avatar
unknown committed
445
  }
unknown's avatar
unknown committed
446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462
}


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
463
  int max_int_part= 0;
unknown's avatar
unknown committed
464
  decimals= 0;
unknown's avatar
unknown committed
465
  unsigned_flag= 1;
unknown's avatar
unknown committed
466
  for (uint i=0 ; i < arg_count ; i++)
unknown's avatar
unknown committed
467
  {
unknown's avatar
unknown committed
468
    set_if_bigger(decimals, args[i]->decimals);
unknown's avatar
unknown committed
469 470
    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
471
  }
unknown's avatar
unknown committed
472 473 474
  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
475 476 477
}


unknown's avatar
unknown committed
478 479 480 481 482 483 484 485
/*
  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
486
{
unknown's avatar
unknown committed
487
  max_length= 0;
unknown's avatar
unknown committed
488
  unsigned_flag= 0;
unknown's avatar
unknown committed
489
  for (uint i=0 ; i < arg_count ; i++)
unknown's avatar
unknown committed
490
  {
unknown's avatar
unknown committed
491
    set_if_bigger(max_length, args[i]->max_length);
unknown's avatar
unknown committed
492 493
    set_if_bigger(unsigned_flag, args[i]->unsigned_flag);
  }
unknown's avatar
unknown committed
494 495
}

unknown's avatar
unknown committed
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 524 525 526 527 528 529 530 531

/*
  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
532 533 534 535 536 537 538 539 540 541
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;
}


542
Item *Item_func::get_tmp_table_item(THD *thd)
543 544 545
{
  if (!with_sum_func && !const_item())
    return new Item_field(result_field);
546
  return copy_or_same(thd);
547 548
}

unknown's avatar
unknown committed
549 550
String *Item_int_func::val_str(String *str)
{
551
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
552 553 554
  longlong nr=val_int();
  if (null_value)
    return 0;
555
  if (!unsigned_flag)
unknown's avatar
unknown committed
556
    str->set(nr,&my_charset_bin);
557
  else
unknown's avatar
unknown committed
558
    str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
559 560 561
  return str;
}

unknown's avatar
unknown committed
562

563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587
void Item_func_connection_id::fix_length_and_dec()
{
  Item_int_func::fix_length_and_dec();
  max_length= 10;
}


bool Item_func_connection_id::fix_fields(THD *thd, Item **ref)
{
  if (Item_int_func::fix_fields(thd, ref))
    return TRUE;

  /*
    To replicate CONNECTION_ID() properly we should use
    pseudo_thread_id on slave, which contains the value of thread_id
    on master.
  */
  value= ((thd->slave_thread) ?
          thd->variables.pseudo_thread_id :
          thd->thread_id);

  return FALSE;
}


588
/*
589 590
  Check arguments here to determine result's type for a numeric
  function of two arguments.
unknown's avatar
unknown committed
591 592 593

  SYNOPSIS
    Item_num_op::find_num_type()
594
*/
unknown's avatar
unknown committed
595 596 597

void Item_num_op::find_num_type(void)
{
unknown's avatar
unknown committed
598 599 600 601 602 603 604 605
  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)
606
  {
unknown's avatar
unknown committed
607 608 609 610 611 612 613 614 615
    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();
  }
616
  else
617
  {
618
    DBUG_ASSERT(r0 == INT_RESULT && r1 == INT_RESULT);
unknown's avatar
unknown committed
619
    decimals= 0;
unknown's avatar
unknown committed
620
    hybrid_type=INT_RESULT;
unknown's avatar
unknown committed
621
    result_precision();
622
  }
unknown's avatar
unknown committed
623 624 625 626 627 628
  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
629 630
}

unknown's avatar
unknown committed
631 632

/*
633 634 635
  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
636 637 638 639

  SYNOPSIS
    Item_func_num1::find_num_type()
*/
640

unknown's avatar
unknown committed
641 642 643 644
void Item_func_num1::find_num_type()
{
  DBUG_ENTER("Item_func_num1::find_num_type");
  DBUG_PRINT("info", ("name %s", func_name()));
645
  switch (hybrid_type= args[0]->result_type()) {
unknown's avatar
unknown committed
646
  case INT_RESULT:
647
    unsigned_flag= args[0]->unsigned_flag;
unknown's avatar
unknown committed
648 649 650 651 652 653 654 655 656 657
    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);
658
  }
unknown's avatar
unknown committed
659 660 661 662 663 664
  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
665 666
}

unknown's avatar
unknown committed
667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682

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
683
{
684
  DBUG_ASSERT(fixed == 1);
685
  switch (hybrid_type) {
unknown's avatar
unknown committed
686 687 688 689
  case DECIMAL_RESULT:
  {
    my_decimal decimal_value, *val;
    if (!(val= decimal_op(&decimal_value)))
690
      return 0;                                 // null is set
unknown's avatar
unknown committed
691 692 693 694 695 696 697
    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
698 699
    if (null_value)
      return 0; /* purecov: inspected */
700
    if (!unsigned_flag)
unknown's avatar
unknown committed
701
      str->set(nr,&my_charset_bin);
702
    else
unknown's avatar
unknown committed
703
      str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
704
    break;
unknown's avatar
unknown committed
705
  }
unknown's avatar
unknown committed
706
  case REAL_RESULT:
unknown's avatar
unknown committed
707
  {
708
    double nr= real_op();
unknown's avatar
unknown committed
709 710
    if (null_value)
      return 0; /* purecov: inspected */
unknown's avatar
unknown committed
711
    str->set(nr,decimals,&my_charset_bin);
unknown's avatar
unknown committed
712 713
    break;
  }
unknown's avatar
unknown committed
714 715
  case STRING_RESULT:
    return str_op(&str_value);
unknown's avatar
unknown committed
716 717
  default:
    DBUG_ASSERT(0);
unknown's avatar
unknown committed
718 719 720 721 722
  }
  return str;
}


unknown's avatar
unknown committed
723 724 725
double Item_func_numhybrid::val_real()
{
  DBUG_ASSERT(fixed == 1);
726
  switch (hybrid_type) {
unknown's avatar
unknown committed
727 728 729 730
  case DECIMAL_RESULT:
  {
    my_decimal decimal_value, *val;
    double result;
731 732
    if (!(val= decimal_op(&decimal_value)))
      return 0.0;                               // null is set
unknown's avatar
unknown committed
733 734 735 736 737 738 739
    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
740 741 742 743 744 745 746 747
  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
748 749 750 751 752 753 754 755 756 757
  default:
    DBUG_ASSERT(0);
  }
  return 0.0;
}


longlong Item_func_numhybrid::val_int()
{
  DBUG_ASSERT(fixed == 1);
758
  switch (hybrid_type) {
unknown's avatar
unknown committed
759 760 761 762
  case DECIMAL_RESULT:
  {
    my_decimal decimal_value, *val;
    if (!(val= decimal_op(&decimal_value)))
763
      return 0;                                 // null is set
unknown's avatar
unknown committed
764 765 766 767 768 769 770
    longlong result;
    my_decimal2int(E_DEC_FATAL_ERROR, val, unsigned_flag, &result);
    return result;
  }
  case INT_RESULT:
    return int_op();
  case REAL_RESULT:
771
    return (longlong) rint(real_op());
unknown's avatar
unknown committed
772 773 774
  case STRING_RESULT:
  {
    int err_not_used;
unknown's avatar
unknown committed
775 776 777 778
    String *res;
    if (!(res= str_op(&str_value)))
      return 0;

779
    char *end= (char*) res->ptr() + res->length();
unknown's avatar
unknown committed
780
    CHARSET_INFO *cs= str_value.charset();
unknown's avatar
unknown committed
781
    return (*(cs->cset->strtoll10))(cs, res->ptr(), &end, &err_not_used);
unknown's avatar
unknown committed
782
  }
unknown's avatar
unknown committed
783 784 785 786 787 788 789 790 791 792 793
  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);
794
  switch (hybrid_type) {
unknown's avatar
unknown committed
795 796 797 798 799 800 801 802 803 804 805
  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:
  {
806
    double result= (double)real_op();
unknown's avatar
unknown committed
807 808 809 810
    double2my_decimal(E_DEC_FATAL_ERROR, result, decimal_value);
    break;
  }
  case STRING_RESULT:
unknown's avatar
unknown committed
811
  {
unknown's avatar
unknown committed
812 813 814 815
    String *res;
    if (!(res= str_op(&str_value)))
      return NULL;

unknown's avatar
unknown committed
816 817 818 819
    str2my_decimal(E_DEC_FATAL_ERROR, (char*) res->ptr(),
                   res->length(), res->charset(), decimal_value);
    break;
  }  
unknown's avatar
unknown committed
820 821 822 823 824 825 826 827
  case ROW_RESULT:
  default:
    DBUG_ASSERT(0);
  }
  return val;
}


828 829
void Item_func_signed::print(String *str)
{
830
  str->append(STRING_WITH_LEN("cast("));
831
  args[0]->print(str);
832
  str->append(STRING_WITH_LEN(" as signed)"));
833 834 835 836

}


837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 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 886 887 888
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;
}


889 890
void Item_func_unsigned::print(String *str)
{
891
  str->append(STRING_WITH_LEN("cast("));
892
  args[0]->print(str);
893
  str->append(STRING_WITH_LEN(" as unsigned)"));
894 895 896 897

}


898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918
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
919 920 921
String *Item_decimal_typecast::val_str(String *str)
{
  my_decimal tmp_buf, *tmp= val_decimal(&tmp_buf);
922 923
  if (null_value)
    return NULL;
924
  my_decimal2string(E_DEC_FATAL_ERROR, tmp, 0, 0, 0, str);
unknown's avatar
unknown committed
925 926 927 928 929 930 931 932
  return str;
}


double Item_decimal_typecast::val_real()
{
  my_decimal tmp_buf, *tmp= val_decimal(&tmp_buf);
  double res;
933 934
  if (null_value)
    return 0.0;
unknown's avatar
unknown committed
935 936 937 938 939 940 941 942 943
  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;
944 945
  if (null_value)
    return 0;
unknown's avatar
unknown committed
946 947 948 949 950 951 952 953
  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);
954 955
  if ((null_value= args[0]->null_value))
    return NULL;
unknown's avatar
unknown committed
956 957 958 959 960
  my_decimal_round(E_DEC_FATAL_ERROR, tmp, decimals, FALSE, dec);
  return dec;
}


961 962
void Item_decimal_typecast::print(String *str)
{
963
  str->append(STRING_WITH_LEN("cast("));
964
  args[0]->print(str);
965
  str->append(STRING_WITH_LEN(" as decimal)"));
966 967 968
}


unknown's avatar
unknown committed
969
double Item_func_plus::real_op()
unknown's avatar
unknown committed
970
{
971
  double value= args[0]->val_real() + args[1]->val_real();
unknown's avatar
unknown committed
972 973 974 975 976
  if ((null_value=args[0]->null_value || args[1]->null_value))
    return 0.0;
  return value;
}

unknown's avatar
unknown committed
977 978

longlong Item_func_plus::int_op()
unknown's avatar
unknown committed
979
{
unknown's avatar
unknown committed
980 981 982 983 984 985 986
  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;
}


987 988 989 990 991 992 993 994 995 996 997 998
/*
  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
999 1000
my_decimal *Item_func_plus::decimal_op(my_decimal *decimal_value)
{
1001 1002 1003
  my_decimal value1, *val1;
  my_decimal value2, *val2;
  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1004 1005
  if ((null_value= args[0]->null_value))
    return 0;
1006
  val2= args[1]->val_decimal(&value2);
1007
  if (!(null_value= (args[1]->null_value ||
1008 1009
                     (my_decimal_add(E_DEC_FATAL_ERROR, decimal_value, val1,
                                     val2) > 3))))
1010 1011
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022
}

/*
  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
1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033
  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
1034 1035
}

1036 1037 1038 1039 1040 1041 1042 1043 1044 1045

/*
  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 &&
1046
      (current_thd->variables.sql_mode & MODE_NO_UNSIGNED_SUBTRACTION))
1047 1048 1049 1050
    unsigned_flag=0;
}


unknown's avatar
unknown committed
1051
double Item_func_minus::real_op()
unknown's avatar
unknown committed
1052
{
1053
  double value= args[0]->val_real() - args[1]->val_real();
unknown's avatar
unknown committed
1054 1055 1056 1057 1058
  if ((null_value=args[0]->null_value || args[1]->null_value))
    return 0.0;
  return value;
}

unknown's avatar
unknown committed
1059 1060

longlong Item_func_minus::int_op()
unknown's avatar
unknown committed
1061
{
unknown's avatar
unknown committed
1062 1063 1064 1065
  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
1066 1067
}

1068

1069 1070
/* See Item_func_plus::decimal_op for comments */

unknown's avatar
unknown committed
1071 1072
my_decimal *Item_func_minus::decimal_op(my_decimal *decimal_value)
{
1073 1074 1075 1076
  my_decimal value1, *val1;
  my_decimal value2, *val2= 

  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1077 1078
  if ((null_value= args[0]->null_value))
    return 0;
1079
  val2= args[1]->val_decimal(&value2);
1080
  if (!(null_value= (args[1]->null_value ||
1081 1082
                     (my_decimal_sub(E_DEC_FATAL_ERROR, decimal_value, val1,
                                     val2) > 3))))
1083 1084
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1085 1086
}

1087

unknown's avatar
unknown committed
1088
double Item_func_mul::real_op()
unknown's avatar
unknown committed
1089
{
1090
  DBUG_ASSERT(fixed == 1);
1091
  double value= args[0]->val_real() * args[1]->val_real();
unknown's avatar
unknown committed
1092
  if ((null_value=args[0]->null_value || args[1]->null_value))
unknown's avatar
unknown committed
1093
    return 0.0;
unknown's avatar
unknown committed
1094 1095 1096
  return value;
}

unknown's avatar
unknown committed
1097 1098

longlong Item_func_mul::int_op()
unknown's avatar
unknown committed
1099
{
1100
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1101 1102 1103 1104
  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
1105 1106 1107
}


1108 1109
/* See Item_func_plus::decimal_op for comments */

unknown's avatar
unknown committed
1110 1111
my_decimal *Item_func_mul::decimal_op(my_decimal *decimal_value)
{
1112 1113 1114
  my_decimal value1, *val1;
  my_decimal value2, *val2;
  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1115 1116
  if ((null_value= args[0]->null_value))
    return 0;
1117
  val2= args[1]->val_decimal(&value2);
1118
  if (!(null_value= (args[1]->null_value ||
1119 1120
                     (my_decimal_mul(E_DEC_FATAL_ERROR, decimal_value, val1,
                                    val2) > 3))))
1121 1122
    return decimal_value;
  return 0;
unknown's avatar
unknown committed
1123 1124 1125
}


unknown's avatar
unknown committed
1126 1127
void Item_func_mul::result_precision()
{
unknown's avatar
unknown committed
1128 1129 1130 1131 1132 1133 1134 1135 1136
  /* 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
1137 1138 1139 1140
}


double Item_func_div::real_op()
unknown's avatar
unknown committed
1141
{
1142
  DBUG_ASSERT(fixed == 1);
1143 1144
  double value= args[0]->val_real();
  double val2= args[1]->val_real();
unknown's avatar
unknown committed
1145 1146 1147 1148 1149
  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
1150
    return 0.0;
unknown's avatar
unknown committed
1151
  }
unknown's avatar
unknown committed
1152 1153 1154
  return value/val2;
}

unknown's avatar
unknown committed
1155

unknown's avatar
unknown committed
1156
my_decimal *Item_func_div::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1157
{
1158 1159
  my_decimal value1, *val1;
  my_decimal value2, *val2;
unknown's avatar
unknown committed
1160
  int err;
1161 1162

  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1163 1164
  if ((null_value= args[0]->null_value))
    return 0;
1165
  val2= args[1]->val_decimal(&value2);
unknown's avatar
unknown committed
1166 1167
  if ((null_value= args[1]->null_value))
    return 0;
unknown's avatar
unknown committed
1168 1169 1170 1171 1172 1173
  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
1174
    return 0;
1175
  }
unknown's avatar
unknown committed
1176
  return decimal_value;
unknown's avatar
unknown committed
1177 1178 1179 1180 1181
}


void Item_func_div::result_precision()
{
unknown's avatar
unknown committed
1182 1183 1184 1185 1186 1187 1188 1189 1190 1191
  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
1192 1193
}

unknown's avatar
unknown committed
1194

unknown's avatar
unknown committed
1195 1196
void Item_func_div::fix_length_and_dec()
{
unknown's avatar
unknown committed
1197
  DBUG_ENTER("Item_func_div::fix_length_and_dec");
unknown's avatar
unknown committed
1198
  prec_increment= current_thd->variables.div_precincrement;
1199
  Item_num_op::fix_length_and_dec();
1200
  switch(hybrid_type) {
unknown's avatar
unknown committed
1201 1202
  case REAL_RESULT:
  {
unknown's avatar
unknown committed
1203
    decimals=max(args[0]->decimals,args[1]->decimals)+prec_increment;
unknown's avatar
unknown committed
1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222
    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
1223 1224
}

1225 1226 1227 1228

/* Integer division */
longlong Item_func_int_div::val_int()
{
1229
  DBUG_ASSERT(fixed == 1);
1230 1231
  longlong value=args[0]->val_int();
  longlong val2=args[1]->val_int();
1232
  if ((null_value= (args[0]->null_value || args[1]->null_value)))
unknown's avatar
unknown committed
1233 1234 1235 1236
    return 0;
  if (val2 == 0)
  {
    signal_divide_by_null();
1237
    return 0;
unknown's avatar
unknown committed
1238
  }
1239 1240 1241
  return (unsigned_flag ?
	  (ulonglong) value / (ulonglong) val2 :
	  value / val2);
1242 1243 1244 1245 1246 1247 1248
}


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
1249
  unsigned_flag=args[0]->unsigned_flag | args[1]->unsigned_flag;
1250 1251 1252
}


unknown's avatar
unknown committed
1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268
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
1269
{
1270
  DBUG_ASSERT(fixed == 1);
1271 1272
  double value= args[0]->val_real();
  double val2=  args[1]->val_real();
unknown's avatar
unknown committed
1273
  if ((null_value= args[0]->null_value || args[1]->null_value))
unknown's avatar
unknown committed
1274
    return 0.0; /* purecov: inspected */
unknown's avatar
unknown committed
1275 1276 1277 1278 1279
  if (val2 == 0.0)
  {
    signal_divide_by_null();
    return 0.0;
  }
unknown's avatar
unknown committed
1280 1281 1282
  return fmod(value,val2);
}

unknown's avatar
unknown committed
1283 1284

my_decimal *Item_func_mod::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1285
{
1286 1287 1288 1289
  my_decimal value1, *val1;
  my_decimal value2, *val2;

  val1= args[0]->val_decimal(&value1);
unknown's avatar
unknown committed
1290 1291
  if ((null_value= args[0]->null_value))
    return 0;
1292
  val2= args[1]->val_decimal(&value2);
unknown's avatar
unknown committed
1293 1294 1295
  if ((null_value= args[1]->null_value))
    return 0;
  switch (my_decimal_mod(E_DEC_FATAL_ERROR & ~E_DEC_DIV_ZERO, decimal_value,
1296
                         val1, val2)) {
unknown's avatar
unknown committed
1297 1298 1299 1300
  case E_DEC_TRUNCATED:
  case E_DEC_OK:
    return decimal_value;
  case E_DEC_DIV_ZERO:
unknown's avatar
unknown committed
1301
    signal_divide_by_null();
unknown's avatar
unknown committed
1302
  default:
1303
    null_value= 1;
unknown's avatar
unknown committed
1304 1305
    return 0;
  }
unknown's avatar
unknown committed
1306 1307
}

unknown's avatar
unknown committed
1308 1309

void Item_func_mod::result_precision()
unknown's avatar
unknown committed
1310
{
unknown's avatar
unknown committed
1311 1312
  decimals= max(args[0]->decimals, args[1]->decimals);
  max_length= max(args[0]->max_length, args[1]->max_length);
unknown's avatar
unknown committed
1313 1314 1315
}


unknown's avatar
unknown committed
1316
double Item_func_neg::real_op()
unknown's avatar
unknown committed
1317
{
1318
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1319
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1320 1321 1322
  return -value;
}

unknown's avatar
unknown committed
1323

unknown's avatar
unknown committed
1324
longlong Item_func_neg::int_op()
unknown's avatar
unknown committed
1325
{
unknown's avatar
unknown committed
1326 1327
  longlong value= args[0]->val_int();
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1328 1329 1330
  return -value;
}

unknown's avatar
unknown committed
1331

unknown's avatar
unknown committed
1332
my_decimal *Item_func_neg::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1333
{
unknown's avatar
unknown committed
1334 1335 1336 1337 1338
  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);
1339
    return decimal_value;
unknown's avatar
unknown committed
1340
  }
1341
  return 0;
unknown's avatar
unknown committed
1342 1343
}

1344

unknown's avatar
unknown committed
1345
void Item_func_neg::fix_num_length_and_dec()
unknown's avatar
unknown committed
1346
{
unknown's avatar
unknown committed
1347 1348 1349
  decimals= args[0]->decimals;
  /* 1 add because sign can appear */
  max_length= args[0]->max_length + 1;
unknown's avatar
unknown committed
1350 1351
}

unknown's avatar
unknown committed
1352

1353
void Item_func_neg::fix_length_and_dec()
unknown's avatar
unknown committed
1354
{
1355
  DBUG_ENTER("Item_func_neg::fix_length_and_dec");
unknown's avatar
unknown committed
1356
  Item_func_num1::fix_length_and_dec();
1357 1358 1359 1360

  /*
    If this is in integer context keep the context as integer if possible
    (This is how multiplication and other integer functions works)
1361 1362
    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.
1363
  */
unknown's avatar
unknown committed
1364 1365
  if (hybrid_type == INT_RESULT &&
      args[0]->type() == INT_ITEM &&
unknown's avatar
unknown committed
1366
      ((ulonglong) args[0]->val_int() >= (ulonglong) LONGLONG_MIN))
unknown's avatar
unknown committed
1367 1368
  {
    /*
1369 1370
      Ensure that result is converted to DECIMAL, as longlong can't hold
      the negated number
unknown's avatar
unknown committed
1371
    */
unknown's avatar
unknown committed
1372 1373
    hybrid_type= DECIMAL_RESULT;
    DBUG_PRINT("info", ("Type changed: DECIMAL_RESULT"));
unknown's avatar
unknown committed
1374
  }
unknown's avatar
unknown committed
1375
  unsigned_flag= 0;
unknown's avatar
unknown committed
1376
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
1377 1378
}

unknown's avatar
unknown committed
1379

unknown's avatar
unknown committed
1380
double Item_func_abs::real_op()
unknown's avatar
unknown committed
1381
{
1382
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1383
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1384 1385 1386
  return fabs(value);
}

unknown's avatar
unknown committed
1387

unknown's avatar
unknown committed
1388
longlong Item_func_abs::int_op()
unknown's avatar
unknown committed
1389
{
unknown's avatar
unknown committed
1390 1391
  longlong value= args[0]->val_int();
  null_value= args[0]->null_value;
unknown's avatar
unknown committed
1392 1393 1394
  return value >= 0 ? value : -value;
}

unknown's avatar
unknown committed
1395

unknown's avatar
unknown committed
1396
my_decimal *Item_func_abs::decimal_op(my_decimal *decimal_value)
unknown's avatar
unknown committed
1397
{
unknown's avatar
unknown committed
1398 1399
  my_decimal val, *value= args[0]->val_decimal(&val);
  if (!(null_value= args[0]->null_value))
unknown's avatar
unknown committed
1400
  {
unknown's avatar
unknown committed
1401 1402 1403
    my_decimal2decimal(value, decimal_value);
    if (decimal_value->sign())
      my_decimal_neg(decimal_value);
1404
    return decimal_value;
unknown's avatar
unknown committed
1405
  }
1406
  return 0;
unknown's avatar
unknown committed
1407 1408 1409 1410 1411 1412
}


void Item_func_abs::fix_length_and_dec()
{
  Item_func_num1::fix_length_and_dec();
unknown's avatar
unknown committed
1413 1414
}

unknown's avatar
unknown committed
1415

1416
/* Gateway to natural LOG function */
1417
double Item_func_ln::val_real()
1418
{
1419
  DBUG_ASSERT(fixed == 1);
1420
  double value= args[0]->val_real();
1421
  if ((null_value= args[0]->null_value))
1422
    return 0.0;
1423
  if (value <= 0.0)
1424 1425
  {
    signal_divide_by_null();
1426
    return 0.0;
1427
  }
1428 1429 1430 1431 1432 1433 1434 1435
  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.
*/ 
1436
double Item_func_log::val_real()
unknown's avatar
unknown committed
1437
{
1438
  DBUG_ASSERT(fixed == 1);
1439
  double value= args[0]->val_real();
1440
  if ((null_value= args[0]->null_value))
1441
    return 0.0;
1442
  if (value <= 0.0)
1443 1444 1445 1446
  {
    signal_divide_by_null();
    return 0.0;
  }
1447 1448
  if (arg_count == 2)
  {
1449
    double value2= args[1]->val_real();
1450
    if ((null_value= args[1]->null_value))
1451
      return 0.0;
1452
    if (value2 <= 0.0 || value == 1.0)
1453 1454 1455 1456
    {
      signal_divide_by_null();
      return 0.0;
    }
1457 1458
    return log(value2) / log(value);
  }
unknown's avatar
unknown committed
1459 1460 1461
  return log(value);
}

1462
double Item_func_log2::val_real()
1463
{
1464
  DBUG_ASSERT(fixed == 1);
1465
  double value= args[0]->val_real();
1466 1467

  if ((null_value=args[0]->null_value))
1468
    return 0.0;
1469
  if (value <= 0.0)
1470 1471 1472 1473
  {
    signal_divide_by_null();
    return 0.0;
  }
1474
  return log(value) / M_LN2;
1475 1476
}

1477
double Item_func_log10::val_real()
unknown's avatar
unknown committed
1478
{
1479
  DBUG_ASSERT(fixed == 1);
1480
  double value= args[0]->val_real();
1481
  if ((null_value= args[0]->null_value))
1482
    return 0.0;
1483
  if (value <= 0.0)
1484 1485 1486 1487
  {
    signal_divide_by_null();
    return 0.0;
  }
unknown's avatar
unknown committed
1488 1489 1490
  return log10(value);
}

1491
double Item_func_exp::val_real()
unknown's avatar
unknown committed
1492
{
1493
  DBUG_ASSERT(fixed == 1);
1494
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1495 1496 1497 1498 1499
  if ((null_value=args[0]->null_value))
    return 0.0; /* purecov: inspected */
  return exp(value);
}

1500
double Item_func_sqrt::val_real()
unknown's avatar
unknown committed
1501
{
1502
  DBUG_ASSERT(fixed == 1);
1503
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1504 1505 1506 1507 1508
  if ((null_value=(args[0]->null_value || value < 0)))
    return 0.0; /* purecov: inspected */
  return sqrt(value);
}

1509
double Item_func_pow::val_real()
unknown's avatar
unknown committed
1510
{
1511
  DBUG_ASSERT(fixed == 1);
1512 1513
  double value= args[0]->val_real();
  double val2= args[1]->val_real();
unknown's avatar
unknown committed
1514 1515 1516 1517 1518 1519 1520
  if ((null_value=(args[0]->null_value || args[1]->null_value)))
    return 0.0; /* purecov: inspected */
  return pow(value,val2);
}

// Trigonometric functions

1521
double Item_func_acos::val_real()
unknown's avatar
unknown committed
1522
{
1523
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1524
  // the volatile's for BUG #2338 to calm optimizer down (because of gcc's bug)
1525
  volatile double value= args[0]->val_real();
unknown's avatar
unknown committed
1526 1527 1528 1529 1530
  if ((null_value=(args[0]->null_value || (value < -1.0 || value > 1.0))))
    return 0.0;
  return fix_result(acos(value));
}

1531
double Item_func_asin::val_real()
unknown's avatar
unknown committed
1532
{
1533
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1534
  // the volatile's for BUG #2338 to calm optimizer down (because of gcc's bug)
1535
  volatile double value= args[0]->val_real();
unknown's avatar
unknown committed
1536 1537 1538 1539 1540
  if ((null_value=(args[0]->null_value || (value < -1.0 || value > 1.0))))
    return 0.0;
  return fix_result(asin(value));
}

1541
double Item_func_atan::val_real()
unknown's avatar
unknown committed
1542
{
1543
  DBUG_ASSERT(fixed == 1);
1544
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1545 1546 1547 1548
  if ((null_value=args[0]->null_value))
    return 0.0;
  if (arg_count == 2)
  {
1549
    double val2= args[1]->val_real();
unknown's avatar
unknown committed
1550 1551 1552 1553 1554 1555 1556
    if ((null_value=args[1]->null_value))
      return 0.0;
    return fix_result(atan2(value,val2));
  }
  return fix_result(atan(value));
}

1557
double Item_func_cos::val_real()
unknown's avatar
unknown committed
1558
{
1559
  DBUG_ASSERT(fixed == 1);
1560
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1561 1562 1563 1564 1565
  if ((null_value=args[0]->null_value))
    return 0.0;
  return fix_result(cos(value));
}

1566
double Item_func_sin::val_real()
unknown's avatar
unknown committed
1567
{
1568
  DBUG_ASSERT(fixed == 1);
1569
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1570 1571 1572 1573 1574
  if ((null_value=args[0]->null_value))
    return 0.0;
  return fix_result(sin(value));
}

1575
double Item_func_tan::val_real()
unknown's avatar
unknown committed
1576
{
1577
  DBUG_ASSERT(fixed == 1);
1578
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589
  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()
{
1590
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604
  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()
{
1605
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620
  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()
{
1621
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638
  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
1639
void Item_func_int_val::fix_num_length_and_dec()
unknown's avatar
unknown committed
1640
{
unknown's avatar
unknown committed
1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692
  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()
{
1693 1694 1695 1696 1697 1698 1699 1700 1701
  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;
1702
    if ((dec= Item_func_ceiling::decimal_op(&dec_buf)))
1703 1704 1705 1706 1707 1708
      my_decimal2int(E_DEC_FATAL_ERROR, dec, unsigned_flag, &result);
    else
      result= 0;
    break;
  }
  default:
1709
    result= (longlong)Item_func_ceiling::real_op();
1710 1711
  };
  return result;
unknown's avatar
unknown committed
1712 1713
}

unknown's avatar
unknown committed
1714 1715

double Item_func_ceiling::real_op()
unknown's avatar
unknown committed
1716
{
unknown's avatar
unknown committed
1717 1718 1719 1720
  /*
    the volatile's for BUG #3051 to calm optimizer down (because of gcc's
    bug)
  */
1721
  volatile double value= args[0]->val_real();
unknown's avatar
unknown committed
1722 1723 1724 1725 1726 1727
  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
1728
{
unknown's avatar
unknown committed
1729
  my_decimal val, *value= args[0]->val_decimal(&val);
1730 1731 1732 1733 1734
  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
1735 1736 1737 1738 1739
}


longlong Item_func_floor::int_op()
{
1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758
  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
1759 1760
}

unknown's avatar
unknown committed
1761 1762

double Item_func_floor::real_op()
unknown's avatar
unknown committed
1763
{
unknown's avatar
unknown committed
1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776
  /*
    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);
1777 1778 1779 1780 1781
  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
1782 1783 1784
}


unknown's avatar
unknown committed
1785
void Item_func_round::fix_length_and_dec()
unknown's avatar
unknown committed
1786
{
unknown's avatar
unknown committed
1787 1788
  unsigned_flag= args[0]->unsigned_flag;
  if (!args[1]->const_item())
unknown's avatar
unknown committed
1789
  {
unknown's avatar
unknown committed
1790 1791 1792 1793 1794 1795
    max_length= args[0]->max_length;
    decimals= args[0]->decimals;
    hybrid_type= REAL_RESULT;
    return;
  }
  
1796
  int decimals_to_set= max((int)args[1]->val_int(), 0);
unknown's avatar
unknown committed
1797 1798 1799 1800 1801 1802 1803 1804
  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
1805
  switch (args[0]->result_type()) {
unknown's avatar
unknown committed
1806 1807 1808 1809 1810 1811 1812
  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
1813
    if (!decimals_to_set &&
1814
        (truncate || (args[0]->decimal_precision() < DECIMAL_LONGLONG_DIGITS)))
unknown's avatar
unknown committed
1815
    {
unknown's avatar
unknown committed
1816 1817
      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
1818 1819 1820 1821 1822
      /* Here we can keep INT_RESULT */
      hybrid_type= INT_RESULT;
      decimals= 0;
      break;
    }
unknown's avatar
unknown committed
1823
    /* fall through */
unknown's avatar
unknown committed
1824 1825 1826 1827 1828
  case DECIMAL_RESULT:
  {
    hybrid_type= DECIMAL_RESULT;
    int decimals_delta= args[0]->decimals - decimals_to_set;
    int precision= args[0]->decimal_precision();
1829 1830 1831 1832
    int length_increase= ((decimals_delta <= 0) || truncate) ? 0:1;

    precision-= decimals_delta - length_increase;
    decimals= decimals_to_set;
unknown's avatar
unknown committed
1833
    max_length= my_decimal_precision_to_length(precision, decimals,
1834
                                               unsigned_flag);
unknown's avatar
unknown committed
1835 1836 1837 1838
    break;
  }
  default:
    DBUG_ASSERT(0); /* This result type isn't handled */
unknown's avatar
unknown committed
1839 1840 1841
  }
}

1842
double my_double_round(double value, int dec, bool truncate)
unknown's avatar
unknown committed
1843
{
1844
  double tmp;
1845
  uint abs_dec= abs(dec);
1846 1847 1848 1849 1850
  /*
    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
1851

1852 1853
  tmp=(abs_dec < array_elements(log_10) ?
       log_10[abs_dec] : pow(10.0,(double) abs_dec));
unknown's avatar
unknown committed
1854 1855

  if (truncate)
unknown's avatar
unknown committed
1856 1857
  {
    if (value >= 0)
1858
      tmp2= dec < 0 ? floor(value/tmp)*tmp : floor(value*tmp)/tmp;
unknown's avatar
unknown committed
1859
    else
1860
      tmp2= dec < 0 ? ceil(value/tmp)*tmp : ceil(value*tmp)/tmp;
unknown's avatar
unknown committed
1861
  }
1862 1863 1864
  else
    tmp2=dec < 0 ? rint(value/tmp)*tmp : rint(value*tmp)/tmp;
  return tmp2;
unknown's avatar
unknown committed
1865 1866 1867
}


1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879
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
1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891
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;
unknown's avatar
unknown committed
1892 1893 1894 1895 1896 1897 1898
  longlong tmp;
  
  if(abs_dec >= array_elements(log_10_int))
    return 0;
  
  tmp= log_10_int[abs_dec];
  
unknown's avatar
unknown committed
1899 1900 1901
  if (truncate)
  {
    if (unsigned_flag)
unknown's avatar
unknown committed
1902
      value= (ulonglong(value)/tmp)*tmp;
unknown's avatar
unknown committed
1903
    else
unknown's avatar
unknown committed
1904
      value= (value/tmp)*tmp;
unknown's avatar
unknown committed
1905 1906
  }
  else
unknown's avatar
unknown committed
1907 1908 1909 1910 1911 1912 1913 1914 1915
  {
    if (unsigned_flag)
      value= ((ulonglong(value)+(tmp>>1))/tmp)*tmp;
    else if ( value >= 0)
      value= ((value+(tmp>>1))/tmp)*tmp;
    else
      value= ((value-(tmp>>1))/tmp)*tmp;
  }
  return value;
unknown's avatar
unknown committed
1916 1917 1918 1919 1920 1921 1922 1923
}


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
1924 1925 1926
  {
    decimals= min(dec, DECIMAL_MAX_SCALE); // to get correct output
  }
1927 1928 1929 1930 1931
  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
1932 1933 1934
}


1935
bool Item_func_rand::fix_fields(THD *thd,Item **ref)
unknown's avatar
unknown committed
1936
{
1937
  if (Item_real_func::fix_fields(thd, ref))
1938
    return TRUE;
1939
  used_tables_cache|= RAND_TABLE_BIT;
unknown's avatar
unknown committed
1940 1941
  if (arg_count)
  {					// Only use argument once in query
1942 1943 1944 1945 1946
    if (!args[0]->const_during_execution())
    {
      my_error(ER_WRONG_ARGUMENTS, MYF(0), "RAND");
      return TRUE;
    }
1947
    /*
unknown's avatar
unknown committed
1948
      Allocate rand structure once: we must use thd->stmt_arena
1949 1950
      to create rand in proper mem_root if it's a prepared statement or
      stored procedure.
1951 1952 1953

      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.
1954 1955
    */
    if (!rand && !(rand= (struct rand_struct*)
unknown's avatar
unknown committed
1956
                   thd->stmt_arena->alloc(sizeof(*rand))))
1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971
      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
1972
  }
1973
  else
unknown's avatar
unknown committed
1974
  {
1975 1976 1977 1978 1979
    /*
      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
    */
1980 1981 1982 1983 1984 1985
    if (!thd->rand_used)
    {
      thd->rand_used= 1;
      thd->rand_saved_seed1= thd->rand.seed1;
      thd->rand_saved_seed2= thd->rand.seed2;
    }
1986
    rand= &thd->rand;
unknown's avatar
unknown committed
1987
  }
1988
  return FALSE;
1989 1990
}

unknown's avatar
unknown committed
1991 1992 1993 1994 1995 1996
void Item_func_rand::update_used_tables()
{
  Item_real_func::update_used_tables();
  used_tables_cache|= RAND_TABLE_BIT;
}

1997

1998
double Item_func_rand::val_real()
1999
{
2000
  DBUG_ASSERT(fixed == 1);
2001
  return my_rnd(rand);
unknown's avatar
unknown committed
2002 2003 2004 2005
}

longlong Item_func_sign::val_int()
{
2006
  DBUG_ASSERT(fixed == 1);
2007
  double value= args[0]->val_real();
unknown's avatar
unknown committed
2008 2009 2010 2011 2012
  null_value=args[0]->null_value;
  return value < 0.0 ? -1 : (value > 0 ? 1 : 0);
}


2013
double Item_func_units::val_real()
unknown's avatar
unknown committed
2014
{
2015
  DBUG_ASSERT(fixed == 1);
2016
  double value= args[0]->val_real();
unknown's avatar
unknown committed
2017 2018 2019 2020 2021 2022 2023 2024
  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
2025
  int max_int_part=0;
unknown's avatar
unknown committed
2026 2027
  decimals=0;
  max_length=0;
unknown's avatar
unknown committed
2028
  maybe_null=0;
unknown's avatar
unknown committed
2029
  cmp_type=args[0]->result_type();
2030

unknown's avatar
unknown committed
2031 2032
  for (uint i=0 ; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
2033 2034
    set_if_bigger(max_length, args[i]->max_length);
    set_if_bigger(decimals, args[i]->decimals);
unknown's avatar
unknown committed
2035
    set_if_bigger(max_int_part, args[i]->decimal_int_part());
unknown's avatar
unknown committed
2036 2037
    if (args[i]->maybe_null)
      maybe_null=1;
unknown's avatar
unknown committed
2038 2039
    cmp_type=item_cmp_type(cmp_type,args[i]->result_type());
  }
unknown's avatar
unknown committed
2040
  if (cmp_type == STRING_RESULT)
2041
    agg_arg_charsets(collation, args, arg_count, MY_COLL_CMP_CONV, 1);
unknown's avatar
unknown committed
2042 2043 2044
  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
2045 2046 2047 2048 2049
}


String *Item_func_min_max::val_str(String *str)
{
2050
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2051 2052 2053 2054 2055 2056
  switch (cmp_type) {
  case INT_RESULT:
  {
    longlong nr=val_int();
    if (null_value)
      return 0;
2057
    if (!unsigned_flag)
unknown's avatar
unknown committed
2058
      str->set(nr,&my_charset_bin);
2059
    else
unknown's avatar
unknown committed
2060
      str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
2061 2062
    return str;
  }
unknown's avatar
unknown committed
2063 2064 2065 2066 2067 2068 2069 2070
  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
2071 2072
  case REAL_RESULT:
  {
2073
    double nr= val_real();
unknown's avatar
unknown committed
2074 2075
    if (null_value)
      return 0; /* purecov: inspected */
unknown's avatar
unknown committed
2076
    str->set(nr,decimals,&my_charset_bin);
unknown's avatar
unknown committed
2077 2078 2079 2080 2081 2082 2083 2084
    return str;
  }
  case STRING_RESULT:
  {
    String *res;
    LINT_INIT(res);
    for (uint i=0; i < arg_count ; i++)
    {
unknown's avatar
unknown committed
2085
      if (i == 0)
unknown's avatar
unknown committed
2086 2087 2088 2089 2090 2091 2092
	res=args[i]->val_str(str);
      else
      {
	String *res2;
	res2= args[i]->val_str(res == str ? &tmp_value : str);
	if (res2)
	{
2093
	  int cmp= sortcmp(res,res2,collation.collation);
unknown's avatar
unknown committed
2094 2095 2096 2097
	  if ((cmp_sign < 0 ? cmp : -cmp) < 0)
	    res=res2;
	}
      }
unknown's avatar
unknown committed
2098
      if ((null_value= args[i]->null_value))
2099
        return 0;
unknown's avatar
unknown committed
2100
    }
2101
    res->set_charset(collation.collation);
unknown's avatar
unknown committed
2102 2103
    return res;
  }
2104
  case ROW_RESULT:
unknown's avatar
unknown committed
2105
  default:
unknown's avatar
unknown committed
2106
    // This case should never be chosen
unknown's avatar
unknown committed
2107 2108
    DBUG_ASSERT(0);
    return 0;
unknown's avatar
unknown committed
2109 2110 2111 2112 2113
  }
  return 0;					// Keep compiler happy
}


2114
double Item_func_min_max::val_real()
unknown's avatar
unknown committed
2115
{
2116
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2117 2118 2119
  double value=0.0;
  for (uint i=0; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
2120
    if (i == 0)
2121
      value= args[i]->val_real();
unknown's avatar
unknown committed
2122 2123
    else
    {
2124
      double tmp= args[i]->val_real();
unknown's avatar
unknown committed
2125 2126 2127
      if (!args[i]->null_value && (tmp < value ? cmp_sign : -cmp_sign) > 0)
	value=tmp;
    }
unknown's avatar
unknown committed
2128 2129
    if ((null_value= args[i]->null_value))
      break;
unknown's avatar
unknown committed
2130 2131 2132 2133 2134 2135 2136
  }
  return value;
}


longlong Item_func_min_max::val_int()
{
2137
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2138 2139 2140
  longlong value=0;
  for (uint i=0; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
2141
    if (i == 0)
unknown's avatar
unknown committed
2142 2143 2144 2145 2146 2147 2148
      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
2149 2150
    if ((null_value= args[i]->null_value))
      break;
unknown's avatar
unknown committed
2151 2152 2153 2154
  }
  return value;
}

unknown's avatar
unknown committed
2155 2156 2157 2158

my_decimal *Item_func_min_max::val_decimal(my_decimal *dec)
{
  DBUG_ASSERT(fixed == 1);
2159 2160 2161
  my_decimal tmp_buf, *tmp, *res;
  LINT_INIT(res);

unknown's avatar
unknown committed
2162 2163
  for (uint i=0; i < arg_count ; i++)
  {
unknown's avatar
unknown committed
2164
    if (i == 0)
unknown's avatar
unknown committed
2165 2166 2167
      res= args[i]->val_decimal(dec);
    else
    {
2168 2169
      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
2170 2171 2172
      {
        if (tmp == &tmp_buf)
        {
2173
          /* Move value out of tmp_buf as this will be reused on next loop */
unknown's avatar
unknown committed
2174 2175 2176 2177 2178 2179 2180
          my_decimal2decimal(tmp, dec);
          res= dec;
        }
        else
          res= tmp;
      }
    }
unknown's avatar
unknown committed
2181
    if ((null_value= args[i]->null_value))
2182 2183
    {
      res= 0;
unknown's avatar
unknown committed
2184
      break;
2185
    }
unknown's avatar
unknown committed
2186 2187 2188 2189 2190
  }
  return res;
}


unknown's avatar
unknown committed
2191 2192
longlong Item_func_length::val_int()
{
2193
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2194 2195 2196 2197 2198 2199 2200 2201 2202 2203
  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
2204

unknown's avatar
unknown committed
2205 2206
longlong Item_func_char_length::val_int()
{
2207
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2208 2209 2210 2211 2212 2213 2214
  String *res=args[0]->val_str(&value);
  if (!res)
  {
    null_value=1;
    return 0; /* purecov: inspected */
  }
  null_value=0;
2215
  return (longlong) res->numchars();
unknown's avatar
unknown committed
2216 2217
}

unknown's avatar
unknown committed
2218

unknown's avatar
unknown committed
2219 2220
longlong Item_func_coercibility::val_int()
{
2221
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2222
  null_value= 0;
2223
  return (longlong) args[0]->collation.derivation;
unknown's avatar
unknown committed
2224
}
unknown's avatar
unknown committed
2225

unknown's avatar
unknown committed
2226

2227 2228 2229
void Item_func_locate::fix_length_and_dec()
{
  maybe_null=0; max_length=11;
2230
  agg_arg_charsets(cmp_collation, args, 2, MY_COLL_CMP_CONV, 1);
2231 2232
}

unknown's avatar
unknown committed
2233

unknown's avatar
unknown committed
2234 2235
longlong Item_func_locate::val_int()
{
2236
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2237 2238 2239 2240 2241 2242 2243 2244
  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;
2245 2246
  uint start=0;
  uint start0=0;
2247
  my_match_t match;
2248

unknown's avatar
unknown committed
2249 2250
  if (arg_count == 3)
  {
2251 2252 2253
    start0= start =(uint) args[2]->val_int()-1;
    start=a->charpos(start);
    
unknown's avatar
unknown committed
2254 2255 2256
    if (start > a->length() || start+b->length() > a->length())
      return 0;
  }
2257

unknown's avatar
unknown committed
2258 2259
  if (!b->length())				// Found empty string at start
    return (longlong) (start+1);
2260
  
2261
  if (!cmp_collation.collation->coll->instr(cmp_collation.collation,
2262
                                            a->ptr()+start, a->length()-start,
2263 2264 2265 2266
                                            b->ptr(), b->length(),
                                            &match, 1))
    return 0;
  return (longlong) match.mblen + start0 + 1;
unknown's avatar
unknown committed
2267 2268 2269
}


2270 2271
void Item_func_locate::print(String *str)
{
2272
  str->append(STRING_WITH_LEN("locate("));
2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284
  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
2285 2286
longlong Item_func_field::val_int()
{
2287
  DBUG_ASSERT(fixed == 1);
2288

2289 2290 2291
  if (cmp_type == STRING_RESULT)
  {
    String *field;
2292 2293
    if (!(field= args[0]->val_str(&value)))
      return 0;
2294
    for (uint i=1 ; i < arg_count ; i++)
2295 2296
    {
      String *tmp_value=args[i]->val_str(&tmp);
unknown's avatar
Fix:  
unknown committed
2297
      if (tmp_value && !sortcmp(field,tmp_value,cmp_collation.collation))
2298
        return (longlong) (i);
2299 2300 2301 2302
    }
  }
  else if (cmp_type == INT_RESULT)
  {
2303
    longlong val= args[0]->val_int();
unknown's avatar
unknown committed
2304 2305
    if (args[0]->null_value)
      return 0;
2306
    for (uint i=1; i < arg_count ; i++)
2307
    {
2308
      if (val == args[i]->val_int() && !args[i]->null_value)
2309
        return (longlong) (i);
2310 2311
    }
  }
unknown's avatar
unknown committed
2312 2313 2314 2315
  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
2316 2317
    if (args[0]->null_value)
      return 0;
unknown's avatar
unknown committed
2318 2319 2320
    for (uint i=1; i < arg_count; i++)
    {
      dec_arg= args[i]->val_decimal(&dec_arg_buf);
unknown's avatar
unknown committed
2321
      if (!args[i]->null_value && !my_decimal_cmp(dec_arg, dec))
unknown's avatar
unknown committed
2322 2323 2324
        return (longlong) (i);
    }
  }
2325
  else
unknown's avatar
unknown committed
2326
  {
2327
    double val= args[0]->val_real();
unknown's avatar
unknown committed
2328 2329
    if (args[0]->null_value)
      return 0;
2330
    for (uint i=1; i < arg_count ; i++)
2331
    {
unknown's avatar
unknown committed
2332
      if (val == args[i]->val_real() && !args[i]->null_value)
2333
        return (longlong) (i);
2334
    }
unknown's avatar
unknown committed
2335 2336 2337 2338
  }
  return 0;
}

unknown's avatar
unknown committed
2339

2340 2341 2342
void Item_func_field::fix_length_and_dec()
{
  maybe_null=0; max_length=3;
2343 2344
  cmp_type= args[0]->result_type();
  for (uint i=1; i < arg_count ; i++)
2345 2346
    cmp_type= item_cmp_type(cmp_type, args[i]->result_type());
  if (cmp_type == STRING_RESULT)
2347
    agg_arg_charsets(cmp_collation, args, arg_count, MY_COLL_CMP_CONV, 1);
2348
}
unknown's avatar
unknown committed
2349

2350

unknown's avatar
unknown committed
2351 2352
longlong Item_func_ascii::val_int()
{
2353
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365
  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()
{
2366
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2367 2368 2369 2370 2371 2372 2373 2374 2375
  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
2376
  if (use_mb(res->charset()))
unknown's avatar
unknown committed
2377 2378
  {
    register const char *str=res->ptr();
2379
    register uint32 n=0, l=my_ismbchar(res->charset(),str,str+res->length());
unknown's avatar
unknown committed
2380 2381
    if (!l)
      return (longlong)((uchar) *str);
unknown's avatar
unknown committed
2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405
    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)
      {
2406 2407
	enum_value= find_type(((Field_enum*) field)->typelib,find->ptr(),
			      find->length(), 0);
unknown's avatar
unknown committed
2408 2409 2410 2411 2412 2413
	enum_bit=0;
	if (enum_value)
	  enum_bit=LL(1) << (enum_value-1);
      }
    }
  }
2414
  agg_arg_charsets(cmp_collation, args, 2, MY_COLL_CMP_CONV, 1);
unknown's avatar
unknown committed
2415 2416 2417 2418 2419 2420
}

static const char separator=',';

longlong Item_func_find_in_set::val_int()
{
2421
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444
  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)
  {
2445 2446 2447 2448 2449 2450 2451 2452 2453
    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
2454
    {
2455 2456 2457
      int symbol_len;
      if ((symbol_len= cs->cset->mb_wc(cs, &wc, (uchar*) str_end, 
                                       (uchar*) real_end)) > 0)
unknown's avatar
unknown committed
2458
      {
2459 2460
        const char *substr_end= str_end + symbol_len;
        bool is_last_item= (substr_end == real_end);
2461 2462
        bool is_separator= (wc == (my_wc_t) separator);
        if (is_separator || is_last_item)
2463 2464
        {
          position++;
2465
          if (is_last_item && !is_separator)
2466 2467 2468 2469 2470 2471 2472 2473 2474
            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
2475
      }
unknown's avatar
unknown committed
2476 2477
      else if (str_end - str_begin == 0 &&
               find_str_len == 0 &&
2478 2479 2480
               wc == (my_wc_t) separator)
        return (longlong) ++position;
      else
unknown's avatar
unknown committed
2481
        return LL(0);
2482
    }
unknown's avatar
unknown committed
2483 2484 2485 2486 2487 2488
  }
  return 0;
}

longlong Item_func_bit_count::val_int()
{
2489
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2490
  ulonglong value= (ulonglong) args[0]->val_int();
2491
  if ((null_value= args[0]->null_value))
unknown's avatar
unknown committed
2492
    return 0; /* purecov: inspected */
unknown's avatar
unknown committed
2493
  return (longlong) my_count_bits(value);
unknown's avatar
unknown committed
2494 2495 2496 2497 2498 2499
}


/****************************************************************************
** Functions to handle dynamic loadable functions
** Original source by: Alexis Mikhailov <root@medinf.chuvashia.su>
2500
** Rewritten by monty.
unknown's avatar
unknown committed
2501 2502 2503 2504
****************************************************************************/

#ifdef HAVE_DLOPEN

2505
void udf_handler::cleanup()
unknown's avatar
unknown committed
2506
{
2507
  if (!not_original)
unknown's avatar
unknown committed
2508
  {
2509
    if (initialized)
unknown's avatar
unknown committed
2510
    {
2511 2512
      if (u_d->func_deinit != NULL)
      {
2513
        Udf_func_deinit deinit= u_d->func_deinit;
2514 2515 2516
        (*deinit)(&initid);
      }
      free_udf(u_d);
2517
      initialized= FALSE;
unknown's avatar
unknown committed
2518
    }
2519 2520
    if (buffers)				// Because of bug in ecc
      delete [] buffers;
2521
    buffers= 0;
unknown's avatar
unknown committed
2522 2523 2524 2525 2526
  }
}


bool
2527
udf_handler::fix_fields(THD *thd, Item_result_field *func,
unknown's avatar
unknown committed
2528 2529
			uint arg_count, Item **arguments)
{
2530 2531 2532
#ifndef EMBEDDED_LIBRARY			// Avoid compiler warning
  char buff[STACK_BUFF_ALLOC];			// Max argument in function
#endif
unknown's avatar
unknown committed
2533 2534
  DBUG_ENTER("Item_udf_func::fix_fields");

2535
  if (check_stack_overrun(thd, STACK_MIN_SIZE, buff))
unknown's avatar
unknown committed
2536
    DBUG_RETURN(TRUE);				// Fatal error flag is set!
unknown's avatar
unknown committed
2537

2538
  udf_func *tmp_udf=find_udf(u_d->name.str,(uint) u_d->name.length,1);
unknown's avatar
unknown committed
2539 2540 2541

  if (!tmp_udf)
  {
2542
    my_error(ER_CANT_FIND_UDF, MYF(0), u_d->name.str, errno);
unknown's avatar
unknown committed
2543
    DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2544 2545 2546 2547 2548
  }
  u_d=tmp_udf;
  args=arguments;

  /* Fix all arguments */
2549
  func->maybe_null=0;
unknown's avatar
unknown committed
2550 2551 2552 2553 2554 2555 2556 2557 2558 2559
  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
2560
      DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2561 2562 2563 2564 2565 2566 2567
    }
    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
2568
      if (!(*arg)->fixed &&
2569
          (*arg)->fix_fields(thd, arg))
unknown's avatar
unknown committed
2570
	DBUG_RETURN(1);
2571
      // we can't assign 'item' before, because fix_fields() can change arg
unknown's avatar
unknown committed
2572
      Item *item= *arg;
2573
      if (item->check_cols(1))
unknown's avatar
unknown committed
2574
	DBUG_RETURN(TRUE);
2575 2576 2577 2578 2579 2580 2581 2582 2583
      /*
	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
2584
        the particular user defined function.
2585
      */
2586 2587
      if (item->collation.collation->state & MY_CS_BINSORT)
	func->collation.set(&my_charset_bin);
unknown's avatar
unknown committed
2588
      if (item->maybe_null)
unknown's avatar
unknown committed
2589
	func->maybe_null=1;
unknown's avatar
unknown committed
2590 2591 2592 2593
      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
2594
    }
unknown's avatar
unknown committed
2595
    //TODO: why all following memory is not allocated with 1 call of sql_alloc?
unknown's avatar
unknown committed
2596 2597
    if (!(buffers=new String[arg_count]) ||
	!(f_args.args= (char**) sql_alloc(arg_count * sizeof(char *))) ||
2598 2599 2600 2601 2602 2603 2604
	!(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
2605 2606
    {
      free_udf(u_d);
unknown's avatar
unknown committed
2607
      DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622
    }
  }
  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;
2623 2624 2625 2626
      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
2627 2628 2629 2630

      switch(arguments[i]->type()) {
      case Item::STRING_ITEM:			// Constant string !
      {
2631
	String *res=arguments[i]->val_str(&buffers[i]);
unknown's avatar
unknown committed
2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645
	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:
2646
	*((double*) to)= arguments[i]->val_real();
unknown's avatar
unknown committed
2647 2648 2649 2650 2651 2652 2653 2654 2655 2656
	if (!arguments[i]->null_value)
	{
	  f_args.args[i]=to;
	  to+= ALIGN_SIZE(sizeof(double));
	}
	break;
      default:					// Skip these
	break;
      }
    }
2657
    thd->net.last_error[0]=0;
2658
    Udf_func_init init= u_d->func_init;
unknown's avatar
unknown committed
2659 2660
    if ((error=(uchar) init(&initid, &f_args, thd->net.last_error)))
    {
2661 2662
      my_error(ER_CANT_INITIALIZE_UDF, MYF(0),
               u_d->name.str, thd->net.last_error);
unknown's avatar
unknown committed
2663
      free_udf(u_d);
unknown's avatar
unknown committed
2664
      DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2665 2666 2667 2668
    }
    func->max_length=min(initid.max_length,MAX_BLOB_WIDTH);
    func->maybe_null=initid.maybe_null;
    const_item_cache=initid.const_item;
2669
    func->decimals=min(initid.decimals,NOT_FIXED_DEC);
unknown's avatar
unknown committed
2670 2671 2672 2673
  }
  initialized=1;
  if (error)
  {
2674 2675
    my_error(ER_CANT_INITIALIZE_UDF, MYF(0),
             u_d->name.str, ER(ER_UNKNOWN_ERROR));
unknown's avatar
unknown committed
2676
    DBUG_RETURN(TRUE);
unknown's avatar
unknown committed
2677
  }
unknown's avatar
unknown committed
2678
  DBUG_RETURN(FALSE);
unknown's avatar
unknown committed
2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692
}


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
2693
    case DECIMAL_RESULT:
unknown's avatar
unknown committed
2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711
      {
	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:
2712
      *((double*) to)= args[i]->val_real();
unknown's avatar
unknown committed
2713 2714 2715 2716 2717 2718
      if (!args[i]->null_value)
      {
	f_args.args[i]=to;
	to+= ALIGN_SIZE(sizeof(double));
      }
      break;
2719
    case ROW_RESULT:
unknown's avatar
unknown committed
2720
    default:
unknown's avatar
unknown committed
2721
      // This case should never be chosen
unknown's avatar
unknown committed
2722
      DBUG_ASSERT(0);
2723
      break;
unknown's avatar
unknown committed
2724 2725 2726 2727 2728 2729 2730 2731 2732
    }
  }
  return 0;
}

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

String *udf_handler::val_str(String *str,String *save_str)
{
2733
  uchar is_null_tmp=0;
unknown's avatar
unknown committed
2734
  ulong res_length;
2735
  DBUG_ENTER("udf_handler::val_str");
unknown's avatar
unknown committed
2736 2737

  if (get_arguments())
2738
    DBUG_RETURN(0);
unknown's avatar
unknown committed
2739 2740 2741 2742 2743 2744 2745 2746 2747
  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;
2748
      DBUG_RETURN(0);
unknown's avatar
unknown committed
2749 2750
    }
  }
2751 2752
  char *res=func(&initid, &f_args, (char*) str->ptr(), &res_length,
		 &is_null_tmp, &error);
2753
  DBUG_PRINT("info", ("udf func returned, res_length: %lu", res_length));
2754
  if (is_null_tmp || !res || error)		// The !res is for safety
unknown's avatar
unknown committed
2755
  {
2756 2757
    DBUG_PRINT("info", ("Null or error"));
    DBUG_RETURN(0);
unknown's avatar
unknown committed
2758 2759 2760 2761
  }
  if (res == str->ptr())
  {
    str->length(res_length);
2762 2763
    DBUG_PRINT("exit", ("str: %s", str->ptr()));
    DBUG_RETURN(str);
unknown's avatar
unknown committed
2764
  }
2765
  save_str->set(res, res_length, str->charset());
2766 2767
  DBUG_PRINT("exit", ("save_str: %s", save_str->ptr()));
  DBUG_RETURN(save_str);
unknown's avatar
unknown committed
2768 2769 2770
}


2771 2772 2773 2774
/*
  For the moment, UDF functions are returning DECIMAL values as strings
*/

unknown's avatar
unknown committed
2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794
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;
  }
2795 2796
  end= res+ res_length;
  str2my_decimal(E_DEC_FATAL_ERROR, res, dec_buf, &end);
unknown's avatar
unknown committed
2797 2798 2799
  return dec_buf;
}

unknown's avatar
unknown committed
2800

2801 2802 2803 2804 2805 2806
void Item_udf_func::cleanup()
{
  udf.cleanup();
  Item_func::cleanup();
}

unknown's avatar
unknown committed
2807

2808
double Item_func_udf_float::val_real()
unknown's avatar
unknown committed
2809
{
2810
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2811 2812 2813 2814 2815 2816 2817 2818 2819
  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)
{
2820
  DBUG_ASSERT(fixed == 1);
2821
  double nr= val_real();
unknown's avatar
unknown committed
2822 2823
  if (null_value)
    return 0;					/* purecov: inspected */
unknown's avatar
unknown committed
2824
  str->set(nr,decimals,&my_charset_bin);
unknown's avatar
unknown committed
2825 2826 2827 2828 2829 2830
  return str;
}


longlong Item_func_udf_int::val_int()
{
2831
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2832 2833 2834 2835 2836 2837 2838
  DBUG_ENTER("Item_func_udf_int::val_int");
  DBUG_RETURN(udf.val_int(&null_value));
}


String *Item_func_udf_int::val_str(String *str)
{
2839
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2840 2841 2842
  longlong nr=val_int();
  if (null_value)
    return 0;
2843
  if (!unsigned_flag)
unknown's avatar
unknown committed
2844
    str->set(nr,&my_charset_bin);
2845
  else
unknown's avatar
unknown committed
2846
    str->set((ulonglong) nr,&my_charset_bin);
unknown's avatar
unknown committed
2847 2848 2849
  return str;
}

unknown's avatar
unknown committed
2850 2851 2852 2853

longlong Item_func_udf_decimal::val_int()
{
  my_decimal dec_buf, *dec= udf.val_decimal(&null_value, &dec_buf);
2854
  longlong result;
unknown's avatar
unknown committed
2855 2856 2857 2858 2859 2860 2861 2862 2863 2864
  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);
2865
  double result;
unknown's avatar
unknown committed
2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902
  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
2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915
/* 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)
{
2916
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2917 2918 2919 2920 2921
  String *res=udf.val_str(str,&str_value);
  null_value = !res;
  return res;
}

2922 2923

/*
2924 2925
   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.)
2926 2927 2928 2929
 */

udf_handler::~udf_handler()
{
2930 2931
  /* Everything should be properly cleaned up by this moment. */
  DBUG_ASSERT(not_original || !(initialized || buffers));
2932 2933
}

unknown's avatar
unknown committed
2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944
#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;

2945
class User_level_lock
unknown's avatar
unknown committed
2946 2947 2948 2949 2950 2951 2952 2953 2954
{
  char *key;
  uint key_length;

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

2957
  User_level_lock(const char *key_arg,uint length, ulong id) 
unknown's avatar
SCRUM  
unknown committed
2958
    :key_length(length),count(1),locked(1), thread_id(id)
unknown's avatar
unknown committed
2959 2960 2961 2962 2963
  {
    key=(char*) my_memdup((byte*) key_arg,length,MYF(0));
    pthread_cond_init(&cond,NULL);
    if (key)
    {
unknown's avatar
SCRUM  
unknown committed
2964
      if (my_hash_insert(&hash_user_locks,(byte*) this))
unknown's avatar
unknown committed
2965 2966 2967 2968 2969 2970
      {
	my_free((gptr) key,MYF(0));
	key=0;
      }
    }
  }
2971
  ~User_level_lock()
unknown's avatar
unknown committed
2972 2973 2974 2975 2976 2977 2978 2979 2980
  {
    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; }
2981 2982 2983
  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
2984 2985
};

2986
char *ull_get_key(const User_level_lock *ull, uint *length,
unknown's avatar
unknown committed
2987 2988 2989 2990 2991 2992
		  my_bool not_used __attribute__((unused)))
{
  *length=(uint) ull->key_length;
  return (char*) ull->key;
}

2993 2994 2995

static bool item_user_lock_inited= 0;

unknown's avatar
unknown committed
2996 2997
void item_user_lock_init(void)
{
2998
  pthread_mutex_init(&LOCK_user_locks,MY_MUTEX_INIT_SLOW);
unknown's avatar
unknown committed
2999 3000
  hash_init(&hash_user_locks,system_charset_info,
	    16,0,0,(hash_get_key) ull_get_key,NULL,0);
3001
  item_user_lock_inited= 1;
unknown's avatar
unknown committed
3002 3003 3004 3005
}

void item_user_lock_free(void)
{
3006 3007 3008 3009 3010 3011
  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
3012 3013
}

3014
void item_user_lock_release(User_level_lock *ull)
unknown's avatar
unknown committed
3015 3016
{
  ull->locked=0;
3017
  ull->thread_id= 0;
unknown's avatar
unknown committed
3018 3019 3020 3021 3022 3023
  if (--ull->count)
    pthread_cond_signal(&ull->cond);
  else
    delete ull;
}

unknown's avatar
unknown committed
3024 3025 3026 3027 3028 3029 3030
/*
   Wait until we are at or past the given position in the master binlog
   on the slave
 */

longlong Item_master_pos_wait::val_int()
{
3031
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3032 3033
  THD* thd = current_thd;
  String *log_name = args[0]->val_str(&value);
3034
  int event_count= 0;
unknown's avatar
unknown committed
3035

unknown's avatar
unknown committed
3036 3037 3038 3039 3040 3041
  null_value=0;
  if (thd->slave_thread || !log_name || !log_name->length())
  {
    null_value = 1;
    return 0;
  }
unknown's avatar
unknown committed
3042
  longlong pos = (ulong)args[1]->val_int();
3043
  longlong timeout = (arg_count==3) ? args[2]->val_int() : 0 ;
unknown's avatar
SCRUM  
unknown committed
3044
#ifdef HAVE_REPLICATION
3045
  if ((event_count = active_mi->rli.wait_for_pos(thd, log_name, pos, timeout)) == -2)
unknown's avatar
unknown committed
3046 3047 3048 3049
  {
    null_value = 1;
    event_count=0;
  }
3050
#endif
unknown's avatar
unknown committed
3051 3052 3053
  return event_count;
}

3054 3055 3056 3057
#ifdef EXTRA_DEBUG
void debug_sync_point(const char* lock_name, uint lock_timeout)
{
  THD* thd=current_thd;
3058
  User_level_lock* ull;
3059
  struct timespec abstime;
3060
  int lock_name_len;
3061 3062 3063 3064 3065 3066 3067 3068 3069
  lock_name_len=strlen(lock_name);
  pthread_mutex_lock(&LOCK_user_locks);

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

3070 3071 3072 3073 3074
  /*
    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
3075
  */
3076
  if (!(ull= ((User_level_lock*) hash_search(&hash_user_locks, lock_name,
3077 3078 3079 3080 3081 3082 3083
				 lock_name_len))))
  {
    pthread_mutex_unlock(&LOCK_user_locks);
    return;
  }
  ull->count++;

3084 3085 3086 3087
  /*
    Structure is now initialized.  Try to get the lock.
    Set up control struct to allow others to abort locks
  */
3088 3089 3090 3091
  thd->proc_info="User lock";
  thd->mysys_var->current_mutex= &LOCK_user_locks;
  thd->mysys_var->current_cond=  &ull->cond;

3092
  set_timespec(abstime,lock_timeout);
unknown's avatar
unknown committed
3093 3094 3095 3096 3097 3098 3099
  while (ull->locked && !thd->killed)
  {
    int error= pthread_cond_timedwait(&ull->cond, &LOCK_user_locks, &abstime);
    if (error == ETIMEDOUT || error == ETIME)
      break;
  }

3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127
  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
3128 3129 3130 3131 3132 3133 3134 3135 3136
/*
  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()
{
3137
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3138 3139 3140 3141
  String *res=args[0]->val_str(&value);
  longlong timeout=args[1]->val_int();
  struct timespec abstime;
  THD *thd=current_thd;
3142
  User_level_lock *ull;
unknown's avatar
unknown committed
3143
  int error;
unknown's avatar
unknown committed
3144

3145 3146 3147 3148 3149 3150 3151 3152 3153 3154
  /*
    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
3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170
  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;
  }

3171 3172 3173
  if (!(ull= ((User_level_lock *) hash_search(&hash_user_locks,
                                              (byte*) res->ptr(),
                                              res->length()))))
unknown's avatar
unknown committed
3174
  {
3175
    ull=new User_level_lock(res->ptr(),res->length(), thd->thread_id);
unknown's avatar
unknown committed
3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189
    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++;

3190 3191 3192 3193
  /*
    Structure is now initialized.  Try to get the lock.
    Set up control struct to allow others to abort locks.
  */
unknown's avatar
unknown committed
3194 3195 3196 3197
  thd->proc_info="User lock";
  thd->mysys_var->current_mutex= &LOCK_user_locks;
  thd->mysys_var->current_cond=  &ull->cond;

3198
  set_timespec(abstime,timeout);
unknown's avatar
unknown committed
3199 3200 3201 3202 3203 3204 3205 3206 3207
  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
3208 3209 3210
  if (ull->locked)
  {
    if (!--ull->count)
unknown's avatar
unknown committed
3211 3212
    {
      DBUG_ASSERT(0);
unknown's avatar
unknown committed
3213
      delete ull;				// Should never happen
unknown's avatar
unknown committed
3214 3215
    }
    if (!error)                                 // Killed (thd->killed != 0)
unknown's avatar
unknown committed
3216 3217 3218 3219 3220
    {
      error=1;
      null_value=1;				// Return NULL
    }
  }
unknown's avatar
unknown committed
3221
  else                                          // We got the lock
unknown's avatar
unknown committed
3222 3223 3224
  {
    ull->locked=1;
    ull->thread=thd->real_id;
3225
    ull->thread_id= thd->thread_id;
unknown's avatar
unknown committed
3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241
    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;
}


/*
3242 3243 3244 3245 3246
  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
3247 3248 3249 3250
*/

longlong Item_func_release_lock::val_int()
{
3251
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3252
  String *res=args[0]->val_str(&value);
3253
  User_level_lock *ull;
unknown's avatar
unknown committed
3254 3255 3256 3257 3258 3259 3260 3261 3262 3263
  longlong result;
  if (!res || !res->length())
  {
    null_value=1;
    return 0;
  }
  null_value=0;

  result=0;
  pthread_mutex_lock(&LOCK_user_locks);
3264 3265 3266
  if (!(ull= ((User_level_lock*) hash_search(&hash_user_locks,
                                             (const byte*) res->ptr(),
                                             res->length()))))
unknown's avatar
unknown committed
3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283
  {
    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;
}


3284
longlong Item_func_last_insert_id::val_int()
unknown's avatar
unknown committed
3285
{
unknown's avatar
unknown committed
3286
  THD *thd= current_thd;
3287
  DBUG_ASSERT(fixed == 1);
3288 3289
  if (arg_count)
  {
unknown's avatar
unknown committed
3290 3291 3292 3293
    longlong value= args[0]->val_int();
    thd->insert_id(value);
    null_value= args[0]->null_value;
    return value;                       // Avoid side effect of insert_id()
3294
  }
unknown's avatar
unknown committed
3295
  thd->lex->uncacheable(UNCACHEABLE_SIDEEFFECT);
unknown's avatar
unknown committed
3296
  return thd->last_insert_id_used ? thd->current_insert_id : thd->insert_id();
unknown's avatar
unknown committed
3297 3298 3299 3300 3301 3302
}

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

longlong Item_func_benchmark::val_int()
{
3303
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3304
  char buff[MAX_FIELD_WIDTH];
unknown's avatar
unknown committed
3305
  String tmp(buff,sizeof(buff), &my_charset_bin);
unknown's avatar
unknown committed
3306 3307 3308 3309 3310 3311
  THD *thd=current_thd;

  for (ulong loop=0 ; loop < loop_count && !thd->killed; loop++)
  {
    switch (args[0]->result_type()) {
    case REAL_RESULT:
3312
      (void) args[0]->val_real();
unknown's avatar
unknown committed
3313 3314 3315 3316 3317 3318 3319
      break;
    case INT_RESULT:
      (void) args[0]->val_int();
      break;
    case STRING_RESULT:
      (void) args[0]->val_str(&tmp);
      break;
3320
    case ROW_RESULT:
unknown's avatar
unknown committed
3321
    default:
unknown's avatar
unknown committed
3322
      // This case should never be chosen
unknown's avatar
unknown committed
3323 3324
      DBUG_ASSERT(0);
      return 0;
unknown's avatar
unknown committed
3325 3326 3327 3328 3329
    }
  }
  return 0;
}

3330

3331 3332
void Item_func_benchmark::print(String *str)
{
3333
  str->append(STRING_WITH_LEN("benchmark("));
3334
  char buffer[20];
unknown's avatar
unknown committed
3335 3336 3337
  // my_charset_bin is good enough for numbers
  String st(buffer, sizeof(buffer), &my_charset_bin);
  st.set((ulonglong)loop_count, &my_charset_bin);
3338 3339 3340 3341 3342 3343
  str->append(st);
  str->append(',');
  args[0]->print(str);
  str->append(')');
}

unknown's avatar
unknown committed
3344

3345 3346 3347 3348
/* This function is just used to create tests with time gaps */

longlong Item_func_sleep::val_int()
{
3349 3350 3351
  THD *thd= current_thd;
  struct timespec abstime;
  pthread_cond_t cond;
3352
  int error;
3353

3354
  DBUG_ASSERT(fixed == 1);
3355

3356
  double time= args[0]->val_real();
3357 3358 3359 3360 3361 3362 3363 3364
  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
3365 3366 3367 3368 3369 3370 3371 3372
  error= 0;
  while (!thd->killed)
  {
    error= pthread_cond_timedwait(&cond, &LOCK_user_locks, &abstime);
    if (error == ETIMEDOUT || error == ETIME)
      break;
    error= 0;
  }
3373 3374 3375 3376 3377 3378 3379 3380 3381

  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
3382
  return test(!error); 		// Return 1 killed
3383 3384 3385
}


unknown's avatar
unknown committed
3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405 3406
#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;
3407
    entry->update_query_id=0;
3408
    entry->collation.set(NULL, DERIVATION_IMPLICIT);
3409 3410 3411 3412 3413 3414 3415 3416 3417 3418
    /*
      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
3419
    entry->used_query_id=current_thd->query_id;
unknown's avatar
unknown committed
3420 3421
    entry->type=STRING_RESULT;
    memcpy(entry->name.str, name.str, name.length+1);
unknown's avatar
SCRUM  
unknown committed
3422
    if (my_hash_insert(hash,(byte*) entry))
unknown's avatar
unknown committed
3423 3424 3425 3426 3427 3428 3429 3430
    {
      my_free((char*) entry,MYF(0));
      return 0;
    }
  }
  return entry;
}

3431
/*
3432 3433
  When a user variable is updated (in a SET command or a query like
  SELECT @a:= ).
3434
*/
3435

3436
bool Item_func_set_user_var::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
3437
{
3438
  DBUG_ASSERT(fixed == 0);
3439
  /* fix_fields will call Item_func_set_user_var::fix_length_and_dec */
3440
  if (Item_func::fix_fields(thd, ref) ||
unknown's avatar
unknown committed
3441
      !(entry= get_variable(&thd->user_vars, name, 1)))
unknown's avatar
unknown committed
3442
    return TRUE;
3443 3444 3445 3446 3447
  /* 
     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).
  */
3448
  entry->update_query_id= thd->query_id;
3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463
  /*
    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.
  */
3464 3465
  null_item= (args[0]->type() == NULL_ITEM);
  if (!entry->collation.collation || !null_item)
3466 3467
    entry->collation.set(args[0]->collation.collation, DERIVATION_IMPLICIT);
  collation.set(entry->collation.collation, DERIVATION_IMPLICIT);
3468
  cached_result_type= args[0]->result_type();
unknown's avatar
unknown committed
3469
  return FALSE;
unknown's avatar
unknown committed
3470 3471 3472 3473 3474 3475 3476 3477 3478
}


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;
3479
  collation.set(args[0]->collation.collation, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3480 3481
}

3482

unknown's avatar
unknown committed
3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502
/*
  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
3503
{
unknown's avatar
unknown committed
3504
  if (set_null)
unknown's avatar
unknown committed
3505 3506 3507 3508
  {
    char *pos= (char*) entry+ ALIGN_SIZE(sizeof(user_var_entry));
    if (entry->value && entry->value != pos)
      my_free(entry->value,MYF(0));
3509 3510
    entry->value= 0;
    entry->length= 0;
unknown's avatar
unknown committed
3511 3512 3513
  }
  else
  {
3514 3515
    if (type == STRING_RESULT)
      length++;					// Store strings with end \0
unknown's avatar
unknown committed
3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535 3536
    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
3537
	  return 1;
unknown's avatar
unknown committed
3538 3539
      }
    }
3540 3541 3542 3543 3544
    if (type == STRING_RESULT)
    {
      length--;					// Fix length change above
      entry->value[length]= 0;			// Store end \0
    }
unknown's avatar
unknown committed
3545
    memcpy(entry->value,ptr,length);
unknown's avatar
unknown committed
3546 3547
    if (type == DECIMAL_RESULT)
      ((my_decimal*)entry->value)->fix_buffer_pointer();
unknown's avatar
unknown committed
3548
    entry->length= length;
3549
    entry->collation.set(cs, dv);
unknown's avatar
unknown committed
3550
  }
3551
  entry->type=type;
3552
  return 0;
unknown's avatar
unknown committed
3553
}
unknown's avatar
unknown committed
3554

unknown's avatar
unknown committed
3555 3556 3557 3558 3559

bool
Item_func_set_user_var::update_hash(void *ptr, uint length, Item_result type,
                                    CHARSET_INFO *cs, Derivation dv)
{
3560 3561 3562 3563 3564 3565
  /*
    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
3566 3567 3568 3569 3570 3571 3572 3573
  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
3574 3575 3576
}


3577 3578
/* Get the value of a variable as a double */

unknown's avatar
unknown committed
3579
double user_var_entry::val_real(my_bool *null_value)
unknown's avatar
unknown committed
3580
{
3581 3582 3583 3584 3585 3586 3587 3588
  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
3589 3590 3591 3592 3593 3594
  case DECIMAL_RESULT:
  {
    double result;
    my_decimal2double(E_DEC_FATAL_ERROR, (my_decimal *)value, &result);
    return result;
  }
3595
  case STRING_RESULT:
unknown's avatar
unknown committed
3596
    return my_atof(value);                      // This is null terminated
3597 3598 3599
  case ROW_RESULT:
    DBUG_ASSERT(1);				// Impossible
    break;
unknown's avatar
unknown committed
3600
  }
3601
  return 0.0;					// Impossible
unknown's avatar
unknown committed
3602 3603 3604
}


3605 3606 3607
/* Get the value of a variable as an integer */

longlong user_var_entry::val_int(my_bool *null_value)
unknown's avatar
unknown committed
3608
{
3609 3610 3611 3612 3613 3614 3615 3616
  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
3617 3618 3619
  case DECIMAL_RESULT:
  {
    longlong result;
3620
    my_decimal2int(E_DEC_FATAL_ERROR, (my_decimal *)value, 0, &result);
unknown's avatar
unknown committed
3621 3622
    return result;
  }
unknown's avatar
unknown committed
3623
  case STRING_RESULT:
unknown's avatar
unknown committed
3624 3625 3626 3627
  {
    int error;
    return my_strtoll10(value, (char**) 0, &error);// String is null terminated
  }
3628 3629 3630
  case ROW_RESULT:
    DBUG_ASSERT(1);				// Impossible
    break;
unknown's avatar
unknown committed
3631
  }
3632 3633 3634 3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645
  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
3646
    str->set(*(double*) value, decimals, &my_charset_bin);
3647 3648
    break;
  case INT_RESULT:
unknown's avatar
unknown committed
3649
    str->set(*(longlong*) value, &my_charset_bin);
3650
    break;
unknown's avatar
unknown committed
3651 3652 3653
  case DECIMAL_RESULT:
    my_decimal2string(E_DEC_FATAL_ERROR, (my_decimal *)value, 0, 0, 0, str);
    break;
3654
  case STRING_RESULT:
unknown's avatar
unknown committed
3655
    if (str->copy(value, length, collation.collation))
3656
      str= 0;					// EOM error
3657 3658 3659
  case ROW_RESULT:
    DBUG_ASSERT(1);				// Impossible
    break;
unknown's avatar
unknown committed
3660
  }
3661
  return(str);
unknown's avatar
unknown committed
3662 3663
}

unknown's avatar
unknown committed
3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690
/* 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);
}

3691 3692
/*
  This functions is invoked on SET @variable or @variable:= expression.
unknown's avatar
unknown committed
3693
  Evaluate (and check expression), store results.
3694 3695 3696 3697 3698

  SYNOPSYS
    Item_func_set_user_var::check()

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

  RETURN
unknown's avatar
unknown committed
3703
    FALSE OK.
3704 3705 3706 3707 3708 3709 3710 3711 3712 3713
*/

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

  switch (cached_result_type) {
  case REAL_RESULT:
  {
3714
    save_result.vreal= args[0]->val_real();
3715 3716 3717 3718 3719 3720 3721 3722 3723 3724 3725 3726
    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
3727 3728 3729 3730 3731
  case DECIMAL_RESULT:
  {
    save_result.vdec= args[0]->val_decimal(&decimal_buff);
    break;
  }
3732 3733
  case ROW_RESULT:
  default:
unknown's avatar
unknown committed
3734
    // This case should never be chosen
3735 3736 3737
    DBUG_ASSERT(0);
    break;
  }
unknown's avatar
unknown committed
3738
  DBUG_RETURN(FALSE);
3739 3740
}

3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752

/*
  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
3753
    0	OK
3754 3755 3756 3757
    1	EOM Error

*/

unknown's avatar
unknown committed
3758 3759 3760
bool
Item_func_set_user_var::update()
{
3761 3762 3763 3764
  bool res;
  DBUG_ENTER("Item_func_set_user_var::update");
  LINT_INIT(res);

unknown's avatar
unknown committed
3765 3766
  switch (cached_result_type) {
  case REAL_RESULT:
unknown's avatar
unknown committed
3767
  {
3768
    res= update_hash((void*) &save_result.vreal,sizeof(save_result.vreal),
3769
		     REAL_RESULT, &my_charset_bin, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3770
    break;
unknown's avatar
unknown committed
3771
  }
unknown's avatar
unknown committed
3772
  case INT_RESULT:
3773
  {
3774
    res= update_hash((void*) &save_result.vint, sizeof(save_result.vint),
3775
		     INT_RESULT, &my_charset_bin, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3776
    break;
unknown's avatar
unknown committed
3777
  }
unknown's avatar
unknown committed
3778
  case STRING_RESULT:
unknown's avatar
unknown committed
3779
  {
3780
    if (!save_result.vstr)					// Null value
unknown's avatar
unknown committed
3781
      res= update_hash((void*) 0, 0, STRING_RESULT, &my_charset_bin,
3782
		       DERIVATION_IMPLICIT);
3783
    else
3784 3785 3786
      res= update_hash((void*) save_result.vstr->ptr(),
		       save_result.vstr->length(), STRING_RESULT,
		       save_result.vstr->charset(),
3787
		       DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3788 3789
    break;
  }
unknown's avatar
unknown committed
3790 3791 3792 3793
  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
3794
                       DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3795 3796 3797
    else
      res= update_hash((void*) save_result.vdec,
                       sizeof(my_decimal), DECIMAL_RESULT,
unknown's avatar
unknown committed
3798
                       &my_charset_bin, DERIVATION_IMPLICIT);
unknown's avatar
unknown committed
3799 3800
    break;
  }
3801
  case ROW_RESULT:
unknown's avatar
unknown committed
3802
  default:
unknown's avatar
unknown committed
3803
    // This case should never be chosen
unknown's avatar
unknown committed
3804 3805 3806
    DBUG_ASSERT(0);
    break;
  }
3807
  DBUG_RETURN(res);
unknown's avatar
unknown committed
3808 3809 3810
}


3811
double Item_func_set_user_var::val_real()
unknown's avatar
unknown committed
3812
{
3813
  DBUG_ASSERT(fixed == 1);
3814
  check();
3815
  update();					// Store expression
unknown's avatar
unknown committed
3816
  return entry->val_real(&null_value);
unknown's avatar
unknown committed
3817 3818
}

3819
longlong Item_func_set_user_var::val_int()
unknown's avatar
unknown committed
3820
{
3821
  DBUG_ASSERT(fixed == 1);
3822
  check();
3823 3824
  update();					// Store expression
  return entry->val_int(&null_value);
unknown's avatar
unknown committed
3825 3826
}

3827
String *Item_func_set_user_var::val_str(String *str)
unknown's avatar
unknown committed
3828
{
3829
  DBUG_ASSERT(fixed == 1);
3830
  check();
3831 3832
  update();					// Store expression
  return entry->val_str(&null_value, str, decimals);
unknown's avatar
unknown committed
3833 3834 3835
}


unknown's avatar
unknown committed
3836 3837 3838 3839 3840 3841 3842 3843 3844
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
3845 3846
void Item_func_set_user_var::print(String *str)
{
3847
  str->append(STRING_WITH_LEN("(@"));
3848
  str->append(name.str, name.length);
3849
  str->append(STRING_WITH_LEN(":="));
unknown's avatar
unknown committed
3850 3851 3852 3853 3854
  args[0]->print(str);
  str->append(')');
}


3855 3856
void Item_func_set_user_var::print_as_stmt(String *str)
{
3857
  str->append(STRING_WITH_LEN("set @"));
3858
  str->append(name.str, name.length);
3859
  str->append(STRING_WITH_LEN(":="));
3860 3861 3862 3863 3864
  args[0]->print(str);
  str->append(')');
}


unknown's avatar
unknown committed
3865 3866 3867
String *
Item_func_get_user_var::val_str(String *str)
{
3868
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3869
  DBUG_ENTER("Item_func_get_user_var::val_str");
3870
  if (!var_entry)
3871
    DBUG_RETURN((String*) 0);			// No such variable
3872
  DBUG_RETURN(var_entry->val_str(&null_value, str, decimals));
unknown's avatar
unknown committed
3873 3874 3875
}


3876
double Item_func_get_user_var::val_real()
unknown's avatar
unknown committed
3877
{
3878
  DBUG_ASSERT(fixed == 1);
3879 3880
  if (!var_entry)
    return 0.0;					// No such variable
unknown's avatar
unknown committed
3881 3882 3883 3884 3885 3886 3887 3888 3889 3890
  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
3891 3892 3893 3894 3895
}


longlong Item_func_get_user_var::val_int()
{
3896
  DBUG_ASSERT(fixed == 1);
3897 3898 3899
  if (!var_entry)
    return LL(0);				// No such variable
  return (var_entry->val_int(&null_value));
unknown's avatar
unknown committed
3900 3901 3902
}


3903
/*
3904 3905 3906 3907 3908 3909 3910 3911 3912 3913
  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.

3914 3915 3916 3917
  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).
3918

3919
  RETURN
3920
    0  OK
unknown's avatar
unknown committed
3921
    1  Failed to put appropriate record into binary log
3922

3923 3924
*/

3925 3926
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
3927
{
3928
  BINLOG_USER_VAR_EVENT *user_var_event;
3929 3930
  user_var_entry *var_entry;
  var_entry= get_variable(&thd->user_vars, name, 0);
3931 3932

  if (!(opt_bin_log && is_update_query(sql_command)))
3933 3934 3935 3936
  {
    *out_entry= var_entry;
    return 0;
  }
3937 3938

  if (!var_entry)
unknown's avatar
unknown committed
3939
  {
3940
    /*
3941 3942 3943 3944
      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
3945 3946
      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;'.
3947 3948 3949
      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();
3950 3951
      this would save memory and time; but calling sql_set_variables() makes
      one unique place to maintain (sql_set_variables()). 
3952 3953 3954

      Manipulation with lex is necessary since free_underlaid_joins
      is going to release memory belonging to the main query.
3955 3956 3957
    */

    List<set_var_base> tmp_var_list;
3958 3959 3960
    LEX *sav_lex= thd->lex, lex_tmp;
    thd->lex= &lex_tmp;
    lex_start(thd, NULL, 0);
3961 3962
    tmp_var_list.push_back(new set_var_user(new Item_func_set_user_var(name,
                                                                       new Item_null())));
3963 3964
    /* Create the variable */
    if (sql_set_variables(thd, &tmp_var_list))
3965 3966
    {
      thd->lex= sav_lex;
3967
      goto err;
3968 3969
    }
    thd->lex= sav_lex;
3970 3971 3972
    if (!(var_entry= get_variable(&thd->user_vars, name, 0)))
      goto err;
  }
3973 3974
  else if (var_entry->used_query_id == thd->query_id ||
           mysql_bin_log.is_query_in_union(thd, var_entry->used_query_id))
3975 3976 3977 3978 3979 3980 3981 3982 3983
  {
    /* 
       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;
  }
3984 3985 3986 3987

  uint size;
  /*
    First we need to store value of var_entry, when the next situation
unknown's avatar
unknown committed
3988
    appears:
3989 3990
    > set @a:=1;
    > insert into t1 values (@a), (@a:=@a+1), (@a:=@a+1);
3991 3992 3993 3994 3995 3996
    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.
3997
  */
3998 3999 4000
  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)))
4001 4002 4003 4004 4005 4006 4007 4008 4009 4010 4011 4012
    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
4013
  }
4014 4015 4016 4017 4018 4019 4020 4021 4022 4023
  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;
4024 4025 4026
  
  *out_entry= var_entry;
  return 0;
4027

unknown's avatar
unknown committed
4028
err:
4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040
  *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;
4041

4042
  error= get_var_with_binlog(thd, thd->lex->sql_command, name, &var_entry);
4043 4044

  if (var_entry)
4045
  {
4046
    collation.set(var_entry->collation);
4047 4048 4049
    switch (var_entry->type) {
    case REAL_RESULT:
      max_length= DBL_DIG + 8;
unknown's avatar
unknown committed
4050
      break;
4051 4052
    case INT_RESULT:
      max_length= MAX_BIGINT_WIDTH;
unknown's avatar
unknown committed
4053
      decimals=0;
4054 4055 4056 4057
      break;
    case STRING_RESULT:
      max_length= MAX_BLOB_WIDTH;
      break;
unknown's avatar
unknown committed
4058
    case DECIMAL_RESULT:
unknown's avatar
unknown committed
4059 4060
      max_length= DECIMAL_MAX_STR_LENGTH;
      decimals= DECIMAL_MAX_SCALE;
unknown's avatar
unknown committed
4061
      break;
unknown's avatar
unknown committed
4062
    case ROW_RESULT:                            // Keep compiler happy
unknown's avatar
unknown committed
4063 4064
    default:
      DBUG_ASSERT(0);
unknown's avatar
unknown committed
4065
      break;
4066 4067
    }
  }
4068
  else
4069 4070
  {
    collation.set(&my_charset_bin, DERIVATION_IMPLICIT);
4071
    null_value= 1;
4072
  }
4073

4074 4075
  if (error)
    thd->fatal_error();
4076

unknown's avatar
unknown committed
4077
  return;
unknown's avatar
unknown committed
4078 4079 4080
}


4081
bool Item_func_get_user_var::const_item() const
4082
{
4083
  return (!var_entry || current_thd->query_id != var_entry->update_query_id);
4084
}
unknown's avatar
unknown committed
4085 4086 4087 4088 4089 4090 4091 4092 4093 4094 4095 4096


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
4097 4098 4099

void Item_func_get_user_var::print(String *str)
{
4100
  str->append(STRING_WITH_LEN("(@"));
unknown's avatar
unknown committed
4101 4102 4103 4104
  str->append(name.str,name.length);
  str->append(')');
}

4105

4106
bool Item_func_get_user_var::eq(const Item *item, bool binary_cmp) const
unknown's avatar
unknown committed
4107 4108 4109 4110 4111 4112
{
  /* 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 ||
4113
      ((Item_func*) item)->functype() != functype())
unknown's avatar
unknown committed
4114 4115 4116 4117 4118 4119 4120
    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));
}


4121
bool Item_func_get_user_var::set_value(THD *thd,
4122
                                       sp_rcontext */*ctx*/, Item **it)
4123
{
4124
  Item_func_set_user_var *suv= new Item_func_set_user_var(get_name(), *it);
4125 4126 4127 4128
  /*
    Item_func_set_user_var is not fixed after construction, call
    fix_fields().
  */
4129
  return (!suv || suv->fix_fields(thd, it) || suv->check() || suv->update());
4130 4131 4132
}


4133
bool Item_user_var_as_out_param::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
4134 4135
{
  DBUG_ASSERT(fixed == 0);
4136
  if (Item::fix_fields(thd, ref) ||
unknown's avatar
unknown committed
4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171 4172 4173 4174 4175 4176 4177 4178 4179 4180 4181 4182 4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193 4194 4195 4196 4197 4198 4199 4200 4201 4202
      !(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);
}


4203 4204 4205 4206 4207 4208 4209 4210 4211 4212 4213 4214
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
4215
Item_func_get_system_var::fix_fields(THD *thd, Item **ref)
4216
{
4217
  Item *item;
4218
  DBUG_ENTER("Item_func_get_system_var::fix_fields");
4219

4220 4221 4222 4223 4224
  /*
    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().
  */
4225
  if (!(item= var->item(thd, var_type, &component)))
4226 4227 4228 4229 4230 4231 4232 4233
    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
4234 4235
longlong Item_func_inet_aton::val_int()
{
4236
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4237 4238 4239 4240 4241
  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];
4242
  int dot_count= 0;
unknown's avatar
unknown committed
4243

unknown's avatar
unknown committed
4244
  String *s,tmp(buff,sizeof(buff),&my_charset_bin);
unknown's avatar
unknown committed
4245 4246 4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260
  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 == '.')
    {
4261
      dot_count++;
unknown's avatar
unknown committed
4262 4263 4264 4265 4266 4267 4268
      result= (result << 8) + (ulonglong) byte_result;
      byte_result = 0;
    }
    else
      goto err;					// Invalid character
  }
  if (c != '.')					// IP number can't end on '.'
4269
  {
4270 4271 4272 4273 4274 4275
    /*
      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
4276
    switch (dot_count) {
4277 4278
    case 1: result<<= 8; /* Fall through */
    case 2: result<<= 8; /* Fall through */
4279
    }
unknown's avatar
unknown committed
4280
    return (result << 8) + (ulonglong) byte_result;
4281
  }
unknown's avatar
unknown committed
4282 4283 4284 4285 4286 4287

err:
  null_value=1;
  return 0;
}

unknown's avatar
unknown committed
4288

4289
void Item_func_match::init_search(bool no_order)
4290
{
unknown's avatar
unknown committed
4291
  DBUG_ENTER("Item_func_match::init_search");
4292 4293

  /* Check if init_search() has been called before */
4294
  if (ft_handler)
unknown's avatar
unknown committed
4295
    DBUG_VOID_RETURN;
4296

4297
  if (key == NO_SUCH_KEY)
4298 4299
  {
    List<Item> fields;
4300
    fields.push_back(new Item_string(" ",1, cmp_collation.collation));
4301 4302
    for (uint i=1; i < arg_count; i++)
      fields.push_back(args[i]);
4303
    concat=new Item_func_concat_ws(fields);
unknown's avatar
unknown committed
4304 4305 4306
    /*
      Above function used only to get value and do not need fix_fields for it:
      Item_string - basic constant
unknown's avatar
unknown committed
4307 4308
      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
4309 4310
    */
    concat->quick_fix_field();
4311
  }
4312

4313 4314
  if (master)
  {
4315 4316
    join_key=master->join_key=join_key|master->join_key;
    master->init_search(no_order);
4317 4318
    ft_handler=master->ft_handler;
    join_key=master->join_key;
unknown's avatar
unknown committed
4319
    DBUG_VOID_RETURN;
4320 4321
  }

unknown's avatar
unknown committed
4322
  String *ft_tmp= 0;
4323

4324
  // MATCH ... AGAINST (NULL) is meaningless, but possible
4325
  if (!(ft_tmp=key_item()->val_str(&value)))
4326
  {
4327 4328
    ft_tmp= &value;
    value.set("",0,cmp_collation.collation);
4329 4330
  }

4331 4332
  if (ft_tmp->charset() != cmp_collation.collation)
  {
4333
    uint dummy_errors;
4334
    search_value.copy(ft_tmp->ptr(), ft_tmp->length(), ft_tmp->charset(),
4335
                      cmp_collation.collation, &dummy_errors);
4336
    ft_tmp= &search_value;
4337 4338
  }

4339 4340
  if (join_key && !no_order)
    flags|=FT_SORTED;
4341
  ft_handler=table->file->ft_init_ext(flags, key, ft_tmp);
4342 4343 4344

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

  DBUG_VOID_RETURN;
4347 4348
}

unknown's avatar
unknown committed
4349

4350
bool Item_func_match::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
4351
{
4352
  DBUG_ASSERT(fixed == 0);
unknown's avatar
unknown committed
4353
  Item *item;
4354
  LINT_INIT(item);				// Safe as arg_count is > 1
unknown's avatar
unknown committed
4355

4356 4357 4358
  maybe_null=1;
  join_key=0;

4359 4360 4361 4362 4363
  /*
    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.
4364
   */
4365
  if (Item_func::fix_fields(thd, ref) ||
4366
      !args[0]->const_during_execution())
4367 4368
  {
    my_error(ER_WRONG_ARGUMENTS,MYF(0),"AGAINST");
unknown's avatar
unknown committed
4369
    return TRUE;
4370
  }
unknown's avatar
unknown committed
4371

4372 4373
  const_item_cache=0;
  for (uint i=1 ; i < arg_count ; i++)
unknown's avatar
unknown committed
4374
  {
4375
    item=args[i];
unknown's avatar
unknown committed
4376
    if (item->type() == Item::REF_ITEM)
4377 4378
      args[i]= item= *((Item_ref *)item)->ref;
    if (item->type() != Item::FIELD_ITEM)
4379
      key=NO_SUCH_KEY;
unknown's avatar
unknown committed
4380
  }
4381 4382
  /*
    Check that all columns come from the same table.
4383
    We've already checked that columns in MATCH are fields so
4384 4385 4386
    PARAM_TABLE_BIT can only appear from AGAINST argument.
  */
  if ((used_tables_cache & ~PARAM_TABLE_BIT) != item->used_tables())
4387
    key=NO_SUCH_KEY;
4388

4389
  if (key == NO_SUCH_KEY && !(flags & FT_BOOL))
4390 4391
  {
    my_error(ER_WRONG_ARGUMENTS,MYF(0),"MATCH");
unknown's avatar
unknown committed
4392
    return TRUE;
4393
  }
4394
  table=((Item_field *)item)->field->table;
unknown's avatar
unknown committed
4395
  if (!(table->file->table_flags() & HA_CAN_FULLTEXT))
4396
  {
unknown's avatar
unknown committed
4397
    my_error(ER_TABLE_CANT_HANDLE_FT, MYF(0));
unknown's avatar
unknown committed
4398
    return 1;
4399
  }
4400
  table->fulltext_searched=1;
4401 4402
  return agg_arg_collations_for_comparison(cmp_collation,
                                           args+1, arg_count-1, 0);
unknown's avatar
unknown committed
4403
}
4404

unknown's avatar
unknown committed
4405 4406 4407
bool Item_func_match::fix_index()
{
  Item_field *item;
4408
  uint ft_to_key[MAX_KEY], ft_cnt[MAX_KEY], fts=0, keynr;
4409
  uint max_cnt=0, mkeys=0, i;
4410

4411
  if (key == NO_SUCH_KEY)
4412
    return 0;
4413 4414 4415
  
  if (!table) 
    goto err;
unknown's avatar
unknown committed
4416

4417
  for (keynr=0 ; keynr < table->s->keys ; keynr++)
unknown's avatar
unknown committed
4418
  {
4419
    if ((table->key_info[keynr].flags & HA_FULLTEXT) &&
4420
        (table->keys_in_use_for_query.is_set(keynr)))
unknown's avatar
unknown committed
4421
    {
4422
      ft_to_key[fts]=keynr;
unknown's avatar
unknown committed
4423 4424 4425 4426 4427 4428
      ft_cnt[fts]=0;
      fts++;
    }
  }

  if (!fts)
4429
    goto err;
unknown's avatar
unknown committed
4430

4431
  for (i=1; i < arg_count; i++)
unknown's avatar
unknown committed
4432
  {
4433
    item=(Item_field*)args[i];
4434
    for (keynr=0 ; keynr < fts ; keynr++)
unknown's avatar
unknown committed
4435
    {
4436
      KEY *ft_key=&table->key_info[ft_to_key[keynr]];
unknown's avatar
unknown committed
4437 4438 4439 4440 4441
      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))
4442
	  ft_cnt[keynr]++;
unknown's avatar
unknown committed
4443 4444 4445 4446
      }
    }
  }

4447
  for (keynr=0 ; keynr < fts ; keynr++)
unknown's avatar
unknown committed
4448
  {
4449
    if (ft_cnt[keynr] > max_cnt)
unknown's avatar
unknown committed
4450
    {
4451
      mkeys=0;
4452 4453
      max_cnt=ft_cnt[mkeys]=ft_cnt[keynr];
      ft_to_key[mkeys]=ft_to_key[keynr];
4454 4455
      continue;
    }
4456
    if (max_cnt && ft_cnt[keynr] == max_cnt)
4457 4458
    {
      mkeys++;
4459 4460
      ft_cnt[mkeys]=ft_cnt[keynr];
      ft_to_key[mkeys]=ft_to_key[keynr];
4461
      continue;
unknown's avatar
unknown committed
4462 4463 4464
    }
  }

4465
  for (keynr=0 ; keynr <= mkeys ; keynr++)
unknown's avatar
unknown committed
4466
  {
4467 4468
    // partial keys doesn't work
    if (max_cnt < arg_count-1 ||
4469
        max_cnt < table->key_info[ft_to_key[keynr]].key_parts)
4470
      continue;
unknown's avatar
unknown committed
4471

4472
    key=ft_to_key[keynr];
4473

4474 4475 4476
    return 0;
  }

4477
err:
4478
  if (flags & FT_BOOL)
4479
  {
4480
    key=NO_SUCH_KEY;
4481 4482
    return 0;
  }
unknown's avatar
unknown committed
4483 4484
  my_message(ER_FT_MATCHING_KEY_NOT_FOUND,
             ER(ER_FT_MATCHING_KEY_NOT_FOUND), MYF(0));
4485
  return 1;
4486 4487
}

4488

4489
bool Item_func_match::eq(const Item *item, bool binary_cmp) const
4490
{
unknown's avatar
unknown committed
4491 4492
  if (item->type() != FUNC_ITEM ||
      ((Item_func*)item)->functype() != FT_FUNC ||
4493
      flags != ((Item_func_match*)item)->flags)
4494 4495 4496 4497 4498
    return 0;

  Item_func_match *ifm=(Item_func_match*) item;

  if (key == ifm->key && table == ifm->table &&
4499
      key_item()->eq(ifm->key_item(), binary_cmp))
4500
    return 1;
unknown's avatar
unknown committed
4501 4502 4503 4504

  return 0;
}

4505

4506
double Item_func_match::val_real()
unknown's avatar
unknown committed
4507
{
4508
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4509
  DBUG_ENTER("Item_func_match::val");
unknown's avatar
unknown committed
4510
  if (ft_handler == NULL)
unknown's avatar
unknown committed
4511
    DBUG_RETURN(-1.0);
unknown's avatar
unknown committed
4512

4513
  if (key != NO_SUCH_KEY && table->null_row) /* NULL row from an outer join */
unknown's avatar
unknown committed
4514 4515
    return 0.0;

unknown's avatar
unknown committed
4516 4517 4518
  if (join_key)
  {
    if (table->file->ft_handler)
unknown's avatar
unknown committed
4519
      DBUG_RETURN(ft_handler->please->get_relevance(ft_handler));
unknown's avatar
unknown committed
4520 4521 4522
    join_key=0;
  }

4523
  if (key == NO_SUCH_KEY)
unknown's avatar
unknown committed
4524
  {
4525 4526
    String *a= concat->val_str(&value);
    if ((null_value= (a == 0)))
unknown's avatar
unknown committed
4527 4528 4529
      DBUG_RETURN(0);
    DBUG_RETURN(ft_handler->please->find_relevance(ft_handler,
				      (byte *)a->ptr(), a->length()));
unknown's avatar
unknown committed
4530 4531
  }
  else
4532 4533
    DBUG_RETURN(ft_handler->please->find_relevance(ft_handler,
                                                   table->record[0], 0));
unknown's avatar
unknown committed
4534 4535
}

4536 4537
void Item_func_match::print(String *str)
{
4538
  str->append(STRING_WITH_LEN("(match "));
unknown's avatar
unknown committed
4539
  print_args(str, 1);
4540
  str->append(STRING_WITH_LEN(" against ("));
4541
  args[0]->print(str);
unknown's avatar
unknown committed
4542
  if (flags & FT_BOOL)
4543
    str->append(STRING_WITH_LEN(" in boolean mode"));
unknown's avatar
unknown committed
4544
  else if (flags & FT_EXPAND)
4545 4546
    str->append(STRING_WITH_LEN(" with query expansion"));
  str->append(STRING_WITH_LEN("))"));
4547
}
unknown's avatar
unknown committed
4548

unknown's avatar
unknown committed
4549 4550
longlong Item_func_bit_xor::val_int()
{
4551
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4552 4553
  ulonglong arg1= (ulonglong) args[0]->val_int();
  ulonglong arg2= (ulonglong) args[1]->val_int();
4554
  if ((null_value= (args[0]->null_value || args[1]->null_value)))
unknown's avatar
unknown committed
4555 4556 4557 4558
    return 0;
  return (longlong) (arg1 ^ arg2);
}

4559

4560 4561 4562 4563
/***************************************************************************
  System variables
****************************************************************************/

4564 4565 4566 4567 4568 4569 4570 4571 4572 4573 4574 4575 4576 4577 4578 4579 4580
/*
  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
*/
4581

4582 4583 4584

Item *get_system_var(THD *thd, enum_var_type var_type, LEX_STRING name,
		     LEX_STRING component)
4585
{
4586 4587 4588
  sys_var *var;
  LEX_STRING *base_name, *component_name;

4589 4590 4591 4592 4593 4594 4595 4596 4597 4598
  if (component.str)
  {
    base_name= &component;
    component_name= &name;
  }
  else
  {
    base_name= &name;
    component_name= &component;			// Empty string
  }
unknown's avatar
unknown committed
4599

4600
  if (!(var= find_sys_var(base_name->str, base_name->length)))
unknown's avatar
unknown committed
4601
    return 0;
4602 4603 4604 4605
  if (component.str)
  {
    if (!var->is_struct())
    {
4606
      my_error(ER_VARIABLE_IS_NOT_STRUCT, MYF(0), base_name->str);
4607 4608 4609
      return 0;
    }
  }
unknown's avatar
unknown committed
4610
  thd->lex->uncacheable(UNCACHEABLE_SIDEEFFECT);
4611

4612 4613
  set_if_smaller(component_name->length, MAX_SYS_VAR_LENGTH);

4614
  return new Item_func_get_system_var(var, var_type, component_name,
4615
                                      NULL, 0);
4616 4617 4618
}


unknown's avatar
unknown committed
4619 4620
/*
  Check a user level lock.
4621 4622 4623 4624 4625 4626 4627 4628

  SYNOPSIS:
    val_int()

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

4631
longlong Item_func_is_free_lock::val_int()
unknown's avatar
unknown committed
4632
{
4633
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4634
  String *res=args[0]->val_str(&value);
4635
  User_level_lock *ull;
unknown's avatar
unknown committed
4636 4637

  null_value=0;
4638
  if (!res || !res->length())
unknown's avatar
unknown committed
4639 4640 4641 4642 4643 4644
  {
    null_value=1;
    return 0;
  }
  
  pthread_mutex_lock(&LOCK_user_locks);
4645
  ull= (User_level_lock *) hash_search(&hash_user_locks, (byte*) res->ptr(),
unknown's avatar
unknown committed
4646 4647 4648 4649 4650 4651
			  res->length());
  pthread_mutex_unlock(&LOCK_user_locks);
  if (!ull || !ull->locked)
    return 1;
  return 0;
}
unknown's avatar
unknown committed
4652

unknown's avatar
SCRUM  
unknown committed
4653 4654
longlong Item_func_is_used_lock::val_int()
{
4655
  DBUG_ASSERT(fixed == 1);
unknown's avatar
SCRUM  
unknown committed
4656
  String *res=args[0]->val_str(&value);
4657
  User_level_lock *ull;
unknown's avatar
SCRUM  
unknown committed
4658 4659 4660 4661 4662 4663

  null_value=1;
  if (!res || !res->length())
    return 0;
  
  pthread_mutex_lock(&LOCK_user_locks);
4664
  ull= (User_level_lock *) hash_search(&hash_user_locks, (byte*) res->ptr(),
unknown's avatar
SCRUM  
unknown committed
4665 4666 4667 4668 4669 4670 4671 4672 4673
			  res->length());
  pthread_mutex_unlock(&LOCK_user_locks);
  if (!ull || !ull->locked)
    return 0;

  null_value=0;
  return ull->thread_id;
}

4674

4675 4676 4677 4678 4679 4680 4681 4682 4683
longlong Item_func_row_count::val_int()
{
  DBUG_ASSERT(fixed == 1);
  THD *thd= current_thd;

  return thd->row_count_func;
}


4684 4685 4686
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)
4687
{
unknown's avatar
unknown committed
4688
  maybe_null= 1;
4689
  m_name->init_qname(current_thd);
unknown's avatar
unknown committed
4690
  dummy_table= (TABLE*) sql_calloc(sizeof(TABLE));
4691 4692
}

unknown's avatar
unknown committed
4693

4694 4695 4696 4697
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)
4698
{
unknown's avatar
unknown committed
4699
  maybe_null= 1;
4700
  m_name->init_qname(current_thd);
unknown's avatar
unknown committed
4701
  dummy_table= (TABLE*) sql_calloc(sizeof(TABLE));
4702 4703
}

4704 4705 4706 4707 4708 4709 4710 4711
void
Item_func_sp::cleanup()
{
  if (result_field)
  {
    delete result_field;
    result_field= NULL;
  }
4712
  m_sp= NULL;
4713 4714
  Item_func::cleanup();
}
unknown's avatar
unknown committed
4715

4716 4717 4718
const char *
Item_func_sp::func_name() const
{
4719
  THD *thd= current_thd;
unknown's avatar
unknown committed
4720
  /* Calculate length to avoid reallocation of string for sure */
4721 4722 4723 4724
  uint len= ((m_name->m_db.length +
              m_name->m_name.length)*2 + //characters*quoting
             2 +                         // ` and `
             1 +                         // .
4725 4726
             1 +                         // end of string
             ALIGN_SIZE(1));             // to avoid String reallocation
unknown's avatar
unknown committed
4727
  String qname((char *)alloc_root(thd->mem_root, len), len,
4728
               system_charset_info);
4729

4730 4731 4732 4733 4734
  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();
4735 4736
}

4737

unknown's avatar
unknown committed
4738 4739 4740
Field *
Item_func_sp::sp_result_field(void) const
{
unknown's avatar
unknown committed
4741
  Field *field;
unknown's avatar
unknown committed
4742
  DBUG_ENTER("Item_func_sp::sp_result_field");
4743 4744 4745 4746
  DBUG_PRINT("info", ("sp: %s, flags: %x, level: %lu",
                      (m_sp ? "YES" : "NO"),
                      (m_sp ? m_sp->m_flags : (uint)0),
                      (m_sp ? m_sp->m_recursion_level : (ulong)0)));
unknown's avatar
unknown committed
4747 4748

  if (!m_sp)
unknown's avatar
unknown committed
4749
  {
4750 4751 4752
    THD *thd= current_thd;
    if (!(m_sp= sp_find_routine(thd, TYPE_ENUM_FUNCTION, m_name,
                                &thd->sp_func_cache, TRUE)))
unknown's avatar
unknown committed
4753
    {
unknown's avatar
unknown committed
4754 4755
      my_error(ER_SP_DOES_NOT_EXIST, MYF(0), "FUNCTION", m_name->m_qname.str);
      DBUG_RETURN(0);
unknown's avatar
unknown committed
4756 4757
    }
  }
unknown's avatar
unknown committed
4758 4759 4760 4761
  if (!dummy_table->s)
  {
    char *empty_name= (char *) "";
    TABLE_SHARE *share;
4762
    dummy_table->s= share= &dummy_table->share_not_to_be_used;
unknown's avatar
unknown committed
4763 4764 4765
    dummy_table->alias = empty_name;
    dummy_table->maybe_null = maybe_null;
    dummy_table->in_use= current_thd;
4766
    dummy_table->copy_blobs= TRUE;
unknown's avatar
unknown committed
4767 4768 4769
    share->table_cache_key = empty_name;
    share->table_name = empty_name;
  }
4770 4771 4772
  if (!(field= m_sp->create_result_field(max_length, name, dummy_table)))
    my_message(ER_OUT_OF_RESOURCES, ER(ER_OUT_OF_RESOURCES), MYF(0));

4773
  DBUG_RETURN(field);
unknown's avatar
unknown committed
4774 4775 4776
}


unknown's avatar
unknown committed
4777 4778 4779 4780 4781 4782 4783 4784
/*
  Execute function & store value in field

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

4785
bool
4786 4787
Item_func_sp::execute(Field **flp)
{
4788
  THD *thd= current_thd;
4789
  Field *f;
4790 4791 4792 4793 4794 4795

  /*
    Get field in virtual tmp table to store result. Create the field if
    invoked first time.
  */
  
4796 4797
  if (!(f= *flp))
  {
4798 4799
    if (!(*flp= f= sp_result_field()))
    {
4800 4801 4802
      /* Error set by sp_result_field() */
      null_value= 1;
      return TRUE;
4803 4804 4805
    }

    f->move_field((f->pack_length() > sizeof(result_buf)) ?
4806 4807 4808 4809
                  sql_alloc(f->pack_length()) : result_buf);
    f->null_ptr= (uchar *)&null_value;
    f->null_bit= 1;
  }
4810 4811 4812 4813 4814 4815 4816 4817 4818 4819 4820 4821 4822 4823 4824

  /* Execute function and store the return value in the field. */

  if (execute_impl(thd, f))
  {
    null_value= 1;
    context->process_error(thd);
    return TRUE;
  }

  /* Check that the field (the value) is not NULL. */

  null_value= f->is_null();

  return null_value;
4825 4826 4827
}


4828 4829
bool
Item_func_sp::execute_impl(THD *thd, Field *return_value_fld)
4830
{
4831
  bool err_status= TRUE;
4832
  Sub_statement_state statement_state;
unknown's avatar
unknown committed
4833
  Security_context *save_security_ctx= thd->security_ctx, *save_ctx_func;
4834

4835 4836
  DBUG_ENTER("Item_func_sp::execute_impl");

4837 4838 4839 4840 4841 4842 4843 4844
#ifndef NO_EMBEDDED_ACCESS_CHECKS
  if (context->security_ctx)
  {
    /* Set view definer security context */
    thd->security_ctx= context->security_ctx;
  }
#endif
  if (find_and_check_access(thd, EXECUTE_ACL, &save_ctx_func))
4845
    goto error;
4846

4847 4848 4849 4850 4851
  /*
    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.
  */
4852
  thd->reset_sub_statement_state(&statement_state, SUB_STMT_FUNCTION);
4853
  err_status= m_sp->execute_function(thd, args, arg_count, return_value_fld);
4854
  thd->restore_sub_statement_state(&statement_state);
unknown's avatar
unknown committed
4855

4856 4857
#ifndef NO_EMBEDDED_ACCESS_CHECKS
  sp_restore_security_context(thd, save_ctx_func);
4858
error:
unknown's avatar
unknown committed
4859
  thd->security_ctx= save_security_ctx;
4860 4861 4862
#else
error:
#endif
4863
  DBUG_RETURN(err_status);
4864 4865
}

4866

unknown's avatar
unknown committed
4867 4868 4869 4870 4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881 4882
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;
}


4883 4884
enum enum_field_types
Item_func_sp::field_type() const
4885
{
unknown's avatar
unknown committed
4886
  Field *field;
4887
  DBUG_ENTER("Item_func_sp::field_type");
4888

4889 4890
  if (result_field)
    DBUG_RETURN(result_field->type());
unknown's avatar
unknown committed
4891
  if ((field= sp_result_field()))
4892
  {
unknown's avatar
unknown committed
4893 4894 4895
    enum_field_types result= field->type();
    delete field;
    DBUG_RETURN(result);
4896
  }
4897
  DBUG_RETURN(MYSQL_TYPE_VARCHAR);
4898 4899
}

4900

4901 4902
Item_result
Item_func_sp::result_type() const
4903
{
unknown's avatar
unknown committed
4904
  Field *field;
4905 4906
  DBUG_ENTER("Item_func_sp::result_type");
  DBUG_PRINT("info", ("m_sp = %p", m_sp));
4907

4908 4909
  if (result_field)
    DBUG_RETURN(result_field->result_type());
unknown's avatar
unknown committed
4910
  if ((field= sp_result_field()))
4911
  {
unknown's avatar
unknown committed
4912 4913 4914
    Item_result result= field->result_type();
    delete field;
    DBUG_RETURN(result);
4915
  }
4916
  DBUG_RETURN(STRING_RESULT);
4917 4918 4919 4920 4921
}

void
Item_func_sp::fix_length_and_dec()
{
unknown's avatar
unknown committed
4922
  Field *field;
4923 4924
  DBUG_ENTER("Item_func_sp::fix_length_and_dec");

4925 4926 4927
  if (result_field)
  {
    decimals= result_field->decimals();
unknown's avatar
unknown committed
4928
    max_length= result_field->field_length;
4929
    collation.set(result_field->charset());
4930 4931 4932
    DBUG_VOID_RETURN;
  }

unknown's avatar
unknown committed
4933
  if (!(field= sp_result_field()))
4934 4935
  {
    context->process_error(current_thd);
unknown's avatar
unknown committed
4936
    DBUG_VOID_RETURN;
4937
  }
unknown's avatar
unknown committed
4938
  decimals= field->decimals();
unknown's avatar
merge  
unknown committed
4939
  max_length= field->field_length;
4940
  collation.set(field->charset());
unknown's avatar
merge  
unknown committed
4941
  maybe_null= 1;
unknown's avatar
unknown committed
4942
  delete field;
4943 4944
  DBUG_VOID_RETURN;
}
unknown's avatar
unknown committed
4945 4946


4947 4948 4949
longlong Item_func_found_rows::val_int()
{
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
4950
  return current_thd->found_rows();
4951
}
unknown's avatar
unknown committed
4952

unknown's avatar
unknown committed
4953

unknown's avatar
unknown committed
4954 4955 4956
Field *
Item_func_sp::tmp_table_field(TABLE *t_arg)
{
4957
  Field *field= 0;
unknown's avatar
unknown committed
4958 4959 4960
  DBUG_ENTER("Item_func_sp::tmp_table_field");

  if (m_sp)
4961
    field= m_sp->create_result_field(max_length, (const char*) name, t_arg);
unknown's avatar
unknown committed
4962
  
4963 4964
  if (!field) 
    field= Item_func::tmp_table_field(t_arg);
unknown's avatar
unknown committed
4965

4966
  if (!field)
4967 4968
    my_message(ER_OUT_OF_RESOURCES, ER(ER_OUT_OF_RESOURCES), MYF(0));

4969
  DBUG_RETURN(field);
unknown's avatar
unknown committed
4970
}
4971

unknown's avatar
unknown committed
4972

4973
/*
4974
  Find the function and check access rights to the function
4975 4976

  SYNOPSIS
unknown's avatar
unknown committed
4977 4978
    find_and_check_access()
    thd           thread handler
4979
    want_access   requested access
unknown's avatar
unknown committed
4980
    save          backup of security context
4981 4982

  RETURN
unknown's avatar
unknown committed
4983 4984
    FALSE    Access granted
    TRUE     Requested access can't be granted or function doesn't exists
unknown's avatar
unknown committed
4985
	     In this case security context is not changed and *save = 0
4986 4987 4988 4989 4990

  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
4991
    error is raised.
4992 4993
    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
4994 4995
    In case of access error or if context is not provided then
    find_and_check_access() switches back to previous security context.
4996
*/
unknown's avatar
unknown committed
4997

4998
bool
unknown's avatar
unknown committed
4999
Item_func_sp::find_and_check_access(THD *thd, ulong want_access,
unknown's avatar
unknown committed
5000
                                    Security_context **save)
5001
{
unknown's avatar
unknown committed
5002 5003 5004
  bool res= TRUE;

  *save= 0;                                     // Safety if error
5005 5006
  if (! m_sp && ! (m_sp= sp_find_routine(thd, TYPE_ENUM_FUNCTION, m_name,
                                         &thd->sp_func_cache, TRUE)))
5007 5008 5009 5010 5011
  {
    my_error(ER_SP_DOES_NOT_EXIST, MYF(0), "FUNCTION", m_name->m_qname.str);
    goto error;
  }

unknown's avatar
unknown committed
5012
#ifndef NO_EMBEDDED_ACCESS_CHECKS
5013
  if (check_routine_access(thd, want_access,
unknown's avatar
unknown committed
5014
			   m_sp->m_db.str, m_sp->m_name.str, 0, FALSE))
5015 5016
    goto error;

unknown's avatar
unknown committed
5017
  sp_change_security_context(thd, m_sp, save);
unknown's avatar
unknown committed
5018 5019 5020 5021 5022 5023 5024 5025 5026 5027
  /*
    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
5028
  if (*save &&
5029
      check_routine_access(thd, want_access,
unknown's avatar
unknown committed
5030
			   m_sp->m_db.str, m_sp->m_name.str, 0, FALSE))
5031
  {
unknown's avatar
unknown committed
5032 5033 5034
    sp_restore_security_context(thd, *save);
    *save= 0;                                   // Safety
    goto error;
5035 5036
  }
#endif
unknown's avatar
unknown committed
5037 5038 5039
  res= FALSE;                                   // no error

error:
5040
  return res;
unknown's avatar
unknown committed
5041
}
5042 5043 5044 5045 5046 5047 5048

bool
Item_func_sp::fix_fields(THD *thd, Item **ref)
{
  bool res;
  DBUG_ASSERT(fixed == 0);
  res= Item_func::fix_fields(thd, ref);
5049
  if (!res && thd->lex->view_prepare_mode)
unknown's avatar
unknown committed
5050
  {
5051 5052 5053 5054 5055 5056 5057 5058 5059 5060
    /*
      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
5061
      procedure has SQL SECURITY DEFINER
5062
    */
unknown's avatar
unknown committed
5063 5064 5065 5066
    Security_context *save_ctx;
    if (!(res= find_and_check_access(thd, EXECUTE_ACL, &save_ctx)))
      sp_restore_security_context(thd, save_ctx);
  }
5067 5068
  return res;
}