/* * ===================================================================================== * * Filename: shm.cpp * * Description: * * Version: 1.0 * Created: 2021年03月22日 16时19分36秒 * Revision: none * Compiler: gcc * * Author: YOUR NAME (), * Organization: * * ===================================================================================== */ #include "shm.h" #include "bh_util.h" #include #include #include namespace bhome_shm { using namespace boost::interprocess; using namespace boost::uuids; namespace { std::string MsgQIdToName(const MQId& id) { return "shmq" + to_string(id); } MQId EmptyId() { return nil_uuid(); } MQId NewId() { return random_generator()(); } } ShmMsgQueue::ShmMsgQueue(MQId id, ShmType &segment, const uint32_t len): SharedQueue(segment, MsgQIdToName(id), id, len, segment.get_segment_manager()) { printf("queue size: %ld cap: %ld\n", data()->size(), data()->capacity()); } ShmMsgQueue::ShmMsgQueue(ShmType &segment, const uint32_t len):ShmMsgQueue(NewId(), segment, len) {} ShmMsgQueue::~ShmMsgQueue() { Remove(); } bool ShmMsgQueue::Send(MQId remote_id, const void *data, const size_t size, const int timeout_ms) { if (data && size) { Queue *remote = find(MsgQIdToName(remote_id)); if (remote) { void *p = shm().allocate(size, std::nothrow); bool r = false; if (p) { Msg buf = { Id(), p, size}; memcpy(p, data, size); if (remote->Write(buf, timeout_ms)) { return true; } else { shm().deallocate(p); } } } } return false; } bool ShmMsgQueue::Recv(MQId &source_id, void *&data, size_t &size, const int timeout_ms) { Msg buf; if (Read(buf, timeout_ms) && buf.size_ > 0) { DEFER1(shm().deallocate(buf.data_.get());); source_id = buf.src_; size = buf.size_; if (data = malloc(size)) { memcpy(data, buf.data_.get(), size); return true; } } source_id = EmptyId(); data = 0; size = 0; return false; } } // namespace bhome_shm