Chinaunix首页 | 论坛 | 博客

孤星赶月zkaka421.blog.chinaunix.net

首页　| 　博文目录　| 　关于我

zkaka421

  • 博客访问： 32130
  • 博文数量： 12
  • 博客积分： 445
  • 博客等级： 下士
  • 技术积分： 150
  • 用 户 组： 普通用户
  • 注册时间： 2012-10-06 16:49
文章分类

全部博文（12）

文章存档

2012年（12）

我的朋友
最近访客
推荐博文
相关博文
u-boot-2010.03在XC2440开发板上的移植—13

分类： 嵌入式

2012-10-21 23:06:31

14、u-boot中添加快捷菜单

    在u-boot启动的时候，在3秒之内按下空格键就可以进入快捷菜单，效果如下：
    
    通过串口输入数字，就可以执行相应的功能，例如：启动内核，重启u-boot，下载镜像到NandFlash。

   这个菜单其实就是u-boot中添加的一个命令，u-boot启动时，如果进入下载模式，就先运行这个命令。

    添加一个名为cmd_menu.c的文件到common目录下，该文件的内容如下：
  1. #include
  2. #include
  3. #include <../drivers/usb/slave/def.h>
  4. #include

  6. extern char console_buffer[];
  7. extern int readline (const char *const prompt);
  8. extern char awaitkey(unsigned long delay, int* error_p);

  10. /**
  11.  * Parses a string into a number. The number stored at ptr is
  12.  * potentially suffixed with K (for kilobytes, or 1024 bytes),
  13.  * M (for megabytes, or 1048576 bytes), or G (for gigabytes, or
  14.  * 1073741824). If the number is suffixed with K, M, or G, then
  15.  * the return value is the number multiplied by one kilobyte, one
  16.  * megabyte, or one gigabyte, respectively.
  17.  *
  18.  * @param ptr where parse begins
  19.  * @param retptr output pointer to next char after parse completes (output)
  20.  * @return resulting unsigned int
  21.  */
  22. static unsigned long memsize_parse2 (const char *const ptr, const char **retptr)
  23. {
  24.     unsigned long ret = simple_strtoul(ptr, (char **)retptr, 0);
  25.     int sixteen = 1;

  27.     switch (**retptr) {
  28.         case 'G':
  29.         case 'g':
  30.             ret <<= 10;
  31.         case 'M':
  32.         case 'm':
  33.             ret <<= 10;
  34.         case 'K':
  35.         case 'k':
  36.             ret <<= 10;
  37.             (*retptr)++;
  38.             sixteen = 0;
  39.         default:
  40.             break;
  41.     }

  43.     if (sixteen)
  44.         return simple_strtoul(ptr, NULL, 16);
  45.     
  46.     return ret;
  47. }


  50. void param_menu_usage(void)
  51. {
  52.     printf("\r\n##### Parameter Menu #####\r\n");
  53.     printf("[1] 查看参数\r\n");
  54.     printf("[2] 设置参数 \r\n");
  55.     printf("[3] 删除参数 \r\n");
  56.     printf("[4] 写入参数到flash \r\n");
  57.     printf("[5] 返回 \r\n");
  58.     printf("Enter your selection: ");
  59. }


  62. void param_menu_shell(void)
  63. {
  64.     char c;
  65.     char cmd_buf[256];

  67.     while (1)
  68.     {
  69.         param_menu_usage();
  70.         c = awaitkey(-1, NULL);
  71.         printf("%c\n", c);
  72.         switch (c)
  73.         {
  74.             case '1':
  75.             {
  76.                 strcpy(cmd_buf, "printenv ");
  77.                 printf("Name(enter to view all paramters): ");
  78.                 readline(NULL);
  79.                 strcat(cmd_buf, console_buffer);
  80.                 run_command(cmd_buf, 0);
  81.                 break;
  82.             }
  83.             
  84.             case '2':
  85.             {
  86.                 sprintf(cmd_buf, "setenv ");

  88.                 printf("Name: ");
  89.                 readline(NULL);
  90.                 strcat(cmd_buf, console_buffer);

  92.                 printf("Value: ");
  93.                 readline(NULL);
  94.                 strcat(cmd_buf, " ");
  95.                 strcat(cmd_buf, console_buffer);

  97.                 run_command(cmd_buf, 0);
  98.                 break;
  99.             }
  100.             
  101.             case '3':
  102.             {
  103.                 sprintf(cmd_buf, "setenv ");

  105.                 printf("Name: ");
  106.                 readline(NULL);
  107.                 strcat(cmd_buf, console_buffer);

  109.                 run_command(cmd_buf, 0);
  110.                 break;
  111.             }
  112.             
  113.             case '4':
  114.             {
  115.                 sprintf(cmd_buf, "saveenv");
  116.                 run_command(cmd_buf, 0);
  117.                 break;
  118.             }
  119.             
  120.             case '5':
  121.             {
  122.                 return;
  123.                 break;
  124.             }
  125.         }
  126.     }
  127. }


  130. void erase_menu_usage(void)
  131. {
  132.     printf("\r\n##### Erase Nand Menu #####\r\n");
  133.     printf("[1] Nand scrub - really clean NAND erasing bad blocks (UNSAFE) \r\n");
  134.     printf("[2] Nand earse - clean NAND eraseing \r\n");
  135.     printf("[q] Return main Menu \r\n");
  136.     printf("Enter your selection: ");
  137. }


  140. void erase_menu_shell(void)
  141. {
  142.     char c;
  143.     char cmd_buf[256];
  144.     char *p = NULL;
  145.     unsigned long size;

  147.     while (1)
  148.     {
  149.         erase_menu_usage();
  150.         c = awaitkey(-1, NULL);
  151.         printf("%c\n", c);
  152.         switch (c)
  153.         {
  154.             case '1':
  155.             {
  156.                 strcpy(cmd_buf, "nand scrub ");
  157.                 run_command(cmd_buf, 0);
  158.                 break;
  159.             }

  161.             case '2':
  162.             {
  163.                 strcpy(cmd_buf, "nand erase ");

  165.                 printf("Start address: ");
  166.                 readline(NULL);
  167.                 strcat(cmd_buf, console_buffer);

  169.                 printf("Size(eg. 4000000, 0x4000000, 64m and so on): ");
  170.                 readline(NULL);
  171.                 p = console_buffer;
  172.                 size = memsize_parse2(p, &p);
  173.                 sprintf(console_buffer, " %x", size);
  174.                 strcat(cmd_buf, console_buffer);

  176.                 run_command(cmd_buf, 0);
  177.                 break;
  178.             }

  180.             case 'q':
  181.             {
  182.                 return;
  183.                 break;
  184.             }
  185.         }
  186.     }
  187. }

  189. void main_menu_usage(void)
  190. {
  191.     printf("[1] 烧写bootloader到nandflash\r\n");
  192.     printf("[2] 烧写Linux Kernel到nandflash\r\n");
  193.     printf("[3] 烧写yaffs2文件系统到nandflash\r\n");
  194.     printf("[4] 下载程序在SDRAM中运行\r\n");
  195.     printf("[5] 设置uboot参数\r\n");
  196.     printf("[6] 格式化Nandflash\r\n");
  197.     printf("[7] 启动Linux系统\r\n");
  198.     printf("[8] 进入Shell模式\r\n");
  199.     printf("[9] 重启uboot\r\n");
  200.     if (bBootFrmNORFlash == 1)
  201.         printf("[0] 下载bootloader到NorFlash\r\n");
  202.     printf("Enter your selection: ");
  203. }


  206. void menu_shell(void)
  207. {
  208.     char c;
  209.     char cmd_buf[200];
  210.     
  211.     while (1)
  212.     {
  213.         main_menu_usage();
  214.         c = awaitkey(-1, NULL);
  215.         printf("%c\n", c);
  216.         switch (c)
  217.         {
  218.             case '1': //download bootloader
  219.             {
  220.                 strcpy(cmd_buf, "usbslave 1 0x30000000; nand erase bios; nand write.jffs2 0x30000000 bios 0x100000");
  221.                 run_command(cmd_buf, 0);
  222.                 break;
  223.             }

  225.             case '2': //kernel
  226.             {
  227.                 strcpy(cmd_buf, "usbslave 1 0x30000000; nand erase kernel; nand write.jffs2 0x30000000 kernel 0x400000");
  228.                 run_command(cmd_buf, 0);                
  229.                 break;
  230.             }

  232.             case '3': //yaffs
  233.             {
  234.                 strcpy(cmd_buf, "usbslave 1 0x30000000; nand erase root; nand write.yaffs 0x30000000 root $(filesize)");
  235.                 run_command(cmd_buf, 0);
  236.                 break;
  237.             }

  239.             case '4': //sdram
  240.             {
  241.                 extern volatile U32 downloadAddress;
  242.                 extern int download_run;
  243.                 
  244.                 download_run = 1;
  245.                 strcpy(cmd_buf, "usbslave 1 0x30000000");
  246.                 run_command(cmd_buf, 0);
  247.                 download_run = 0;
  248.                 sprintf(cmd_buf, "go %x", downloadAddress);
  249.                 run_command(cmd_buf, 0);
  250.                 break;
  251.             }

  253.             case '5': //set param
  254.             {
  255.                 param_menu_shell();
  256.                 break;
  257.             }
  258.             
  259.             case '6': //format
  260.             {
  261.                 erase_menu_shell();
  262.                 break;
  263.             }

  265.             case '7': //boot
  266.             {
  267.                 printf("Start Linux ...\n");
  268.                 strcpy(cmd_buf, "nand read 0x30008000 0x120000 0x400000; go 0x30008000");
  269.                 run_command(cmd_buf, 0);
  270.                 break;
  271.             }

  273.             case '8': //quit
  274.             {
  275.                 return;    
  276.                 break;
  277.             }

  279.             case '9': //reset
  280.             {
  281.                 strcpy(cmd_buf, "reset");
  282.                 run_command(cmd_buf, 0);
  283.                 break;
  284.             }
  285.             
  286.             case '0': //download uboot to NOR
  287.             {
  288.                 if (bBootFrmNORFlash == 1)
  289.                 {
  290.                     strcpy(cmd_buf, "usbslave 1 0x30000000; protect off all; erase 0 +$(filesize); cp.b 0x30000000 0 $(filesize)");
  291.                     run_command(cmd_buf, 0);
  292.                 }
  293.                 break;
  294.             }

  296.         }
  297.                 
  298.     }
  299. }

  301. int do_menu (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  302. {
  303.     menu_shell();
  304.     return 0;
  305. }

  307. U_BOOT_CMD(
  308.     menu,    3,    0,    do_menu,
  309.     "menu - display a menu, to select the items to do something\n",
  310.     " - display a menu, to select the items to do something"
  311. );
     
    修改common目录下的Makefile文件，加入cmd_menu.c的编译选项，代码如下：
  1. COBJS-$(CONFIG_UPDATE_TFTP) += update.o
  2. COBJS-$(CONFIG_USB_KEYBOARD) += usb_kbd.o
  3. COBJS-$(CONFIG_CMD_MENU) += cmd_menu.o
    然后修改smdk2440.h文件，添加CONFIG_CMD_MENU命令，这样就会编译cmd_menu.c文件，修改的代码如下：
  1. #define CONFIG_CMD_CACHE
  2. #define CONFIG_CMD_DATE
  3. #define CONFIG_CMD_ELF
  4. #define CONFIG_CMD_NAND
  5. #define CONFIG_CMD_MTDPARTS 
  6. #define CONFIG_CMD_PING 
  7. #define CONFIG_CMD_MENU 

    在main.c的main_loop函数中调用menu命令，在bootdelay时间没有按下空格键，如果是NorFlash启动的话则运行menu命令，如果是NandFlash启动则启动linux内核，修改代码如下：

  1. if (bootdelay >= 0 && s && !abortboot (bootdelay)) {
  2. # ifdef CONFIG_AUTOBOOT_KEYED
  3.         int prev = disable_ctrlc(1);    /* disable Control C checking */
  4. # endif

  6. #ifdef CONFIG_CMD_MENU
  7.     if (bBootFrmNORFlash == 1)    
  8.     {
  9.         run_command("menu", 0);    
  10.     }
  11.     else
  12.     {
  13.         printf("Booting Linux ...\n");
  14.     }
  15. #else
  16.     printf("Booting Linux ...\n");
  17. #endif    
    
     u-boot源代码中，在bootdelay时间内按下空格键的话，则会进入shell命令行模式，我们修改代码使其运行menu命令，进入快捷菜单界面，修改的代码如下：

  1. #ifdef CONFIG_AMIGAONEG3SE
  2.     {
  3.      extern void video_banner(void);
  4.      video_banner();
  5.     }
  6. #endif

  8. #ifdef CONFIG_CMD_MENU
  9.     run_command("menu", 0);
  10. #endif

    另外，在start.S文件中定义变量bBootFrmNORFlash，如果是从NorFlash驱动的话，则把该变量置为1，修改的代码如下：
  1. .globl _bss_start
  2. _bss_start:
  3.     .word __bss_start

  5. .globl _bss_end
  6. _bss_end:
  7.     .word _end
  8.    
  9. .globl bBootFrmNORFlash      
  10. bBootFrmNORFlash:
  11.     .word 0

  在NorFlash启动的时候，把该变量置1
copy_loop:
ldmia r0!, {r3-r10} /* copy from source address [r0]    */
stmia r1!, {r3-r10} /* copy to   target address [r1]    */
cmp r0, r2 /* until source end addreee [r2]    */
ble copy_loop
#endif /* CONFIG_SKIP_RELOCATE_UBOOT */

SetBootFlag:
ldr r0, =bBootFrmNORFlash
mov r1, #1
str r1, [r0]

/* Set up the stack     */
stack_setup:
ldr r0, _TEXT_BASE /* upper 128 KiB: relocated uboot   */
sub r0, r0, #CONFIG_SYS_MALLOC_LEN
sub r0, r0, #CONFIG_SYS_GBL_DATA_SIZE /* bdinfo */


    修改include/asm-arm/目录下的u-boot-arm.h文件，这样在main.c里就可以使用bBootFrmNORFlash这个变量了，修改的代码如下：
  1. extern ulong IRQ_STACK_START;    /* top of IRQ stack */
  2. extern ulong FIQ_STACK_START;    /* top of FIQ stack */
  3. extern ulong bBootFrmNORFlash; 
    
    重新编译u-boot，下载到NandFlash中，就可以看到效果了。

    我们还可以继续修改代码，添加自己的打印消息，如下所示：

    在main.c中添加一个show_myinfo函数，修改的代码如下：
  1. static void show_myinfo(void)
  2. {
  3.     printf("****************************************\n");
  4.     printf("******* zjw 制作 2012.10.20 ********\n");
  5.     printf("*** u-boot-2010.03 for smdk2440 ***\n");
  6.     if (bBootFrmNORFlash == 1)
  7.         printf("*** NorFlash启动方式 ***\n");
  8.     else
  9.         printf("*** NandFlash启动方式 ***\n");
  10.     printf("****************************************\n");
  11. }

    在abortboot函数中调用show_myinfo函数，修改的代码如下：
  1. static __inline__ int abortboot(int bootdelay)
  2. {
  3.     int abort = 0;
  4.     show_myinfo(); 

    
    还可以在board.c中继续修改和添加自己的信息，修改version_string[]如下：
  1. const char version_string[] =
  2.     U_BOOT_VERSION"<-->"CONFIG_IDENT_STRING;
   
    在smdk2440.h中定义CONFIG_IDENT_STRING宏，代码如下：
  1. #define CONFIG_IDENT_STRING " zjw modify for SMDK2440 "
   
    修改board.c中的display_banner函数，修改的代码如下：
  1. static int display_banner (void)
  2. {
  3.     printf ("\n\n%s\n\n", version_string);
  4.     printf ("##########********** Board Info **********##########\n");
  5.     debug ("U-Boot code: %08lX -> %08lX BSS: -> %08lX\n",
  6.      _armboot_start, _bss_start, _bss_end);

    修改board.c中的start_armboot函数，修改的代码如下：
  1. printf ("##########******************************##########\n\n"); 
  2.     /* main_loop() can return to retry autoboot, if so just run it again. */
  3.     for (;;) {
  4.         main_loop ();
  5.     }

    最后重新编译u-boot，下载到NandFlas中运行，就会出现如本节最上面的图片一样的效果。

阅读(1192) | 评论(0) | 转发(1) |
0

上一篇：u-boot-2010.03在XC2440开发板上的移植—12

下一篇：没有了

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6