关闭

linux c模拟ls命令

2551人阅读 评论(1) 收藏 举报
/*
 模拟ls命令,实现参数-tariRl.有些代码重复出现,可改进
 可以在该程序的基础上增加参数,该程序思路清晰,容易扩展
 */

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<sys/types.h>
#include<unistd.h>
#include<sys/stat.h>
#include<dirent.h>
#include<sys/ioctl.h>
#include<pwd.h>
#include<grp.h>


#define LL (1<<0)  /*带l参数标志*/
#define IL (1<<1)  /*带i参数标志*/
#define RU (1<<2)  /*带R参数标志*/
#define TL (1<<16) /*带t参数标志*/
#define AL (1<<17) /*带a参数标志*/
#define RL (1<<18) /*带r参数标志*/

#define LENGTH 500 /*字符串最大长度*/
#define MAX_FILE_COUNT 500/*文件列表最大长度*/

/*函数声明*/
void error(char* error_message,int line);
void display(struct stat* stalist,char (*namelist)[LENGTH],int n,int mode);
int file_list(char* path,struct stat* stalist,char (*namelist)[LENGTH],int mode);
void display_simply(char (*namelist)[LENGTH],int n);
void display_LL(struct stat * stalist,char (*namelist)[LENGTH],int n);
void display_IL(struct stat * stalist,char (*namelist)[LENGTH],int n);
void display_LL_IL(struct stat * stalist,char (*namelist)[LENGTH],int n);
void display_RU(struct stat* stalist,char (*namelist)[LENGTH],int n,int mode);
void display_RU_LL(struct stat * stalist,char (*namelist)[LENGTH],int n,int mode);
void display_RU_IL(struct stat* stalist,char (*namelist)[LENGTH],int n,int mode);
void display_RU_IL_LL(struct stat * stalist,char (*namelist)[LENGTH],int n,int mode);

/*主函数*/
int main(int argc,char **argv)
{
    int i,j;
    /*mode的前16个位用来标志那些能影响显示的参数,16位之后的位用来标志不影响输出格式的参数。这个程序只用到了0-3位,16-18位,如果要在该程序的基础上增加参数,可以用其他没有使用的位。
    mode的第0位为1表示带l参数,mode的第1位为1表示带i参数,mode的第2位为1表示带R参数,mode的第16位为1表示带t参数,mode的第17位为1表示带a参数,mode的第18位为1表示带r参数*/
    int mode=0;
    char error_message[LENGTH]={0};/*错误信息*/
    char path[LENGTH]={0};/*路径*/
    struct stat stalist[MAX_FILE_COUNT];/*文件详细信息数组*/
    char  namelist[MAX_FILE_COUNT][LENGTH];/*文件名数组*/
    int flag=0;/*判断是否指定了目录*/
    int count;/*文件个数*/

    /*解析参数*/
    for(i=1;i<argc;i++){
        if(argv[i][0]=='-'){/*如果是选项参数*/
	    for(j=1;j<strlen(argv[i]);j++){
		if(argv[i][j]=='l')
		    mode=mode|LL;
		else if(argv[i][j]=='i')
		    mode=mode|IL;
		else if(argv[i][j]=='R')
		    mode=mode|RU;
		else if(argv[i][j]=='r')
		    mode=mode|RL;
		else if(argv[i][j]=='a')
		    mode=mode|AL;
		else if(argv[i][j]=='t')
		    mode=mode|TL;
		else{
		    snprintf(error_message,LENGTH,"no option of %c",argv[i][j]);
		    error(error_message,__LINE__);
		}					
	    }
	}else{/*参数为目录或文件*/
	    if(flag==1)
		error("can not specify more then two dir or file",__LINE__);
	    else
		flag=1;
            if(argv[i][0]!='/'){/*相对路径*/
		strcat(path,get_current_dir_name());
		strcat(path,"/");
		strncat(path,argv[i],strlen(argv[i]));
	    }else/*绝对路径*/
		strcpy(path,argv[i]);
	}
    }

    if(flag==0)/*未指定任何目录或文件,则使用默认当前目录*/
	strcat(path,get_current_dir_name());

    /*根据mode获取指定目录下的文件*/
    count=file_list(path,stalist,namelist,mode);
    
    /*根据mode显示文件*/
    display(stalist,namelist,count,mode);

    return 0;
}


