thr_lock.c 50.7 KB
Newer Older
Marc Alff's avatar
Marc Alff committed
1
/* Copyright (C) 2000 MySQL AB, 2008-2009 Sun Microsystems, Inc
unknown's avatar
unknown committed
2 3 4

   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
unknown's avatar
unknown committed
5
   the Free Software Foundation; version 2 of the License.
unknown's avatar
unknown committed
6 7

   This program is distributed in the hope that it will be useful,
unknown's avatar
unknown committed
8
   but WITHOUT ANY WARRANTY; without even the implied warranty of
unknown's avatar
unknown committed
9 10 11 12 13 14
   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 */
unknown's avatar
unknown committed
15 16 17 18 19 20 21 22 23 24 25 26 27

/*
Read and write locks for Posix threads. All tread must acquire
all locks it needs through thr_multi_lock() to avoid dead-locks.
A lock consists of a master lock (THR_LOCK), and lock instances
(THR_LOCK_DATA).
Any thread can have any number of lock instances (read and write:s) on
any lock. All lock instances must be freed.
Locks are prioritized according to:

The current lock types are:

TL_READ	 		# Low priority read
unknown's avatar
unknown committed
28
TL_READ_WITH_SHARED_LOCKS
unknown's avatar
unknown committed
29 30 31 32 33 34 35 36 37 38 39 40 41 42
TL_READ_HIGH_PRIORITY	# High priority read
TL_READ_NO_INSERT	# Read without concurrent inserts
TL_WRITE_ALLOW_WRITE	# Write lock that allows other writers
TL_WRITE_CONCURRENT_INSERT
			# Insert that can be mixed when selects
TL_WRITE_DELAYED	# Used by delayed insert
			# Allows lower locks to take over
TL_WRITE_LOW_PRIORITY	# Low priority write
TL_WRITE		# High priority write
TL_WRITE_ONLY		# High priority write
			# Abort all new lock request with an error

Locks are prioritized according to:

43
WRITE_ALLOW_WRITE, WRITE_CONCURRENT_INSERT, WRITE_DELAYED,
unknown's avatar
unknown committed
44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
WRITE_LOW_PRIORITY, READ, WRITE, READ_HIGH_PRIORITY and WRITE_ONLY

Locks in the same privilege level are scheduled in first-in-first-out order.

To allow concurrent read/writes locks, with 'WRITE_CONCURRENT_INSERT' one
should put a pointer to the following functions in the lock structure:
(If the pointer is zero (default), the function is not called)

check_status:
	 Before giving a lock of type TL_WRITE_CONCURRENT_INSERT,
         we check if this function exists and returns 0.
	 If not, then the lock is upgraded to TL_WRITE_LOCK
	 In MyISAM this is a simple check if the insert can be done
	 at the end of the datafile.
update_status:
	Before a write lock is released, this function is called.
	In MyISAM this functions updates the count and length of the datafile
get_status:
	When one gets a lock this functions is called.
	In MyISAM this stores the number of rows and size of the datafile
	for concurrent reads.

66 67
The lock algorithm allows one to have one TL_WRITE_CONCURRENT_INSERT or
one TL_WRITE_DELAYED lock at the same time as multiple read locks.
unknown's avatar
unknown committed
68 69 70 71

*/

#if !defined(MAIN) && !defined(DBUG_OFF) && !defined(EXTRA_DEBUG)
72
#define FORCE_DBUG_OFF
unknown's avatar
unknown committed
73 74 75
#endif

#include "mysys_priv.h"
unknown's avatar
unknown committed
76 77

#ifdef THREAD
unknown's avatar
unknown committed
78 79 80 81 82
#include "thr_lock.h"
#include <m_string.h>
#include <errno.h>

my_bool thr_lock_inited=0;
unknown's avatar
unknown committed
83
ulong locks_immediate = 0L, locks_waited = 0L;
84
enum thr_lock_type thr_upgraded_concurrent_insert_lock = TL_WRITE;
unknown's avatar
unknown committed
85 86 87 88 89 90

/* The following constants are only for debug output */
#define MAX_THREADS 100
#define MAX_LOCKS   100


unknown's avatar
unknown committed
91
LIST *thr_lock_thread_list;			/* List of threads in use */
unknown's avatar
unknown committed
92 93
ulong max_write_lock_count= ~(ulong) 0L;

Marc Alff's avatar
Marc Alff committed
94
static inline mysql_cond_t *get_cond(void)
unknown's avatar
unknown committed
95 96 97 98 99 100 101 102 103 104 105 106 107 108
{
  return &my_thread_var->suspend;
}

/*
** For the future (now the thread specific cond is alloced by my_pthread.c)
*/

my_bool init_thr_lock()
{
  thr_lock_inited=1;
  return 0;
}

109
static inline my_bool
110
thr_lock_owner_equal(THR_LOCK_INFO *rhs, THR_LOCK_INFO *lhs)
111 112 113 114 115
{
  return rhs == lhs;
}


unknown's avatar
unknown committed
116
#ifdef EXTRA_DEBUG
unknown's avatar
unknown committed
117 118
#define MAX_FOUND_ERRORS	10		/* Report 10 first errors */
static uint found_errors=0;
unknown's avatar
unknown committed
119 120

static int check_lock(struct st_lock_list *list, const char* lock_type,
121
		      const char *where, my_bool same_owner, my_bool no_cond)
unknown's avatar
unknown committed
122 123 124
{
  THR_LOCK_DATA *data,**prev;
  uint count=0;
125
  THR_LOCK_INFO *UNINIT_VAR(first_owner);
unknown's avatar
unknown committed
126 127 128 129 130 131

  prev= &list->data;
  if (list->data)
  {
    enum thr_lock_type last_lock_type=list->data->type;

132 133
    if (same_owner && list->data)
      first_owner= list->data->owner;
unknown's avatar
unknown committed
134 135 136 137 138 139 140 141 142 143 144
    for (data=list->data; data && count++ < MAX_LOCKS ; data=data->next)
    {
      if (data->type != last_lock_type)
	last_lock_type=TL_IGNORE;
      if (data->prev != prev)
      {
	fprintf(stderr,
		"Warning: prev link %d didn't point at previous lock at %s: %s\n",
		count, lock_type, where);
	return 1;
      }
145 146
      if (same_owner &&
          !thr_lock_owner_equal(data->owner, first_owner) &&
unknown's avatar
unknown committed
147 148 149 150 151 152 153
	  last_lock_type != TL_WRITE_ALLOW_WRITE)
      {
	fprintf(stderr,
		"Warning: Found locks from different threads in %s: %s\n",
		lock_type,where);
	return 1;
      }
unknown's avatar
unknown committed
154 155 156 157 158 159 160
      if (no_cond && data->cond)
      {
	fprintf(stderr,
		"Warning: Found active lock with not reset cond %s: %s\n",
		lock_type,where);
	return 1;
      }
unknown's avatar
unknown committed
161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
      prev= &data->next;
    }
    if (data)
    {
      fprintf(stderr,"Warning: found too many locks at %s: %s\n",
	      lock_type,where);
      return 1;
    }
  }
  if (prev != list->last)
  {
    fprintf(stderr,"Warning: last didn't point at last lock at %s: %s\n",
	    lock_type, where);
    return 1;
  }
  return 0;
}

179

