From 0fc836ce8f80493591a217ebee34a2db52620e67 Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Fri, 18 Jan 2019 16:24:26 +0300
Subject: [PATCH] Make td::as safe.

GitOrigin-RevId: 84a76a434f88e09d9c74644aee139a96bb34239d
---
 tdutils/td/utils/as.h | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tdutils/td/utils/as.h b/tdutils/td/utils/as.h
index e3fc8a24d..815e777d3 100644
--- a/tdutils/td/utils/as.h
+++ b/tdutils/td/utils/as.h
@@ -7,6 +7,7 @@
 #pragma once
 
 #include <cstring>
+#include <type_traits>
 
 namespace td {
 
@@ -60,12 +61,14 @@ class ConstAs {
 
 }  // namespace detail
 
-template <class ToT, class FromT>
+template <class ToT, class FromT,
+          std::enable_if_t<std::is_trivially_copyable<ToT>::value && std::is_trivially_copyable<FromT>::value, int> = 0>
 detail::As<ToT> as(FromT *from) {
   return detail::As<ToT>(from);
 }
 
-template <class ToT, class FromT>
+template <class ToT, class FromT,
+          std::enable_if_t<std::is_trivially_copyable<ToT>::value && std::is_trivially_copyable<FromT>::value, int> = 0>
 const detail::ConstAs<ToT> as(const FromT *from) {
   return detail::ConstAs<ToT>(from);
 }