add: 添加函数fota.file, 从文件读取升级数据

wendal · wendal · commit 42cc7efca780 · 2023-03-23T22:30:59.000+08:00
diff --git a/luat/modules/luat_lib_fota.c b/luat/modules/luat_lib_fota.c
@@ -11,6 +11,9 @@
 #include "luat_fota.h"
 #include "luat_zbuff.h"
 #include "luat_spi.h"
+#include "luat_fs.h"
+#include "luat_malloc.h"
+
 #define LUAT_LOG_TAG "fota"
 #include "luat_log.h"
 
@@ -108,6 +111,79 @@ static int l_fota_write(lua_State* L)
 	return 3;
 }
 
+/**
+&#20174;&#25351;&#23450;&#25991;&#20214;&#35835;&#21462;fota&#25968;&#25454;
+@api fota.file(path)
+@string &#25991;&#20214;&#36335;&#24452;
+@return boolean &#26377;&#24322;&#24120;&#36820;&#22238;false&#65292;&#26080;&#24322;&#24120;&#36820;&#22238;true
+@return boolean &#25509;&#25910;&#21040;&#26368;&#21518;&#19968;&#22359;&#36820;&#22238;true
+@return int &#36824;&#26410;&#20889;&#20837;&#30340;&#25968;&#25454;&#37327;&#65292;&#36229;&#36807;64K&#24517;&#39035;&#20570;&#31561;&#24453;
+@usage
+local result, isDone, cache = fota.file("/xxx.bin") -- &#20889;&#20837;fota&#27969;&#31243;
+-- &#26412;API&#20110;2023.03.23 &#28155;&#21152;
+*/
+static int l_fota_file(lua_State* L)
+{
+    int result = 0;
+	const char *path = luaL_checkstring(L, 1);
+    FILE* fd = luat_fs_fopen(path, "rb");
+    if (fd == NULL) {
+        LLOGE("no such file for FOTA %s", path);
+        lua_pushboolean(L, 0);
+    	lua_pushboolean(L, 0);
+        lua_pushinteger(L, 0);
+        return 3;
+    }
+    #define BUFF_SIZE (4096)
+    char buff = luat_heap_malloc(BUFF_SIZE);
+    if (buff == NULL) {
+        luat_fs_fclose(fd);
+        LLOGE("out of memory when reading file %s", path);
+        lua_pushboolean(L, 0);
+    	lua_pushboolean(L, 0);
+        lua_pushinteger(L, 0);
+        return 3;
+    }
+    size_t len  = 0;
+    while (1) {
+        len = luat_fs_fread(buff + len, BUFF_SIZE - len, 1, fd);
+        if (len < 1) {
+            // EOF &#32467;&#26463;&#20102;
+            break;
+        }
+        result = luat_fota_write((uint8_t*)buff, len);
+        if (result < 0) {
+            break;
+        }
+        result = len;
+        if (len >= BUFF_SIZE) {
+            LLOGD("too many data to write! bug??");
+            result = -7;
+            break;
+        }
+    }
+    luat_heap_free(buff);
+    luat_fs_fclose(fd);
+
+    if (result > 0)
+    {
+    	lua_pushboolean(L, 1);
+    	lua_pushboolean(L, 0);
+    }
+    else if (result == 0)
+    {
+    	lua_pushboolean(L, 1);
+    	lua_pushboolean(L, 1);
+    }
+    else
+    {
+    	lua_pushboolean(L, 0);
+    	lua_pushboolean(L, 1);
+    }
+    lua_pushinteger(L, result);
+	return 3;
+}
+
 /**
 &#31561;&#24453;&#24213;&#23618;fota&#27969;&#31243;&#23436;&#25104;
 @api fota.isDone()
@@ -160,6 +236,7 @@ static const rotable_Reg_t reg_fota[] =
 	{ "run",		ROREG_FUNC(l_fota_write)},
 	{ "isDone",		ROREG_FUNC(l_fota_done)},
 	{ "finish",		ROREG_FUNC(l_fota_end)},
+    { "file",       ROREG_FUNC(l_fota_file)},
 	{ NULL,         ROREG_INT(0) }
 };
 
