view: security net and explicit arrays to avoid regressions

This commit is contained in:
Michele Caini
2022-05-06 09:31:23 +02:00
parent 7aa276df5a
commit e0a1a338a5
2 changed files with 27 additions and 26 deletions

View File

@@ -25,16 +25,14 @@ namespace entt {
namespace internal {
template<typename, typename, typename>
class view_iterator;
template<typename Type, std::size_t... AllOf, std::size_t... AnyOf>
class view_iterator<Type, std::index_sequence<AllOf...>, std::index_sequence<AnyOf...>> final {
static constexpr std::size_t pool_count = sizeof...(AllOf) + sizeof...(AnyOf);
template<typename Type, std::size_t Component, std::size_t Exclude>
class view_iterator final {
using iterator_type = typename Type::const_iterator;
[[nodiscard]] bool valid() const noexcept {
return ((sizeof...(AllOf) != 0u) || (*it != tombstone)) && (std::get<AllOf>(pools)->contains(*it) && ...) && (!std::get<AnyOf>(pools)->contains(*it) && ...);
return ((Component != 0u) || (*it != tombstone))
&& std::apply([entt = *it](const auto *...curr) { return (curr->contains(entt) && ...); }, pools)
&& std::apply([entt = *it](const auto *...curr) { return (!curr->contains(entt) && ...); }, filter);
}
public:
@@ -46,10 +44,11 @@ public:
constexpr view_iterator() noexcept = default;
view_iterator(iterator_type curr, iterator_type to, std::array<const Type *, pool_count> ref) noexcept
view_iterator(iterator_type curr, iterator_type to, std::array<const Type *, Component> all_of, std::array<const Type *, Exclude> none_of) noexcept
: it{curr},
last{to},
pools{ref} {
pools{all_of},
filter{none_of} {
if(it != last && !valid()) {
++(*this);
}
@@ -73,21 +72,22 @@ public:
return *operator->();
}
template<typename LhsType, typename... LhsArgs, typename RhsType, typename... RhsArgs>
template<typename LhsType, auto... LhsArgs, typename RhsType, auto... RhsArgs>
friend constexpr bool operator==(const view_iterator<LhsType, LhsArgs...> &, const view_iterator<RhsType, RhsArgs...> &) noexcept;
private:
iterator_type it;
iterator_type last;
std::array<const Type *, pool_count> pools;
std::array<const Type *, Component> pools;
std::array<const Type *, Exclude> filter;
};
template<typename LhsType, typename... LhsArgs, typename RhsType, typename... RhsArgs>
template<typename LhsType, auto... LhsArgs, typename RhsType, auto... RhsArgs>
[[nodiscard]] constexpr bool operator==(const view_iterator<LhsType, LhsArgs...> &lhs, const view_iterator<RhsType, RhsArgs...> &rhs) noexcept {
return lhs.it == rhs.it;
}
template<typename LhsType, typename... LhsArgs, typename RhsType, typename... RhsArgs>
template<typename LhsType, auto... LhsArgs, typename RhsType, auto... RhsArgs>
[[nodiscard]] constexpr bool operator!=(const view_iterator<LhsType, LhsArgs...> &lhs, const view_iterator<RhsType, RhsArgs...> &rhs) noexcept {
return !(lhs == rhs);
}
@@ -191,13 +191,14 @@ class basic_view<Entity, get_t<Component...>, exclude_t<Exclude...>> {
template<typename Comp>
using storage_type = constness_as_t<typename storage_traits<Entity, std::remove_const_t<Comp>>::storage_type, Comp>;
template<std::size_t... Index>
[[nodiscard]] auto opaque_set(std::index_sequence<Index...>) const noexcept {
std::size_t pos{};
std::array<const base_type *, sizeof...(Component) + sizeof...(Exclude) - 1u> arr{};
(static_cast<void>(std::get<Index>(pools) == view ? void() : void(arr[pos++] = std::get<Index>(pools))), ...);
std::apply([&arr, &pos](auto *...curr) mutable { ((arr[pos++] = curr), ...); }, filter);
return arr;
[[nodiscard]] auto opaque_check() const noexcept {
std::array<const base_type *, sizeof...(Component) - 1u> other{};
std::apply([&other, pos = 0u, view = view](const auto *...curr) mutable { ((curr == view ? void() : void(other[pos++] = curr)), ...); }, pools);
return other;
}
[[nodiscard]] auto opaque_filter() const noexcept {
return std::apply([](const auto *...curr) { return std::array<const base_type *, sizeof...(Exclude)>{curr...}; }, filter);
}
template<std::size_t Comp, std::size_t Other, typename... Args>
@@ -239,7 +240,7 @@ public:
/*! @brief Common type among all storage types. */
using base_type = std::common_type_t<typename storage_type<Component>::base_type...>;
/*! @brief Bidirectional iterator type. */
using iterator = internal::view_iterator<base_type, std::make_index_sequence<sizeof...(Component) - 1u>, std::index_sequence_for<Exclude...>>;
using iterator = internal::view_iterator<base_type, sizeof...(Component) - 1u, sizeof...(Exclude)>;
/*! @brief Iterable view type. */
using iterable = iterable_adaptor<internal::extended_view_iterator<iterator, storage_type<Component>...>>;
@@ -336,7 +337,7 @@ public:
* @return An iterator to the first entity of the view.
*/
[[nodiscard]] iterator begin() const noexcept {
return iterator{view->begin(), view->end(), opaque_set(std::index_sequence_for<Component...>{})};
return iterator{view->begin(), view->end(), opaque_check(), opaque_filter()};
}
/**
@@ -349,7 +350,7 @@ public:
* @return An iterator to the entity following the last entity of the view.
*/
[[nodiscard]] iterator end() const noexcept {
return iterator{view->end(), view->end(), opaque_set(std::index_sequence_for<Component...>{})};
return iterator{view->end(), view->end(), opaque_check(), opaque_filter()};
}
/**
@@ -380,7 +381,7 @@ public:
* iterator otherwise.
*/
[[nodiscard]] iterator find(const entity_type entt) const noexcept {
return contains(entt) ? iterator{view->find(entt), view->end(), opaque_set(std::index_sequence_for<Component...>{})} : end();
return contains(entt) ? iterator{view->find(entt), view->end(), opaque_check(), opaque_filter()} : end();
}
/**

View File

@@ -169,7 +169,7 @@ TEST(SingleComponentView, Each) {
entt::registry registry;
entt::entity entity[2]{registry.create(), registry.create()};
auto view = registry.view<int>();
auto view = registry.view<int>(entt::exclude<double>);
auto cview = std::as_const(registry).view<const int>();
registry.emplace<int>(entity[0u], 0);
@@ -673,7 +673,7 @@ TEST(MultiComponentView, Each) {
entt::registry registry;
entt::entity entity[2]{registry.create(), registry.create()};
auto view = registry.view<int, char>();
auto view = registry.view<int, char>(entt::exclude<double>);
auto cview = std::as_const(registry).view<const int, const char>();
registry.emplace<int>(entity[0u], 0);