| | |
| | | */ |
| | | |
| | | #include "shm.h" |
| | | #include "bh_util.h" |
| | | #include <mutex> |
| | | #include <boost/uuid/uuid_io.hpp> |
| | | #include <boost/uuid/uuid_generators.hpp> |
| | | |
| | | 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(const MQId &id, ShmType &segment, const uint32_t len): |
| | | SharedQueue(segment, MsgQIdToName(id), id, len, segment.get_segment_manager()) |
| | | namespace bhome_shm |
| | | { |
| | | 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() |
| | | SharedMemory::SharedMemory(const std::string &name, const uint64_t size) : |
| | | mshm_t(open_or_create, name.c_str(), size, 0, AllowAll()), |
| | | name_(name) |
| | | { |
| | | Remove(); |
| | | pmutex_ = FindOrCreate<Mutex>("ShmSelfControlMutex0"); |
| | | } |
| | | |
| | | bool ShmMsgQueue::Send(const MQId &remote_id, const void *data, const size_t size, const int timeout_ms) |
| | | SharedMemory::~SharedMemory() |
| | | { |
| | | if (data && size) { |
| | | Queue *remote = find(MsgQIdToName(remote_id)); |
| | | if (remote) { |
| | | void *p = shm().allocate(sizeof(MsgMetaV1) + size, std::nothrow); |
| | | bool r = false; |
| | | if (p) { |
| | | MsgMetaV1 meta; |
| | | meta.data_size_ = size; |
| | | memcpy(meta.src_id_, &Id(), sizeof(MQId)); |
| | | meta.Pack(p); |
| | | |
| | | memcpy(static_cast<char*>(p) + sizeof(meta), data, size); |
| | | if (remote->Write(p, 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 msg; |
| | | if (Read(msg, timeout_ms) && msg) { |
| | | DEFER1(shm().deallocate(msg.get());); |
| | | MsgMetaV1 meta; |
| | | meta.Parse(msg.get()); |
| | | memcpy(&source_id, meta.src_id_, sizeof(MQId)); |
| | | size = meta.data_size_; |
| | | if (data = malloc(size)) { |
| | | memcpy(data, static_cast<char*>(msg.get()) + meta.self_size_, size); |
| | | return true; |
| | | } |
| | | } |
| | | source_id = EmptyId(); |
| | | data = 0; |
| | | size = 0; |
| | | return false; |
| | | } |
| | | |
| | | } // namespace bhome_shm |