Fix overflow in modwin erase. Do not refresh modwin yet, since it is
[coreboot.git] / payloads / coreinfo / coreinfo.c
1 /*
2  * This file is part of the coreinfo project.
3  *
4  * Copyright (C) 2008 Advanced Micro Devices, Inc.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; version 2 of the License.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
18  */
19
20 #include "coreinfo.h"
21
22 #define SCREEN_Y 25
23 #define SCREEN_X 80
24
25 #define KEY_ESC 27
26
27 extern struct coreinfo_module cpuinfo_module;
28 extern struct coreinfo_module pci_module;
29 extern struct coreinfo_module coreboot_module;
30 extern struct coreinfo_module nvram_module;
31 extern struct coreinfo_module bootlog_module;
32 extern struct coreinfo_module ramdump_module;
33 extern struct coreinfo_module lar_module;
34
35 struct coreinfo_module *system_modules[] = {
36 #ifdef CONFIG_MODULE_CPUINFO
37         &cpuinfo_module,
38 #endif
39 #ifdef CONFIG_MODULE_PCI
40         &pci_module,
41 #endif
42 #ifdef CONFIG_MODULE_NVRAM
43         &nvram_module,
44 #endif
45 #ifdef CONFIG_MODULE_RAMDUMP
46         &ramdump_module,
47 #endif
48 };
49
50 struct coreinfo_module *coreboot_modules[] = {
51 #ifdef CONFIG_MODULE_COREBOOT
52         &coreboot_module,
53 #endif
54 #ifdef CONFIG_MODULE_BOOTLOG
55         &bootlog_module,
56 #endif
57 #ifdef CONFIG_MODULE_LAR
58         &lar_module
59 #endif
60 };
61
62 struct coreinfo_cat {
63         char name[15];
64         int cur;
65         int count;
66         struct coreinfo_module **modules;
67 } categories[] = {
68         {
69                 .name = "System",
70                 .modules = system_modules,
71                 .count = ARRAY_SIZE(system_modules),
72         },
73         {
74                 .name = "Coreboot",
75                 .modules = coreboot_modules,
76                 .count = ARRAY_SIZE(coreboot_modules),
77         }
78 };
79
80 static WINDOW *modwin, *menuwin;
81 static int curwin;
82
83 void print_module_title(WINDOW *win, const char *title)
84 {
85         int i;
86
87         wattrset(win, COLOR_PAIR(2));
88         mvwprintw(win, 0, 1, title);
89
90         wmove(win, 1, 1);
91         for (i = 0; i < 78; i++)
92                 waddch(win, ACS_HLINE);
93 }
94
95 static void print_submenu(struct coreinfo_cat *cat)
96 {
97         int i, j;
98         char menu[80];
99         char *ptr = menu;
100
101         wmove(menuwin, 0, 0);
102
103         for (j = 0; j < SCREEN_X; j++)
104                 waddch(menuwin, ' ');
105
106         if (!cat->count)
107                 return;
108
109         for (i = 0; i < cat->count; i++)
110                 ptr += sprintf(ptr, "[%c: %s] ", 'A' + i,
111                                cat->modules[i]->name);
112
113         mvwprintw(menuwin, 0, 0, menu);
114 }
115
116 #ifdef CONFIG_SHOW_DATE_TIME
117 static void print_time_and_date(void)
118 {
119         struct tm tm;
120
121         while (nvram_updating())
122                 mdelay(10);
123
124         rtc_read_clock(&tm);
125
126         mvwprintw(menuwin, 0, 57, "%02d/%02d/%04d - %02d:%02d:%02d",
127                   tm.tm_mon, tm.tm_mday, 1900 + tm.tm_year, tm.tm_hour,
128                   tm.tm_min, tm.tm_sec);
129 }
130 #endif
131
132 static void print_menu(void)
133 {
134         int i, j;
135         char menu[80];
136         char *ptr = menu;
137
138         wmove(menuwin, 1, 0);
139         for (j = 0; j < SCREEN_X; j++)
140                 waddch(menuwin, ' ');
141
142         for (i = 0; i < ARRAY_SIZE(categories); i++) {
143                 if (categories[i].count == 0)
144                         continue;
145
146                 ptr += sprintf(ptr, "F%d: %s ", i + 1, categories[i].name);
147         }
148
149         mvwprintw(menuwin, 1, 0, menu);
150
151 #ifdef CONFIG_SHOW_DATE_TIME
152         print_time_and_date();
153 #endif
154 }
155
156 static void center(int row, const char *str)
157 {
158         int j, len = strlen(str);
159
160         wmove(stdscr, row, 0);
161         for (j = 0; j < SCREEN_X; j++)
162                 waddch(stdscr, ' ');
163
164         mvprintw(row, (SCREEN_X - len) / 2, str);
165 }
166
167 /* FIXME: Currently unused. */
168 #if 0
169 static void header(int row, const char *str)
170 {
171         char buf[SCREEN_X];
172         char *ptr = buf;
173         int i;
174         int len = strlen(str) + 4;
175
176         for (i = 0; i < (SCREEN_X - len) / 2; i++)
177                 ptr += sprintf(ptr, "=");
178
179         ptr += sprintf(ptr, "[ %s ]", str);
180
181         for (i = ((SCREEN_X - len) / 2) + len; i < SCREEN_X; i++)
182                 ptr += sprintf(ptr, "=");
183
184         mvprintw(row, 0, buf);
185 }
186 #endif
187
188 static void redraw_module(struct coreinfo_cat *cat)
189 {
190         if (cat->count == 0)
191                 return;
192
193         wclear(modwin);
194         cat->modules[cat->cur]->redraw(modwin);
195         wrefresh(modwin);
196 }
197
198 static void handle_category_key(struct coreinfo_cat *cat, int key)
199 {
200         if (key >= 'a' && key <= 'z') {
201                 int index = key - 'a';
202                 if (index < cat->count) {
203                         cat->cur = index;
204                         redraw_module(cat);
205                         return;
206                 }
207         }
208
209         if (cat->count && cat->modules[cat->cur]->handle) {
210                 if (cat->modules[cat->cur]->handle(key))
211                         redraw_module(cat);
212         }
213 }
214
215 static void loop(void)
216 {
217         int key;
218
219         center(0, CONFIG_PAYLOAD_INFO_NAME " " CONFIG_PAYLOAD_INFO_VERSION);
220         refresh();
221
222         print_menu();
223         print_submenu(&categories[curwin]);
224         redraw_module(&categories[curwin]);
225
226         halfdelay(10);
227
228         while (1) {
229 #ifdef CONFIG_SHOW_DATE_TIME
230                 print_time_and_date();
231                 wrefresh(menuwin);
232 #endif
233
234                 key = getch();
235
236                 if (key == ERR)
237                         continue;
238
239                 if (key >= KEY_F(1) && key <= KEY_F(9)) {
240                         unsigned char ch = key - KEY_F(1);
241
242                         if (ch <= ARRAY_SIZE(categories)) {
243                                 if (ch == ARRAY_SIZE(categories))
244                                         continue;
245                                 if (categories[ch].count == 0)
246                                         continue;
247
248                                 curwin = ch;
249                                 print_submenu(&categories[curwin]);
250                                 redraw_module(&categories[curwin]);
251                                 continue;
252                         }
253                 }
254
255                 if (key == KEY_ESC)
256                         return;
257
258                 handle_category_key(&categories[curwin], key);
259         }
260 }
261
262 int main(void)
263 {
264         int i, j;
265
266         initscr();
267
268         init_pair(1, COLOR_WHITE, COLOR_GREEN);
269         init_pair(2, COLOR_BLACK, COLOR_WHITE);
270         init_pair(3, COLOR_WHITE, COLOR_WHITE);
271
272         modwin = newwin(SCREEN_Y - 3, SCREEN_X, 1, 0);
273         menuwin = newwin(2, SCREEN_X, SCREEN_Y - 2, 0);
274
275         wattrset(stdscr, COLOR_PAIR(1) | A_BOLD);
276         wattrset(modwin, COLOR_PAIR(2));
277         wattrset(menuwin, COLOR_PAIR(1) | A_BOLD);
278
279         werase(modwin);
280
281         for (i = 0; i < ARRAY_SIZE(categories); i++) {
282                 for (j = 0; j < categories[i].count; j++)
283                         categories[i].modules[j]->init();
284         }
285
286         loop();
287
288         return 0;
289 }
290
291 PAYLOAD_INFO(name, CONFIG_PAYLOAD_INFO_NAME);
292 PAYLOAD_INFO(listname, CONFIG_PAYLOAD_INFO_LISTNAME);
293 PAYLOAD_INFO(desc, CONFIG_PAYLOAD_INFO_DESC);