my_lockmem.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:3k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. /* Alloc a block of locked memory */
  14. #include "mysys_priv.h"
  15. #include "mysys_err.h"
  16. #include <my_list.h>
  17. #ifdef HAVE_MLOCK
  18. #include <sys/mman.h>
  19. struct st_mem_list
  20. {
  21.   LIST list;
  22.   byte *page;
  23.   uint size;
  24. };
  25. LIST *mem_list;
  26. byte *my_malloc_lock(uint size,myf MyFlags)
  27. {
  28.   int success;
  29.   uint pagesize=sysconf(_SC_PAGESIZE);
  30.   byte *ptr;
  31.   struct st_mem_list *element;
  32.   DBUG_ENTER("my_malloc_lock");
  33.   size=((size-1) & ~(pagesize-1))+pagesize;
  34.   if (!(ptr=memalign(pagesize,size)))
  35.   {
  36.     if (MyFlags & (MY_FAE+MY_WME))
  37.       my_error(EE_OUTOFMEMORY, MYF(ME_BELL+ME_WAITTANG),size);
  38.     DBUG_RETURN(0);
  39.   }
  40.   success = mlock((byte*) ptr,size);
  41.   if (success != 0 && geteuid() == 0)
  42.   {
  43.     DBUG_PRINT("warning",("Failed to lock memory. errno %dn",
  44.   errno));
  45.     fprintf(stderr, "Warning: Failed to lock memory. errno %dn",
  46.     errno);
  47.   }
  48.   else
  49.   {
  50.     /* Add block in a list for munlock */
  51.     if (!(element=(struct st_mem_list*) my_malloc(sizeof(*element),MyFlags)))
  52.     {
  53.       VOID(munlock((byte*) ptr,size));
  54.       free(ptr);
  55.       DBUG_RETURN(0);
  56.     }
  57.     element->list.data=(byte*) element;
  58.     element->page=ptr;
  59.     element->size=size;
  60.     pthread_mutex_lock(&THR_LOCK_malloc);
  61.     mem_list=list_add(mem_list,&element->list);
  62.     pthread_mutex_unlock(&THR_LOCK_malloc);
  63.   }
  64.   DBUG_RETURN(ptr);
  65. }
  66. void my_free_lock(byte *ptr,myf Myflags __attribute__((unused)))
  67. {
  68.   LIST *list;
  69.   struct st_mem_list *element=0;
  70.   pthread_mutex_lock(&THR_LOCK_malloc);
  71.   for (list=mem_list ; list ; list=list->next)
  72.   {
  73.     element=(struct st_mem_list*) list->data;
  74.     if (ptr == element->page)
  75.     { /* Found locked mem */
  76.       VOID(munlock((byte*) ptr,element->size));
  77.       mem_list=list_delete(mem_list,list);
  78.       break;
  79.     }
  80.   }
  81.   pthread_mutex_unlock(&THR_LOCK_malloc);
  82.   if (element)
  83.     my_free((gptr) element,MYF(0));
  84.   free(ptr); /* Free even if not locked */
  85. }
  86. #endif /* HAVE_MLOCK */