Harden failure signal handler in the face of memory corruptions
[folly.git] / folly / dynamic-inl.h
1 /*
2  * Copyright 2014 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #ifndef FOLLY_DYNAMIC_INL_H_
18 #define FOLLY_DYNAMIC_INL_H_
19
20 #include <functional>
21 #include <boost/iterator/iterator_adaptor.hpp>
22 #include <boost/iterator/iterator_facade.hpp>
23 #include "folly/Likely.h"
24 #include "folly/Conv.h"
25 #include "folly/Format.h"
26
27 //////////////////////////////////////////////////////////////////////
28
29 namespace std {
30
31 template<>
32 struct hash< ::folly::dynamic> {
33   size_t operator()(::folly::dynamic const& d) const {
34     return d.hash();
35   }
36 };
37
38 }
39
40 //////////////////////////////////////////////////////////////////////
41
42 // This is a higher-order preprocessor macro to aid going from runtime
43 // types to the compile time type system.
44 #define FB_DYNAMIC_APPLY(type, apply) do {         \
45   switch ((type)) {                             \
46   case NULLT:   apply(void*);          break;   \
47   case ARRAY:   apply(Array);          break;   \
48   case BOOL:    apply(bool);           break;   \
49   case DOUBLE:  apply(double);         break;   \
50   case INT64:   apply(int64_t);        break;   \
51   case OBJECT:  apply(ObjectImpl);     break;   \
52   case STRING:  apply(fbstring);       break;   \
53   default:      CHECK(0); abort();              \
54   }                                             \
55 } while (0)
56
57 //////////////////////////////////////////////////////////////////////
58
59 namespace folly {
60
61 struct TypeError : std::runtime_error {
62   explicit TypeError(const std::string& expected, dynamic::Type actual)
63     : std::runtime_error(to<std::string>("TypeError: expected dynamic "
64         "type `", expected, '\'', ", but had type `",
65         dynamic::typeName(actual), '\''))
66   {}
67   explicit TypeError(const std::string& expected,
68       dynamic::Type actual1, dynamic::Type actual2)
69     : std::runtime_error(to<std::string>("TypeError: expected dynamic "
70         "types `", expected, '\'', ", but had types `",
71         dynamic::typeName(actual1), "' and `", dynamic::typeName(actual2),
72         '\''))
73   {}
74 };
75
76
77 //////////////////////////////////////////////////////////////////////
78
79 namespace detail {
80
81   // This helper is used in destroy() to be able to run destructors on
82   // types like "int64_t" without a compiler error.
83   struct Destroy {
84     template<class T> static void destroy(T* t) { t->~T(); }
85   };
86
87   /*
88    * The enable_if junk here is necessary to avoid ambiguous
89    * conversions relating to bool and double when you implicitly
90    * convert an int or long to a dynamic.
91    */
92   template<class T, class Enable = void> struct ConversionHelper;
93   template<class T>
94   struct ConversionHelper<
95     T,
96     typename std::enable_if<
97       std::is_integral<T>::value && !std::is_same<T,bool>::value
98     >::type
99   > {
100     typedef int64_t type;
101   };
102   template<class T>
103   struct ConversionHelper<
104     T,
105     typename std::enable_if<
106       (!std::is_integral<T>::value || std::is_same<T,bool>::value) &&
107       !std::is_same<T,std::nullptr_t>::value
108     >::type
109   > {
110     typedef T type;
111   };
112   template<class T>
113   struct ConversionHelper<
114     T,
115     typename std::enable_if<
116       std::is_same<T,std::nullptr_t>::value
117     >::type
118   > {
119     typedef void* type;
120   };
121
122   /*
123    * Helper for implementing numeric conversions in operators on
124    * numbers.  Just promotes to double when one of the arguments is
125    * double, or throws if either is not a numeric type.
126    */
127   template<template<class> class Op>
128   dynamic numericOp(dynamic const& a, dynamic const& b) {
129     if (!a.isNumber() || !b.isNumber()) {
130       throw TypeError("numeric", a.type(), b.type());
131     }
132     if (a.type() != b.type()) {
133       auto& integ  = a.isInt() ? a : b;
134       auto& nonint = a.isInt() ? b : a;
135       return Op<double>()(to<double>(integ.asInt()), nonint.asDouble());
136     }
137     if (a.isDouble()) {
138       return Op<double>()(a.asDouble(), b.asDouble());
139     }
140     return Op<int64_t>()(a.asInt(), b.asInt());
141   }
142
143 }
144
145 //////////////////////////////////////////////////////////////////////
146
147 /*
148  * We're doing this instead of a simple member typedef to avoid the
149  * undefined behavior of parameterizing std::unordered_map<> with an
150  * incomplete type.
151  *
152  * Note: Later we may add separate order tracking here (a multi-index
153  * type of thing.)
154  */
155 struct dynamic::ObjectImpl : std::unordered_map<dynamic, dynamic> {};
156
157 //////////////////////////////////////////////////////////////////////
158
159 // Helper object for creating objects conveniently.  See object and
160 // the dynamic::dynamic(ObjectMaker&&) ctor.
161 struct dynamic::ObjectMaker {
162   friend struct dynamic;
163
164   explicit ObjectMaker() : val_(dynamic::object) {}
165   explicit ObjectMaker(dynamic const& key, dynamic val)
166     : val_(dynamic::object)
167   {
168     val_.insert(key, std::move(val));
169   }
170   explicit ObjectMaker(dynamic&& key, dynamic val)
171     : val_(dynamic::object)
172   {
173     val_.insert(std::move(key), std::move(val));
174   }
175
176   // Make sure no one tries to save one of these into an lvalue with
177   // auto or anything like that.
178   ObjectMaker(ObjectMaker&&) = default;
179   ObjectMaker(ObjectMaker const&) = delete;
180   ObjectMaker& operator=(ObjectMaker const&) = delete;
181   ObjectMaker& operator=(ObjectMaker&&) = delete;
182
183   // These return rvalue-references instead of lvalue-references to allow
184   // constructs like this to moved instead of copied:
185   //  dynamic a = dynamic::object("a", "b")("c", "d")
186   ObjectMaker&& operator()(dynamic const& key, dynamic val) {
187     val_.insert(key, std::move(val));
188     return std::move(*this);
189   }
190
191   ObjectMaker&& operator()(dynamic&& key, dynamic val) {
192     val_.insert(std::move(key), std::move(val));
193     return std::move(*this);
194   }
195
196 private:
197   dynamic val_;
198 };
199
200 // This looks like a case for perfect forwarding, but our use of
201 // std::initializer_list for constructing dynamic arrays makes it less
202 // functional than doing this manually.
203 inline dynamic::ObjectMaker dynamic::object() { return ObjectMaker(); }
204 inline dynamic::ObjectMaker dynamic::object(dynamic&& a, dynamic&& b) {
205   return ObjectMaker(std::move(a), std::move(b));
206 }
207 inline dynamic::ObjectMaker dynamic::object(dynamic const& a, dynamic&& b) {
208   return ObjectMaker(a, std::move(b));
209 }
210 inline dynamic::ObjectMaker dynamic::object(dynamic&& a, dynamic const& b) {
211   return ObjectMaker(std::move(a), b);
212 }
213 inline dynamic::ObjectMaker
214 dynamic::object(dynamic const& a, dynamic const& b) {
215   return ObjectMaker(a, b);
216 }
217
218 //////////////////////////////////////////////////////////////////////
219
220 struct dynamic::const_item_iterator
221   : boost::iterator_adaptor<dynamic::const_item_iterator,
222                             dynamic::ObjectImpl::const_iterator> {
223   /* implicit */ const_item_iterator(base_type b) : iterator_adaptor_(b) { }
224
225  private:
226   friend class boost::iterator_core_access;
227 };
228
229 struct dynamic::const_key_iterator
230   : boost::iterator_adaptor<dynamic::const_key_iterator,
231                             dynamic::ObjectImpl::const_iterator,
232                             dynamic const> {
233   /* implicit */ const_key_iterator(base_type b) : iterator_adaptor_(b) { }
234
235  private:
236   dynamic const& dereference() const {
237     return base_reference()->first;
238   }
239   friend class boost::iterator_core_access;
240 };
241
242 struct dynamic::const_value_iterator
243   : boost::iterator_adaptor<dynamic::const_value_iterator,
244                             dynamic::ObjectImpl::const_iterator,
245                             dynamic const> {
246   /* implicit */ const_value_iterator(base_type b) : iterator_adaptor_(b) { }
247
248  private:
249   dynamic const& dereference() const {
250     return base_reference()->second;
251   }
252   friend class boost::iterator_core_access;
253 };
254
255 //////////////////////////////////////////////////////////////////////
256
257 inline dynamic::dynamic(ObjectMaker (*)())
258   : type_(OBJECT)
259 {
260   new (getAddress<ObjectImpl>()) ObjectImpl();
261 }
262
263 inline dynamic::dynamic(char const* s)
264   : type_(STRING)
265 {
266   new (&u_.string) fbstring(s);
267 }
268
269 inline dynamic::dynamic(std::string const& s)
270   : type_(STRING)
271 {
272   new (&u_.string) fbstring(s);
273 }
274
275 inline dynamic::dynamic(std::initializer_list<dynamic> il)
276   : type_(ARRAY)
277 {
278   new (&u_.array) Array(il.begin(), il.end());
279 }
280
281 inline dynamic::dynamic(ObjectMaker&& maker)
282   : type_(OBJECT)
283 {
284   new (getAddress<ObjectImpl>())
285     ObjectImpl(std::move(*maker.val_.getAddress<ObjectImpl>()));
286 }
287
288 inline dynamic::dynamic(dynamic const& o)
289   : type_(NULLT)
290 {
291   *this = o;
292 }
293
294 inline dynamic::dynamic(dynamic&& o)
295   : type_(NULLT)
296 {
297   *this = std::move(o);
298 }
299
300 inline dynamic::~dynamic() { destroy(); }
301
302 template<class T>
303 dynamic::dynamic(T t) {
304   typedef typename detail::ConversionHelper<T>::type U;
305   type_ = TypeInfo<U>::type;
306   new (getAddress<U>()) U(std::move(t));
307 }
308
309 template<class Iterator>
310 dynamic::dynamic(Iterator first, Iterator last)
311   : type_(ARRAY)
312 {
313   new (&u_.array) Array(first, last);
314 }
315
316 //////////////////////////////////////////////////////////////////////
317
318 inline dynamic::const_iterator dynamic::begin() const {
319   return get<Array>().begin();
320 }
321 inline dynamic::const_iterator dynamic::end() const {
322   return get<Array>().end();
323 }
324
325 template <class It>
326 struct dynamic::IterableProxy {
327   typedef It const_iterator;
328   typedef typename It::value_type value_type;
329
330   /* implicit */ IterableProxy(const dynamic::ObjectImpl* o) : o_(o) { }
331
332   It begin() const {
333     return o_->begin();
334   }
335
336   It end() const {
337     return o_->end();
338   }
339
340  private:
341   const dynamic::ObjectImpl* o_;
342 };
343
344 inline dynamic::IterableProxy<dynamic::const_key_iterator> dynamic::keys()
345   const {
346   return &(get<ObjectImpl>());
347 }
348
349 inline dynamic::IterableProxy<dynamic::const_value_iterator> dynamic::values()
350   const {
351   return &(get<ObjectImpl>());
352 }
353
354 inline dynamic::IterableProxy<dynamic::const_item_iterator> dynamic::items()
355   const {
356   return &(get<ObjectImpl>());
357 }
358
359 inline bool dynamic::isString() const { return get_nothrow<fbstring>(); }
360 inline bool dynamic::isObject() const { return get_nothrow<ObjectImpl>(); }
361 inline bool dynamic::isBool()   const { return get_nothrow<bool>(); }
362 inline bool dynamic::isArray()  const { return get_nothrow<Array>(); }
363 inline bool dynamic::isDouble() const { return get_nothrow<double>(); }
364 inline bool dynamic::isInt()    const { return get_nothrow<int64_t>(); }
365 inline bool dynamic::isNull()   const { return get_nothrow<void*>(); }
366 inline bool dynamic::isNumber() const { return isInt() || isDouble(); }
367
368 inline dynamic::Type dynamic::type() const {
369   return type_;
370 }
371
372 inline fbstring dynamic::asString() const { return asImpl<fbstring>(); }
373 inline double   dynamic::asDouble() const { return asImpl<double>(); }
374 inline int64_t  dynamic::asInt()    const { return asImpl<int64_t>(); }
375 inline bool     dynamic::asBool()   const { return asImpl<bool>(); }
376
377 inline const char* dynamic::data()  const { return get<fbstring>().data();  }
378 inline const char* dynamic::c_str() const { return get<fbstring>().c_str(); }
379
380 template<class T>
381 struct dynamic::CompareOp {
382   static bool comp(T const& a, T const& b) { return a < b; }
383 };
384 template<>
385 struct dynamic::CompareOp<dynamic::ObjectImpl> {
386   static bool comp(ObjectImpl const& a, ObjectImpl const& b) {
387     // This code never executes; it is just here for the compiler.
388     return false;
389   }
390 };
391
392 inline bool dynamic::operator<(dynamic const& o) const {
393   if (UNLIKELY(type_ == OBJECT || o.type_ == OBJECT)) {
394     throw TypeError("object", type_);
395   }
396   if (type_ != o.type_) {
397     return type_ < o.type_;
398   }
399
400 #define FB_X(T) return CompareOp<T>::comp(*getAddress<T>(),   \
401                                           *o.getAddress<T>())
402   FB_DYNAMIC_APPLY(type_, FB_X);
403 #undef FB_X
404 }
405
406 inline bool dynamic::operator==(dynamic const& o) const {
407   if (type() != o.type()) {
408     if (isNumber() && o.isNumber()) {
409       auto& integ = isInt() ? *this : o;
410       auto& doubl = isInt() ? o     : *this;
411       return integ.asInt() == doubl.asDouble();
412     }
413     return false;
414   }
415
416 #define FB_X(T) return *getAddress<T>() == *o.getAddress<T>();
417   FB_DYNAMIC_APPLY(type_, FB_X);
418 #undef FB_X
419 }
420
421 inline dynamic& dynamic::operator+=(dynamic const& o) {
422   if (type() == STRING && o.type() == STRING) {
423     *getAddress<fbstring>() += *o.getAddress<fbstring>();
424     return *this;
425   }
426   *this = detail::numericOp<std::plus>(*this, o);
427   return *this;
428 }
429
430 inline dynamic& dynamic::operator-=(dynamic const& o) {
431   *this = detail::numericOp<std::minus>(*this, o);
432   return *this;
433 }
434
435 inline dynamic& dynamic::operator*=(dynamic const& o) {
436   *this = detail::numericOp<std::multiplies>(*this, o);
437   return *this;
438 }
439
440 inline dynamic& dynamic::operator/=(dynamic const& o) {
441   *this = detail::numericOp<std::divides>(*this, o);
442   return *this;
443 }
444
445 #define FB_DYNAMIC_INTEGER_OP(op)                           \
446   inline dynamic& dynamic::operator op(dynamic const& o) {  \
447     if (!isInt() || !o.isInt()) {                           \
448       throw TypeError("int64", type(), o.type());           \
449     }                                                       \
450     *getAddress<int64_t>() op o.asInt();                    \
451     return *this;                                           \
452   }
453
454 FB_DYNAMIC_INTEGER_OP(%=)
455 FB_DYNAMIC_INTEGER_OP(|=)
456 FB_DYNAMIC_INTEGER_OP(&=)
457 FB_DYNAMIC_INTEGER_OP(^=)
458
459 #undef FB_DYNAMIC_INTEGER_OP
460
461 inline dynamic& dynamic::operator++() {
462   ++get<int64_t>();
463   return *this;
464 }
465
466 inline dynamic& dynamic::operator--() {
467   --get<int64_t>();
468   return *this;
469 }
470
471 inline dynamic& dynamic::operator=(dynamic const& o) {
472   if (&o != this) {
473     destroy();
474 #define FB_X(T) new (getAddress<T>()) T(*o.getAddress<T>())
475     FB_DYNAMIC_APPLY(o.type_, FB_X);
476 #undef FB_X
477     type_ = o.type_;
478   }
479   return *this;
480 }
481
482 inline dynamic& dynamic::operator=(dynamic&& o) {
483   if (&o != this) {
484     destroy();
485 #define FB_X(T) new (getAddress<T>()) T(std::move(*o.getAddress<T>()))
486     FB_DYNAMIC_APPLY(o.type_, FB_X);
487 #undef FB_X
488     type_ = o.type_;
489   }
490   return *this;
491 }
492
493 inline dynamic& dynamic::operator[](dynamic const& k) {
494   if (!isObject() && !isArray()) {
495     throw TypeError("object/array", type());
496   }
497   if (isArray()) {
498     return at(k);
499   }
500   auto& obj = get<ObjectImpl>();
501   auto ret = obj.insert({k, nullptr});
502   return ret.first->second;
503 }
504
505 inline dynamic const& dynamic::operator[](dynamic const& idx) const {
506   return at(idx);
507 }
508
509 inline dynamic dynamic::getDefault(const dynamic& k, const dynamic& v) const {
510   auto& obj = get<ObjectImpl>();
511   auto it = obj.find(k);
512   return it == obj.end() ? v : it->second;
513 }
514
515 inline dynamic&& dynamic::getDefault(const dynamic& k, dynamic&& v) const {
516   auto& obj = get<ObjectImpl>();
517   auto it = obj.find(k);
518   if (it != obj.end()) {
519     v = it->second;
520   }
521
522   return std::move(v);
523 }
524
525 template<class K, class V> inline dynamic& dynamic::setDefault(K&& k, V&& v) {
526   auto& obj = get<ObjectImpl>();
527   return obj.insert(std::make_pair(std::forward<K>(k),
528                                    std::forward<V>(v))).first->second;
529 }
530
531 inline dynamic* dynamic::get_ptr(dynamic const& idx) {
532   return const_cast<dynamic*>(const_cast<dynamic const*>(this)->get_ptr(idx));
533 }
534
535 inline const dynamic* dynamic::get_ptr(dynamic const& idx) const {
536   if (auto* parray = get_nothrow<Array>()) {
537     if (!idx.isInt()) {
538       throw TypeError("int64", idx.type());
539     }
540     if (idx >= parray->size()) {
541       return nullptr;
542     }
543     return &(*parray)[idx.asInt()];
544   } else if (auto* pobject = get_nothrow<ObjectImpl>()) {
545     auto it = pobject->find(idx);
546     if (it == pobject->end()) {
547       return nullptr;
548     }
549     return &it->second;
550   } else {
551     throw TypeError("object/array", type());
552   }
553 }
554
555 inline dynamic& dynamic::at(dynamic const& idx) {
556   return const_cast<dynamic&>(const_cast<dynamic const*>(this)->at(idx));
557 }
558
559 inline dynamic const& dynamic::at(dynamic const& idx) const {
560   if (auto* parray = get_nothrow<Array>()) {
561     if (!idx.isInt()) {
562       throw TypeError("int64", idx.type());
563     }
564     if (idx >= parray->size()) {
565       throw std::out_of_range("out of range in dynamic array");
566     }
567     return (*parray)[idx.asInt()];
568   } else if (auto* pobject = get_nothrow<ObjectImpl>()) {
569     auto it = pobject->find(idx);
570     if (it == pobject->end()) {
571       throw std::out_of_range(to<std::string>(
572           "couldn't find key ", idx.asString(), " in dynamic object"));
573     }
574     return it->second;
575   } else {
576     throw TypeError("object/array", type());
577   }
578 }
579
580 inline bool dynamic::empty() const {
581   if (isNull()) {
582     return true;
583   }
584   return !size();
585 }
586
587 inline std::size_t dynamic::size() const {
588   if (auto* ar = get_nothrow<Array>()) {
589     return ar->size();
590   }
591   if (auto* obj = get_nothrow<ObjectImpl>()) {
592     return obj->size();
593   }
594   if (auto* str = get_nothrow<fbstring>()) {
595     return str->size();
596   }
597   throw TypeError("array/object", type());
598 }
599
600 inline std::size_t dynamic::count(dynamic const& key) const {
601   return find(key) != items().end();
602 }
603
604 inline dynamic::const_item_iterator dynamic::find(dynamic const& key) const {
605   return get<ObjectImpl>().find(key);
606 }
607
608 template<class K, class V> inline void dynamic::insert(K&& key, V&& val) {
609   auto& obj = get<ObjectImpl>();
610   auto rv = obj.insert(std::make_pair(std::forward<K>(key),
611                                       std::forward<V>(val)));
612   if (!rv.second) {
613     // note, the second use of std:forward<V>(val) is only correct
614     // if the first one did not result in a move. obj[key] = val
615     // would be preferrable but doesn't compile because dynamic
616     // is (intentionally) not default constructable
617     rv.first->second = std::forward<V>(val);
618   }
619 }
620
621 inline std::size_t dynamic::erase(dynamic const& key) {
622   auto& obj = get<ObjectImpl>();
623   return obj.erase(key);
624 }
625
626 inline dynamic::const_iterator dynamic::erase(const_iterator it) {
627   auto& arr = get<Array>();
628   // std::vector doesn't have an erase method that works on const iterators,
629   // even though the standard says it should, so this hack converts to a
630   // non-const iterator before calling erase.
631   return get<Array>().erase(arr.begin() + (it - arr.begin()));
632 }
633
634 inline dynamic::const_iterator
635 dynamic::erase(const_iterator first, const_iterator last) {
636   auto& arr = get<Array>();
637   return get<Array>().erase(
638     arr.begin() + (first - arr.begin()),
639     arr.begin() + (last - arr.begin()));
640 }
641
642 inline dynamic::const_key_iterator dynamic::erase(const_key_iterator it) {
643   return const_key_iterator(get<ObjectImpl>().erase(it.base()));
644 }
645
646 inline dynamic::const_key_iterator dynamic::erase(const_key_iterator first,
647                                                   const_key_iterator last) {
648   return const_key_iterator(get<ObjectImpl>().erase(first.base(),
649                                                     last.base()));
650 }
651
652 inline dynamic::const_value_iterator dynamic::erase(const_value_iterator it) {
653   return const_value_iterator(get<ObjectImpl>().erase(it.base()));
654 }
655
656 inline dynamic::const_value_iterator dynamic::erase(const_value_iterator first,
657                                                     const_value_iterator last) {
658   return const_value_iterator(get<ObjectImpl>().erase(first.base(),
659                                                       last.base()));
660 }
661
662 inline dynamic::const_item_iterator dynamic::erase(const_item_iterator it) {
663   return const_item_iterator(get<ObjectImpl>().erase(it.base()));
664 }
665
666 inline dynamic::const_item_iterator dynamic::erase(const_item_iterator first,
667                                                    const_item_iterator last) {
668   return const_item_iterator(get<ObjectImpl>().erase(first.base(),
669                                                      last.base()));
670 }
671
672 inline void dynamic::resize(std::size_t sz, dynamic const& c) {
673   auto& array = get<Array>();
674   array.resize(sz, c);
675 }
676
677 inline void dynamic::push_back(dynamic const& v) {
678   auto& array = get<Array>();
679   array.push_back(v);
680 }
681
682 inline void dynamic::push_back(dynamic&& v) {
683   auto& array = get<Array>();
684   array.push_back(std::move(v));
685 }
686
687 inline void dynamic::pop_back() {
688   auto& array = get<Array>();
689   array.pop_back();
690 }
691
692 inline std::size_t dynamic::hash() const {
693   switch (type()) {
694   case OBJECT:
695   case ARRAY:
696   case NULLT:
697     throw TypeError("not null/object/array", type());
698   case INT64:
699     return std::hash<int64_t>()(asInt());
700   case DOUBLE:
701     return std::hash<double>()(asDouble());
702   case BOOL:
703     return std::hash<bool>()(asBool());
704   case STRING:
705     return std::hash<fbstring>()(asString());
706   default:
707     CHECK(0); abort();
708   }
709 }
710
711 //////////////////////////////////////////////////////////////////////
712
713 template<class T> struct dynamic::TypeInfo {
714   static char const name[];
715   static Type const type;
716 };
717
718 #define FB_DEC_TYPE(T)                                      \
719   template<> char const dynamic::TypeInfo<T>::name[];       \
720   template<> dynamic::Type const dynamic::TypeInfo<T>::type
721
722 FB_DEC_TYPE(void*);
723 FB_DEC_TYPE(bool);
724 FB_DEC_TYPE(fbstring);
725 FB_DEC_TYPE(dynamic::Array);
726 FB_DEC_TYPE(double);
727 FB_DEC_TYPE(int64_t);
728 FB_DEC_TYPE(dynamic::ObjectImpl);
729
730 #undef FB_DEC_TYPE
731
732 template<class T>
733 T dynamic::asImpl() const {
734   switch (type()) {
735   case INT64:    return to<T>(*get_nothrow<int64_t>());
736   case DOUBLE:   return to<T>(*get_nothrow<double>());
737   case BOOL:     return to<T>(*get_nothrow<bool>());
738   case STRING:   return to<T>(*get_nothrow<fbstring>());
739   default:
740     throw TypeError("int/double/bool/string", type());
741   }
742 }
743
744 // Return a T* to our type, or null if we're not that type.
745 template<class T>
746 T* dynamic::get_nothrow() {
747   if (type_ != TypeInfo<T>::type) {
748     return nullptr;
749   }
750   return getAddress<T>();
751 }
752
753 template<class T>
754 T const* dynamic::get_nothrow() const {
755   return const_cast<dynamic*>(this)->get_nothrow<T>();
756 }
757
758 // Return T* for where we can put a T, without type checking.  (Memory
759 // might be uninitialized, even.)
760 template<class T>
761 T* dynamic::getAddress() {
762   return GetAddrImpl<T>::get(u_);
763 }
764
765 template<class T>
766 T const* dynamic::getAddress() const {
767   return const_cast<dynamic*>(this)->getAddress<T>();
768 }
769
770 template<class T> struct dynamic::GetAddrImpl {};
771 template<> struct dynamic::GetAddrImpl<void*> {
772   static void** get(Data& d) { return &d.nul; }
773 };
774 template<> struct dynamic::GetAddrImpl<dynamic::Array> {
775   static Array* get(Data& d) { return &d.array; }
776 };
777 template<> struct dynamic::GetAddrImpl<bool> {
778   static bool* get(Data& d) { return &d.boolean; }
779 };
780 template<> struct dynamic::GetAddrImpl<int64_t> {
781   static int64_t* get(Data& d) { return &d.integer; }
782 };
783 template<> struct dynamic::GetAddrImpl<double> {
784   static double* get(Data& d) { return &d.doubl; }
785 };
786 template<> struct dynamic::GetAddrImpl<fbstring> {
787   static fbstring* get(Data& d) { return &d.string; }
788 };
789 template<> struct dynamic::GetAddrImpl<dynamic::ObjectImpl> {
790   static_assert(sizeof(ObjectImpl) <= sizeof(Data::objectBuffer),
791     "In your implementation, std::unordered_map<> apparently takes different"
792     " amount of space depending on its template parameters.  This is "
793     "weird.  Make objectBuffer bigger if you want to compile dynamic.");
794
795   static ObjectImpl* get(Data& d) {
796     void* data = &d.objectBuffer;
797     return static_cast<ObjectImpl*>(data);
798   }
799 };
800
801 template<class T>
802 T& dynamic::get() {
803   if (auto* p = get_nothrow<T>()) {
804     return *p;
805   }
806   throw TypeError(TypeInfo<T>::name, type());
807 }
808
809 template<class T>
810 T const& dynamic::get() const {
811   return const_cast<dynamic*>(this)->get<T>();
812 }
813
814 inline char const* dynamic::typeName(Type t) {
815 #define FB_X(T) return TypeInfo<T>::name
816   FB_DYNAMIC_APPLY(t, FB_X);
817 #undef FB_X
818 }
819
820 inline void dynamic::destroy() {
821   // This short-circuit speeds up some microbenchmarks.
822   if (type_ == NULLT) return;
823
824 #define FB_X(T) detail::Destroy::destroy(getAddress<T>())
825   FB_DYNAMIC_APPLY(type_, FB_X);
826 #undef FB_X
827   type_ = NULLT;
828   u_.nul = nullptr;
829 }
830
831 //////////////////////////////////////////////////////////////////////
832
833 /*
834  * Helper for implementing operator<<.  Throws if the type shouldn't
835  * support it.
836  */
837 template<class T>
838 struct dynamic::PrintImpl {
839   static void print(dynamic const&, std::ostream& out, T const& t) {
840     out << t;
841   }
842 };
843 template<>
844 struct dynamic::PrintImpl<dynamic::ObjectImpl> {
845   static void print(dynamic const& d,
846                     std::ostream& out,
847                     dynamic::ObjectImpl const&) {
848     d.print_as_pseudo_json(out);
849   }
850 };
851 template<>
852 struct dynamic::PrintImpl<dynamic::Array> {
853   static void print(dynamic const& d,
854                     std::ostream& out,
855                     dynamic::Array const&) {
856     d.print_as_pseudo_json(out);
857   }
858 };
859
860 inline void dynamic::print(std::ostream& out) const {
861 #define FB_X(T) PrintImpl<T>::print(*this, out, *getAddress<T>())
862   FB_DYNAMIC_APPLY(type_, FB_X);
863 #undef FB_X
864 }
865
866 inline std::ostream& operator<<(std::ostream& out, dynamic const& d) {
867   d.print(out);
868   return out;
869 }
870
871 //////////////////////////////////////////////////////////////////////
872
873 // Secialization of FormatValue so dynamic objects can be formatted
874 template <>
875 class FormatValue<dynamic> {
876  public:
877   explicit FormatValue(const dynamic& val) : val_(val) { }
878
879   template <class FormatCallback>
880   void format(FormatArg& arg, FormatCallback& cb) const {
881     switch (val_.type()) {
882     case dynamic::NULLT:
883       FormatValue<std::nullptr_t>(nullptr).format(arg, cb);
884       break;
885     case dynamic::BOOL:
886       FormatValue<bool>(val_.asBool()).format(arg, cb);
887       break;
888     case dynamic::INT64:
889       FormatValue<int64_t>(val_.asInt()).format(arg, cb);
890       break;
891     case dynamic::STRING:
892       FormatValue<fbstring>(val_.asString()).format(arg, cb);
893       break;
894     case dynamic::DOUBLE:
895       FormatValue<double>(val_.asDouble()).format(arg, cb);
896       break;
897     case dynamic::ARRAY:
898       FormatValue(val_.at(arg.splitIntKey())).format(arg, cb);
899       break;
900     case dynamic::OBJECT:
901       FormatValue(val_.at(arg.splitKey().toFbstring())).format(arg, cb);
902       break;
903     }
904   }
905
906  private:
907   const dynamic& val_;
908 };
909
910 }
911
912 #undef FB_DYNAMIC_APPLY
913
914 #endif