| Index: pdf/pdfium/pdfium_api_string_buffer_adapter.h
|
| diff --git a/pdf/pdfium/pdfium_api_string_buffer_adapter.h b/pdf/pdfium/pdfium_api_string_buffer_adapter.h
|
| index a36879f112bc745dc264b997192fdd8cf9c54e88..449588b6dfc284828afef1eacc475604693a2a00 100644
|
| --- a/pdf/pdfium/pdfium_api_string_buffer_adapter.h
|
| +++ b/pdf/pdfium/pdfium_api_string_buffer_adapter.h
|
| @@ -48,6 +48,41 @@ class PDFiumAPIStringBufferAdapter {
|
| DISALLOW_COPY_AND_ASSIGN(PDFiumAPIStringBufferAdapter);
|
| };
|
|
|
| +// Helper to deal with the fact that many PDFium APIs write the null-terminator
|
| +// into string buffers that are passed to them, but the PDF plugin likes to pass
|
| +// in std::strings / base::string16s, where one should not count on the internal
|
| +// string buffers to be null-terminated. This version is suitable for APIs that
|
| +// work in terms of number of bytes instead of the number of characters.
|
| +template <class StringType>
|
| +class PDFiumAPIStringBufferSizeInBytesAdapter {
|
| + public:
|
| + // |str| is the string to write into.
|
| + // |expected_size| is the number of bytes the PDFium API will write,
|
| + // including the null-terminator. It should be at least the size of a
|
| + // character in bytes.
|
| + // |check_expected_size| whether to check the actual number of bytes
|
| + // written into |str| against |expected_size| when calling Close().
|
| + PDFiumAPIStringBufferSizeInBytesAdapter(StringType* str,
|
| + size_t expected_size,
|
| + bool check_expected_size);
|
| + ~PDFiumAPIStringBufferSizeInBytesAdapter();
|
| +
|
| + // Returns a pointer to |str_|'s buffer. The buffer's size is large enough to
|
| + // hold |expected_size_| + sizeof(StringType::value_type) bytes, so the PDFium
|
| + // API that uses the pointer has space to write a null-terminator.
|
| + void* GetData();
|
| +
|
| + // Resizes |str_| to |actual_size| - sizeof(StringType::value_type) bytes,
|
| + // thereby removing the extra null-terminator. This must be called prior to
|
| + // the adapter's destruction. The pointer returned by GetData() should be
|
| + // considered invalid.
|
| + void Close(int actual_size);
|
| + void Close(size_t actual_size);
|
| +
|
| + private:
|
| + PDFiumAPIStringBufferAdapter<StringType> adapter_;
|
| +};
|
| +
|
| } // namespace chrome_pdf
|
|
|
| #endif // PDF_PDFIUM_PDFIUM_API_STRING_BUFFER_ADAPTER_H_
|
|
|