unknown's avatar
unknown committed
180 181 182
static void check_locks(THR_LOCK *lock, const char *where,
			my_bool allow_no_locks)
{
unknown's avatar
unknown committed
183
  uint old_found_errors=found_errors;
184 185
  DBUG_ENTER("check_locks");

unknown's avatar
unknown committed
186
  if (found_errors < MAX_FOUND_ERRORS)
unknown's avatar
unknown committed
187
  {
unknown's avatar
unknown committed
188 189 190 191
    if (check_lock(&lock->write,"write",where,1,1) |
	check_lock(&lock->write_wait,"write_wait",where,0,0) |
	check_lock(&lock->read,"read",where,0,1) |
	check_lock(&lock->read_wait,"read_wait",where,0,0))
unknown's avatar
unknown committed
192
      found_errors++;
unknown's avatar
unknown committed
193

unknown's avatar
unknown committed
194
    if (found_errors < MAX_FOUND_ERRORS)
unknown's avatar
unknown committed
195 196 197 198 199 200 201
    {
      uint count=0;
      THR_LOCK_DATA *data;
      for (data=lock->read.data ; data ; data=data->next)
      {
	if ((int) data->type == (int) TL_READ_NO_INSERT)
	  count++;
202 203
        /* Protect against infinite loop. */
        DBUG_ASSERT(count <= lock->read_no_write_count);
unknown's avatar
unknown committed
204 205 206
      }
      if (count != lock->read_no_write_count)
      {
unknown's avatar
unknown committed
207
	found_errors++;
unknown's avatar
unknown committed
208 209 210 211 212 213 214 215 216
	fprintf(stderr,
		"Warning at '%s': Locks read_no_write_count was %u when it should have been %u\n", where, lock->read_no_write_count,count);
      }      

      if (!lock->write.data)
      {
	if (!allow_no_locks && !lock->read.data &&
	    (lock->write_wait.data || lock->read_wait.data))
	{
unknown's avatar
unknown committed
217
	  found_errors++;
unknown's avatar
unknown committed
218 219 220 221 222 223 224 225
	  fprintf(stderr,
		  "Warning at '%s': No locks in use but locks are in wait queue\n",
		  where);
	}
	if (!lock->write_wait.data)
	{
	  if (!allow_no_locks && lock->read_wait.data)
	  {
unknown's avatar
unknown committed
226
	    found_errors++;
unknown's avatar
unknown committed
227 228 229 230 231 232 233 234 235 236 237 238 239 240
	    fprintf(stderr,
		    "Warning at '%s': No write locks and waiting read locks\n",
		    where);
	  }
	}
	else
	{
	  if (!allow_no_locks &&
	      (((lock->write_wait.data->type == TL_WRITE_CONCURRENT_INSERT ||
		 lock->write_wait.data->type == TL_WRITE_ALLOW_WRITE) &&
		!lock->read_no_write_count) ||
	       (lock->write_wait.data->type == TL_WRITE_DELAYED &&
		!lock->read.data)))
	  {
unknown's avatar
unknown committed
241
	    found_errors++;
unknown's avatar
unknown committed
242 243 244 245 246 247 248 249 250 251 252 253 254
	    fprintf(stderr,
		    "Warning at '%s': Write lock %d waiting while no exclusive read locks\n",where,(int) lock->write_wait.data->type);
	  }
	}	      
      }
      else
      {						/* Have write lock */
	if (lock->write_wait.data)
	{
	  if (!allow_no_locks && 
	      lock->write.data->type == TL_WRITE_ALLOW_WRITE &&
	      lock->write_wait.data->type == TL_WRITE_ALLOW_WRITE)
	  {
unknown's avatar
unknown committed
255
	    found_errors++;
unknown's avatar
unknown committed
256 257 258 259 260 261 262
	    fprintf(stderr,
		    "Warning at '%s': Found WRITE_ALLOW_WRITE lock waiting for WRITE_ALLOW_WRITE lock\n",
		    where);
	  }
	}
	if (lock->read.data)
	{
263 264
          if (!thr_lock_owner_equal(lock->write.data->owner,
                                    lock->read.data->owner) &&
265 266 267 268 269
	      ((lock->write.data->type > TL_WRITE_DELAYED &&
		lock->write.data->type != TL_WRITE_ONLY) ||
	       ((lock->write.data->type == TL_WRITE_CONCURRENT_INSERT ||
		 lock->write.data->type == TL_WRITE_ALLOW_WRITE) &&
		lock->read_no_write_count)))
unknown's avatar
unknown committed
270
	  {
unknown's avatar
unknown committed
271
	    found_errors++;
unknown's avatar
unknown committed
272
	    fprintf(stderr,
unknown's avatar
unknown committed
273 274
		    "Warning at '%s': Found lock of type %d that is write and read locked\n",
		    where, lock->write.data->type);
275 276 277
	    DBUG_PRINT("warning",("At '%s': Found lock of type %d that is write and read locked\n",
		    where, lock->write.data->type));

unknown's avatar
unknown committed
278 279 280 281 282 283 284
	  }
	}
	if (lock->read_wait.data)
	{
	  if (!allow_no_locks && lock->write.data->type <= TL_WRITE_DELAYED &&
	      lock->read_wait.data->type <= TL_READ_HIGH_PRIORITY)
	  {
unknown's avatar
unknown committed
285
	    found_errors++;
unknown's avatar
unknown committed
286 287 288 289 290 291 292 293 294
	    fprintf(stderr,
		    "Warning at '%s': Found read lock of type %d waiting for write lock of type %d\n",
		    where,
		    (int) lock->read_wait.data->type,
		    (int) lock->write.data->type);
	  }
	}
      }
    }
unknown's avatar
unknown committed
295
    if (found_errors != old_found_errors)
unknown's avatar
unknown committed
296 297 298 299
    {
      DBUG_PRINT("error",("Found wrong lock"));
    }
  }
300
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
301 302 303 304 305 306 307 308 309 310 311 312 313
}

#else /* EXTRA_DEBUG */
#define check_locks(A,B,C)
#endif


	/* Initialize a lock */

void thr_lock_init(THR_LOCK *lock)
{
  DBUG_ENTER("thr_lock_init");
  bzero((char*) lock,sizeof(*lock));
Marc Alff's avatar
Marc Alff committed
314
  mysql_mutex_init(key_THR_LOCK_mutex, &lock->mutex, MY_MUTEX_INIT_FAST);
unknown's avatar
unknown committed
315 316 317 318 319
  lock->read.last= &lock->read.data;
  lock->read_wait.last= &lock->read_wait.data;
  lock->write_wait.last= &lock->write_wait.data;
  lock->write.last= &lock->write.data;

Marc Alff's avatar
Marc Alff committed
320
  mysql_mutex_lock(&THR_LOCK_lock);              /* Add to locks in use */
unknown's avatar
unknown committed
321
  lock->list.data=(void*) lock;
unknown's avatar
unknown committed
322
  thr_lock_thread_list=list_add(thr_lock_thread_list,&lock->list);
Marc Alff's avatar
Marc Alff committed
323
  mysql_mutex_unlock(&THR_LOCK_lock);
unknown's avatar
unknown committed
324 325 326 327 328 329 330
  DBUG_VOID_RETURN;
}


void thr_lock_delete(THR_LOCK *lock)
{
  DBUG_ENTER("thr_lock_delete");
Marc Alff's avatar
Marc Alff committed
331
  mysql_mutex_lock(&THR_LOCK_lock);
unknown's avatar
unknown committed
332
  thr_lock_thread_list=list_delete(thr_lock_thread_list,&lock->list);
Marc Alff's avatar
Marc Alff committed
333 334
  mysql_mutex_unlock(&THR_LOCK_lock);
  mysql_mutex_destroy(&lock->mutex);
unknown's avatar
unknown committed
335 336 337
  DBUG_VOID_RETURN;
}

338 339 340

void thr_lock_info_init(THR_LOCK_INFO *info)
{
unknown's avatar
unknown committed
341 342 343
  struct st_my_thread_var *tmp= my_thread_var;
  info->thread=    tmp->pthread_self;
  info->thread_id= tmp->id;
344 345
}

unknown's avatar
unknown committed
346 347 348 349 350 351
	/* Initialize a lock instance */

void thr_lock_data_init(THR_LOCK *lock,THR_LOCK_DATA *data, void *param)
{
  data->lock=lock;
  data->type=TL_UNLOCK;
352
  data->owner= 0;                               /* no owner yet */
unknown's avatar
unknown committed
353
  data->status_param=param;
unknown's avatar
unknown committed
354
  data->cond=0;
unknown's avatar
unknown committed
355 356 357
}


358
static inline my_bool
359
has_old_lock(THR_LOCK_DATA *data, THR_LOCK_INFO *owner)
unknown's avatar
unknown committed
360 361 362
{
  for ( ; data ; data=data->next)
  {
363
    if (thr_lock_owner_equal(data->owner, owner))
unknown's avatar
unknown committed
364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380
      return 1;					/* Already locked by thread */
  }
  return 0;
}

static inline my_bool have_specific_lock(THR_LOCK_DATA *data,
					 enum thr_lock_type type)
{
  for ( ; data ; data=data->next)
  {
    if (data->type == type)
      return 1;
  }
  return 0;
}


381 382 383
static void wake_up_waiters(THR_LOCK *lock);


384 385
static enum enum_thr_lock_result
wait_for_lock(struct st_lock_list *wait, THR_LOCK_DATA *data,
386
              my_bool in_wait_list, ulong lock_wait_timeout)
