1 files changed, 85 insertions, 0 deletions
diff --git a/internal/xmpmeta/base64.cc b/internal/xmpmeta/base64.cc
new file mode 100644
index 0000000..a21d8b5
--- /dev/null
+++ b/internal/xmpmeta/base64.cc
@@ -0,0 +1,85 @@
+#include "xmpmeta/base64.h"
+
+#include "android-base/logging.h"
+#include "strings/escaping.h"
+
+namespace photos_editing_formats {
+namespace {
+
+bool EncodeBase64RawData(const uint8* data, size_t data_size, string* output) {
+  // Disable linting because string_view doesn't appear to support uint8_t.
+  dynamic_depth::strings::Base64Escape(data, data_size, output,
+                                       false);  // NOLINT
+  return output->length() > 0;
+}
+
+template <typename T>
+bool InternalEncodeArrayBase64(const std::vector<T>& data, string* output) {
+  size_t buffer_size = data.size() * sizeof(T);
+  return EncodeBase64RawData(reinterpret_cast<const uint8_t*>(data.data()),
+                             buffer_size, output);
+}
+
+template <typename T>
+bool InternalDecodeArrayBase64(const string& data, std::vector<T>* output) {
+  string bytes;
+  if (!DecodeBase64(data, &bytes)) {
+    return false;
+  }
+
+  const int count = bytes.size() / sizeof(T);
+  output->clear();
+  output->resize(count);
+  memcpy(output->data(), bytes.data(), output->size() * sizeof(T));
+  return !output->empty();
+}
+
+}  // namespace
+
+// Decodes the base64-encoded input range.
+bool DecodeBase64(const string& data, string* output) {
+  // Support decoding of both web-safe and regular base64.
+  // "Web-safe" base-64 replaces + with - and / with _, and omits
+  // trailing = padding characters.
+  if (dynamic_depth::absl::Base64Unescape(data, output)) {
+    return true;
+  }
+  return dynamic_depth::absl::WebSafeBase64Unescape(data, output);
+}
+
+// Base64-encodes the given data.
+bool EncodeBase64(const string& data, string* output) {
+  return EncodeBase64RawData(reinterpret_cast<const uint8*>(data.c_str()),
+                             data.length(), output);
+}
+
+// Base64-encodes the given int array.
+bool EncodeIntArrayBase64(const std::vector<int32_t>& data, string* output) {
+  return InternalEncodeArrayBase64<int32_t>(data, output);
+}
+
+// Base64-decodes the given base64-encoded string.
+bool DecodeIntArrayBase64(const string& data, std::vector<int32_t>* output) {
+  return InternalDecodeArrayBase64<int32_t>(data, output);
+}
+
+// Base64-encodes the given float array.
+bool EncodeFloatArrayBase64(const std::vector<float>& data, string* output) {
+  return InternalEncodeArrayBase64<float>(data, output);
+}
+
+// Base64-decodes the given base64-encoded string.
+bool DecodeFloatArrayBase64(const string& data, std::vector<float>* output) {
+  return InternalDecodeArrayBase64<float>(data, output);
+}
+
+// Base64-encodes the given double array.
+bool EncodeDoubleArrayBase64(const std::vector<double>& data, string* output) {
+  return InternalEncodeArrayBase64<double>(data, output);
+}
+
+// Base64-decodes the given base64-encoded string.
+bool DecodeDoubleArrayBase64(const string& data, std::vector<double>* output) {
+  return InternalDecodeArrayBase64<double>(data, output);
+}
+}  // namespace photos_editing_formats