/*获取文件列表*/
int file_list(char* path,struct stat* stalist,char (*namelist)[LENGTH],int mode)
{
    int i=0,index,j,k;
    DIR *dir;
    struct dirent *ptr;
    char str[LENGTH];
    char nametemp[LENGTH];
    struct stat statemp;

    if(stat(path,&stalist[0])<0){
//	puts(path);
	error("the specify file is not exist",__LINE__);
    }
    if(S_ISDIR(stalist[0].st_mode)){/*如果该路径指向的是一个目录*/
        if((dir=opendir(path))==NULL){/*打开目录*/
	    sprintf(str,"can not open %s",path);
	    error(str,__LINE__);
	}
        chdir(path);/*改变当前工作目录到path*/
        i=0;
	while((ptr=readdir(dir))!=NULL){/*依次读取目录下的所有文件*/
	    if(ptr->d_name[0]=='.' && !(mode & AL))/*忽略隐藏文件*/
		continue;
	    /*按字母顺序插入到stalist数组及namelist数组*/
	    for(j=i;j>0;j--){
		if(strcmp(ptr->d_name,namelist[j-1])>0)
		    break;
	    }
	    for(k=i;k>j;k--){
		strcpy(namelist[k],namelist[k-1]);
		stalist[k]=stalist[k-1];
	    }

            strcpy(namelist[j],ptr->d_name);
            stat(ptr->d_name,&stalist[j]);
	    i++;
	    if(i==MAX_FILE_COUNT)
		error("file count beyond MAX_FILE_COUNT",__LINE__);
	}
	closedir(dir);
    }else{
	j=strlen(path)-1;
	while(j>=0 && path[j]!='/'){
	    j--;
	}
	if(j<0)
	    error("path error",__LINE__);
	j++;
	k=0;
        while(path[j]){
	    namelist[0][k]=path[j];
            j++;
 	    k++;
	}
        namelist[0][k]=0;
        return 1;
    }

    if(mode & TL){/*按文件修改时间排序*/
	for(j=0;j<i;j++){
	    index=j;
	    for(k=j+1;k<i;k++)
		if(stalist[index].st_mtime<stalist[k].st_mtime){
		    index=k;
		}
	    statemp=stalist[j];
	    stalist[j]=stalist[index];
	    stalist[index]=statemp;
	    strcpy(nametemp,namelist[j]);
	    strcpy(namelist[j],namelist[index]);
	    strcpy(namelist[index],nametemp);
	}
    }

    if(mode & RL){/*反序排列*/
	for(j=0;j<i/2;j++){
	    strcpy(nametemp,namelist[j]);
	    strcpy(namelist[j],namelist[i-j-1]);
	    strcpy(namelist[i-j-1],nametemp);
	    statemp=stalist[j];
	    stalist[j]=stalist[i-j-1];
	    stalist[i-j-1]=statemp;
	}
    }
    return i;
}

/*根据mode选择不同的函数进行显示*/
void display(struct stat* stalist,char (*namelist)[LENGTH],int n,int mode)
{
    /*使mode的高16位为0*/
    int m = mode & ((1<<16)-1);
    switch(m){
	case 0:
	    display_simply(namelist,n);
	    break;
	case 1:
	    display_LL(stalist,namelist,n);
	    break;
	case 2:
	    display_IL(stalist,namelist,n);
	    break;
	case 3:
	    display_LL_IL(stalist,namelist,n);
	    break;
	case 4:
	    display_RU(stalist,namelist,n,mode);
	    break;
	case 5:
	    display_RU_LL(stalist,namelist,n,mode);
	    break;
        case 6:
	    display_RU_IL(stalist,namelist,n,mode);
	    break;
	case 7:
	    display_RU_IL_LL(stalist,namelist,n,mode);
	    break;
    }
}

/*不带显示选项,简单显示文件名*/
void display_simply(char (*namelist)[LENGTH],int n)
{
    int i,maxlength=0,cols;
    struct winsize ts;
    for(i=0;i<n;i++)
	if(strlen(namelist[i])>maxlength)
	    maxlength=strlen(namelist[i]);
    /*获取终端的大小*/
    ioctl(STDIN_FILENO,TIOCGWINSZ,&ts);
    cols=ts.ws_col;
    cols/=(maxlength+1);
    //lines=ts.ws_row;
    for(i=0;i<n;i++){
       if(i!=0 && i%cols==0)
	   puts("");
       printf("%*s",-(maxlength+1),namelist[i]);
    }
    putchar('\n');
}