unknown's avatar
unknown committed
387
{
388
  struct st_my_thread_var *thread_var= my_thread_var;
Marc Alff's avatar
Marc Alff committed
389
  mysql_cond_t *cond= &thread_var->suspend;
390 391
  struct timespec wait_timeout;
  enum enum_thr_lock_result result= THR_LOCK_ABORTED;
Konstantin Osipov's avatar
Konstantin Osipov committed
392
  const char *old_proc_info;
393
  DBUG_ENTER("wait_for_lock");
unknown's avatar
unknown committed
394

Ingo Struewing's avatar
Ingo Struewing committed
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416
  /*
    One can use this to signal when a thread is going to wait for a lock.
    See debug_sync.cc.

    Beware of waiting for a signal here. The lock has aquired its mutex.
    While waiting on a signal here, the locking thread could not aquire
    the mutex to release the lock. One could lock up the table
    completely.

    In detail it works so: When thr_lock() tries to acquire a table
    lock, it locks the lock->mutex, checks if it can have the lock, and
    if not, it calls wait_for_lock(). Here it unlocks the table lock
    while waiting on a condition. The sync point is located before this
    wait for condition. If we have a waiting action here, we hold the
    the table locks mutex all the time. Any attempt to look at the table
    lock by another thread blocks it immediately on lock->mutex. This
    can easily become an unexpected and unobvious blockage. So be
    warned: Do not request a WAIT_FOR action for the 'wait_for_lock'
    sync point unless you really know what you do.
  */
  DEBUG_SYNC_C("wait_for_lock");

unknown's avatar
unknown committed
417 418 419 420 421 422 423
  if (!in_wait_list)
  {
    (*wait->last)=data;				/* Wait for lock */
    data->prev= wait->last;
    wait->last= &data->next;
  }

424 425
  statistic_increment(locks_waited, &THR_LOCK_lock);

unknown's avatar
unknown committed
426 427 428
  /* Set up control struct to allow others to abort locks */
  thread_var->current_mutex= &data->lock->mutex;
  thread_var->current_cond=  cond;
429
  data->cond= cond;
unknown's avatar
unknown committed
430

Konstantin Osipov's avatar
Konstantin Osipov committed
431 432 433
  old_proc_info= proc_info_hook(NULL, "Table lock",
                                __func__, __FILE__, __LINE__);

434
  set_timespec(wait_timeout, lock_wait_timeout);
unknown's avatar
unknown committed
435
  while (!thread_var->abort || in_wait_list)
unknown's avatar
unknown committed
436
  {
437
    int rc= mysql_cond_timedwait(cond, &data->lock->mutex, &wait_timeout);
438 439 440 441 442 443 444 445 446 447 448 449 450 451
    /*
      We must break the wait if one of the following occurs:
      - the connection has been aborted (!thread_var->abort), but
        this is not a delayed insert thread (in_wait_list). For a delayed
        insert thread the proper action at shutdown is, apparently, to
        acquire the lock and complete the insert.
      - the lock has been granted (data->cond is set to NULL by the granter),
        or the waiting has been aborted (additionally data->type is set to
        TL_UNLOCK).
      - the wait has timed out (rc == ETIMEDOUT)
      Order of checks below is important to not report about timeout
      if the predicate is true.
    */
    if (data->cond == 0)
452 453
    {
      DBUG_PRINT("thr_lock", ("lock granted/aborted"));
454
      break;
455
    }
unknown's avatar
unknown committed
456
    if (rc == ETIMEDOUT || rc == ETIME)
457
    {
458 459
      /* purecov: begin inspected */
      DBUG_PRINT("thr_lock", ("lock timed out"));
460
      result= THR_LOCK_WAIT_TIMEOUT;
unknown's avatar
unknown committed
461
      break;
462
      /* purecov: end */
463
    }
unknown's avatar
unknown committed
464
  }
465 466
  DBUG_PRINT("thr_lock", ("aborted: %d  in_wait_list: %d",
                          thread_var->abort, in_wait_list));
unknown's avatar
unknown committed
467 468 469

  if (data->cond || data->type == TL_UNLOCK)
  {
470
    if (data->cond)                             /* aborted or timed out */
unknown's avatar
unknown committed
471 472 473 474 475
    {
      if (((*data->prev)=data->next))		/* remove from wait-list */
	data->next->prev= data->prev;
      else
	wait->last=data->prev;
476
      data->type= TL_UNLOCK;                    /* No lock */
477 478 479 480 481
      check_locks(data->lock, "killed or timed out wait_for_lock", 1);
      wake_up_waiters(data->lock);
    }
    else
    {
482
      DBUG_PRINT("thr_lock", ("lock aborted"));
483
      check_locks(data->lock, "aborted wait_for_lock", 0);
unknown's avatar
unknown committed
484 485 486 487
    }
  }
  else
  {
488
    result= THR_LOCK_SUCCESS;
unknown's avatar
unknown committed
489
    if (data->lock->get_status)
490
      (*data->lock->get_status)(data->status_param, 0);
unknown's avatar
unknown committed
491 492
    check_locks(data->lock,"got wait_for_lock",0);
  }
Marc Alff's avatar
Marc Alff committed
493
  mysql_mutex_unlock(&data->lock->mutex);
unknown's avatar
unknown committed
494 495

  /* The following must be done after unlock of lock->mutex */
Marc Alff's avatar
Marc Alff committed
496
  mysql_mutex_lock(&thread_var->mutex);
unknown's avatar
unknown committed
497 498
  thread_var->current_mutex= 0;
  thread_var->current_cond=  0;
Marc Alff's avatar
Marc Alff committed
499
  mysql_mutex_unlock(&thread_var->mutex);
Konstantin Osipov's avatar
Konstantin Osipov committed
500 501 502

  proc_info_hook(NULL, old_proc_info, __func__, __FILE__, __LINE__);

503
  DBUG_RETURN(result);
unknown's avatar
unknown committed
504 505 506
}


507
enum enum_thr_lock_result
508
thr_lock(THR_LOCK_DATA *data, THR_LOCK_INFO *owner,
509
         enum thr_lock_type lock_type, ulong lock_wait_timeout)
unknown's avatar
unknown committed
510 511
{
  THR_LOCK *lock=data->lock;
512 513
  enum enum_thr_lock_result result= THR_LOCK_SUCCESS;
  struct st_lock_list *wait_queue;
unknown's avatar
unknown committed
514 515 516
  DBUG_ENTER("thr_lock");

  data->next=0;
unknown's avatar
unknown committed
517
  data->cond=0;					/* safety */
unknown's avatar
unknown committed
518
  data->type=lock_type;
519
  data->owner= owner;                           /* Must be reset ! */
Marc Alff's avatar
Marc Alff committed
520
  mysql_mutex_lock(&lock->mutex);
521
  DBUG_PRINT("lock",("data: 0x%lx  thread: 0x%lx  lock: 0x%lx  type: %d",
522
                     (long) data, data->owner->thread_id,
unknown's avatar
unknown committed
523
                     (long) lock, (int) lock_type));
unknown's avatar
unknown committed
524 525 526 527 528 529 530
  check_locks(lock,(uint) lock_type <= (uint) TL_READ_NO_INSERT ?
	      "enter read_lock" : "enter write_lock",0);
  if ((int) lock_type <= (int) TL_READ_NO_INSERT)
  {
    /* Request for READ lock */
    if (lock->write.data)
    {
531 532 533 534 535 536 537 538 539
      /*
        We can allow a read lock even if there is already a
        write lock on the table if they are owned by the same
        thread or if they satisfy the following lock
        compatibility matrix:

           Request
          /-------
         H|++++  WRITE_ALLOW_WRITE
540 541 542
         e|+++-  WRITE_CONCURRENT_INSERT
         l|++++  WRITE_DELAYED
         d ||||
543 544 545 546
           |||\= READ_NO_INSERT
           ||\ = READ_HIGH_PRIORITY
           |\  = READ_WITH_SHARED_LOCKS
           \   = READ
547
          
548 549 550 551 552 553 554 555

        + = Request can be satisified.
        - = Request cannot be satisified.

        READ_NO_INSERT and WRITE_ALLOW_WRITE should in principle
        be incompatible. However this will cause starvation of
        LOCK TABLE READ in InnoDB under high write load.
        See Bug#42147 for more information.
unknown's avatar
unknown committed
556 557
      */

558
      DBUG_PRINT("lock",("write locked 1 by thread: 0x%lx",
559
			 lock->write.data->owner->thread_id));
560
      if (thr_lock_owner_equal(data->owner, lock->write.data->owner) ||
unknown's avatar
unknown committed
561 562
	  (lock->write.data->type <= TL_WRITE_DELAYED &&
	   (((int) lock_type <= (int) TL_READ_HIGH_PRIORITY) ||
563
	    (lock->write.data->type != TL_WRITE_CONCURRENT_INSERT))))
unknown's avatar
unknown committed
564 565 566 567
      {						/* Already got a write lock */
	(*lock->read.last)=data;		/* Add to running FIFO */
	data->prev=lock->read.last;
	lock->read.last= &data->next;
568
	if (lock_type == TL_READ_NO_INSERT)
unknown's avatar
unknown committed
569 570 571
	  lock->read_no_write_count++;
	check_locks(lock,"read lock with old write lock",0);
	if (lock->get_status)
572
	  (*lock->get_status)(data->status_param, 0);
573
	statistic_increment(locks_immediate,&THR_LOCK_lock);
unknown's avatar
unknown committed
574 575 576 577 578 579
	goto end;
      }
      if (lock->write.data->type == TL_WRITE_ONLY)
      {
	/* We are not allowed to get a READ lock in this case */
	data->type=TL_UNLOCK;
580
        result= THR_LOCK_ABORTED;               /* Can't wait for this one */
unknown's avatar
unknown committed
581 582 583 584 585 586
	goto end;
      }
    }
    else if (!lock->write_wait.data ||
	     lock->write_wait.data->type <= TL_WRITE_LOW_PRIORITY ||
	     lock_type == TL_READ_HIGH_PRIORITY ||
587
	     has_old_lock(lock->read.data, data->owner)) /* Has old read lock */
unknown's avatar
unknown committed
588 589 590 591 592
    {						/* No important write-locks */
      (*lock->read.last)=data;			/* Add to running FIFO */
      data->prev=lock->read.last;
      lock->read.last= &data->next;
      if (lock->get_status)
593
	(*lock->get_status)(data->status_param, 0);
594
      if (lock_type == TL_READ_NO_INSERT)
unknown's avatar
unknown committed
595 596
	lock->read_no_write_count++;
      check_locks(lock,"read lock with no write locks",0);
597
      statistic_increment(locks_immediate,&THR_LOCK_lock);
unknown's avatar
unknown committed
598 599
      goto end;
    }
600
    /*
601
      We're here if there is an active write lock or no write
602 603 604 605
      lock but a high priority write waiting in the write_wait queue.
      In the latter case we should yield the lock to the writer.
    */
    wait_queue= &lock->read_wait;
unknown's avatar
unknown committed
606 607 608 609 610 611 612 613
  }
  else						/* Request for WRITE lock */
  {
    if (lock_type == TL_WRITE_DELAYED)
    {
      if (lock->write.data && lock->write.data->type == TL_WRITE_ONLY)
      {
	data->type=TL_UNLOCK;
614
        result= THR_LOCK_ABORTED;               /* Can't wait for this one */
unknown's avatar
unknown committed
615 616
	goto end;
      }
617
      if (lock->write.data || lock->read.data)
unknown's avatar
unknown committed
618 619 620 621 622 623
      {
	/* Add delayed write lock to write_wait queue, and return at once */
	(*lock->write_wait.last)=data;
	data->prev=lock->write_wait.last;
	lock->write_wait.last= &data->next;
	data->cond=get_cond();
624 625 626 627
        /*
          We don't have to do get_status here as we will do it when we change
          the delayed lock to a real write lock
        */
628
	statistic_increment(locks_immediate,&THR_LOCK_lock);
unknown's avatar
unknown committed
629 630 631 632
	goto end;
      }
    }
    else if (lock_type == TL_WRITE_CONCURRENT_INSERT && ! lock->check_status)
633
      data->type=lock_type= thr_upgraded_concurrent_insert_lock;
unknown's avatar
unknown committed
634 635 636 637 638

    if (lock->write.data)			/* If there is a write lock */
    {
      if (lock->write.data->type == TL_WRITE_ONLY)
      {
Konstantin Osipov's avatar
Konstantin Osipov committed
639
        /* purecov: begin tested */
640 641 642 643 644 645 646 647
        /* Allow lock owner to bypass TL_WRITE_ONLY. */
        if (!thr_lock_owner_equal(data->owner, lock->write.data->owner))
        {
          /* We are not allowed to get a lock in this case */
          data->type=TL_UNLOCK;
          result= THR_LOCK_ABORTED;               /* Can't wait for this one */
          goto end;
        }
Konstantin Osipov's avatar
Konstantin Osipov committed
648
        /* purecov: end */
unknown's avatar
unknown committed
649 650 651
      }

      /*
652 653 654 655 656
        The idea is to allow us to get a lock at once if we already have
        a write lock or if there is no pending write locks and if all
        write locks are of TL_WRITE_ALLOW_WRITE type.

        Note that, since lock requests for the same table are sorted in
657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
        such way that requests with higher thr_lock_type value come first
        (with one exception (*)), lock being requested usually (**) has
        equal or "weaker" type than one which thread might have already
        acquired.
        *)  The only exception to this rule is case when type of old lock
            is TL_WRITE_LOW_PRIORITY and type of new lock is changed inside
            of thr_lock() from TL_WRITE_CONCURRENT_INSERT to TL_WRITE since
            engine turns out to be not supporting concurrent inserts.
            Note that since TL_WRITE has the same compatibility rules as
            TL_WRITE_LOW_PRIORITY (their only difference is priority),
            it is OK to grant new lock without additional checks in such
            situation.
        **) The exceptions are situations when:
            - when old lock type is TL_WRITE_DELAYED
            But these should never happen within MySQL.
672 673 674 675 676 677
        Therefore it is OK to allow acquiring write lock on the table if
        this thread already holds some write lock on it.

        (INSERT INTO t1 VALUES (f1()), where f1() is stored function which
        tries to update t1, is an example of statement which requests two
        different types of write lock on the same table).
unknown's avatar
unknown committed
678
      */
679
      DBUG_ASSERT(! has_old_lock(lock->write.data, data->owner) ||
680 681 682
                  ((lock_type <= lock->write.data->type ||
                    (lock_type == TL_WRITE &&
                     lock->write.data->type == TL_WRITE_LOW_PRIORITY)) &&
683
                   lock->write.data->type != TL_WRITE_DELAYED));
684 685 686 687 688

      if ((lock_type == TL_WRITE_ALLOW_WRITE &&
           ! lock->write_wait.data &&
           lock->write.data->type == TL_WRITE_ALLOW_WRITE) ||
          has_old_lock(lock->write.data, data->owner))
unknown's avatar
unknown committed
689
      {
unknown's avatar
unknown committed
690 691 692 693 694 695 696 697
	/*
          We have already got a write lock or all locks are
          TL_WRITE_ALLOW_WRITE
        */
        DBUG_PRINT("info", ("write_wait.data: 0x%lx  old_type: %d",
                            (ulong) lock->write_wait.data,
                            lock->write.data->type));

unknown's avatar
unknown committed
698 699 700 701 702
	(*lock->write.last)=data;	/* Add to running fifo */
	data->prev=lock->write.last;
	lock->write.last= &data->next;
	check_locks(lock,"second write lock",0);
	if (data->lock->get_status)
703
	  (*data->lock->get_status)(data->status_param, 0);
704
	statistic_increment(locks_immediate,&THR_LOCK_lock);
unknown's avatar
unknown committed
705 706
	goto end;
      }
707
      DBUG_PRINT("lock",("write locked 2 by thread: 0x%lx",
708
			 lock->write.data->owner->thread_id));
unknown's avatar
unknown committed
709 710 711
    }
    else
    {
unknown's avatar
unknown committed
712 713
      DBUG_PRINT("info", ("write_wait.data: 0x%lx",
                          (ulong) lock->write_wait.data));
unknown's avatar
unknown committed
714 715
      if (!lock->write_wait.data)
      {						/* no scheduled write locks */
716 717 718 719 720 721 722 723 724 725
        my_bool concurrent_insert= 0;
	if (lock_type == TL_WRITE_CONCURRENT_INSERT)
        {
          concurrent_insert= 1;
          if ((*lock->check_status)(data->status_param))
          {
            concurrent_insert= 0;
            data->type=lock_type= thr_upgraded_concurrent_insert_lock;
          }
        }
unknown's avatar
unknown committed
726 727 728 729 730 731 732 733 734 735 736

	if (!lock->read.data ||
	    (lock_type <= TL_WRITE_DELAYED &&
	     ((lock_type != TL_WRITE_CONCURRENT_INSERT &&
	       lock_type != TL_WRITE_ALLOW_WRITE) ||
	      !lock->read_no_write_count)))
	{
	  (*lock->write.last)=data;		/* Add as current write lock */
	  data->prev=lock->write.last;
	  lock->write.last= &data->next;
	  if (data->lock->get_status)
737
	    (*data->lock->get_status)(data->status_param, concurrent_insert);
unknown's avatar
unknown committed
738
	  check_locks(lock,"only write lock",0);
739
	  statistic_increment(locks_immediate,&THR_LOCK_lock);
unknown's avatar
unknown committed
740 741 742
	  goto end;
	}
      }
743
      DBUG_PRINT("lock",("write locked 3 by thread: 0x%lx  type: %d",
744
			 lock->read.data->owner->thread_id, data->type));
unknown's avatar
unknown committed
745
    }
746
    wait_queue= &lock->write_wait;
unknown's avatar
unknown committed
747
  }
748
  /* Can't get lock yet;  Wait for it */
749
  DBUG_RETURN(wait_for_lock(wait_queue, data, 0, lock_wait_timeout));
unknown's avatar
unknown committed
750
end:
Marc Alff's avatar
Marc Alff committed
751
  mysql_mutex_unlock(&lock->mutex);
unknown's avatar
unknown committed
752 753 754 755 756
  DBUG_RETURN(result);
}


