mirror of
https://github.com/DragonOS-Community/DragonOS.git
synced 2025-06-24 21:33:27 +00:00
新增rust ffi (#77)
* 引入cargo * 取消对Cargo.lock的跟踪 * 解决vscode报错问题 * new: rust的代码能够调用c语言的printk_color * 1、将原本run.sh的工作拆解，变为几个不同的make命令 2、在docker镜像中编译rust * 更改workflow * update workflow * new: 解决workflow无法通过编译的问题
This commit is contained in:
login
GitHub
13
kernel/src/smp/Makefile
Normal file
13
kernel/src/smp/Makefile
Normal file
@ -0,0 +1,13 @@
|
||||
|
||||
CFLAGS += -I .
|
||||
|
||||
|
||||
all: apu_boot.o smp.o
|
||||
|
||||
|
||||
apu_boot.o: apu_boot.S
|
||||
$(CC) -E apu_boot.S > _apu_boot.s # 预处理
|
||||
as $(ASFLAGS) -o apu_boot.o _apu_boot.s
|
||||
|
||||
smp.o: smp.c
|
||||
$(CC) $(CFLAGS) -c smp.c -o smp.o
|
||||
121
kernel/src/smp/apu_boot.S
Normal file
121
kernel/src/smp/apu_boot.S
Normal file
@ -0,0 +1,121 @@
|
||||
#include "../common/asm.h"
|
||||
|
||||
|
||||
.align 0x1000 // 按照4k对齐
|
||||
|
||||
.section .text
|
||||
.code16
|
||||
|
||||
ENTRY(_apu_boot_start)
|
||||
_apu_boot_base = .
|
||||
cli
|
||||
wbinvd // 将处理器缓存同步到内存中
|
||||
|
||||
mov %cs, %ax
|
||||
mov %ax, %ds
|
||||
mov %ax, %es
|
||||
mov %ax, %ss
|
||||
mov %ax, %fs
|
||||
mov %ax, %gs
|
||||
|
||||
// 设置栈指针
|
||||
movl $(_apu_boot_tmp_stack_end - _apu_boot_base), %esp
|
||||
|
||||
// 计算ap处理器引导程序的基地址
|
||||
mov %cs, %ax
|
||||
movzx %ax, %esi
|
||||
shll $4, %esi
|
||||
|
||||
|
||||
// set gdt and 32bit/64bit code addr
|
||||
|
||||
leal (_apu_code32 - _apu_boot_base)(%esi), %eax
|
||||
movl %eax, (_apu_code32_vector - _apu_boot_base)
|
||||
|
||||
leal (_apu_code64 - _apu_boot_base)(%esi), %eax
|
||||
movl %eax, (_apu_code64_vector - _apu_boot_base)
|
||||
|
||||
leal (_apu_tmp_gdt - _apu_boot_base)(%esi), %eax
|
||||
movl %eax, (_apu_tmp_gdt + 2 - _apu_boot_base)
|
||||
|
||||
// 从实模式切换到保护模式
|
||||
|
||||
lidtl _apu_tmp_idt - _apu_boot_base
|
||||
lgdtl _apu_tmp_gdt - _apu_boot_base
|
||||
|
||||
// 操作cr0控制器，使能保护模式
|
||||
smsw %ax
|
||||
bts $0, %ax
|
||||
lmsw %ax
|
||||
|
||||
// 转到保护模式
|
||||
ljmpl *(_apu_code32_vector - _apu_boot_base)
|
||||
|
||||
|
||||
.code32
|
||||
.align 0x1000
|
||||
_apu_code32:
|
||||
# 转到长模式
|
||||
|
||||
mov $0x10, %ax
|
||||
mov %ax, %ds
|
||||
mov %ax, %es
|
||||
mov %ax, %ss
|
||||
mov %ax, %fs
|
||||
mov %ax, %gs
|
||||
|
||||
// 设置栈指针
|
||||
leal (_apu_boot_tmp_stack_end - _apu_boot_base)(%esi), %eax
|
||||
movl %eax, %esp
|
||||
|
||||
// 1. 允许 PAE
|
||||
mov %cr4, %eax
|
||||
or $(1<<5), %eax
|
||||
mov %eax, %cr4
|
||||
|
||||
movl $enter_head_from_ap_boot, %eax
|
||||
jmpl *%eax
|
||||
hlt
|
||||
|
||||
|
||||
|
||||
.code64
|
||||
.align 0x1000
|
||||
_apu_code64:
|
||||
|
||||
hlt
|
||||
|
||||
|
||||
.align 0x1000
|
||||
_apu_tmp_idt:
|
||||
.word 0
|
||||
.word 0,0
|
||||
|
||||
.align 0x1000
|
||||
_apu_tmp_gdt:
|
||||
.short _apu_tmp_gdt_end - _apu_tmp_gdt -1
|
||||
.long _apu_tmp_gdt - _apu_boot_base
|
||||
.short 0
|
||||
.quad 0x00cf9a000000ffff
|
||||
.quad 0x00cf92000000ffff
|
||||
.quad 0x0020980000000000
|
||||
.quad 0x0000920000000000
|
||||
_apu_tmp_gdt_end:
|
||||
|
||||
.align 0x1000
|
||||
_apu_code32_vector:
|
||||
.long _apu_code32 - _apu_boot_base
|
||||
.word 0x08,0
|
||||
|
||||
.align 0x1000
|
||||
_apu_code64_vector:
|
||||
.long _apu_code64 - _apu_boot_base
|
||||
.word 0x18,0
|
||||
|
||||
.align 0x1000
|
||||
_apu_boot_tmp_stack_start:
|
||||
// .org 0x400
|
||||
_apu_boot_tmp_stack_end:
|
||||
|
||||
ENTRY(_apu_boot_end)
|
||||
|
||||
39
kernel/src/smp/ipi.h
Normal file
39
kernel/src/smp/ipi.h
Normal file
@ -0,0 +1,39 @@
|
||||
#pragma once
|
||||
|
||||
#include <arch/arch.h>
|
||||
#if ARCH(I386) || ARCH(X86_64)
|
||||
#include <arch/x86_64/x86_64_ipi.h>
|
||||
#else
|
||||
#error "error type of arch!"
|
||||
#endif
|
||||
|
||||
/**
|
||||
* @brief 发送ipi消息
|
||||
*
|
||||
* @param dest_mode 目标模式
|
||||
* @param deliver_status 投递模式
|
||||
* @param level 信号驱动电平
|
||||
* @param trigger 触发模式
|
||||
* @param vector 中断向量
|
||||
* @param deliver_mode 投递模式
|
||||
* @param dest_shorthand 投递目标速记值
|
||||
* @param apic_type apic的类型 （0:xapic 1: x2apic）
|
||||
* @param destination 投递目标
|
||||
*/
|
||||
extern void ipi_send_IPI(uint32_t dest_mode, uint32_t deliver_status, uint32_t level, uint32_t trigger,
|
||||
uint32_t vector, uint32_t deliver_mode, uint32_t dest_shorthand, uint32_t destination);
|
||||
|
||||
/**
|
||||
* @brief ipi中断处理注册函数
|
||||
*
|
||||
* @param irq_num 中断向量号
|
||||
* @param arg 参数
|
||||
* @param handler 处理函数
|
||||
* @param param 参数
|
||||
* @param controller 当前为NULL
|
||||
* @param irq_name ipi中断名
|
||||
* @return int 成功：0
|
||||
*/
|
||||
extern int ipi_regiserIPI(uint64_t irq_num, void *arg,
|
||||
void (*handler)(uint64_t irq_num, uint64_t param, struct pt_regs *regs),
|
||||
uint64_t param, hardware_intr_controller *controller, char *irq_name);
|
||||
190
kernel/src/smp/smp.c
Normal file
190
kernel/src/smp/smp.c
Normal file
@ -0,0 +1,190 @@
|
||||
#include "smp.h"
|
||||
#include <common/kprint.h>
|
||||
#include <driver/interrupt/apic/apic.h>
|
||||
#include <exception/gate.h>
|
||||
#include <common/cpu.h>
|
||||
#include <mm/slab.h>
|
||||
#include <process/process.h>
|
||||
#include <common/spinlock.h>
|
||||
|
||||
#include <sched/sched.h>
|
||||
|
||||
#include "ipi.h"
|
||||
|
||||
void ipi_0xc8_handler(uint64_t irq_num, uint64_t param, struct pt_regs *regs); // 由BSP转发的HPET中断处理函数
|
||||
|
||||
static spinlock_t multi_core_starting_lock = {1}; // 多核启动锁
|
||||
|
||||
static struct acpi_Processor_Local_APIC_Structure_t *proc_local_apic_structs[MAX_SUPPORTED_PROCESSOR_NUM];
|
||||
static uint32_t total_processor_num = 0;
|
||||
int current_starting_cpu = 0;
|
||||
|
||||
int num_cpu_started = 1;
|
||||
|
||||
void smp_init()
|
||||
{
|
||||
spin_init(&multi_core_starting_lock); // 初始化多核启动锁
|
||||
ul tmp_vaddr[MAX_SUPPORTED_PROCESSOR_NUM] = {0};
|
||||
|
||||
apic_get_ics(ACPI_ICS_TYPE_PROCESSOR_LOCAL_APIC, tmp_vaddr, &total_processor_num);
|
||||
|
||||
// kdebug("processor num=%d", total_processor_num);
|
||||
for (int i = 0; i < total_processor_num; ++i)
|
||||
{
|
||||
io_mfence();
|
||||
proc_local_apic_structs[i] = (struct acpi_Processor_Local_APIC_Structure_t *)(tmp_vaddr[i]);
|
||||
}
|
||||
|
||||
// 将引导程序复制到物理地址0x20000处
|
||||
memcpy((unsigned char *)phys_2_virt(0x20000), _apu_boot_start, (unsigned long)&_apu_boot_end - (unsigned long)&_apu_boot_start);
|
||||
io_mfence();
|
||||
// 设置多核IPI中断门
|
||||
for (int i = 200; i < 210; ++i)
|
||||
set_intr_gate(i, 0, SMP_interrupt_table[i - 200]);
|
||||
memset((void *)SMP_IPI_desc, 0, sizeof(irq_desc_t) * SMP_IRQ_NUM);
|
||||
|
||||
io_mfence();
|
||||
|
||||
// 注册接收bsp处理器的hpet中断转发的处理函数
|
||||
ipi_regiserIPI(0xc8, NULL, &ipi_0xc8_handler, NULL, NULL, "IPI 0xc8");
|
||||
io_mfence();
|
||||
ipi_send_IPI(DEST_PHYSICAL, IDLE, ICR_LEVEL_DE_ASSERT, EDGE_TRIGGER, 0x00, ICR_INIT, ICR_ALL_EXCLUDE_Self, 0x00);
|
||||
|
||||
kdebug("total_processor_num=%d", total_processor_num);
|
||||
kdebug("rflags=%#018lx", get_rflags());
|
||||
// total_processor_num = 3;
|
||||
for (int i = 0; i < total_processor_num; ++i) // i从1开始，不初始化bsp
|
||||
{
|
||||
io_mfence();
|
||||
|
||||
// 跳过BSP
|
||||
kdebug("[core %d] acpi processor UID=%d, APIC ID=%d, flags=%#010lx", i, proc_local_apic_structs[i]->ACPI_Processor_UID, proc_local_apic_structs[i]->local_apic_id, proc_local_apic_structs[i]->flags);
|
||||
if (proc_local_apic_structs[i]->local_apic_id == 0)
|
||||
{
|
||||
--total_processor_num;
|
||||
continue;
|
||||
}
|
||||
if (!((proc_local_apic_structs[i]->flags & 0x1) || (proc_local_apic_structs[i]->flags & 0x2)))
|
||||
{
|
||||
--total_processor_num;
|
||||
kdebug("processor %d cannot be enabled.", proc_local_apic_structs[i]->ACPI_Processor_UID);
|
||||
continue;
|
||||
}
|
||||
// continue;
|
||||
io_mfence();
|
||||
spin_lock(&multi_core_starting_lock);
|
||||
preempt_enable(); // 由于ap处理器的pcb与bsp的不同，因此ap处理器放锁时，bsp的自旋锁持有计数不会发生改变,需要手动恢复preempt count
|
||||
current_starting_cpu = proc_local_apic_structs[i]->ACPI_Processor_UID;
|
||||
io_mfence();
|
||||
// 为每个AP处理器分配栈空间
|
||||
cpu_core_info[current_starting_cpu].stack_start = (uint64_t)kmalloc(STACK_SIZE, 0) + STACK_SIZE;
|
||||
cpu_core_info[current_starting_cpu].ist_stack_start = (uint64_t)(kmalloc(STACK_SIZE, 0)) + STACK_SIZE;
|
||||
io_mfence();
|
||||
memset((void *)cpu_core_info[current_starting_cpu].stack_start - STACK_SIZE, 0, STACK_SIZE);
|
||||
memset((void *)cpu_core_info[current_starting_cpu].ist_stack_start - STACK_SIZE, 0, STACK_SIZE);
|
||||
io_mfence();
|
||||
|
||||
// 设置ap处理器的中断栈及内核栈中的cpu_id
|
||||
((struct process_control_block *)(cpu_core_info[current_starting_cpu].stack_start - STACK_SIZE))->cpu_id = proc_local_apic_structs[i]->local_apic_id;
|
||||
((struct process_control_block *)(cpu_core_info[current_starting_cpu].ist_stack_start - STACK_SIZE))->cpu_id = proc_local_apic_structs[i]->local_apic_id;
|
||||
|
||||
cpu_core_info[current_starting_cpu].tss_vaddr = (uint64_t)&initial_tss[current_starting_cpu];
|
||||
|
||||
memset(&initial_tss[current_starting_cpu], 0, sizeof(struct tss_struct));
|
||||
|
||||
set_tss_descriptor(10 + (current_starting_cpu * 2), (void *)(cpu_core_info[current_starting_cpu].tss_vaddr));
|
||||
io_mfence();
|
||||
set_tss64((uint *)cpu_core_info[current_starting_cpu].tss_vaddr, cpu_core_info[current_starting_cpu].stack_start, cpu_core_info[current_starting_cpu].stack_start, cpu_core_info[current_starting_cpu].stack_start,
|
||||
cpu_core_info[current_starting_cpu].ist_stack_start, cpu_core_info[current_starting_cpu].ist_stack_start, cpu_core_info[current_starting_cpu].ist_stack_start, cpu_core_info[current_starting_cpu].ist_stack_start, cpu_core_info[current_starting_cpu].ist_stack_start, cpu_core_info[current_starting_cpu].ist_stack_start, cpu_core_info[current_starting_cpu].ist_stack_start);
|
||||
io_mfence();
|
||||
|
||||
// 连续发送两次start-up IPI
|
||||
ipi_send_IPI(DEST_PHYSICAL, IDLE, ICR_LEVEL_DE_ASSERT, EDGE_TRIGGER, 0x20, ICR_Start_up, ICR_No_Shorthand, proc_local_apic_structs[i]->local_apic_id);
|
||||
io_mfence();
|
||||
ipi_send_IPI(DEST_PHYSICAL, IDLE, ICR_LEVEL_DE_ASSERT, EDGE_TRIGGER, 0x20, ICR_Start_up, ICR_No_Shorthand, proc_local_apic_structs[i]->local_apic_id);
|
||||
|
||||
}
|
||||
io_mfence();
|
||||
while (num_cpu_started != total_processor_num)
|
||||
pause();
|
||||
|
||||
kinfo("Cleaning page table remapping...\n");
|
||||
|
||||
// 由于ap处理器初始化过程需要用到0x00处的地址，因此初始化完毕后才取消内存地址的重映射
|
||||
uint64_t *global_CR3 = get_CR3();
|
||||
for (int i = 0; i < 256; ++i)
|
||||
{
|
||||
io_mfence();
|
||||
*(ul *)(phys_2_virt(global_CR3) + i) = 0UL;
|
||||
}
|
||||
kdebug("init proc's preempt_count=%ld", current_pcb->preempt_count);
|
||||
kinfo("Successfully cleaned page table remapping!\n");
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief AP处理器启动后执行的第一个函数
|
||||
*
|
||||
*/
|
||||
void smp_ap_start()
|
||||
{
|
||||
|
||||
// 切换栈基地址
|
||||
// uint64_t stack_start = (uint64_t)kmalloc(STACK_SIZE, 0) + STACK_SIZE;
|
||||
__asm__ __volatile__("movq %0, %%rbp \n\t" ::"m"(cpu_core_info[current_starting_cpu].stack_start)
|
||||
: "memory");
|
||||
__asm__ __volatile__("movq %0, %%rsp \n\t" ::"m"(cpu_core_info[current_starting_cpu].stack_start)
|
||||
: "memory");
|
||||
|
||||
ksuccess("AP core %d successfully started!", current_starting_cpu);
|
||||
io_mfence();
|
||||
++num_cpu_started;
|
||||
|
||||
apic_init_ap_core_local_apic();
|
||||
|
||||
// ============ 为ap处理器初始化IDLE进程 =============
|
||||
memset(current_pcb, 0, sizeof(struct process_control_block));
|
||||
|
||||
barrier();
|
||||
current_pcb->state = PROC_RUNNING;
|
||||
current_pcb->flags = PF_KTHREAD;
|
||||
current_pcb->mm = &initial_mm;
|
||||
|
||||
list_init(¤t_pcb->list);
|
||||
current_pcb->addr_limit = KERNEL_BASE_LINEAR_ADDR;
|
||||
current_pcb->priority = 2;
|
||||
current_pcb->virtual_runtime = 0;
|
||||
|
||||
current_pcb->thread = (struct thread_struct *)(current_pcb + 1); // 将线程结构体放置在pcb后方
|
||||
current_pcb->thread->rbp = _stack_start;
|
||||
current_pcb->thread->rsp = _stack_start;
|
||||
current_pcb->thread->fs = KERNEL_DS;
|
||||
current_pcb->thread->gs = KERNEL_DS;
|
||||
current_pcb->cpu_id = current_starting_cpu;
|
||||
|
||||
initial_proc[proc_current_cpu_id] = current_pcb;
|
||||
barrier();
|
||||
load_TR(10 + current_starting_cpu * 2);
|
||||
current_pcb->preempt_count = 0;
|
||||
|
||||
io_mfence();
|
||||
spin_unlock(&multi_core_starting_lock);
|
||||
preempt_disable(); // 由于ap处理器的pcb与bsp的不同，因此ap处理器放锁时，需要手动恢复preempt count
|
||||
io_mfence();
|
||||
sti();
|
||||
|
||||
while (1)
|
||||
hlt();
|
||||
|
||||
while (1)
|
||||
{
|
||||
printk_color(BLACK, WHITE, "CPU:%d IDLE process.\n", proc_current_cpu_id);
|
||||
}
|
||||
while (1) // 这里要循环hlt，原因是当收到中断后，核心会被唤醒，处理完中断之后不会自动hlt
|
||||
hlt();
|
||||
}
|
||||
|
||||
// 由BSP转发的HPET中断处理函数
|
||||
void ipi_0xc8_handler(uint64_t irq_num, uint64_t param, struct pt_regs *regs)
|
||||
{
|
||||
sched_update_jiffies();
|
||||
}
|
||||
18
kernel/src/smp/smp.h
Normal file
18
kernel/src/smp/smp.h
Normal file
@ -0,0 +1,18 @@
|
||||
#pragma once
|
||||
#include <common/glib.h>
|
||||
|
||||
#include <common/asm.h>
|
||||
#include <driver/acpi/acpi.h>
|
||||
#include <driver/interrupt/apic/apic.h>
|
||||
|
||||
#define MAX_SUPPORTED_PROCESSOR_NUM 1024
|
||||
|
||||
|
||||
|
||||
extern uchar _apu_boot_start[];
|
||||
extern uchar _apu_boot_end[];
|
||||
/**
|
||||
* @brief 初始化对称多核处理器
|
||||
*
|
||||
*/
|
||||
void smp_init();
|
||||
Reference in New Issue
Block a user