| | |
| | | // ShmMsg is safe to be stored in shared memory, so POD data or offset_ptr is required. |
| | | // message content layout: (meta) / header_size + header + data_size + data |
| | | |
| | | class ShmMsg : private StaticDataRef<SharedMemory, ShmMsg> |
| | | class ShmMsg |
| | | { |
| | | public: |
| | | static inline SharedMemory &shm() { return GetData(); } |
| | | |
| | | private: |
| | | static ShmSocket &Sender(); |
| | | |
| | | // store ref count, msgs shareing the same data should also hold a pointer of the same RefCount object. |
| | | class RefCount : private boost::noncopyable |
| | | { |
| | |
| | | typedef int64_t OffsetType; |
| | | static OffsetType Addr(void *ptr) { return reinterpret_cast<OffsetType>(ptr); } |
| | | static void *Ptr(const OffsetType offset) { return reinterpret_cast<void *>(offset); } |
| | | static inline OffsetType BaseAddr() |
| | | { |
| | | static const OffsetType base = Addr(shm().get_address()); // cache value. |
| | | return base; |
| | | } |
| | | OffsetType BaseAddr() const { return Addr(shm().get_address()); } |
| | | |
| | | static const uint32_t kMsgTag = 0xf1e2d3c4; |
| | | struct Meta { |
| | |
| | | capacity_(size), id_(NewId()), timestamp_(NowSec()) {} |
| | | }; |
| | | OffsetType offset_; |
| | | static void *Alloc(const size_t size) |
| | | SharedMemory *pshm_; |
| | | void *Alloc(const size_t size) |
| | | { |
| | | void *p = shm().Alloc(sizeof(Meta) + size); |
| | | if (p) { |
| | |
| | | if (!addr) { |
| | | return false; |
| | | } |
| | | ShmMsg(addr).swap(*this); |
| | | offset_ = Addr(addr) - BaseAddr(); |
| | | return true; |
| | | } |
| | | ShmMsg(void *p) : |
| | | offset_(p ? (Addr(p) - BaseAddr()) : 0) {} |
| | | |
| | | template <class T = void> |
| | | T *get() const { return offset_ != 0 ? static_cast<T *>(Ptr(offset_ + BaseAddr())) : nullptr; } |
| | | |
| | | public: |
| | | static bool BindShm(SharedMemory &shm) { return SetData(shm); } |
| | | ShmMsg() : |
| | | offset_(0) {} |
| | | explicit ShmMsg(const OffsetType offset) : |
| | | offset_(offset) {} |
| | | explicit ShmMsg(SharedMemory &shm) : |
| | | offset_(0), pshm_(&shm) {} |
| | | ShmMsg(const OffsetType offset, SharedMemory &shm) : |
| | | offset_(offset), pshm_(&shm) {} |
| | | OffsetType Offset() const { return offset_; } |
| | | OffsetType &OffsetRef() { return offset_; } |
| | | void swap(ShmMsg &a) { std::swap(offset_, a.offset_); } |
| | | SharedMemory &shm() const { return *pshm_; } |
| | | |
| | | void swap(ShmMsg &a) |
| | | { |
| | | std::swap(offset_, a.offset_); |
| | | std::swap(pshm_, a.pshm_); |
| | | } |
| | | bool valid() const { return offset_ != 0 && meta()->tag_ == kMsgTag; } |
| | | int64_t id() const { return valid() ? meta()->id_ : 0; } |
| | | int64_t timestamp() const { return valid() ? meta()->timestamp_.load() : 0; } |