Add a WriteString method to IFX_WriteStream

This Cl allows passing a CFX_ByteStringC to IFX_WriteStream along with the
buffer method. This makes it easier to pass C-style strings to the stream.

Change-Id: I1051eb3ba17c7fbd42984c14dc60cbce24d72f3f
Reviewed-on: https://pdfium-review.googlesource.com/5430
Reviewed-by: Nicolás Peña <npm@chromium.org>
Commit-Queue: dsinclair <dsinclair@chromium.org>

2 files changed, 7 insertions, 1 deletions

diff --git a/core/fxcrt/fx_stream.cpp b/core/fxcrt/fx_stream.cpp
index 32b6ca3495..ba21461098 100644
--- a/core/fxcrt/fx_stream.cpp
+++ b/core/fxcrt/fx_stream.cpp
@@ -93,3 +93,7 @@ size_t IFX_SeekableReadStream::ReadBlock(void* buffer, size_t size) {
 bool IFX_SeekableStream::WriteBlock(const void* buffer, size_t size) {
   return WriteBlock(buffer, GetSize(), size);
 }
+
+bool IFX_SeekableStream::WriteString(const CFX_ByteStringC& str) {
+  return WriteBlock(str.c_str(), str.GetLength());
+}
diff --git a/core/fxcrt/fx_stream.h b/core/fxcrt/fx_stream.h
index 5237a7cb94..622818173b 100644
--- a/core/fxcrt/fx_stream.h
+++ b/core/fxcrt/fx_stream.h
@@ -42,13 +42,13 @@ wchar_t FX_GetFolderSeparator();
 class IFX_WriteStream : virtual public CFX_Retainable {
  public:
   virtual bool WriteBlock(const void* pData, size_t size) = 0;
+  virtual bool WriteString(const CFX_ByteStringC& str) = 0;
 };
 
 class IFX_ArchiveStream : public IFX_WriteStream {
  public:
   virtual bool WriteByte(uint8_t byte) = 0;
   virtual bool WriteDWord(uint32_t i) = 0;
-  virtual bool WriteString(const CFX_ByteStringC& str) = 0;
   virtual FX_FILESIZE CurrentOffset() const = 0;
 };
 
@@ -108,6 +108,8 @@ class IFX_SeekableStream : public IFX_SeekableReadStream,
                   FX_FILESIZE offset,
                   size_t size) override = 0;
   bool WriteBlock(const void* buffer, size_t size) override;
+  bool WriteString(const CFX_ByteStringC& str) override;
+
   bool Flush() override = 0;
 };