static inline void free_all_read_locks(THR_LOCK *lock,
757
				       my_bool using_concurrent_insert)
unknown's avatar
unknown committed
758 759 760 761 762 763 764 765 766 767 768 769 770 771 772
{
  THR_LOCK_DATA *data=lock->read_wait.data;

  check_locks(lock,"before freeing read locks",1);

  /* move all locks from read_wait list to read list */
  (*lock->read.last)=data;
  data->prev=lock->read.last;
  lock->read.last=lock->read_wait.last;

  /* Clear read_wait list */
  lock->read_wait.last= &lock->read_wait.data;

  do
  {
Marc Alff's avatar
Marc Alff committed
773
    mysql_cond_t *cond= data->cond;
unknown's avatar
unknown committed
774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792
    if ((int) data->type == (int) TL_READ_NO_INSERT)
    {
      if (using_concurrent_insert)
      {
	/*
	  We can't free this lock; 
	  Link lock away from read chain back into read_wait chain
	*/
	if (((*data->prev)=data->next))
	  data->next->prev=data->prev;
	else
	  lock->read.last=data->prev;
	*lock->read_wait.last= data;
	data->prev= lock->read_wait.last;
	lock->read_wait.last= &data->next;
	continue;
      }
      lock->read_no_write_count++;
    }      
unknown's avatar
unknown committed
793
    /* purecov: begin inspected */
794
    DBUG_PRINT("lock",("giving read lock to thread: 0x%lx",
795
		       data->owner->thread_id));
unknown's avatar
unknown committed
796
    /* purecov: end */
unknown's avatar
unknown committed
797
    data->cond=0;				/* Mark thread free */
Marc Alff's avatar
Marc Alff committed
798
    mysql_cond_signal(cond);
unknown's avatar
unknown committed
799 800 801 802 803 804 805 806 807 808 809 810 811 812
  } while ((data=data->next));
  *lock->read_wait.last=0;
  if (!lock->read_wait.data)
    lock->write_lock_count=0;
  check_locks(lock,"after giving read locks",0);
}

	/* Unlock lock and free next thread on same lock */

void thr_unlock(THR_LOCK_DATA *data)
{
  THR_LOCK *lock=data->lock;
  enum thr_lock_type lock_type=data->type;
  DBUG_ENTER("thr_unlock");
813
  DBUG_PRINT("lock",("data: 0x%lx  thread: 0x%lx  lock: 0x%lx",
814
                     (long) data, data->owner->thread_id, (long) lock));
Marc Alff's avatar
Marc Alff committed
815
  mysql_mutex_lock(&lock->mutex);
unknown's avatar
unknown committed
816 817 818 819 820 821 822 823
  check_locks(lock,"start of release lock",0);

  if (((*data->prev)=data->next))		/* remove from lock-list */
    data->next->prev= data->prev;
  else if (lock_type <= TL_READ_NO_INSERT)
    lock->read.last=data->prev;
  else if (lock_type == TL_WRITE_DELAYED && data->cond)
  {
unknown's avatar
unknown committed
824 825 826 827
    /*
      This only happens in extreme circumstances when a 
      write delayed lock that is waiting for a lock
    */
unknown's avatar
unknown committed
828 829 830 831
    lock->write_wait.last=data->prev;		/* Put it on wait queue */
  }
  else
    lock->write.last=data->prev;
832 833 834 835 836 837 838 839 840 841
  if (lock_type >= TL_WRITE_CONCURRENT_INSERT)
  {
    if (lock->update_status)
      (*lock->update_status)(data->status_param);
  }
  else
  {
    if (lock->restore_status)
      (*lock->restore_status)(data->status_param);
  }
unknown's avatar
unknown committed
842 843 844 845
  if (lock_type == TL_READ_NO_INSERT)
    lock->read_no_write_count--;
  data->type=TL_UNLOCK;				/* Mark unlocked */
  check_locks(lock,"after releasing lock",1);
846
  wake_up_waiters(lock);
Marc Alff's avatar
Marc Alff committed
847
  mysql_mutex_unlock(&lock->mutex);
848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865
  DBUG_VOID_RETURN;
}


/**
  @brief  Wake up all threads which pending requests for the lock
          can be satisfied.

  @param  lock  Lock for which threads should be woken up

*/

