feat: No more kernel page table switching upon interrupt.
[lunaix-os.git] / lunaix-os / kernel / lxinit.c
1 #include <hal/cpu.h>
2 #include <lunaix/syslog.h>
3 #include <lunaix/mm/kalloc.h>
4 #include <lunaix/mm/vmm.h>
5 #include <lunaix/spike.h>
6 #include <lunaix/clock.h>
7 #include <lunaix/timer.h>
8 #include <lunaix/keyboard.h>
9 #include <lunaix/tty/tty.h>
10 #include <lunaix/lunistd.h>
11 #include <lunaix/proc.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 "Forked %d\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("I am parent, my child (%d) terminated with code: %d.\n", child, status);
45 #endif
46     
47     // 这里是就是LunaixOS的第一个进程了!
48     for (size_t i = 0; i < 10; i++)
49     {
50         pid_t pid = 0;
51         if (!(pid = fork())) {
52             sleep(i);
53             if (i == 3) {
54                 i = *(int*)0xdeadc0de;    // seg fault!
55             }
56             tty_put_char('0'+i);
57             tty_put_char('\n');
58             _exit(0);
59         }
60         kprintf(KINFO "Forked %d\n", pid);
61     }
62
63     char buf[64];
64
65     kprintf(KINFO "Hello processes!\n");
66
67     cpu_get_brand(buf);
68     kprintf("CPU: %s\n\n", buf);
69
70     // no lxmalloc here! This can only be used within kernel, but here, we are in a dedicated process!
71     // any access to kernel method must be done via syscall
72
73     struct kdb_keyinfo_pkt keyevent;
74     while (1)
75     {
76         if (!kbd_recv_key(&keyevent)) {
77             // yield();
78             continue;
79         }
80         if ((keyevent.state & KBD_KEY_FPRESSED) && (keyevent.keycode & 0xff00) <= KEYPAD) {
81             tty_put_char((char)(keyevent.keycode & 0x00ff));
82             tty_sync_cursor();
83         }
84     }
85     
86
87     spin();
88 }