summaryrefslogtreecommitdiffstats
path: root/src/lib/cc/user_context.cc
blob: 74104006ef2349d6aeb60c56fc7232abce6f5280 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
// Copyright (C) 2017-2018 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include <config.h>

#include <cc/user_context.h>

using namespace isc::data;

namespace isc {
namespace dhcp {

void
UserContext::contextToElement(ElementPtr map) const {
    // Set user-context extracting comment
    ConstElementPtr context = getContext();
    if (!isNull(context)) {
        if ((context->getType() == Element::map) &&
            context->contains("comment")) {
            ElementPtr copied = isc::data::copy(context);
            map->set("comment", copied->get("comment"));
            copied->remove("comment");
            if (copied->size() > 0) {
                map->set("user-context", copied);
            }
        } else {
            map->set("user-context", context);
        }
    }
}

ElementPtr
UserContext::toElement(ConstElementPtr map) {
    ElementPtr result = isc::data::copy(map);
    // Protect against argument not map
    if (result->getType() != Element::map) {
        return (result);
    }
    ConstElementPtr ctx = result->get("user-context");
    // Protect against user context not map
    if (!ctx || (ctx->getType() != Element::map)) {
        return (result);
    }
    // Extract comment
    if (ctx->contains("comment")) {
        ElementPtr ctx_copy = isc::data::copy(ctx);
        result->set("comment", ctx_copy->get("comment"));
        ctx_copy->remove("comment");
        result->remove("user-context");
        if (ctx_copy->size() > 0) {
            result->set("user-context", ctx_copy);
        }
    }
    return (result);
}

}; // end of isc::dhcp namespace
}; // end of isc namespace