blob: ebd44ad525bd81884584f37a15d5cf8dec454869 [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{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000720 PyObject *strptime_module = PyImport_ImportModuleNoBlock("_strptime");
721 PyObject *strptime_result;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200722 _Py_IDENTIFIER(_strptime_time);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000723
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000724 if (!strptime_module)
725 return NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200726 strptime_result = _PyObject_CallMethodId(strptime_module,
727 &PyId__strptime_time, "O", args);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000728 Py_DECREF(strptime_module);
729 return strptime_result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000730}
731
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000732
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000733PyDoc_STRVAR(strptime_doc,
Brett Cannon20def8b2003-07-01 05:16:08 +0000734"strptime(string, format) -> struct_time\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000735\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000736Parse a string to a time tuple according to a format specification.\n\
Alexander Belopolskycf774542012-10-02 18:39:16 -0400737See the library reference manual for formatting codes (same as\n\
738strftime()).\n\
739\n" STRFTIME_FORMAT_CODES);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000740
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000741static PyObject *
742_asctime(struct tm *timeptr)
743{
744 /* Inspired by Open Group reference implementation available at
745 * http://pubs.opengroup.org/onlinepubs/009695399/functions/asctime.html */
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200746 static const char wday_name[7][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000747 "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
748 };
Serhiy Storchaka2d06e842015-12-25 19:53:18 +0200749 static const char mon_name[12][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000750 "Jan", "Feb", "Mar", "Apr", "May", "Jun",
751 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
752 };
Victor Stinner499dfcf2011-03-21 13:26:24 +0100753 return PyUnicode_FromFormat(
754 "%s %s%3d %.2d:%.2d:%.2d %d",
755 wday_name[timeptr->tm_wday],
756 mon_name[timeptr->tm_mon],
757 timeptr->tm_mday, timeptr->tm_hour,
758 timeptr->tm_min, timeptr->tm_sec,
759 1900 + timeptr->tm_year);
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000760}
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000761
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000762static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000763time_asctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000764{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000765 PyObject *tup = NULL;
766 struct tm buf;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000767
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000768 if (!PyArg_UnpackTuple(args, "asctime", 0, 1, &tup))
769 return NULL;
770 if (tup == NULL) {
771 time_t tt = time(NULL);
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400772 if (_PyTime_localtime(tt, &buf) != 0)
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100773 return NULL;
774
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000775 } else if (!gettmarg(tup, &buf) || !checktm(&buf))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000776 return NULL;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000777 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000778}
779
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000780PyDoc_STRVAR(asctime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000781"asctime([tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000782\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000783Convert a time tuple to a string, e.g. 'Sat Jun 06 16:26:11 1998'.\n\
784When the time tuple is not present, current time as returned by localtime()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000785is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000786
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000787static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000788time_ctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000789{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000790 time_t tt;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100791 struct tm buf;
792 if (!parse_time_t_args(args, "|O:ctime", &tt))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000793 return NULL;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -0400794 if (_PyTime_localtime(tt, &buf) != 0)
Alexander Belopolsky5da468f2011-01-04 17:15:52 +0000795 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100796 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000797}
798
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000799PyDoc_STRVAR(ctime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000800"ctime(seconds) -> string\n\
801\n\
802Convert a time in seconds since the Epoch to a string in local time.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000803This is equivalent to asctime(localtime(seconds)). When the time tuple is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000804not present, current time as returned by localtime() is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000805
Guido van Rossum60cd8131998-03-06 17:16:21 +0000806#ifdef HAVE_MKTIME
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000807static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100808time_mktime(PyObject *self, PyObject *tup)
Guido van Rossum234f9421993-06-17 12:35:49 +0000809{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000810 struct tm buf;
811 time_t tt;
812 if (!gettmarg(tup, &buf))
813 return NULL;
Victor Stinner1ac42612014-02-21 09:27:17 +0100814#ifdef _AIX
815 /* year < 1902 or year > 2037 */
816 if (buf.tm_year < 2 || buf.tm_year > 137) {
817 /* Issue #19748: On AIX, mktime() doesn't report overflow error for
818 * timestamp < -2^31 or timestamp > 2**31-1. */
819 PyErr_SetString(PyExc_OverflowError,
820 "mktime argument out of range");
821 return NULL;
822 }
823#else
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000824 buf.tm_wday = -1; /* sentinel; original value ignored */
Victor Stinner1ac42612014-02-21 09:27:17 +0100825#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000826 tt = mktime(&buf);
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000827 /* Return value of -1 does not necessarily mean an error, but tm_wday
Ezio Melotti13925002011-03-16 11:05:33 +0200828 * cannot remain set to -1 if mktime succeeded. */
Victor Stinner93037492013-06-25 22:54:35 +0200829 if (tt == (time_t)(-1)
830#ifndef _AIX
831 /* Return value of -1 does not necessarily mean an error, but
832 * tm_wday cannot remain set to -1 if mktime succeeded. */
833 && buf.tm_wday == -1
834#else
835 /* on AIX, tm_wday is always sets, even on error */
836#endif
837 )
838 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000839 PyErr_SetString(PyExc_OverflowError,
840 "mktime argument out of range");
841 return NULL;
842 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100843 return PyFloat_FromDouble((double)tt);
Guido van Rossum234f9421993-06-17 12:35:49 +0000844}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000845
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000846PyDoc_STRVAR(mktime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000847"mktime(tuple) -> floating point number\n\
848\n\
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400849Convert a time tuple in local time to seconds since the Epoch.\n\
850Note that mktime(gmtime(0)) will not generally return zero for most\n\
851time zones; instead the returned value will either be equal to that\n\
852of the timezone or altzone attributes on the time module.");
Guido van Rossum60cd8131998-03-06 17:16:21 +0000853#endif /* HAVE_MKTIME */
Guido van Rossum234f9421993-06-17 12:35:49 +0000854
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000855#ifdef HAVE_WORKING_TZSET
Martin v. Löwis1a214512008-06-11 05:26:20 +0000856static void PyInit_timezone(PyObject *module);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000857
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000858static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000859time_tzset(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000860{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000861 PyObject* m;
Fred Drake9bb74322002-04-01 14:49:59 +0000862
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000863 m = PyImport_ImportModuleNoBlock("time");
864 if (m == NULL) {
865 return NULL;
866 }
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000867
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000868 tzset();
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000869
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000870 /* Reset timezone, altzone, daylight and tzname */
871 PyInit_timezone(m);
872 Py_DECREF(m);
Victor Stinner2ff51b82013-07-17 21:42:45 +0200873 if (PyErr_Occurred())
874 return NULL;
Tim Peters1b6f7a92004-06-20 02:50:16 +0000875
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000876 Py_INCREF(Py_None);
877 return Py_None;
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000878}
879
880PyDoc_STRVAR(tzset_doc,
R. David Murray4d55bf92010-12-14 00:55:46 +0000881"tzset()\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000882\n\
883Initialize, or reinitialize, the local timezone to the value stored in\n\
884os.environ['TZ']. The TZ environment variable should be specified in\n\
Neal Norwitzdc8e1942004-07-20 22:34:37 +0000885standard Unix timezone format as documented in the tzset man page\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000886(eg. 'US/Eastern', 'Europe/Amsterdam'). Unknown timezones will silently\n\
887fall back to UTC. If the TZ environment variable is not set, the local\n\
888timezone is set to the systems best guess of wallclock time.\n\
889Changing the TZ environment variable without calling tzset *may* change\n\
890the local timezone used by methods such as localtime, but this behaviour\n\
891should not be relied on.");
892#endif /* HAVE_WORKING_TZSET */
893
Victor Stinnerae586492014-09-02 23:18:25 +0200894static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200895pymonotonic(_Py_clock_info_t *info)
Victor Stinnerb94b2662012-01-18 01:50:21 +0100896{
Victor Stinner4bfb4602015-03-27 22:27:24 +0100897 _PyTime_t t;
898 double d;
899 if (_PyTime_GetMonotonicClockWithInfo(&t, info) < 0) {
Victor Stinnerae586492014-09-02 23:18:25 +0200900 assert(info != NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100901 return NULL;
902 }
Victor Stinner4bfb4602015-03-27 22:27:24 +0100903 d = _PyTime_AsSecondsDouble(t);
904 return PyFloat_FromDouble(d);
Victor Stinner8b302012012-02-07 23:29:46 +0100905}
906
Victor Stinner071eca32012-03-15 01:17:09 +0100907static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200908time_monotonic(PyObject *self, PyObject *unused)
Victor Stinner071eca32012-03-15 01:17:09 +0100909{
Victor Stinnerec895392012-04-29 02:41:27 +0200910 return pymonotonic(NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100911}
912
Victor Stinnerec895392012-04-29 02:41:27 +0200913PyDoc_STRVAR(monotonic_doc,
914"monotonic() -> float\n\
Victor Stinner8b302012012-02-07 23:29:46 +0100915\n\
Victor Stinnerec895392012-04-29 02:41:27 +0200916Monotonic clock, cannot go backward.");
Victor Stinnerec919cc2012-03-15 00:58:32 +0100917
Victor Stinnerec895392012-04-29 02:41:27 +0200918static PyObject*
919perf_counter(_Py_clock_info_t *info)
920{
Victor Stinner54884492014-08-29 16:51:33 +0200921#ifdef WIN32_PERF_COUNTER
922 return win_perf_counter(info);
923#else
Victor Stinnerae586492014-09-02 23:18:25 +0200924 return pymonotonic(info);
Victor Stinner54884492014-08-29 16:51:33 +0200925#endif
Victor Stinnerec895392012-04-29 02:41:27 +0200926}
927
928static PyObject *
929time_perf_counter(PyObject *self, PyObject *unused)
930{
931 return perf_counter(NULL);
932}
933
934PyDoc_STRVAR(perf_counter_doc,
935"perf_counter() -> float\n\
936\n\
937Performance counter for benchmarking.");
938
939static PyObject*
940py_process_time(_Py_clock_info_t *info)
941{
942#if defined(MS_WINDOWS)
943 HANDLE process;
944 FILETIME creation_time, exit_time, kernel_time, user_time;
945 ULARGE_INTEGER large;
946 double total;
947 BOOL ok;
948
949 process = GetCurrentProcess();
950 ok = GetProcessTimes(process, &creation_time, &exit_time, &kernel_time, &user_time);
951 if (!ok)
952 return PyErr_SetFromWindowsErr(0);
953
954 large.u.LowPart = kernel_time.dwLowDateTime;
955 large.u.HighPart = kernel_time.dwHighDateTime;
956 total = (double)large.QuadPart;
957 large.u.LowPart = user_time.dwLowDateTime;
958 large.u.HighPart = user_time.dwHighDateTime;
959 total += (double)large.QuadPart;
960 if (info) {
961 info->implementation = "GetProcessTimes()";
962 info->resolution = 1e-7;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400963 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200964 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200965 }
966 return PyFloat_FromDouble(total * 1e-7);
967#else
968
969#if defined(HAVE_SYS_RESOURCE_H)
970 struct rusage ru;
971#endif
972#ifdef HAVE_TIMES
973 struct tms t;
974 static long ticks_per_second = -1;
975#endif
976
977#if defined(HAVE_CLOCK_GETTIME) \
978 && (defined(CLOCK_PROCESS_CPUTIME_ID) || defined(CLOCK_PROF))
979 struct timespec tp;
980#ifdef CLOCK_PROF
981 const clockid_t clk_id = CLOCK_PROF;
982 const char *function = "clock_gettime(CLOCK_PROF)";
983#else
984 const clockid_t clk_id = CLOCK_PROCESS_CPUTIME_ID;
985 const char *function = "clock_gettime(CLOCK_PROCESS_CPUTIME_ID)";
986#endif
987
988 if (clock_gettime(clk_id, &tp) == 0) {
989 if (info) {
990 struct timespec res;
991 info->implementation = function;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400992 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200993 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200994 if (clock_getres(clk_id, &res) == 0)
995 info->resolution = res.tv_sec + res.tv_nsec * 1e-9;
996 else
997 info->resolution = 1e-9;
998 }
999 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
1000 }
1001#endif
1002
1003#if defined(HAVE_SYS_RESOURCE_H)
1004 if (getrusage(RUSAGE_SELF, &ru) == 0) {
1005 double total;
1006 total = ru.ru_utime.tv_sec + ru.ru_utime.tv_usec * 1e-6;
1007 total += ru.ru_stime.tv_sec + ru.ru_stime.tv_usec * 1e-6;
1008 if (info) {
1009 info->implementation = "getrusage(RUSAGE_SELF)";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001010 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001011 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001012 info->resolution = 1e-6;
1013 }
1014 return PyFloat_FromDouble(total);
1015 }
1016#endif
1017
1018#ifdef HAVE_TIMES
1019 if (times(&t) != (clock_t)-1) {
1020 double total;
1021
1022 if (ticks_per_second == -1) {
1023#if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
1024 ticks_per_second = sysconf(_SC_CLK_TCK);
1025 if (ticks_per_second < 1)
1026 ticks_per_second = -1;
1027#elif defined(HZ)
1028 ticks_per_second = HZ;
1029#else
1030 ticks_per_second = 60; /* magic fallback value; may be bogus */
1031#endif
1032 }
1033
1034 if (ticks_per_second != -1) {
1035 total = (double)t.tms_utime / ticks_per_second;
1036 total += (double)t.tms_stime / ticks_per_second;
1037 if (info) {
1038 info->implementation = "times()";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001039 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001040 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001041 info->resolution = 1.0 / ticks_per_second;
1042 }
1043 return PyFloat_FromDouble(total);
1044 }
1045 }
1046#endif
1047
Victor Stinner53e22bf2016-07-08 17:55:01 +02001048 /* Currently, Python 3 requires clock() to build: see issue #22624 */
Victor Stinnerec895392012-04-29 02:41:27 +02001049 return floatclock(info);
1050#endif
1051}
1052
1053static PyObject *
1054time_process_time(PyObject *self, PyObject *unused)
1055{
1056 return py_process_time(NULL);
1057}
1058
1059PyDoc_STRVAR(process_time_doc,
1060"process_time() -> float\n\
1061\n\
1062Process time for profiling: sum of the kernel and user-space CPU time.");
1063
1064
Victor Stinnerec895392012-04-29 02:41:27 +02001065static PyObject *
1066time_get_clock_info(PyObject *self, PyObject *args)
1067{
1068 char *name;
Victor Stinnerec895392012-04-29 02:41:27 +02001069 _Py_clock_info_t info;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001070 PyObject *obj = NULL, *dict, *ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001071
1072 if (!PyArg_ParseTuple(args, "s:get_clock_info", &name))
1073 return NULL;
1074
1075#ifdef Py_DEBUG
1076 info.implementation = NULL;
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001077 info.monotonic = -1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001078 info.adjustable = -1;
Victor Stinnerec895392012-04-29 02:41:27 +02001079 info.resolution = -1.0;
1080#else
1081 info.implementation = "";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001082 info.monotonic = 0;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001083 info.adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001084 info.resolution = 1.0;
1085#endif
1086
1087 if (strcmp(name, "time") == 0)
1088 obj = floattime(&info);
1089#ifdef PYCLOCK
1090 else if (strcmp(name, "clock") == 0)
1091 obj = pyclock(&info);
1092#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001093 else if (strcmp(name, "monotonic") == 0)
1094 obj = pymonotonic(&info);
Victor Stinnerec895392012-04-29 02:41:27 +02001095 else if (strcmp(name, "perf_counter") == 0)
1096 obj = perf_counter(&info);
1097 else if (strcmp(name, "process_time") == 0)
1098 obj = py_process_time(&info);
1099 else {
1100 PyErr_SetString(PyExc_ValueError, "unknown clock");
1101 return NULL;
1102 }
1103 if (obj == NULL)
1104 return NULL;
1105 Py_DECREF(obj);
1106
Victor Stinnerbda4b882012-06-12 22:11:44 +02001107 dict = PyDict_New();
1108 if (dict == NULL)
Victor Stinnerec895392012-04-29 02:41:27 +02001109 return NULL;
1110
1111 assert(info.implementation != NULL);
1112 obj = PyUnicode_FromString(info.implementation);
1113 if (obj == NULL)
1114 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001115 if (PyDict_SetItemString(dict, "implementation", obj) == -1)
1116 goto error;
1117 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001118
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001119 assert(info.monotonic != -1);
1120 obj = PyBool_FromLong(info.monotonic);
Victor Stinnerec895392012-04-29 02:41:27 +02001121 if (obj == NULL)
1122 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001123 if (PyDict_SetItemString(dict, "monotonic", obj) == -1)
1124 goto error;
1125 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001126
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001127 assert(info.adjustable != -1);
1128 obj = PyBool_FromLong(info.adjustable);
Victor Stinnerec895392012-04-29 02:41:27 +02001129 if (obj == NULL)
1130 goto error;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001131 if (PyDict_SetItemString(dict, "adjustable", obj) == -1)
Victor Stinnerbda4b882012-06-12 22:11:44 +02001132 goto error;
1133 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001134
1135 assert(info.resolution > 0.0);
1136 assert(info.resolution <= 1.0);
1137 obj = PyFloat_FromDouble(info.resolution);
1138 if (obj == NULL)
1139 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001140 if (PyDict_SetItemString(dict, "resolution", obj) == -1)
1141 goto error;
1142 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001143
Victor Stinnerbda4b882012-06-12 22:11:44 +02001144 ns = _PyNamespace_New(dict);
1145 Py_DECREF(dict);
1146 return ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001147
1148error:
Victor Stinnerbda4b882012-06-12 22:11:44 +02001149 Py_DECREF(dict);
1150 Py_XDECREF(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001151 return NULL;
1152}
1153
1154PyDoc_STRVAR(get_clock_info_doc,
1155"get_clock_info(name: str) -> dict\n\
1156\n\
1157Get information of the specified clock.");
Victor Stinner8b302012012-02-07 23:29:46 +01001158
Martin v. Löwisd218dc12008-04-07 03:17:54 +00001159static void
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001160get_zone(char *zone, int n, struct tm *p)
1161{
1162#ifdef HAVE_STRUCT_TM_TM_ZONE
1163 strncpy(zone, p->tm_zone ? p->tm_zone : " ", n);
1164#else
1165 tzset();
1166 strftime(zone, n, "%Z", p);
1167#endif
1168}
1169
1170static int
1171get_gmtoff(time_t t, struct tm *p)
1172{
1173#ifdef HAVE_STRUCT_TM_TM_ZONE
1174 return p->tm_gmtoff;
1175#else
1176 return timegm(p) - t;
1177#endif
1178}
1179
1180static void
Martin v. Löwis1a214512008-06-11 05:26:20 +00001181PyInit_timezone(PyObject *m) {
1182 /* This code moved from PyInit_time wholesale to allow calling it from
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001183 time_tzset. In the future, some parts of it can be moved back
1184 (for platforms that don't HAVE_WORKING_TZSET, when we know what they
1185 are), and the extraneous calls to tzset(3) should be removed.
1186 I haven't done this yet, as I don't want to change this code as
1187 little as possible when introducing the time.tzset and time.tzsetwall
1188 methods. This should simply be a method of doing the following once,
1189 at the top of this function and removing the call to tzset() from
1190 time_tzset():
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001191
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001192 #ifdef HAVE_TZSET
1193 tzset()
1194 #endif
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001195
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001196 And I'm lazy and hate C so nyer.
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001197 */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001198#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001199 PyObject *otz0, *otz1;
1200 tzset();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001201 PyModule_AddIntConstant(m, "timezone", timezone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001202#ifdef HAVE_ALTZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001203 PyModule_AddIntConstant(m, "altzone", altzone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001204#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001205 PyModule_AddIntConstant(m, "altzone", timezone-3600);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001206#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001207 PyModule_AddIntConstant(m, "daylight", daylight);
Victor Stinner1b579672011-12-17 05:47:23 +01001208 otz0 = PyUnicode_DecodeLocale(tzname[0], "surrogateescape");
1209 otz1 = PyUnicode_DecodeLocale(tzname[1], "surrogateescape");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001210 PyModule_AddObject(m, "tzname", Py_BuildValue("(NN)", otz0, otz1));
Guido van Rossum10b164a2001-09-25 13:59:01 +00001211#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001212 {
Guido van Rossum234f9421993-06-17 12:35:49 +00001213#define YEAR ((time_t)((365 * 24 + 6) * 3600))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 time_t t;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -04001215 struct tm p;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001216 long janzone, julyzone;
1217 char janname[10], julyname[10];
1218 t = (time((time_t *)0) / YEAR) * YEAR;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -04001219 _PyTime_localtime(t, &p);
1220 get_zone(janname, 9, &p);
1221 janzone = -get_gmtoff(t, &p);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001222 janname[9] = '\0';
1223 t += YEAR/2;
Alexander Belopolsky3e7a3cb2016-09-28 17:31:35 -04001224 _PyTime_localtime(t, &p);
1225 get_zone(julyname, 9, &p);
1226 julyzone = -get_gmtoff(t, &p);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001227 julyname[9] = '\0';
Guido van Rossum10b164a2001-09-25 13:59:01 +00001228
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001229 if( janzone < julyzone ) {
1230 /* DST is reversed in the southern hemisphere */
1231 PyModule_AddIntConstant(m, "timezone", julyzone);
1232 PyModule_AddIntConstant(m, "altzone", janzone);
1233 PyModule_AddIntConstant(m, "daylight",
1234 janzone != julyzone);
1235 PyModule_AddObject(m, "tzname",
1236 Py_BuildValue("(zz)",
1237 julyname, janname));
1238 } else {
1239 PyModule_AddIntConstant(m, "timezone", janzone);
1240 PyModule_AddIntConstant(m, "altzone", julyzone);
1241 PyModule_AddIntConstant(m, "daylight",
1242 janzone != julyzone);
1243 PyModule_AddObject(m, "tzname",
1244 Py_BuildValue("(zz)",
1245 janname, julyname));
1246 }
1247 }
Tim Peters26ae7cd2001-03-20 03:26:49 +00001248#ifdef __CYGWIN__
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001249 tzset();
1250 PyModule_AddIntConstant(m, "timezone", _timezone);
1251 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
1252 PyModule_AddIntConstant(m, "daylight", _daylight);
1253 PyModule_AddObject(m, "tzname",
1254 Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
Tim Peters26ae7cd2001-03-20 03:26:49 +00001255#endif /* __CYGWIN__ */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001256#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001257}
1258
1259
1260static PyMethodDef time_methods[] = {
Victor Stinner4195b5c2012-02-08 23:03:19 +01001261 {"time", time_time, METH_NOARGS, time_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001262#ifdef PYCLOCK
Victor Stinner4195b5c2012-02-08 23:03:19 +01001263 {"clock", time_clock, METH_NOARGS, clock_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001264#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001265#ifdef HAVE_CLOCK_GETTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001266 {"clock_gettime", time_clock_gettime, METH_VARARGS, clock_gettime_doc},
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001267#endif
1268#ifdef HAVE_CLOCK_SETTIME
Victor Stinner30d79472012-04-03 00:45:07 +02001269 {"clock_settime", time_clock_settime, METH_VARARGS, clock_settime_doc},
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001270#endif
1271#ifdef HAVE_CLOCK_GETRES
Victor Stinner4195b5c2012-02-08 23:03:19 +01001272 {"clock_getres", time_clock_getres, METH_VARARGS, clock_getres_doc},
Victor Stinnere0be4232011-10-25 13:06:09 +02001273#endif
Victor Stinnercb29f012015-03-27 13:31:18 +01001274 {"sleep", time_sleep, METH_O, sleep_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001275 {"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
1276 {"localtime", time_localtime, METH_VARARGS, localtime_doc},
1277 {"asctime", time_asctime, METH_VARARGS, asctime_doc},
1278 {"ctime", time_ctime, METH_VARARGS, ctime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001279#ifdef HAVE_MKTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001280 {"mktime", time_mktime, METH_O, mktime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001281#endif
1282#ifdef HAVE_STRFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001283 {"strftime", time_strftime, METH_VARARGS, strftime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001284#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001285 {"strptime", time_strptime, METH_VARARGS, strptime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001286#ifdef HAVE_WORKING_TZSET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001287 {"tzset", time_tzset, METH_NOARGS, tzset_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001288#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001289 {"monotonic", time_monotonic, METH_NOARGS, monotonic_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001290 {"process_time", time_process_time, METH_NOARGS, process_time_doc},
1291 {"perf_counter", time_perf_counter, METH_NOARGS, perf_counter_doc},
1292 {"get_clock_info", time_get_clock_info, METH_VARARGS, get_clock_info_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001293 {NULL, NULL} /* sentinel */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001294};
1295
1296
1297PyDoc_STRVAR(module_doc,
1298"This module provides various functions to manipulate time values.\n\
1299\n\
1300There are two standard representations of time. One is the number\n\
1301of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
1302or a floating point number (to represent fractions of seconds).\n\
1303The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
1304The actual value can be retrieved by calling gmtime(0).\n\
1305\n\
1306The other representation is a tuple of 9 integers giving local time.\n\
1307The tuple items are:\n\
Alexander Belopolsky03163ac2011-05-02 12:20:52 -04001308 year (including century, e.g. 1998)\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001309 month (1-12)\n\
1310 day (1-31)\n\
1311 hours (0-23)\n\
1312 minutes (0-59)\n\
1313 seconds (0-59)\n\
1314 weekday (0-6, Monday is 0)\n\
1315 Julian day (day in the year, 1-366)\n\
1316 DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
1317If the DST flag is 0, the time is given in the regular time zone;\n\
1318if it is 1, the time is given in the DST time zone;\n\
1319if it is -1, mktime() should guess based on the date and time.\n\
1320\n\
1321Variables:\n\
1322\n\
1323timezone -- difference in seconds between UTC and local standard time\n\
1324altzone -- difference in seconds between UTC and local DST time\n\
1325daylight -- whether local time should reflect DST\n\
1326tzname -- tuple of (standard time zone name, DST time zone name)\n\
1327\n\
1328Functions:\n\
1329\n\
1330time() -- return current time in seconds since the Epoch as a float\n\
1331clock() -- return CPU time since process start as a float\n\
1332sleep() -- delay for a number of seconds given as a float\n\
1333gmtime() -- convert seconds since Epoch to UTC tuple\n\
1334localtime() -- convert seconds since Epoch to local time tuple\n\
1335asctime() -- convert time tuple to string\n\
1336ctime() -- convert time in seconds to string\n\
1337mktime() -- convert local time tuple to seconds since Epoch\n\
1338strftime() -- convert time tuple to string according to format specification\n\
1339strptime() -- parse string to time tuple according to format specification\n\
1340tzset() -- change the local timezone");
1341
1342
Martin v. Löwis1a214512008-06-11 05:26:20 +00001343
1344static struct PyModuleDef timemodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001345 PyModuleDef_HEAD_INIT,
1346 "time",
1347 module_doc,
1348 -1,
1349 time_methods,
1350 NULL,
1351 NULL,
1352 NULL,
1353 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001354};
1355
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001356PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001357PyInit_time(void)
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001358{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001359 PyObject *m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001360 m = PyModule_Create(&timemodule);
1361 if (m == NULL)
1362 return NULL;
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001363
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001364 /* Set, or reset, module variables like time.timezone */
1365 PyInit_timezone(m);
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001366
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001367#ifdef CLOCK_REALTIME
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001368 PyModule_AddIntMacro(m, CLOCK_REALTIME);
Benjamin Peterson37098cd2016-09-13 22:55:09 -07001369#endif
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001370#ifdef CLOCK_MONOTONIC
1371 PyModule_AddIntMacro(m, CLOCK_MONOTONIC);
1372#endif
1373#ifdef CLOCK_MONOTONIC_RAW
1374 PyModule_AddIntMacro(m, CLOCK_MONOTONIC_RAW);
1375#endif
1376#ifdef CLOCK_HIGHRES
1377 PyModule_AddIntMacro(m, CLOCK_HIGHRES);
1378#endif
1379#ifdef CLOCK_PROCESS_CPUTIME_ID
1380 PyModule_AddIntMacro(m, CLOCK_PROCESS_CPUTIME_ID);
1381#endif
1382#ifdef CLOCK_THREAD_CPUTIME_ID
1383 PyModule_AddIntMacro(m, CLOCK_THREAD_CPUTIME_ID);
1384#endif
Alexander Belopolsky18f3a9b2016-09-11 22:55:16 -04001385
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001386 if (!initialized) {
Victor Stinner1c8f0592013-07-22 22:24:54 +02001387 if (PyStructSequence_InitType2(&StructTimeType,
1388 &struct_time_type_desc) < 0)
1389 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001390 }
1391 Py_INCREF(&StructTimeType);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -04001392 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 11);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001393 PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
1394 initialized = 1;
1395 return m;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001396}
1397
Victor Stinner071eca32012-03-15 01:17:09 +01001398static PyObject*
Victor Stinnerec895392012-04-29 02:41:27 +02001399floattime(_Py_clock_info_t *info)
Victor Stinner4195b5c2012-02-08 23:03:19 +01001400{
Victor Stinnera47b8812015-03-27 18:16:17 +01001401 _PyTime_t t;
1402 double d;
1403 if (_PyTime_GetSystemClockWithInfo(&t, info) < 0) {
Victor Stinner00111242014-08-29 16:31:59 +02001404 assert(info != NULL);
1405 return NULL;
1406 }
Victor Stinnera47b8812015-03-27 18:16:17 +01001407 d = _PyTime_AsSecondsDouble(t);
1408 return PyFloat_FromDouble(d);
Victor Stinner4195b5c2012-02-08 23:03:19 +01001409}
1410
1411
Victor Stinnercb29f012015-03-27 13:31:18 +01001412/* Implement pysleep() for various platforms.
Guido van Rossumb6775db1994-08-01 11:34:53 +00001413 When interrupted (or when another error occurs), return -1 and
1414 set an exception; else return 0. */
1415
1416static int
Victor Stinnercb29f012015-03-27 13:31:18 +01001417pysleep(_PyTime_t secs)
Guido van Rossum426035c1991-02-19 12:27:35 +00001418{
Victor Stinnercb29f012015-03-27 13:31:18 +01001419 _PyTime_t deadline, monotonic;
Victor Stinner79d68f92015-03-19 21:54:09 +01001420#ifndef MS_WINDOWS
1421 struct timeval timeout;
Victor Stinner79d68f92015-03-19 21:54:09 +01001422 int err = 0;
1423#else
Victor Stinnercb29f012015-03-27 13:31:18 +01001424 _PyTime_t millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001425 unsigned long ul_millis;
1426 DWORD rc;
1427 HANDLE hInterruptEvent;
Victor Stinner0c2fd892015-03-17 10:49:17 +01001428#endif
Victor Stinner79d68f92015-03-19 21:54:09 +01001429
Victor Stinnercb29f012015-03-27 13:31:18 +01001430 deadline = _PyTime_GetMonotonicClock() + secs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001431
1432 do {
1433#ifndef MS_WINDOWS
Victor Stinner869e1772015-03-30 03:49:14 +02001434 if (_PyTime_AsTimeval(secs, &timeout, _PyTime_ROUND_CEILING) < 0)
Victor Stinnercb29f012015-03-27 13:31:18 +01001435 return -1;
Victor Stinner79d68f92015-03-19 21:54:09 +01001436
1437 Py_BEGIN_ALLOW_THREADS
1438 err = select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &timeout);
1439 Py_END_ALLOW_THREADS
1440
1441 if (err == 0)
1442 break;
1443
1444 if (errno != EINTR) {
Victor Stinner0c2fd892015-03-17 10:49:17 +01001445 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001446 return -1;
1447 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001448#else
Victor Stinner869e1772015-03-30 03:49:14 +02001449 millisecs = _PyTime_AsMilliseconds(secs, _PyTime_ROUND_CEILING);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001450 if (millisecs > (double)ULONG_MAX) {
1451 PyErr_SetString(PyExc_OverflowError,
1452 "sleep length is too large");
1453 return -1;
1454 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001455
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001456 /* Allow sleep(0) to maintain win32 semantics, and as decreed
1457 * by Guido, only the main thread can be interrupted.
1458 */
1459 ul_millis = (unsigned long)millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001460 if (ul_millis == 0 || !_PyOS_IsMainThread()) {
1461 Py_BEGIN_ALLOW_THREADS
Victor Stinner0eac1302015-03-20 03:06:12 +01001462 Sleep(ul_millis);
Victor Stinner79d68f92015-03-19 21:54:09 +01001463 Py_END_ALLOW_THREADS
1464 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001465 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001466
1467 hInterruptEvent = _PyOS_SigintEvent();
1468 ResetEvent(hInterruptEvent);
1469
1470 Py_BEGIN_ALLOW_THREADS
1471 rc = WaitForSingleObjectEx(hInterruptEvent, ul_millis, FALSE);
Victor Stinner0c2fd892015-03-17 10:49:17 +01001472 Py_END_ALLOW_THREADS
Victor Stinner79d68f92015-03-19 21:54:09 +01001473
1474 if (rc != WAIT_OBJECT_0)
1475 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001476#endif
Victor Stinner0c2fd892015-03-17 10:49:17 +01001477
Victor Stinner79d68f92015-03-19 21:54:09 +01001478 /* sleep was interrupted by SIGINT */
1479 if (PyErr_CheckSignals())
1480 return -1;
1481
Victor Stinnercb29f012015-03-27 13:31:18 +01001482 monotonic = _PyTime_GetMonotonicClock();
1483 secs = deadline - monotonic;
Victor Stinner6aa446c2015-03-30 21:33:51 +02001484 if (secs < 0)
Victor Stinner79d68f92015-03-19 21:54:09 +01001485 break;
1486 /* retry with the recomputed delay */
1487 } while (1);
1488
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001489 return 0;
Guido van Rossum80c9d881991-04-16 08:47:51 +00001490}