static void wake_up_waiters(THR_LOCK *lock)
{
  THR_LOCK_DATA *data;
  enum thr_lock_type lock_type;

  DBUG_ENTER("wake_up_waiters");
unknown's avatar
unknown committed
866

unknown's avatar
unknown committed
867
  if (!lock->write.data)			/* If no active write locks */
unknown's avatar
unknown committed
868 869 870 871 872 873 874
  {
    data=lock->write_wait.data;
    if (!lock->read.data)			/* If no more locks in use */
    {
      /* Release write-locks with TL_WRITE or TL_WRITE_ONLY priority first */
      if (data &&
	  (data->type != TL_WRITE_LOW_PRIORITY || !lock->read_wait.data ||
unknown's avatar
unknown committed
875
	   lock->read_wait.data->type < TL_READ_HIGH_PRIORITY))
unknown's avatar
unknown committed
876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900
      {
	if (lock->write_lock_count++ > max_write_lock_count)
	{
	  /* Too many write locks in a row;  Release all waiting read locks */
	  lock->write_lock_count=0;
	  if (lock->read_wait.data)
	  {
	    DBUG_PRINT("info",("Freeing all read_locks because of max_write_lock_count"));
	    free_all_read_locks(lock,0);
	    goto end;
	  }
	}
	for (;;)
	{
	  if (((*data->prev)=data->next))	/* remove from wait-list */
	    data->next->prev= data->prev;
	  else
	    lock->write_wait.last=data->prev;
	  (*lock->write.last)=data;		/* Put in execute list */
	  data->prev=lock->write.last;
	  data->next=0;
	  lock->write.last= &data->next;
	  if (data->type == TL_WRITE_CONCURRENT_INSERT &&
	      (*lock->check_status)(data->status_param))
	    data->type=TL_WRITE;			/* Upgrade lock */
unknown's avatar
unknown committed
901
          /* purecov: begin inspected */
902
	  DBUG_PRINT("lock",("giving write lock of type %d to thread: 0x%lx",
903
			     data->type, data->owner->thread_id));
unknown's avatar
unknown committed
904
          /* purecov: end */
unknown's avatar
unknown committed
905
	  {
Marc Alff's avatar
Marc Alff committed
906
            mysql_cond_t *cond= data->cond;
unknown's avatar
unknown committed
907
	    data->cond=0;				/* Mark thread free */
Marc Alff's avatar
Marc Alff committed
908
            mysql_cond_signal(cond);                    /* Start waiting thread */
unknown's avatar
unknown committed
909 910 911 912 913 914 915 916
	  }
	  if (data->type != TL_WRITE_ALLOW_WRITE ||
	      !lock->write_wait.data ||
	      lock->write_wait.data->type != TL_WRITE_ALLOW_WRITE)
	    break;
	  data=lock->write_wait.data;		/* Free this too */
	}
	if (data->type >= TL_WRITE_LOW_PRIORITY)
917
          goto end;
unknown's avatar
unknown committed
918 919 920 921 922 923 924 925 926
	/* Release possible read locks together with the write lock */
      }
      if (lock->read_wait.data)
	free_all_read_locks(lock,
			    data &&
			    (data->type == TL_WRITE_CONCURRENT_INSERT ||
			     data->type == TL_WRITE_ALLOW_WRITE));
      else
      {
unknown's avatar
unknown committed
927
	DBUG_PRINT("lock",("No waiting read locks to free"));
unknown's avatar
unknown committed
928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948
      }
    }
    else if (data &&
	     (lock_type=data->type) <= TL_WRITE_DELAYED &&
	     ((lock_type != TL_WRITE_CONCURRENT_INSERT &&
	       lock_type != TL_WRITE_ALLOW_WRITE) ||
	      !lock->read_no_write_count))
    {
      /*
	For DELAYED, ALLOW_READ, WRITE_ALLOW_WRITE or CONCURRENT_INSERT locks
	start WRITE locks together with the READ locks
      */
      if (lock_type == TL_WRITE_CONCURRENT_INSERT &&
	  (*lock->check_status)(data->status_param))
      {
	data->type=TL_WRITE;			/* Upgrade lock */
	if (lock->read_wait.data)
	  free_all_read_locks(lock,0);
	goto end;
      }
      do {
Marc Alff's avatar
Marc Alff committed
949
        mysql_cond_t *cond= data->cond;
unknown's avatar
unknown committed
950 951 952 953
	if (((*data->prev)=data->next))		/* remove from wait-list */
	  data->next->prev= data->prev;
	else
	  lock->write_wait.last=data->prev;
unknown's avatar
unknown committed
954
	(*lock->write.last)=data;		/* Put in execute list */
unknown's avatar
unknown committed
955 956 957 958
	data->prev=lock->write.last;
	lock->write.last= &data->next;
	data->next=0;				/* Only one write lock */
	data->cond=0;				/* Mark thread free */
Marc Alff's avatar
Marc Alff committed
959
        mysql_cond_signal(cond);                /* Start waiting thread */
unknown's avatar
unknown committed
960 961 962 963 964 965 966 967
      } while (lock_type == TL_WRITE_ALLOW_WRITE &&
	       (data=lock->write_wait.data) &&
	       data->type == TL_WRITE_ALLOW_WRITE);
      if (lock->read_wait.data)
	free_all_read_locks(lock,
			    (lock_type == TL_WRITE_CONCURRENT_INSERT ||
			     lock_type == TL_WRITE_ALLOW_WRITE));
    }
unknown's avatar
unknown committed
968
    else if (!data && lock->read_wait.data)
unknown's avatar
unknown committed
969 970 971
      free_all_read_locks(lock,0);
  }
end:
972
  check_locks(lock, "after waking up waiters", 0);
unknown's avatar
unknown committed
973 974 975 976 977 978 979 980 981 982 983
  DBUG_VOID_RETURN;
}


/*
** Get all locks in a specific order to avoid dead-locks
** Sort acording to lock position and put write_locks before read_locks if
** lock on same lock.
*/


984
#define LOCK_CMP(A,B) ((uchar*) (A->lock) - (uint) ((A)->type) < (uchar*) (B->lock)- (uint) ((B)->type))
unknown's avatar
unknown committed
985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006

static void sort_locks(THR_LOCK_DATA **data,uint count)
{
  THR_LOCK_DATA **pos,**end,**prev,*tmp;

  /* Sort locks with insertion sort (fast because almost always few locks) */

  for (pos=data+1,end=data+count; pos < end ; pos++)
  {
    tmp= *pos;
    if (LOCK_CMP(tmp,pos[-1]))
    {
      prev=pos;
      do {
	prev[0]=prev[-1];
      } while (--prev != data && LOCK_CMP(tmp,prev[-1]));
      prev[0]=tmp;
    }
  }
}


1007
enum enum_thr_lock_result
1008
thr_multi_lock(THR_LOCK_DATA **data, uint count, THR_LOCK_INFO *owner,
1009
               ulong lock_wait_timeout)
unknown's avatar
unknown committed
1010 1011 1012
{
  THR_LOCK_DATA **pos,**end;
  DBUG_ENTER("thr_multi_lock");
unknown's avatar
unknown committed
1013
  DBUG_PRINT("lock",("data: 0x%lx  count: %d", (long) data, count));
unknown's avatar
unknown committed
1014 1015 1016 1017 1018
  if (count > 1)
    sort_locks(data,count);
  /* lock everything */
  for (pos=data,end=data+count; pos < end ; pos++)
  {
1019 1020
    enum enum_thr_lock_result result= thr_lock(*pos, owner, (*pos)->type,
                                               lock_wait_timeout);
1021
    if (result != THR_LOCK_SUCCESS)
unknown's avatar
unknown committed
1022 1023
    {						/* Aborted */
      thr_multi_unlock(data,(uint) (pos-data));
1024
      DBUG_RETURN(result);
unknown's avatar
unknown committed
1025
    }
1026
    DEBUG_SYNC_C("thr_multi_lock_after_thr_lock");
unknown's avatar
unknown committed
1027
#ifdef MAIN
1028
    printf("Thread: %s  Got lock: 0x%lx  type: %d\n",my_thread_name(),
unknown's avatar
unknown committed
1029 1030 1031
	   (long) pos[0]->lock, pos[0]->type); fflush(stdout);
#endif
  }
Konstantin Osipov's avatar
Konstantin Osipov committed
1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065
  thr_lock_merge_status(data, count);
  DBUG_RETURN(THR_LOCK_SUCCESS);
}


/**
  Ensure that all locks for a given table have the same
  status_param.

  This is a MyISAM and possibly Maria specific crutch. MyISAM
  engine stores data file length, record count and other table
  properties in status_param member of handler. When a table is
  locked, connection-local copy is made from a global copy
  (myisam_share) by mi_get_status(). When a table is unlocked,
  the changed status is transferred back to the global share by
  mi_update_status().

  One thing MyISAM doesn't do is to ensure that when the same
  table is opened twice in a connection all instances share the
  same status_param. This is necessary, however: for one, to keep
  all instances of a connection "on the same page" with regard to
  the current state of the table. For other, unless this is done,
  myisam_share will always get updated from the last unlocked
  instance (in mi_update_status()), and when this instance was not
  the one that was used to update data, records may be lost.

  For each table, this function looks up the last lock_data in the
  list of acquired locks, and makes sure that all other instances
  share status_param with it.
*/

