From 026bbfaf2b5d73a26b8e2fa49158883ef64c211b Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期四, 27 五月 2021 13:51:26 +0800 Subject: [PATCH] tcp server call center to send proxy requests. --- src/msg.h | 47 ++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 40 insertions(+), 7 deletions(-) diff --git a/src/msg.h b/src/msg.h index 12922b5..f0ed840 100644 --- a/src/msg.h +++ b/src/msg.h @@ -75,12 +75,16 @@ }; OffsetType offset_; SharedMemory *pshm_; - void *Alloc(const size_t size) + + void *Alloc(const size_t size, const void *src = nullptr) { void *p = shm().Alloc(sizeof(Meta) + size); if (p) { auto pmeta = new (p) Meta(size); p = pmeta + 1; + if (src) { + memcpy(p, src, size); + } } return p; } @@ -108,16 +112,35 @@ } return addr; } - - void *Pack(const std::string &content) + void *Pack(const BHMsgHead &head, const uint32_t head_len, const std::string &body_content) { void *addr = get(); if (addr) { - memcpy(addr, content.data(), content.size()); - meta()->size_ = content.size(); + auto p = static_cast<char *>(addr); + auto Pack1 = [&p](uint32_t len, auto &&writer) { + Put32(p, len); + p += sizeof(len); + writer(p, len); + p += len; + }; + Pack1(head_len, [&](void *p, int len) { head.SerializeToArray(p, len); }); + Pack1(body_content.size(), [&](void *p, int len) { memcpy(p, body_content.data(), len); }); + meta()->size_ = 4 + head_len + 4 + body_content.size(); } return addr; } + + void *Pack(const void *src, const size_t size) + { + void *addr = get(); + if (addr && src) { + memcpy(addr, src, size); + meta()->size_ = size; + } + return addr; + } + + void *Pack(const std::string &content) { return Pack(content.data(), content.size()); } bool Make(void *addr) { @@ -166,6 +189,13 @@ uint32_t size = sizeof(head_len) + head_len + sizeof(body_len) + body_len; return Make(size) && Pack(head, head_len, body, body_len); } + inline bool Make(const BHMsgHead &head, const std::string &body_content) + { + uint32_t head_len = head.ByteSizeLong(); + uint32_t size = sizeof(head_len) + head_len + sizeof(uint32_t) + body_content.size(); + return Make(size) && Pack(head, head_len, body_content); + } + template <class Body> inline bool Fill(const BHMsgHead &head, const Body &body) { @@ -175,8 +205,11 @@ return valid() && (meta()->capacity_ >= size) && Pack(head, head_len, body, body_len); } - inline bool Make(const std::string &content) { return Make(content.size()) && Pack(content); } - inline bool Fill(const std::string &content) { return valid() && (meta()->capacity_ >= content.size()) && Pack(content); } + inline bool Make(const void *src, const size_t size) { return Make(Alloc(size, src)); } + inline bool Fill(const void *src, const size_t size) { return valid() && (meta()->capacity_ >= size) && Pack(src, size); } + + inline bool Make(const std::string &content) { return Make(content.data(), content.size()); } + inline bool Fill(const std::string &content) { return Fill(content.data(), content.size()); } inline bool Make(const size_t size) { return Make(Alloc(size)); } -- Gitblit v1.8.0