lunaix-os/kernel/lxinit.c

   1 #include <hal/cpu.h>
   2 #include <lunaix/clock.h>
   3 #include <lunaix/keyboard.h>
   4 #include <lunaix/lunistd.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 #include <lunaix/tty/tty.h>
  12
  13 extern uint8_t __kernel_start;
  14
  15 LOG_MODULE("INIT")
  16
  17 // #define FORK_BOMB_DEMO
  18 #define WAIT_DEMO
  19
  20 void
  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 #ifdef WAIT_DEMO
  34     // 测试wait
  35     kprintf("I am parent, going to fork my child and wait.\n");
  36     if (!fork()) {
  37         kprintf("I am child, going to sleep for 2 seconds\n");
  38         sleep(2);
  39         kprintf("I am child, I am about to terminated\n");
  40         _exit(1);
  41     }
  42     int status;
  43     pid_t child = wait(&status);
  44     kprintf(
  45       "I am parent, my child (%d) terminated with code: %d.\n", child, status);
  46 #endif
  47
  48     sleep(5);
  49
  50     // 这里是就是LunaixOS的第一个进程了！
  51     for (size_t i = 0; i < 10; i++) {
  52         pid_t pid = 0;
  53         if (!(pid = fork())) {
  54             sleep(i);
  55             if (i == 3) {
  56                 i = *(int*)0xdeadc0de; // seg fault!
  57             }
  58             tty_put_char('0' + i);
  59             tty_put_char('\n');
  60             _exit(0);
  61         }
  62         kprintf(KINFO "Forked %d\n", pid);
  63     }
  64
  65     char buf[64];
  66
  67     kprintf(KINFO "Hello processes!\n");
  68
  69     cpu_get_brand(buf);
  70     kprintf("CPU: %s\n\n", buf);
  71
  72     // no lxmalloc here! This can only be used within kernel, but here, we are
  73     // in a dedicated process! any access to kernel method must be done via
  74     // syscall
  75
  76     struct kdb_keyinfo_pkt keyevent;
  77     while (1) {
  78         if (!kbd_recv_key(&keyevent)) {
  79             // yield();
  80             continue;
  81         }
  82         if ((keyevent.state & KBD_KEY_FPRESSED) &&
  83             (keyevent.keycode & 0xff00) <= KEYPAD) {
  84             tty_put_char((char)(keyevent.keycode & 0x00ff));
  85             tty_sync_cursor();
  86         }
  87     }
  88
  89     spin();
  90 }