From f6ab28ae5be33aa6b662cafc7033f06fe373039b Mon Sep 17 00:00:00 2001
From: "Kirill Lipatov (Leency)" <lipatov.kiril@gmail.com>
Date: Sun, 31 May 2020 14:37:09 +0000
Subject: [PATCH] osupdate: final

git-svn-id: svn://kolibrios.org@8013 a494cfbc-eb01-0410-851d-a64ba20cac60
---
 programs/cmm/installer/install.c            |   2 +-
 programs/cmm/lib/obj/libimg.h               |   1 -
 programs/cmm/lib/patterns/restart_process.h |  14 ++
 programs/cmm/misc/osupdate.c                | 190 ++++++++++++--------
 4 files changed, 128 insertions(+), 79 deletions(-)

diff --git a/programs/cmm/installer/install.c b/programs/cmm/installer/install.c
index e7001c0474..991e95ee9a 100644
--- a/programs/cmm/installer/install.c
+++ b/programs/cmm/installer/install.c
@@ -130,4 +130,4 @@ void EventInstall()
 	draw_window();
 }
 
-void Operation_Draw_Progress(dword filename) { debug("copying: "); debugln(filename); }
\ No newline at end of file
+void Operation_Draw_Progress(dword filename) { }
\ No newline at end of file
diff --git a/programs/cmm/lib/obj/libimg.h b/programs/cmm/lib/obj/libimg.h
index ac5b02d07a..26b87fe25b 100644
--- a/programs/cmm/lib/obj/libimg.h
+++ b/programs/cmm/lib/obj/libimg.h
@@ -299,7 +299,6 @@ struct libimg_image
         bg = _bg;
         i32.load("/sys/icons32.png");
         i32.replace_color(0x00000000, bg);
-        debugln("wolo");
     }
     if (icon_n>=0) i32.draw(x, y, 32, 32, 0, icon_n*32);
 }
diff --git a/programs/cmm/lib/patterns/restart_process.h b/programs/cmm/lib/patterns/restart_process.h
index 234b6dffdd..29c00bbb8a 100644
--- a/programs/cmm/lib/patterns/restart_process.h
+++ b/programs/cmm/lib/patterns/restart_process.h
@@ -53,5 +53,19 @@ enum {
 	RunProgram(proc_name, "");	
 }
 
