dict_ext.c

Go to the documentation of this file.

/*
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the Free Software
 *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */
 
/** Extensions to dictionary structures
 *
 * @file src/lib/util/dict_ext.c
 *
 * @copyright 2020 The FreeRADIUS server project
 * @copyright 2020,2024 Arran Cudbard-Bell <a.cudbardb@freeradius.org>
 */
RCSID("$Id: e64fb640bf096fdf88ff31d6da368c25a8275969 $")
 
#include <freeradius-devel/util/dict_priv.h>
#include <freeradius-devel/util/dict_ext_priv.h>
 
static fr_table_num_ordered_t const dict_attr_ext_table[] = {
        { L("name"),                    FR_DICT_ATTR_EXT_NAME                   },
        { L("children"),                FR_DICT_ATTR_EXT_CHILDREN               },
        { L("ref"),                     FR_DICT_ATTR_EXT_REF                    },
        { L("vendor"),                  FR_DICT_ATTR_EXT_VENDOR                 },
        { L("da_stack"),                FR_DICT_ATTR_EXT_DA_STACK               },
        { L("enumv"),                   FR_DICT_ATTR_EXT_ENUMV                  },
        { L("namespace"),               FR_DICT_ATTR_EXT_NAMESPACE              },
        { L("protocol-specific"),       FR_DICT_ATTR_EXT_PROTOCOL_SPECIFIC      }
};
static size_t dict_attr_ext_table_len = NUM_ELEMENTS(dict_attr_ext_table);
 
/** Fixup name pointer on realloc
 *
 */
static int fr_dict_attr_ext_name_fixup(UNUSED int ext,
                                       TALLOC_CTX *chunk,
                                       void *ext_ptr, UNUSED size_t ext_ptr_len)
{
        fr_dict_attr_t                  *da = talloc_get_type_abort(chunk, fr_dict_attr_t);
 
        da->name = ext_ptr;
 
        return 0;
}
 
/** Copy all enumeration values from one attribute to another
 *
 */
static int fr_dict_attr_ext_enumv_copy(UNUSED int ext,
                                       TALLOC_CTX *chunk_dst,
                                       UNUSED void *dst_ext_ptr, UNUSED size_t dst_ext_len,
                                       TALLOC_CTX const *chunk_src,
                                       void *src_ext_ptr, UNUSED size_t src_ext_len)
{
        fr_dict_attr_t const            *da_src = talloc_get_type_abort_const(chunk_src, fr_dict_attr_t);
        fr_dict_attr_t                  *da_dst = talloc_get_type_abort(chunk_dst, fr_dict_attr_t);
        fr_dict_attr_ext_enumv_t        *src_ext = src_ext_ptr;
        fr_hash_iter_t                  iter;
        fr_dict_enum_value_t                    *enumv;
        bool                            has_child = fr_dict_attr_is_key_field(da_src);
 
        if (!src_ext->value_by_name) return 0;
 
        /*
         *      Add all the enumeration values from
         *      the old attribute to the new attribute.
         */
        for (enumv = fr_hash_table_iter_init(src_ext->value_by_name, &iter);
             enumv;
             enumv = fr_hash_table_iter_next(src_ext->value_by_name, &iter)) {
                fr_dict_attr_t *child_struct;
 
                if (!has_child) {
                        child_struct = NULL;
                } else {
                        fr_dict_t *dict = dict_by_da(enumv->child_struct[0]);
 
                        /*
                         *      Copy the child_struct, and all if it's children recursively.
                         */
                        child_struct = dict_attr_acopy(dict->pool, enumv->child_struct[0], NULL);
                        if (!child_struct) return -1;
 
                        child_struct->parent = da_dst; /* we need to re-parent this attribute */
 
                        if (dict_attr_children(enumv->child_struct[0])) {
                                if (dict_attr_acopy_children(dict, child_struct, enumv->child_struct[0]) < 0) return -1;
                        }
                }
 
                if (dict_attr_enum_add_name(da_dst, enumv->name, enumv->value,
                                            true, true, child_struct) < 0) return -1;
        }
 
        return 0;
}
 
