【Linux】进程替换以及实现简易shell

1、进程程序替换

1.1程序替换的原理

在使用fork()函数创建子进程后，子进程往往需要使用exec函数将程序进行替换，已执行另一个程序，当进程调用exec函数时，该进程的程序代码以及数据将完全被新程序替换。从新程序的启动开始执行，调用exec函数不会创建子进程，因为调用exec函数前后该进程的pid没有发生变化。

2、替换函数

有六种exec开头的函数。

函数的统一解释：

如果函数执行成功就从新程序的启动代码开始执行，没有返回值。

如果函数调用出错就会返回-1。

exce系列函数调用成功没有返回值，调用失败返回-1。

函数的命名：

l(list) : 表示参数采用列表

v(vector) : 参数用数组

p(path) : 有p自动搜索环境变量

PATH e(env) : 表示自己维护环境变量

3实现简易shell

#include<stdio.h>
#include<unistd.h>
#include<sys/types.h>
#include<sys/wait.h>
#include<string.h>
#include<stdlib.h>

#define NUM 1024
#define SIZE 64
#define SEP " "
#define Debug 1

char cwd[1024];
char enval[1024];
int lastcode = 0;

const char* getUsername()
{
  const char* name = getenv("USER");
  if(name)
  {
    return name;
  }
  else
  {
    return "none";
  }
}

const char *getHostname()
{
  const char* hostname = getenv("HOSTNAME");

  if(hostname)
  {
    return hostname;
  }
  else
  {
    return "none";
  }
}


const char *getCwd()
{
  const char *cwd = getenv("PWD");
  
  if(cwd)
  {
    return cwd;
  }
  else
  {
    return "none";
  }

}






int getusercommand(char* command,int n)
{
  printf("[%s@%s %s]#",getUsername(),getHostname(),getCwd());
  char *r = fgets(command,n,stdin);
  if(r == NULL)
  {
    return -1;
  }
  command[strlen(command) - 1] = '\0';

  return strlen(command);

}


void commandSplit(char *in,char **out)
{
  int argc = 0;
  out[argc++] = strtok(in,SEP);

  while(out[argc++] = strtok(NULL,SEP))
  {
    ;
  }

#ifdef Debug
  for(int i = 0; out[i]; i++)
  {
    printf("%d:%s\n",i,out[i]);
  }

#endif
}


int execute(char** argv)
{
  pid_t id = fork();
  if(id == 0)
  {
    execvp(argv[0],argv);
    exit(1);
  }
  else if(id < 0)
  {
    return -1;
  }
  else
  {
    int status = 0;
    pid_t rid = waitpid(id,&status,0);
    if(rid > 0)
    {
      lastcode = WEXITSTATUS(status);
    }
  }
  return 0;
}


void cd(const char * path)
{
    chdir(path);
    char  tmp[1024];
    getcwd(tmp,sizeof(tmp));
    sprintf(cwd,"PWD=%s",tmp);
    putenv(cwd);

}


int DoBuildin(char **argv)
{
  if(strcmp(argv[0],"cd") == 0)
  {
    char *path = NULL;
    if(argv[1] == NULL)
    {
      path = ".";
    }
    else
    {
      path  = argv[1];
    }
    cd(path);
    return 1;
  }
  else if(strcmp(argv[0],"export") == 0)
  {
    if(argv[1] == NULL)
    {
      return 1;
    }
    strcpy(enval,argv[1]);
    putenv(enval);
    return 1;
  }
  else if(strcmp(argv[0], "echo") == 0)
  {
    char *val = argv[1] + 1;
    if(strcmp(val,"?") == 0)
    {
      printf("%d\n",lastcode);
      lastcode = 0;
    }
    else
    {
      printf("%s\n",getenv(val));
    }
    return 1;
  }
  else if(0){}



  return 0;
}


int main()
{
  while(1)
  {
    char entercommand[NUM];
    char *argv[SIZE];

    int n = getusercommand(entercommand,sizeof(entercommand));

   if(n <= 0)
   {
     continue;
   }

    //分割字符串
    
    commandSplit(entercommand,argv);
    //处理内建命令
     n = DoBuildin(argv);
    execute(argv);

    if(n)
    {
      continue;
    }
  }
  return 0;
}