From 050e645ef2380d14a2833edb509d4f7498f6d035 Mon Sep 17 00:00:00 2001
From: Paul Sokolovsky <pfalcon@users.sourceforge.net>
Date: Fri, 15 Apr 2016 22:07:28 +0300
Subject: [PATCH] esp8266/modmachine: Add reset_cause() function.

---
 esp8266/modmachine.c   | 6 ++++++
 esp8266/qstrdefsport.h | 1 +
 2 files changed, 7 insertions(+)

diff --git a/esp8266/modmachine.c b/esp8266/modmachine.c
index 8fb88fafd7..83fb8d3504 100644
--- a/esp8266/modmachine.c
+++ b/esp8266/modmachine.c
@@ -64,6 +64,11 @@ STATIC mp_obj_t machine_reset(void) {
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_0(machine_reset_obj, machine_reset);
 
+STATIC mp_obj_t machine_reset_cause(void) {
+    return MP_OBJ_NEW_SMALL_INT(system_get_rst_info()->reason);
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_0(machine_reset_cause_obj, machine_reset_cause);
+
 STATIC mp_obj_t machine_unique_id(void) {
     uint32_t id = system_get_chip_id();
     return mp_obj_new_bytes((byte*)&id, sizeof(id));
@@ -153,6 +158,7 @@ STATIC const mp_rom_map_elem_t machine_module_globals_table[] = {
 
     { MP_ROM_QSTR(MP_QSTR_freq), MP_ROM_PTR(&machine_freq_obj) },
     { MP_ROM_QSTR(MP_QSTR_reset), MP_ROM_PTR(&machine_reset_obj) },
+    { MP_ROM_QSTR(MP_QSTR_reset_cause), MP_ROM_PTR(&machine_reset_cause_obj) },
     { MP_ROM_QSTR(MP_QSTR_unique_id), MP_ROM_PTR(&machine_unique_id_obj) },
 
     { MP_ROM_QSTR(MP_QSTR_Timer), MP_ROM_PTR(&esp_timer_type) },
diff --git a/esp8266/qstrdefsport.h b/esp8266/qstrdefsport.h
index cb5168016b..34f7452069 100644
--- a/esp8266/qstrdefsport.h
+++ b/esp8266/qstrdefsport.h
@@ -214,6 +214,7 @@ Q(time)
 
 // machine
 Q(reset)
+Q(reset_cause)
 Q(Timer)
 Q(callback)
 Q(deinit)