blob: 640f04a79bb13acd507abcf2353e99b339bf70da [file] [log] [blame]
Tao Bao337db142015-08-20 14:52:57 -07001/*
2 * Copyright (C) 2014 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
Tao Bao736d59c2017-01-03 10:15:33 -080017#include "wear_ui.h"
18
Tao Bao337db142015-08-20 14:52:57 -070019#include <errno.h>
20#include <fcntl.h>
Tao Bao337db142015-08-20 14:52:57 -070021#include <stdarg.h>
Tao Bao337db142015-08-20 14:52:57 -070022#include <stdlib.h>
23#include <string.h>
24#include <sys/stat.h>
25#include <sys/time.h>
26#include <sys/types.h>
27#include <time.h>
28#include <unistd.h>
29
Tao Bao736d59c2017-01-03 10:15:33 -080030#include <string>
Tao Bao337db142015-08-20 14:52:57 -070031#include <vector>
32
Tao Bao0ecbd762017-01-16 21:16:58 -080033#include <android-base/properties.h>
34#include <android-base/strings.h>
35#include <android-base/stringprintf.h>
36#include <minui/minui.h>
37
Tao Bao337db142015-08-20 14:52:57 -070038#include "common.h"
39#include "device.h"
Tao Bao337db142015-08-20 14:52:57 -070040
Tao Bao337db142015-08-20 14:52:57 -070041// There's only (at most) one of these objects, and global callbacks
42// (for pthread_create, and the input event system) need to find it,
43// so use a global variable.
44static WearRecoveryUI* self = NULL;
45
46// Return the current time as a double (including fractions of a second).
47static double now() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -070048 struct timeval tv;
49 gettimeofday(&tv, NULL);
50 return tv.tv_sec + tv.tv_usec / 1000000.0;
Tao Bao337db142015-08-20 14:52:57 -070051}
52
Tao Bao5d2e3bd2017-06-23 22:23:50 -070053WearRecoveryUI::WearRecoveryUI()
54 : progress_bar_y(259), outer_height(0), outer_width(0), menu_unusable_rows(0) {
55 intro_frames = 22;
56 loop_frames = 60;
57 animation_fps = 30;
Tao Bao337db142015-08-20 14:52:57 -070058
Tao Bao5d2e3bd2017-06-23 22:23:50 -070059 for (size_t i = 0; i < 5; i++) backgroundIcon[i] = NULL;
Tao Bao337db142015-08-20 14:52:57 -070060
Tao Bao5d2e3bd2017-06-23 22:23:50 -070061 self = this;
Tao Bao337db142015-08-20 14:52:57 -070062}
63
Tao Bao99b2d772017-06-23 22:47:03 -070064int WearRecoveryUI::GetProgressBaseline() const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -070065 return progress_bar_y;
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -070066}
67
Tao Bao337db142015-08-20 14:52:57 -070068// Draw background frame on the screen. Does not flip pages.
69// Should only be called with updateMutex locked.
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -070070// TODO merge drawing routines with screen_ui
Tao Bao5d2e3bd2017-06-23 22:23:50 -070071void WearRecoveryUI::draw_background_locked() {
72 pagesIdentical = false;
73 gr_color(0, 0, 0, 255);
74 gr_fill(0, 0, gr_fb_width(), gr_fb_height());
Tao Bao337db142015-08-20 14:52:57 -070075
Tao Bao5d2e3bd2017-06-23 22:23:50 -070076 if (currentIcon != NONE) {
77 GRSurface* surface;
78 if (currentIcon == INSTALLING_UPDATE || currentIcon == ERASING) {
79 if (!intro_done) {
80 surface = introFrames[current_frame];
81 } else {
82 surface = loopFrames[current_frame];
83 }
84 } else {
85 surface = backgroundIcon[currentIcon];
Tao Bao337db142015-08-20 14:52:57 -070086 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -070087
88 int width = gr_get_width(surface);
89 int height = gr_get_height(surface);
90
91 int x = (gr_fb_width() - width) / 2;
92 int y = (gr_fb_height() - height) / 2;
93
94 gr_blit(surface, 0, 0, width, height, x, y);
95 }
Tao Bao337db142015-08-20 14:52:57 -070096}
97
Tao Bao337db142015-08-20 14:52:57 -070098static const char* HEADERS[] = {
Tao Bao5d2e3bd2017-06-23 22:23:50 -070099 "Swipe up/down to move.",
100 "Swipe left/right to select.",
101 "",
102 NULL
Tao Bao337db142015-08-20 14:52:57 -0700103};
104
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700105// TODO merge drawing routines with screen_ui
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700106void WearRecoveryUI::draw_screen_locked() {
107 char cur_selection_str[50];
Tao Bao337db142015-08-20 14:52:57 -0700108
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700109 draw_background_locked();
110 if (!show_text) {
111 draw_foreground_locked();
112 } else {
113 SetColor(TEXT_FILL);
114 gr_fill(0, 0, gr_fb_width(), gr_fb_height());
Tao Bao337db142015-08-20 14:52:57 -0700115
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700116 int y = outer_height;
117 int x = outer_width;
118 if (show_menu) {
119 std::string recovery_fingerprint =
120 android::base::GetProperty("ro.bootimage.build.fingerprint", "");
121 SetColor(HEADER);
122 DrawTextLine(x + 4, &y, "Android Recovery", true);
123 for (auto& chunk : android::base::Split(recovery_fingerprint, ":")) {
124 DrawTextLine(x + 4, &y, chunk.c_str(), false);
125 }
Tao Bao337db142015-08-20 14:52:57 -0700126
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700127 // This is actually the help strings.
128 DrawTextLines(x + 4, &y, HEADERS);
129 SetColor(HEADER);
130 DrawTextLines(x + 4, &y, menu_headers_);
Tao Bao337db142015-08-20 14:52:57 -0700131
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700132 // Show the current menu item number in relation to total number if
133 // items don't fit on the screen.
134 if (menu_items > menu_end - menu_start) {
135 sprintf(cur_selection_str, "Current item: %d/%d", menu_sel + 1, menu_items);
136 gr_text(gr_sys_font(), x + 4, y, cur_selection_str, 1);
137 y += char_height_ + 4;
138 }
Tao Bao337db142015-08-20 14:52:57 -0700139
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700140 // Menu begins here
141 SetColor(MENU);
Tao Bao337db142015-08-20 14:52:57 -0700142
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700143 for (int i = menu_start; i < menu_end; ++i) {
144 if (i == menu_sel) {
145 // draw the highlight bar
146 SetColor(MENU_SEL_BG);
147 gr_fill(x, y - 2, gr_fb_width() - x, y + char_height_ + 2);
148 // white text of selected item
149 SetColor(MENU_SEL_FG);
150 if (menu_[i][0]) {
151 gr_text(gr_sys_font(), x + 4, y, menu_[i], 1);
152 }
153 SetColor(MENU);
154 } else if (menu_[i][0]) {
155 gr_text(gr_sys_font(), x + 4, y, menu_[i], 0);
Tao Bao337db142015-08-20 14:52:57 -0700156 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700157 y += char_height_ + 4;
158 }
159 SetColor(MENU);
160 y += 4;
161 gr_fill(0, y, gr_fb_width(), y + 2);
162 y += 4;
Tao Bao337db142015-08-20 14:52:57 -0700163 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700164
165 SetColor(LOG);
166
167 // display from the bottom up, until we hit the top of the
168 // screen, the bottom of the menu, or we've displayed the
169 // entire text buffer.
170 int ty;
171 int row = (text_top_ + text_rows_ - 1) % text_rows_;
172 size_t count = 0;
173 for (int ty = gr_fb_height() - char_height_ - outer_height; ty > y + 2 && count < text_rows_;
174 ty -= char_height_, ++count) {
175 gr_text(gr_sys_font(), x + 4, ty, text_[row], 0);
176 --row;
177 if (row < 0) row = text_rows_ - 1;
178 }
179 }
Tao Bao337db142015-08-20 14:52:57 -0700180}
181
Damien Bargiacchiad8b5a62016-09-09 08:18:06 -0700182// TODO merge drawing routines with screen_ui
183void WearRecoveryUI::update_progress_locked() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700184 draw_screen_locked();
185 gr_flip();
Tao Bao337db142015-08-20 14:52:57 -0700186}
187
Sen Jiangd5304492016-12-09 16:20:49 -0800188bool WearRecoveryUI::InitTextParams() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700189 if (!ScreenRecoveryUI::InitTextParams()) {
190 return false;
191 }
Tao Bao337db142015-08-20 14:52:57 -0700192
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700193 text_cols_ = (gr_fb_width() - (outer_width * 2)) / char_width_;
Tao Bao337db142015-08-20 14:52:57 -0700194
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700195 if (text_rows_ > kMaxRows) text_rows_ = kMaxRows;
196 if (text_cols_ > kMaxCols) text_cols_ = kMaxCols;
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700197
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700198 visible_text_rows = (gr_fb_height() - (outer_height * 2)) / char_height_;
199 return true;
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700200}
Tao Bao337db142015-08-20 14:52:57 -0700201
Tao Bao736d59c2017-01-03 10:15:33 -0800202bool WearRecoveryUI::Init(const std::string& locale) {
203 if (!ScreenRecoveryUI::Init(locale)) {
204 return false;
205 }
Tao Bao337db142015-08-20 14:52:57 -0700206
Tao Bao736d59c2017-01-03 10:15:33 -0800207 LoadBitmap("icon_error", &backgroundIcon[ERROR]);
208 backgroundIcon[NO_COMMAND] = backgroundIcon[ERROR];
Tim Kryger825b6b02016-11-29 13:33:29 -0800209
Tao Bao736d59c2017-01-03 10:15:33 -0800210 // This leaves backgroundIcon[INSTALLING_UPDATE] and backgroundIcon[ERASING]
211 // as NULL which is fine since draw_background_locked() doesn't use them.
Tim Kryger825b6b02016-11-29 13:33:29 -0800212
Tao Bao736d59c2017-01-03 10:15:33 -0800213 return true;
Prashant Malanif7f9e502016-03-10 03:40:20 +0000214}
215
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700216void WearRecoveryUI::SetStage(int current, int max) {
217}
Tao Bao337db142015-08-20 14:52:57 -0700218
Tao Bao736d59c2017-01-03 10:15:33 -0800219void WearRecoveryUI::Print(const char* fmt, ...) {
220 char buf[256];
221 va_list ap;
222 va_start(ap, fmt);
223 vsnprintf(buf, 256, fmt, ap);
224 va_end(ap);
Tao Bao337db142015-08-20 14:52:57 -0700225
Tao Bao736d59c2017-01-03 10:15:33 -0800226 fputs(buf, stdout);
Tao Bao337db142015-08-20 14:52:57 -0700227
Tao Bao736d59c2017-01-03 10:15:33 -0800228 // This can get called before ui_init(), so be careful.
229 pthread_mutex_lock(&updateMutex);
230 if (text_rows_ > 0 && text_cols_ > 0) {
231 char* ptr;
232 for (ptr = buf; *ptr != '\0'; ++ptr) {
233 if (*ptr == '\n' || text_col_ >= text_cols_) {
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700234 text_[text_row_][text_col_] = '\0';
Tao Bao736d59c2017-01-03 10:15:33 -0800235 text_col_ = 0;
236 text_row_ = (text_row_ + 1) % text_rows_;
237 if (text_row_ == text_top_) text_top_ = (text_top_ + 1) % text_rows_;
238 }
239 if (*ptr != '\n') text_[text_row_][text_col_++] = *ptr;
Tao Bao337db142015-08-20 14:52:57 -0700240 }
Tao Bao736d59c2017-01-03 10:15:33 -0800241 text_[text_row_][text_col_] = '\0';
242 update_screen_locked();
243 }
244 pthread_mutex_unlock(&updateMutex);
Tao Bao337db142015-08-20 14:52:57 -0700245}
246
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700247void WearRecoveryUI::StartMenu(const char* const* headers, const char* const* items,
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700248 int initial_selection) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700249 pthread_mutex_lock(&updateMutex);
250 if (text_rows_ > 0 && text_cols_ > 0) {
251 menu_headers_ = headers;
252 size_t i = 0;
253 // "i < text_rows_" is removed from the loop termination condition,
254 // which is different from the one in ScreenRecoveryUI::StartMenu().
255 // Because WearRecoveryUI supports scrollable menu, it's fine to have
256 // more entries than text_rows_. The menu may be truncated otherwise.
257 // Bug: 23752519
258 for (; items[i] != nullptr; i++) {
259 strncpy(menu_[i], items[i], text_cols_ - 1);
260 menu_[i][text_cols_ - 1] = '\0';
Tao Bao337db142015-08-20 14:52:57 -0700261 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700262 menu_items = i;
263 show_menu = true;
264 menu_sel = initial_selection;
265 menu_start = 0;
266 menu_end = visible_text_rows - 1 - menu_unusable_rows;
267 if (menu_items <= menu_end) menu_end = menu_items;
268 update_screen_locked();
269 }
270 pthread_mutex_unlock(&updateMutex);
Tao Bao337db142015-08-20 14:52:57 -0700271}
272
273int WearRecoveryUI::SelectMenu(int sel) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700274 int old_sel;
275 pthread_mutex_lock(&updateMutex);
276 if (show_menu) {
277 old_sel = menu_sel;
278 menu_sel = sel;
279 if (menu_sel < 0) menu_sel = 0;
280 if (menu_sel >= menu_items) menu_sel = menu_items - 1;
281 if (menu_sel < menu_start) {
282 menu_start--;
283 menu_end--;
284 } else if (menu_sel >= menu_end && menu_sel < menu_items) {
285 menu_end++;
286 menu_start++;
Tao Bao337db142015-08-20 14:52:57 -0700287 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700288 sel = menu_sel;
289 if (menu_sel != old_sel) update_screen_locked();
290 }
291 pthread_mutex_unlock(&updateMutex);
292 return sel;
Tao Bao337db142015-08-20 14:52:57 -0700293}
294
Tao Bao337db142015-08-20 14:52:57 -0700295void WearRecoveryUI::ShowFile(FILE* fp) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700296 std::vector<off_t> offsets;
297 offsets.push_back(ftello(fp));
298 ClearText();
Tao Bao337db142015-08-20 14:52:57 -0700299
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700300 struct stat sb;
301 fstat(fileno(fp), &sb);
Tao Bao337db142015-08-20 14:52:57 -0700302
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700303 bool show_prompt = false;
304 while (true) {
305 if (show_prompt) {
306 Print("--(%d%% of %d bytes)--",
307 static_cast<int>(100 * (double(ftello(fp)) / double(sb.st_size))),
308 static_cast<int>(sb.st_size));
309 Redraw();
310 while (show_prompt) {
311 show_prompt = false;
312 int key = WaitKey();
313 if (key == KEY_POWER || key == KEY_ENTER) {
314 return;
315 } else if (key == KEY_UP || key == KEY_VOLUMEUP) {
316 if (offsets.size() <= 1) {
Tao Bao337db142015-08-20 14:52:57 -0700317 show_prompt = true;
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700318 } else {
319 offsets.pop_back();
320 fseek(fp, offsets.back(), SEEK_SET);
321 }
Tao Bao337db142015-08-20 14:52:57 -0700322 } else {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700323 if (feof(fp)) {
324 return;
325 }
326 offsets.push_back(ftello(fp));
Tao Bao337db142015-08-20 14:52:57 -0700327 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700328 }
329 ClearText();
Tao Bao337db142015-08-20 14:52:57 -0700330 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700331
332 int ch = getc(fp);
333 if (ch == EOF) {
334 text_row_ = text_top_ = text_rows_ - 2;
335 show_prompt = true;
336 } else {
337 PutChar(ch);
338 if (text_col_ == 0 && text_row_ >= text_rows_ - 2) {
339 text_top_ = text_row_;
340 show_prompt = true;
341 }
342 }
343 }
Tao Bao337db142015-08-20 14:52:57 -0700344}
345
346void WearRecoveryUI::PutChar(char ch) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700347 pthread_mutex_lock(&updateMutex);
348 if (ch != '\n') text_[text_row_][text_col_++] = ch;
349 if (ch == '\n' || text_col_ >= text_cols_) {
350 text_col_ = 0;
351 ++text_row_;
352 }
353 pthread_mutex_unlock(&updateMutex);
Tao Bao337db142015-08-20 14:52:57 -0700354}
355
356void WearRecoveryUI::ShowFile(const char* filename) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700357 FILE* fp = fopen_path(filename, "re");
358 if (fp == nullptr) {
359 Print(" Unable to open %s: %s\n", filename, strerror(errno));
360 return;
361 }
362 ShowFile(fp);
363 fclose(fp);
Tao Bao337db142015-08-20 14:52:57 -0700364}
365
366void WearRecoveryUI::ClearText() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700367 pthread_mutex_lock(&updateMutex);
368 text_col_ = 0;
369 text_row_ = 0;
370 text_top_ = 1;
371 for (size_t i = 0; i < text_rows_; ++i) {
372 memset(text_[i], 0, text_cols_ + 1);
373 }
374 pthread_mutex_unlock(&updateMutex);
Tao Bao337db142015-08-20 14:52:57 -0700375}
Prashant Malani0eb41c32016-02-25 18:27:03 -0800376
377void WearRecoveryUI::PrintOnScreenOnly(const char *fmt, ...) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700378 va_list ap;
379 va_start(ap, fmt);
380 PrintV(fmt, false, ap);
381 va_end(ap);
Prashant Malani0eb41c32016-02-25 18:27:03 -0800382}
383
384void WearRecoveryUI::PrintV(const char* fmt, bool copy_to_stdout, va_list ap) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700385 std::string str;
386 android::base::StringAppendV(&str, fmt, ap);
Prashant Malani0eb41c32016-02-25 18:27:03 -0800387
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700388 if (copy_to_stdout) {
389 fputs(str.c_str(), stdout);
390 }
Prashant Malani0eb41c32016-02-25 18:27:03 -0800391
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700392 pthread_mutex_lock(&updateMutex);
393 if (text_rows_ > 0 && text_cols_ > 0) {
394 for (const char* ptr = str.c_str(); *ptr != '\0'; ++ptr) {
395 if (*ptr == '\n' || text_col_ >= text_cols_) {
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700396 text_[text_row_][text_col_] = '\0';
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700397 text_col_ = 0;
398 text_row_ = (text_row_ + 1) % text_rows_;
399 if (text_row_ == text_top_) text_top_ = (text_top_ + 1) % text_rows_;
400 }
401 if (*ptr != '\n') text_[text_row_][text_col_++] = *ptr;
Prashant Malani0eb41c32016-02-25 18:27:03 -0800402 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700403 text_[text_row_][text_col_] = '\0';
404 update_screen_locked();
405 }
406 pthread_mutex_unlock(&updateMutex);
Prashant Malani0eb41c32016-02-25 18:27:03 -0800407}