shmhelp.cpp

著名的 helix realplayer 基于手机 symbian 系统的 播放器全套源代码

/* ***** BEGIN LICENSE BLOCK ***** 
 * Version: RCSL 1.0/RPSL 1.0 
 *  
 * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved. 
 *      
 * The contents of this file, and the files included with this file, are 
 * subject to the current version of the RealNetworks Public Source License 
 * Version 1.0 (the "RPSL") available at 
 * http://www.helixcommunity.org/content/rpsl unless you have licensed 
 * the file under the RealNetworks Community Source License Version 1.0 
 * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl, 
 * in which case the RCSL will apply. You may also obtain the license terms 
 * directly from RealNetworks.  You may not use this file except in 
 * compliance with the RPSL or, if you have a valid RCSL with RealNetworks 
 * applicable to this file, the RCSL.  Please see the applicable RPSL or 
 * RCSL for the rights, obligations and limitations governing use of the 
 * contents of the file.  
 *  
 * This file is part of the Helix DNA Technology. RealNetworks is the 
 * developer of the Original Code and owns the copyrights in the portions 
 * it created. 
 *  
 * This file, and the files included with this file, is distributed and made 
 * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 
 * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES, 
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS 
 * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 
 * 
 * Technology Compatibility Kit Test Suite(s) Location: 
 *    http://www.helixcommunity.org/content/tck 
 * 
 * Contributor(s): 
 *  
 * ***** END LICENSE BLOCK ***** */ 

// for shared memory
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/utsname.h>

#include "hlxclib/string.h"
#include "shmhelp.h"
#include "hxassert.h"

//Work around AIX problem.
#ifdef _AIX
#  define MAX_SHARED_REGIONS ((UINT32)1)
#else
#  define MAX_SHARED_REGIONS ((UINT32)9999)
#endif

//Keeps track of how many shared memory segments we have used.
int      ShmHelp::zm_nSegment = 0;
Display* ShmHelp::zm_pDisplay = NULL;
BOOL     ShmHelp::zm_bUseShm  = FALSE;

void ShmHelp::Init(Display* pDisplay)
{
   HX_ASSERT( pDisplay );

   zm_pDisplay = pDisplay;
   zm_bUseShm  = FALSE;
   
   //See if our XServer supports ShmMemory....
#if defined(_AIX)  
   int majorVersion, minorVersion;
   Bool sharedPixmaps;
   // doesn't seem to work with multiple presentation panes...
   zm_bUseShm = XShmQueryVersion( zm_pDisplay,
                                  &majorVersion,
                                  &minorVersion, &sharedPixmaps); 
   if (zm_nSegment > AIX_MAX_SHARED_REGIONS) 
      zm_bUseShm = FALSE;
#else
#  ifdef _SOLARIS
   zm_bUseShm = FALSE;
#  else    
   zm_bUseShm = XShmQueryExtension(zm_pDisplay);
#  endif    
#endif    

   //Make sure we only use it if we are displaying locally.  The
   //DISPLAY string can start with ':', '0' or the machine name.
   if(zm_bUseShm)
   {
      char *disp = getenv("DISPLAY");
      HX_ASSERT( disp != NULL && "Your display variable isn't set!" );
      if(!disp)
      {
         zm_bUseShm = FALSE;
      }
      else if( disp[0]!=':' && disp[0]!='0' )
      {
         //See if our machine name is in the dipsplay string.
         struct utsname inf;
         uname(&inf);
         if( (strlen(disp)<=strlen(inf.machine)) ||
             (strncmp(disp, inf.machine, strlen(inf.machine))) )
         {
            zm_bUseShm = FALSE;
         }
      }
   }
}

//This ctor implies someone has called the one that sets the
//display var.
ShmHelp::ShmHelp()
{
   HX_ASSERT( zm_pDisplay );
}

BOOL ShmHelp::ShmAvailable()
{
   return zm_bUseShm;
}

HX_RESULT ShmHelp::CreateSharedRegion(INT32 nSize, UCHAR**ppMem, int* pnShmID, XShmSegmentInfo* pInfo ) 
{
    HX_RESULT theErr   = HXR_OK;
    int       nRetCode = 0;

    HX_ASSERT( ppMem );
    HX_ASSERT( pnShmID );

    //Make sure we were inited...
    HX_ASSERT(zm_pDisplay);
    if( !zm_pDisplay )
       return HXR_UNEXPECTED;

    *ppMem   = NULL;
    *pnShmID = -1;
    
    //get the shared memory segment, etc.
    if( !zm_bUseShm ) 
    {
	theErr = HXR_UNEXPECTED;
    }
    
#ifdef _AIX
    if (zm_nSegment > AIX_MAX_SHARED_REGIONS)
    {
	theErr = HXR_UNEXPECTED;
    }
#endif

    if( theErr == HXR_OK )
    {
        // Request a new shared memory ID
        *pnShmID = ::shmget(0, nSize, IPC_CREAT | 0777);
    
        if( *pnShmID == -1 ) 
        {
#ifdef _DEBUG
            perror("shmget");
#endif            
            theErr = HXR_FAIL;
        }
        else
        {
            // Attach shared memory segment to process's address space
            *ppMem = (UCHAR*)::shmat(*pnShmID, 0, 0);
            if( ((int)*ppMem) == -1) 
            {
                theErr = HXR_FAIL;
#ifdef _DEBUG            
                perror("shmat");
#endif            
            }
            else
            {
                //Request that segment be removed after last reference to it 
                //is released
                pInfo->shmseg   = ++zm_nSegment;
                pInfo->shmid    = *pnShmID;
                pInfo->shmaddr  = (char*)*ppMem;
                pInfo->readOnly = True;
                nRetCode = XShmAttach(zm_pDisplay, pInfo);
                XSync(zm_pDisplay, False);
                ::shmctl(*pnShmID, IPC_RMID, NULL); 
    
                if( !nRetCode )
                {
                    theErr = HXR_FAIL;
                }
                else
                {
                    theErr = HXR_OK;
                }
            }
        }

        if( HXR_OK != theErr )
        {
            // in case of failure, make a best case effort to detach the region
            ::shmctl(*pnShmID, IPC_RMID, NULL);
            zm_bUseShm = FALSE;
        }
    }
    return theErr;
}

HX_RESULT ShmHelp::DetachSharedRegion(UCHAR**ppMem, XShmSegmentInfo* pInfo ) 
{
   HX_ASSERT(zm_bUseShm);
   HX_ASSERT(pInfo);
   HX_ASSERT(*ppMem);
   
   //Make sure we were inited...
   HX_ASSERT(zm_pDisplay);
   if( !zm_pDisplay )
      return HXR_UNEXPECTED;
   
   if(!zm_bUseShm)
      return HXR_UNEXPECTED;
   
   XShmDetach(zm_pDisplay, pInfo);
   
   ::shmdt((char*)*ppMem);
   *ppMem = NULL;
   
   --zm_nSegment;
   
   return HXR_OK;
}