0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-09-29 04:08:54 +02:00

ircd: Split base64 from lexical.

This commit is contained in:
Jason Volk 2018-01-18 18:17:02 -08:00
parent 054c50edcc
commit 0a6ea976a5
4 changed files with 208 additions and 183 deletions

27
include/ircd/base64.h Normal file
View file

@ -0,0 +1,27 @@
// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 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_BASE64_H
namespace ircd
{
// Binary -> Base64 conversion suite
string_view b64encode(const mutable_buffer &out, const const_raw_buffer &in);
std::string b64encode(const const_raw_buffer &in);
// Binary -> Base64 conversion without padding
string_view b64encode_unpadded(const mutable_buffer &out, const const_raw_buffer &in);
std::string b64encode_unpadded(const const_raw_buffer &in);
// Base64 -> Binary conversion (padded or unpadded)
const_raw_buffer b64decode(const mutable_raw_buffer &out, const string_view &in);
std::string b64decode(const string_view &in);
}

View file

@ -47,18 +47,6 @@ namespace ircd
// Binary <-> Hex conversion suite // Binary <-> Hex conversion suite
string_view u2a(const mutable_buffer &out, const const_raw_buffer &in); string_view u2a(const mutable_buffer &out, const const_raw_buffer &in);
const_raw_buffer a2u(const mutable_raw_buffer &out, const const_buffer &in); const_raw_buffer a2u(const mutable_raw_buffer &out, const const_buffer &in);
// Binary -> Base64 conversion suite
string_view b64encode(const mutable_buffer &out, const const_raw_buffer &in);
std::string b64encode(const const_raw_buffer &in);
// Binary -> Base64 conversion without padding
string_view b64encode_unpadded(const mutable_buffer &out, const const_raw_buffer &in);
std::string b64encode_unpadded(const const_raw_buffer &in);
// Base64 -> Binary conversion (padded or unpadded)
const_raw_buffer b64decode(const mutable_raw_buffer &out, const string_view &in);
std::string b64decode(const string_view &in);
} }
namespace ircd namespace ircd

181
ircd/base64.cc Normal file
View file

@ -0,0 +1,181 @@
// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 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.
#include <boost/archive/iterators/base64_from_binary.hpp>
#include <boost/archive/iterators/binary_from_base64.hpp>
#include <boost/archive/iterators/transform_width.hpp>
namespace ircd
{
const char _b64_pad_
{
'='
};
using _b64_encoder = std::function<string_view (const mutable_buffer &, const const_raw_buffer &)>;
static std::string _b64encode(const const_raw_buffer &in, const _b64_encoder &);
}
/// Allocate and return a string without padding from the encoding of in
std::string
ircd::b64encode_unpadded(const const_raw_buffer &in)
{
return _b64encode(in, [](const auto &out, const auto &in)
{
return b64encode_unpadded(out, in);
});
}
/// Allocate and return a string from the encoding of in
std::string
ircd::b64encode(const const_raw_buffer &in)
{
return _b64encode(in, [](const auto &out, const auto &in)
{
return b64encode(out, in);
});
}
/// Internal; dedupes encoding functions that create and return a string
static std::string
ircd::_b64encode(const const_raw_buffer &in,
const _b64_encoder &encoder)
{
// Allocate a buffer 1.33 times larger than input with pessimistic
// extra space for any padding and nulling.
const auto max
{
ceil(size(in) * (4.0 / 3.0)) + 4
};
std::string ret(max, char{});
const mutable_buffer buf
{
const_cast<char *>(ret.data()), ret.size()
};
const string_view encoded
{
encoder(buf, in)
};
assert(size(encoded) <= ret.size());
ret.resize(size(encoded));
return ret;
}
/// Encoding in to base64 at out. Out must be 1.33+ larger than in
/// padding is not present in the returned view.
ircd::string_view
ircd::b64encode(const mutable_buffer &out,
const const_raw_buffer &in)
{
const auto pads
{
(3 - size(in) % 3) % 3
};
const auto encoded
{
b64encode_unpadded(out, in)
};
assert(size(encoded) + pads <= size(out));
memset(data(out) + size(encoded), _b64_pad_, pads);
const auto len
{
size(encoded) + pads
};
return { data(out), len };
}
/// Encoding in to base64 at out. Out must be 1.33+ larger than in.
ircd::string_view
ircd::b64encode_unpadded(const mutable_buffer &out,
const const_raw_buffer &in)
{
namespace iterators = boost::archive::iterators;
using transform = iterators::transform_width<unsigned char *, 6, 8>;
using b64fb = iterators::base64_from_binary<transform>;
const auto cpsz
{
std::min(size(in), size_t(size(out) * (3.0 / 4.0)))
};
const auto end
{
std::copy(b64fb(data(in)), b64fb(data(in) + cpsz), begin(out))
};
const auto len
{
size_t(std::distance(begin(out), end))
};
assert(len <= size(out));
return { data(out), len };
}
std::string
ircd::b64decode(const string_view &in)
{
// Allocate a buffer 75% than input with pessimistic extra space
const auto max
{
ceil(size(in) * 0.75) + 4
};
std::string ret(max, char{});
const mutable_raw_buffer buf
{
reinterpret_cast<unsigned char *>(const_cast<char *>(ret.data())), ret.size()
};
const auto decoded
{
b64decode(buf, in)
};
assert(size(decoded) <= ret.size());
ret.resize(size(decoded));
return ret;
}
/// Decode base64 from in to the buffer at out; out can be 75% of the size
/// of in.
ircd::const_raw_buffer
ircd::b64decode(const mutable_raw_buffer &out,
const string_view &in)
{
namespace iterators = boost::archive::iterators;
using b64bf = iterators::binary_from_base64<const char *>;
using transform = iterators::transform_width<b64bf, 8, 6>;
const auto pads
{
endswith_count(in, _b64_pad_)
};
const auto e
{
std::copy(transform(begin(in)), transform(begin(in) + size(in) - pads), begin(out))
};
const auto len
{
std::distance(begin(out), e)
};
assert(size_t(len) <= size(out));
return { data(out), size_t(len) };
}

