LuatOS/olddemo/socket/Air105/ota_demo.lua

-- 用http方式下载ota固件并实现升级
-- 注意下载速度比写flash速度要快的多，而ram有限，w5500目前无法控制下载速度，所以需要先擦除flash，预估一定时间后，才能开始下载
--[[
-- 这个是用uart3直接把升级包发给mcu，作为本地升级的一种参考，高速串口单次发送不要超过4K
local rbuff = zbuff.create(16384)
local function uartRx(id, len)
    uart.rx(id, rbuff)
    if rbuff:used() > 0 then 
        local succ,fotaDone,nCache = fota.run(rbuff)
        
        if succ then
            
        else
            log.error("fota写入异常，请至少在1秒后重试")
            fota.finish(false)
            return
        end
        rbuff:del()
        -- log.info("收到服务器数据，长度", rbuff:used(), "fota结果", succ, done, "总共", filelen)
        if fotaDone then
            log.info("下载完成")
            sys.publish("downloadOK")
        end
    end
end

local function otaTask()
    local spiFlash = spi.deviceSetup(spi.SPI_1,pin.PA7,0,0,8,24*1000*1000,spi.MSB,1,1)
    fota.init(0, 0x00200000, spiFlash)
    while not fota.wait() do
        sys.wait(100)
    end
    
    uart.setup(3,1500000)
    uart.on(3, "receive", uartRx)
    uart.write(3,"ready")
    sys.waitUntil("downloadOK")                       
    while true do
        local succ,fotaDone  = fota.isDone()
        if fotaDone then
            fota.finish(true)
            log.info("FOTA完成")
            rtos.reboot()   --如果还有其他事情要做，就不要立刻reboot
            break
        end
        sys.wait(100)
    end
end
function otaDemo()
    sys.taskInit(otaTask)
end
]]


local libnet = require "libnet"

local taskName = "OTA_TASK"
local function netCB(msg)
    log.info("未处理消息", msg[1], msg[2], msg[3], msg[4])
end

local function otaTask()
    local spiFlash = spi.deviceSetup(spi.SPI_1,pin.PA7,0,0,8,24*1000*1000,spi.MSB,1,1)
    fota.init(0, 0x00200000, spiFlash)
    while not fota.wait() do
        sys.wait(100)
    end
    -- 如果确定只下载脚本，可以下载到文件系统里
    -- fota.init("/update.bin", 0, nil)
    -- os.remove("/update.bin")
    local succ, param, ip, port, total, findhead, filelen, rcvCache,d1,d2,statusCode,retry,rspHead,rcvChunked,done,fotaDone,nCache
    local tbuff = zbuff.create(512)
    local rbuff = zbuff.create(2048)
    local netc = socket.create(socket.ETH0, taskName)
    socket.config(netc, nil, nil, nil) -- http用的普通TCP连接
    -- 用的iot平台，用芯片名称+24byte id（实际上是12byte,96bit）
    -- 当然也可以用自己设置的id规则
    local imei,_ = mcu.unique_id():toHex()
    imei = rtos.bsp() .. string.sub(imei, 9, 32)
    log.info(imei, rtos.firmware())
    filelen = 0
    total = 0
    retry = 0
    done = false
    rspHead = {}
    local result = libnet.waitLink(taskName, 0, netc)
    while retry < 3 and not done do
        result = libnet.connect(taskName, 5000, netc, "iot.openluat.com", 80) --后续出了http库则直接用http来处理
        tbuff:del()
        -- 用的iot平台，所以固件名称和版本号需要对应处理
        -- 用的自建平台，可以自主定制规则
        local v = rtos.version()
        v = tonumber(v:sub(2, 5))
        tbuff:copy(0, "GET /api/site/firmware_upgrade?project_key=" .. _G.PRODUCT_KEY .. "&imei=".. imei .. "&device_key=&firmware_name=" .. _G.PROJECT.. "_LuatOS-SoC_" .. rtos.bsp() .. "&version=" .. v .. "." .. _G.VERSION .. " HTTP/1.1\r\n")
        tbuff:copy(nil,"Host: iot.openluat.com:80\r\n")
        if filelen > 0 then --断网重连的话，只需要下载剩余的部分就行了
            tbuff:copy(nil,"Range: bytes=" .. total .. "-\r\n") 
        end
        
        tbuff:copy(nil,"Accept: application/octet-stream\r\n\r\n")
        log.info(tbuff:query())
        result = libnet.tx(taskName, 5000, netc, tbuff)
        rbuff:del()
        findhead = false
        while result do
            succ, param, ip, port = socket.rx(netc, rbuff)
            if not succ then
                log.info("服务器断开了", succ, param, ip, port)
                break
            end
            if rbuff:used() > 0 then
                if findhead then
                    succ,fotaDone,nCache = fota.run(rbuff)
                    
                    if succ then
                        total = total + rbuff:used()
                    else
                        log.error("fota写入异常，请至少在1秒后重试")
                        fota.finish(false)
                        done = true
                        break
                    end
                    rbuff:del()
                    -- log.info("收到服务器数据，长度", rbuff:used(), "fota结果", succ, done, "总共", filelen)
                    if fotaDone then
                        log.info("下载完成")
                        while true do
                            succ,fotaDone  = fota.isDone()
                            if fotaDone then
                                fota.finish(true)
                                log.info("FOTA完成")
                                done = true
                                rtos.reboot()   --如果还有其他事情要做，就不要立刻reboot
                                break
                            end
                            sys.wait(100)
                        end
                        break
                    end
                else
                    rcvCache = rbuff:query()
                    d1,d2 = rcvCache:find("\r\n\r\n")
                    -- 打印出http response head
                    -- log.info(rcvCache:sub(1, d2))    
                    if d2 then
                        --状态行
                        _,d1,statusCode = rcvCache:find("%s(%d+)%s.-\r\n")
                        if not statusCode then
                            log.info("http没有状态返回")
                            break
                        end
                        statusCode = tonumber(statusCode)
                        if statusCode ~= 200 and statusCode ~= 206 then
                            log.info("http应答不OK", statusCode)
                            done = true
                            break
                        end
                        --应答头
                        for k,v in string.gmatch(rcvCache:sub(d1+1,d2-2),"(.-):%s*(.-)\r\n") do
                            rspHead[k] = v
                            if (string.upper(k)==string.upper("Transfer-Encoding")) and (string.upper(v)==string.upper("chunked")) then rcvChunked = true end
                        end
                        if filelen == 0 and not rcvChunked then 
                            if not rcvChunked then
                                filelen = tonumber(rspHead["Content-Length"] or "2147483647")
                            end
                        end
                        --未处理的body数据
                        rbuff:del(0, d2)
                        succ,fotaDone,nCache = fota.run(rbuff)
                        if not succ then
                            total = total + rbuff:used()
                        else
                            log.error("fota写入异常，请至少在1秒后重试")
                            fota.finish(false)
                            done = true
                            break
                        end
                        rbuff:del()
                        -- log.info("收到服务器数据，长度", rbuff:used(), "fota结果", succ, done, "总共", filelen)
                    else
                        break
                    end
                    findhead = true
                end
            end 
            result, param = libnet.wait(taskName, 5000, netc)
            if not result then
                log.info("服务器断开了", result, param)
                break
            elseif not param then
                log.info("服务器没有数据", result, param)
                break
            end
        end
        libnet.close(taskName, 5000, netc)
        retry = retry + 1
    end
    socket.release(netc)
    sysplus.taskDel(taskName)
    fota.finish(false)
end

function otaDemo()
    sysplus.taskInitEx(otaTask, taskName, netCB)
end