/** Rediscover the parent of this attribute, and cache it
 *
 */
static int fr_dict_attr_ext_vendor_copy(UNUSED int ext,
                                        TALLOC_CTX *chunk_dst,
                                        void *dst_ext_ptr, UNUSED size_t dst_ext_len,
                                        UNUSED TALLOC_CTX const *chunk_src,
                                        void *src_ext_ptr, UNUSED size_t src_ext_len)
{
        fr_dict_attr_t                  *da_dst = talloc_get_type_abort(chunk_dst, fr_dict_attr_t);
        fr_dict_attr_ext_vendor_t       *dst_ext = dst_ext_ptr, *src_ext = src_ext_ptr;
        fr_dict_attr_t const            **da_stack;
        fr_dict_attr_t const            *old_vendor = src_ext->vendor;
        fr_dict_attr_t const            *new_vendor, *da;
 
        if (!old_vendor) {
                dst_ext->vendor = NULL;
                return 0;
        }
 
        /*
         *      If we have a da stack, see if we can
         *      find a vendor at the same depth as
         *      the old depth.
         */
        da_stack = fr_dict_attr_da_stack(da_dst);
        if (da_stack) {
                new_vendor = da_stack[old_vendor->depth];
                if ((new_vendor->type == old_vendor->type) && (new_vendor->attr == old_vendor->attr)) {
                        dst_ext->vendor = new_vendor;
                        return 0;
                }
        }
 
        /*
         *      Otherwise traverse the parent list
         *      looking for the vendor.
         *
         *      Theoretically the attribute could
         *      have been moved to a different depth.
         */
        for (da = da_dst->parent; da; da = da->parent) {
                if ((da->type == old_vendor->type) && (da->attr == old_vendor->attr)) {
                        dst_ext->vendor = da;
                        return 0;
                }
        }
 
        return -1;
}
 
static int dict_ext_protocol_specific_copy(UNUSED int ext,
                                           TALLOC_CTX *dst_chunk,
                                           void *dst_ext_ptr, size_t dst_ext_len,
                                           TALLOC_CTX const *src_chunk,
                                           void *src_ext_ptr, size_t src_ext_len)
{
        fr_dict_attr_t const            *from = talloc_get_type_abort_const(src_chunk, fr_dict_attr_t);
        fr_dict_protocol_t const        *from_proto = fr_dict_protocol(from->dict);
        fr_dict_attr_t                  *to = talloc_get_type_abort(dst_chunk, fr_dict_attr_t);
        fr_dict_protocol_t const        *to_proto = fr_dict_protocol(to->dict);
 
        /*
         *      Whilst it's not strictly disallowed, we can't do anything
         *      sane without an N x N matrix of copy functions for different
         *      protocols.  Maybe we should add that at some point, but for
         *      now, just ignore the copy.
         */
        if (from->dict != to->dict) return 0;
 
        /*
         *      Sanity checks...
         */
        if (unlikely(from_proto->attr.flags.len != src_ext_len)) {
                fr_strerror_printf("Protocol specific extension length mismatch in source attribute %s.  Expected %zu, got %zu",
                                   from->name,
                                   fr_dict_protocol(from->dict)->attr.flags.len, fr_dict_protocol(to->dict)->attr.flags.len);
                return -1;
        }
 
        if (unlikely(to_proto->attr.flags.len != dst_ext_len)) {
                fr_strerror_printf("Protocol specific extension length mismatch in destintion attribute %s.  Expected %zu, got %zu",
                                   to->name,
                                   fr_dict_protocol(to->dict)->attr.flags.len, fr_dict_protocol(to->dict)->attr.flags.len);
                return -1;
        }
 
        /*
         *      The simple case... No custom copy function, just memcpy
         */
        if (!to_proto->attr.flags.copy) {
                memcpy(dst_ext_ptr, src_ext_ptr, src_ext_len);
                return 0;
        }
 
        /*
         *      Call the custom copy function.  This is only needed if
         *      there are heap allocated values, like strings, which
         *      need copying from sources flags to the destination.
         */
        return to_proto->attr.flags.copy(dst_chunk, dst_ext_ptr, src_ext_ptr);
}
 
