|  | @@ -20,6 +20,7 @@
 | 
		
	
		
			
			| 20 | 20 |  #include <console.h>
 | 
		
	
		
			
			| 21 | 21 |  #include <unistd.h>
 | 
		
	
		
			
			| 22 | 22 |  #include <config/general.h>
 | 
		
	
		
			
			|  | 23 | +#include <gpxe/keys.h>
 | 
		
	
		
			
			| 23 | 24 |  #include <gpxe/shell_banner.h>
 | 
		
	
		
			
			| 24 | 25 |  
 | 
		
	
		
			
			| 25 | 26 |  /** @file
 | 
		
	
	
		
			
			|  | @@ -34,22 +35,21 @@
 | 
		
	
		
			
			| 34 | 35 |   * @ret	enter_shell		User wants to enter shell
 | 
		
	
		
			
			| 35 | 36 |   */
 | 
		
	
		
			
			| 36 | 37 |  int shell_banner ( void ) {
 | 
		
	
		
			
			| 37 |  | -	int wait_count = 0;
 | 
		
	
		
			
			| 38 | 38 |  	int enter_shell = 0;
 | 
		
	
		
			
			|  | 39 | +	int wait_count;
 | 
		
	
		
			
			| 39 | 40 |  	int key;
 | 
		
	
		
			
			| 40 | 41 |  
 | 
		
	
		
			
			| 41 | 42 |  	printf ( "\nPress Ctrl-B for the gPXE command line..." );
 | 
		
	
		
			
			| 42 | 43 |  
 | 
		
	
		
			
			| 43 | 44 |  	/* Wait for key */
 | 
		
	
		
			
			| 44 |  | -	while ( wait_count < BANNER_TIMEOUT ) {
 | 
		
	
		
			
			|  | 45 | +	for ( wait_count = 0 ; wait_count < BANNER_TIMEOUT ; wait_count++ ) {
 | 
		
	
		
			
			| 45 | 46 |  		if ( iskey() ) {
 | 
		
	
		
			
			| 46 | 47 |  			key = getchar();
 | 
		
	
		
			
			| 47 |  | -			if ( key == 0x02 /* Ctrl-B */ )
 | 
		
	
		
			
			|  | 48 | +			if ( key == CTRL_B )
 | 
		
	
		
			
			| 48 | 49 |  				enter_shell = 1;
 | 
		
	
		
			
			| 49 | 50 |  			break;
 | 
		
	
		
			
			| 50 | 51 |  		}
 | 
		
	
		
			
			| 51 | 52 |  		mdelay(100);
 | 
		
	
		
			
			| 52 |  | -		wait_count++;
 | 
		
	
		
			
			| 53 | 53 |  	}
 | 
		
	
		
			
			| 54 | 54 |  
 | 
		
	
		
			
			| 55 | 55 |  	/* Clear the "Press Ctrl-B" line */
 |