203 lines
4.2 KiB
C
203 lines
4.2 KiB
C
/* Copyright (C) 2009-2015 Free Software Foundation, Inc.
|
|
This file is part of the GNU C Library.
|
|
|
|
The GNU C Library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 of the License, or (at your option) any later version.
|
|
|
|
The GNU C Library 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
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with the GNU C Library; if not, see
|
|
<http://www.gnu.org/licenses/>. */
|
|
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <printf.h>
|
|
#include <stdlib.h>
|
|
#include <wchar.h>
|
|
#include <libc-lock.h>
|
|
|
|
|
|
struct printf_modifier_record
|
|
{
|
|
struct printf_modifier_record *next;
|
|
int bit;
|
|
wchar_t str[0];
|
|
};
|
|
|
|
struct printf_modifier_record **__printf_modifier_table attribute_hidden;
|
|
|
|
__libc_lock_define_initialized (static, lock)
|
|
|
|
/* Bits to hand out. */
|
|
static int next_bit;
|
|
|
|
|
|
int
|
|
__register_printf_modifier (const wchar_t *str)
|
|
{
|
|
if (str[0] == L'\0')
|
|
{
|
|
einval:
|
|
__set_errno (EINVAL);
|
|
return -1;
|
|
}
|
|
|
|
const wchar_t *wc = str;
|
|
while (*wc != L'\0')
|
|
if (*wc < 0 || *wc > (wchar_t) UCHAR_MAX)
|
|
goto einval;
|
|
else
|
|
++wc;
|
|
|
|
if (next_bit / 8 == sizeof (((struct printf_info *) NULL)->user))
|
|
{
|
|
__set_errno (ENOSPC);
|
|
return -1;
|
|
}
|
|
|
|
int result = -1;
|
|
__libc_lock_lock (lock);
|
|
|
|
if (__printf_modifier_table == NULL)
|
|
{
|
|
__printf_modifier_table = calloc (UCHAR_MAX,
|
|
sizeof (*__printf_modifier_table));
|
|
if (__printf_modifier_table == NULL)
|
|
goto out;
|
|
}
|
|
|
|
/* Create enough room for the string. But we don't need the first
|
|
character. */
|
|
struct printf_modifier_record *newp = malloc (sizeof (*newp)
|
|
+ ((wc - str)
|
|
* sizeof (wchar_t)));
|
|
if (newp == NULL)
|
|
goto out;
|
|
|
|
newp->next = __printf_modifier_table[(unsigned char) *str];
|
|
newp->bit = 1 << next_bit++;
|
|
__wmemcpy (newp->str, str + 1, wc - str);
|
|
|
|
__printf_modifier_table[(unsigned char) *str] = newp;
|
|
|
|
result = newp->bit;
|
|
|
|
out:
|
|
__libc_lock_unlock (lock);
|
|
|
|
return result;
|
|
}
|
|
weak_alias (__register_printf_modifier, register_printf_modifier)
|
|
|
|
|
|
#include <stdio.h>
|
|
int
|
|
attribute_hidden
|
|
__handle_registered_modifier_mb (const unsigned char **format,
|
|
struct printf_info *info)
|
|
{
|
|
struct printf_modifier_record *runp = __printf_modifier_table[**format];
|
|
|
|
int best_bit = 0;
|
|
int best_len = 0;
|
|
const unsigned char *best_cp = NULL;
|
|
|
|
while (runp != NULL)
|
|
{
|
|
const unsigned char *cp = *format + 1;
|
|
wchar_t *fcp = runp->str;
|
|
|
|
while (*cp != '\0' && *fcp != L'\0')
|
|
if (*cp != *fcp)
|
|
break;
|
|
else
|
|
++cp, ++fcp;
|
|
|
|
if (*fcp == L'\0' && cp - *format > best_len)
|
|
{
|
|
best_cp = cp;
|
|
best_len = cp - *format;
|
|
best_bit = runp->bit;
|
|
}
|
|
|
|
runp = runp->next;
|
|
}
|
|
|
|
if (best_bit != 0)
|
|
{
|
|
info->user |= best_bit;
|
|
*format = best_cp;
|
|
return 0;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
int
|
|
attribute_hidden
|
|
__handle_registered_modifier_wc (const unsigned int **format,
|
|
struct printf_info *info)
|
|
{
|
|
struct printf_modifier_record *runp = __printf_modifier_table[**format];
|
|
|
|
int best_bit = 0;
|
|
int best_len = 0;
|
|
const unsigned int *best_cp = NULL;
|
|
|
|
while (runp != NULL)
|
|
{
|
|
const unsigned int *cp = *format + 1;
|
|
wchar_t *fcp = runp->str;
|
|
|
|
while (*cp != '\0' && *fcp != L'\0')
|
|
if (*cp != *fcp)
|
|
break;
|
|
else
|
|
++cp, ++fcp;
|
|
|
|
if (*fcp == L'\0' && cp - *format > best_len)
|
|
{
|
|
best_cp = cp;
|
|
best_len = cp - *format;
|
|
best_bit = runp->bit;
|
|
}
|
|
|
|
runp = runp->next;
|
|
}
|
|
|
|
if (best_bit != 0)
|
|
{
|
|
info->user |= best_bit;
|
|
*format = best_cp;
|
|
return 0;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
libc_freeres_fn (free_mem)
|
|
{
|
|
if (__printf_modifier_table != NULL)
|
|
{
|
|
for (int i = 0; i < UCHAR_MAX; ++i)
|
|
{
|
|
struct printf_modifier_record *runp = __printf_modifier_table[i];
|
|
while (runp != NULL)
|
|
{
|
|
struct printf_modifier_record *oldp = runp;
|
|
runp = runp->next;
|
|
free (oldp);
|
|
}
|
|
}
|
|
free (__printf_modifier_table);
|
|
}
|
|
}
|