/*-l 参数,显示详细文件信息*/
void display_LL(struct stat * stalist,char (*namelist)[LENGTH],int n)
{
    int i,mode;
    char* str;
    for(i=0;i<n;i++){
        mode=stalist[i].st_mode;
        if(S_ISDIR(mode))
	    printf("d");
	else
	    printf("-");
	if(mode & (1<<8))
	    printf("r");
	else
	    printf("-");
        if(mode & (1<<7))
	    printf("w");
	else
	    printf("-");
        if(mode & (1<<6))
	    printf("x");
	else
	    printf("-");
        if(mode & (1<<5))
	    printf("r");
	else
	    printf("-");
        if(mode & (1<<4))
	    printf("w");
	else
	    printf("-");
        if(mode & (1<<3))
	    printf("x");
	else
	    printf("-");
        if(mode & (1<<2))
	    printf("r");
	else
	    printf("-");
        if(mode & (1<<1))
	    printf("w");
	else
	    printf("-");
        if(mode & (1<<0))
	    printf("x");
	else
	    printf("-");
	printf(" %-3d",stalist[i].st_nlink);
	printf(" %-6s",getpwuid(stalist[i].st_uid)->pw_name);
	printf(" %-6s",getgrgid(stalist[i].st_gid)->gr_name);
	printf(" %-10d",stalist[i].st_size);
	str=ctime(&stalist[i].st_mtime);
	str[strlen(str)-2]=0;
	printf(" %s",str);
	printf(" %s\n",namelist[i]);
    }
}

/*-i 参数,显示文件名及节点号*/
void display_IL(struct stat* stalist,char (*namelist)[LENGTH],int n)
{
    int i,maxlength=0,cols;
    struct winsize ts;
    for(i=0;i<n;i++)
	if(strlen(namelist[i])>maxlength)
	    maxlength=strlen(namelist[i]);
    /*获取终端的大小*/
    ioctl(STDIN_FILENO,TIOCGWINSZ,&ts);
    cols=ts.ws_col;
    cols/=(maxlength+9);
    //lines=ts.ws_row;
    
    for(i=0;i<n;i++){
       if(i!=0 && i%cols==0)
	   puts("");
       printf("%-8d ",stalist[i].st_ino);
       printf("%*s",-(maxlength+1),namelist[i]);
    }
    putchar('\n');   
}

/*-li 参数,显示详细文件信息和节点号*/
void display_LL_IL(struct stat * stalist,char (*namelist)[LENGTH],int n)
{
    int i,mode;
    char* str;

    for(i=0;i<n;i++){
        printf("%-8d ",stalist[i].st_ino);
        mode=stalist[i].st_mode;
        if(S_ISDIR(mode))
	    printf("d");
	else
	    printf("-");
	if(mode & (1<<8))
	    printf("r");
	else
	    printf("-");
        if(mode & (1<<7))
	    printf("w");
	else
	    printf("-");
        if(mode & (1<<6))
	    printf("x");
	else
	    printf("-");
        if(mode & (1<<5))
	    printf("r");
	else
	    printf("-");
        if(mode & (1<<4))
	    printf("w");
	else
	    printf("-");
        if(mode & (1<<3))
	    printf("x");
	else
	    printf("-");
        if(mode & (1<<2))
	    printf("r");
	else
	    printf("-");
        if(mode & (1<<1))
	    printf("w");
	else
	    printf("-");
        if(mode & (1<<0))
	    printf("x");
	else
	    printf("-");
	printf(" %-3d",stalist[i].st_nlink);
	printf(" %-6s",getpwuid(stalist[i].st_uid)->pw_name);
	printf(" %-6s",getgrgid(stalist[i].st_gid)->gr_name);
	printf(" %-10d",stalist[i].st_size);
	str=ctime(&stalist[i].st_mtime);
	str[strlen(str)-2]=0;
	printf(" %s",str);
	printf(" %s\n",namelist[i]);
	
    }

}

