]> granicus.if.org Git - icinga2/blobdiff - lib/base/serializer.cpp
Add missing lock in DeserializeObject().
[icinga2] / lib / base / serializer.cpp
index 77126db5c1a5eab9ff0817d51832184aaec86df8..88c2ad7ac669229edb12534d2a86eb0950cff853 100644 (file)
@@ -1,6 +1,6 @@
 /******************************************************************************
 * Icinga 2                                                                   *
-* Copyright (C) 2012-2013 Icinga Development Team (http://www.icinga.org/)   *
+* Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org)    *
 *                                                                            *
 * This program is free software; you can redistribute it and/or              *
 * modify it under the terms of the GNU General Public License                *
 
 #include "base/serializer.h"
 #include "base/type.h"
-#include <boost/smart_ptr/make_shared.hpp>
+#include "base/application.h"
+#include "base/objectlock.h"
+#include <boost/foreach.hpp>
+#include <cJSON.h>
 
 using namespace icinga;
 
-Dictionary::Ptr Serializer::Serialize(const Object::Ptr& object, int attributeTypes)
+/**
+ * Serializes a Value into a JSON string.
+ *
+ * @returns A string representing the Value.
+ */
+String icinga::JsonSerialize(const Value& value)
 {
-       const Type *type = object->GetReflectionType();
+       cJSON *json = value.ToJson();
 
-       Dictionary::Ptr update = boost::make_shared<Dictionary>();
+       char *jsonString;
+
+       if (Application::IsDebugging())
+               jsonString = cJSON_Print(json);
+       else
+               jsonString = cJSON_PrintUnformatted(json);
+
+       cJSON_Delete(json);
+
+       String result = jsonString;
+
+       free(jsonString);
+
+       return result;
+}
+
+/**
+ * Deserializes the string representation of a Value.
+ *
+ * @param data A JSON string obtained from JsonSerialize
+ * @returns The newly deserialized Value.
+ */
+Value icinga::JsonDeserialize(const String& data)
+{
+       cJSON *json = cJSON_Parse(data.CStr());
+
+       if (!json)
+               BOOST_THROW_EXCEPTION(std::runtime_error("Invalid JSON String: " + data));
+
+       Value value = Value::FromJson(json);
+       cJSON_Delete(json);
+
+       return value;
+}
+
+static Array::Ptr SerializeArray(const Array::Ptr& input, int attributeTypes)
+{
+       Array::Ptr result = make_shared<Array>();
+
+       ObjectLock olock(input);
+
+       BOOST_FOREACH(const Value& value, input) {
+               result->Add(Serialize(value, attributeTypes));
+       }
+
+       return result;
+}
+
+static Dictionary::Ptr SerializeDictionary(const Dictionary::Ptr& input, int attributeTypes)
+{
+       Dictionary::Ptr result = make_shared<Dictionary>();
+
+       ObjectLock olock(input);
+
+       BOOST_FOREACH(const Dictionary::Pair& kv, input) {
+               result->Set(kv.first, Serialize(kv.second, attributeTypes));
+       }
+
+       return result;
+}
+
+static Object::Ptr SerializeObject(const Object::Ptr& input, int attributeTypes)
+{
+       const Type *type = input->GetReflectionType();
+
+       VERIFY(type);
+
+       Dictionary::Ptr fields = make_shared<Dictionary>();
 
        for (int i = 0; i < type->GetFieldCount(); i++) {
                Field field = type->GetFieldInfo(i);
@@ -35,25 +110,129 @@ Dictionary::Ptr Serializer::Serialize(const Object::Ptr& object, int attributeTy
                if ((field.Attributes & attributeTypes) == 0)
                        continue;
 
-               update->Set(field.Name, object->GetField(i));
+               fields->Set(field.Name, Serialize(input->GetField(i), attributeTypes));
        }
 
-       return update;
+       fields->Set("type", type->GetName());
+
+       return fields;
 }
 
-void Serializer::Deserialize(const Object::Ptr& object, const Dictionary::Ptr& update, int attributeTypes)
+static Array::Ptr DeserializeArray(const Array::Ptr& input, bool safe_mode, int attributeTypes)
 {
-       const Type *type = object->GetReflectionType();
+       Array::Ptr result = make_shared<Array>();
 
-       for (int i = 0; i < type->GetFieldCount(); i++) {
-               Field field = type->GetFieldInfo(i);
+       ObjectLock olock(input);
 
-               if ((field.Attributes & attributeTypes) == 0)
+       BOOST_FOREACH(const Value& value, input) {
+               result->Add(Deserialize(value, safe_mode, attributeTypes));
+       }
+
+       return result;
+}
+
+static Dictionary::Ptr DeserializeDictionary(const Dictionary::Ptr& input, bool safe_mode, int attributeTypes)
+{
+       Dictionary::Ptr result = make_shared<Dictionary>();
+
+       ObjectLock olock(input);
+
+       BOOST_FOREACH(const Dictionary::Pair& kv, input) {
+               result->Set(kv.first, Deserialize(kv.second, attributeTypes));
+       }
+
+       return result;
+}
+
+static Object::Ptr DeserializeObject(const Object::Ptr& object, const Dictionary::Ptr& input, bool safe_mode, int attributeTypes)
+{
+       const Type *type;
+
+       if (object)
+               type = object->GetReflectionType();
+       else
+               type = Type::GetByName(input->Get("type"));
+
+       if (!type)
+               return object;
+
+       Object::Ptr instance = object;
+
+       if (!instance) {
+               if (safe_mode && !type->IsSafe())
+                       BOOST_THROW_EXCEPTION(std::runtime_error("Tried to instantiate type '" + type->GetName() + "' which is not marked as safe."));
+
+               instance = type->Instantiate();
+       }
+
+       ObjectLock olock(input);
+       BOOST_FOREACH(const Dictionary::Pair& kv, input) {
+               if (kv.first.IsEmpty())
+                       continue;
+
+               int fid = type->GetFieldId(kv.first);
+       
+               if (fid < 0)
                        continue;
 
-               if (!update->Contains(field.Name))
+               Field field = type->GetFieldInfo(fid);
+
+               if ((field.Attributes & attributeTypes) == 0)
                        continue;
 
-               object->SetField(i, update->Get(field.Name));
+               try {
+                       instance->SetField(fid, Deserialize(kv.second, safe_mode, attributeTypes));
+               } catch (const std::exception&) {
+                       instance->SetField(fid, Empty);
+               }
        }
+
+       return instance;
+}
+
+Value icinga::Serialize(const Value& value, int attributeTypes)
+{
+       if (!value.IsObject())
+               return value;
+
+       Object::Ptr input = value;
+
+       Array::Ptr array = dynamic_pointer_cast<Array>(input);
+
+       if (array != NULL)
+               return SerializeArray(array, attributeTypes);
+
+       Dictionary::Ptr dict = dynamic_pointer_cast<Dictionary>(input);
+
+       if (dict != NULL)
+               return SerializeDictionary(dict, attributeTypes);
+
+       return SerializeObject(input, attributeTypes);
+}
+
+Value icinga::Deserialize(const Value& value, bool safe_mode, int attributeTypes)
+{
+       return Deserialize(Object::Ptr(), value, safe_mode, attributeTypes);
+}
+
+Value icinga::Deserialize(const Object::Ptr& object, const Value& value, bool safe_mode, int attributeTypes)
+{
+       if (!value.IsObject())
+               return value;
+
+       Object::Ptr input = value;
+
+       Array::Ptr array = dynamic_pointer_cast<Array>(input);
+
+       if (array != NULL)
+               return DeserializeArray(array, safe_mode, attributeTypes);
+
+       Dictionary::Ptr dict = dynamic_pointer_cast<Dictionary>(input);
+
+       ASSERT(dict != NULL);
+
+       if (!dict->Contains("type"))
+               return DeserializeDictionary(dict, safe_mode, attributeTypes);
+
+       return DeserializeObject(object, dict, safe_mode, attributeTypes);
 }