/** Holds additional information about extension structures
 *
 */
fr_ext_t const fr_dict_attr_ext_def = {
        .offset_of_exts = offsetof(fr_dict_attr_t, ext),
        .name_table     = dict_attr_ext_table,
        .name_table_len = &dict_attr_ext_table_len,
        .max            = FR_DICT_ATTR_EXT_MAX,
        .info           =  (fr_ext_info_t[]){ /* -Wgnu-flexible-array-initializer */
                [FR_DICT_ATTR_EXT_NAME]         = {
                                                        .min = sizeof(char),
                                                        .has_hdr = true,
                                                        .fixup = fr_dict_attr_ext_name_fixup,
                                                        .can_copy = false,              /* Name may change, and we can only set it once */
                                                },
                [FR_DICT_ATTR_EXT_CHILDREN]     = {
                                                        .min = sizeof(fr_dict_attr_ext_children_t),
                                                        .can_copy = false,              /* Limitation in hashing scheme we use */
                                                },
                [FR_DICT_ATTR_EXT_REF]          = {
                                                        .min = sizeof(fr_dict_attr_ext_ref_t),
                                                        .can_copy = true,
                                                },
                [FR_DICT_ATTR_EXT_VENDOR]       = {
                                                        .min = sizeof(fr_dict_attr_ext_vendor_t),
                                                        .can_copy = true,
                                                        .copy = fr_dict_attr_ext_vendor_copy
                                                },
                [FR_DICT_ATTR_EXT_DA_STACK]     = {
                                                        .min = sizeof(fr_dict_attr_ext_da_stack_t),
                                                        .has_hdr = true,
                                                        .can_copy = false               /* Reinitialised for each new attribute */
                                                },
                [FR_DICT_ATTR_EXT_ENUMV]        = {
                                                        .min = sizeof(fr_dict_attr_ext_enumv_t),
                                                        .can_copy = true,
                                                        .copy = fr_dict_attr_ext_enumv_copy
                                                },
                [FR_DICT_ATTR_EXT_NAMESPACE]    = {
                                                        .min = sizeof(fr_dict_attr_ext_namespace_t),
                                                        .can_copy = false,              /* Same limitation as ext_children */
                                                },
                [FR_DICT_ATTR_EXT_PROTOCOL_SPECIFIC] = {
                                                        .min = FR_EXT_ALIGNMENT,        /* allow for one byte of protocol stuff */
                                                        .has_hdr = true,                /* variable sized */
                                                        .copy = dict_ext_protocol_specific_copy,
                                                        .can_copy = true                /* Use the attr.flags.copy function */
                                                },
                [FR_DICT_ATTR_EXT_MAX]          = {}
        }
};
 
static fr_table_num_ordered_t const dict_enum_ext_table[] = {
        { L("union_ref"),       FR_DICT_ENUM_EXT_UNION_REF      }
};
static size_t dict_enum_ext_table_len = NUM_ELEMENTS(dict_enum_ext_table);
 
void fr_dict_attr_ext_debug(fr_dict_attr_t const *da)
{
        fr_ext_debug(&fr_dict_attr_ext_def, da->name, da);
}
 
/** Holds additional information about extension structures
 *
 */
fr_ext_t const fr_dict_enum_ext_def = {
        .offset_of_exts = offsetof(fr_dict_enum_value_t, ext),
        .name_table     = dict_enum_ext_table,
        .name_table_len = &dict_enum_ext_table_len,
        .max            = FR_DICT_ENUM_EXT_MAX,
        .info           = (fr_ext_info_t[]){ /* -Wgnu-flexible-array-initializer */
                [FR_DICT_ENUM_EXT_UNION_REF]    = {
                                                        .min = sizeof(fr_dict_enum_ext_union_ref_t),
                                                        .can_copy = true
                                                },
                [FR_DICT_ENUM_EXT_MAX]          = {}
        }
};