void
thr_lock_merge_status(THR_LOCK_DATA **data, uint count)
{
unknown's avatar
unknown committed
1066
#if !defined(DONT_USE_RW_LOCKS)
Konstantin Osipov's avatar
Konstantin Osipov committed
1067 1068
  THR_LOCK_DATA **pos= data;
  THR_LOCK_DATA **end= data + count;
unknown's avatar
unknown committed
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117
  if (count > 1)
  {
    THR_LOCK_DATA *last_lock= end[-1];
    pos=end-1;
    do
    {
      pos--;
      if (last_lock->lock == (*pos)->lock &&
	  last_lock->lock->copy_status)
      {
	if (last_lock->type <= TL_READ_NO_INSERT)
	{
	  THR_LOCK_DATA **read_lock;
	  /*
	    If we are locking the same table with read locks we must ensure
	    that all tables share the status of the last write lock or
	    the same read lock.
	  */
	  for (;
	       (*pos)->type <= TL_READ_NO_INSERT &&
		 pos != data &&
		 pos[-1]->lock == (*pos)->lock ;
	       pos--) ;

	  read_lock = pos+1;
	  do
	  {
	    (last_lock->lock->copy_status)((*read_lock)->status_param,
					   (*pos)->status_param);
	  } while (*(read_lock++) != last_lock);
	  last_lock= (*pos);			/* Point at last write lock */
	}
	else
	  (*last_lock->lock->copy_status)((*pos)->status_param,
					  last_lock->status_param);
      }
      else
	last_lock=(*pos);
    } while (pos != data);
  }
#endif
}

  /* free all locks */

void thr_multi_unlock(THR_LOCK_DATA **data,uint count)
{
  THR_LOCK_DATA **pos,**end;
  DBUG_ENTER("thr_multi_unlock");
unknown's avatar
unknown committed
1118
  DBUG_PRINT("lock",("data: 0x%lx  count: %d", (long) data, count));
unknown's avatar
unknown committed
1119 1120 1121 1122

  for (pos=data,end=data+count; pos < end ; pos++)
  {
#ifdef MAIN
1123
    printf("Thread: %s  Rel lock: 0x%lx  type: %d\n",
unknown's avatar
unknown committed
1124 1125 1126 1127 1128 1129 1130
	   my_thread_name(), (long) pos[0]->lock, pos[0]->type);
    fflush(stdout);
#endif
    if ((*pos)->type != TL_UNLOCK)
      thr_unlock(*pos);
    else
    {
unknown's avatar
unknown committed
1131
      DBUG_PRINT("lock",("Free lock: data: 0x%lx  thread: 0x%lx  lock: 0x%lx",
1132
                         (long) *pos, (*pos)->owner->thread_id,
unknown's avatar
unknown committed
1133
                         (long) (*pos)->lock));
unknown's avatar
unknown committed
1134 1135 1136 1137 1138
    }
  }
  DBUG_VOID_RETURN;
}

unknown's avatar
unknown committed
1139 1140
/*
  Abort all threads waiting for a lock. The lock will be upgraded to
unknown's avatar
unknown committed
1141 1142 1143
  TL_WRITE_ONLY to abort any new accesses to the lock
*/

1144
void thr_abort_locks(THR_LOCK *lock, my_bool upgrade_lock)
unknown's avatar
unknown committed
1145 1146 1147
{
  THR_LOCK_DATA *data;
  DBUG_ENTER("thr_abort_locks");
Marc Alff's avatar
Marc Alff committed
1148
  mysql_mutex_lock(&lock->mutex);
unknown's avatar
unknown committed
1149 1150 1151 1152

  for (data=lock->read_wait.data; data ; data=data->next)
  {
    data->type=TL_UNLOCK;			/* Mark killed */
1153
    /* It's safe to signal the cond first: we're still holding the mutex. */
Marc Alff's avatar
Marc Alff committed
1154
    mysql_cond_signal(data->cond);
unknown's avatar
unknown committed
1155 1156 1157 1158 1159
    data->cond=0;				/* Removed from list */
  }
  for (data=lock->write_wait.data; data ; data=data->next)
  {
    data->type=TL_UNLOCK;
Marc Alff's avatar
Marc Alff committed
1160
    mysql_cond_signal(data->cond);
unknown's avatar
unknown committed
1161 1162 1163 1164 1165
    data->cond=0;
  }
  lock->read_wait.last= &lock->read_wait.data;
  lock->write_wait.last= &lock->write_wait.data;
  lock->read_wait.data=lock->write_wait.data=0;
unknown's avatar
unknown committed
1166
  if (upgrade_lock && lock->write.data)
unknown's avatar
unknown committed
1167
    lock->write.data->type=TL_WRITE_ONLY;
Marc Alff's avatar
Marc Alff committed
1168
  mysql_mutex_unlock(&lock->mutex);
unknown's avatar
unknown committed
1169 1170 1171 1172
  DBUG_VOID_RETURN;
}


1173 1174 1175 1176 1177 1178
/*
  Abort all locks for specific table/thread combination

  This is used to abort all locks for a specific thread
*/

unknown's avatar
unknown committed
1179
my_bool thr_abort_locks_for_thread(THR_LOCK *lock, my_thread_id thread_id)
1180 1181
{
  THR_LOCK_DATA *data;
unknown's avatar
unknown committed
1182
  my_bool found= FALSE;
1183 1184
  DBUG_ENTER("thr_abort_locks_for_thread");

Marc Alff's avatar
Marc Alff committed
1185
  mysql_mutex_lock(&lock->mutex);
1186 1187
  for (data= lock->read_wait.data; data ; data= data->next)
  {
1188
    if (data->owner->thread_id == thread_id)    /* purecov: tested */
1189 1190 1191
    {
      DBUG_PRINT("info",("Aborting read-wait lock"));
      data->type= TL_UNLOCK;			/* Mark killed */
1192
      /* It's safe to signal the cond first: we're still holding the mutex. */
unknown's avatar
unknown committed
1193
      found= TRUE;
Marc Alff's avatar
Marc Alff committed
1194
      mysql_cond_signal(data->cond);
1195 1196 1197 1198 1199 1200 1201 1202 1203 1204
      data->cond= 0;				/* Removed from list */

      if (((*data->prev)= data->next))
	data->next->prev= data->prev;
      else
	lock->read_wait.last= data->prev;
    }
  }
  for (data= lock->write_wait.data; data ; data= data->next)
  {
1205
    if (data->owner->thread_id == thread_id) /* purecov: tested */
1206 1207 1208
    {
      DBUG_PRINT("info",("Aborting write-wait lock"));
      data->type= TL_UNLOCK;
unknown's avatar
unknown committed
1209
      found= TRUE;
Marc Alff's avatar
Marc Alff committed
1210
      mysql_cond_signal(data->cond);
1211 1212 1213 1214 1215 1216 1217 1218
      data->cond= 0;

      if (((*data->prev)= data->next))
	data->next->prev= data->prev;
      else
	lock->write_wait.last= data->prev;
    }
  }
1219
  wake_up_waiters(lock);
Marc Alff's avatar
Marc Alff committed
1220
  mysql_mutex_unlock(&lock->mutex);
unknown's avatar
unknown committed
1221
  DBUG_RETURN(found);
1222 1223 1224
}


unknown's avatar
unknown committed
1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237
/*
  Downgrade a WRITE_* to a lower WRITE level
  SYNOPSIS
    thr_downgrade_write_lock()
    in_data                   Lock data of thread downgrading its lock
    new_lock_type             New write lock type
  RETURN VALUE
    NONE
  DESCRIPTION
    This can be used to downgrade a lock already owned. When the downgrade
    occurs also other waiters, both readers and writers can be allowed to
    start.
    The previous lock is often TL_WRITE_ONLY but can also be
1238
    TL_WRITE. The normal downgrade variants are:
unknown's avatar
unknown committed
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255
    TL_WRITE_ONLY => TL_WRITE after a short exclusive lock while holding a
    write table lock
    TL_WRITE_ONLY => TL_WRITE_ALLOW_WRITE After a short exclusive lock after
    already earlier having dongraded lock to TL_WRITE_ALLOW_WRITE
    The implementation is conservative and rather don't start rather than
    go on unknown paths to start, the common cases are handled.

    NOTE:
    In its current implementation it is only allowed to downgrade from
    TL_WRITE_ONLY. In this case there are no waiters. Thus no wake up
    logic is required.
*/

void thr_downgrade_write_lock(THR_LOCK_DATA *in_data,
                              enum thr_lock_type new_lock_type)
{
  THR_LOCK *lock=in_data->lock;
unknown's avatar
unknown committed
1256
#ifndef DBUG_OFF
unknown's avatar
unknown committed
1257
  enum thr_lock_type old_lock_type= in_data->type;
unknown's avatar
unknown committed
1258
#endif
unknown's avatar
unknown committed
1259 1260
  DBUG_ENTER("thr_downgrade_write_only_lock");

Marc Alff's avatar
Marc Alff committed
1261
  mysql_mutex_lock(&lock->mutex);
unknown's avatar
unknown committed
1262 1263 1264 1265
  DBUG_ASSERT(old_lock_type == TL_WRITE_ONLY);
  DBUG_ASSERT(old_lock_type > new_lock_type);
  in_data->type= new_lock_type;
  check_locks(lock,"after downgrading lock",0);
unknown's avatar
unknown committed
1266

Marc Alff's avatar
Marc Alff committed
1267
  mysql_mutex_unlock(&lock->mutex);
unknown's avatar
unknown committed
1268 1269
  DBUG_VOID_RETURN;
}
1270

unknown's avatar
unknown committed
1271 1272
/* Upgrade a WRITE_DELAY lock to a WRITE_LOCK */

1273
my_bool thr_upgrade_write_delay_lock(THR_LOCK_DATA *data,
1274 1275
                                     enum thr_lock_type new_lock_type,
                                     ulong lock_wait_timeout)
