item_func.cc 122 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
}

197 198 199

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

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

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

unknown's avatar
unknown committed
238

unknown's avatar
unknown committed
239

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


279 280
/* See comments in Item_cmp_func::split_sum_func() */

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


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

308 309 310 311 312 313 314

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


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


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

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

367

unknown's avatar
unknown committed
368
Field *Item_func::tmp_table_field(TABLE *table)
369
{
unknown's avatar
unknown committed
370 371
  Field *field;
  LINT_INIT(field);
372

unknown's avatar
unknown committed
373
  switch (result_type()) {
374 375
  case INT_RESULT:
    if (max_length > 11)
unknown's avatar
unknown committed
376
      field= new Field_longlong(max_length, maybe_null, name, unsigned_flag);
377
    else
unknown's avatar
unknown committed
378
      field= new Field_long(max_length, maybe_null, name, unsigned_flag);
379 380
    break;
  case REAL_RESULT:
unknown's avatar
unknown committed
381
    field= new Field_double(max_length, maybe_null, name, decimals);
382 383
    break;
  case STRING_RESULT:
unknown's avatar
unknown committed
384
    return make_string_field(table);
385
    break;
unknown's avatar
unknown committed
386
  case DECIMAL_RESULT:
unknown's avatar
unknown committed
387 388 389 390
    field= new Field_new_decimal(my_decimal_precision_to_length(decimal_precision(),
                                                                decimals,
                                                                unsigned_flag),
                                 maybe_null, name, 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
    DBUG_ASSERT(0);
unknown's avatar
unknown committed
396
    field= 0;
unknown's avatar
unknown committed
397
    break;
398
  }
unknown's avatar
unknown committed
399 400 401
  if (field)
    field->init(table);
  return field;
402 403
}

unknown's avatar
unknown committed
404

unknown's avatar
unknown committed
405 406 407 408 409 410 411
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;
}

412

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


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


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


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

unknown's avatar
unknown committed
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 532 533 534 535

/*
  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
536 537 538 539 540 541 542 543 544 545
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;
}


546
Item *Item_func::get_tmp_table_item(THD *thd)
547 548 549
{
  if (!with_sum_func && !const_item())
    return new Item_field(result_field);
550
  return copy_or_same(thd);
551 552
}

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

unknown's avatar
unknown committed
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 588
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;
}


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

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

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

unknown's avatar
unknown committed
632 633

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

  SYNOPSIS
    Item_func_num1::find_num_type()
*/
641

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

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

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


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


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

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

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


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

}


835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 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
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;
}


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

}


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


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


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


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

unknown's avatar
unknown committed
975 976

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


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

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

1034 1035 1036 1037 1038 1039 1040 1041 1042 1043

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


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

unknown's avatar
unknown committed
1057 1058

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

1066

1067 1068
/* See Item_func_plus::decimal_op for comments */

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

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

1085

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

unknown's avatar
unknown committed
1095 1096

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


1106 1107
/* See Item_func_plus::decimal_op for comments */

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


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


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

unknown's avatar
unknown committed
1153

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

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


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

unknown's avatar
unknown committed
1192

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

1223 1224 1225 1226

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


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


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

unknown's avatar
unknown committed
1281 1282

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

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

unknown's avatar
unknown committed
1306 1307

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


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

unknown's avatar
unknown committed
1321

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

unknown's avatar
unknown committed
1329

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

1342

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

unknown's avatar
unknown committed
1350

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

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

unknown's avatar
unknown committed
1377

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

unknown's avatar
unknown committed
1385

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

unknown's avatar
unknown committed
1393

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


void Item_func_abs::fix_length_and_dec()
{
  Item_func_num1::fix_length_and_dec();
unknown's avatar
unknown committed
1411 1412
}

unknown's avatar
unknown committed
1413

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

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

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

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

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

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

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

// Trigonometric functions

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

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

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

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

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

