#include "absl/strings/str_cat.h"
#include <assert.h>
#include <cstddef>
#include <cstdint>
#include <cstring>
#include <initializer_list>
#include <limits>
#include <string>
#include "absl/base/config.h"
#include "absl/base/internal/raw_logging.h"
#include "absl/base/nullability.h"
#include "absl/strings/internal/append_and_overwrite.h"
#include "absl/strings/resize_and_overwrite.h"
#include "absl/strings/string_view.h"
namespace absl {
ABSL_NAMESPACE_BEGIN
namespace {
inline char* absl_nonnull Append(char* absl_nonnull out, const AlphaNum& x) {
char* after = out + x.size();
if (x.size() != 0) {
memcpy(out, x.data(), x.size());
}
return after;
}
}
std::string StrCat(const AlphaNum& a, const AlphaNum& b) {
std::string result;
constexpr uint64_t kMaxSize = uint64_t{std::numeric_limits<size_t>::max()};
const uint64_t result_size =
static_cast<uint64_t>(a.size()) + static_cast<uint64_t>(b.size());
ABSL_INTERNAL_CHECK(result_size <= kMaxSize, "size_t overflow");
absl::StringResizeAndOverwrite(result, static_cast<size_t>(result_size),
[&a, &b](char* const begin, size_t buf_size) {
char* out = begin;
out = Append(out, a);
out = Append(out, b);
assert(out == begin + buf_size);
return buf_size;
});
return result;
}
std::string StrCat(const AlphaNum& a, const AlphaNum& b, const AlphaNum& c) {
std::string result;
constexpr uint64_t kMaxSize = uint64_t{std::numeric_limits<size_t>::max()};
const uint64_t result_size = static_cast<uint64_t>(a.size()) +
static_cast<uint64_t>(b.size()) +
static_cast<uint64_t>(c.size());
ABSL_INTERNAL_CHECK(result_size <= kMaxSize, "size_t overflow");
absl::StringResizeAndOverwrite(
result, static_cast<size_t>(result_size),
[&a, &b, &c](char* const begin, size_t buf_size) {
char* out = begin;
out = Append(out, a);
out = Append(out, b);
out = Append(out, c);
assert(out == begin + buf_size);
return buf_size;
});
return result;
}
std::string StrCat(const AlphaNum& a, const AlphaNum& b, const AlphaNum& c,
const AlphaNum& d) {
std::string result;
constexpr uint64_t kMaxSize = uint64_t{std::numeric_limits<size_t>::max()};
const uint64_t result_size =
static_cast<uint64_t>(a.size()) + static_cast<uint64_t>(b.size()) +
static_cast<uint64_t>(c.size()) + static_cast<uint64_t>(d.size());
ABSL_INTERNAL_CHECK(result_size <= kMaxSize, "size_t overflow");
absl::StringResizeAndOverwrite(
result, static_cast<size_t>(result_size),
[&a, &b, &c, &d](char* const begin, size_t buf_size) {
char* out = begin;
out = Append(out, a);
out = Append(out, b);
out = Append(out, c);
out = Append(out, d);
assert(out == begin + buf_size);
return buf_size;
});
return result;
}
namespace strings_internal {
std::string CatPieces(std::initializer_list<absl::string_view> pieces) {
std::string result;
constexpr uint64_t kMaxSize = uint64_t{std::numeric_limits<size_t>::max()};
uint64_t total_size = 0;
for (absl::string_view piece : pieces) {
total_size += piece.size();
}
ABSL_INTERNAL_CHECK(total_size <= kMaxSize, "size_t overflow");
absl::StringResizeAndOverwrite(result, static_cast<size_t>(total_size),
[&pieces](char* const begin, size_t buf_size) {
char* out = begin;
for (absl::string_view piece : pieces) {
const size_t this_size = piece.size();
if (this_size != 0) {
memcpy(out, piece.data(), this_size);
out += this_size;
}
}
assert(out == begin + buf_size);
return buf_size;
});
return result;
}
#define ASSERT_NO_OVERLAP(dest, src) \
assert(((src).size() == 0) || \
(uintptr_t((src).data() - (dest).data()) > uintptr_t((dest).size())))
void AppendPieces(std::string* absl_nonnull dest,
std::initializer_list<absl::string_view> pieces) {
size_t to_append = 0;
for (absl::string_view piece : pieces) {
ASSERT_NO_OVERLAP(*dest, piece);
to_append += piece.size();
}
StringAppendAndOverwrite(*dest, to_append,
[&pieces](char* const buf, size_t buf_size) {
char* out = buf;
for (absl::string_view piece : pieces) {
const size_t this_size = piece.size();
if (this_size != 0) {
memcpy(out, piece.data(), this_size);
out += this_size;
}
}
assert(out == buf + buf_size);
return buf_size;
});
}
}
void StrAppend(std::string* absl_nonnull dest, const AlphaNum& a) {
ASSERT_NO_OVERLAP(*dest, a);
strings_internal::StringAppendAndOverwrite(
*dest, a.size(), [&a](char* const buf, size_t buf_size) {
char* out = buf;
out = Append(out, a);
assert(out == buf + buf_size);
return buf_size;
});
}
void StrAppend(std::string* absl_nonnull dest, const AlphaNum& a,
const AlphaNum& b) {
ASSERT_NO_OVERLAP(*dest, a);
ASSERT_NO_OVERLAP(*dest, b);
strings_internal::StringAppendAndOverwrite(
*dest, a.size() + b.size(), [&a, &b](char* const buf, size_t buf_size) {
char* out = buf;
out = Append(out, a);
out = Append(out, b);
assert(out == buf + buf_size);
return buf_size;
});
}
void StrAppend(std::string* absl_nonnull dest, const AlphaNum& a,
const AlphaNum& b, const AlphaNum& c) {
ASSERT_NO_OVERLAP(*dest, a);
ASSERT_NO_OVERLAP(*dest, b);
ASSERT_NO_OVERLAP(*dest, c);
strings_internal::StringAppendAndOverwrite(
*dest, a.size() + b.size() + c.size(),
[&a, &b, &c](char* const buf, size_t buf_size) {
char* out = buf;
out = Append(out, a);
out = Append(out, b);
out = Append(out, c);
assert(out == buf + buf_size);
return buf_size;
});
}
void StrAppend(std::string* absl_nonnull dest, const AlphaNum& a,
const AlphaNum& b, const AlphaNum& c, const AlphaNum& d) {
ASSERT_NO_OVERLAP(*dest, a);
ASSERT_NO_OVERLAP(*dest, b);
ASSERT_NO_OVERLAP(*dest, c);
ASSERT_NO_OVERLAP(*dest, d);
strings_internal::StringAppendAndOverwrite(
*dest, a.size() + b.size() + c.size() + d.size(),
[&a, &b, &c, &d](char* const buf, size_t buf_size) {
char* out = buf;
out = Append(out, a);
out = Append(out, b);
out = Append(out, c);
out = Append(out, d);
assert(out == buf + buf_size);
return buf_size;
});
}
ABSL_NAMESPACE_END
}