Update

Author: MRNIU

devlog/devlog20200306.md

@@ -44,4 +44,11 @@
 
 # SimpleKernel 开发日志 20200312
 
-思考了一下发现，为啥我不直接用 pmm_alloc 进行分配？有物理地址对齐需求的也就只有这里了，用 pmm 刚好啊
+思考了一下发现，为啥我不直接用 pmm_alloc 进行分配？有物理地址对齐需求的也就只有这里了，用 pmm 刚好啊。
+
+我真傻，真的。我要对齐的是线性地址，关物理内存什么事？我是个沙雕吧
+
+好的，那么新的解决方案是这样：
+
+- 添加一个专门分配栈空间的函数，负责申请物理内存+映射，保证返回的地址可以用 slab 管理
+- 为了避免线性空间冲突的问题，需要在每次映射页之前判断一下有没有冲突，旧有的 alloc_page 函数与新的函数都需要

iso/boot/kernel.bin

@@ -78,7 +78,7 @@ void unmap(pgd_t * pgd_now, ptr_t va) {
 	return;
 }
 
-uint32_t get_mapping(pgd_t * pgd_now, ptr_t va, ptr_t pa) {
+uint32_t get_mapping(pgd_t * pgd_now, ptr_t va, ptr_t * pa) {
 	uint32_t pgd_idx = VMM_PGD_INDEX(va);
 	uint32_t pte_idx = VMM_PTE_INDEX(va);
 
@@ -89,8 +89,10 @@ uint32_t get_mapping(pgd_t * pgd_now, ptr_t va, ptr_t pa) {
 	// 转换到内核线性地址
 	pte = (pte_t *)VMM_PA_LA( (ptr_t)pte);
 	// 如果地址有效而且指针不为 NULL
-	if( ( (void *)pte[pte_idx] != NULL) && ( (void *)pa != NULL) ) {
-		pa = pte[pte_idx] & VMM_PAGE_MASK;
+	if( (void *)pte[pte_idx] != NULL) {
+		if( (void *)pa != NULL) {
+			pa = pte[pte_idx] & VMM_PAGE_MASK;
+		}
 		return 1;
 	}
 	return 0;

simplekernel.iso

@@ -32,6 +32,7 @@ ListEntry * wait_list = NULL;
 static inline ptr_t alloc_stack(void);
 static inline ptr_t alloc_stack(void) {
 
+	return 0;
 }
 
 // 返回一个空的任务控制块

src/arch/x86_64/mm/vmm.c

@@ -98,7 +98,7 @@ void unmap(pgd_t * pgd_now, ptr_t va);
 
 // 如果虚拟地址 va 映射到物理地址则返回 1
 // 同时如果 pa 不是空指针则把物理地址写入 pa 参数
-uint32_t get_mapping(pgd_t * pgd_now, ptr_t va, ptr_t pa);
+uint32_t get_mapping(pgd_t * pgd_now, ptr_t va, ptr_t * pa);
 
 // 更换当前页目录
 void switch_pgd(ptr_t pd);

src/arch/x86_64/task/task.c

@@ -260,13 +260,39 @@ ptr_t alloc_page(ptr_t va, size_t page) {
 	for(ptr_t va_start = va, pa_start = pa ;
 	    pa_start < pa + VMM_PAGE_SIZE * page ;
 	    pa_start += VMM_PAGE_SIZE, va_start += VMM_PAGE_SIZE) {
-		map(pgd_kernel, va_start, pa_start, VMM_PAGE_PRESENT | VMM_PAGE_RW);
+		// 如果当前线性地址没有映射
+		if(get_mapping(pgd_kernel, va_start, (ptr_t *)NULL) == 0) {
+			map(pgd_kernel, va_start, pa_start, VMM_PAGE_PRESENT | VMM_PAGE_RW);
+		}
+		else {
+			// 如果有部分映射了，则全部 unmap
+			ptr_t addr = (ptr_t)NULL;
+			for(addr = va ; addr < va_start ; addr += VMM_PAGE_SIZE) {
+				unmap(pgd_kernel, addr);
+			}
+			// 并重新计算 va 的值
+			// 运行到这里时 addr 尚未被检测，所以下面的代码从 addr 开始
+			// 计算方法：addr~addr+page*VMM_PAGE_SIZE 的 get_mapping 结果全部为 0
+			ptr_t tmp = addr;
+			while(addr < tmp + page * VMM_PAGE_SIZE) {
+				// 如果遇到映射过的
+				if(get_mapping(pgd_kernel, addr, (ptr_t *)NULL) != 0) {
+					// 更新 addr 地址
+					addr += VMM_PAGE_SIZE;
+					tmp = addr;
+				}
+				addr += VMM_PAGE_SIZE;
+			}
+			// 这时 addr 就是符合要求的地址
+			// 全部映射即可
+			map(pgd_kernel, addr, pa_start, VMM_PAGE_PRESENT | VMM_PAGE_RW);
+			return addr;
+		}
 	}
 	bzero( (void *)va, VMM_PAGE_SIZE * page);
 	return va;
 }
 
-// align 参数为按照 align 地址对齐
 ptr_t alloc(size_t byte) {
 	// 所有申请的内存长度(限制最小大小)加上管理头的长度
 	size_t len = (byte > SLAB_MIN) ? byte : SLAB_MIN;