stdio-common/reg-printf.c

   1 /* Copyright (C) 1991-2017 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3
   4    The GNU C Library is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU Lesser General Public
   6    License as published by the Free Software Foundation; either
   7    version 2.1 of the License, or (at your option) any later version.
   8
   9    The GNU C Library is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12    Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public
  15    License along with the GNU C Library; if not, see
  16    <http://www.gnu.org/licenses/>.  */
  17
  18 #include <errno.h>
  19 #include <limits.h>
  20 #include <printf.h>
  21 #include <stddef.h>
  22 #include <stdlib.h>
  23 #include <libc-lock.h>
  24
  25
  26 /* Array of functions indexed by format character.  */
  27 libc_freeres_ptr (printf_arginfo_size_function **__printf_arginfo_table)
  28   attribute_hidden;
  29 printf_function **__printf_function_table attribute_hidden;
  30
  31 __libc_lock_define_initialized (static, lock)
  32
  33 int __register_printf_specifier (int, printf_function,
  34                                  printf_arginfo_size_function);
  35 int __register_printf_function (int, printf_function,
  36                                 printf_arginfo_function);
  37
  38
  39 /* Register FUNC to be called to format SPEC specifiers.  */
  40 int
  41 __register_printf_specifier (int spec, printf_function converter,
  42                              printf_arginfo_size_function arginfo)
  43 {
  44   if (spec < 0 || spec > (int) UCHAR_MAX)
  45     {
  46       __set_errno (EINVAL);
  47       return -1;
  48     }
  49
  50   int result = 0;
  51   __libc_lock_lock (lock);
  52
  53   if (__printf_function_table == NULL)
  54     {
  55       __printf_arginfo_table = (printf_arginfo_size_function **)
  56         calloc (UCHAR_MAX + 1, sizeof (void *) * 2);
  57       if (__printf_arginfo_table == NULL)
  58         {
  59           result = -1;
  60           goto out;
  61         }
  62
  63       __printf_function_table = (printf_function **)
  64         (__printf_arginfo_table + UCHAR_MAX + 1);
  65     }
  66
  67   __printf_function_table[spec] = converter;
  68   __printf_arginfo_table[spec] = arginfo;
  69
  70  out:
  71   __libc_lock_unlock (lock);
  72
  73   return result;
  74 }
  75 weak_alias (__register_printf_specifier, register_printf_specifier)
  76
  77
  78 /* Register FUNC to be called to format SPEC specifiers.  */
  79 int
  80 __register_printf_function (int spec, printf_function converter,
  81                             printf_arginfo_function arginfo)
  82 {
  83   return __register_printf_specifier (spec, converter,
  84                                       (printf_arginfo_size_function*) arginfo);
  85 }
  86 weak_alias (__register_printf_function, register_printf_function)