lunaix-os/kernel/lxinit.c

   1 #include <hal/cpu.h>
   2 #include <lunaix/clock.h>
   3 #include <lunaix/lunistd.h>
   4 #include <lunaix/lxconsole.h>
   5 #include <lunaix/mm/kalloc.h>
   6 #include <lunaix/mm/vmm.h>
   7 #include <lunaix/proc.h>
   8 #include <lunaix/spike.h>
   9 #include <lunaix/syslog.h>
  10 #include <lunaix/timer.h>
  11
  12 extern uint8_t __kernel_start;
  13
  14 LOG_MODULE("INIT")
  15
  16 // #define FORK_BOMB_DEMO
  17 #define WAIT_DEMO
  18 #define IN_USER_MODE
  19
  20 void __USER__
  21 _lxinit_main()
  22 {
  23 #ifdef FORK_BOMB_DEMO
  24     // fork炸弹
  25     for (;;) {
  26         pid_t p;
  27         if ((p = fork())) {
  28             kprintf(KDEBUG "Pinkie Pie #%d: FUN!\n", p);
  29         }
  30     }
  31 #endif
  32
  33     int status;
  34 #ifdef WAIT_DEMO
  35     // 测试wait
  36     kprintf("I am parent, going to fork my child and wait.\n");
  37     if (!fork()) {
  38         kprintf("I am child, going to sleep for 2 seconds\n");
  39         sleep(2);
  40         kprintf("I am child, I am about to terminated\n");
  41         _exit(1);
  42     }
  43     wait(&status);
  44     pid_t child = wait(&status);
  45     kprintf("I am parent, my child (%d) terminated normally with code: %d.\n",
  46             child,
  47             WEXITSTATUS(status));
  48 #endif
  49
  50     pid_t p = 0;
  51
  52     if (!fork()) {
  53         kprintf("Test no hang!\n");
  54         sleep(6);
  55         _exit(0);
  56     }
  57
  58     waitpid(-1, &status, WNOHANG);
  59
  60     for (size_t i = 0; i < 5; i++) {
  61         pid_t pid = 0;
  62         if (!(pid = fork())) {
  63             sleep(i);
  64             if (i == 3) {
  65                 i = *(int*)0xdeadc0de; // seg fault!
  66             }
  67             kprintf(KINFO "%d\n", i);
  68             _exit(0);
  69         }
  70         kprintf(KINFO "Forked %d\n", pid);
  71     }
  72
  73     while ((p = wait(&status)) >= 0) {
  74         short code = WEXITSTATUS(status);
  75         if (WIFEXITED(status)) {
  76             kprintf(KINFO "Process %d exited with code %d\n", p, code);
  77         } else {
  78             kprintf(KWARN "Process %d aborted with code %d\n", p, code);
  79         }
  80     }
  81
  82     char buf[64];
  83
  84     kprintf(KINFO "Hello processes!\n");
  85
  86     cpu_get_brand(buf);
  87     kprintf("CPU: %s\n\n", buf);
  88
  89     _exit(0);
  90 }