blob: a3319a289cbc93ce3c1ee244b9bd0bf7a5d09c3a [file] [log] [blame]
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001/* Time module */
2
Barry Warsaw9a2a8a81996-12-06 23:32:14 +00003#include "Python.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +00004
Guido van Rossum87ce7bb1998-06-09 16:30:31 +00005#include <ctype.h>
6
Victor Stinnerec895392012-04-29 02:41:27 +02007#ifdef HAVE_SYS_TIMES_H
8#include <sys/times.h>
9#endif
10
Thomas Wouters0e3f5912006-08-11 14:57:12 +000011#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000012#include <sys/types.h>
Victor Stinnerec895392012-04-29 02:41:27 +020013#endif
14
15#if defined(HAVE_SYS_RESOURCE_H)
16#include <sys/resource.h>
17#endif
Guido van Rossum6d946f91992-08-14 13:49:30 +000018
Guido van Rossumb6775db1994-08-01 11:34:53 +000019#ifdef QUICKWIN
20#include <io.h>
21#endif
22
Guido van Rossum7bf22de1997-12-02 20:34:19 +000023#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossumbceeac81996-05-23 22:53:47 +000024#include <i86.h>
25#else
Guido van Rossumcac6c721996-09-06 13:34:02 +000026#ifdef MS_WINDOWS
Mark Hammond975e3922002-07-16 01:29:19 +000027#define WIN32_LEAN_AND_MEAN
Guido van Rossum258ccd42001-03-02 06:53:29 +000028#include <windows.h>
Mark Hammond975e3922002-07-16 01:29:19 +000029#include "pythread.h"
Guido van Rossumcac6c721996-09-06 13:34:02 +000030#endif /* MS_WINDOWS */
Guido van Rossum7bf22de1997-12-02 20:34:19 +000031#endif /* !__WATCOMC__ || __QNX__ */
Guido van Rossum234f9421993-06-17 12:35:49 +000032
33/* Forward declarations */
Victor Stinnercb29f012015-03-27 13:31:18 +010034static int pysleep(_PyTime_t);
Victor Stinnerec895392012-04-29 02:41:27 +020035static PyObject* floattime(_Py_clock_info_t *info);
36
Barry Warsaw9a2a8a81996-12-06 23:32:14 +000037static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +010038time_time(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000039{
Victor Stinnerec895392012-04-29 02:41:27 +020040 return floattime(NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +000041}
42
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000043PyDoc_STRVAR(time_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +010044"time() -> floating point number\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +000045\n\
46Return the current time in seconds since the Epoch.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000047Fractions of a second may be present if the system clock provides them.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +000048
Victor Stinner85fdfa82012-01-27 00:38:48 +010049#if defined(HAVE_CLOCK)
50
51#ifndef CLOCKS_PER_SEC
52#ifdef CLK_TCK
53#define CLOCKS_PER_SEC CLK_TCK
54#else
55#define CLOCKS_PER_SEC 1000000
56#endif
57#endif
58
Victor Stinner4195b5c2012-02-08 23:03:19 +010059static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +020060floatclock(_Py_clock_info_t *info)
Victor Stinner85fdfa82012-01-27 00:38:48 +010061{
Victor Stinner4195b5c2012-02-08 23:03:19 +010062 clock_t value;
63 value = clock();
64 if (value == (clock_t)-1) {
Victor Stinner85fdfa82012-01-27 00:38:48 +010065 PyErr_SetString(PyExc_RuntimeError,
66 "the processor time used is not available "
67 "or its value cannot be represented");
Victor Stinner4195b5c2012-02-08 23:03:19 +010068 return NULL;
Victor Stinner85fdfa82012-01-27 00:38:48 +010069 }
Victor Stinnerec895392012-04-29 02:41:27 +020070 if (info) {
71 info->implementation = "clock()";
72 info->resolution = 1.0 / (double)CLOCKS_PER_SEC;
Benjamin Peterson49a69e42012-05-01 09:38:34 -040073 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +020074 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +020075 }
Victor Stinner4195b5c2012-02-08 23:03:19 +010076 return PyFloat_FromDouble((double)value / CLOCKS_PER_SEC);
Victor Stinner85fdfa82012-01-27 00:38:48 +010077}
78#endif /* HAVE_CLOCK */
79
Victor Stinnerf427a142014-10-22 12:33:23 +020080#ifdef MS_WINDOWS
Victor Stinnerec895392012-04-29 02:41:27 +020081#define WIN32_PERF_COUNTER
Victor Stinner9122fdd2011-07-04 13:55:40 +020082/* Win32 has better clock replacement; we have our own version, due to Mark
83 Hammond and Tim Peters */
Victor Stinner54884492014-08-29 16:51:33 +020084static PyObject*
85win_perf_counter(_Py_clock_info_t *info)
Guido van Rossum3917c221997-04-02 05:35:28 +000086{
Victor Stinner8b302012012-02-07 23:29:46 +010087 static LONGLONG cpu_frequency = 0;
Victor Stinner4195b5c2012-02-08 23:03:19 +010088 static LONGLONG ctrStart;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000089 LARGE_INTEGER now;
Victor Stinner4195b5c2012-02-08 23:03:19 +010090 double diff;
Guido van Rossum3917c221997-04-02 05:35:28 +000091
Victor Stinner8b302012012-02-07 23:29:46 +010092 if (cpu_frequency == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000093 LARGE_INTEGER freq;
Victor Stinner8b302012012-02-07 23:29:46 +010094 QueryPerformanceCounter(&now);
Victor Stinner4195b5c2012-02-08 23:03:19 +010095 ctrStart = now.QuadPart;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000096 if (!QueryPerformanceFrequency(&freq) || freq.QuadPart == 0) {
Victor Stinner54884492014-08-29 16:51:33 +020097 PyErr_SetFromWindowsErr(0);
98 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000099 }
Victor Stinner8b302012012-02-07 23:29:46 +0100100 cpu_frequency = freq.QuadPart;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000101 }
102 QueryPerformanceCounter(&now);
Victor Stinner4195b5c2012-02-08 23:03:19 +0100103 diff = (double)(now.QuadPart - ctrStart);
Victor Stinnerec895392012-04-29 02:41:27 +0200104 if (info) {
105 info->implementation = "QueryPerformanceCounter()";
106 info->resolution = 1.0 / (double)cpu_frequency;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400107 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200108 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200109 }
Victor Stinner54884492014-08-29 16:51:33 +0200110 return PyFloat_FromDouble(diff / (double)cpu_frequency);
Guido van Rossum3917c221997-04-02 05:35:28 +0000111}
Victor Stinnerf427a142014-10-22 12:33:23 +0200112#endif /* MS_WINDOWS */
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000113
Victor Stinnerec895392012-04-29 02:41:27 +0200114#if defined(WIN32_PERF_COUNTER) || defined(HAVE_CLOCK)
115#define PYCLOCK
116static PyObject*
117pyclock(_Py_clock_info_t *info)
118{
119#ifdef WIN32_PERF_COUNTER
Victor Stinner54884492014-08-29 16:51:33 +0200120 return win_perf_counter(info);
121#else
Victor Stinnerec895392012-04-29 02:41:27 +0200122 return floatclock(info);
Victor Stinner54884492014-08-29 16:51:33 +0200123#endif
Victor Stinnerec895392012-04-29 02:41:27 +0200124}
125
Victor Stinner9122fdd2011-07-04 13:55:40 +0200126static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100127time_clock(PyObject *self, PyObject *unused)
Victor Stinner9122fdd2011-07-04 13:55:40 +0200128{
Victor Stinnerec895392012-04-29 02:41:27 +0200129 return pyclock(NULL);
Victor Stinner9122fdd2011-07-04 13:55:40 +0200130}
Victor Stinner9122fdd2011-07-04 13:55:40 +0200131
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000132PyDoc_STRVAR(clock_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +0100133"clock() -> floating point number\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000134\n\
135Return the CPU time or real time since the start of the process or since\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000136the first call to clock(). This has as much precision as the system\n\
137records.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000138#endif
139
Victor Stinnere0be4232011-10-25 13:06:09 +0200140#ifdef HAVE_CLOCK_GETTIME
141static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100142time_clock_gettime(PyObject *self, PyObject *args)
Victor Stinnere0be4232011-10-25 13:06:09 +0200143{
144 int ret;
Victor Stinnerb8d01692012-04-13 23:44:05 +0200145 int clk_id;
Victor Stinnere0be4232011-10-25 13:06:09 +0200146 struct timespec tp;
147
Victor Stinner4195b5c2012-02-08 23:03:19 +0100148 if (!PyArg_ParseTuple(args, "i:clock_gettime", &clk_id))
Victor Stinnere0be4232011-10-25 13:06:09 +0200149 return NULL;
150
151 ret = clock_gettime((clockid_t)clk_id, &tp);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100152 if (ret != 0) {
Victor Stinnera734af32014-07-31 13:07:17 +0200153 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100154 return NULL;
155 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100156 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
Victor Stinnere0be4232011-10-25 13:06:09 +0200157}
158
159PyDoc_STRVAR(clock_gettime_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +0100160"clock_gettime(clk_id) -> floating point number\n\
Victor Stinnere0be4232011-10-25 13:06:09 +0200161\n\
162Return the time of the specified clock clk_id.");
Benjamin Peterson37098cd2016-09-13 22:55:09 -0700163#endif /* HAVE_CLOCK_GETTIME */
Victor Stinner30d79472012-04-03 00:45:07 +0200164
Benjamin Peterson37098cd2016-09-13 22:55:09 -0700165#ifdef HAVE_CLOCK_SETTIME
Victor Stinner30d79472012-04-03 00:45:07 +0200166static PyObject *
167time_clock_settime(PyObject *self, PyObject *args)
168{
Victor Stinnerb8d01692012-04-13 23:44:05 +0200169 int clk_id;
Victor Stinner30d79472012-04-03 00:45:07 +0200170 PyObject *obj;
Victor Stinnerb3b45442015-03-28 04:09:41 +0100171 _PyTime_t t;
Victor Stinner30d79472012-04-03 00:45:07 +0200172 struct timespec tp;
173 int ret;
174
175 if (!PyArg_ParseTuple(args, "iO:clock_settime", &clk_id, &obj))
176 return NULL;
177
Victor Stinner02937aa2015-03-28 05:02:39 +0100178 if (_PyTime_FromSecondsObject(&t, obj, _PyTime_ROUND_FLOOR) < 0)
Victor Stinner30d79472012-04-03 00:45:07 +0200179 return NULL;
Victor Stinnerb3b45442015-03-28 04:09:41 +0100180
181 if (_PyTime_AsTimespec(t, &tp) == -1)
182 return NULL;
Victor Stinner30d79472012-04-03 00:45:07 +0200183
184 ret = clock_settime((clockid_t)clk_id, &tp);
185 if (ret != 0) {
Victor Stinnera734af32014-07-31 13:07:17 +0200186 PyErr_SetFromErrno(PyExc_OSError);
Victor Stinner30d79472012-04-03 00:45:07 +0200187 return NULL;
188 }
189 Py_RETURN_NONE;
190}
191
192PyDoc_STRVAR(clock_settime_doc,
193"clock_settime(clk_id, time)\n\
194\n\
195Set the time of the specified clock clk_id.");
Benjamin Peterson37098cd2016-09-13 22:55:09 -0700196#endif /* HAVE_CLOCK_SETTIME */
Victor Stinnere0be4232011-10-25 13:06:09 +0200197
Benjamin Peterson37098cd2016-09-13 22:55:09 -0700198#ifdef HAVE_CLOCK_GETRES
Victor Stinnere0be4232011-10-25 13:06:09 +0200199static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100200time_clock_getres(PyObject *self, PyObject *args)
Victor Stinnere0be4232011-10-25 13:06:09 +0200201{
202 int ret;
Victor Stinnerb8d01692012-04-13 23:44:05 +0200203 int clk_id;
Victor Stinnere0be4232011-10-25 13:06:09 +0200204 struct timespec tp;
205
Victor Stinner4195b5c2012-02-08 23:03:19 +0100206 if (!PyArg_ParseTuple(args, "i:clock_getres", &clk_id))
Victor Stinnere0be4232011-10-25 13:06:09 +0200207 return NULL;
208
209 ret = clock_getres((clockid_t)clk_id, &tp);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100210 if (ret != 0) {
Victor Stinnera734af32014-07-31 13:07:17 +0200211 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100212 return NULL;
213 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100214
215 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
Victor Stinnere0be4232011-10-25 13:06:09 +0200216}
217
218PyDoc_STRVAR(clock_getres_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +0100219"clock_getres(clk_id) -> floating point number\n\
Victor Stinnere0be4232011-10-25 13:06:09 +0200220\n\
221Return the resolution (precision) of the specified clock clk_id.");
Benjamin Peterson37098cd2016-09-13 22:55:09 -0700222#endif /* HAVE_CLOCK_GETRES */
Victor Stinnere0be4232011-10-25 13:06:09 +0200223
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000224static PyObject *
Victor Stinnercb29f012015-03-27 13:31:18 +0100225time_sleep(PyObject *self, PyObject *obj)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000226{
Victor Stinnercb29f012015-03-27 13:31:18 +0100227 _PyTime_t secs;
Victor Stinner869e1772015-03-30 03:49:14 +0200228 if (_PyTime_FromSecondsObject(&secs, obj, _PyTime_ROUND_CEILING))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000229 return NULL;
Victor Stinner7f53a502011-07-05 22:00:25 +0200230 if (secs < 0) {
231 PyErr_SetString(PyExc_ValueError,
232 "sleep length must be non-negative");
233 return NULL;
234 }
Victor Stinnercb29f012015-03-27 13:31:18 +0100235 if (pysleep(secs) != 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000236 return NULL;
237 Py_INCREF(Py_None);
238 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000239}
240
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000241PyDoc_STRVAR(sleep_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000242"sleep(seconds)\n\
243\n\
244Delay execution for a given number of seconds. The argument may be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000245a floating point number for subsecond precision.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000246
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000247static PyStructSequence_Field struct_time_type_fields[] = {
Alexander Belopolsky69f3fd02010-06-05 15:04:51 +0000248 {"tm_year", "year, for example, 1993"},
249 {"tm_mon", "month of year, range [1, 12]"},
250 {"tm_mday", "day of month, range [1, 31]"},
251 {"tm_hour", "hours, range [0, 23]"},
252 {"tm_min", "minutes, range [0, 59]"},
253 {"tm_sec", "seconds, range [0, 61])"},
254 {"tm_wday", "day of week, range [0, 6], Monday is 0"},
255 {"tm_yday", "day of year, range [1, 366]"},
256 {"tm_isdst", "1 if summer time is in effect, 0 if not, and -1 if unknown"},
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400257 {"tm_zone", "abbreviation of timezone name"},
258 {"tm_gmtoff", "offset from UTC in seconds"},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000259 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000260};
261
262static PyStructSequence_Desc struct_time_type_desc = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000263 "time.struct_time",
Alexander Belopolsky69f3fd02010-06-05 15:04:51 +0000264 "The time value as returned by gmtime(), localtime(), and strptime(), and\n"
265 " accepted by asctime(), mktime() and strftime(). May be considered as a\n"
266 " sequence of 9 integers.\n\n"
267 " Note that several fields' values are not the same as those defined by\n"
268 " the C language standard for struct tm. For example, the value of the\n"
269 " field tm_year is the actual year, not year - 1900. See individual\n"
270 " fields' descriptions for details.",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000271 struct_time_type_fields,
272 9,
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000273};
Tim Peters9ad4b682002-02-13 05:14:18 +0000274
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000275static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000276static PyTypeObject StructTimeType;
277
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400278
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000279static PyObject *
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400280tmtotuple(struct tm *p
281#ifndef HAVE_STRUCT_TM_TM_ZONE
282 , const char *zone, int gmtoff
283#endif
284)
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000285{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286 PyObject *v = PyStructSequence_New(&StructTimeType);
287 if (v == NULL)
288 return NULL;
Tim Peters9ad4b682002-02-13 05:14:18 +0000289
Christian Heimes217cfd12007-12-02 14:31:20 +0000290#define SET(i,val) PyStructSequence_SET_ITEM(v, i, PyLong_FromLong((long) val))
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000291
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000292 SET(0, p->tm_year + 1900);
293 SET(1, p->tm_mon + 1); /* Want January == 1 */
294 SET(2, p->tm_mday);
295 SET(3, p->tm_hour);
296 SET(4, p->tm_min);
297 SET(5, p->tm_sec);
298 SET(6, (p->tm_wday + 6) % 7); /* Want Monday == 0 */
299 SET(7, p->tm_yday + 1); /* Want January, 1 == 1 */
300 SET(8, p->tm_isdst);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400301#ifdef HAVE_STRUCT_TM_TM_ZONE
302 PyStructSequence_SET_ITEM(v, 9,
303 PyUnicode_DecodeLocale(p->tm_zone, "surrogateescape"));
304 SET(10, p->tm_gmtoff);
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400305#else
306 PyStructSequence_SET_ITEM(v, 9,
307 PyUnicode_DecodeLocale(zone, "surrogateescape"));
308 SET(10, gmtoff);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400309#endif /* HAVE_STRUCT_TM_TM_ZONE */
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000310#undef SET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000311 if (PyErr_Occurred()) {
312 Py_XDECREF(v);
313 return NULL;
314 }
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000315
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000316 return v;
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000317}
318
Fred Drakef901abd2004-08-03 17:58:55 +0000319/* Parse arg tuple that can contain an optional float-or-None value;
320 format needs to be "|O:name".
321 Returns non-zero on success (parallels PyArg_ParseTuple).
322*/
323static int
Serhiy Storchakaef1585e2015-12-25 20:01:53 +0200324parse_time_t_args(PyObject *args, const char *format, time_t *pwhen)
Fred Drakef901abd2004-08-03 17:58:55 +0000325{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000326 PyObject *ot = NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100327 time_t whent;
Fred Drakef901abd2004-08-03 17:58:55 +0000328
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000329 if (!PyArg_ParseTuple(args, format, &ot))
330 return 0;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100331 if (ot == NULL || ot == Py_None) {
332 whent = time(NULL);
333 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000334 else {
Victor Stinner02937aa2015-03-28 05:02:39 +0100335 if (_PyTime_ObjectToTime_t(ot, &whent, _PyTime_ROUND_FLOOR) == -1)
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100336 return 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000337 }
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100338 *pwhen = whent;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000339 return 1;
Fred Drakef901abd2004-08-03 17:58:55 +0000340}
341
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000342static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000343time_gmtime(PyObject *self, PyObject *args)
Guido van Rossum234f9421993-06-17 12:35:49 +0000344{
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100345 time_t when;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400346 struct tm buf;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100347
348 if (!parse_time_t_args(args, "|O:gmtime", &when))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000349 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100350
351 errno = 0;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400352 if (_PyTime_gmtime(when, &buf) != 0)
353 return NULL;
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400354#ifdef HAVE_STRUCT_TM_TM_ZONE
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100355 return tmtotuple(&buf);
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400356#else
357 return tmtotuple(&buf, "UTC", 0);
358#endif
Guido van Rossum234f9421993-06-17 12:35:49 +0000359}
360
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400361#ifndef HAVE_TIMEGM
362static time_t
363timegm(struct tm *p)
364{
365 /* XXX: the following implementation will not work for tm_year < 1970.
366 but it is likely that platforms that don't have timegm do not support
367 negative timestamps anyways. */
368 return p->tm_sec + p->tm_min*60 + p->tm_hour*3600 + p->tm_yday*86400 +
369 (p->tm_year-70)*31536000 + ((p->tm_year-69)/4)*86400 -
370 ((p->tm_year-1)/100)*86400 + ((p->tm_year+299)/400)*86400;
371}
372#endif
373
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000374PyDoc_STRVAR(gmtime_doc,
Christian Heimes9a371592007-12-28 14:08:13 +0000375"gmtime([seconds]) -> (tm_year, tm_mon, tm_mday, tm_hour, tm_min,\n\
Fred Drake193a3f62002-03-12 21:38:49 +0000376 tm_sec, tm_wday, tm_yday, tm_isdst)\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000377\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000378Convert seconds since the Epoch to a time tuple expressing UTC (a.k.a.\n\
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400379GMT). When 'seconds' is not passed in, convert the current time instead.\n\
380\n\
381If the platform supports the tm_gmtoff and tm_zone, they are available as\n\
382attributes only.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000383
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000384static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000385time_localtime(PyObject *self, PyObject *args)
Guido van Rossum234f9421993-06-17 12:35:49 +0000386{
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100387 time_t when;
388 struct tm buf;
389
390 if (!parse_time_t_args(args, "|O:localtime", &when))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000391 return NULL;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400392 if (_PyTime_localtime(when, &buf) != 0)
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100393 return NULL;
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400394#ifdef HAVE_STRUCT_TM_TM_ZONE
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100395 return tmtotuple(&buf);
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400396#else
397 {
398 struct tm local = buf;
399 char zone[100];
400 int gmtoff;
Steve Dowerc3c6f712016-12-14 11:22:05 -0800401 strftime(zone, sizeof(zone), "%Z", &buf);
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -0400402 gmtoff = timegm(&buf) - when;
403 return tmtotuple(&local, zone, gmtoff);
404 }
405#endif
Guido van Rossum234f9421993-06-17 12:35:49 +0000406}
407
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000408PyDoc_STRVAR(localtime_doc,
Christian Heimes9a371592007-12-28 14:08:13 +0000409"localtime([seconds]) -> (tm_year,tm_mon,tm_mday,tm_hour,tm_min,\n\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000410 tm_sec,tm_wday,tm_yday,tm_isdst)\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000411\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000412Convert seconds since the Epoch to a time tuple expressing local time.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000413When 'seconds' is not passed in, convert the current time instead.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000414
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000415/* Convert 9-item tuple to tm structure. Return 1 on success, set
416 * an exception and return 0 on error.
417 */
Guido van Rossum9e90a671993-06-24 11:10:19 +0000418static int
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000419gettmarg(PyObject *args, struct tm *p)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000420{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000421 int y;
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000422
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000423 memset((void *) p, '\0', sizeof(struct tm));
Guido van Rossumb9081262007-08-25 03:14:09 +0000424
Alexander Belopolsky610e5442011-01-06 21:57:06 +0000425 if (!PyTuple_Check(args)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000426 PyErr_SetString(PyExc_TypeError,
427 "Tuple or struct_time argument required");
428 return 0;
429 }
Skip Montanaro41cfce92007-08-24 21:11:00 +0000430
Alexander Belopolsky610e5442011-01-06 21:57:06 +0000431 if (!PyArg_ParseTuple(args, "iiiiiiiii",
432 &y, &p->tm_mon, &p->tm_mday,
433 &p->tm_hour, &p->tm_min, &p->tm_sec,
434 &p->tm_wday, &p->tm_yday, &p->tm_isdst))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000435 return 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000436 p->tm_year = y - 1900;
437 p->tm_mon--;
438 p->tm_wday = (p->tm_wday + 1) % 7;
439 p->tm_yday--;
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400440#ifdef HAVE_STRUCT_TM_TM_ZONE
441 if (Py_TYPE(args) == &StructTimeType) {
442 PyObject *item;
443 item = PyTuple_GET_ITEM(args, 9);
Serhiy Storchaka06515832016-11-20 09:13:07 +0200444 p->tm_zone = item == Py_None ? NULL : PyUnicode_AsUTF8(item);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400445 item = PyTuple_GET_ITEM(args, 10);
446 p->tm_gmtoff = item == Py_None ? 0 : PyLong_AsLong(item);
447 if (PyErr_Occurred())
448 return 0;
449 }
450#endif /* HAVE_STRUCT_TM_TM_ZONE */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000451 return 1;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000452}
453
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000454/* Check values of the struct tm fields before it is passed to strftime() and
455 * asctime(). Return 1 if all values are valid, otherwise set an exception
456 * and returns 0.
457 */
Victor Stinneref128102010-10-07 01:00:52 +0000458static int
459checktm(struct tm* buf)
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000460{
Victor Stinneref128102010-10-07 01:00:52 +0000461 /* Checks added to make sure strftime() and asctime() does not crash Python by
462 indexing blindly into some array for a textual representation
463 by some bad index (fixes bug #897625 and #6608).
464
465 Also support values of zero from Python code for arguments in which
466 that is out of range by forcing that value to the lowest value that
467 is valid (fixed bug #1520914).
468
469 Valid ranges based on what is allowed in struct tm:
470
471 - tm_year: [0, max(int)] (1)
472 - tm_mon: [0, 11] (2)
473 - tm_mday: [1, 31]
474 - tm_hour: [0, 23]
475 - tm_min: [0, 59]
476 - tm_sec: [0, 60]
477 - tm_wday: [0, 6] (1)
478 - tm_yday: [0, 365] (2)
479 - tm_isdst: [-max(int), max(int)]
480
481 (1) gettmarg() handles bounds-checking.
482 (2) Python's acceptable range is one greater than the range in C,
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000483 thus need to check against automatic decrement by gettmarg().
484 */
485 if (buf->tm_mon == -1)
486 buf->tm_mon = 0;
487 else if (buf->tm_mon < 0 || buf->tm_mon > 11) {
488 PyErr_SetString(PyExc_ValueError, "month out of range");
489 return 0;
490 }
491 if (buf->tm_mday == 0)
492 buf->tm_mday = 1;
493 else if (buf->tm_mday < 0 || buf->tm_mday > 31) {
494 PyErr_SetString(PyExc_ValueError, "day of month out of range");
495 return 0;
496 }
497 if (buf->tm_hour < 0 || buf->tm_hour > 23) {
498 PyErr_SetString(PyExc_ValueError, "hour out of range");
499 return 0;
500 }
501 if (buf->tm_min < 0 || buf->tm_min > 59) {
502 PyErr_SetString(PyExc_ValueError, "minute out of range");
503 return 0;
504 }
505 if (buf->tm_sec < 0 || buf->tm_sec > 61) {
506 PyErr_SetString(PyExc_ValueError, "seconds out of range");
507 return 0;
508 }
509 /* tm_wday does not need checking of its upper-bound since taking
510 ``% 7`` in gettmarg() automatically restricts the range. */
511 if (buf->tm_wday < 0) {
512 PyErr_SetString(PyExc_ValueError, "day of week out of range");
513 return 0;
514 }
515 if (buf->tm_yday == -1)
516 buf->tm_yday = 0;
517 else if (buf->tm_yday < 0 || buf->tm_yday > 365) {
518 PyErr_SetString(PyExc_ValueError, "day of year out of range");
519 return 0;
520 }
521 return 1;
522}
523
Victor Stinnerc1f32ca2011-10-14 02:36:13 +0200524#ifdef MS_WINDOWS
525 /* wcsftime() doesn't format correctly time zones, see issue #10653 */
526# undef HAVE_WCSFTIME
527#endif
Alexander Belopolskycf774542012-10-02 18:39:16 -0400528#define STRFTIME_FORMAT_CODES \
529"Commonly used format codes:\n\
530\n\
531%Y Year with century as a decimal number.\n\
532%m Month as a decimal number [01,12].\n\
533%d Day of the month as a decimal number [01,31].\n\
534%H Hour (24-hour clock) as a decimal number [00,23].\n\
535%M Minute as a decimal number [00,59].\n\
536%S Second as a decimal number [00,61].\n\
537%z Time zone offset from UTC.\n\
538%a Locale's abbreviated weekday name.\n\
539%A Locale's full weekday name.\n\
540%b Locale's abbreviated month name.\n\
541%B Locale's full month name.\n\
542%c Locale's appropriate date and time representation.\n\
543%I Hour (12-hour clock) as a decimal number [01,12].\n\
544%p Locale's equivalent of either AM or PM.\n\
545\n\
546Other codes may be available on your platform. See documentation for\n\
547the C library strftime function.\n"
Victor Stinnerc1f32ca2011-10-14 02:36:13 +0200548
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000549#ifdef HAVE_STRFTIME
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000550#ifdef HAVE_WCSFTIME
551#define time_char wchar_t
552#define format_time wcsftime
553#define time_strlen wcslen
554#else
555#define time_char char
556#define format_time strftime
557#define time_strlen strlen
558#endif
559
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000560static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000561time_strftime(PyObject *self, PyObject *args)
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000562{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000563 PyObject *tup = NULL;
564 struct tm buf;
565 const time_char *fmt;
Victor Stinnerb2904782010-09-29 10:34:19 +0000566#ifdef HAVE_WCSFTIME
567 wchar_t *format;
568#else
569 PyObject *format;
570#endif
Victor Stinneref128102010-10-07 01:00:52 +0000571 PyObject *format_arg;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000572 size_t fmtlen, buflen;
Victor Stinnerb2904782010-09-29 10:34:19 +0000573 time_char *outbuf = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 size_t i;
Victor Stinnerb2904782010-09-29 10:34:19 +0000575 PyObject *ret = NULL;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000576
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000577 memset((void *) &buf, '\0', sizeof(buf));
Guido van Rossum1f41f841998-04-27 19:04:26 +0000578
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000579 /* Will always expect a unicode string to be passed as format.
580 Given that there's no str type anymore in py3k this seems safe.
581 */
Victor Stinneref128102010-10-07 01:00:52 +0000582 if (!PyArg_ParseTuple(args, "U|O:strftime", &format_arg, &tup))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000583 return NULL;
Thomas Woutersfe385252001-01-19 23:16:56 +0000584
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000585 if (tup == NULL) {
586 time_t tt = time(NULL);
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400587 if (_PyTime_localtime(tt, &buf) != 0)
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100588 return NULL;
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000589 }
590 else if (!gettmarg(tup, &buf) || !checktm(&buf))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000591 return NULL;
Guido van Rossum10b164a2001-09-25 13:59:01 +0000592
Victor Stinner36b82d82013-06-25 02:33:53 +0200593#if defined(_MSC_VER) || defined(sun) || defined(_AIX)
Victor Stinner73ea29c2011-01-08 01:56:31 +0000594 if (buf.tm_year + 1900 < 1 || 9999 < buf.tm_year + 1900) {
Victor Stinner6f0e4f92011-03-21 02:14:53 +0100595 PyErr_SetString(PyExc_ValueError,
596 "strftime() requires year in [1; 9999]");
Alexander Belopolsky0dd06f42011-01-08 01:23:02 +0000597 return NULL;
Alexander Belopolskyc64708a2011-01-07 19:59:19 +0000598 }
Victor Stinner73ea29c2011-01-08 01:56:31 +0000599#endif
Alexander Belopolskyc64708a2011-01-07 19:59:19 +0000600
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000601 /* Normalize tm_isdst just in case someone foolishly implements %Z
602 based on the assumption that tm_isdst falls within the range of
603 [-1, 1] */
604 if (buf.tm_isdst < -1)
605 buf.tm_isdst = -1;
606 else if (buf.tm_isdst > 1)
607 buf.tm_isdst = 1;
Brett Cannond1080a32004-03-02 04:38:10 +0000608
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000609#ifdef HAVE_WCSFTIME
Victor Stinnerbeb4135b2010-10-07 01:02:42 +0000610 format = PyUnicode_AsWideCharString(format_arg, NULL);
Victor Stinnerb2904782010-09-29 10:34:19 +0000611 if (format == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000612 return NULL;
Victor Stinnerb2904782010-09-29 10:34:19 +0000613 fmt = format;
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000614#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000615 /* Convert the unicode string to an ascii one */
Victor Stinner1b579672011-12-17 05:47:23 +0100616 format = PyUnicode_EncodeLocale(format_arg, "surrogateescape");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000617 if (format == NULL)
618 return NULL;
619 fmt = PyBytes_AS_STRING(format);
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000620#endif
Amaury Forgeot d'Arcb5be6d42009-03-02 23:52:57 +0000621
Stefan Krah4aea7d32012-02-27 16:30:26 +0100622#if defined(MS_WINDOWS) && !defined(HAVE_WCSFTIME)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000623 /* check that the format string contains only valid directives */
Steve Dowere5b58952015-09-06 19:20:51 -0700624 for (outbuf = strchr(fmt, '%');
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 outbuf != NULL;
Victor Stinner5a3ff792011-10-16 19:08:23 +0200626 outbuf = strchr(outbuf+2, '%'))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000627 {
Steve Dowere5b58952015-09-06 19:20:51 -0700628 if (outbuf[1] == '#')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000629 ++outbuf; /* not documented by python, */
Steve Dowere5b58952015-09-06 19:20:51 -0700630 if (outbuf[1] == '\0')
631 break;
632 if ((outbuf[1] == 'y') && buf.tm_year < 0) {
Tim Golden6e51b8f2013-11-12 12:36:54 +0000633 PyErr_SetString(PyExc_ValueError,
634 "format %y requires year >= 1900 on Windows");
635 Py_DECREF(format);
636 return NULL;
637 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000638 }
Victor Stinner93965f72013-11-23 14:59:33 +0100639#elif (defined(_AIX) || defined(sun)) && defined(HAVE_WCSFTIME)
Steve Dowere5b58952015-09-06 19:20:51 -0700640 for (outbuf = wcschr(fmt, '%');
Victor Stinner55329f82013-11-17 23:39:21 +0100641 outbuf != NULL;
642 outbuf = wcschr(outbuf+2, '%'))
643 {
Steve Dowere5b58952015-09-06 19:20:51 -0700644 if (outbuf[1] == L'\0')
645 break;
Victor Stinner55329f82013-11-17 23:39:21 +0100646 /* Issue #19634: On AIX, wcsftime("y", (1899, 1, 1, 0, 0, 0, 0, 0, 0))
647 returns "0/" instead of "99" */
648 if (outbuf[1] == L'y' && buf.tm_year < 0) {
649 PyErr_SetString(PyExc_ValueError,
650 "format %y requires year >= 1900 on AIX");
Victor Stinner55329f82013-11-17 23:39:21 +0100651 return NULL;
652 }
653 }
Amaury Forgeot d'Arcb5be6d42009-03-02 23:52:57 +0000654#endif
655
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000656 fmtlen = time_strlen(fmt);
Guido van Rossumc222ec21999-02-23 00:00:10 +0000657
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000658 /* I hate these functions that presume you know how big the output
659 * will be ahead of time...
660 */
661 for (i = 1024; ; i += i) {
662 outbuf = (time_char *)PyMem_Malloc(i*sizeof(time_char));
663 if (outbuf == NULL) {
Victor Stinnerb2904782010-09-29 10:34:19 +0000664 PyErr_NoMemory();
665 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000666 }
Steve Dower57ab1cd2015-09-22 14:51:42 -0700667#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
668 errno = 0;
669#endif
Steve Dowerd9ef74e2015-05-22 15:08:34 -0700670 _Py_BEGIN_SUPPRESS_IPH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000671 buflen = format_time(outbuf, i, fmt, &buf);
Steve Dowerd9ef74e2015-05-22 15:08:34 -0700672 _Py_END_SUPPRESS_IPH
Victor Stinner136ea492011-12-17 22:37:18 +0100673#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Steve Dower97cded92015-09-08 19:12:51 -0700674 /* VisualStudio .NET 2005 does this properly */
675 if (buflen == 0 && errno == EINVAL) {
676 PyErr_SetString(PyExc_ValueError, "Invalid format string");
677 PyMem_Free(outbuf);
678 break;
679 }
Victor Stinner136ea492011-12-17 22:37:18 +0100680#endif
Steve Dower97cded92015-09-08 19:12:51 -0700681 if (buflen > 0 || i >= 256 * fmtlen) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000682 /* If the buffer is 256 times as long as the format,
683 it's probably not failing for lack of room!
684 More likely, the format yields an empty result,
685 e.g. an empty format, or %Z when the timezone
686 is unknown. */
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000687#ifdef HAVE_WCSFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000688 ret = PyUnicode_FromWideChar(outbuf, buflen);
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000689#else
Victor Stinner1b579672011-12-17 05:47:23 +0100690 ret = PyUnicode_DecodeLocaleAndSize(outbuf, buflen,
691 "surrogateescape");
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000692#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000693 PyMem_Free(outbuf);
Victor Stinnerb2904782010-09-29 10:34:19 +0000694 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000695 }
696 PyMem_Free(outbuf);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000697 }
Victor Stinnerb2904782010-09-29 10:34:19 +0000698#ifdef HAVE_WCSFTIME
699 PyMem_Free(format);
700#else
701 Py_DECREF(format);
702#endif
703 return ret;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000704}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000705
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000706#undef time_char
707#undef format_time
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000708PyDoc_STRVAR(strftime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000709"strftime(format[, tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000710\n\
711Convert a time tuple to a string according to a format specification.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000712See the library reference manual for formatting codes. When the time tuple\n\
Alexander Belopolskycf774542012-10-02 18:39:16 -0400713is not present, current time as returned by localtime() is used.\n\
714\n" STRFTIME_FORMAT_CODES);
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000715#endif /* HAVE_STRFTIME */
716
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000717static PyObject *
718time_strptime(PyObject *self, PyObject *args)
719{
Victor Stinnerdbe28d22016-12-09 00:38:53 +0100720 PyObject *module, *func, *result;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200721 _Py_IDENTIFIER(_strptime_time);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000722
Victor Stinnerdbe28d22016-12-09 00:38:53 +0100723 module = PyImport_ImportModuleNoBlock("_strptime");
724 if (!module)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000725 return NULL;
Victor Stinnerdbe28d22016-12-09 00:38:53 +0100726
727 func = _PyObject_GetAttrId(module, &PyId__strptime_time);
728 Py_DECREF(module);
729 if (!func) {
730 return NULL;
731 }
732
733 result = PyObject_Call(func, args, NULL);
734 Py_DECREF(func);
735 return result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000736}
737
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000738
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000739PyDoc_STRVAR(strptime_doc,
Brett Cannon20def8b2003-07-01 05:16:08 +0000740"strptime(string, format) -> struct_time\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000741\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000742Parse a string to a time tuple according to a format specification.\n\
Alexander Belopolskycf774542012-10-02 18:39:16 -0400743See the library reference manual for formatting codes (same as\n\
744strftime()).\n\
745\n" STRFTIME_FORMAT_CODES);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000746
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000747static PyObject *
748_asctime(struct tm *timeptr)
749{
750 /* Inspired by Open Group reference implementation available at
751 * http://pubs.opengroup.org/onlinepubs/009695399/functions/asctime.html */
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200752 static const char wday_name[7][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000753 "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
754 };
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200755 static const char mon_name[12][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000756 "Jan", "Feb", "Mar", "Apr", "May", "Jun",
757 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
758 };
Victor Stinner499dfcf2011-03-21 13:26:24 +0100759 return PyUnicode_FromFormat(
760 "%s %s%3d %.2d:%.2d:%.2d %d",
761 wday_name[timeptr->tm_wday],
762 mon_name[timeptr->tm_mon],
763 timeptr->tm_mday, timeptr->tm_hour,
764 timeptr->tm_min, timeptr->tm_sec,
765 1900 + timeptr->tm_year);
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000766}
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000767
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000768static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000769time_asctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000770{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000771 PyObject *tup = NULL;
772 struct tm buf;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000774 if (!PyArg_UnpackTuple(args, "asctime", 0, 1, &tup))
775 return NULL;
776 if (tup == NULL) {
777 time_t tt = time(NULL);
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400778 if (_PyTime_localtime(tt, &buf) != 0)
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100779 return NULL;
780
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000781 } else if (!gettmarg(tup, &buf) || !checktm(&buf))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000782 return NULL;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000783 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000784}
785
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000786PyDoc_STRVAR(asctime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000787"asctime([tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000788\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000789Convert a time tuple to a string, e.g. 'Sat Jun 06 16:26:11 1998'.\n\
790When the time tuple is not present, current time as returned by localtime()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000791is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000792
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000793static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000794time_ctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000795{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000796 time_t tt;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100797 struct tm buf;
798 if (!parse_time_t_args(args, "|O:ctime", &tt))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000799 return NULL;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400800 if (_PyTime_localtime(tt, &buf) != 0)
Alexander Belopolsky5da468f2011-01-04 17:15:52 +0000801 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100802 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000803}
804
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000805PyDoc_STRVAR(ctime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000806"ctime(seconds) -> string\n\
807\n\
808Convert a time in seconds since the Epoch to a string in local time.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000809This is equivalent to asctime(localtime(seconds)). When the time tuple is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000810not present, current time as returned by localtime() is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000811
Guido van Rossum60cd8131998-03-06 17:16:21 +0000812#ifdef HAVE_MKTIME
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000813static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100814time_mktime(PyObject *self, PyObject *tup)
Guido van Rossum234f9421993-06-17 12:35:49 +0000815{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000816 struct tm buf;
817 time_t tt;
818 if (!gettmarg(tup, &buf))
819 return NULL;
Victor Stinner1ac42612014-02-21 09:27:17 +0100820#ifdef _AIX
821 /* year < 1902 or year > 2037 */
822 if (buf.tm_year < 2 || buf.tm_year > 137) {
823 /* Issue #19748: On AIX, mktime() doesn't report overflow error for
824 * timestamp < -2^31 or timestamp > 2**31-1. */
825 PyErr_SetString(PyExc_OverflowError,
826 "mktime argument out of range");
827 return NULL;
828 }
829#else
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000830 buf.tm_wday = -1; /* sentinel; original value ignored */
Victor Stinner1ac42612014-02-21 09:27:17 +0100831#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000832 tt = mktime(&buf);
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000833 /* Return value of -1 does not necessarily mean an error, but tm_wday
Ezio Melotti13925002011-03-16 11:05:33 +0200834 * cannot remain set to -1 if mktime succeeded. */
Victor Stinner93037492013-06-25 22:54:35 +0200835 if (tt == (time_t)(-1)
836#ifndef _AIX
837 /* Return value of -1 does not necessarily mean an error, but
838 * tm_wday cannot remain set to -1 if mktime succeeded. */
839 && buf.tm_wday == -1
840#else
841 /* on AIX, tm_wday is always sets, even on error */
842#endif
843 )
844 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000845 PyErr_SetString(PyExc_OverflowError,
846 "mktime argument out of range");
847 return NULL;
848 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100849 return PyFloat_FromDouble((double)tt);
Guido van Rossum234f9421993-06-17 12:35:49 +0000850}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000851
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000852PyDoc_STRVAR(mktime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000853"mktime(tuple) -> floating point number\n\
854\n\
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400855Convert a time tuple in local time to seconds since the Epoch.\n\
856Note that mktime(gmtime(0)) will not generally return zero for most\n\
857time zones; instead the returned value will either be equal to that\n\
858of the timezone or altzone attributes on the time module.");
Guido van Rossum60cd8131998-03-06 17:16:21 +0000859#endif /* HAVE_MKTIME */
Guido van Rossum234f9421993-06-17 12:35:49 +0000860
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000861#ifdef HAVE_WORKING_TZSET
Martin v. Löwis1a214512008-06-11 05:26:20 +0000862static void PyInit_timezone(PyObject *module);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000863
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000864static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000865time_tzset(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000866{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000867 PyObject* m;
Fred Drake9bb74322002-04-01 14:49:59 +0000868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000869 m = PyImport_ImportModuleNoBlock("time");
870 if (m == NULL) {
871 return NULL;
872 }
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000873
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000874 tzset();
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000875
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000876 /* Reset timezone, altzone, daylight and tzname */
877 PyInit_timezone(m);
878 Py_DECREF(m);
Victor Stinner2ff51b82013-07-17 21:42:45 +0200879 if (PyErr_Occurred())
880 return NULL;
Tim Peters1b6f7a92004-06-20 02:50:16 +0000881
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000882 Py_INCREF(Py_None);
883 return Py_None;
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000884}
885
886PyDoc_STRVAR(tzset_doc,
R. David Murray4d55bf92010-12-14 00:55:46 +0000887"tzset()\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000888\n\
889Initialize, or reinitialize, the local timezone to the value stored in\n\
890os.environ['TZ']. The TZ environment variable should be specified in\n\
Neal Norwitzdc8e1942004-07-20 22:34:37 +0000891standard Unix timezone format as documented in the tzset man page\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000892(eg. 'US/Eastern', 'Europe/Amsterdam'). Unknown timezones will silently\n\
893fall back to UTC. If the TZ environment variable is not set, the local\n\
894timezone is set to the systems best guess of wallclock time.\n\
895Changing the TZ environment variable without calling tzset *may* change\n\
896the local timezone used by methods such as localtime, but this behaviour\n\
897should not be relied on.");
898#endif /* HAVE_WORKING_TZSET */
899
Victor Stinnerae586492014-09-02 23:18:25 +0200900static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200901pymonotonic(_Py_clock_info_t *info)
Victor Stinnerb94b2662012-01-18 01:50:21 +0100902{
Victor Stinner4bfb4602015-03-27 22:27:24 +0100903 _PyTime_t t;
904 double d;
905 if (_PyTime_GetMonotonicClockWithInfo(&t, info) < 0) {
Victor Stinnerae586492014-09-02 23:18:25 +0200906 assert(info != NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100907 return NULL;
908 }
Victor Stinner4bfb4602015-03-27 22:27:24 +0100909 d = _PyTime_AsSecondsDouble(t);
910 return PyFloat_FromDouble(d);
Victor Stinner8b302012012-02-07 23:29:46 +0100911}
912
Victor Stinner071eca32012-03-15 01:17:09 +0100913static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200914time_monotonic(PyObject *self, PyObject *unused)
Victor Stinner071eca32012-03-15 01:17:09 +0100915{
Victor Stinnerec895392012-04-29 02:41:27 +0200916 return pymonotonic(NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100917}
918
Victor Stinnerec895392012-04-29 02:41:27 +0200919PyDoc_STRVAR(monotonic_doc,
920"monotonic() -> float\n\
Victor Stinner8b302012012-02-07 23:29:46 +0100921\n\
Victor Stinnerec895392012-04-29 02:41:27 +0200922Monotonic clock, cannot go backward.");
Victor Stinnerec919cc2012-03-15 00:58:32 +0100923
Victor Stinnerec895392012-04-29 02:41:27 +0200924static PyObject*
925perf_counter(_Py_clock_info_t *info)
926{
Victor Stinner54884492014-08-29 16:51:33 +0200927#ifdef WIN32_PERF_COUNTER
928 return win_perf_counter(info);
929#else
Victor Stinnerae586492014-09-02 23:18:25 +0200930 return pymonotonic(info);
Victor Stinner54884492014-08-29 16:51:33 +0200931#endif
Victor Stinnerec895392012-04-29 02:41:27 +0200932}
933
934static PyObject *
935time_perf_counter(PyObject *self, PyObject *unused)
936{
937 return perf_counter(NULL);
938}
939
940PyDoc_STRVAR(perf_counter_doc,
941"perf_counter() -> float\n\
942\n\
943Performance counter for benchmarking.");
944
945static PyObject*
946py_process_time(_Py_clock_info_t *info)
947{
948#if defined(MS_WINDOWS)
949 HANDLE process;
950 FILETIME creation_time, exit_time, kernel_time, user_time;
951 ULARGE_INTEGER large;
952 double total;
953 BOOL ok;
954
955 process = GetCurrentProcess();
956 ok = GetProcessTimes(process, &creation_time, &exit_time, &kernel_time, &user_time);
957 if (!ok)
958 return PyErr_SetFromWindowsErr(0);
959
960 large.u.LowPart = kernel_time.dwLowDateTime;
961 large.u.HighPart = kernel_time.dwHighDateTime;
962 total = (double)large.QuadPart;
963 large.u.LowPart = user_time.dwLowDateTime;
964 large.u.HighPart = user_time.dwHighDateTime;
965 total += (double)large.QuadPart;
966 if (info) {
967 info->implementation = "GetProcessTimes()";
968 info->resolution = 1e-7;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400969 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200970 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200971 }
972 return PyFloat_FromDouble(total * 1e-7);
973#else
974
975#if defined(HAVE_SYS_RESOURCE_H)
976 struct rusage ru;
977#endif
978#ifdef HAVE_TIMES
979 struct tms t;
980 static long ticks_per_second = -1;
981#endif
982
983#if defined(HAVE_CLOCK_GETTIME) \
984 && (defined(CLOCK_PROCESS_CPUTIME_ID) || defined(CLOCK_PROF))
985 struct timespec tp;
986#ifdef CLOCK_PROF
987 const clockid_t clk_id = CLOCK_PROF;
988 const char *function = "clock_gettime(CLOCK_PROF)";
989#else
990 const clockid_t clk_id = CLOCK_PROCESS_CPUTIME_ID;
991 const char *function = "clock_gettime(CLOCK_PROCESS_CPUTIME_ID)";
992#endif
993
994 if (clock_gettime(clk_id, &tp) == 0) {
995 if (info) {
996 struct timespec res;
997 info->implementation = function;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400998 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200999 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001000 if (clock_getres(clk_id, &res) == 0)
1001 info->resolution = res.tv_sec + res.tv_nsec * 1e-9;
1002 else
1003 info->resolution = 1e-9;
1004 }
1005 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
1006 }
1007#endif
1008
1009#if defined(HAVE_SYS_RESOURCE_H)
1010 if (getrusage(RUSAGE_SELF, &ru) == 0) {
1011 double total;
1012 total = ru.ru_utime.tv_sec + ru.ru_utime.tv_usec * 1e-6;
1013 total += ru.ru_stime.tv_sec + ru.ru_stime.tv_usec * 1e-6;
1014 if (info) {
1015 info->implementation = "getrusage(RUSAGE_SELF)";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001016 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001017 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001018 info->resolution = 1e-6;
1019 }
1020 return PyFloat_FromDouble(total);
1021 }
1022#endif
1023
1024#ifdef HAVE_TIMES
1025 if (times(&t) != (clock_t)-1) {
1026 double total;
1027
1028 if (ticks_per_second == -1) {
1029#if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
1030 ticks_per_second = sysconf(_SC_CLK_TCK);
1031 if (ticks_per_second < 1)
1032 ticks_per_second = -1;
1033#elif defined(HZ)
1034 ticks_per_second = HZ;
1035#else
1036 ticks_per_second = 60; /* magic fallback value; may be bogus */
1037#endif
1038 }
1039
1040 if (ticks_per_second != -1) {
1041 total = (double)t.tms_utime / ticks_per_second;
1042 total += (double)t.tms_stime / ticks_per_second;
1043 if (info) {
1044 info->implementation = "times()";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001045 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001046 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001047 info->resolution = 1.0 / ticks_per_second;
1048 }
1049 return PyFloat_FromDouble(total);
1050 }
1051 }
1052#endif
1053
Victor Stinner53e22bf2016-07-08 17:55:01 +02001054 /* Currently, Python 3 requires clock() to build: see issue #22624 */
Victor Stinnerec895392012-04-29 02:41:27 +02001055 return floatclock(info);
1056#endif
1057}
1058
1059static PyObject *
1060time_process_time(PyObject *self, PyObject *unused)
1061{
1062 return py_process_time(NULL);
1063}
1064
1065PyDoc_STRVAR(process_time_doc,
1066"process_time() -> float\n\
1067\n\
1068Process time for profiling: sum of the kernel and user-space CPU time.");
1069
1070
Victor Stinnerec895392012-04-29 02:41:27 +02001071static PyObject *
1072time_get_clock_info(PyObject *self, PyObject *args)
1073{
1074 char *name;
Victor Stinnerec895392012-04-29 02:41:27 +02001075 _Py_clock_info_t info;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001076 PyObject *obj = NULL, *dict, *ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001077
1078 if (!PyArg_ParseTuple(args, "s:get_clock_info", &name))
1079 return NULL;
1080
1081#ifdef Py_DEBUG
1082 info.implementation = NULL;
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001083 info.monotonic = -1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001084 info.adjustable = -1;
Victor Stinnerec895392012-04-29 02:41:27 +02001085 info.resolution = -1.0;
1086#else
1087 info.implementation = "";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001088 info.monotonic = 0;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001089 info.adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001090 info.resolution = 1.0;
1091#endif
1092
1093 if (strcmp(name, "time") == 0)
1094 obj = floattime(&info);
1095#ifdef PYCLOCK
1096 else if (strcmp(name, "clock") == 0)
1097 obj = pyclock(&info);
1098#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001099 else if (strcmp(name, "monotonic") == 0)
1100 obj = pymonotonic(&info);
Victor Stinnerec895392012-04-29 02:41:27 +02001101 else if (strcmp(name, "perf_counter") == 0)
1102 obj = perf_counter(&info);
1103 else if (strcmp(name, "process_time") == 0)
1104 obj = py_process_time(&info);
1105 else {
1106 PyErr_SetString(PyExc_ValueError, "unknown clock");
1107 return NULL;
1108 }
1109 if (obj == NULL)
1110 return NULL;
1111 Py_DECREF(obj);
1112
Victor Stinnerbda4b882012-06-12 22:11:44 +02001113 dict = PyDict_New();
1114 if (dict == NULL)
Victor Stinnerec895392012-04-29 02:41:27 +02001115 return NULL;
1116
1117 assert(info.implementation != NULL);
1118 obj = PyUnicode_FromString(info.implementation);
1119 if (obj == NULL)
1120 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001121 if (PyDict_SetItemString(dict, "implementation", obj) == -1)
1122 goto error;
1123 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001124
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001125 assert(info.monotonic != -1);
1126 obj = PyBool_FromLong(info.monotonic);
Victor Stinnerec895392012-04-29 02:41:27 +02001127 if (obj == NULL)
1128 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001129 if (PyDict_SetItemString(dict, "monotonic", obj) == -1)
1130 goto error;
1131 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001132
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001133 assert(info.adjustable != -1);
1134 obj = PyBool_FromLong(info.adjustable);
Victor Stinnerec895392012-04-29 02:41:27 +02001135 if (obj == NULL)
1136 goto error;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001137 if (PyDict_SetItemString(dict, "adjustable", obj) == -1)
Victor Stinnerbda4b882012-06-12 22:11:44 +02001138 goto error;
1139 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001140
1141 assert(info.resolution > 0.0);
1142 assert(info.resolution <= 1.0);
1143 obj = PyFloat_FromDouble(info.resolution);
1144 if (obj == NULL)
1145 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001146 if (PyDict_SetItemString(dict, "resolution", obj) == -1)
1147 goto error;
1148 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001149
Victor Stinnerbda4b882012-06-12 22:11:44 +02001150 ns = _PyNamespace_New(dict);
1151 Py_DECREF(dict);
1152 return ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001153
1154error:
Victor Stinnerbda4b882012-06-12 22:11:44 +02001155 Py_DECREF(dict);
1156 Py_XDECREF(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001157 return NULL;
1158}
1159
1160PyDoc_STRVAR(get_clock_info_doc,
1161"get_clock_info(name: str) -> dict\n\
1162\n\
1163Get information of the specified clock.");
Victor Stinner8b302012012-02-07 23:29:46 +01001164
Martin v. Löwisd218dc12008-04-07 03:17:54 +00001165static void
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001166get_zone(char *zone, int n, struct tm *p)
1167{
1168#ifdef HAVE_STRUCT_TM_TM_ZONE
1169 strncpy(zone, p->tm_zone ? p->tm_zone : " ", n);
1170#else
1171 tzset();
1172 strftime(zone, n, "%Z", p);
1173#endif
1174}
1175
1176static int
1177get_gmtoff(time_t t, struct tm *p)
1178{
1179#ifdef HAVE_STRUCT_TM_TM_ZONE
1180 return p->tm_gmtoff;
1181#else
1182 return timegm(p) - t;
1183#endif
1184}
1185
1186static void
Martin v. Löwis1a214512008-06-11 05:26:20 +00001187PyInit_timezone(PyObject *m) {
1188 /* This code moved from PyInit_time wholesale to allow calling it from
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001189 time_tzset. In the future, some parts of it can be moved back
1190 (for platforms that don't HAVE_WORKING_TZSET, when we know what they
1191 are), and the extraneous calls to tzset(3) should be removed.
1192 I haven't done this yet, as I don't want to change this code as
1193 little as possible when introducing the time.tzset and time.tzsetwall
1194 methods. This should simply be a method of doing the following once,
1195 at the top of this function and removing the call to tzset() from
1196 time_tzset():
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001197
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001198 #ifdef HAVE_TZSET
1199 tzset()
1200 #endif
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 And I'm lazy and hate C so nyer.
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001203 */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001204#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001205 PyObject *otz0, *otz1;
1206 tzset();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001207 PyModule_AddIntConstant(m, "timezone", timezone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001208#ifdef HAVE_ALTZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001209 PyModule_AddIntConstant(m, "altzone", altzone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001210#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001211 PyModule_AddIntConstant(m, "altzone", timezone-3600);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001212#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001213 PyModule_AddIntConstant(m, "daylight", daylight);
Victor Stinner1b579672011-12-17 05:47:23 +01001214 otz0 = PyUnicode_DecodeLocale(tzname[0], "surrogateescape");
1215 otz1 = PyUnicode_DecodeLocale(tzname[1], "surrogateescape");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001216 PyModule_AddObject(m, "tzname", Py_BuildValue("(NN)", otz0, otz1));
Guido van Rossum10b164a2001-09-25 13:59:01 +00001217#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001218 {
Guido van Rossum234f9421993-06-17 12:35:49 +00001219#define YEAR ((time_t)((365 * 24 + 6) * 3600))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001220 time_t t;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -04001221 struct tm p;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001222 long janzone, julyzone;
1223 char janname[10], julyname[10];
1224 t = (time((time_t *)0) / YEAR) * YEAR;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -04001225 _PyTime_localtime(t, &p);
1226 get_zone(janname, 9, &p);
1227 janzone = -get_gmtoff(t, &p);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001228 janname[9] = '\0';
1229 t += YEAR/2;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -04001230 _PyTime_localtime(t, &p);
1231 get_zone(julyname, 9, &p);
1232 julyzone = -get_gmtoff(t, &p);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001233 julyname[9] = '\0';
Guido van Rossum10b164a2001-09-25 13:59:01 +00001234
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001235 if( janzone < julyzone ) {
1236 /* DST is reversed in the southern hemisphere */
1237 PyModule_AddIntConstant(m, "timezone", julyzone);
1238 PyModule_AddIntConstant(m, "altzone", janzone);
1239 PyModule_AddIntConstant(m, "daylight",
1240 janzone != julyzone);
1241 PyModule_AddObject(m, "tzname",
1242 Py_BuildValue("(zz)",
1243 julyname, janname));
1244 } else {
1245 PyModule_AddIntConstant(m, "timezone", janzone);
1246 PyModule_AddIntConstant(m, "altzone", julyzone);
1247 PyModule_AddIntConstant(m, "daylight",
1248 janzone != julyzone);
1249 PyModule_AddObject(m, "tzname",
1250 Py_BuildValue("(zz)",
1251 janname, julyname));
1252 }
1253 }
Tim Peters26ae7cd2001-03-20 03:26:49 +00001254#ifdef __CYGWIN__
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001255 tzset();
1256 PyModule_AddIntConstant(m, "timezone", _timezone);
1257 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
1258 PyModule_AddIntConstant(m, "daylight", _daylight);
1259 PyModule_AddObject(m, "tzname",
1260 Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
Tim Peters26ae7cd2001-03-20 03:26:49 +00001261#endif /* __CYGWIN__ */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001262#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001263}
1264
1265
1266static PyMethodDef time_methods[] = {
Victor Stinner4195b5c2012-02-08 23:03:19 +01001267 {"time", time_time, METH_NOARGS, time_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001268#ifdef PYCLOCK
Victor Stinner4195b5c2012-02-08 23:03:19 +01001269 {"clock", time_clock, METH_NOARGS, clock_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001270#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001271#ifdef HAVE_CLOCK_GETTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001272 {"clock_gettime", time_clock_gettime, METH_VARARGS, clock_gettime_doc},
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001273#endif
1274#ifdef HAVE_CLOCK_SETTIME
Victor Stinner30d79472012-04-03 00:45:07 +02001275 {"clock_settime", time_clock_settime, METH_VARARGS, clock_settime_doc},
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001276#endif
1277#ifdef HAVE_CLOCK_GETRES
Victor Stinner4195b5c2012-02-08 23:03:19 +01001278 {"clock_getres", time_clock_getres, METH_VARARGS, clock_getres_doc},
Victor Stinnere0be4232011-10-25 13:06:09 +02001279#endif
Victor Stinnercb29f012015-03-27 13:31:18 +01001280 {"sleep", time_sleep, METH_O, sleep_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001281 {"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
1282 {"localtime", time_localtime, METH_VARARGS, localtime_doc},
1283 {"asctime", time_asctime, METH_VARARGS, asctime_doc},
1284 {"ctime", time_ctime, METH_VARARGS, ctime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001285#ifdef HAVE_MKTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001286 {"mktime", time_mktime, METH_O, mktime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001287#endif
1288#ifdef HAVE_STRFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001289 {"strftime", time_strftime, METH_VARARGS, strftime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001290#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001291 {"strptime", time_strptime, METH_VARARGS, strptime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001292#ifdef HAVE_WORKING_TZSET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001293 {"tzset", time_tzset, METH_NOARGS, tzset_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001294#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001295 {"monotonic", time_monotonic, METH_NOARGS, monotonic_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001296 {"process_time", time_process_time, METH_NOARGS, process_time_doc},
1297 {"perf_counter", time_perf_counter, METH_NOARGS, perf_counter_doc},
1298 {"get_clock_info", time_get_clock_info, METH_VARARGS, get_clock_info_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001299 {NULL, NULL} /* sentinel */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001300};
1301
1302
1303PyDoc_STRVAR(module_doc,
1304"This module provides various functions to manipulate time values.\n\
1305\n\
1306There are two standard representations of time. One is the number\n\
1307of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
1308or a floating point number (to represent fractions of seconds).\n\
1309The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
1310The actual value can be retrieved by calling gmtime(0).\n\
1311\n\
1312The other representation is a tuple of 9 integers giving local time.\n\
1313The tuple items are:\n\
Alexander Belopolsky03163ac2011-05-02 12:20:52 -04001314 year (including century, e.g. 1998)\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001315 month (1-12)\n\
1316 day (1-31)\n\
1317 hours (0-23)\n\
1318 minutes (0-59)\n\
1319 seconds (0-59)\n\
1320 weekday (0-6, Monday is 0)\n\
1321 Julian day (day in the year, 1-366)\n\
1322 DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
1323If the DST flag is 0, the time is given in the regular time zone;\n\
1324if it is 1, the time is given in the DST time zone;\n\
1325if it is -1, mktime() should guess based on the date and time.\n\
1326\n\
1327Variables:\n\
1328\n\
1329timezone -- difference in seconds between UTC and local standard time\n\
1330altzone -- difference in seconds between UTC and local DST time\n\
1331daylight -- whether local time should reflect DST\n\
1332tzname -- tuple of (standard time zone name, DST time zone name)\n\
1333\n\
1334Functions:\n\
1335\n\
1336time() -- return current time in seconds since the Epoch as a float\n\
1337clock() -- return CPU time since process start as a float\n\
1338sleep() -- delay for a number of seconds given as a float\n\
1339gmtime() -- convert seconds since Epoch to UTC tuple\n\
1340localtime() -- convert seconds since Epoch to local time tuple\n\
1341asctime() -- convert time tuple to string\n\
1342ctime() -- convert time in seconds to string\n\
1343mktime() -- convert local time tuple to seconds since Epoch\n\
1344strftime() -- convert time tuple to string according to format specification\n\
1345strptime() -- parse string to time tuple according to format specification\n\
1346tzset() -- change the local timezone");
1347
1348
Martin v. Löwis1a214512008-06-11 05:26:20 +00001349
1350static struct PyModuleDef timemodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001351 PyModuleDef_HEAD_INIT,
1352 "time",
1353 module_doc,
1354 -1,
1355 time_methods,
1356 NULL,
1357 NULL,
1358 NULL,
1359 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001360};
1361
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001362PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001363PyInit_time(void)
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001364{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001365 PyObject *m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001366 m = PyModule_Create(&timemodule);
1367 if (m == NULL)
1368 return NULL;
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001370 /* Set, or reset, module variables like time.timezone */
1371 PyInit_timezone(m);
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001372
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001373#ifdef CLOCK_REALTIME
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001374 PyModule_AddIntMacro(m, CLOCK_REALTIME);
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001375#endif
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001376#ifdef CLOCK_MONOTONIC
1377 PyModule_AddIntMacro(m, CLOCK_MONOTONIC);
1378#endif
1379#ifdef CLOCK_MONOTONIC_RAW
1380 PyModule_AddIntMacro(m, CLOCK_MONOTONIC_RAW);
1381#endif
1382#ifdef CLOCK_HIGHRES
1383 PyModule_AddIntMacro(m, CLOCK_HIGHRES);
1384#endif
1385#ifdef CLOCK_PROCESS_CPUTIME_ID
1386 PyModule_AddIntMacro(m, CLOCK_PROCESS_CPUTIME_ID);
1387#endif
1388#ifdef CLOCK_THREAD_CPUTIME_ID
1389 PyModule_AddIntMacro(m, CLOCK_THREAD_CPUTIME_ID);
1390#endif
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001391
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001392 if (!initialized) {
Victor Stinner1c8f0592013-07-22 22:24:54 +02001393 if (PyStructSequence_InitType2(&StructTimeType,
1394 &struct_time_type_desc) < 0)
1395 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001396 }
1397 Py_INCREF(&StructTimeType);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -04001398 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 11);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001399 PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
1400 initialized = 1;
1401 return m;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001402}
1403
Victor Stinner071eca32012-03-15 01:17:09 +01001404static PyObject*
Victor Stinnerec895392012-04-29 02:41:27 +02001405floattime(_Py_clock_info_t *info)
Victor Stinner4195b5c2012-02-08 23:03:19 +01001406{
Victor Stinnera47b8812015-03-27 18:16:17 +01001407 _PyTime_t t;
1408 double d;
1409 if (_PyTime_GetSystemClockWithInfo(&t, info) < 0) {
Victor Stinner00111242014-08-29 16:31:59 +02001410 assert(info != NULL);
1411 return NULL;
1412 }
Victor Stinnera47b8812015-03-27 18:16:17 +01001413 d = _PyTime_AsSecondsDouble(t);
1414 return PyFloat_FromDouble(d);
Victor Stinner4195b5c2012-02-08 23:03:19 +01001415}
1416
1417
Victor Stinnercb29f012015-03-27 13:31:18 +01001418/* Implement pysleep() for various platforms.
Guido van Rossumb6775db1994-08-01 11:34:53 +00001419 When interrupted (or when another error occurs), return -1 and
1420 set an exception; else return 0. */
1421
1422static int
Victor Stinnercb29f012015-03-27 13:31:18 +01001423pysleep(_PyTime_t secs)
Guido van Rossum426035c1991-02-19 12:27:35 +00001424{
Victor Stinnercb29f012015-03-27 13:31:18 +01001425 _PyTime_t deadline, monotonic;
Victor Stinner79d68f92015-03-19 21:54:09 +01001426#ifndef MS_WINDOWS
1427 struct timeval timeout;
Victor Stinner79d68f92015-03-19 21:54:09 +01001428 int err = 0;
1429#else
Victor Stinnercb29f012015-03-27 13:31:18 +01001430 _PyTime_t millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001431 unsigned long ul_millis;
1432 DWORD rc;
1433 HANDLE hInterruptEvent;
Victor Stinner0c2fd892015-03-17 10:49:17 +01001434#endif
Victor Stinner79d68f92015-03-19 21:54:09 +01001435
Victor Stinnercb29f012015-03-27 13:31:18 +01001436 deadline = _PyTime_GetMonotonicClock() + secs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001437
1438 do {
1439#ifndef MS_WINDOWS
Victor Stinner869e1772015-03-30 03:49:14 +02001440 if (_PyTime_AsTimeval(secs, &timeout, _PyTime_ROUND_CEILING) < 0)
Victor Stinnercb29f012015-03-27 13:31:18 +01001441 return -1;
Victor Stinner79d68f92015-03-19 21:54:09 +01001442
1443 Py_BEGIN_ALLOW_THREADS
1444 err = select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &timeout);
1445 Py_END_ALLOW_THREADS
1446
1447 if (err == 0)
1448 break;
1449
1450 if (errno != EINTR) {
Victor Stinner0c2fd892015-03-17 10:49:17 +01001451 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001452 return -1;
1453 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001454#else
Victor Stinner869e1772015-03-30 03:49:14 +02001455 millisecs = _PyTime_AsMilliseconds(secs, _PyTime_ROUND_CEILING);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001456 if (millisecs > (double)ULONG_MAX) {
1457 PyErr_SetString(PyExc_OverflowError,
1458 "sleep length is too large");
1459 return -1;
1460 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001461
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001462 /* Allow sleep(0) to maintain win32 semantics, and as decreed
1463 * by Guido, only the main thread can be interrupted.
1464 */
1465 ul_millis = (unsigned long)millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001466 if (ul_millis == 0 || !_PyOS_IsMainThread()) {
1467 Py_BEGIN_ALLOW_THREADS
Victor Stinner0eac1302015-03-20 03:06:12 +01001468 Sleep(ul_millis);
Victor Stinner79d68f92015-03-19 21:54:09 +01001469 Py_END_ALLOW_THREADS
1470 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001471 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001472
1473 hInterruptEvent = _PyOS_SigintEvent();
1474 ResetEvent(hInterruptEvent);
1475
1476 Py_BEGIN_ALLOW_THREADS
1477 rc = WaitForSingleObjectEx(hInterruptEvent, ul_millis, FALSE);
Victor Stinner0c2fd892015-03-17 10:49:17 +01001478 Py_END_ALLOW_THREADS
Victor Stinner79d68f92015-03-19 21:54:09 +01001479
1480 if (rc != WAIT_OBJECT_0)
1481 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001482#endif
Victor Stinner0c2fd892015-03-17 10:49:17 +01001483
Victor Stinner79d68f92015-03-19 21:54:09 +01001484 /* sleep was interrupted by SIGINT */
1485 if (PyErr_CheckSignals())
1486 return -1;
1487
Victor Stinnercb29f012015-03-27 13:31:18 +01001488 monotonic = _PyTime_GetMonotonicClock();
1489 secs = deadline - monotonic;
Victor Stinner6aa446c2015-03-30 21:33:51 +02001490 if (secs < 0)
Victor Stinner79d68f92015-03-19 21:54:09 +01001491 break;
1492 /* retry with the recomputed delay */
1493 } while (1);
1494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001495 return 0;
Guido van Rossum80c9d881991-04-16 08:47:51 +00001496}