+:void RestartAllProcess() {
+	int i;
+	proc_info Process, Self;
+	GetProcessInfo(#Self, -1);
+	for (i=0; i<MAX_PROCESS_COUNT; i++;)
+	{
+		GetProcessInfo(#Process, i);
+		if (Process.name) 
+		&& (!streq(#Process.name, "OS"))
+		&& (Process.ID != Self.ID)
+			KillProcess(Process.ID);
+	}
+	RunProgram("/sys/launcher", NULL);
+}
 
 #endif
\ No newline at end of file
diff --git a/programs/cmm/misc/osupdate.c b/programs/cmm/misc/osupdate.c
index c211c7c7a3..2d49a20a2d 100644
--- a/programs/cmm/misc/osupdate.c
+++ b/programs/cmm/misc/osupdate.c
@@ -1,4 +1,4 @@
-#define MEMSIZE 1024*460
+#define MEMSIZE 1024*50
 
 #include "../lib/io.h"
 #include "../lib/gui.h"
@@ -17,147 +17,183 @@
 #include "lang.h--"
 #endif
 
-_http http;
-proc_info Form;
 bool install_complete = false;
-sensor progress;
+_http http;
 
 #define WINW 460
+#define WINH 380
 
 //#define LANG_RUS 1
 
 #ifdef LANG_RUS
-#define WINH 345
 #define T_WINDOW_TITLE "Онлайн обновление KolibriOS"
 #define T_TITLE_H1 "ОНЛАЙН ОБНОВЛЕНИЕ"
 #define T_INTRO "Данное приложение скачает последнюю версию KolibriOS и распакует ее на RAM-диск. При этом ядро не будет перезапущено, для этого необходимо сохранить образ и перезагрузится. Пожалуйста, закройте все открытые приложения перед началом обновления.
-
 ВНИМАНИЕ: Все изменные файлы на RAM-диске будут перезаписаны!"; 
 #define T_INSTALL "Обновить"
 #define T_COMPLETE "Обновление завершено"
 #define T_EXIT "Выход"
 #define IMG_URL "http://builds.kolibrios.org/rus/data/data/kolibri.img"
+#define KS "Сохранить настройки"
+#define RA "Перезапустить все программы"
 #else
-#define WINH 305
 #define T_WINDOW_TITLE "KolibriOS Online Updater"
 #define T_TITLE_H1 "ONLINE UPDATE"
 #define T_INTRO "This app will download the latest KolibriOS dirsto and update your RAM-disk with it. Kernel won't be restarted. 
 Please close all opened apps before start.
-
-ALERT: All chages on RAM-disk will be lost!"; 
+Note that all changes on RAM-disk will be lost."; 
 #define T_INSTALL "Update"
 #define T_COMPLETE "Update complete"
 #define T_EXIT "Exit"
 #define IMG_URL "http://builds.kolibrios.org/eng/data/data/kolibri.img"
+#define KS "Keep settings folder"
+#define RA "Restart all apps"
 #endif
 char accept_language[]="en"; //not used, necessary for http.get()
+void Operation_Draw_Progress(dword f) {} //not used, necessary for copyf()
+
+checkbox keep_settings = { KS, true };
+checkbox restart_apps = { RA, true };
 
 void main()
 {
-	//load_dll(libini, #lib_init,1);
+	int btn;
+	sensor progress;
 	load_dll(libio, #libio_init,1);
 	load_dll(libimg, #libimg_init,1);
 	load_dll(libHTTP,   #http_lib_init,1);
-	SetEventMask(EVM_REDRAW + EVM_KEY + EVM_BUTTON + EVM_STACK);
-	loop() switch(WaitEventTimeout(300) & 0xFF)
+	@SetEventMask(EVM_REDRAW + EVM_KEY + EVM_BUTTON + EVM_STACK);
+	loop() switch(@WaitEventTimeout(300))
 	{
-		case evButton:
-			if (GetButtonID() == 1) ExitProcess(); else EventInstall();
+		case evButton: 
+			btn = @GetButtonID();
+			if (btn<=2) ExitProcess();
+			if (btn==9) goto _INSTALL; 
+			keep_settings.click(btn);
+			restart_apps.click(btn);
 			break;
-	  
+
 		case evKey:
-			GetKeys();
-			if (key_scancode == SCAN_CODE_ESC) ExitProcess();
-			if (key_scancode == SCAN_CODE_ENTER) {
-				if (install_complete) ExitProcess();
-				else EventInstall();
+			switch (@GetKeyScancode()) {
+				case SCAN_CODE_ESC: ExitProcess();
+				case SCAN_CODE_ENTER: 
+					if (install_complete) ExitProcess();
+					else {
+						_INSTALL: 
+						http.get(IMG_URL); 
+						goto _DRAW_WINDOW;
+						}
 			}
 			break;
-		 
+
 		case evReDraw:
-			draw_window();
-			break;
+			_DRAW_WINDOW:
+			sc.get();
+			DefineAndDrawWindow(screen.width-WINW/2,screen.height-WINH/2,
+				WINW+9,WINH+skin_height,0x34,sc.work,T_WINDOW_TITLE,0);
+			WriteText(30, 20, 0x81, 0xEC008C, T_TITLE_H1);
+			if (!install_complete) {
+					DrawTextViewArea(30, 50, WINW-60, WINH-80, T_INTRO, -1, sc.work_text);
+					progress.set_size(30, WINH-130, WINW-60, 20);
+					if (http.transfer<=0) {
+						DrawCaptButton(WINW-110/2, WINH-70, 110, 28, 9, 0x0092D8, 0xFFFfff, T_INSTALL);
+						keep_settings.draw(30, WINH - 210);
+						restart_apps.draw(30, WINH - 185);
+					}
+			} else {
+					DrawIcon32(WINW-32/2, 140, sc.work, 49);
+					WriteTextCenter(0,185, WINW, sc.work_text, T_COMPLETE);
+					DrawCaptButton(WINW-110/2, WINH-70, 110, 28, 2, 
+						0x0092D8, 0xFFFfff, T_EXIT);
+			}
 
 		case evNetwork:
 			if (http.transfer <= 0) break;
 			http.receive();
 			if (http.content_length) {
-				progress.draw_progress(http.content_length - http.content_received * progress.w / http.content_length);
+				progress.draw_progress(http.content_length - http.content_received 
+					* progress.w / http.content_length);
+			}
+			if (!http.receive_result) {
+				CreateFile(http.content_received, 
+					http.content_pointer, "/tmp0/1/latest.img");
+				http.free();
+				EventDownloadComplete();
 			}
-			if (http.receive_result == 0) EventDownloadComplete();
 	}
 }
 
-void draw_window()
+dword GetFreeSpaceOfRamdisk()
 {
-	sc.get();
-	DefineAndDrawWindow(screen.width-WINW/2,screen.height-WINH/2,
-		WINW+9,WINH+skin_height,0x34,sc.work,T_WINDOW_TITLE,0);
-	GetProcessInfo(#Form, -1);
-	WriteText(30, 20, 0x81, 0xEC008C, T_TITLE_H1);
-	if (install_complete) {
-		DrawInstallComplete(); 
-	} else {
-		DrawIntro();
-		progress.set_size(30, WINH-130, WINW-60, 20);
-	}
+	dword rdempty = malloc(1440*1024);
+	CreateFile(0, 1440*1024, rdempty, "/rd/1/rdempty");
+	free(rdempty);
+	file_size stdcall ("/rd/1/rdempty");
+	rdempty = EBX;
+	DeleteFile("/rd/1/rdempty");
+	return rdempty;
 }
 
-void DrawIntro()
+signed CheckFreeSpace(dword _latest, _combined)
 {
-	DrawTextViewArea(30, 50, WINW-60, WINH-80, 
-		T_INTRO, -1, sc.work_text);
-	DrawCaptButton(WINW-110/2, WINH-70, 110, 28, 9999, 
-		0x0092D8, 0xFFFfff, T_INSTALL);
-}
+	dword cur_size, new_size, empty;
+	DIR_SIZE dir_size;
 
-void DrawInstallComplete()
-{
-	DrawIcon32(WINW-32/2, 140, sc.work, 49);
-	WriteTextCenter(0,185, WINW, sc.work_text, T_COMPLETE);
-	DrawCaptButton(WINW-110/2, WINH-70, 110, 28, 1, 
-		0x0092D8, 0xFFFfff, T_EXIT);
-}
+	dir_size.get("/sys");
+	cur_size = dir_size.bytes;
 
-void EventInstall()
-{
-	http.get(IMG_URL);
+	copyf("/sys", _combined);
+	copyf(_latest, _combined);
+	dir_size.get(_combined);
+	new_size = dir_size.bytes;
+
+	empty = GetFreeSpaceOfRamdisk();
+
+	return cur_size + empty - new_size / 1024;
 }
 
 void EventDownloadComplete()
 {
 	dword unimg_id, slot_n;
+	signed space_delta;
+	int i=0;
 
-	CreateFile(http.content_received, http.content_pointer, "/tmp0/1/latest.img");
-	http.free();
+	char osupdate[32];
+	char latest[40];
+	char backup[40];
+	char combined[40];
+	char exract_param[64];
+	char backup_settings[64];
 
-	unimg_id = RunProgram("/sys/unimg", "/tmp0/1/latest.img /tmp0/1/latest_img -e");
+	do  { sprintf(#osupdate, "/tmp0/1/osupdate%d", i); i++;
+	} while (dir_exists(#osupdate));
+	CreateDir(#osupdate);
+
+	sprintf(#latest, "%s/latest", #osupdate);
+	sprintf(#backup, "%s/rdbackup", #osupdate);
+	sprintf(#combined, "%s/combined", #osupdate);
+	sprintf(#backup_settings, "%s/settings", #backup);
+	sprintf(#exract_param, "/tmp0/1/latest.img %s -e", #latest);
+
+
+	unimg_id = RunProgram("/sys/unimg", #exract_param);
 	do {
 		slot_n = GetProcessSlot(unimg_id);
 		pause(10);
 	} while (slot_n!=0);
 
-	copyf("/rd/1/settings", "/tmp0/1/settings_backup");
-	copyf("/tmp0/1/latest_img", "/rd/1");
-	copyf("/tmp0/1/settings_backup", "/rd/1/settings");
-
-	RestartAllProcess();
-	install_complete = true;
-	draw_window();
-}
-
-void RestartAllProcess() {
-	int i;
-	proc_info Process;
-	for (i=0; i<MAX_PROCESS_COUNT; i++;)
-	{
-		GetProcessInfo(#Process, i);
-		if (Process.name) 
-		&& (!streq(#Process.name, "OS"))
-		&& (Process.ID != Form.ID)
-			KillProcess(Process.ID);
+	space_delta = CheckFreeSpace(#latest, #combined);
+	if (space_delta<0) {
+		sprintf(#param, "'Not enought free space! You need %d Kb more.'E", -space_delta);
+		notify(#param);
+	} else {
+		copyf("/sys", #backup);
+		copyf(#latest, "/sys");
+		if (keep_settings.checked) copyf(#backup_settings, "/rd/1/settings");
+		if (restart_apps.checked) RestartAllProcess();
+		install_complete = true;		
 	}
-	RunProgram("/sys/launcher", NULL);
 }
 
-void Operation_Draw_Progress(dword filename) { debug("copying: "); debugln(filename); }
\ No newline at end of file
+
+