1 files changed, 49 insertions, 0 deletions
diff --git a/unix/modtime.c b/unix/modtime.c
new file mode 100644
index 0000000000..032528947c
--- /dev/null
+++ b/unix/modtime.c
@@ -0,0 +1,49 @@
+#include <unistd.h>
+#include <string.h>
+#include <time.h>
+#include <sys/time.h>
+#include <math.h>
+
+#include "misc.h"
+#include "mpconfig.h"
+#include "qstr.h"
+#include "obj.h"
+#include "runtime.h"
+
+STATIC mp_obj_t mod_time_time() {
+    return mp_obj_new_int((machine_int_t)time(NULL));
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_0(mod_time_time_obj, mod_time_time);
+
+// Note: this is deprecated since CPy3.3, but pystone still uses it.
+STATIC mp_obj_t mod_time_clock() {
+//    return mp_obj_new_int((machine_int_t)clock());
+    // POSIX requires CLOCKS_PER_SEC equals 1000000, so that's what we assume
+    // float cannot represent full range of int32 precisely, so we pre-divide
+    // int to reduce resolution, and then actually do float division hoping
+    // to preserve integer part resolution.
+    return mp_obj_new_float((float)(clock() / 1000) / 1000.0);
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_0(mod_time_clock_obj, mod_time_clock);
+
+STATIC mp_obj_t mod_time_sleep(mp_obj_t arg) {
+#if MICROPY_ENABLE_FLOAT
+    struct timeval tv;
+    mp_float_t val = mp_obj_get_float(arg);
+    double ipart;
+    tv.tv_usec = round(modf(val, &ipart) * 1000000);
+    tv.tv_sec = ipart;
+    select(0, NULL, NULL, NULL, &tv);
+#else
+    sleep(mp_obj_get_int(arg));
+#endif
+    return mp_const_none;
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_time_sleep_obj, mod_time_sleep);
+
+void time_init() {
+    mp_obj_t m = mp_obj_new_module(QSTR_FROM_STR_STATIC("time"));
+    mp_store_attr(m, QSTR_FROM_STR_STATIC("time"), (mp_obj_t)&mod_time_time_obj);
+    mp_store_attr(m, QSTR_FROM_STR_STATIC("clock"), (mp_obj_t)&mod_time_clock_obj);
+    mp_store_attr(m, QSTR_FROM_STR_STATIC("sleep"), (mp_obj_t)&mod_time_sleep_obj);
+}