| | |
| | | * Revision: none |
| | | * Compiler: gcc |
| | | * |
| | | * Author: YOUR NAME (), |
| | | * Author: Li Chao (), |
| | | * Organization: |
| | | * |
| | | * ===================================================================================== |
| | | */ |
| | | |
| | | #include "shm.h" |
| | | #include "bh_util.h" |
| | | #include <mutex> |
| | | |
| | | namespace bhome_shm { |
| | | using namespace boost::interprocess; |
| | | |
| | | namespace { |
| | | std::string MsgQIdToName(const int id) { return "shmq" + std::to_string(id); } |
| | | } |
| | | |
| | | ShmMsgQueue::ShmMsgQueue(MQId id, ShmType &segment, const std::string &name, const uint32_t len): |
| | | SharedQueue(segment, name, id, len, segment.get_segment_manager()) |
| | | { |
| | | printf("queue size: %ld cap: %ld\n", data()->size(), data()->capacity()); |
| | | } |
| | | |
| | | ShmMsgQueue::ShmMsgQueue(MQId id, ShmType &segment, const uint32_t len): |
| | | ShmMsgQueue(id, segment, MsgQIdToName(id), len) |
| | | {} |
| | | |
| | | ShmMsgQueue::~ShmMsgQueue() |
| | | namespace bhome_shm |
| | | { |
| | | |
| | | } |
| | | |
| | | bool ShmMsgQueue::Send(MQId remote_id, const void *data, const size_t size, const int timeout_ms) |
| | | SharedMemory::SharedMemory(const std::string &name, const uint64_t size) : |
| | | mshm_t(open_or_create, name.c_str(), size, 0, AllowAll()), |
| | | name_(name) |
| | | { |
| | | 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; |
| | | pmutex_ = FindOrCreate<Mutex>("ShmSelfControlMutex0"); |
| | | } |
| | | |
| | | bool ShmMsgQueue::Recv(MQId &source_id, void *&data, size_t &size, const int timeout_ms) |
| | | SharedMemory::~SharedMemory() |
| | | { |
| | | 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 = 0; |
| | | data = 0; |
| | | size = 0; |
| | | return false; |
| | | } |
| | | |
| | | } // namespace bhome_shm |