动态生成菜单的程序(配置文件驱动)

libinary · 发表于 2003-7-30 13:19:46

前两天突然想写个程序练练手，大程序一下又写不了，就写了一个动态生成菜单的小程序，大家看看吧，说不定有点参考价值。
因为是突然想起来写的，所以程序的结构一开始没有好好规划，而且一开始也没有注释，现在有的都是后来加上的。

程序主要有3个模块：
主模块：
文件：main.c
说明：包含main函数，比较简单
menu模块：
文件：menu.h、menu.c
说明：菜单的初始化和删除
display模块：
文件：display.h display.c
说明：显示菜单，接受输入
其他文件：
menudef.h：包含一些常数和结构的定义
my.c：动态库(菜单中用到的函数)
menurc：测试用的配置文件

代码还有一些问题，不是很满意，作为半成品大家讨论一下吧。

libinary · 发表于 2003-7-30 13:22:32

menudef.h文件：

#ifndef _MENUDEF_H_
#define _MENUDEF_H_
#define MAXWIDTH 80 /* 菜单最大宽度 */
#define MAXCOUNT 25 /* 最大菜单项数 */
#define MAXLINE 1024 /* 读入行最大长度 */
#define MAXPATH 256 /* 文件名(含路径)最大长度 */
enum{ /* 常数定义 */
ERR_ITEM = 1, /* MENUITEM::options 错误的菜单项 */
SUBMENU = 2, /* MENUITEM::options 子菜单 */
FUNCTION = 3, /* MENUITEM::options 动态库函数 */
COMMAND = 4, /* MENUITEM::options 外部命令 */
SYSMENU = 0x100, /* MENUITEM::options 系统菜单 */
SYS_EXIT = 0, /* MENUITEM::options 系统菜单，退出 */
SYS_ROOT = 1, /* MENUITEM::options 系统菜单，root菜单 */
SYS_RET = 2, /* MENUITEM::options 系统菜单，上层菜单 */
};
struct MENUITEM{ /* 菜单项 */
char name[MAXWIDTH + 1];
char accel;
void *proc;
int options;
};
struct MENU{ /* 菜单 */
char name[MAXWIDTH + 1];
int count;
struct MENUITEM item[MAXCOUNT + 2];
};
struct MENUSTACK{ /* 菜单堆栈 */
struct MENU *stack[100];
int current;
};
#endif /* _MENUDEF_H_ */

复制代码

libinary · 发表于 2003-7-30 13:24:14

main.c文件：

#include <stdio.h>
#include <string.h>
#include <libgen.h>
#include "menudef.h"
#include "menu.h"
#include "display.h"
extern char rcpath[MAXPATH];
extern char libpath[MAXPATH];
void getexepath(char *);
int
main(void)
{
char exepath[MAXPATH];
char menurc[MAXPATH];
/* 得到程序路径，并将配置文件路径和库文件路径都设为程序路径 */
getexepath(exepath);
strncpy(rcpath, exepath, MAXPATH - 1);
strncpy(libpath, exepath, MAXPATH - 1);
/* 得到配置文件名(全路径)，并初始化菜单结构 */
strncpy(menurc, rcpath, MAXPATH - 1);
strcat(menurc, "/menurc");
if(initmenu(menurc) == -1)
exit(-1);
run();
destroymenu();
exit(0);
}
void
getexepath(char *path)
{
char link[MAXPATH], exe[MAXPATH];
sprintf(link, "/proc/%d/exe", getpid());
readlink(link, exe, MAXPATH - 1);
strcpy(path, dirname(exe));
}

复制代码

libinary · 发表于 2003-7-30 13:29:57

menu.h menu.c文件：
有些不满意的地方，比如暴露了menusize变量，一开始是因为display模块要用到，后来想想，用一个next函数比较好，还有其他一些地方也不太满意。

#ifndef _MENU_H_
#define _MENU_H_
#include "menudef.h"
int initmenu(char *);
void destroymenu(void);
#endif /* _MENU_H_ */

复制代码

