123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428 |
- #include "process.h"
- #include "../exception/gate.h"
- #include "../common/printk.h"
- #include "../common/kprint.h"
- #include "../syscall/syscall.h"
- #include "../syscall/syscall_num.h"
- #include <mm/slab.h>
- #include <sched/sched.h>
- extern void system_call(void);
- ul _stack_start;
- struct mm_struct initial_mm = {0};
- struct thread_struct initial_thread =
- {
- .rbp = (ul)(initial_proc_union.stack + STACK_SIZE / sizeof(ul)),
- .rsp = (ul)(initial_proc_union.stack + STACK_SIZE / sizeof(ul)),
- .fs = KERNEL_DS,
- .gs = KERNEL_DS,
- .cr2 = 0,
- .trap_num = 0,
- .err_code = 0};
- union proc_union initial_proc_union __attribute__((__section__(".data.init_proc_union"))) = {INITIAL_PROC(initial_proc_union.pcb)};
- struct process_control_block *initial_proc[MAX_CPU_NUM] = {&initial_proc_union.pcb, 0};
- struct tss_struct initial_tss[MAX_CPU_NUM] = {[0 ... MAX_CPU_NUM - 1] = INITIAL_TSS};
- void __switch_to(struct process_control_block *prev, struct process_control_block *next)
- {
- initial_tss[proc_current_cpu_id].rsp0 = next->thread->rbp;
- kdebug("next_rsp = %#018lx ", next->thread->rsp);
-
-
- __asm__ __volatile__("movq %%fs, %0 \n\t"
- : "=a"(prev->thread->fs));
- __asm__ __volatile__("movq %%gs, %0 \n\t"
- : "=a"(prev->thread->gs));
- __asm__ __volatile__("movq %0, %%fs \n\t" ::"a"(next->thread->fs));
- __asm__ __volatile__("movq %0, %%gs \n\t" ::"a"(next->thread->gs));
-
- }
- void user_level_function()
- {
-
-
-
-
-
- long ret = 0;
-
- char string[] = "User level process.\n";
-
- long err_code = 1;
- ul addr = (ul)string;
- __asm__ __volatile__(
- "movq %2, %%r8 \n\t"
- "int $0x80 \n\t"
- : "=a"(err_code)
- : "a"(SYS_PUT_STRING), "m"(addr)
- : "memory", "r8");
- if (err_code == 0)
- {
- char str[] = "errno is 0";
- addr = (ul)str;
- __asm__ __volatile__(
- "movq %2, %%r8 \n\t"
- "int $0x80 \n\t"
- : "=a"(err_code)
- : "a"(SYS_PUT_STRING), "m"(addr)
- : "memory", "r8");
- }
-
-
- while (1)
- pause();
- }
- ul do_execve(struct pt_regs *regs)
- {
-
- regs->rip = 0x800000;
- regs->rsp = 0xa00000;
- regs->cs = USER_CS | 3;
- regs->ds = USER_DS | 3;
- regs->ss = USER_DS | 0x3;
- regs->rflags = 0x200246;
- regs->rax = 1;
- regs->es = 0;
-
-
-
- uint64_t addr = 0x800000UL;
-
- mm_map_phys_addr_user(addr, alloc_pages(ZONE_NORMAL, 1, PAGE_PGT_MAPPED)->addr_phys, PAGE_2M_SIZE, PAGE_USER_PAGE);
- if (!(current_pcb->flags & PF_KTHREAD))
- current_pcb->addr_limit = KERNEL_BASE_LINEAR_ADDR;
-
- memcpy((void *)0x800000, user_level_function, 1024);
-
- return 0;
- }
- ul initial_kernel_thread(ul arg)
- {
-
- kdebug("6666");
- struct pt_regs *regs;
- current_pcb->thread->rip = (ul)ret_from_system_call;
- current_pcb->thread->rsp = (ul)current_pcb + STACK_SIZE - sizeof(struct pt_regs);
-
-
- regs = (struct pt_regs *)current_pcb->thread->rsp;
-
- current_pcb->flags = 0;
-
- __asm__ __volatile__("movq %1, %%rsp \n\t"
- "pushq %2 \n\t"
- "jmp do_execve \n\t" ::"D"(current_pcb->thread->rsp),
- "m"(current_pcb->thread->rsp), "m"(current_pcb->thread->rip)
- : "memory");
- return 1;
- }
- ul process_thread_do_exit(ul code)
- {
- kinfo("thread_exiting..., code is %#018lx.", code);
- while (1)
- ;
- }
- extern void kernel_thread_func(void);
- int kernel_thread(unsigned long (*fn)(unsigned long), unsigned long arg, unsigned long flags)
- {
- struct pt_regs regs;
- memset(®s, 0, sizeof(regs));
-
- regs.rbx = (ul)fn;
-
- regs.rdx = (ul)arg;
- regs.ds = KERNEL_DS;
- regs.es = KERNEL_DS;
- regs.cs = KERNEL_CS;
- regs.ss = KERNEL_DS;
-
- regs.rflags = (1 << 9);
-
- regs.rip = (ul)kernel_thread_func;
- kdebug("kernel_thread_func=%#018lx", kernel_thread_func);
- kdebug("&kernel_thread_func=%#018lx", &kernel_thread_func);
- kdebug("1111\tregs.rip = %#018lx", regs.rip);
- return do_fork(®s, flags, 0, 0);
- }
- void process_init()
- {
- kinfo("Initializing process...");
- initial_mm.pgd = (pml4t_t *)global_CR3;
- initial_mm.code_addr_start = memory_management_struct.kernel_code_start;
- initial_mm.code_addr_end = memory_management_struct.kernel_code_end;
- initial_mm.data_addr_start = (ul)&_data;
- initial_mm.data_addr_end = memory_management_struct.kernel_data_end;
- initial_mm.rodata_addr_start = (ul)&_rodata;
- initial_mm.rodata_addr_end = (ul)&_erodata;
- initial_mm.brk_start = 0;
- initial_mm.brk_end = memory_management_struct.kernel_end;
- initial_mm.stack_start = _stack_start;
-
-
-
- initial_tss[proc_current_cpu_id].rsp0 = initial_thread.rbp;
-
-
- list_init(&initial_proc_union.pcb.list);
- kernel_thread(initial_kernel_thread, 10, CLONE_FS | CLONE_FILES | CLONE_SIGNAL);
- initial_proc_union.pcb.state = PROC_RUNNING;
- initial_proc_union.pcb.preempt_count = 0;
-
-
-
-
-
- }
- unsigned long do_fork(struct pt_regs *regs, unsigned long clone_flags, unsigned long stack_start, unsigned long stack_size)
- {
- struct process_control_block *tsk = NULL;
- kdebug("222\tregs.rip = %#018lx", regs->rip);
-
- struct Page *pp = alloc_pages(ZONE_NORMAL, 1, PAGE_PGT_MAPPED | PAGE_KERNEL);
- tsk = (struct process_control_block *)phys_2_virt(pp->addr_phys);
- memset(tsk, 0, sizeof(struct process_control_block));
-
- *tsk = *current_pcb;
-
-
- list_init(&tsk->list);
-
- tsk->priority = 2;
- tsk->preempt_count = 0;
- ++(tsk->pid);
- tsk->cpu_id = proc_current_cpu_id;
- tsk->state = PROC_UNINTERRUPTIBLE;
- list_init(&tsk->list);
- list_add(&initial_proc_union.pcb.list, &tsk->list);
-
- struct thread_struct *thd = (struct thread_struct *)(tsk + 1);
- memset(thd, 0, sizeof(struct thread_struct));
- tsk->thread = thd;
- kdebug("333\tregs.rip = %#018lx", regs->rip);
-
- memcpy((void *)((ul)tsk + STACK_SIZE - sizeof(struct pt_regs)), regs, sizeof(struct pt_regs));
- kdebug("regs.rip = %#018lx", regs->rip);
-
- thd->rbp = (ul)tsk + STACK_SIZE;
- thd->rip = regs->rip;
- thd->rsp = (ul)tsk + STACK_SIZE - sizeof(struct pt_regs);
- thd->fs = KERNEL_DS;
- thd->gs = KERNEL_DS;
-
-
- if (!(tsk->flags & PF_KTHREAD))
- thd->rip = regs->rip = (ul)ret_from_system_call;
- else
- kdebug("is kernel proc.");
- tsk->state = PROC_RUNNING;
- sched_cfs_enqueue(tsk);
- return 0;
- }
|