Add new private API InputSource read methods

Improve support for reading into std::string objects.

Add new private API InputSource read methods
Improve support for reading into std::string objects.
m-holger
1 parent 61804943
Showing 2 changed files with 21 additions and 0 deletions
include/qpdf/InputSource.hh
libqpdf/qpdf/InputSource_private.hh
@@ -76,6 +76,8 @@ class QPDF_DLL_CLASS InputSource
     virtual void unreadCh(char ch) = 0;
     // The following methods are for internal use by qpdf only.
+    inline size_t read(std::string& str, size_t count, qpdf_offset_t at = -1);
+    inline std::string read(size_t count, qpdf_offset_t at = -1);
     inline qpdf_offset_t fastTell();
     inline bool fastRead(char&);
     inline void fastUnread(bool);
@@ -3,6 +3,25 @@
 #include <qpdf/InputSource.hh>
+inline size_t
+InputSource::read(std::string& str, size_t count, qpdf_offset_t at)
+{
+    if (at >= 0) {
+        seek(at, SEEK_SET);
+    }
+    str.resize(count);
+    str.resize(read(str.data(), count));
+    return str.size();
+}
+
+inline std::string
+InputSource::read(size_t count, qpdf_offset_t at)
+{
+    std::string result(count, '\0');
+    (void)read(result, count, at);
+    return result;
+}
+
 inline void
 InputSource::loadBuffer()
 {