xref: /aosp_15_r20/external/pdfium/testing/fx_string_testhelpers.cpp (revision 3ac0a46f773bac49fa9476ec2b1cf3f8da5ec3a4)
1 // Copyright 2014 The PDFium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #include "testing/fx_string_testhelpers.h"
6 
7 #include <iomanip>
8 #include <ios>
9 #include <ostream>
10 
11 #include "core/fxcrt/cfx_datetime.h"
12 #include "core/fxcrt/fx_string.h"
13 #include "third_party/base/check_op.h"
14 #include "third_party/base/containers/span.h"
15 
operator <<(std::ostream & os,const CFX_DateTime & dt)16 std::ostream& operator<<(std::ostream& os, const CFX_DateTime& dt) {
17   os << dt.GetYear() << "-" << std::to_string(dt.GetMonth()) << "-"
18      << std::to_string(dt.GetDay()) << " " << std::to_string(dt.GetHour())
19      << ":" << std::to_string(dt.GetMinute()) << ":"
20      << std::to_string(dt.GetSecond()) << "."
21      << std::to_string(dt.GetMillisecond());
22   return os;
23 }
24 
StringSplit(const std::string & str,char delimiter)25 std::vector<std::string> StringSplit(const std::string& str, char delimiter) {
26   std::vector<std::string> result;
27   size_t pos = 0;
28   while (true) {
29     size_t found = str.find(delimiter, pos);
30     if (found == std::string::npos)
31       break;
32 
33     result.push_back(str.substr(pos, found - pos));
34     pos = found + 1;
35   }
36   result.push_back(str.substr(pos));
37   return result;
38 }
39 
GetPlatformString(FPDF_WIDESTRING wstr)40 std::string GetPlatformString(FPDF_WIDESTRING wstr) {
41   WideString wide_string =
42       WideString::FromUTF16LE(wstr, WideString::WStringLength(wstr));
43   return std::string(wide_string.ToUTF8().c_str());
44 }
45 
GetPlatformWString(FPDF_WIDESTRING wstr)46 std::wstring GetPlatformWString(FPDF_WIDESTRING wstr) {
47   if (!wstr)
48     return std::wstring();
49 
50   size_t characters = 0;
51   while (wstr[characters])
52     ++characters;
53 
54   std::wstring platform_string;
55   platform_string.reserve(characters);
56   for (size_t i = 0; i < characters; ++i) {
57     const unsigned char* ptr = reinterpret_cast<const unsigned char*>(&wstr[i]);
58     platform_string.push_back(ptr[0] + 256 * ptr[1]);
59   }
60   return platform_string;
61 }
62 
GetFPDFWideString(const std::wstring & wstr)63 ScopedFPDFWideString GetFPDFWideString(const std::wstring& wstr) {
64   size_t length = sizeof(uint16_t) * (wstr.size() + 1);
65   ScopedFPDFWideString result(static_cast<FPDF_WCHAR*>(malloc(length)));
66   pdfium::span<uint8_t> result_span(reinterpret_cast<uint8_t*>(result.get()),
67                                     length);
68   size_t i = 0;
69   for (wchar_t w : wstr) {
70     result_span[i++] = w & 0xff;
71     result_span[i++] = (w >> 8) & 0xff;
72   }
73   result_span[i++] = 0;
74   result_span[i] = 0;
75   return result;
76 }
77 
GetFPDFWideStringBuffer(size_t length_bytes)78 std::vector<FPDF_WCHAR> GetFPDFWideStringBuffer(size_t length_bytes) {
79   DCHECK_EQ(length_bytes % sizeof(FPDF_WCHAR), 0u);
80   return std::vector<FPDF_WCHAR>(length_bytes / sizeof(FPDF_WCHAR));
81 }
82