f0552f88135fb0d4eac17235c84d0d4e113afb26
[sliver-openvswitch.git] / tests / test-timeval.c
1 /*
2  * Copyright (c) 2009, 2010 Nicira Networks.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at:
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include <config.h>
18
19 #include "timeval.h"
20
21 #include <errno.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <sys/time.h>
25 #include <sys/types.h>
26 #include <unistd.h>
27
28 #include "command-line.h"
29 #include "daemon.h"
30 #include "util.h"
31
32 #undef NDEBUG
33 #include <assert.h>
34
35 static long long int
36 gettimeofday_in_msec(void)
37 {
38     struct timeval tv;
39
40     assert(!gettimeofday(&tv, NULL));
41     return timeval_to_msec(&tv);
42 }
43
44 static void
45 do_test(void)
46 {
47     /* Wait until we are awakened by a signal (typically EINTR due to the
48      * setitimer()).  Then ensure that, if time has really advanced by
49      * TIME_UPDATE_INTERVAL, then time_msec() reports that it advanced.
50      */
51     long long int start_time_msec, start_time_wall;
52     long long int start_gtod;
53
54     start_time_msec = time_msec();
55     start_time_wall = time_wall_msec();
56     start_gtod = gettimeofday_in_msec();
57     for (;;) {
58         /* Wait up to 1 second.  Using select() to do the timeout avoids
59          * interfering with the interval timer. */
60         struct timeval timeout;
61         int retval;
62
63         timeout.tv_sec = 1;
64         timeout.tv_usec = 0;
65         retval = select(0, NULL, NULL, NULL, &timeout);
66         if (retval != -1) {
67             ovs_fatal(0, "select returned %d", retval);
68         } else if (errno != EINTR) {
69             ovs_fatal(errno, "select reported unexpected error");
70         }
71
72         if (gettimeofday_in_msec() - start_gtod >= TIME_UPDATE_INTERVAL) {
73             assert(time_msec() - start_time_msec >= TIME_UPDATE_INTERVAL);
74             assert(time_wall_msec() - start_time_wall >= TIME_UPDATE_INTERVAL);
75             break;
76         }
77     }
78 }
79
80 static void
81 usage(void)
82 {
83     ovs_fatal(0, "usage: %s TEST, where TEST is \"plain\" or \"daemon\"",
84               program_name);
85 }
86
87 int
88 main(int argc, char *argv[])
89 {
90     proctitle_init(argc, argv);
91     set_program_name(argv[0]);
92     time_init();
93
94     if (argc != 2) {
95         usage();
96     } else if (!strcmp(argv[1], "plain")) {
97         do_test();
98     } else if (!strcmp(argv[1], "daemon")) {
99         /* Test that time still advances even in a daemon.  This is an
100          * interesting test because fork() cancels the interval timer. */
101         char cwd[1024], *pidfile;
102         FILE *success;
103
104         assert(getcwd(cwd, sizeof cwd) == cwd);
105
106         unlink("test-timeval.success");
107
108         /* Daemonize, with a pidfile in the current directory. */
109         set_detach();
110         pidfile = xasprintf("%s/test-timeval.pid", cwd);
111         set_pidfile(pidfile);
112         free(pidfile);
113         set_no_chdir();
114         daemonize();
115
116         /* Run the test. */
117         do_test();
118
119         /* Report success by writing out a file, since the ultimate invoker of
120          * test-timeval can't wait on the daemonized process. */
121         success = fopen("test-timeval.success", "w");
122         if (!success) {
123             ovs_fatal(errno, "test-timeval.success: create failed");
124         }
125         fprintf(success, "success\n");
126         fclose(success);
127     } else {
128         usage();
129     }
130
131     return 0;
132 }