自制shell

tigerli

浏览: 101106 次
性别:
来自: 北京

最近访客更多访客>>

lmz123456

码农gg

geeksun

NTdiary

博主相关

博客

微博

相册

留言

关于我

文章分类

社区版块

存档分类

博客分类：

C/C++

C C++C#Access F#

自制shell，MINI型 http://blog.csdn.net/manio/category/218491.aspx

像普通C程序一样编译运行即可。
mini.h

/*filename:mini.h*/
#define WHITESPACE    " ., &"
#define MAX_ARG_LEN        10
#define MAX_DIRS           20
#define MAX_PATH_LEN       100
#define STD_INPUT          0
#define STD_OUTPUT         1
#define MAX_ARGS           5
struct command_t {
int argc;
char *name;
char *argv[MAX_ARGS];
};

mini.c

/************************************************************
*MiniShell
*filename:mini.c
*it's a simple shell written by Manio(maniosterATgmail.com)
*Bugs:1.the commands must be seprated by only one whitespace
*and more and more bugs here...
*the part of pipe havn't be written.
*date,addr:08/16/2006 Yangshuo,China
************************************************************/
#include <fcntl.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>
#include <unistd.h>
#include "mini.h"

//#define DEBUG 1

void printLineHead();
void parseCommand(char *,char **);
char *GetChkFullPath(char **,char **);

main(){
int i,j;/*for common use*/
int stat;/*used by wait()*/
int isInRedir,isOutRedir,isPipe;
int fid;/*for file opening*/
char orginalCmd[100];/*store the command user typed*/
char *dir[MAX_DIRS];

struct command_t command;
char *tempstr[7];

/*initial argv
   *maybe a error will be occured if argv is not be initialed
*/
for( i = 0 ; i < MAX_ARGS ; i++ )
    command.argv[i] = (char *)malloc(MAX_ARG_LEN);

      while( 1 ){
               printLineHead();
               gets(orginalCmd);  /* read user's input */
        #ifdef DEBUG
                      printf("::::after gets(orginalCmd):::: ");
                      printf("orginalCmd=%s ",orginalCmd);
                      printf(":::::::::::::::::::::::::::::: ");
        #endif
        /*if "exit" inputed,exit the shell*/
        if(strcmp(orginalCmd,"exit") == 0)
            break;
        /*devide command line to several parts*/
              parseCommand(orginalCmd,command.argv);
              /*get the amount of argv*/
              for(i = 0;command.argv[i] != NULL;i++);
              command.argc = i;
#ifdef DEBUG
              printf("command.argc=%d ",command.argc);
#endif
#ifdef DEBUG
    printf("::::parseCommand command.argv:::: ");
    printf("command.name = %s ",command.name);
    int commandi;
    for(commandi=0;command.argv[commandi]!=NULL;commandi++)
        printf("command.argv[%d]=%s ",commandi,command.argv[commandi]);
    printf("::::END command.argv:::: ");
#endif

            /*get the dirs from PATH;PATH is a enviroment var*/
#ifdef DEBUG
            printf("getDir start=========================== ");
#endif
            getDir(dir);

#ifdef DEBUG
    int diri=0;
    for(;dir[diri]!=NULL;diri++)
    {
        printf("in main dir[%d] is %s ",diri,dir[diri]);
    }
#endif
              /*get the name of command,this must get the full path of the file,but here is for debug*/
            command.name = GetChkFullPath(command.argv,dir);
#ifdef DEBUG
            printf("in main after GetChkFullPath command.name=%s ",command.name);
#endif
            if(command.name == NULL ){
                /*the command inputed is not valid or is empty*/
                continue;
                fprintf(stderr, "Command %s not found ", command.argv[0]);
            }
            /*scan  < > | ,isPipe isInRedir and isOutRedir store the position of "< > |" */
            isPipe = isInRedir = isOutRedir = 0;
            for( i= 0 ; command.argv[i] != NULL ; i++ )
            {
                if( strcmp( command.argv[i] , "<" ) == 0 ){
                    isInRedir = i;
                }
                if( strcmp( command.argv[i] , ">" ) == 0 ){
                    isOutRedir = i;
                }
                if( strcmp( command.argv[i] ,  "|" ) == 0 ){
                    isPipe = i;
                }
            }
#ifdef DEBUG
            printf("isin,inout,ispipe:%d%d%d ",isInRedir,isOutRedir,isPipe);
#endif
#ifdef DEBUG
            command.argv[3] = NULL;
            printf("show the argvs which is not set ");
            for(i = 0 ; i < MAX_ARGS ; i++)
                printf("argv[%d]=%s ",i,command.argv[i]);
#endif
                /*excute a commmand*/
        if(fork() == 0){
#ifdef DEBUG
            printf("::::::::::::::::In fork==0:::::::::::::::::: ");
#endif
            if(isInRedir){
                fid = open(command.argv[isInRedir + 1],O_RDONLY);
                close(STD_INPUT);//close standard input
                dup(fid);//redirect fid to standard input
                close(fid);
                command.argv[isInRedir] = NULL; /*ignore the word after isInRedir*/
#ifdef DEBUG
    printf("::::parseCommand command.argv:::: ");
    printf("command.name = %s ",command.name);
    for(commandi=0;command.argv[commandi]!=NULL;commandi++)
        printf("command.argv[%d]=%s ",commandi,command.argv[commandi]);
    printf("::::END command.argv:::: ");
#endif
            }
            if(isOutRedir){
                fid = open(command.argv[isOutRedir + 1],O_WRONLY|O_CREAT);
                close(STD_OUTPUT);
                dup(fid);
                close(fid);
                command.argv[isOutRedir] = NULL;
#ifdef DEBUG
    printf("::::parseCommand command.argv:::: ");
    printf("command.name = %s ",command.name);
    for(commandi=0;command.argv[commandi]!=NULL;commandi++)
        printf("command.argv[%d]=%s ",commandi,command.argv[commandi]);
    printf("::::END command.argv:::: ");
#endif
            }
            execv(command.name,command.argv);/*excute the command*/
#ifdef DEBUG
            printf("::::::::::::::::In fork==0 END:::::::::::::::::: ");
#endif
        }else{
#ifdef DEBUG
            printf("::::::::::::::::In fork!=0:::::::::::::::::: ");
#endif
            wait(&stat);/*wait for the end of child process*/
#ifdef DEBUG
            printf("::::::::child process done!::IN FORK!=0:END::::: ");
#endif
        }
//        loopcount = 0;
      }/*while*/
#ifdef DEBUG
      printf("Shell terminaled ");
#endif
}

