trunk/bwapi/Util/Source/Util/SharedStructure.h

Go to the documentation of this file.

#pragma once

//  A memory buffer which can be shared with other processes          //
//                                                                    //

namespace Util
{
  template<typename T> class SharedStructure;
}

#include "MemoryFrame.h"
#include "RemoteProcess.h"
#include "SharedMemory.h"
#define WIN32_LEAN_AND_MEAN
#include "windows.h"

namespace Util
{
  template<typename T> class SharedStructure
  {
  public:
    //----------------------- TYPES -----------------------------------------------------
    template<typename S> struct Pointer
    {
      friend class SharedStructure;
    private:
      SharedMemory::Pointer<S> pointer;
    };
    struct Export
    {
      friend class SharedStructure;

      Export(){};
      bool isValid() const
      {
        return smemExport.isValid();
      }
    private:
      SharedMemory::Export smemExport;
    };
    //----------------------- CONSTRUCTION ----------------------------------------------
    SharedStructure()
      : smem()
    {
    }
    ~SharedStructure()
    {
    }
    //----------------------- CREATE ----------------------------------------------------
    void create()
    {
      smem.create(sizeof(T));
    }
    bool create(const char* systemName)
    {
      return smem.create(sizeof(T), systemName);
    }
    //----------------------- IMPORT ----------------------------------------------------
    void import(Export source)
    {
      smem.import(source.smemExport);
    }
    //----------------------- RELEASE ---------------------------------------------------
    void release()
    {
      smem.release();
    }
    //----------------------- EXPORT TO PROCESS -----------------------------------------
    Export exportToProcess(RemoteProcess &target, bool readOnly) const
    {
      Export exp;
      exp.smemExport = smem.exportToProcess(target, readOnly);
      return exp;
    }
    //----------------------- UNPACK ----------------------------------------------------
    template<typename S>
      S *unpack(Pointer<S> p) const // to process address space
      {
        return smem.unpack(p.pointer);
      }
    //----------------------- PACK ------------------------------------------------------
    template<typename S>
      Pointer<S> pack(S *pt) const  // to shared address space
      {
        Pointer<S> retval;
        retval.pointer = smem.pack(pt);
        return retval;
      }
    //----------------------- GET MEMORY ------------------------------------------------
    T &get() const
    {
      return smem.getMemory().getAs<T>();
    }
    //----------------------- -----------------------------------------------------------
  private:
    SharedStructure(SharedStructure&);                // no copying

    SharedMemory smem;
  };
}