/*-R 参数,简单显示所有文件,包括目录下面的子目录*/
void display_RU(struct stat* stalist,char (*namelist)[LENGTH],int n,int mode)
{
    int i,count;
    char path[LENGTH]={0},temp[LENGTH]={0};
    struct stat sta[MAX_FILE_COUNT];
    char name[MAX_FILE_COUNT][LENGTH];
    puts(get_current_dir_name());
    display_simply(namelist,n);
    putchar('\n');
    strcpy(path,get_current_dir_name());
    
    for(i=0;i<n;i++){
        if(strcmp(namelist[i],".")==0 || strcmp(namelist[i],"..")==0)
     	    continue;
	if(S_ISDIR(stalist[i].st_mode)){
	    strcpy(temp,path);
	    strcat(path,"/");
	    strcat(path,namelist[i]);
	    count=file_list(path,sta,name,mode);
	    display_RU(sta,name,count,mode);
            strcpy(path,temp);
	}
    }
}

/*-Rl 参数,显示所有文件,包括目录下面的子目录的详细信息*/
void display_RU_LL(struct stat * stalist,char (*namelist)[LENGTH],int n,int mode)
{
    int i,count;
    char path[LENGTH]={0},temp[LENGTH]={0};
    struct stat sta[MAX_FILE_COUNT];
    char name[MAX_FILE_COUNT][LENGTH];

    puts(get_current_dir_name());
    display_LL(stalist,namelist,n);
    putchar('\n');
    strcpy(path,get_current_dir_name());
    
    for(i=0;i<n;i++){
	if(strcmp(namelist[i],".")==0 || strcmp(namelist[i],"..")==0)
     	    continue;

	if(S_ISDIR(stalist[i].st_mode)){
	    strcpy(temp,path);
	    strcat(path,"/");
	    strcat(path,namelist[i]);
	    count=file_list(path,sta,name,mode);
	    display_RU_LL(sta,name,count,mode);
	    strcpy(path,temp);
	}
    }
}

/*-Ri 参数,简单显示所有文件,包括目录下的子目录,及节点号*/
void display_RU_IL(struct stat* stalist,char (*namelist)[LENGTH],int n,int mode)
{
    int i,count;
    char path[LENGTH]={0},temp[LENGTH]={0};
    struct stat sta[MAX_FILE_COUNT];
    char name[MAX_FILE_COUNT][LENGTH];
    puts(get_current_dir_name());
    display_IL(stalist,namelist,n);
    putchar('\n');
    strcpy(path,get_current_dir_name());
    
    for(i=0;i<n;i++){
	if(strcmp(namelist[i],".")==0 || strcmp(namelist[i],"..")==0)
     	    continue;

	if(S_ISDIR(stalist[i].st_mode)){
	    strcpy(temp,path);
	    strcat(path,"/");
	    strcat(path,namelist[i]);
	    count=file_list(path,sta,name,mode);
	    display_RU_IL(sta,name,count,mode);
	    strcpy(path,temp);
	}
    }

}

/*-Ril 参数,显示所有文件,包括目录下面的子目录的详细信息,及节点号*/
void display_RU_IL_LL(struct stat * stalist,char (*namelist)[LENGTH],int n,int mode)
{
    int i,count;
    char path[LENGTH]={0},temp[LENGTH]={0};
    struct stat sta[MAX_FILE_COUNT];
    char name[MAX_FILE_COUNT][LENGTH];
    puts(get_current_dir_name());
    display_LL_IL(stalist,namelist,n);
    putchar('\n');
    strcpy(path,get_current_dir_name());
    
    for(i=0;i<n;i++){
	if(strcmp(namelist[i],".")==0 || strcmp(namelist[i],"..")==0)
     	    continue;

	if(S_ISDIR(stalist[i].st_mode)){
	    strcpy(temp,path);
	    strcat(path,"/");
	    strcat(path,namelist[i]);
	    count=file_list(path,sta,name,mode);
	    display_RU_IL_LL(sta,name,count,mode);
	    strcpy(path,temp);
	}
    }
}

//错误处理函数
void error(char* error_message,int line)
{
    char str[5];
    sprintf(str,"%d:",line);
    strcat(str,error_message);
    perror(str);
    exit(EXIT_FAILURE);
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:509682次
    • 积分:4658
    • 等级:
    • 排名:第6693名
    • 原创:46篇
    • 转载:2篇
    • 译文:0篇
    • 评论:191条
    最新评论