1573
double Item_func_tan::val_real()
unknown's avatar
unknown committed
1574
{
1575
  DBUG_ASSERT(fixed == 1);
1576
  double value= args[0]->val_real();
unknown's avatar
unknown committed
1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587
  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()
{
1588
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602
  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()
{
1603
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618
  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()
{
1619
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636
  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
1637
void Item_func_int_val::fix_num_length_and_dec()
unknown's avatar
unknown committed
1638
{
unknown's avatar
unknown committed
1639 1640 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
  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()
{
1691 1692 1693 1694 1695 1696 1697 1698 1699
  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;
1700
    if ((dec= Item_func_ceiling::decimal_op(&dec_buf)))
1701 1702 1703 1704 1705 1706
      my_decimal2int(E_DEC_FATAL_ERROR, dec, unsigned_flag, &result);
    else
      result= 0;
    break;
  }
  default:
1707
    result= (longlong)Item_func_ceiling::real_op();
1708 1709
  };
  return result;
unknown's avatar
unknown committed
1710 1711
}

unknown's avatar
unknown committed
1712 1713

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


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

unknown's avatar
unknown committed
1759 1760

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


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

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

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

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

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


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


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


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

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

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

1995

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

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


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

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


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


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


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

unknown's avatar
unknown committed
2150 2151 2152 2153

my_decimal *Item_func_min_max::val_decimal(my_decimal *dec)
{
  DBUG_ASSERT(fixed == 1);
2154 2155 2156
  my_decimal tmp_buf, *tmp, *res;
  LINT_INIT(res);

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


unknown's avatar
unknown committed
2186 2187
longlong Item_func_length::val_int()
{
2188
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2189 2190 2191 2192 2193 2194 2195 2196 2197 2198
  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
2199

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

unknown's avatar
unknown committed
2213

unknown's avatar
unknown committed
2214 2215
longlong Item_func_coercibility::val_int()
{
2216
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2217
  null_value= 0;
2218
  return (longlong) args[0]->collation.derivation;
unknown's avatar
unknown committed
2219
}
unknown's avatar
unknown committed
2220

unknown's avatar
unknown committed
2221

2222 2223 2224
void Item_func_locate::fix_length_and_dec()
{
  maybe_null=0; max_length=11;
unknown's avatar
unknown committed
2225
  agg_arg_charsets(cmp_collation, args, 2, MY_COLL_CMP_CONV);
2226 2227
}

unknown's avatar
unknown committed
2228

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

unknown's avatar
unknown committed
2244 2245
  if (arg_count == 3)
  {
2246 2247 2248
    start0= start =(uint) args[2]->val_int()-1;
    start=a->charpos(start);
    
unknown's avatar
unknown committed
2249 2250 2251
    if (start > a->length() || start+b->length() > a->length())
      return 0;
  }
2252

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


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

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

unknown's avatar
unknown committed
2334

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

2345

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

static const char separator=',';

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

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


/****************************************************************************
** Functions to handle dynamic loadable functions
** Original source by: Alexis Mikhailov <root@medinf.chuvashia.su>
2495
** Rewritten by monty.
unknown's avatar
unknown committed
2496 2497 2498 2499
****************************************************************************/

#ifdef HAVE_DLOPEN

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


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

2531
  if (check_stack_overrun(thd, STACK_MIN_SIZE, buff))
unknown's avatar
unknown committed
2532
    DBUG_RETURN(TRUE);				// Fatal error flag is set!
unknown's avatar
unknown committed
2533

2534
  udf_func *tmp_udf=find_udf(u_d->name.str,(uint) u_d->name.length,1);
unknown's avatar
unknown committed
2535 2536 2537

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

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

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


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

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

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

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


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

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

unknown's avatar
unknown committed
2798

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

unknown's avatar
unknown committed
2805

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


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


String *Item_func_udf_int::val_str(String *str)
{
2837
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2838 2839 2840
  longlong nr=val_int();
  if (null_value)
    return 0;
2841
  str->set_int(nr, unsigned_flag, &my_charset_bin);
unknown's avatar
unknown committed
2842 2843 2844
  return str;
}

unknown's avatar
unknown committed
2845 2846 2847 2848

longlong Item_func_udf_decimal::val_int()
{
  my_decimal dec_buf, *dec= udf.val_decimal(&null_value, &dec_buf);
2849
  longlong result;
unknown's avatar
unknown committed
2850 2851 2852 2853 2854 2855 2856 2857 2858 2859
  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);
2860
  double result;
unknown's avatar
unknown committed
2861 2862 2863 2864 2865 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
  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
2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910
/* 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)
{
2911
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
2912 2913 2914 2915 2916
  String *res=udf.val_str(str,&str_value);
  null_value = !res;
  return res;
}

2917 2918

/*
2919 2920
   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.)
2921 2922 2923 2924
 */

udf_handler::~udf_handler()
{
2925 2926
  /* Everything should be properly cleaned up by this moment. */
  DBUG_ASSERT(not_original || !(initialized || buffers));
2927 2928
}

unknown's avatar
unknown committed
2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939
#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;

2940
class User_level_lock
unknown's avatar
unknown committed
2941 2942 2943 2944 2945 2946 2947 2948 2949
{
  char *key;
  uint key_length;

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

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

2981
char *ull_get_key(const User_level_lock *ull, uint *length,
unknown's avatar
unknown committed
2982 2983 2984 2985 2986 2987
		  my_bool not_used __attribute__((unused)))
{
  *length=(uint) ull->key_length;
  return (char*) ull->key;
}

2988 2989 2990

static bool item_user_lock_inited= 0;

unknown's avatar
unknown committed
2991 2992
void item_user_lock_init(void)
{
2993
  pthread_mutex_init(&LOCK_user_locks,MY_MUTEX_INIT_SLOW);
unknown's avatar
unknown committed
2994 2995
  hash_init(&hash_user_locks,system_charset_info,
	    16,0,0,(hash_get_key) ull_get_key,NULL,0);
2996
  item_user_lock_inited= 1;
unknown's avatar
unknown committed
2997 2998 2999 3000
}

void item_user_lock_free(void)
{
3001 3002 3003 3004 3005 3006
  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
3007 3008
}

3009
void item_user_lock_release(User_level_lock *ull)
unknown's avatar
unknown committed
3010 3011
{
  ull->locked=0;
3012
  ull->thread_id= 0;
unknown's avatar
unknown committed
3013 3014 3015 3016 3017 3018
  if (--ull->count)
    pthread_cond_signal(&ull->cond);
  else
    delete ull;
}

unknown's avatar
unknown committed
3019 3020 3021 3022 3023 3024 3025
/*
   Wait until we are at or past the given position in the master binlog
   on the slave
 */

longlong Item_master_pos_wait::val_int()
{
3026
  DBUG_ASSERT(fixed == 1);
unknown's avatar
unknown committed
3027 3028
  THD* thd = current_thd;
  String *log_name = args[0]->val_str(&value);
3029
  int event_count= 0;
unknown's avatar
unknown committed
3030

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

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

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

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

3079 3080 3081 3082
  /*
    Structure is now initialized.  Try to get the lock.
    Set up control struct to allow others to abort locks
  */
3083 3084 3085 3086
  thd->proc_info="User lock";
  thd->mysys_var->current_mutex= &LOCK_user_locks;
  thd->mysys_var->current_cond=  &ull->cond;

3087
  set_timespec(abstime,lock_timeout);
unknown's avatar
unknown committed
3088 3089 3090 3091 3092 3093 3094
  while (ull->locked && !thd->killed)
  {
    int error= pthread_cond_timedwait(&ull->cond, &LOCK_user_locks, &abstime);
    if (error == ETIMEDOUT || error == ETIME)
      break;
  }

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

3140 3141 3142 3143 3144 3145 3146 3147 3148 3149
  /*
    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
3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165
  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;
  }

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

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

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


/*
3237 3238 3239 3240 3241
  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
3242 3243 3244 3245
*/

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

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


3279
longlong Item_func_last_insert_id::val_int()
unknown's avatar
unknown committed
3280
{
unknown's avatar
unknown committed
3281
  THD *thd= current_thd;
3282
  DBUG_ASSERT(fixed == 1);
3283 3284
  if (arg_count)
  {
unknown's avatar
unknown committed
3285 3286
    longlong value= args[0]->val_int();
    null_value= args[0]->null_value;
3287 3288 3289 3290 3291 3292 3293 3294 3295 3296
    /*
      LAST_INSERT_ID(X) must affect the client's mysql_insert_id() as
      documented in the manual. We don't want to touch
      first_successful_insert_id_in_cur_stmt because it would make
      LAST_INSERT_ID(X) take precedence over an generated auto_increment
      value for this row.
    */
    thd->arg_of_last_insert_id_function= TRUE;
    thd->first_successful_insert_id_in_prev_stmt= value;
    return value;
3297
  }
unknown's avatar
unknown committed
3298
  thd->lex->uncacheable(UNCACHEABLE_SIDEEFFECT);
3299
  return thd->read_first_successful_insert_id_in_prev_stmt();
unknown's avatar
unknown committed
3300 3301 3302 3303 3304 3305
}

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

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

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

3333

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

unknown's avatar
unknown committed
3347

3348 3349 3350 3351
/* This function is just used to create tests with time gaps */

longlong Item_func_sleep::val_int()
{
3352 3353 3354
  THD *thd= current_thd;
  struct timespec abstime;
  pthread_cond_t cond;
3355
  int error;
3356

3357
  DBUG_ASSERT(fixed == 1);
3358

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

  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
3385
  return test(!error); 		// Return 1 killed
3386 3387 3388
}


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

3434
/*
3435 3436
  When a user variable is updated (in a SET command or a query like
  SELECT @a:= ).
3437
*/
3438

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


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

3485

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

unknown's avatar
unknown committed
3558 3559 3560 3561 3562

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


3580 3581
/* Get the value of a variable as a double */

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


3608 3609 3610
/* Get the value of a variable as an integer */

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

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

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

  SYNOPSYS
    Item_func_set_user_var::check()

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

  RETURN
unknown's avatar
unknown committed
3706
    FALSE OK.
3707 3708 3709 3710 3711 3712 3713 3714 3715 3716
*/

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

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

3744 3745 3746 3747 3748 3749 3750 3751 3752 3753 3754 3755

/*
  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
3756
    0	OK
3757 3758 3759 3760
    1	EOM Error

*/

unknown's avatar
unknown committed
3761 3762 3763
bool
Item_func_set_user_var::update()
{
3764 3765 3766 3767
  bool res;
  DBUG_ENTER("Item_func_set_user_var::update");
  LINT_INIT(res);

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


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

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

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


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


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


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


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


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


3906
/*
3907 3908 3909 3910 3911 3912 3913 3914 3915 3916
  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.

3917 3918 3919 3920
  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).
3921

3922
  RETURN
3923
    0  OK
unknown's avatar
unknown committed
3924
    1  Failed to put appropriate record into binary log
3925

3926 3927
*/

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

  if (!(opt_bin_log && is_update_query(sql_command)))
3936 3937 3938 3939
  {
    *out_entry= var_entry;
    return 0;
  }
3940 3941

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

      Manipulation with lex is necessary since free_underlaid_joins
      is going to release memory belonging to the main query.
3958 3959 3960
    */

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

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

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

4045
  error= get_var_with_binlog(thd, thd->lex->sql_command, name, &var_entry);
4046 4047

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

4077 4078
  if (error)
    thd->fatal_error();
4079

unknown's avatar
unknown committed
4080
  return;
unknown's avatar
unknown committed
4081 4082 4083
}


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


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
4100 4101 4102

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

4108

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


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


4136
bool Item_user_var_as_out_param::fix_fields(THD *thd, Item **ref)
unknown's avatar
unknown committed
4137 4138
{
  DBUG_ASSERT(fixed == 0);
4139
  if (Item::fix_fields(thd, ref) ||
unknown's avatar
unknown committed
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 4203 4204 4205
      !(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);
}


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

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

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

err:
  null_value=1;
  return 0;
}

unknown's avatar
unknown committed
4291

4292
void Item_func_match::init_search(bool no_order)
4293
{
unknown's avatar
unknown committed
4294
  DBUG_ENTER("Item_func_match::init_search");
4295 4296

  /* Check if init_search() has been called before */
4297
  if (ft_handler)
unknown's avatar
unknown committed
4298
    DBUG_VOID_RETURN;
4299

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

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

unknown's avatar
unknown committed
4325
  String *ft_tmp= 0;
4326

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

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

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

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

  DBUG_VOID_RETURN;
4350 4351
}

unknown's avatar
unknown committed
4352

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

4359 4360 4361
  maybe_null=1;
  join_key=0;

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

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

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

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

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

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

  if (!fts)
4431
    goto err;
unknown's avatar
unknown committed
4432

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

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

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

4474
    key=ft_to_key[keynr];
4475

4476 4477 4478
    return 0;
  }

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

4490

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

  Item_func_match *ifm=(Item_func_match*) item;

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

  return 0;
}

4507

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

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

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

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

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

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

4560

4561 4562 4563 4564
/***************************************************************************
  System variables
****************************************************************************/

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

4583 4584 4585

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

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

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

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

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


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

  SYNOPSIS:
    val_int()

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

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

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

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

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

  null_value=0;
  return ull->thread_id;
}

4675

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

  return thd->row_count_func;
}


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

unknown's avatar
unknown committed
4695

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

unknown's avatar
unknown committed
4707

4708 4709 4710 4711 4712 4713 4714 4715
void
Item_func_sp::cleanup()
{
  if (result_field)
  {
    delete result_field;
    result_field= NULL;
  }
4716
  m_sp= NULL;
4717 4718
  Item_func::cleanup();
}
unknown's avatar
unknown committed
4719

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

4734 4735 4736 4737 4738
  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();
4739 4740
}

4741

unknown's avatar
unknown committed
4742 4743 4744
Field *
Item_func_sp::sp_result_field(void) const
{
unknown's avatar
unknown committed
4745
  Field *field;
unknown's avatar
unknown committed
4746
  DBUG_ENTER("Item_func_sp::sp_result_field");
4747 4748 4749 4750
  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
4751 4752

  if (!m_sp)
unknown's avatar
unknown committed
4753
  {
4754 4755 4756
    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
4757
    {
unknown's avatar
unknown committed
4758 4759
      my_error(ER_SP_DOES_NOT_EXIST, MYF(0), "FUNCTION", m_name->m_qname.str);
      DBUG_RETURN(0);
unknown's avatar
unknown committed
4760 4761
    }
  }
unknown's avatar
unknown committed
4762
  if (!dummy_table->alias)
unknown's avatar
unknown committed
4763 4764
  {
    char *empty_name= (char *) "";
unknown's avatar
unknown committed
4765 4766
    dummy_table->alias= empty_name;
    dummy_table->maybe_null= maybe_null;
unknown's avatar
unknown committed
4767
    dummy_table->in_use= current_thd;
4768
    dummy_table->copy_blobs= TRUE;
unknown's avatar
unknown committed
4769 4770 4771
    dummy_table->s->table_cache_key.str = empty_name;
    dummy_table->s->table_name.str= empty_name;
    dummy_table->s->db.str= empty_name;
unknown's avatar
unknown committed
4772
  }
4773 4774 4775
  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));

4776
  DBUG_RETURN(field);
unknown's avatar
unknown committed
4777 4778 4779
}


