1 /** @file
2 * @brief Measure CPU time.
3 */
4 /* Copyright (C) 2009,2015,2018,2020 Olly Betts
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as
8 * published by the Free Software Foundation; either version 2 of the
9 * License, or (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <config.h>
22
23 #include "cputimer.h"
24
25 #include "errno_to_string.h"
26 #include "testsuite.h"
27
28 #ifdef HAVE_GETRUSAGE
29 # include <sys/time.h>
30 # include <sys/resource.h>
31 #elif defined HAVE_TIMES
32 # include <sys/times.h>
33 # ifdef HAVE_SYSCONF
34 # include "safeunistd.h"
35 # endif
36 #else
37 # include "realtime.h"
38 #endif
39
40 #include <cerrno>
41 #include <cstdlib>
42 #include <cstring>
43 #include <string>
44
45 using namespace std;
46
47 double
get_current_cputime() const48 CPUTimer::get_current_cputime() const
49 {
50 #ifdef XAPIAN_DEBUG_LOG
51 SKIP_TEST("Skipping timed test because configured with --enable-log");
52 #else
53 static bool skip = (getenv("AUTOMATED_TESTING") != NULL);
54 if (skip) {
55 SKIP_TEST("Skipping timed test because $AUTOMATED_TESTING is set");
56 }
57
58 double t = 0;
59 #ifdef HAVE_GETRUSAGE
60 struct rusage r;
61 if (getrusage(RUSAGE_SELF, &r) == -1) {
62 FAIL_TEST("Couldn't measure CPU for self: " << errno_to_string(errno));
63 }
64
65 t = r.ru_utime.tv_sec + r.ru_stime.tv_sec;
66 t += (r.ru_utime.tv_usec + r.ru_stime.tv_usec) * 0.000001;
67 #elif defined HAVE_TIMES
68 struct tms b;
69 if (times(&b) == clock_t(-1)) {
70 FAIL_TEST("Couldn't measure CPU: " << errno_to_string(errno));
71 }
72 t = (double)(b.tms_utime + b.tms_stime);
73 # ifdef HAVE_SYSCONF
74 t /= sysconf(_SC_CLK_TCK);
75 # else
76 t /= CLK_TCK;
77 # endif
78 #else
79 // FIXME: Fallback to just using wallclock time, which is probably only
80 // going to be used on Microsoft Windows, where nobody has implemented
81 // the code required to get the CPU time used by a process.
82 t = RealTime::now();
83 #endif
84
85 return t;
86 #endif
87 }
88