#include <stdio.h>
#include <string.h>
#include <dlfcn.h>
#include "menu.h"
struct MENU *menu; /* 全局菜单变量 */
int menusize; /* 菜单大小(菜单个数) */
char rcpath[MAXPATH]; /* 菜单配置文件路径 */
char libpath[MAXPATH]; /* 动态库路径 */
enum{root, sub}; /* 用于createmenu的参数 */
static struct MENU *createmenu(int);
static void readmenuitems();
/* 读配置文件初始化全局变量menu，分配内存
* 参数：menurc：配置文件名
* 返回值：0：正确；-1：错误
*/
int
initmenu(char *menurc)
{
FILE *fp;
char line[MAXLINE], *t;
struct MENU *mp;
if((fp = fopen(menurc, "r")) == NULL){
sprintf(line, "cannot open file %s", menurc);
perror(line);
return(-1);
}
while(fgets(line, MAXLINE - 1, fp)){
line[strlen(line) - 1] = '\0';
if((t = strtok(line, " \t")) == NULL)
continue;
if(!strcasecmp(t, "rootmenu")){ /* 创建根菜单 */
if((mp = createmenu(root)) == NULL){
destroymenu();
return(-1);
}
readmenuitems(mp, fp);
mp->count++; /* 增加exit菜单项 */
strcpy(mp->item[mp->count - 1].name, "exit");
mp->item[mp->count - 1].accel = 'x';
mp->item[mp->count - 1].options = SYSMENU | SYS_EXIT;
}else if(strcasecmp(t, "submenu") == 0){ /* 创建子菜单 */
if((mp = createmenu(sub)) == NULL){
destroymenu();
return(-1);
}
readmenuitems(mp, fp);
mp->count++; /* 增加return和root菜单项 */
strcpy(mp->item[mp->count - 1].name, "return");
mp->item[mp->count - 1].accel = 't';
mp->item[mp->count - 1].options = SYSMENU | SYS_RET;
mp->count++;
strcpy(mp->item[mp->count - 1].name, "root");
mp->item[mp->count - 1].accel = 'r';
mp->item[mp->count - 1].options = SYSMENU | SYS_ROOT;
}
}
}
void
destroymenu(void)
{
int i, j;
for(i = 0; i < menusize; i++)
for(j = 0; j < menu[i].count; j++)
if(menu[i].item[j].options == SUBMENU || menu[i].item[j].options == COMMAND)
free(menu[i].item[j].proc);
free(menu);
}
/* 创建菜单
* 参数：type：菜单类型(root，sub)
* 返回值：新分配菜单的指针；NULL：错误
*/
static struct MENU *
createmenu(int type)
{
char *t;
if((menu = (struct MENU *)realloc(menu, (menusize + 1) * sizeof(struct MENU))) == NULL){
fprintf(stderr, "realloc menu error\n");
return(NULL);
}
if(type == root)
strcpy(menu[menusize].name, "root");
else if(type == sub){
t = strtok(NULL, " \t");
strcpy(menu[menusize].name, t);
}
menu[menusize].count = 0;
menusize++;
return(&menu[menusize - 1]);
}
/* 读配置文件初始化菜单项
* 参数：m：包含菜单项的菜单；fp：配置文件指针
*/
static void
readmenuitems(struct MENU *mp, FILE *fp)
{
char line[MAXLINE], *name, *type, *menuname, *libname, *funcname, *command;
int i = -1;
char accel[] = "1234567890abcdefghijklmno"; /* 为简单起见使用指定快捷键的办法 */
while(fgets(line, MAXLINE - 1, fp)){
line[strlen(line) - 1] = '\0';
if((name = strtok(line, " \t")) == NULL)
continue;
if(!strcasecmp(name, "endmenu"))
break;
++i;
strncpy(mp->item[i].name, name, MAXWIDTH);
mp->count++;
mp->item[i].accel = accel[i];
if((type = strtok(NULL, " \t")) == NULL){
mp->item[i].options = ERR_ITEM;
continue;
}
if(!strcasecmp(type, "submenu")){
mp->item[i].options = SUBMENU;
if((menuname = strtok(NULL, " \t")) == NULL){
mp->item[i].options = ERR_ITEM;
continue;
}
mp->item[i].proc = (void *)malloc(strlen(menuname) + 1);
strcpy(mp->item[i].proc, menuname);
}else if(!strcasecmp(type, "function")){
void *libc, (*func)();
char *libfullname;
mp->item[i].options = FUNCTION;
if((libname = strtok(NULL, " \t")) == NULL){
mp->item[i].options = ERR_ITEM;
continue;
}
libfullname = (char *)malloc(strlen(libpath) + strlen(libname) + 2);
sprintf(libfullname, "%s/%s", libpath, libname);
if((libc = (void *)dlopen(libfullname, RTLD_LAZY)) == NULL)
mp->item[i].options = ERR_ITEM;
else if((funcname = strtok(NULL, " \t")) == NULL)
mp->item[i].options = ERR_ITEM;
else if((func = (void (*)())dlsym(libc, funcname)) == NULL)
mp->item[i].options = ERR_ITEM;
else
mp->item[i].proc = func;
free(libfullname);
}else if(!strcasecmp(type, "command")){
mp->item[i].options = COMMAND;
command = type + strlen(type) + 1; /* 这里缺少错误检查 */
mp->item[i].proc = (void *)malloc(strlen(command) + 1);
strcpy(mp->item[i].proc, command);
}
}
}

复制代码

libinary · 发表于 2003-7-30 13:31:32

display.h display.c文件：