View file

@ -30,9 +30,6 @@
#include <RB_INC_BOOST_TOKENIZER_HPP #include <RB_INC_BOOST_TOKENIZER_HPP
#include <RB_INC_BOOST_LEXICAL_CAST_HPP #include <RB_INC_BOOST_LEXICAL_CAST_HPP
#include <boost/archive/iterators/base64_from_binary.hpp>
#include <boost/archive/iterators/binary_from_base64.hpp>
#include <boost/archive/iterators/transform_width.hpp>
/////////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////////
// //
@ -665,174 +662,6 @@ ircd::replace(const string_view &s,
return ret; return ret;
} }
namespace ircd
{
const char _b64_pad_
{
'='
};
using _b64_encoder = std::function<string_view (const mutable_buffer &, const const_raw_buffer &)>;
static std::string _b64encode(const const_raw_buffer &in, const _b64_encoder &);
}
/// Allocate and return a string without padding from the encoding of in
std::string
ircd::b64encode_unpadded(const const_raw_buffer &in)
{
return _b64encode(in, [](const auto &out, const auto &in)
{
return b64encode_unpadded(out, in);
});
}
/// Allocate and return a string from the encoding of in
std::string
ircd::b64encode(const const_raw_buffer &in)
{
return _b64encode(in, [](const auto &out, const auto &in)
{
return b64encode(out, in);
});
}
/// Internal; dedupes encoding functions that create and return a string
static std::string
ircd::_b64encode(const const_raw_buffer &in,
const _b64_encoder &encoder)
{
// Allocate a buffer 1.33 times larger than input with pessimistic
// extra space for any padding and nulling.
const auto max
{
ceil(size(in) * (4.0 / 3.0)) + 4
};
std::string ret(max, char{});
const mutable_buffer buf
{
const_cast<char *>(ret.data()), ret.size()
};
const string_view encoded
{
encoder(buf, in)
};
assert(size(encoded) <= ret.size());
ret.resize(size(encoded));
return ret;
}
/// Encoding in to base64 at out. Out must be 1.33+ larger than in
/// padding is not present in the returned view.
ircd::string_view
ircd::b64encode(const mutable_buffer &out,
const const_raw_buffer &in)
{
const auto pads
{
(3 - size(in) % 3) % 3
};
const auto encoded
{
b64encode_unpadded(out, in)
};
assert(size(encoded) + pads <= size(out));
memset(data(out) + size(encoded), _b64_pad_, pads);
const auto len
{
size(encoded) + pads
};
return { data(out), len };
}
/// Encoding in to base64 at out. Out must be 1.33+ larger than in.
ircd::string_view
ircd::b64encode_unpadded(const mutable_buffer &out,
const const_raw_buffer &in)
{
namespace iterators = boost::archive::iterators;
using transform = iterators::transform_width<unsigned char *, 6, 8>;
using b64fb = iterators::base64_from_binary<transform>;
const auto cpsz
{
std::min(size(in), size_t(size(out) * (3.0 / 4.0)))
};
const auto end
{
std::copy(b64fb(data(in)), b64fb(data(in) + cpsz), begin(out))
};
const auto len
{
size_t(std::distance(begin(out), end))
};
assert(len <= size(out));
return { data(out), len };
}
std::string
ircd::b64decode(const string_view &in)
{
// Allocate a buffer 75% than input with pessimistic extra space
const auto max
{
ceil(size(in) * 0.75) + 4
};
std::string ret(max, char{});
const mutable_raw_buffer buf
{
reinterpret_cast<unsigned char *>(const_cast<char *>(ret.data())), ret.size()
};
const auto decoded
{
b64decode(buf, in)
};
assert(size(decoded) <= ret.size());
ret.resize(size(decoded));
return ret;
}
/// Decode base64 from in to the buffer at out; out can be 75% of the size
/// of in.
ircd::const_raw_buffer
ircd::b64decode(const mutable_raw_buffer &out,
const string_view &in)
{
namespace iterators = boost::archive::iterators;
using b64bf = iterators::binary_from_base64<const char *>;
using transform = iterators::transform_width<b64bf, 8, 6>;
const auto pads
{
endswith_count(in, _b64_pad_)
};
const auto e
{
std::copy(transform(begin(in)), transform(begin(in) + size(in) - pads), begin(out))
};
const auto len
{
std::distance(begin(out), e)
};
assert(size_t(len) <= size(out));
return { data(out), size_t(len) };
}
ircd::const_raw_buffer ircd::const_raw_buffer
ircd::a2u(const mutable_raw_buffer &out, ircd::a2u(const mutable_raw_buffer &out,
const const_buffer &in) const const_buffer &in)