Linux小程序: 手写自己的shell

注意: 本文章只是为了理解shell内部的工作原理, 所以并没有完成shell的所有工作, 只是完成了shell里的一小部分工作

c 复制代码
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <assert.h>
#include <sys/types.h> 
#include <sys/wait.h>

#define LEFT "["
#define RIGHT "]"
#define END "#"
#define SPLIT " "
#define LINE_SIZE 1024
#define ARGV_SIZE 32
#define EXIT_CODE 66
#define PWD_SIZE 100

int lastexit = 0;
int quit = 0;
char pwd[PWD_SIZE];

const char* getusername() {
  return getenv("USER");
}

const char* mygethostname() {
  return getenv("HOSTNAME");
}

void getpwd() {
  getcwd(pwd, PWD_SIZE);
}

void display() {
  getpwd();
  printf(LEFT"%s@%s %s"RIGHT""END" ", getusername(), mygethostname(), pwd);
}

void getinput(char* cline, int size) {
  char* s = fgets(cline, size, stdin);
  assert(s != NULL);
  cline[strlen(cline) - 1] = '\0';
  //printf("%s\n", cline);
}

int split(char* cline, char* argv[]) {
  int i = 0;
  //printf("%s", cline);
  argv[i ++] = strtok(cline, SPLIT);
  while (argv[i ++] = strtok(NULL, SPLIT));
  return i - 1;
}

void normalExe(char* argv[]) {
    extern char** environ;
    pid_t id = fork();
    if (id < 0) {
      perror("fork");
      //continue;
    } else if (id == 0) {
      // child
      execvpe(argv[0], argv, environ);
      exit(EXIT_CODE);
    }
    // father
    int status = 0;
    pid_t ret = waitpid(id, &status, 0);
  //  if (ret > 0) printf("wait success pid: %d\n", ret);
    if (ret > 0) lastexit = WEXITSTATUS(status);
}

int neijian(char* argv[], int argc) {
  if (argc == 2 && strcmp(argv[0], "cd") == 0) {
    chdir(argv[1]);
    getpwd();
    return 1;
  }
  return 0;
}

int main() {
  //int quit = 0;
  char cline[LINE_SIZE];
  char* argv[ARGV_SIZE] = { NULL };
  extern char** environ;
  while (!quit) {
    display();
    getinput(cline, LINE_SIZE);
    int tt = split(cline, argv);
    if (tt == 0) continue;

    //printf("%d\n", tt);
    /*
    int i = 0;
    for (i = 0; argv[i]; ++ i)
      printf("%s ", argv[i]);
    printf("\n");
    */

    // 内建命令执行
    //内建命令本质上就是shell内部的一个函数
    
    int n = neijian(argv, tt);
    //printf("%d\n", n); 

    //普通命令执行
    if (!n) normalExe(argv);
  }
  return 0;
}
相关推荐
舒克起飞了几秒前
linux系统编程——Makefile、GDB调试
linux·运维·服务器
背影疾风9 分钟前
C++之路:类基础、构造析构、拷贝构造函数
linux·开发语言·c++
deming_su17 分钟前
轻松上手:使用Nginx实现高效负载均衡
运维·nginx·负载均衡
mg66840 分钟前
微信小程序入门实例_____打造你的专属单词速记小程序
微信小程序·小程序
程序员陆通1 小时前
Vibe Coding开发微信小程序实战案例
微信小程序·小程序·notepad++·ai编程
「、皓子~1 小时前
后台管理系统的诞生 - 利用AI 1天完成整个后台管理系统的微服务后端+前端
前端·人工智能·微服务·小程序·go·ai编程·ai写作
m0_694845571 小时前
服务器如何配置防火墙规则开放/关闭端口?
linux·服务器·安全·云计算
降世神童2 小时前
华为云Flexus+DeepSeek征文| 使用华为云CCE容器部署Dify-LLM高可用方案的验证与测试
运维·华为云·aigc
降世神童2 小时前
华为云Flexus+DeepSeek征文| 基于华为云Dify-LLM高可用平台开发运维故障处理智能体
运维·华为云·aigc
nbsaas-boot2 小时前
[特殊字符] 分享裂变新姿势:用 UniApp + Vue3 玩转小程序页面分享跳转!
小程序·uniapp·notepad++