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