SHM_Transporter.unix.cpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:2k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 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. #include <ndb_global.h>
  14. #include "SHM_Transporter.hpp"
  15. #include "TransporterInternalDefinitions.hpp"
  16. #include <TransporterCallback.hpp>
  17. #include <NdbSleep.h>
  18. #include <NdbOut.hpp>
  19. #include <sys/ipc.h>
  20. #include <sys/shm.h>
  21. bool
  22. SHM_Transporter::ndb_shm_create()
  23. {
  24.   shmId = shmget(shmKey, shmSize, IPC_CREAT | 960);
  25.   if(shmId == -1) {
  26.     perror("shmget: ");
  27.     return false;
  28.   }
  29.   return true;
  30. }
  31. bool
  32. SHM_Transporter::ndb_shm_get()
  33. {
  34.   shmId = shmget(shmKey, shmSize, 0);
  35.   if(shmId == -1) {
  36.     perror("shmget: ");
  37.     return false;
  38.   }
  39.   return true;
  40. }
  41. bool
  42. SHM_Transporter::ndb_shm_attach()
  43. {
  44.   shmBuf = (char *)shmat(shmId, 0, 0);
  45.   if(shmBuf == 0) {
  46.     perror("shmat: ");
  47.     return false;
  48.   }
  49.   return true;
  50. }
  51. bool
  52. SHM_Transporter::checkConnected(){
  53.   struct shmid_ds info;
  54.   const int res = shmctl(shmId, IPC_STAT, &info);
  55.   if(res == -1){
  56.     report_error(TE_SHM_IPC_STAT);
  57.     return false;
  58.   }
  59.  
  60.   if(info.shm_nattch != 2){
  61.     report_error(TE_SHM_DISCONNECT);
  62.     return false;
  63.   }
  64.   return true;
  65. }
  66. void
  67. SHM_Transporter::disconnectImpl(){
  68.   if(_attached){
  69.     const int res = shmdt(shmBuf);
  70.     if(res == -1){
  71.       perror("shmdelete: ");
  72.       return;   
  73.     }
  74.     _attached = false;
  75.     if(!isServer && _shmSegCreated)
  76.       _shmSegCreated = false;
  77.   }
  78.   
  79.   if(isServer && _shmSegCreated){
  80.     const int res = shmctl(shmId, IPC_RMID, 0);
  81.     if(res == -1){
  82.       report_error(TE_SHM_UNABLE_TO_REMOVE_SEGMENT);
  83.       return;
  84.     }
  85.     _shmSegCreated = false;
  86.   }
  87.   setupBuffersDone=false;
  88. }