stdio-common/reg-printf.c

   1 /* Copyright (C) 1991-2015 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 (spec, converter, arginfo)
  42      int spec;
  43      printf_function converter;
  44      printf_arginfo_size_function arginfo;
  45 {
  46   if (spec < 0 || spec > (int) UCHAR_MAX)
  47     {
  48       __set_errno (EINVAL);
  49       return -1;
  50     }
  51
  52   int result = 0;
  53   __libc_lock_lock (lock);
  54
  55   if (__printf_function_table == NULL)
  56     {
  57       __printf_arginfo_table = (printf_arginfo_size_function **)
  58         calloc (UCHAR_MAX + 1, sizeof (void *) * 2);
  59       if (__printf_arginfo_table == NULL)
  60         {
  61           result = -1;
  62           goto out;
  63         }
  64
  65       __printf_function_table = (printf_function **)
  66         (__printf_arginfo_table + UCHAR_MAX + 1);
  67     }
  68
  69   __printf_function_table[spec] = converter;
  70   __printf_arginfo_table[spec] = arginfo;
  71
  72  out:
  73   __libc_lock_unlock (lock);
  74
  75   return result;
  76 }
  77 weak_alias (__register_printf_specifier, register_printf_specifier)
  78
  79
  80 /* Register FUNC to be called to format SPEC specifiers.  */
  81 int
  82 __register_printf_function (spec, converter, arginfo)
  83      int spec;
  84      printf_function converter;
  85      printf_arginfo_function arginfo;
  86 {
  87   return __register_printf_specifier (spec, converter,
  88                                       (printf_arginfo_size_function*) arginfo);
  89 }
  90 weak_alias (__register_printf_function, register_printf_function)