glibc/timezone/tst-tzset.c

202 lines
5.0 KiB
C

/* tzset tests with crafted time zone data.
Copyright (C) 2015-2018 Free Software Foundation, Inc.
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/>. */
#define _GNU_SOURCE 1
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/resource.h>
#include <time.h>
#include <unistd.h>
#define TIMEOUT 5
static int do_test (void);
#define TEST_FUNCTION do_test ()
#include "../test-skeleton.c"
/* Returns the name of a large TZ file. */
static char *
create_tz_file (off64_t size)
{
char *path;
int fd = create_temp_file ("tst-tzset-", &path);
if (fd < 0)
exit (1);
// Reopen for large-file support.
close (fd);
fd = open64 (path, O_WRONLY);
if (fd < 0)
{
printf ("open64 (%s) failed: %m\n", path);
exit (1);
}
static const char data[] = {
0x54, 0x5a, 0x69, 0x66, 0x32, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x58, 0x54, 0x47, 0x00, 0x00, 0x00,
0x54, 0x5a, 0x69, 0x66, 0x32, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01,
0x00, 0x00, 0x00, 0x04, 0xf8, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x58, 0x54, 0x47, 0x00, 0x00,
0x00, 0x0a, 0x58, 0x54, 0x47, 0x30, 0x0a
};
ssize_t ret = write (fd, data, sizeof (data));
if (ret < 0)
{
printf ("write failed: %m\n");
exit (1);
}
if ((size_t) ret != sizeof (data))
{
printf ("Short write\n");
exit (1);
}
if (lseek64 (fd, size, SEEK_CUR) < 0)
{
printf ("lseek failed: %m\n");
close (fd);
return NULL;
}
if (write (fd, "", 1) != 1)
{
printf ("Single-byte write failed\n");
close (fd);
return NULL;
}
if (close (fd) != 0)
{
printf ("close failed: %m\n");
exit (1);
}
return path;
}
static void
test_tz_file (off64_t size)
{
char *path = create_tz_file (size);
if (setenv ("TZ", path, 1) < 0)
{
printf ("setenv failed: %m\n");
exit (1);
}
tzset ();
free (path);
}
static int
do_test (void)
{
/* Limit the size of the process. Otherwise, some of the tests will
consume a lot of resources. */
{
struct rlimit limit;
if (getrlimit (RLIMIT_AS, &limit) != 0)
{
printf ("getrlimit (RLIMIT_AS) failed: %m\n");
return 1;
}
long target = 512 * 1024 * 1024;
if (limit.rlim_cur == RLIM_INFINITY || limit.rlim_cur > target)
{
limit.rlim_cur = 512 * 1024 * 1024;
if (setrlimit (RLIMIT_AS, &limit) != 0)
{
printf ("setrlimit (RLIMIT_AS) failed: %m\n");
return 1;
}
}
}
int errors = 0;
for (int i = 1; i <= 4; ++i)
{
char tz[16];
snprintf (tz, sizeof (tz), "XT%d", i);
if (setenv ("TZ", tz, 1) < 0)
{
printf ("setenv failed: %m\n");
return 1;
}
tzset ();
if (strcmp (tzname[0], tz) == 0)
{
printf ("Unexpected success for %s\n", tz);
++errors;
}
}
/* Large TZ files. */
/* This will succeed on 64-bit architectures, and fail on 32-bit
architectures. It used to crash on 32-bit. */
test_tz_file (64 * 1024 * 1024);
/* This will fail on 64-bit and 32-bit architectures. It used to
cause a test timeout on 64-bit and crash on 32-bit if the TZ file
open succeeded for some reason (it does not use O_LARGEFILE in
regular builds). */
test_tz_file (4LL * 1024 * 1024 * 1024 - 6);
/* Large TZ variables. */
{
size_t length = 64 * 1024 * 1024;
char *value = malloc (length + 1);
if (value == NULL)
{
puts ("malloc failed: %m");
return 1;
}
value[length] = '\0';
memset (value, ' ', length);
value[0] = 'U';
value[1] = 'T';
value[2] = 'C';
if (setenv ("TZ", value, 1) < 0)
{
printf ("setenv failed: %m\n");
return 1;
}
tzset ();
memset (value, '0', length);
value[0] = '<';
value[length - 1] = '>';
if (setenv ("TZ", value, 1) < 0)
{
printf ("setenv failed: %m\n");
return 1;
}
tzset ();
}
return errors > 0;
}