unknown's avatar
unknown committed
4780 4781 4782 4783 4784 4785 4786 4787
/*
  Execute function & store value in field

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

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

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

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

  /* 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;
4828 4829 4830
}


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

4838 4839
  DBUG_ENTER("Item_func_sp::execute_impl");

4840 4841 4842 4843 4844 4845 4846 4847
#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))
4848
    goto error;
4849

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

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

4869

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


4886 4887
enum enum_field_types
Item_func_sp::field_type() const
4888
{
unknown's avatar
unknown committed
4889
  Field *field;
4890
  DBUG_ENTER("Item_func_sp::field_type");
4891

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

4903

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

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

void
Item_func_sp::fix_length_and_dec()
{
unknown's avatar
unknown committed
4925
  Field *field;
4926 4927
  DBUG_ENTER("Item_func_sp::fix_length_and_dec");

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

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


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

unknown's avatar
unknown committed
4956

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

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

4969
  if (!field)
4970 4971
    my_message(ER_OUT_OF_RESOURCES, ER(ER_OUT_OF_RESOURCES), MYF(0));

4972
  DBUG_RETURN(field);
unknown's avatar
unknown committed
4973
}
4974

unknown's avatar
unknown committed
4975

4976
/*
4977
  Find the function and check access rights to the function
4978 4979

  SYNOPSIS
unknown's avatar
unknown committed
4980 4981
    find_and_check_access()
    thd           thread handler
4982
    want_access   requested access
unknown's avatar
unknown committed
4983
    save          backup of security context
4984 4985

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

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

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

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

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

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

error:
5043
  return res;
unknown's avatar
unknown committed
5044
}
5045 5046 5047 5048 5049 5050 5051

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