kernel/tools/testing/selftests/timers/leap-a-day.c

   1 /* Leap second stress test
   2  *              by: John Stultz (john.stultz@linaro.org)
   3  *              (C) Copyright IBM 2012
   4  *              (C) Copyright 2013, 2015 Linaro Limited
   5  *              Licensed under the GPLv2
   6  *
   7  *  This test signals the kernel to insert a leap second
   8  *  every day at midnight GMT. This allows for stessing the
   9  *  kernel's leap-second behavior, as well as how well applications
  10  *  handle the leap-second discontinuity.
  11  *
  12  *  Usage: leap-a-day [-s] [-i <num>]
  13  *
  14  *  Options:
  15  *      -s:     Each iteration, set the date to 10 seconds before midnight GMT.
  16  *              This speeds up the number of leapsecond transitions tested,
  17  *              but because it calls settimeofday frequently, advancing the
  18  *              time by 24 hours every ~16 seconds, it may cause application
  19  *              disruption.
  20  *
  21  *      -i:     Number of iterations to run (default: infinite)
  22  *
  23  *  Other notes: Disabling NTP prior to running this is advised, as the two
  24  *               may conflict in their commands to the kernel.
  25  *
  26  *  To build:
  27  *      $ gcc leap-a-day.c -o leap-a-day -lrt
  28  *
  29  *   This program is free software: you can redistribute it and/or modify
  30  *   it under the terms of the GNU General Public License as published by
  31  *   the Free Software Foundation, either version 2 of the License, or
  32  *   (at your option) any later version.
  33  *
  34  *   This program is distributed in the hope that it will be useful,
  35  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
  36  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  37  *   GNU General Public License for more details.
  38  */
  39
  40
  41
  42 #include <stdio.h>
  43 #include <stdlib.h>
  44 #include <time.h>
  45 #include <sys/time.h>
  46 #include <sys/timex.h>
  47 #include <sys/errno.h>
  48 #include <string.h>
  49 #include <signal.h>
  50 #include <unistd.h>
  51 #ifdef KTEST
  52 #include "../kselftest.h"
  53 #else
  54 static inline int ksft_exit_pass(void)
  55 {
  56         exit(0);
  57 }
  58 static inline int ksft_exit_fail(void)
  59 {
  60         exit(1);
  61 }
  62 #endif
  63
  64 #define NSEC_PER_SEC 1000000000ULL
  65 #define CLOCK_TAI 11
  66
  67 time_t next_leap;
  68 int error_found;
  69
  70 /* returns 1 if a <= b, 0 otherwise */
  71 static inline int in_order(struct timespec a, struct timespec b)
  72 {
  73         if (a.tv_sec < b.tv_sec)
  74                 return 1;
  75         if (a.tv_sec > b.tv_sec)
  76                 return 0;
  77         if (a.tv_nsec > b.tv_nsec)
  78                 return 0;
  79         return 1;
  80 }
  81
  82 struct timespec timespec_add(struct timespec ts, unsigned long long ns)
  83 {
  84         ts.tv_nsec += ns;
  85         while (ts.tv_nsec >= NSEC_PER_SEC) {
  86                 ts.tv_nsec -= NSEC_PER_SEC;
  87                 ts.tv_sec++;
  88         }
  89         return ts;
  90 }
  91
  92 char *time_state_str(int state)
  93 {
  94         switch (state) {
  95         case TIME_OK:   return "TIME_OK";
  96         case TIME_INS:  return "TIME_INS";
  97         case TIME_DEL:  return "TIME_DEL";
  98         case TIME_OOP:  return "TIME_OOP";
  99         case TIME_WAIT: return "TIME_WAIT";
 100         case TIME_BAD:  return "TIME_BAD";
 101         }
 102         return "ERROR";
 103 }
 104
 105 /* clear NTP time_status & time_state */
 106 int clear_time_state(void)
 107 {
 108         struct timex tx;
 109         int ret;
 110
 111         /*
 112          * We have to call adjtime twice here, as kernels
 113          * prior to 6b1859dba01c7 (included in 3.5 and
 114          * -stable), had an issue with the state machine
 115          * and wouldn't clear the STA_INS/DEL flag directly.
 116          */
 117         tx.modes = ADJ_STATUS;
 118         tx.status = STA_PLL;
 119         ret = adjtimex(&tx);
 120
 121         /* Clear maxerror, as it can cause UNSYNC to be set */
 122         tx.modes = ADJ_MAXERROR;
 123         tx.maxerror = 0;
 124         ret = adjtimex(&tx);
 125
 126         /* Clear the status */
 127         tx.modes = ADJ_STATUS;
 128         tx.status = 0;
 129         ret = adjtimex(&tx);
 130
 131         return ret;
 132 }
 133
 134 /* Make sure we cleanup on ctrl-c */
 135 void handler(int unused)
 136 {
 137         clear_time_state();
 138         exit(0);
 139 }
 140
 141 void sigalarm(int signo)
 142 {
 143         struct timex tx;
 144         int ret;
 145
 146         tx.modes = 0;
 147         ret = adjtimex(&tx);
 148
 149         if (tx.time.tv_sec < next_leap) {
 150                 printf("Error: Early timer expiration! (Should be %ld)\n", next_leap);
 151                 error_found = 1;
 152                 printf("adjtimex: %10ld sec + %6ld us (%i)\t%s\n",
 153                                         tx.time.tv_sec,
 154                                         tx.time.tv_usec,
 155                                         tx.tai,
 156                                         time_state_str(ret));
 157         }
 158         if (ret != TIME_WAIT) {
 159                 printf("Error: Timer seeing incorrect NTP state? (Should be TIME_WAIT)\n");
 160                 error_found = 1;
 161                 printf("adjtimex: %10ld sec + %6ld us (%i)\t%s\n",
 162                                         tx.time.tv_sec,
 163                                         tx.time.tv_usec,
 164                                         tx.tai,
 165                                         time_state_str(ret));
 166         }
 167 }
 168
 169
 170 /* Test for known hrtimer failure */
 171 void test_hrtimer_failure(void)
 172 {
 173         struct timespec now, target;
 174
 175         clock_gettime(CLOCK_REALTIME, &now);
 176         target = timespec_add(now, NSEC_PER_SEC/2);
 177         clock_nanosleep(CLOCK_REALTIME, TIMER_ABSTIME, &target, NULL);
 178         clock_gettime(CLOCK_REALTIME, &now);
 179
 180         if (!in_order(target, now)) {
 181                 printf("ERROR: hrtimer early expiration failure observed.\n");
 182                 error_found = 1;
 183         }
 184 }
 185
 186 int main(int argc, char **argv)
 187 {
 188         timer_t tm1;
 189         struct itimerspec its1;
 190         struct sigevent se;
 191         struct sigaction act;
 192         int signum = SIGRTMAX;
 193         int settime = 0;
 194         int tai_time = 0;
 195         int insert = 1;
 196         int iterations = -1;
 197         int opt;
 198
 199         /* Process arguments */
 200         while ((opt = getopt(argc, argv, "sti:")) != -1) {
 201                 switch (opt) {
 202                 case 's':
 203                         printf("Setting time to speed up testing\n");
 204                         settime = 1;
 205                         break;
 206                 case 'i':
 207                         iterations = atoi(optarg);
 208                         break;
 209                 case 't':
 210                         tai_time = 1;
 211                         break;
 212                 default:
 213                         printf("Usage: %s [-s] [-i <iterations>]\n", argv[0]);
 214                         printf("        -s: Set time to right before leap second each iteration\n");
 215                         printf("        -i: Number of iterations\n");
 216                         printf("        -t: Print TAI time\n");
 217                         exit(-1);
 218                 }
 219         }
 220
 221         /* Make sure TAI support is present if -t was used */
 222         if (tai_time) {
 223                 struct timespec ts;
 224
 225                 if (clock_gettime(CLOCK_TAI, &ts)) {
 226                         printf("System doesn't support CLOCK_TAI\n");
 227                         ksft_exit_fail();
 228                 }
 229         }
 230
 231         signal(SIGINT, handler);
 232         signal(SIGKILL, handler);
 233
 234         /* Set up timer signal handler: */
 235         sigfillset(&act.sa_mask);
 236         act.sa_flags = 0;
 237         act.sa_handler = sigalarm;
 238         sigaction(signum, &act, NULL);
 239
 240         if (iterations < 0)
 241                 printf("This runs continuously. Press ctrl-c to stop\n");
 242         else
 243                 printf("Running for %i iterations. Press ctrl-c to stop\n", iterations);
 244
 245         printf("\n");
 246         while (1) {
 247                 int ret;
 248                 struct timespec ts;
 249                 struct timex tx;
 250                 time_t now;
 251
 252                 /* Get the current time */
 253                 clock_gettime(CLOCK_REALTIME, &ts);
 254
 255                 /* Calculate the next possible leap second 23:59:60 GMT */
 256                 next_leap = ts.tv_sec;
 257                 next_leap += 86400 - (next_leap % 86400);
 258
 259                 if (settime) {
 260                         struct timeval tv;
 261
 262                         tv.tv_sec = next_leap - 10;
 263                         tv.tv_usec = 0;
 264                         settimeofday(&tv, NULL);
 265                         printf("Setting time to %s", ctime(&tv.tv_sec));
 266                 }
 267
 268                 /* Reset NTP time state */
 269                 clear_time_state();
 270
 271                 /* Set the leap second insert flag */
 272                 tx.modes = ADJ_STATUS;
 273                 if (insert)
 274                         tx.status = STA_INS;
 275                 else
 276                         tx.status = STA_DEL;
 277                 ret = adjtimex(&tx);
 278                 if (ret < 0) {
 279                         printf("Error: Problem setting STA_INS/STA_DEL!: %s\n",
 280                                                         time_state_str(ret));
 281                         return ksft_exit_fail();
 282                 }
 283
 284                 /* Validate STA_INS was set */
 285                 tx.modes = 0;
 286                 ret = adjtimex(&tx);
 287                 if (tx.status != STA_INS && tx.status != STA_DEL) {
 288                         printf("Error: STA_INS/STA_DEL not set!: %s\n",
 289                                                         time_state_str(ret));
 290                         return ksft_exit_fail();
 291                 }
 292
 293                 if (tai_time) {
 294                         printf("Using TAI time,"
 295                                 " no inconsistencies should be seen!\n");
 296                 }
 297
 298                 printf("Scheduling leap second for %s", ctime(&next_leap));
 299
 300                 /* Set up timer */
 301                 printf("Setting timer for %ld -  %s", next_leap, ctime(&next_leap));
 302                 memset(&se, 0, sizeof(se));
 303                 se.sigev_notify = SIGEV_SIGNAL;
 304                 se.sigev_signo = signum;
 305                 se.sigev_value.sival_int = 0;
 306                 if (timer_create(CLOCK_REALTIME, &se, &tm1) == -1) {
 307                         printf("Error: timer_create failed\n");
 308                         return ksft_exit_fail();
 309                 }
 310                 its1.it_value.tv_sec = next_leap;
 311                 its1.it_value.tv_nsec = 0;
 312                 its1.it_interval.tv_sec = 0;
 313                 its1.it_interval.tv_nsec = 0;
 314                 timer_settime(tm1, TIMER_ABSTIME, &its1, NULL);
 315
 316                 /* Wake up 3 seconds before leap */
 317                 ts.tv_sec = next_leap - 3;
 318                 ts.tv_nsec = 0;
 319
 320
 321                 while (clock_nanosleep(CLOCK_REALTIME, TIMER_ABSTIME, &ts, NULL))
 322                         printf("Something woke us up, returning to sleep\n");
 323
 324                 /* Validate STA_INS is still set */
 325                 tx.modes = 0;
 326                 ret = adjtimex(&tx);
 327                 if (tx.status != STA_INS && tx.status != STA_DEL) {
 328                         printf("Something cleared STA_INS/STA_DEL, setting it again.\n");
 329                         tx.modes = ADJ_STATUS;
 330                         if (insert)
 331                                 tx.status = STA_INS;
 332                         else
 333                                 tx.status = STA_DEL;
 334                         ret = adjtimex(&tx);
 335                 }
 336
 337                 /* Check adjtimex output every half second */
 338                 now = tx.time.tv_sec;
 339                 while (now < next_leap + 2) {
 340                         char buf[26];
 341                         struct timespec tai;
 342                         int ret;
 343
 344                         tx.modes = 0;
 345                         ret = adjtimex(&tx);
 346
 347                         if (tai_time) {
 348                                 clock_gettime(CLOCK_TAI, &tai);
 349                                 printf("%ld sec, %9ld ns\t%s\n",
 350                                                 tai.tv_sec,
 351                                                 tai.tv_nsec,
 352                                                 time_state_str(ret));
 353                         } else {
 354                                 ctime_r(&tx.time.tv_sec, buf);
 355                                 buf[strlen(buf)-1] = 0; /*remove trailing\n */
 356
 357                                 printf("%s + %6ld us (%i)\t%s\n",
 358                                                 buf,
 359                                                 tx.time.tv_usec,
 360                                                 tx.tai,
 361                                                 time_state_str(ret));
 362                         }
 363                         now = tx.time.tv_sec;
 364                         /* Sleep for another half second */
 365                         ts.tv_sec = 0;
 366                         ts.tv_nsec = NSEC_PER_SEC / 2;
 367                         clock_nanosleep(CLOCK_MONOTONIC, 0, &ts, NULL);
 368                 }
 369                 /* Switch to using other mode */
 370                 insert = !insert;
 371
 372                 /* Note if kernel has known hrtimer failure */
 373                 test_hrtimer_failure();
 374
 375                 printf("Leap complete\n");
 376                 if (error_found) {
 377                         printf("Errors observed\n");
 378                         clear_time_state();
 379                         return ksft_exit_fail();
 380                 }
 381                 printf("\n");
 382                 if ((iterations != -1) && !(--iterations))
 383                         break;
 384         }
 385
 386         clear_time_state();
 387         return ksft_exit_pass();
 388 }