unknown's avatar
unknown committed
1276 1277 1278 1279
{
  THR_LOCK *lock=data->lock;
  DBUG_ENTER("thr_upgrade_write_delay_lock");

Marc Alff's avatar
Marc Alff committed
1280
  mysql_mutex_lock(&lock->mutex);
1281
  if (data->type == TL_UNLOCK || data->type >= TL_WRITE_LOW_PRIORITY)
unknown's avatar
unknown committed
1282
  {
Marc Alff's avatar
Marc Alff committed
1283
    mysql_mutex_unlock(&lock->mutex);
1284
    DBUG_RETURN(data->type == TL_UNLOCK);	/* Test if Aborted */
unknown's avatar
unknown committed
1285 1286 1287
  }
  check_locks(lock,"before upgrading lock",0);
  /* TODO:  Upgrade to TL_WRITE_CONCURRENT_INSERT in some cases */
1288
  data->type= new_lock_type;                    /* Upgrade lock */
unknown's avatar
unknown committed
1289 1290 1291 1292 1293 1294 1295

  /* Check if someone has given us the lock */
  if (!data->cond)
  {
    if (!lock->read.data)			/* No read locks */
    {						/* We have the lock */
      if (data->lock->get_status)
1296
	(*data->lock->get_status)(data->status_param, 0);
Marc Alff's avatar
Marc Alff committed
1297
      mysql_mutex_unlock(&lock->mutex);
unknown's avatar
unknown committed
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313
      DBUG_RETURN(0);
    }

    if (((*data->prev)=data->next))		/* remove from lock-list */
      data->next->prev= data->prev;
    else
      lock->write.last=data->prev;

    if ((data->next=lock->write_wait.data))	/* Put first in lock_list */
      data->next->prev= &data->next;
    else
      lock->write_wait.last= &data->next;
    data->prev= &lock->write_wait.data;
    lock->write_wait.data=data;
    check_locks(lock,"upgrading lock",0);
  }
unknown's avatar
unknown committed
1314 1315 1316 1317
  else
  {
    check_locks(lock,"waiting for lock",0);
  }
1318
  DBUG_RETURN(wait_for_lock(&lock->write_wait,data,1, lock_wait_timeout));
unknown's avatar
unknown committed
1319 1320 1321 1322 1323
}


/* downgrade a WRITE lock to a WRITE_DELAY lock if there is pending locks */

1324 1325
my_bool thr_reschedule_write_lock(THR_LOCK_DATA *data,
                                  ulong lock_wait_timeout)
unknown's avatar
unknown committed
1326 1327
{
  THR_LOCK *lock=data->lock;
1328
  enum thr_lock_type write_lock_type;
unknown's avatar
unknown committed
1329 1330
  DBUG_ENTER("thr_reschedule_write_lock");

Marc Alff's avatar
Marc Alff committed
1331
  mysql_mutex_lock(&lock->mutex);
unknown's avatar
unknown committed
1332 1333
  if (!lock->read_wait.data)			/* No waiting read locks */
  {
Marc Alff's avatar
Marc Alff committed
1334
    mysql_mutex_unlock(&lock->mutex);
unknown's avatar
unknown committed
1335 1336 1337
    DBUG_RETURN(0);
  }

1338
  write_lock_type= data->type;
unknown's avatar
unknown committed
1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355
  data->type=TL_WRITE_DELAYED;
  if (lock->update_status)
    (*lock->update_status)(data->status_param);
  if (((*data->prev)=data->next))		/* remove from lock-list */
    data->next->prev= data->prev;
  else
    lock->write.last=data->prev;

  if ((data->next=lock->write_wait.data))	/* Put first in lock_list */
    data->next->prev= &data->next;
  else
    lock->write_wait.last= &data->next;
  data->prev= &lock->write_wait.data;
  data->cond=get_cond();			/* This was zero */
  lock->write_wait.data=data;
  free_all_read_locks(lock,0);

Marc Alff's avatar
Marc Alff committed
1356
  mysql_mutex_unlock(&lock->mutex);
1357 1358
  DBUG_RETURN(thr_upgrade_write_delay_lock(data, write_lock_type,
                                           lock_wait_timeout));
unknown's avatar
unknown committed
1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374
}


#include <my_sys.h>

static void thr_print_lock(const char* name,struct st_lock_list *list)
{
  THR_LOCK_DATA *data,**prev;
  uint count=0;

  if (list->data)
  {
    printf("%-10s: ",name);
    prev= &list->data;
    for (data=list->data; data && count++ < MAX_LOCKS ; data=data->next)
    {
1375
      printf("0x%lx (%lu:%d); ", (ulong) data, data->owner->thread_id,
1376
             (int) data->type);
unknown's avatar
unknown committed
1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391
      if (data->prev != prev)
	printf("\nWarning: prev didn't point at previous lock\n");
      prev= &data->next;
    }
    puts("");
    if (prev != list->last)
      printf("Warning: last didn't point at last lock\n");
  }
}

void thr_print_locks(void)
{
  LIST *list;
  uint count=0;

Marc Alff's avatar
Marc Alff committed
1392
  mysql_mutex_lock(&THR_LOCK_lock);
unknown's avatar
unknown committed
1393
  puts("Current locks:");
1394 1395
  for (list= thr_lock_thread_list; list && count++ < MAX_THREADS;
       list= list_rest(list))
unknown's avatar
unknown committed
1396 1397
  {
    THR_LOCK *lock=(THR_LOCK*) list->data;
Marc Alff's avatar
Marc Alff committed
1398
    mysql_mutex_lock(&lock->mutex);
1399
    printf("lock: 0x%lx:",(ulong) lock);
unknown's avatar
unknown committed
1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415
    if ((lock->write_wait.data || lock->read_wait.data) &&
	(! lock->read.data && ! lock->write.data))
      printf(" WARNING: ");
    if (lock->write.data)
      printf(" write");
    if (lock->write_wait.data)
      printf(" write_wait");
    if (lock->read.data)
      printf(" read");
    if (lock->read_wait.data)
      printf(" read_wait");
    puts("");
    thr_print_lock("write",&lock->write);
    thr_print_lock("write_wait",&lock->write_wait);
    thr_print_lock("read",&lock->read);
    thr_print_lock("read_wait",&lock->read_wait);
Marc Alff's avatar
Marc Alff committed
1416
    mysql_mutex_unlock(&lock->mutex);
unknown's avatar
unknown committed
1417 1418 1419
    puts("");
  }
  fflush(stdout);
Marc Alff's avatar
Marc Alff committed
1420
  mysql_mutex_unlock(&THR_LOCK_lock);
unknown's avatar
unknown committed
1421 1422
}

unknown's avatar
unknown committed
1423 1424 1425 1426 1427 1428
#endif /* THREAD */

/*****************************************************************************
** Test of thread locks
****************************************************************************/

unknown's avatar
unknown committed
1429 1430
#ifdef MAIN

unknown's avatar
unknown committed
1431 1432
#ifdef THREAD

unknown's avatar
unknown committed
1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
struct st_test {
  uint lock_nr;
  enum thr_lock_type lock_type;
};

THR_LOCK locks[5];			/* 4 locks */

struct st_test test_0[] = {{0,TL_READ}};	/* One lock */
struct st_test test_1[] = {{0,TL_READ},{0,TL_WRITE}}; /* Read and write lock of lock 0 */
struct st_test test_2[] = {{1,TL_WRITE},{0,TL_READ},{2,TL_READ}};
struct st_test test_3[] = {{2,TL_WRITE},{1,TL_READ},{0,TL_READ}}; /* Deadlock with test_2 ? */
struct st_test test_4[] = {{0,TL_WRITE},{0,TL_READ},{0,TL_WRITE},{0,TL_READ}};
struct st_test test_5[] = {{0,TL_READ},{1,TL_READ},{2,TL_READ},{3,TL_READ}}; /* Many reads */
struct st_test test_6[] = {{0,TL_WRITE},{1,TL_WRITE},{2,TL_WRITE},{3,TL_WRITE}}; /* Many writes */
struct st_test test_7[] = {{3,TL_READ}};
struct st_test test_8[] = {{1,TL_READ_NO_INSERT},{2,TL_READ_NO_INSERT},{3,TL_READ_NO_INSERT}};	/* Should be quick */
struct st_test test_9[] = {{4,TL_READ_HIGH_PRIORITY}};
struct st_test test_10[] ={{4,TL_WRITE}};
struct st_test test_11[] = {{0,TL_WRITE_LOW_PRIORITY},{1,TL_WRITE_LOW_PRIORITY},{2,TL_WRITE_LOW_PRIORITY},{3,TL_WRITE_LOW_PRIORITY}}; /* Many writes */
1452 1453 1454 1455
struct st_test test_12[] = {{0,TL_WRITE_CONCURRENT_INSERT},{1,TL_WRITE_CONCURRENT_INSERT},{2,TL_WRITE_CONCURRENT_INSERT},{3,TL_WRITE_CONCURRENT_INSERT}};
struct st_test test_13[] = {{0,TL_WRITE_CONCURRENT_INSERT},{1,TL_READ}};
struct st_test test_14[] = {{0,TL_WRITE_ALLOW_WRITE},{1,TL_READ}};
struct st_test test_15[] = {{0,TL_WRITE_ALLOW_WRITE},{1,TL_WRITE_ALLOW_WRITE}};
unknown's avatar
unknown committed
1456 1457 1458

struct st_test *tests[] = {test_0,test_1,test_2,test_3,test_4,test_5,test_6,
			   test_7,test_8,test_9,test_10,test_11,test_12,
1459
			   test_13,test_14,test_15};
unknown's avatar
unknown committed
1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474
int lock_counts[]= {sizeof(test_0)/sizeof(struct st_test),
		    sizeof(test_1)/sizeof(struct st_test),
		    sizeof(test_2)/sizeof(struct st_test),
		    sizeof(test_3)/sizeof(struct st_test),
		    sizeof(test_4)/sizeof(struct st_test),
		    sizeof(test_5)/sizeof(struct st_test),
		    sizeof(test_6)/sizeof(struct st_test),
		    sizeof(test_7)/sizeof(struct st_test),
		    sizeof(test_8)/sizeof(struct st_test),
		    sizeof(test_9)/sizeof(struct st_test),
		    sizeof(test_10)/sizeof(struct st_test),
		    sizeof(test_11)/sizeof(struct st_test),
		    sizeof(test_12)/sizeof(struct st_test),
		    sizeof(test_13)/sizeof(struct st_test),
		    sizeof(test_14)/sizeof(struct st_test),
1475
		    sizeof(test_15)/sizeof(struct st_test)
unknown's avatar
unknown committed
1476 1477 1478
};


