email_lib: add MAIL::to_str

grommunio · Aug 30, 2024 · 8358916 · 8358916
1 parent a07be3c
commit 8358916
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/include/gromox/mail.hpp b/include/gromox/mail.hpp
@@ -23,6 +23,7 @@ struct GX_EXPORT MAIL {
 	bool serialize(STREAM *) const;
 	bool emit(MIME::write_func, void *) const;
 	gromox::errno_t to_fd(int) const;
+	gromox::errno_t to_str(std::string &) const;
 	ssize_t get_length() const;
 	MIME *add_head();
 	MIME *get_head();

diff --git a/lib/email/mail.cpp b/lib/email/mail.cpp
@@ -242,6 +242,22 @@ errno_t MAIL::to_fd(int fd) const
 	return 0;
 }
 
+errno_t MAIL::to_str(std::string &out) const try
+{
+	STREAM st;
+	if (!serialize(&st))
+		return ENOMEM;
+	char *data;
+	unsigned int size = STREAM_BLOCK_SIZE;
+	while ((data = static_cast<char *>(st.get_read_buf(&size))) != nullptr) {
+		out.append(data, size);
+		size = STREAM_BLOCK_SIZE;
+	}
+	return 0;
+} catch (const std::bad_alloc &) {
+	return ENOMEM;
+}
+
 /*
  *	calculate the mail object length in bytes
  *	@param