glibc/stdio-common/reg-modifier.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 <bits/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);
}
}