Marc Alff's avatar
Marc Alff committed
1479 1480
static mysql_cond_t COND_thread_count;
static mysql_mutex_t LOCK_thread_count;
unknown's avatar
unknown committed
1481 1482 1483 1484
static uint thread_count;
static ulong sum=0;

#define MAX_LOCK_COUNT 8
1485
#define TEST_TIMEOUT 100000
unknown's avatar
unknown committed
1486 1487 1488

/* The following functions is for WRITE_CONCURRENT_INSERT */

1489 1490
static void test_get_status(void* param __attribute__((unused)),
                            int concurrent_insert __attribute__((unused)))
unknown's avatar
unknown committed
1491 1492 1493
{
}

unknown's avatar
unknown committed
1494 1495 1496 1497
static void test_update_status(void* param __attribute__((unused)))
{
}

unknown's avatar
unknown committed
1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512
static void test_copy_status(void* to __attribute__((unused)) ,
			     void *from __attribute__((unused)))
{
}

static my_bool test_check_status(void* param __attribute__((unused)))
{
  return 0;
}


static void *test_thread(void *arg)
{
  int i,j,param=*((int*) arg);
  THR_LOCK_DATA data[MAX_LOCK_COUNT];
1513
  THR_LOCK_INFO lock_info;
unknown's avatar
unknown committed
1514 1515 1516 1517 1518
  THR_LOCK_DATA *multi_locks[MAX_LOCK_COUNT];
  my_thread_init();

  printf("Thread %s (%d) started\n",my_thread_name(),param); fflush(stdout);

1519 1520

  thr_lock_info_init(&lock_info);
unknown's avatar
unknown committed
1521 1522 1523 1524 1525 1526 1527 1528 1529
  for (i=0; i < lock_counts[param] ; i++)
    thr_lock_data_init(locks+tests[param][i].lock_nr,data+i,NULL);
  for (j=1 ; j < 10 ; j++)		/* try locking 10 times */
  {
    for (i=0; i < lock_counts[param] ; i++)
    {					/* Init multi locks */
      multi_locks[i]= &data[i];
      data[i].type= tests[param][i].lock_type;
    }
1530
    thr_multi_lock(multi_locks, lock_counts[param], &lock_info, TEST_TIMEOUT);
Marc Alff's avatar
Marc Alff committed
1531
    mysql_mutex_lock(&LOCK_thread_count);
unknown's avatar
unknown committed
1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544
    {
      int tmp=rand() & 7;			/* Do something from 0-2 sec */
      if (tmp == 0)
	sleep(1);
      else if (tmp == 1)
	sleep(2);
      else
      {
	ulong k;
	for (k=0 ; k < (ulong) (tmp-2)*100000L ; k++)
	  sum+=k;
      }
    }
Marc Alff's avatar
Marc Alff committed
1545
    mysql_mutex_unlock(&LOCK_thread_count);
unknown's avatar
unknown committed
1546 1547 1548
    thr_multi_unlock(multi_locks,lock_counts[param]);
  }

unknown's avatar
unknown committed
1549
  printf("Thread %s (%d) ended\n",my_thread_name(),param); fflush(stdout);
unknown's avatar
unknown committed
1550
  thr_print_locks();
Marc Alff's avatar
Marc Alff committed
1551
  mysql_mutex_lock(&LOCK_thread_count);
unknown's avatar
unknown committed
1552
  thread_count--;
Marc Alff's avatar
Marc Alff committed
1553 1554
  mysql_cond_signal(&COND_thread_count); /* Tell main we are ready */
  mysql_mutex_unlock(&LOCK_thread_count);
1555
  free((uchar*) arg);
unknown's avatar
unknown committed
1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570
  return 0;
}


int main(int argc __attribute__((unused)),char **argv __attribute__((unused)))
{
  pthread_t tid;
  pthread_attr_t thr_attr;
  int i,*param,error;
  MY_INIT(argv[0]);
  if (argc > 1 && argv[1][0] == '-' && argv[1][1] == '#')
    DBUG_PUSH(argv[1]+2);

  printf("Main thread: %s\n",my_thread_name());

Marc Alff's avatar
Marc Alff committed
1571
  if ((error= mysql_cond_init(0, &COND_thread_count, NULL)))
unknown's avatar
unknown committed
1572
  {
Marc Alff's avatar
Marc Alff committed
1573
    fprintf(stderr, "Got error: %d from mysql_cond_init (errno: %d)",
unknown's avatar
unknown committed
1574 1575 1576
	    error,errno);
    exit(1);
  }
Marc Alff's avatar
Marc Alff committed
1577
  if ((error= mysql_mutex_init(0, &LOCK_thread_count, MY_MUTEX_INIT_FAST)))
unknown's avatar
unknown committed
1578
  {
Marc Alff's avatar
Marc Alff committed
1579
    fprintf(stderr, "Got error: %d from mysql_cond_init (errno: %d)",
unknown's avatar
unknown committed
1580 1581 1582 1583 1584 1585 1586 1587
	    error,errno);
    exit(1);
  }

  for (i=0 ; i < (int) array_elements(locks) ; i++)
  {
    thr_lock_init(locks+i);
    locks[i].check_status= test_check_status;
unknown's avatar
unknown committed
1588
    locks[i].update_status=test_update_status;
unknown's avatar
unknown committed
1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613
    locks[i].copy_status=  test_copy_status;
    locks[i].get_status=   test_get_status;
  }
  if ((error=pthread_attr_init(&thr_attr)))
  {
    fprintf(stderr,"Got error: %d from pthread_attr_init (errno: %d)",
	    error,errno);
    exit(1);
  }
  if ((error=pthread_attr_setdetachstate(&thr_attr,PTHREAD_CREATE_DETACHED)))
  {
    fprintf(stderr,
	    "Got error: %d from pthread_attr_setdetachstate (errno: %d)",
	    error,errno);
    exit(1);
  }
#ifndef pthread_attr_setstacksize		/* void return value */
  if ((error=pthread_attr_setstacksize(&thr_attr,65536L)))
  {
    fprintf(stderr,"Got error: %d from pthread_attr_setstacksize (errno: %d)",
	    error,errno);
    exit(1);
  }
#endif
#ifdef HAVE_THR_SETCONCURRENCY
Konstantin Osipov's avatar
Konstantin Osipov committed
1614
  (void) thr_setconcurrency(2);
unknown's avatar
unknown committed
1615 1616 1617 1618 1619 1620
#endif
  for (i=0 ; i < (int) array_elements(lock_counts) ; i++)
  {
    param=(int*) malloc(sizeof(int));
    *param=i;

Marc Alff's avatar
Marc Alff committed
1621
    if ((error= mysql_mutex_lock(&LOCK_thread_count)))
unknown's avatar
unknown committed
1622
    {
Marc Alff's avatar
Marc Alff committed
1623 1624
      fprintf(stderr, "Got error: %d from mysql_mutex_lock (errno: %d)",
              error, errno);
unknown's avatar
unknown committed
1625 1626
      exit(1);
    }
Marc Alff's avatar
Marc Alff committed
1627 1628 1629
    if ((error= mysql_thread_create(0,
                                    &tid, &thr_attr, test_thread,
                                    (void*) param)))
unknown's avatar
unknown committed
1630
    {
Marc Alff's avatar
Marc Alff committed
1631 1632 1633
      fprintf(stderr, "Got error: %d from mysql_thread_create (errno: %d)\n",
              error, errno);
      mysql_mutex_unlock(&LOCK_thread_count);
unknown's avatar
unknown committed
1634 1635 1636
      exit(1);
    }
    thread_count++;
Marc Alff's avatar
Marc Alff committed
1637
    mysql_mutex_unlock(&LOCK_thread_count);
unknown's avatar
unknown committed
1638 1639 1640
  }

  pthread_attr_destroy(&thr_attr);
Marc Alff's avatar
Marc Alff committed
1641 1642
  if ((error= mysql_mutex_lock(&LOCK_thread_count)))
    fprintf(stderr, "Got error: %d from mysql_mutex_lock\n", error);
unknown's avatar
unknown committed
1643 1644
  while (thread_count)
  {
Marc Alff's avatar
Marc Alff committed
1645 1646
    if ((error= mysql_cond_wait(&COND_thread_count, &LOCK_thread_count)))
      fprintf(stderr, "Got error: %d from mysql_cond_wait\n", error);
unknown's avatar
unknown committed
1647
  }
Marc Alff's avatar
Marc Alff committed
1648 1649
  if ((error= mysql_mutex_unlock(&LOCK_thread_count)))
    fprintf(stderr, "Got error: %d from mysql_mutex_unlock\n", error);
unknown's avatar
unknown committed
1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660
  for (i=0 ; i < (int) array_elements(locks) ; i++)
    thr_lock_delete(locks+i);
#ifdef EXTRA_DEBUG
  if (found_errors)
    printf("Got %d warnings\n",found_errors);
  else
#endif
    printf("Test succeeded\n");
  return 0;
}

unknown's avatar
unknown committed
1661 1662 1663 1664 1665 1666 1667 1668 1669 1670
#else /* THREAD */

int main(int argc __attribute__((unused)),char **argv __attribute__((unused)))
{
  printf("thr_lock disabled because we are not using threads\n");
  exit(1);
}

#endif /* THREAD */
#endif /* MAIN */