mirror of
https://github.com/HowardHinnant/date.git
synced 2025-01-13 17:34:45 +08:00
Add support for links to links
This commit is contained in:
parent
cd3c57932f
commit
8f8336f42b
54
src/tz.cpp
54
src/tz.cpp
@ -3711,6 +3711,48 @@ get_tzdb()
|
|||||||
return get_tzdb_list().front();
|
return get_tzdb_list().front();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
class recursion_limiter
|
||||||
|
{
|
||||||
|
unsigned depth_ = 0;
|
||||||
|
unsigned limit_;
|
||||||
|
|
||||||
|
class restore_recursion_depth
|
||||||
|
{
|
||||||
|
recursion_limiter* rc_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
~restore_recursion_depth() {--(rc_->depth_);}
|
||||||
|
restore_recursion_depth(restore_recursion_depth&&) = default;
|
||||||
|
|
||||||
|
explicit restore_recursion_depth(recursion_limiter* rc) noexcept
|
||||||
|
: rc_{rc}
|
||||||
|
{}
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
recursion_limiter(recursion_limiter const&) = delete;
|
||||||
|
recursion_limiter& operator=(recursion_limiter const&) = delete;
|
||||||
|
|
||||||
|
explicit recursion_limiter(unsigned limit) noexcept
|
||||||
|
: limit_{limit}
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
restore_recursion_depth
|
||||||
|
count()
|
||||||
|
{
|
||||||
|
++depth_;
|
||||||
|
if (depth_ > limit_)
|
||||||
|
throw std::runtime_error("recursion limit of " +
|
||||||
|
std::to_string(limit_) + " exceeded");
|
||||||
|
return restore_recursion_depth{this};
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
} // unnamed namespace
|
||||||
|
|
||||||
const time_zone*
|
const time_zone*
|
||||||
#if HAS_STRING_VIEW
|
#if HAS_STRING_VIEW
|
||||||
tzdb::locate_zone(std::string_view tz_name) const
|
tzdb::locate_zone(std::string_view tz_name) const
|
||||||
@ -3718,6 +3760,10 @@ tzdb::locate_zone(std::string_view tz_name) const
|
|||||||
tzdb::locate_zone(const std::string& tz_name) const
|
tzdb::locate_zone(const std::string& tz_name) const
|
||||||
#endif
|
#endif
|
||||||
{
|
{
|
||||||
|
// If a link-to-link chain exceeds this limit, give up
|
||||||
|
thread_local recursion_limiter rc{10};
|
||||||
|
auto restore_count = rc.count();
|
||||||
|
|
||||||
auto zi = std::lower_bound(zones.begin(), zones.end(), tz_name,
|
auto zi = std::lower_bound(zones.begin(), zones.end(), tz_name,
|
||||||
#if HAS_STRING_VIEW
|
#if HAS_STRING_VIEW
|
||||||
[](const time_zone& z, const std::string_view& nm)
|
[](const time_zone& z, const std::string_view& nm)
|
||||||
@ -3741,13 +3787,7 @@ tzdb::locate_zone(const std::string& tz_name) const
|
|||||||
});
|
});
|
||||||
if (li != links.end() && li->name() == tz_name)
|
if (li != links.end() && li->name() == tz_name)
|
||||||
{
|
{
|
||||||
zi = std::lower_bound(zones.begin(), zones.end(), li->target(),
|
return locate_zone(li->target());
|
||||||
[](const time_zone& z, const std::string& nm)
|
|
||||||
{
|
|
||||||
return z.name() < nm;
|
|
||||||
});
|
|
||||||
if (zi != zones.end() && zi->name() == li->target())
|
|
||||||
return &*zi;
|
|
||||||
}
|
}
|
||||||
#endif // !USE_OS_TZDB
|
#endif // !USE_OS_TZDB
|
||||||
throw std::runtime_error(std::string(tz_name) + " not found in timezone database");
|
throw std::runtime_error(std::string(tz_name) + " not found in timezone database");
|
||||||
|
Loading…
x
Reference in New Issue
Block a user