]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - stand/forth/menu.4th
e1000: Lock nvm print sysctl
[FreeBSD/FreeBSD.git] / stand / forth / menu.4th
1 \ Copyright (c) 2003 Scott Long <scottl@FreeBSD.org>
2 \ Copyright (c) 2003 Aleksander Fafula <alex@fafula.com>
3 \ Copyright (c) 2006-2015 Devin Teske <dteske@FreeBSD.org>
4 \ All rights reserved.
5
6 \ Redistribution and use in source and binary forms, with or without
7 \ modification, are permitted provided that the following conditions
8 \ are met:
9 \ 1. Redistributions of source code must retain the above copyright
10 \    notice, this list of conditions and the following disclaimer.
11 \ 2. Redistributions in binary form must reproduce the above copyright
12 \    notice, this list of conditions and the following disclaimer in the
13 \    documentation and/or other materials provided with the distribution.
14
15 \ THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 \ ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 \ IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 \ ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 \ FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 \ DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 \ OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 \ HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 \ LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 \ OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 \ SUCH DAMAGE.
26
27 \ $FreeBSD$
28
29 marker task-menu.4th
30
31 \ Frame drawing
32 include /boot/frames.4th
33
34 vocabulary menu-infrastructure
35 vocabulary menu-namespace
36 vocabulary menu-command-helpers
37
38 only forth also menu-infrastructure definitions
39
40 f_double        \ Set frames to double (see frames.4th). Replace with
41                 \ f_single if you want single frames.
42 46 constant dot \ ASCII definition of a period (in decimal)
43
44  5 constant menu_default_x         \ default column position of timeout
45 10 constant menu_default_y         \ default row position of timeout msg
46  4 constant menu_timeout_default_x \ default column position of timeout
47 23 constant menu_timeout_default_y \ default row position of timeout msg
48 10 constant menu_timeout_default   \ default timeout (in seconds)
49
50 \ Customize the following values with care
51
52   1 constant menu_start \ Numerical prefix of first menu item
53 dot constant bullet     \ Menu bullet (appears after numerical prefix)
54   5 constant menu_x     \ Row position of the menu (from the top)
55  10 constant menu_y     \ Column position of the menu (from left side)
56
57 \ Menu Appearance
58 variable menuidx   \ Menu item stack for number prefixes
59 variable menurow   \ Menu item stack for positioning
60 variable menubllt  \ Menu item bullet
61
62 \ Menu Positioning
63 variable menuX     \ Menu X offset (columns)
64 variable menuY     \ Menu Y offset (rows)
65
66 \ Menu-item elements
67 variable menurebootadded
68
69 \ Parsing of kernels into menu-items
70 variable kernidx
71 variable kernlen
72 variable kernmenuidx
73
74 \ Menu timer [count-down] variables
75 variable menu_timeout_enabled \ timeout state (internal use only)
76 variable menu_time            \ variable for tracking the passage of time
77 variable menu_timeout         \ determined configurable delay duration
78 variable menu_timeout_x       \ column position of timeout message
79 variable menu_timeout_y       \ row position of timeout message
80
81 \ Containers for parsing kernels into menu-items
82 create kerncapbuf 64 allot
83 create kerndefault 64 allot
84 create kernelsbuf 256 allot
85
86 only forth also menu-namespace definitions
87
88 \ Menu-item key association/detection
89 variable menukey1
90 variable menukey2
91 variable menukey3
92 variable menukey4
93 variable menukey5
94 variable menukey6
95 variable menukey7
96 variable menukey8
97 variable menureboot
98 variable menuacpi
99 variable menuoptions
100 variable menukernel
101
102 \ Menu initialization status variables
103 variable init_state1
104 variable init_state2
105 variable init_state3
106 variable init_state4
107 variable init_state5
108 variable init_state6
109 variable init_state7
110 variable init_state8
111
112 \ Boolean option status variables
113 variable toggle_state1
114 variable toggle_state2
115 variable toggle_state3
116 variable toggle_state4
117 variable toggle_state5
118 variable toggle_state6
119 variable toggle_state7
120 variable toggle_state8
121
122 \ Array option status variables
123 variable cycle_state1
124 variable cycle_state2
125 variable cycle_state3
126 variable cycle_state4
127 variable cycle_state5
128 variable cycle_state6
129 variable cycle_state7
130 variable cycle_state8
131
132 \ Containers for storing the initial caption text
133 create init_text1 64 allot
134 create init_text2 64 allot
135 create init_text3 64 allot
136 create init_text4 64 allot
137 create init_text5 64 allot
138 create init_text6 64 allot
139 create init_text7 64 allot
140 create init_text8 64 allot
141
142 only forth definitions
143
144 : arch-i386? ( -- BOOL ) \ Returns TRUE (-1) on i386, FALSE (0) otherwise.
145         s" arch-i386" environment? dup if
146                 drop
147         then
148 ;
149
150 : acpipresent? ( -- flag ) \ Returns TRUE if ACPI is present, FALSE otherwise
151         s" hint.acpi.0.rsdp" getenv
152         dup -1 = if
153                 drop false exit
154         then
155         2drop
156         true
157 ;
158
159 : acpienabled? ( -- flag ) \ Returns TRUE if ACPI is enabled, FALSE otherwise
160         s" hint.acpi.0.disabled" getenv
161         dup -1 <> if
162                 s" 0" compare 0<> if
163                         false exit
164                 then
165         else
166                 drop
167         then
168         true
169 ;
170
171 : +c! ( N C-ADDR/U K -- C-ADDR/U )
172         3 pick 3 pick   ( n c-addr/u k -- n c-addr/u k n c-addr )
173         rot + c!        ( n c-addr/u k n c-addr -- n c-addr/u )
174         rot drop        ( n c-addr/u -- c-addr/u )
175 ;
176
177 only forth also menu-namespace definitions
178
179 \ Forth variables
180 : namespace     ( C-ADDR/U N -- ) also menu-namespace +c! evaluate previous ;
181 : menukeyN      ( N -- ADDR )   s" menukeyN"       7 namespace ;
182 : init_stateN   ( N -- ADDR )   s" init_stateN"   10 namespace ;
183 : toggle_stateN ( N -- ADDR )   s" toggle_stateN" 12 namespace ;
184 : cycle_stateN  ( N -- ADDR )   s" cycle_stateN"  11 namespace ;
185 : init_textN    ( N -- C-ADDR ) s" init_textN"     9 namespace ;
186
187 \ Environment variables
188 : kernel[x]          ( N -- C-ADDR/U )   s" kernel[x]"           7 +c! ;
189 : menu_init[x]       ( N -- C-ADDR/U )   s" menu_init[x]"       10 +c! ;
190 : menu_command[x]    ( N -- C-ADDR/U )   s" menu_command[x]"    13 +c! ;
191 : menu_caption[x]    ( N -- C-ADDR/U )   s" menu_caption[x]"    13 +c! ;
192 : ansi_caption[x]    ( N -- C-ADDR/U )   s" ansi_caption[x]"    13 +c! ;
193 : menu_keycode[x]    ( N -- C-ADDR/U )   s" menu_keycode[x]"    13 +c! ;
194 : toggled_text[x]    ( N -- C-ADDR/U )   s" toggled_text[x]"    13 +c! ;
195 : toggled_ansi[x]    ( N -- C-ADDR/U )   s" toggled_ansi[x]"    13 +c! ;
196 : menu_caption[x][y] ( N M -- C-ADDR/U ) s" menu_caption[x][y]" 16 +c! 13 +c! ;
197 : ansi_caption[x][y] ( N M -- C-ADDR/U ) s" ansi_caption[x][y]" 16 +c! 13 +c! ;
198
199 also menu-infrastructure definitions
200
201 \ This function prints a menu item at menuX (row) and menuY (column), returns
202 \ the incremental decimal ASCII value associated with the menu item, and
203 \ increments the cursor position to the next row for the creation of the next
204 \ menu item. This function is called by the menu-create function. You need not
205 \ call it directly.
206
207 : printmenuitem ( menu_item_str -- ascii_keycode )
208
209         loader_color? if [char] ^ escc! then
210
211         menurow dup @ 1+ swap ! ( increment menurow )
212         menuidx dup @ 1+ swap ! ( increment menuidx )
213
214         \ Calculate the menuitem row position
215         menurow @ menuY @ +
216
217         \ Position the cursor at the menuitem position
218         dup menuX @ swap at-xy
219
220         \ Print the value of menuidx
221         loader_color? dup ( -- bool bool )
222         if b then
223         menuidx @ .
224         if me then
225
226         \ Move the cursor forward 1 column
227         dup menuX @ 1+ swap at-xy
228
229         menubllt @ emit \ Print the menu bullet using the emit function
230
231         \ Move the cursor to the 3rd column from the current position
232         \ to allow for a space between the numerical prefix and the
233         \ text caption
234         menuX @ 3 + swap at-xy
235
236         \ Print the menu caption (we expect a string to be on the stack
237         \ prior to invoking this function)
238         type
239
240         \ Here we will add the ASCII decimal of the numerical prefix
241         \ to the stack (decimal ASCII for `1' is 49) as a "return value"
242         menuidx @ 48 +
243 ;
244
245 \ This function prints the appropriate menuitem basename to the stack if an
246 \ ACPI option is to be presented to the user, otherwise returns -1. Used
247 \ internally by menu-create, you need not (nor should you) call this directly.
248
249 : acpimenuitem ( -- C-Addr/U | -1 )
250
251         arch-i386? if
252                 acpipresent? if
253                         acpienabled? if
254                                 loader_color? if
255                                         s" toggled_ansi[x]"
256                                 else
257                                         s" toggled_text[x]"
258                                 then
259                         else
260                                 loader_color? if
261                                         s" ansi_caption[x]"
262                                 else
263                                         s" menu_caption[x]"
264                                 then
265                         then
266                 else
267                         menuidx dup @ 1+ swap ! ( increment menuidx )
268                         -1
269                 then
270         else
271                 -1
272         then
273 ;
274
275 : delim? ( C -- BOOL )
276         dup  32 =               ( c -- c bool )         \ [sp] space
277         over  9 = or            ( c bool -- c bool )    \ [ht] horizontal tab
278         over 10 = or            ( c bool -- c bool )    \ [nl] newline
279         over 13 = or            ( c bool -- c bool )    \ [cr] carriage return
280         over [char] , = or      ( c bool -- c bool )    \ comma
281         swap drop               ( c bool -- bool )      \ return boolean
282 ;
283
284 \ This function parses $kernels into variables that are used by the menu to
285 \ display which kernel to boot when the [overloaded] `boot' word is interpreted.
286 \ Used internally by menu-create, you need not (nor should you) call this
287 \ directly.
288
289 : parse-kernels ( N -- ) \ kernidx
290         kernidx ! ( n -- )      \ store provided `x' value
291         [char] 0 kernmenuidx !  \ initialize `y' value for menu_caption[x][y]
292
293         \ Attempt to get a list of kernels, fall back to sensible default
294         s" kernels" getenv dup -1 = if
295                 drop ( cruft )
296                 s" kernel kernel.old"
297         then ( -- c-addr/u )
298
299         \ Check to see if the user has altered $kernel by comparing it against
300         \ $kernel[N] where N is kernel_state (the actively displayed kernel).
301         s" kernel_state" evaluate @ 48 + s" kernel[N]" 7 +c! getenv
302         dup -1 <> if
303                 s" kernel" getenv dup -1 = if
304                         drop ( cruft ) s" "
305                 then
306                 2swap 2over compare 0= if
307                         2drop FALSE ( skip below conditional )
308                 else \ User has changed $kernel
309                         TRUE ( slurp in new value )
310                 then
311         else \ We haven't yet parsed $kernels into $kernel[N]
312                 drop ( getenv cruft )
313                 s" kernel" getenv dup -1 = if
314                         drop ( cruft ) s" "
315                 then
316                 TRUE ( slurp in initial value )
317         then ( c-addr/u -- c-addr/u c-addr/u,-1 | 0 )
318         if \ slurp new value into kerndefault
319                 kerndefault 1+ 0 2swap strcat swap 1- c!
320         then
321
322         \ Clear out existing parsed-kernels
323         kernidx @ [char] 0
324         begin
325                 dup kernel[x] unsetenv
326                 2dup menu_caption[x][y] unsetenv
327                 2dup ansi_caption[x][y] unsetenv
328                 1+ dup [char] 8 >
329         until
330         2drop
331
332         \ Step through the string until we find the end
333         begin
334                 0 kernlen ! \ initialize length of value
335
336                 \ Skip leading whitespace and/or comma delimiters
337                 begin
338                         dup 0<> if
339                                 over c@ delim? ( c-addr/u -- c-addr/u bool )
340                         else
341                                 false ( c-addr/u -- c-addr/u bool )
342                         then
343                 while
344                         1- swap 1+ swap ( c-addr/u -- c-addr'/u' )
345                 repeat
346                 ( c-addr/u -- c-addr'/u' )
347
348                 dup 0= if \ end of string while eating whitespace
349                         2drop ( c-addr/u -- )
350                         kernmenuidx @ [char] 0 <> if \ found at least one
351                                 exit \ all done
352                         then
353
354                         \ No entries in $kernels; use $kernel instead
355                         s" kernel" getenv dup -1 = if
356                                 drop ( cruft ) s" "
357                         then ( -- c-addr/u )
358                         dup kernlen ! \ store entire value length as kernlen
359                 else
360                         \ We're still within $kernels parsing toward the end;
361                         \ find delimiter/end to determine kernlen
362                         2dup ( c-addr/u -- c-addr/u c-addr/u )
363                         begin dup 0<> while
364                                 over c@ delim? if
365                                         drop 0 ( break ) \ found delimiter
366                                 else
367                                         kernlen @ 1+ kernlen ! \ incrememnt
368                                         1- swap 1+ swap \ c-addr++ u--
369                                 then
370                         repeat
371                         2drop ( c-addr/u c-addr'/u' -- c-addr/u )
372
373                         \ If this is the first entry, compare it to $kernel
374                         \ If different, then insert $kernel beforehand
375                         kernmenuidx @ [char] 0 = if
376                                 over kernlen @ kerndefault count compare if
377                                         kernelsbuf 0 kerndefault count strcat
378                                         s" ," strcat 2swap strcat
379                                         kerndefault count swap drop kernlen !
380                                 then
381                         then
382                 then
383                 ( c-addr/u -- c-addr'/u' )
384
385                 \ At this point, we should have something on the stack to store
386                 \ as the next kernel menu option; start assembling variables
387
388                 over kernlen @ ( c-addr/u -- c-addr/u c-addr/u2 )
389
390                 \ Assign first to kernel[x]
391                 2dup kernmenuidx @ kernel[x] setenv
392
393                 \ Assign second to menu_caption[x][y]
394                 kerncapbuf 0 s" [K]ernel: " strcat
395                 2over strcat
396                 kernidx @ kernmenuidx @ menu_caption[x][y]
397                 setenv
398
399                 \ Assign third to ansi_caption[x][y]
400                 kerncapbuf 0 s" @[1mK@[mernel: " [char] @ escc! strcat
401                 kernmenuidx @ [char] 0 = if
402                         s" default/@[32m"
403                 else
404                         s" @[34;1m"
405                 then
406                 [char] @ escc! strcat
407                 2over strcat
408                 s" @[m" [char] @ escc! strcat
409                 kernidx @ kernmenuidx @ ansi_caption[x][y]
410                 setenv
411
412                 2drop ( c-addr/u c-addr/u2 -- c-addr/u )
413
414                 kernmenuidx @ 1+ dup kernmenuidx ! [char] 8 > if
415                         2drop ( c-addr/u -- ) exit
416                 then
417
418                 kernlen @ - swap kernlen @ + swap ( c-addr/u -- c-addr'/u' )
419         again
420 ;
421
422 \ This function goes through the kernels that were discovered by the
423 \ parse-kernels function [above], adding " (# of #)" text to the end of each
424 \ caption.
425
426 : tag-kernels ( -- )
427         kernidx @ ( -- x ) dup 0= if exit then
428         [char] 0 s"  (Y of Z)" ( x -- x y c-addr/u )
429         kernmenuidx @ -rot 7 +c! \ Replace 'Z' with number of kernels parsed
430         begin
431                 2 pick 1+ -rot 2 +c! \ Replace 'Y' with current ASCII num
432
433                 2over menu_caption[x][y] getenv dup -1 <> if
434                         2dup + 1- c@ [char] ) = if
435                                 2drop \ Already tagged
436                         else
437                                 kerncapbuf 0 2swap strcat
438                                 2over strcat
439                                 5 pick 5 pick menu_caption[x][y] setenv
440                         then
441                 else
442                         drop ( getenv cruft )
443                 then
444
445                 2over ansi_caption[x][y] getenv dup -1 <> if
446                         2dup + 1- c@ [char] ) = if
447                                 2drop \ Already tagged
448                         else
449                                 kerncapbuf 0 2swap strcat
450                                 2over strcat
451                                 5 pick 5 pick ansi_caption[x][y] setenv
452                         then
453                 else
454                         drop ( getenv cruft )
455                 then
456
457                 rot 1+ dup [char] 8 > if
458                         -rot 2drop TRUE ( break )
459                 else
460                         -rot FALSE
461                 then
462         until
463         2drop ( x y -- )
464 ;
465
466 \ This function creates the list of menu items. This function is called by the
467 \ menu-display function. You need not call it directly.
468
469 : menu-create ( -- )
470
471         \ Print the frame caption at (x,y)
472         s" loader_menu_title" getenv dup -1 = if
473                 drop s" Welcome to FreeBSD"
474         then
475         TRUE ( use default alignment )
476         s" loader_menu_title_align" getenv dup -1 <> if
477                 2dup s" left" compare-insensitive 0= if ( 1 )
478                         2drop ( c-addr/u ) drop ( bool )
479                         menuX @ menuY @ 1-
480                         FALSE ( don't use default alignment )
481                 else ( 1 ) 2dup s" right" compare-insensitive 0= if ( 2 )
482                         2drop ( c-addr/u ) drop ( bool )
483                         menuX @ 42 + 4 - over - menuY @ 1-
484                         FALSE ( don't use default alignment )
485                 else ( 2 ) 2drop ( c-addr/u ) then ( 1 ) then
486         else
487                 drop ( getenv cruft )
488         then
489         if ( use default center alignement? )
490                 menuX @ 19 + over 2 / - menuY @ 1-
491         then
492         swap 1- swap
493         at-xy dup 0= if
494                 2drop ( empty loader_menu_title )
495         else
496                 space type space
497         then
498
499         \ If $menu_init is set, evaluate it (allowing for whole menus to be
500         \ constructed dynamically -- as this function could conceivably set
501         \ the remaining environment variables to construct the menu entirely).
502         \ 
503         s" menu_init" getenv dup -1 <> if
504                 evaluate
505         else
506                 drop
507         then
508
509         \ Print our menu options with respective key/variable associations.
510         \ `printmenuitem' ends by adding the decimal ASCII value for the
511         \ numerical prefix to the stack. We store the value left on the stack
512         \ to the key binding variable for later testing against a character
513         \ captured by the `getkey' function.
514
515         \ Note that any menu item beyond 9 will have a numerical prefix on the
516         \ screen consisting of the first digit (ie. 1 for the tenth menu item)
517         \ and the key required to activate that menu item will be the decimal
518         \ ASCII of 48 plus the menu item (ie. 58 for the tenth item, aka. `:')
519         \ which is misleading and not desirable.
520         \ 
521         \ Thus, we do not allow more than 8 configurable items on the menu
522         \ (with "Reboot" as the optional ninth and highest numbered item).
523
524         \ 
525         \ Initialize the ACPI option status.
526         \ 
527         0 menuacpi !
528         s" menu_acpi" getenv -1 <> if
529                 c@ dup 48 > over 57 < and if ( '1' <= c1 <= '8' )
530                         menuacpi !
531                         arch-i386? if acpipresent? if
532                                 \ 
533                                 \ Set menu toggle state to active state
534                                 \ (required by generic toggle_menuitem)
535                                 \ 
536                                 acpienabled? menuacpi @ toggle_stateN !
537                         then then
538                 else
539                         drop
540                 then
541         then
542
543         \ 
544         \ Initialize kernel captions after parsing $kernels
545         \ 
546         0 menukernel !
547         s" menu_kernel" getenv -1 <> if
548                 c@ dup 48 > over 57 < and if ( '1' <= c1 <= '8' )
549                         dup menukernel !
550                         dup parse-kernels tag-kernels
551
552                         \ Get the current cycle state (entry to use)
553                         s" kernel_state" evaluate @ 48 + ( n -- n y )
554
555                         \ If state is invalid, reset
556                         dup kernmenuidx @ 1- > if
557                                 drop [char] 0 ( n y -- n 48 )
558                                 0 s" kernel_state" evaluate !
559                                 over s" init_kernel" evaluate drop
560                         then
561
562                         \ Set the current non-ANSI caption
563                         2dup swap dup ( n y -- n y y n n )
564                         s" set menu_caption[x]=$menu_caption[x][y]"
565                         17 +c! 34 +c! 37 +c! evaluate
566                         ( n y y n n c-addr/u -- n y  )
567
568                         \ Set the current ANSI caption
569                         2dup swap dup ( n y -- n y y n n )
570                         s" set ansi_caption[x]=$ansi_caption[x][y]"
571                         17 +c! 34 +c! 37 +c! evaluate
572                         ( n y y n n c-addr/u -- n y )
573
574                         \ Initialize cycle state from stored value
575                         48 - ( n y -- n k )
576                         s" init_cyclestate" evaluate ( n k -- n )
577
578                         \ Set $kernel to $kernel[y]
579                         s" activate_kernel" evaluate ( n -- n )
580                 then
581                 drop
582         then
583
584         \ 
585         \ Initialize the menu_options visual separator.
586         \ 
587         0 menuoptions !
588         s" menu_options" getenv -1 <> if
589                 c@ dup 48 > over 57 < and if ( '1' <= c1 <= '8' )
590                         menuoptions !
591                 else
592                         drop
593                 then
594         then
595
596         \ Initialize "Reboot" menu state variable (prevents double-entry)
597         false menurebootadded !
598
599         menu_start
600         1- menuidx !    \ Initialize the starting index for the menu
601         0 menurow !     \ Initialize the starting position for the menu
602
603         49 \ Iterator start (loop range 49 to 56; ASCII '1' to '8')
604         begin
605                 \ If the "Options:" separator, print it.
606                 dup menuoptions @ = if
607                         \ Optionally add a reboot option to the menu
608                         s" menu_reboot" getenv -1 <> if
609                                 drop
610                                 s" Reboot" printmenuitem menureboot !
611                                 true menurebootadded !
612                         then
613
614                         menuX @
615                         menurow @ 2 + menurow !
616                         menurow @ menuY @ +
617                         at-xy
618                         s" menu_optionstext" getenv dup -1 <> if
619                                 type
620                         else
621                                 drop ." Options:"
622                         then
623                 then
624
625                 \ If this is the ACPI menu option, act accordingly.
626                 dup menuacpi @ = if
627                         dup acpimenuitem ( n -- n n c-addr/u | n n -1 )
628                         dup -1 <> if
629                                 13 +c! ( n n c-addr/u -- n c-addr/u )
630                                        \ replace 'x' with n
631                         else
632                                 swap drop ( n n -1 -- n -1 )
633                                 over menu_command[x] unsetenv
634                         then
635                 else
636                         \ make sure we have not already initialized this item
637                         dup init_stateN dup @ 0= if
638                                 1 swap !
639
640                                 \ If this menuitem has an initializer, run it
641                                 dup menu_init[x]
642                                 getenv dup -1 <> if
643                                         evaluate
644                                 else
645                                         drop
646                                 then
647                         else
648                                 drop
649                         then
650
651                         dup
652                         loader_color? if
653                                 ansi_caption[x]
654                         else
655                                 menu_caption[x]
656                         then
657                 then
658
659                 dup -1 <> if
660                         \ test for environment variable
661                         getenv dup -1 <> if
662                                 printmenuitem ( c-addr/u -- n )
663                                 dup menukeyN !
664                         else
665                                 drop
666                         then
667                 else
668                         drop
669                 then
670
671                 1+ dup 56 > \ add 1 to iterator, continue if less than 57
672         until
673         drop \ iterator
674
675         \ Optionally add a reboot option to the menu
676         menurebootadded @ true <> if
677                 s" menu_reboot" getenv -1 <> if
678                         drop       \ no need for the value
679                         s" Reboot" \ menu caption (required by printmenuitem)
680
681                         printmenuitem
682                         menureboot !
683                 else
684                         0 menureboot !
685                 then
686         then
687 ;
688
689 \ Takes a single integer on the stack and updates the timeout display. The
690 \ integer must be between 0 and 9 (we will only update a single digit in the
691 \ source message).
692
693 : menu-timeout-update ( N -- )
694
695         \ Enforce minimum/maximum
696         dup 9 > if drop 9 then
697         dup 0 < if drop 0 then
698
699         s" Autoboot in N seconds. [Space] to pause" ( n -- n c-addr/u )
700
701         2 pick 0> if
702                 rot 48 + -rot ( n c-addr/u -- n' c-addr/u ) \ convert to ASCII
703                 12 +c!        ( n' c-addr/u -- c-addr/u )   \ replace 'N' above
704
705                 menu_timeout_x @ menu_timeout_y @ at-xy \ position cursor
706                 type ( c-addr/u -- ) \ print message
707         else
708                 menu_timeout_x @ menu_timeout_y @ at-xy \ position cursor
709                 spaces ( n c-addr/u -- n c-addr ) \ erase message
710                 2drop ( n c-addr -- )
711         then
712
713         0 25 at-xy ( position cursor back at bottom-left )
714 ;
715
716 \ This function blocks program flow (loops forever) until a key is pressed.
717 \ The key that was pressed is added to the top of the stack in the form of its
718 \ decimal ASCII representation. This function is called by the menu-display
719 \ function. You need not call it directly.
720
721 : getkey ( -- ascii_keycode )
722
723         begin \ loop forever
724
725                 menu_timeout_enabled @ 1 = if
726                         ( -- )
727                         seconds ( get current time: -- N )
728                         dup menu_time @ <> if ( has time elapsed?: N N N -- N )
729
730                                 \ At least 1 second has elapsed since last loop
731                                 \ so we will decrement our "timeout" (really a
732                                 \ counter, insuring that we do not proceed too
733                                 \ fast) and update our timeout display.
734
735                                 menu_time ! ( update time record: N -- )
736                                 menu_timeout @ ( "time" remaining: -- N )
737                                 dup 0> if ( greater than 0?: N N 0 -- N )
738                                         1- ( decrement counter: N -- N )
739                                         dup menu_timeout !
740                                                 ( re-assign: N N Addr -- N )
741                                 then
742                                 ( -- N )
743
744                                 dup 0= swap 0< or if ( N <= 0?: N N -- )
745                                         \ halt the timer
746                                         0 menu_timeout ! ( 0 Addr -- )
747                                         0 menu_timeout_enabled ! ( 0 Addr -- )
748                                 then
749
750                                 \ update the timer display ( N -- )
751                                 menu_timeout @ menu-timeout-update
752
753                                 menu_timeout @ 0= if
754                                         \ We've reached the end of the timeout
755                                         \ (user did not cancel by pressing ANY
756                                         \ key)
757
758                                         s" menu_timeout_command"  getenv dup
759                                         -1 = if
760                                                 drop \ clean-up
761                                         else
762                                                 evaluate
763                                         then
764                                 then
765
766                         else ( -- N )
767                                 \ No [detectable] time has elapsed (in seconds)
768                                 drop ( N -- )
769                         then
770                         ( -- )
771                 then
772
773                 key? if \ Was a key pressed? (see loader(8))
774
775                         \ An actual key was pressed (if the timeout is running,
776                         \ kill it regardless of which key was pressed)
777                         menu_timeout @ 0<> if
778                                 0 menu_timeout !
779                                 0 menu_timeout_enabled !
780
781                                 \ clear screen of timeout message
782                                 0 menu-timeout-update
783                         then
784
785                         \ get the key that was pressed and exit (if we
786                         \ get a non-zero ASCII code)
787                         key dup 0<> if
788                                 exit
789                         else
790                                 drop
791                         then
792                 then
793                 50 ms \ sleep for 50 milliseconds (see loader(8))
794
795         again
796 ;
797
798 : menu-erase ( -- ) \ Erases menu and resets positioning variable to position 1.
799
800         \ Clear the screen area associated with the interactive menu
801         menuX @ menuY @
802         2dup at-xy 38 spaces 1+         2dup at-xy 38 spaces 1+
803         2dup at-xy 38 spaces 1+         2dup at-xy 38 spaces 1+
804         2dup at-xy 38 spaces 1+         2dup at-xy 38 spaces 1+
805         2dup at-xy 38 spaces 1+         2dup at-xy 38 spaces 1+
806         2dup at-xy 38 spaces 1+         2dup at-xy 38 spaces 1+
807         2dup at-xy 38 spaces 1+         2dup at-xy 38 spaces
808         2drop
809
810         \ Reset the starting index and position for the menu
811         menu_start 1- menuidx !
812         0 menurow !
813 ;
814
815 only forth
816 also menu-infrastructure
817 also menu-namespace
818 also menu-command-helpers definitions
819
820 : toggle_menuitem ( N -- N ) \ toggles caption text and internal menuitem state
821
822         \ ASCII numeral equal to user-selected menu item must be on the stack.
823         \ We do not modify the stack, so the ASCII numeral is left on top.
824
825         dup init_textN c@ 0= if
826                 \ NOTE: no need to check toggle_stateN since the first time we
827                 \ are called, we will populate init_textN. Further, we don't
828                 \ need to test whether menu_caption[x] (ansi_caption[x] when
829                 \ loader_color?=1) is available since we would not have been
830                 \ called if the caption was NULL.
831
832                 \ base name of environment variable
833                 dup ( n -- n n ) \ key pressed
834                 loader_color? if
835                         ansi_caption[x]
836                 else
837                         menu_caption[x]
838                 then    
839                 getenv dup -1 <> if
840
841                         2 pick ( n c-addr/u -- n c-addr/u n )
842                         init_textN ( n c-addr/u n -- n c-addr/u c-addr )
843
844                         \ now we have the buffer c-addr on top
845                         \ ( followed by c-addr/u of current caption )
846
847                         \ Copy the current caption into our buffer
848                         2dup c! -rot \ store strlen at first byte
849                         begin
850                                 rot 1+    \ bring alt addr to top and increment
851                                 -rot -rot \ bring buffer addr to top
852                                 2dup c@ swap c! \ copy current character
853                                 1+     \ increment buffer addr
854                                 rot 1- \ bring buffer len to top and decrement
855                                 dup 0= \ exit loop if buffer len is zero
856                         until
857                         2drop \ buffer len/addr
858                         drop  \ alt addr
859
860                 else
861                         drop
862                 then
863         then
864
865         \ Now we are certain to have init_textN populated with the initial
866         \ value of menu_caption[x] (ansi_caption[x] with loader_color enabled).
867         \ We can now use init_textN as the untoggled caption and
868         \ toggled_text[x] (toggled_ansi[x] with loader_color enabled) as the
869         \ toggled caption and store the appropriate value into menu_caption[x]
870         \ (again, ansi_caption[x] with loader_color enabled). Last, we'll
871         \ negate the toggled state so that we reverse the flow on subsequent
872         \ calls.
873
874         dup toggle_stateN @ 0= if
875                 \ state is OFF, toggle to ON
876
877                 dup ( n -- n n ) \ key pressed
878                 loader_color? if
879                         toggled_ansi[x]
880                 else
881                         toggled_text[x]
882                 then
883                 getenv dup -1 <> if
884                         \ Assign toggled text to menu caption
885                         2 pick ( n c-addr/u -- n c-addr/u n ) \ key pressed
886                         loader_color? if
887                                 ansi_caption[x]
888                         else
889                                 menu_caption[x]
890                         then
891                         setenv
892                 else
893                         \ No toggled text, keep the same caption
894                         drop ( n -1 -- n ) \ getenv cruft
895                 then
896
897                 true \ new value of toggle state var (to be stored later)
898         else
899                 \ state is ON, toggle to OFF
900
901                 dup init_textN count ( n -- n c-addr/u )
902
903                 \ Assign init_textN text to menu caption
904                 2 pick ( n c-addr/u -- n c-addr/u n ) \ key pressed
905                 loader_color? if
906                         ansi_caption[x]
907                 else
908                         menu_caption[x]
909                 then
910                 setenv
911
912                 false \ new value of toggle state var (to be stored below)
913         then
914
915         \ now we'll store the new toggle state (on top of stack)
916         over toggle_stateN !
917 ;
918
919 : cycle_menuitem ( N -- N ) \ cycles through array of choices for a menuitem
920
921         \ ASCII numeral equal to user-selected menu item must be on the stack.
922         \ We do not modify the stack, so the ASCII numeral is left on top.
923
924         dup cycle_stateN dup @ 1+ \ get value and increment
925
926         \ Before assigning the (incremented) value back to the pointer,
927         \ let's test for the existence of this particular array element.
928         \ If the element exists, we'll store index value and move on.
929         \ Otherwise, we'll loop around to zero and store that.
930
931         dup 48 + ( n addr k -- n addr k k' )
932                  \ duplicate array index and convert to ASCII numeral
933
934         3 pick swap ( n addr k k' -- n addr k n k' ) \ (n,k') as (x,y)
935         loader_color? if
936                 ansi_caption[x][y]
937         else
938                 menu_caption[x][y]
939         then
940         ( n addr k n k' -- n addr k c-addr/u )
941
942         \ Now test for the existence of our incremented array index in the
943         \ form of $menu_caption[x][y] ($ansi_caption[x][y] with loader_color
944         \ enabled) as set in loader.rc(5), et. al.
945
946         getenv dup -1 = if
947                 \ No caption set for this array index. Loop back to zero.
948
949                 drop ( n addr k -1 -- n addr k ) \ getenv cruft
950                 drop 0 ( n addr k -- n addr 0 )  \ new value to store later
951
952                 2 pick [char] 0 ( n addr 0 -- n addr 0 n 48 ) \ (n,48) as (x,y)
953                 loader_color? if
954                         ansi_caption[x][y]
955                 else
956                         menu_caption[x][y]
957                 then
958                 ( n addr 0 n 48 -- n addr 0 c-addr/u )
959                 getenv dup -1 = if
960                         \ Highly unlikely to occur, but to ensure things move
961                         \ along smoothly, allocate a temporary NULL string
962                         drop ( cruft ) s" "
963                 then
964         then
965
966         \ At this point, we should have the following on the stack (in order,
967         \ from bottom to top):
968         \ 
969         \    n        - Ascii numeral representing the menu choice (inherited)
970         \    addr     - address of our internal cycle_stateN variable
971         \    k        - zero-based number we intend to store to the above
972         \    c-addr/u - string value we intend to store to menu_caption[x]
973         \               (or ansi_caption[x] with loader_color enabled)
974         \ 
975         \ Let's perform what we need to with the above.
976
977         \ Assign array value text to menu caption
978         4 pick ( n addr k c-addr/u -- n addr k c-addr/u n )
979         loader_color? if
980                 ansi_caption[x]
981         else
982                 menu_caption[x]
983         then
984         setenv
985
986         swap ! ( n addr k -- n ) \ update array state variable
987 ;
988
989 only forth definitions also menu-infrastructure
990
991 \ Erase and redraw the menu. Useful if you change a caption and want to
992 \ update the menu to reflect the new value.
993
994 : menu-redraw ( -- )
995         menu-erase
996         menu-create
997 ;
998
999 \ This function initializes the menu. Call this from your `loader.rc' file
1000 \ before calling any other menu-related functions.
1001
1002 : menu-init ( -- )
1003         menu_start
1004         1- menuidx !    \ Initialize the starting index for the menu
1005         0 menurow !     \ Initialize the starting position for the menu
1006
1007         \ Assign configuration values
1008         s" loader_menu_y" getenv dup -1 = if
1009                 drop \ no custom row position
1010                 menu_default_y
1011         else
1012                 \ make sure custom position is a number
1013                 ?number 0= if
1014                         menu_default_y \ or use default
1015                 then
1016         then
1017         menuY !
1018         s" loader_menu_x" getenv dup -1 = if
1019                 drop \ no custom column position
1020                 menu_default_x
1021         else
1022                 \ make sure custom position is a number
1023                 ?number 0= if
1024                         menu_default_x \ or use default
1025                 then
1026         then
1027         menuX !
1028
1029         \ Interpret a custom frame type for the menu
1030         TRUE ( draw a box? default yes, but might be altered below )
1031         s" loader_menu_frame" getenv dup -1 = if ( 1 )
1032                 drop \ no custom frame type
1033         else ( 1 )  2dup s" single" compare-insensitive 0= if ( 2 )
1034                 f_single ( see frames.4th )
1035         else ( 2 )  2dup s" double" compare-insensitive 0= if ( 3 )
1036                 f_double ( see frames.4th )
1037         else ( 3 ) s" none" compare-insensitive 0= if ( 4 )
1038                 drop FALSE \ don't draw a box
1039         ( 4 ) then ( 3 ) then ( 2 )  then ( 1 ) then
1040         if
1041                 42 13 menuX @ 3 - menuY @ 1- box \ Draw frame (w,h,x,y)
1042         then
1043
1044         0 25 at-xy \ Move cursor to the bottom for output
1045 ;
1046
1047 also menu-namespace
1048
1049 \ Main function. Call this from your `loader.rc' file.
1050
1051 : menu-display ( -- )
1052
1053         0 menu_timeout_enabled ! \ start with automatic timeout disabled
1054
1055         \ check indication that automatic execution after delay is requested
1056         s" menu_timeout_command" getenv -1 <> if ( Addr C -1 -- | Addr )
1057                 drop ( just testing existence right now: Addr -- )
1058
1059                 \ initialize state variables
1060                 seconds menu_time ! ( store the time we started )
1061                 1 menu_timeout_enabled ! ( enable automatic timeout )
1062
1063                 \ read custom time-duration (if set)
1064                 s" autoboot_delay" getenv dup -1 = if
1065                         drop \ no custom duration (remove dup'd bunk -1)
1066                         menu_timeout_default \ use default setting
1067                 else
1068                         2dup ?number 0= if ( if not a number )
1069                                 \ disable timeout if "NO", else use default
1070                                 s" NO" compare-insensitive 0= if
1071                                         0 menu_timeout_enabled !
1072                                         0 ( assigned to menu_timeout below )
1073                                 else
1074                                         menu_timeout_default
1075                                 then
1076                         else
1077                                 -rot 2drop
1078
1079                                 \ boot immediately if less than zero
1080                                 dup 0< if
1081                                         drop
1082                                         menu-create
1083                                         0 25 at-xy
1084                                         0 boot
1085                                 then
1086                         then
1087                 then
1088                 menu_timeout ! ( store value on stack from above )
1089
1090                 menu_timeout_enabled @ 1 = if
1091                         \ read custom column position (if set)
1092                         s" loader_menu_timeout_x" getenv dup -1 = if
1093                                 drop \ no custom column position
1094                                 menu_timeout_default_x \ use default setting
1095                         else
1096                                 \ make sure custom position is a number
1097                                 ?number 0= if
1098                                         menu_timeout_default_x \ or use default
1099                                 then
1100                         then
1101                         menu_timeout_x ! ( store value on stack from above )
1102         
1103                         \ read custom row position (if set)
1104                         s" loader_menu_timeout_y" getenv dup -1 = if
1105                                 drop \ no custom row position
1106                                 menu_timeout_default_y \ use default setting
1107                         else
1108                                 \ make sure custom position is a number
1109                                 ?number 0= if
1110                                         menu_timeout_default_y \ or use default
1111                                 then
1112                         then
1113                         menu_timeout_y ! ( store value on stack from above )
1114                 then
1115         then
1116
1117         menu-create
1118
1119         begin \ Loop forever
1120
1121                 0 25 at-xy \ Move cursor to the bottom for output
1122                 getkey     \ Block here, waiting for a key to be pressed
1123
1124                 dup -1 = if
1125                         drop exit \ Caught abort (abnormal return)
1126                 then
1127
1128                 \ Boot if the user pressed Enter/Ctrl-M (13) or
1129                 \ Ctrl-Enter/Ctrl-J (10)
1130                 dup over 13 = swap 10 = or if
1131                         drop ( no longer needed )
1132                         s" boot" evaluate
1133                         exit ( pedantic; never reached )
1134                 then
1135
1136                 dup menureboot @ = if 0 reboot then
1137
1138                 \ Evaluate the decimal ASCII value against known menu item
1139                 \ key associations and act accordingly
1140
1141                 49 \ Iterator start (loop range 49 to 56; ASCII '1' to '8')
1142                 begin
1143                         dup menukeyN @
1144                         rot tuck = if
1145
1146                                 \ Adjust for missing ACPI menuitem on non-i386
1147                                 arch-i386? true <> menuacpi @ 0<> and if
1148                                         menuacpi @ over 2dup < -rot = or
1149                                         over 58 < and if
1150                                         ( key >= menuacpi && key < 58: N -- N )
1151                                                 1+
1152                                         then
1153                                 then
1154
1155                                 \ Test for the environment variable
1156                                 dup menu_command[x]
1157                                 getenv dup -1 <> if
1158                                         \ Execute the stored procedure
1159                                         evaluate
1160
1161                                         \ We expect there to be a non-zero
1162                                         \  value left on the stack after
1163                                         \ executing the stored procedure.
1164                                         \ If so, continue to run, else exit.
1165
1166                                         0= if
1167                                                 drop \ key pressed
1168                                                 drop \ loop iterator
1169                                                 exit
1170                                         else
1171                                                 swap \ need iterator on top
1172                                         then
1173                                 then
1174
1175                                 \ Re-adjust for missing ACPI menuitem
1176                                 arch-i386? true <> menuacpi @ 0<> and if
1177                                         swap
1178                                         menuacpi @ 1+ over 2dup < -rot = or
1179                                         over 59 < and if
1180                                                 1-
1181                                         then
1182                                         swap
1183                                 then
1184                         else
1185                                 swap \ need iterator on top
1186                         then
1187
1188                         \ 
1189                         \ Check for menu keycode shortcut(s)
1190                         \ 
1191                         dup menu_keycode[x]
1192                         getenv dup -1 = if
1193                                 drop
1194                         else
1195                                 ?number 0<> if
1196                                         rot tuck = if
1197                                                 swap
1198                                                 dup menu_command[x]
1199                                                 getenv dup -1 <> if
1200                                                         evaluate
1201                                                         0= if
1202                                                                 2drop
1203                                                                 exit
1204                                                         then
1205                                                 else
1206                                                         drop
1207                                                 then
1208                                         else
1209                                                 swap
1210                                         then
1211                                 then
1212                         then
1213
1214                         1+ dup 56 > \ increment iterator
1215                                     \ continue if less than 57
1216                 until
1217                 drop \ loop iterator
1218                 drop \ key pressed
1219
1220         again   \ Non-operational key was pressed; repeat
1221 ;
1222
1223 \ This function unsets all the possible environment variables associated with
1224 \ creating the interactive menu.
1225
1226 : menu-unset ( -- )
1227
1228         49 \ Iterator start (loop range 49 to 56; ASCII '1' to '8')
1229         begin
1230                 dup menu_init[x]    unsetenv    \ menu initializer
1231                 dup menu_command[x] unsetenv    \ menu command
1232                 dup menu_caption[x] unsetenv    \ menu caption
1233                 dup ansi_caption[x] unsetenv    \ ANSI caption
1234                 dup menu_keycode[x] unsetenv    \ menu keycode
1235                 dup toggled_text[x] unsetenv    \ toggle_menuitem caption
1236                 dup toggled_ansi[x] unsetenv    \ toggle_menuitem ANSI caption
1237
1238                 48 \ Iterator start (inner range 48 to 57; ASCII '0' to '9')
1239                 begin
1240                         \ cycle_menuitem caption and ANSI caption
1241                         2dup menu_caption[x][y] unsetenv
1242                         2dup ansi_caption[x][y] unsetenv
1243                         1+ dup 57 >
1244                 until
1245                 drop \ inner iterator
1246
1247                 0 over menukeyN      !  \ used by menu-create, menu-display
1248                 0 over init_stateN   !  \ used by menu-create
1249                 0 over toggle_stateN !  \ used by toggle_menuitem
1250                 0 over init_textN   c!  \ used by toggle_menuitem
1251                 0 over cycle_stateN  !  \ used by cycle_menuitem
1252
1253                 1+ dup 56 >     \ increment, continue if less than 57
1254         until
1255         drop \ iterator
1256
1257         s" menu_timeout_command" unsetenv       \ menu timeout command
1258         s" menu_reboot"          unsetenv       \ Reboot menu option flag
1259         s" menu_acpi"            unsetenv       \ ACPI menu option flag
1260         s" menu_kernel"          unsetenv       \ Kernel menu option flag
1261         s" menu_options"         unsetenv       \ Options separator flag
1262         s" menu_optionstext"     unsetenv       \ separator display text
1263         s" menu_init"            unsetenv       \ menu initializer
1264
1265         0 menureboot !
1266         0 menuacpi !
1267         0 menuoptions !
1268 ;
1269
1270 only forth definitions also menu-infrastructure
1271
1272 \ This function both unsets menu variables and visually erases the menu area
1273 \ in-preparation for another menu.
1274
1275 : menu-clear ( -- )
1276         menu-unset
1277         menu-erase
1278 ;
1279
1280 bullet menubllt !
1281
1282 also menu-namespace
1283
1284 \ Initialize our menu initialization state variables
1285 0 init_state1 !
1286 0 init_state2 !
1287 0 init_state3 !
1288 0 init_state4 !
1289 0 init_state5 !
1290 0 init_state6 !
1291 0 init_state7 !
1292 0 init_state8 !
1293
1294 \ Initialize our boolean state variables
1295 0 toggle_state1 !
1296 0 toggle_state2 !
1297 0 toggle_state3 !
1298 0 toggle_state4 !
1299 0 toggle_state5 !
1300 0 toggle_state6 !
1301 0 toggle_state7 !
1302 0 toggle_state8 !
1303
1304 \ Initialize our array state variables
1305 0 cycle_state1 !
1306 0 cycle_state2 !
1307 0 cycle_state3 !
1308 0 cycle_state4 !
1309 0 cycle_state5 !
1310 0 cycle_state6 !
1311 0 cycle_state7 !
1312 0 cycle_state8 !
1313
1314 \ Initialize string containers
1315 0 init_text1 c!
1316 0 init_text2 c!
1317 0 init_text3 c!
1318 0 init_text4 c!
1319 0 init_text5 c!
1320 0 init_text6 c!
1321 0 init_text7 c!
1322 0 init_text8 c!
1323
1324 only forth definitions