sys/contrib/libnv/dnvlist.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause
   3  *
   4  * Copyright (c) 2013 The FreeBSD Foundation
   5  *
   6  * This software was developed by Pawel Jakub Dawidek under sponsorship from
   7  * the FreeBSD Foundation.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
  19  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
  22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28  * SUCH DAMAGE.
  29  */
  30
  31 #include <sys/cdefs.h>
  32 __FBSDID("$FreeBSD$");
  33
  34 #ifdef _KERNEL
  35
  36 #include <sys/types.h>
  37 #include <sys/param.h>
  38 #include <sys/kernel.h>
  39 #include <sys/systm.h>
  40 #include <sys/malloc.h>
  41
  42 #include <machine/stdarg.h>
  43
  44 #else
  45 #include <stdarg.h>
  46 #include <stdbool.h>
  47 #include <stdint.h>
  48 #include <stdlib.h>
  49 #endif
  50
  51 #include <sys/dnv.h>
  52 #include <sys/nv.h>
  53
  54 #include "nv_impl.h"
  55
  56 #define DNVLIST_GET(ftype, type)                                        \
  57 ftype                                                                   \
  58 dnvlist_get_##type(const nvlist_t *nvl, const char *name, ftype defval) \
  59 {                                                                       \
  60                                                                         \
  61         if (nvlist_exists_##type(nvl, name))                            \
  62                 return (nvlist_get_##type(nvl, name));                  \
  63         else                                                            \
  64                 return (defval);                                        \
  65 }
  66
  67 DNVLIST_GET(bool, bool)
  68 DNVLIST_GET(uint64_t, number)
  69 DNVLIST_GET(const char *, string)
  70 DNVLIST_GET(const nvlist_t *, nvlist)
  71 #ifndef _KERNEL
  72 DNVLIST_GET(int, descriptor)
  73 #endif
  74
  75 #undef  DNVLIST_GET
  76
  77 const void *
  78 dnvlist_get_binary(const nvlist_t *nvl, const char *name, size_t *sizep,
  79     const void *defval, size_t defsize)
  80 {
  81         const void *value;
  82
  83         if (nvlist_exists_binary(nvl, name))
  84                 value = nvlist_get_binary(nvl, name, sizep);
  85         else {
  86                 if (sizep != NULL)
  87                         *sizep = defsize;
  88                 value = defval;
  89         }
  90         return (value);
  91 }
  92
  93 #define DNVLIST_TAKE(ftype, type)                                       \
  94 ftype                                                                   \
  95 dnvlist_take_##type(nvlist_t *nvl, const char *name, ftype defval)      \
  96 {                                                                       \
  97                                                                         \
  98         if (nvlist_exists_##type(nvl, name))                            \
  99                 return (nvlist_take_##type(nvl, name));                 \
 100         else                                                            \
 101                 return (defval);                                        \
 102 }
 103
 104 DNVLIST_TAKE(bool, bool)
 105 DNVLIST_TAKE(uint64_t, number)
 106 DNVLIST_TAKE(char *, string)
 107 DNVLIST_TAKE(nvlist_t *, nvlist)
 108 #ifndef _KERNEL
 109 DNVLIST_TAKE(int, descriptor)
 110 #endif
 111
 112 #undef  DNVLIST_TAKE
 113
 114 void *
 115 dnvlist_take_binary(nvlist_t *nvl, const char *name, size_t *sizep,
 116     void *defval, size_t defsize)
 117 {
 118         void *value;
 119
 120         if (nvlist_exists_binary(nvl, name))
 121                 value = nvlist_take_binary(nvl, name, sizep);
 122         else {
 123                 if (sizep != NULL)
 124                         *sizep = defsize;
 125                 value = defval;
 126         }
 127         return (value);
 128 }
 129