void printLineHead()
{
printf(" MANIO's miniSHELL>");
}
/**************************************************************
   parseCommand()
   *seprate cLine by WHITESPACE,put them to pchar[]
   *#define WHITESPACE    " ., &"
   *bugs:This code does not handle multiple WHITESPACE characters
***************************************************************
*/
void parseCommand(char *cLine, char *pchar[]) {
    int argc;
    char *tCLine;
    char **clPtr;
// Initialization
    tCLine = cLine;
    clPtr = &tCLine;
    argc = 0;
// This code does not handle multiple WHITESPACE characters
    while((pchar[argc++] = strsep(clPtr, WHITESPACE)) != NULL);

    pchar[argc--] = NULL;    // Null terminated list of strings
}

/****************************************************************
   *get the path in PATH seprated by ":"
*****************************************************************
*/
int  getDir(char *dir[])
{
    int i;
    char *s;
    char *oldpath;
    char *ppath;
    s=(char *)getenv("PATH");
#ifdef DEBUG
    printf("s is %s ",s);
    printf("s2=====%c ",s[2]);
#endif
    oldpath = (char *)malloc(strlen(s)+1);
    strcpy(oldpath,s);
#ifdef DEBUG
    printf("oldpath is %s ",oldpath);
#endif
    ppath = oldpath;
#ifdef DEBUG
    printf("ppath char  is %c ",*ppath);
#endif

    for(i=0;*ppath!='';ppath++)
    {
#ifdef DEBUG
    printf("in for ppath char  is %c ",*ppath);
#endif

        if(*ppath==':')
        {
            dir[i++] = oldpath;
            *ppath='';
            oldpath = ppath+1;
        }
    }
    dir[i] = NULL;
#ifdef DEBUG
    int diri=0;
    for(;dir[diri]!=NULL;diri++)
    {
        printf("dir[%d] is %s ",diri,dir[diri]);
    }
#endif
    return 1;

}

/****************************************************************
   *get and check the full path which the user inputed
    *dir :dirs in PATH
    *argv:argv in command_t
*****************************************************************
*/
char *GetChkFullPath(char **argv,char **dir)
{
#ifdef DEBUG
    printf(":::::::::::GetChkFullPath start ");
    printf("argv[0]=%s dir[0]=%s ",argv[0],dir[0]);
#endif
    char fullpath[MAX_PATH_LEN];
    char *result;
    int i;

    result = NULL;
    if( *argv[0] != '/' )
    {
        // *argv[0] != ''
#ifdef DEBUG
        printf("JUGE argv[0]!=/ ");
#endif
        for(i = 0;dir[i] != NULL ;i++)
        {
            strcpy(fullpath,dir[i]);
            strcat(fullpath,"/");
            strcat(fullpath,argv[0]);
#ifdef DEBUG
            printf("after strcat fullpath:%s ",fullpath);
#endif
            if(access(fullpath, X_OK | F_OK) != -1){
#ifdef DEBUG
                printf("FOUND %s in %s ",argv[0],dir[i]);
#endif
                result = (char *)malloc(strlen(fullpath)+1);
                strcpy(result,fullpath);
                break;
            }
        }
    }else{
#ifdef DEBUG
    printf("JUGE argv[0]==/ ");
#endif
        if(access(argv[0], X_OK | F_OK) != -1){
#ifdef DEBUG
            printf("FOUND %s ",argv[0]);
#endif
            result = (char *)malloc(strlen(argv[0])+1);
            strcpy(result , argv[0]);
        }

    }

    if(result == NULL ){
        printf("%s is not a command. ",argv[0]);
        return NULL;
    }else{
#ifdef DEBUG
        printf("GetChkFullPath end result == %s ",result);
#endif
        return result;
    }
}

分享到：