timezone/tst-tzset.c

   1 /* tzset tests with crafted time zone data.
   2    Copyright (C) 2015 Free Software Foundation, Inc.
   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 #define _GNU_SOURCE 1
  19
  20 #include <errno.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <sys/resource.h>
  25 #include <time.h>
  26 #include <unistd.h>
  27
  28 static int do_test (void);
  29 #define TEST_FUNCTION do_test ()
  30 #include "../test-skeleton.c"
  31
  32 /* Returns the name of a large TZ file.  */
  33 static char *
  34 create_tz_file (off64_t size)
  35 {
  36   char *path;
  37   int fd = create_temp_file ("tst-tzset-", &path);
  38   if (fd < 0)
  39     exit (1);
  40
  41   // Reopen for large-file support.
  42   close (fd);
  43   fd = open64 (path, O_WRONLY);
  44   if (fd < 0)
  45     {
  46       printf ("open64 (%s) failed: %m\n", path);
  47       exit (1);
  48     }
  49
  50   static const char data[] = {
  51     0x54, 0x5a, 0x69, 0x66, 0x32, 0x00, 0x00, 0x00,
  52     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  53     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
  54     0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
  55     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
  56     0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00,
  57     0x00, 0x00, 0x58, 0x54, 0x47, 0x00, 0x00, 0x00,
  58     0x54, 0x5a, 0x69, 0x66, 0x32, 0x00, 0x00, 0x00,
  59     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  60     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
  61     0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
  62     0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01,
  63     0x00, 0x00, 0x00, 0x04, 0xf8, 0x00, 0x00, 0x00,
  64     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  65     0x00, 0x00, 0x00, 0x58, 0x54, 0x47, 0x00, 0x00,
  66     0x00, 0x0a, 0x58, 0x54, 0x47, 0x30, 0x0a
  67   };
  68   ssize_t ret = write (fd, data, sizeof (data));
  69   if (ret < 0)
  70     {
  71       printf ("write failed: %m\n");
  72       exit (1);
  73     }
  74   if ((size_t) ret != sizeof (data))
  75     {
  76       printf ("Short write\n");
  77       exit (1);
  78     }
  79   if (lseek64 (fd, size, SEEK_CUR) < 0)
  80     {
  81       printf ("lseek failed: %m\n");
  82       close (fd);
  83       return NULL;
  84     }
  85   if (write (fd, "", 1) != 1)
  86     {
  87       printf ("Single-byte write failed\n");
  88       close (fd);
  89       return NULL;
  90     }
  91   if (close (fd) != 0)
  92     {
  93       printf ("close failed: %m\n");
  94       exit (1);
  95     }
  96   return path;
  97 }
  98
  99 static void
 100 test_tz_file (off64_t size)
 101 {
 102   char *path = create_tz_file (size);
 103   if (setenv ("TZ", path, 1) < 0)
 104     {
 105       printf ("setenv failed: %m\n");
 106       exit (1);
 107     }
 108   tzset ();
 109   free (path);
 110 }
 111
 112 static int
 113 do_test (void)
 114 {
 115   /* Limit the size of the process.  Otherwise, some of the tests will
 116      consume a lot of resources.  */
 117   {
 118     struct rlimit limit;
 119     if (getrlimit (RLIMIT_AS, &limit) != 0)
 120       {
 121         printf ("getrlimit (RLIMIT_AS) failed: %m\n");
 122         return 1;
 123       }
 124     long target = 512 * 1024 * 1024;
 125     if (limit.rlim_cur == RLIM_INFINITY || limit.rlim_cur > target)
 126       {
 127         limit.rlim_cur = 512 * 1024 * 1024;
 128         if (setrlimit (RLIMIT_AS, &limit) != 0)
 129           {
 130             printf ("setrlimit (RLIMIT_AS) failed: %m\n");
 131             return 1;
 132           }
 133       }
 134   }
 135
 136   int errors = 0;
 137   for (int i = 1; i <= 4; ++i)
 138     {
 139       char tz[16];
 140       snprintf (tz, sizeof (tz), "XT%d", i);
 141       if (setenv ("TZ", tz, 1) < 0)
 142         {
 143           printf ("setenv failed: %m\n");
 144           return 1;
 145         }
 146       tzset ();
 147       if (strcmp (tzname[0], tz) == 0)
 148         {
 149           printf ("Unexpected success for %s\n", tz);
 150           ++errors;
 151         }
 152     }
 153
 154   /* Large TZ files.  */
 155
 156   /* This will succeed on 64-bit architectures, and fail on 32-bit
 157      architectures.  It used to crash on 32-bit.  */
 158   test_tz_file (64 * 1024 * 1024);
 159
 160   /* This will fail on 64-bit and 32-bit architectures.  It used to
 161      cause a test timeout on 64-bit and crash on 32-bit if the TZ file
 162      open succeeded for some reason (it does not use O_LARGEFILE in
 163      regular builds).  */
 164   test_tz_file (4LL * 1024 * 1024 * 1024 - 6);
 165
 166   /* Large TZ variables.  */
 167   {
 168     size_t length = 64 * 1024 * 1024;
 169     char *value = malloc (length + 1);
 170     if (value == NULL)
 171       {
 172         puts ("malloc failed: %m");
 173         return 1;
 174       }
 175     value[length] = '\0';
 176
 177     memset (value, ' ', length);
 178     value[0] = 'U';
 179     value[1] = 'T';
 180     value[2] = 'C';
 181     if (setenv ("TZ", value, 1) < 0)
 182       {
 183         printf ("setenv failed: %m\n");
 184         return 1;
 185       }
 186     tzset ();
 187
 188     memset (value, '0', length);
 189     value[0] = '<';
 190     value[length - 1] = '>';
 191     if (setenv ("TZ", value, 1) < 0)
 192       {
 193         printf ("setenv failed: %m\n");
 194         return 1;
 195       }
 196     tzset ();
 197   }
 198
 199   return errors > 0;
 200 }