57 lines
1.6 KiB
C++
57 lines
1.6 KiB
C++
#include <iostream>
|
|
#include <stdio.h>
|
|
#include "update.h"
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <sys/reboot.h>
|
|
#include "download.h"
|
|
|
|
bool update_ok = false;
|
|
void handle_upgrade_callback(void *user_data, RK_Upgrade_Status_t status){
|
|
if(status == RK_UPGRADE_FINISHED)
|
|
update_ok = true;
|
|
}
|
|
|
|
void handle_upgrade_callback_reboot(void *user_data, RK_Upgrade_Status_t status){
|
|
if(status == RK_UPGRADE_FINISHED){
|
|
update_ok = true;
|
|
reboot(RB_AUTOBOOT);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]){
|
|
//char url[256] = "http://148.70.52.169:8080/update.img";
|
|
printf("version 1.0.0\n");
|
|
if(argc >= 2){
|
|
if(strcmp(argv[1], "check") == 0){
|
|
if(RK_ota_check_version(argv[2])){
|
|
printf("version check ok, need update..\n");
|
|
return 0;
|
|
}
|
|
else{
|
|
return -1;
|
|
}
|
|
}else if(strcmp(argv[1], "update") == 0){
|
|
if(argc == 3){
|
|
RK_ota_set_url(argv[2]);
|
|
RK_ota_start(handle_upgrade_callback);
|
|
RK_ota_get_progress();
|
|
if(update_ok)
|
|
return 0;
|
|
else
|
|
return -1;
|
|
}else if(argc == 4){
|
|
RK_ota_set_url(argv[2]);
|
|
RK_ota_start(handle_upgrade_callback_reboot);
|
|
RK_ota_get_progress();
|
|
}
|
|
}
|
|
}else{
|
|
printf("============useage===============\n");
|
|
printf("1. update_engine check url\n");
|
|
printf("2. update_engine update url\n");
|
|
printf("3. update_engine update url reboot\n");
|
|
}
|
|
return 0;
|
|
}
|