wifi/cell driver improved

1 files changed, 8 insertions, 3 deletions

diff --git a/fw/fe310/eos/net/pwr.c b/fw/fe310/eos/net/pwr.c
index 734e3cd..89f1817 100644
--- a/fw/fe310/eos/net/pwr.c
+++ b/fw/fe310/eos/net/pwr.c
@@ -8,6 +8,7 @@
 #include "soc/pwr.h"
 #include "soc/spi.h"
 #include "dev/spi.h"
+#include "dev/lcd.h"
 #include "eve/eve.h"
 
 #include "pwr.h"
@@ -42,9 +43,13 @@ static void pwr_handle_btn(unsigned char type, unsigned char *buffer, uint16_t l
     }
     if (!power_btn_down) return;
 
-    eos_spi_select(EOS_SPI_DEV_EVE);
-    eve_pwr_sleep();
-    eos_spi_deselect();
+    rv = eos_lcd_sleep();
+
+    rv = eos_spi_select(EOS_SPI_DEV_EVE);
+    if (!rv) {
+        eve_pwr_sleep();
+        eos_spi_deselect();
+    }
 
     rv = eos_pwr_sleep();
 }