#ifndef _DISPLAY_H_
#define _DISPLAY_H_
#include "menudef.h"
void displaymenu(void);
void run(void);
#endif /* _DISPLAY_H_ */

复制代码

#include <stdio.h>
#include <string.h>
#include "display.h"
struct MENUSTACK menustack;
extern struct MENU *menu;
extern int menusize;
static void pushmenu(char *);
static void popmenu(void);
static void rootmenu(void);
void
displaymenu(void)
{
struct MENU *t;
int i;
t = menustack.stack[menustack.current];
for(i = 0; i < t->count; i++)
printf("%c) %s\n", t->item[i].accel, t->item[i].name);
}
void
run(void)
{
char c, s[80];
int i;
struct MENU *t;
menustack.current = -1; /* 因为pushmenu会先menustack.current++，为了把menustack.stack[0] */
pushmenu("root"); /* 初始化为root，把menustack.current赋值为-1 */
displaymenu();
printf("-----------------\n");
printf("Enter your choice: ");
while(fgets(s, 79, stdin) != NULL){
c = s[0];
t = menustack.stack[menustack.current];
for(i = 0; i < t->count; i++)
if(t->item[i].accel == c)
break;
if(i == t->count || t->item[i].options == ERR_ITEM){
printf("\n");
displaymenu();
printf("-----------------\n");
printf("Enter your choice(again): ");
continue;
}
switch(t->item[i].options){
case SYSMENU | SYS_EXIT:
return;
break;
case SYSMENU | SYS_ROOT:
rootmenu();
break;
case SYSMENU | SYS_RET:
popmenu();
break;
case SUBMENU:
pushmenu((char *)t->item[i].proc);
break;
case FUNCTION:
printf("**********\n");
(*(void (*)())t->item[i].proc)();
printf("**********\n");
break;
case COMMAND:
printf("**********\n");
system((char *)t->item[i].proc);
printf("**********\n");
break;
}
printf("\n");
displaymenu();
printf("-----------------\n");
printf("Enter your choice: ");
}
}
void
pushmenu(char *name)
{
struct MENU *t;
for(t = menu; t < menu + menusize; t++)
if(!strcasecmp(name, t->name)){
menustack.current++;
menustack.stack[menustack.current] = t;
break;
}
}
void
popmenu(void)
{
if(menustack.current > 0)
menustack.current--;
}
static void
rootmenu(void)
{
menustack.current = 0;
}

复制代码

libinary · 发表于 2003-7-30 13:32:41

my.c文件：

#include <stdio.h>
#include <dirent.h>
void
printhw(void)
{
printf("hello, world!\n");
return;
}
void
printhelp(void)
{
printf("help!\n");
printf("Help!!\n");
printf("HELP!!!\n");
return;
}
void
myls(void)
{
DIR *dp;
struct dirent *dirp;
char path[256];
printf("Please enter a directory(default .): ");
fgets(path, 255, stdin);
path[strlen(path) - 1] = '\0';
if(path[0] == '\0')
path[0] = '.';
if((dp = opendir(path)) == NULL){
fprintf(stderr, "cannot open current directory.\n");
return;
}
while(dirp = readdir(dp))
printf("%s\n", dirp->d_name);
closedir(dp);
}

复制代码

libinary · 发表于 2003-7-30 13:33:34

menurc文件：

rootmenu
printhello function libmy.so printhw
printhelp function libmy.so printhelp
cmd submenu cmd
func submenu func
endmenu
submenu cmd
ls command ls
ll command ls -l
tree command tree
sh command bash
endmenu
submenu func
myls function libmy.so myls
endmenu

复制代码

libinary · 发表于 2003-7-30 13:36:27

编译：
gcc main.c menu.c display.c -ldl -o mn
gcc -shared my.c -o libmy.so
把menurc和mn(可执行文件)放在同一目录下。

libinary · 发表于 2003-7-30 13:50:36

关于程序还是有一些其他的考虑。
menu模块的接口比较简单，这是因为把菜单结构全部暴露了出来，考虑一下感觉应该用函数实现，比如getName、action之类的，把数据表示封装起来。
display模块里包含了显示和运行的功能，到底应不应该分成两个模块？
面向对象里的“文档-视图-控制”模型比较有参考价值，这个程序应该可以直接套用一下。

实际这个程序主要是要完善menu模块，只要menu模块写好了display模块应该可以很容易的改一种实现，比如用ncurses实现菜单的显示和程序的输入，现在的实现用的是一个菜单堆栈，如果要用ncuses把菜单显示在屏幕中间的话还可以，如果要实现屏幕顶部的菜单条就不行了，必须换一种实现。

kj501 · 发表于 2003-7-30 13:53:53

控制台下用curses写菜单界面的程序不少。libinary兄的这个有什么新特点？
我还没有仔细看呢。

		自动登录	找回密码
密码			注册