---

communShMem.cpp

/*
 *                         Sun Public License Notice
 *
 * The contents of this file are subject to the Sun Public License
 * Version 1.0 (the "License"); you may not use this file except
 * in compliance with the License. A copy of the License is available
 * at http://www.sun.com/
 *
 * The Original Code is the Java Profiler module.  The Initial Developers
 * of the Original Code are Jan Stola, Pavel Vacha, Michal Pise, Petr Luner,
 * Lukas Petru and Marek Przeczek.
 *
 * Portions created by Jan Stola are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Pavel Vacha are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Michal Pise are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Petr Luner are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Lukas Petru are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Portions created by Marek Przeczek are Copyright (C) 2000-2001.
 * All Rights Reserved.
 *
 * Contributors: Jan Stola, Pavel Vacha, Michal Pise, Petr Luner,
 *               Lukas Petru and Marek Przeczek.
 */

#include "../commun3/communShMem.h"

CommunShMem::CommunShMem( const String& shmemId, int size) :

        _sem1(  String( "1") + shmemId + "XXXXXXXXXX"),
        _sem2(  String( "2") + shmemId + "XXXXXXXXXX"),

        _shmem( String( "3") + shmemId + "XXXXXXXXXX", size) {
}

CommunShMem::~CommunShMem() {

        *(jint*)(_shmem.getAddress()) = -1;

        _sem1.reset();
        _sem1.release();

        _sem2.reset();
        _sem2.release();
}

Commun& CommunShMem::operator>>( Buffer& b) {

        void* p = _shmem.getAddress();
        int shmemSize = _shmem.getSize();

        jint size  = *(jint*)p;

        if( size < 0 || !checkClient()) _failed = 1;
        else if( size) {

                jint num = size/shmemSize;

                char* buf = new char[size];
                char* q = buf;

                for( jint i = 0; i < num; i++, q += shmemSize) {

                        _sem2.release();
                        _sem1.wait();

                        _failed = !checkClient();
                        if( _failed) break;

                        memcpy( q, p, shmemSize);
                }

                _failed = !checkClient();

                jint rest = size%shmemSize;
                if( rest && !_failed) {

                        _sem2.release();
                        _sem1.wait();

                        _failed = !checkClient();
                        if( !_failed) memcpy( q, p, rest);
                }

                if( !_failed) b = Buffer( buf, size);
                delete[] buf;
        }
        else b.clear();

        return *this;
}

Commun& CommunShMem::operator<<( const Buffer& b) {

        void* p = _shmem.getAddress();
        int shmemSize = _shmem.getSize();

        if( *(jint*)p < 0 || !checkClient()) _failed = 1;
        else {

                jint size = b.getSize();
                *(jint*)p = size;

                if( size) {

                        jint num = size/shmemSize;
                        const char* q = b.getBuffer();

                        for( int i = 0; i < num; i++, q += shmemSize) {

                                _sem2.release();
                                _sem1.wait();

                                _failed = !checkClient();
                                if( _failed) break;
                
                                memcpy( p, q, shmemSize);
                        }

                        _failed = !checkClient();

                        jint rest = size%shmemSize;
                        if( rest && !_failed) {

                                _sem2.release();
                                _sem1.wait();

                                _failed = !checkClient();
                                if( !_failed) memcpy( p, q, rest);
                        }
                }
        }

        _sem2.release();

        return *this;
}

int CommunShMem::isReady() {

        _sem1.wait();

        return 1;
}

int CommunShMem::initialize() {

        _sem1.reset();
        _sem1.wait();

        _failed = ( *(jint*)_shmem.getAddress() < 0);

#ifdef WIN32
        DWORD* p = (DWORD*)((jint*)(_shmem.getAddress())+1);

        _pid = *p;
        *p = GetCurrentProcessId();
#else
        pid_t* p = (pid_t*)((jint*)(_shmem.getAddress())+1);

        _pid = *p;
        *p = getpid();
#endif

        _sem2.release();

        return !_failed;
}

int CommunShMem::checkClient() {

#ifdef WIN32
        return GetProcessVersion( _pid);
#else
        return !kill( _pid, 0);
#endif
}

---