0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-10-31 10:58:54 +01:00
construct/include/ircd/buffer/copy.h

92 lines
2.5 KiB
C++

// The Construct
//
// Copyright (C) The Construct Developers, Authors & Contributors
// Copyright (C) 2016-2020 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
#pragma once
#define HAVE_IRCD_BUFFER_COPY_H
namespace ircd::buffer
{
// copy single character
char *&copy(char *&dest, char *const &stop, char src);
size_t copy(const mutable_buffer &dst, const char src);
// copy non-overlapping regions
char *&copy(char *&dest, char *const &stop, const const_buffer &src);
size_t copy(const mutable_buffer &dst, const const_buffer &src);
template<size_t SIZE> size_t copy(const mutable_buffer &dst, const char (&)[SIZE]);
}
template<size_t SIZE>
#ifndef __clang__
__attribute__((error
#else
__attribute__((unavailable
#endif
(
"Copy source is an array. Is this a string literal? Do you want to copy the \\0?"
" Disambiguate this by typing the source string_view or const_buffer."
)))
inline size_t
ircd::buffer::copy(const mutable_buffer &dst,
const char (&buf)[SIZE])
{
return copy(dst, const_buffer{buf});
}
inline size_t
ircd::buffer::copy(const mutable_buffer &dst,
const const_buffer &src)
{
char *const &s(begin(dst)), *e(s);
e = copy(e, end(dst), src);
assert(std::distance(s, e) >= 0);
return std::distance(s, e);
}
inline size_t
ircd::buffer::copy(const mutable_buffer &dst,
const char src)
{
char *const &s(begin(dst)), *e(s);
e = copy(e, end(dst), src);
assert(std::distance(s, e) >= 0);
return std::distance(s, e);
}
inline char *&
__attribute__((always_inline))
ircd::buffer::copy(char *&dest,
char *const &stop,
const const_buffer &src)
{
assert(dest <= stop);
const size_t remain(std::distance(dest, stop));
const size_t cpsz(std::min(size(src), remain));
assert(!overlap(const_buffer(dest, cpsz), src));
assert(cpsz <= size(src));
assert(cpsz <= remain);
__builtin_memcpy(dest, data(src), cpsz);
dest += cpsz;
assert(dest <= stop);
return dest;
}
inline char *&
ircd::buffer::copy(char *&dest,
char *const &stop,
char src)
{
assert(dest <= stop);
const bool cpsz(dest != stop);
(cpsz? *dest : src) = src;
dest += cpsz;
assert(dest <= stop);
return dest;
}