Skip to content

wa
wa

wa-lang / wa
大约 1 年 前同步成功

通知 68
Star 655
Fork 45
wa
wa

体验新版 GitCode，发现更多精彩内容 >>

提交 dd2f0fe3 编写于 作者: chai2010's avatar chai2010
浏览文件
操作

wat 后端汇编代码拆分到相关 wa 包中

上级 1b663af6
隐藏空白更改
内联 并排
Showing with 380 addition and 267 deletion
internal/backends/compiler_wat/Makefile 已删除 100644 → 0
浏览文件 @ 1b663af6
# 版权 @2022 凹语言 作者。保留所有权利。
# wasmer
# https://github.com/wasmerio/wasmer/releases
default:
wasmer run base.wat
clean:
internal/backends/compiler_wat/base.wat 已删除 100644 → 0
浏览文件 @ 1b663af6
;; 版权 @2022 凹语言 作者。保留所有权利。
(module $walang
;; ----------------------------------------------------
;; import 必须最先定义
;; ----------------------------------------------------
;; WASI 最小子集
;; 用于输出字符串
(import "wasi_snapshot_preview1" "fd_write"
(func $fd_write (param i32 i32 i32 i32) (result i32))
)
;; ----------------------------------------------------
;; 内存和入口
;; ----------------------------------------------------
(memory $memory 1)
(export "memory" (memory $memory))
(export "_start" (func $_start))
(export "main.main" (func $main.main))
;; ----------------------------------------------------
;; WASM 约定栈和内存管理
;; 相关全局变量地址必须和 base_wasm.go 保持一致
;; ----------------------------------------------------
;; heap 和 stack 状态(__heap_base 只读)
;; | 0 <-- stack --> | <-- static-data --> | <-- heap --> |
(global $__stack_ptr (mut i32) (i32.const 1024)) ;; index=0
(global $__heap_base i32 (i32.const 2048)) ;; index=1
;; ----------------------------------------------------
;; Stack 辅助函数
;; ----------------------------------------------------
;; 获取栈顶地址
(func $waStackPtr (result i32)
(global.get $__stack_ptr)
)
;; 栈上分配空间
(func $waStackAlloc (param $size i32) (result i32)
;; $__stack_ptr -= $size
(global.set $__stack_ptr (i32.sub (global.get $__stack_ptr) (local.get $size)))
;; return $__stack_ptr
(return (global.get $__stack_ptr))
)
;; 释放栈上的空间
(func $waStackFree (param $size i32)
;; $__stack_ptr += $size
(global.set $__stack_ptr (i32.add (global.get $__stack_ptr) (local.get $size)))
)
;; ----------------------------------------------------
;; Heap 辅助函数
;; ----------------------------------------------------
;; 获取堆地址
(func $waHeapPtr (result i32)
(global.get $__heap_base)
)
;; 堆上分配内存(没有记录大小)
(func $waAlloc (param $size i32) (result i32)
;; {{$waAlloc/body/begin}}
unreachable
;; {{$waAlloc/body/end}}
)
;; 内存复用(引用加一)
(func $waRetain(param $ptr i32) (result i32)
;; {{$waRetain/body/begin}}
unreachable
;; {{$waRetain/body/end}}
)
;; 释放内存(引用减一)
(func $waFree (param $ptr i32)
;; {{$waFree/body/begin}}
unreachable
;; {{$waFree/body/end}}
)
;; ----------------------------------------------------
;; 输出函数
;; ----------------------------------------------------
;; 打印字符串
(func $puts (param $str i32) (param $len i32)
;; {{$puts/body/begin}}
(local $sp i32)
(local $p_iov i32)
(local $p_nwritten i32)
(local $stdout i32)
;; 保存栈指针状态
(local.set $sp (global.get $__stack_ptr))
;; 分配 iov 结构体
(local.set $p_iov (call $waStackAlloc (i32.const 8)))
;; 返回地址
(local.set $p_nwritten (call $waStackAlloc (i32.const 4)))
;; 设置字符串指针和长度
(i32.store offset=0 align=1 (local.get $p_iov) (local.get $str))
(i32.store offset=4 align=1 (local.get $p_iov) (local.get $len))
;; 标准输出
(local.set $stdout (i32.const 1))
;; 输出字符串
(call $fd_write
(local.get $stdout)
(local.get $p_iov) (i32.const 1)
(local.get $p_nwritten)
)
;; 重置栈指针
(global.set $__stack_ptr (local.get $sp))
drop
;; {{$puts/body/end}}
)
;; 打印字符
(func $putchar (param $ch i32)
;; {{$putchar/body/begin}}
(local $sp i32)
(local $p_ch i32)
;; 保存栈指针状态
(local.set $sp (global.get $__stack_ptr))
;; 分配字符
(local.set $p_ch (call $waStackAlloc (i32.const 4)))
(i32.store offset=0 align=1 (local.get $p_ch) (local.get $ch))
;; 输出字符
(call $puts (local.get $p_ch) (i32.const 1))
;; 重置栈指针
(global.set $__stack_ptr (local.get $sp))
;; {{$putchar/body/begin}}
)
;; 打印整数
(func $print_i32 (param $x i32)
;; {{$print_i32/body/begin}}
(local $div i32)
(local $rem i32)
;; if $x == 0 { print '0'; return }
(i32.eq (local.get $x) (i32.const 0))
if
(call $putchar (i32.const 48)) ;; '0'
(call $putchar (i32.const 10)) ;; '\n'
(return)
end
;; if $x < 0 { $x = 0-$x; print '-'; }
(i32.lt_s (local.get $x) (i32.const 0))
if
(local.set $x (i32.sub (i32.const 0) (local.get $x)))
(call $putchar (i32.const 45)) ;; '-'
end
;; print_i32($x / 10)
;; puchar($x%10 + '0')
(local.set $div (i32.div_s (local.get $x) (i32.const 10)))
(local.set $rem (i32.rem_s (local.get $x) (i32.const 10)))
(call $print_i32 (local.get $div))
(call $putchar (i32.add (local.get $rem) (i32.const 48))) ;; '0'
;; {{$print_i32/body/end}}
)
;; ----------------------------------------------------
;; _start 函数
;; ----------------------------------------------------
;; _start 函数
(func $_start
;; {{$_start/body/begin}}
(call $main.init)
(call $main.main)
;; {{$_start/body/end}}
)
(func $main.init
;; {{$main.init/body/begin}}
;; 需要有防止重入的措施
;; {{$main.init/body/end}}
)
(func $main.main
;; {{$main.main/body/begin}}
(local $str.hello.ptr i32)
(local $str.hello.len i32)
;; println("AaA")
(call $putchar (i32.const 65)) ;; 'A'
(call $putchar (i32.const 97)) ;; 'a'
(call $putchar (i32.const 65)) ;; 'A'
(call $putchar (i32.const 10)) ;; '\n'
;; print("hello world\n")
;; [104 101 108 108 111 32 119 111 114 108 100 10]
(local.set $str.hello.ptr (global.get $__heap_base))
(local.set $str.hello.len (i32.const 12))
(i32.store offset=0 align=1 (local.get $str.hello.ptr) (i32.const 104))
(i32.store offset=1 align=1 (local.get $str.hello.ptr) (i32.const 101))
(i32.store offset=2 align=1 (local.get $str.hello.ptr) (i32.const 108))
(i32.store offset=3 align=1 (local.get $str.hello.ptr) (i32.const 108))
(i32.store offset=4 align=1 (local.get $str.hello.ptr) (i32.const 111))
(i32.store offset=5 align=1 (local.get $str.hello.ptr) (i32.const 32))
(i32.store offset=6 align=1 (local.get $str.hello.ptr) (i32.const 119))
(i32.store offset=7 align=1 (local.get $str.hello.ptr) (i32.const 111))
(i32.store offset=8 align=1 (local.get $str.hello.ptr) (i32.const 114))
(i32.store offset=9 align=1 (local.get $str.hello.ptr) (i32.const 108))
(i32.store offset=10 align=1 (local.get $str.hello.ptr) (i32.const 100))
(i32.store offset=11 align=1 (local.get $str.hello.ptr) (i32.const 10))
(call $puts (local.get $str.hello.ptr) (local.get $str.hello.len))
;; println(123)
(call $print_i32 (i32.const 123))
(call $putchar (i32.const 10)) ;; '\n'
;; {{$main.main/body/end}}
)
;; ----------------------------------------------------
;; 编译器填充的代码
;; ----------------------------------------------------
;; {{$wa.compiler.output.code}}
;; ----------------------------------------------------
;; END
;; ----------------------------------------------------
)
internal/backends/compiler_wat/compile.go
浏览文件 @ dd2f0fe3
......@@ -3,14 +3,15 @@
package compiler_wat
import (
"fmt"
"sort"
"strings"
"wa-lang.org/wa/internal/backends/compiler_wat/wir"
"wa-lang.org/wa/internal/backends/compiler_wat/wir/wat"
"wa-lang.org/wa/internal/config"
"wa-lang.org/wa/internal/loader"
"wa-lang.org/wa/internal/ssa"
"wa-lang.org/wa/internal/waroot"
)
type Compiler struct {
......@@ -65,7 +66,7 @@ func (p *Compiler) Compile(prog *loader.Program, mainFunc string) (output string
func (p *Compiler) CompileWsFiles(prog *loader.Program) {
var sb strings.Builder
sb.WriteString(modBaseWat_wasi)
sb.WriteString(waroot.GetBaseWsCode(config.WaBackend_wat))
sb.WriteString("\n")
var pkgpathList = make([]string, 0, len(prog.Pkgs))
......@@ -74,14 +75,28 @@ func (p *Compiler) CompileWsFiles(prog *loader.Program) {
}
sort.Strings(pkgpathList)
var lineCommentSep = ";; -" + strings.Repeat("-", 60-4) + "\n"
for _, pkgpath := range pkgpathList {
pkg := prog.Pkgs[pkgpath]
for _, sf := range pkg.WsFiles {
sb.WriteString(fmt.Sprintf(";; ---- %s:%s begin ----\n\n", pkgpath, sf.Name))
sb.WriteString(sf.Code)
sb.WriteString("\n")
sb.WriteString(fmt.Sprintf(";; ---- %s:%s end ----\n\n", pkgpath, sf.Name))
if len(pkg.WsFiles) == 0 {
continue
}
func() {
sb.WriteString(lineCommentSep)
sb.WriteString(";; package: " + pkgpath + "\n")
sb.WriteString(lineCommentSep)
sb.WriteString("\n")
for _, sf := range pkg.WsFiles {
sb.WriteString(";; file: " + sf.Name + "\n")
sb.WriteString("\n")
sb.WriteString(strings.TrimSpace(sf.Code))
sb.WriteString("\n")
}
}()
}
p.module.BaseWat = sb.String()
......
internal/waroot/_waroot/src/math/sqrt.wat.ws 0 → 100644
浏览文件 @ dd2f0fe3
;; Copyright 2023 The Wa Authors. All rights reserved.
(func $$waSqrtF64 (param $x f64) (result f64)
local.get $x
f64.sqrt
)
internal/backends/compiler_wat/base.wat.go → internal/waroot/_waroot/src/runtime/heap.wat.ws
浏览文件 @ dd2f0fe3
// 版权 @2022 凹语言 作者。保留所有权利。
package compiler_wat
const modBaseWat_wasi = `
(memory $memory 1024)
(export "memory" (memory $memory))
;; --------------------------------------------------------
;; | 0 <-- stack --> | <-- static-data --> | <-- heap --> |
;; --------------------------------------------------------
(global $__stack_ptr (mut i32) (i32.const 1024)) ;; index=0
(global $__heap_base (mut i32) (i32.const 2048)) ;; index=1
(global $__heap_max i32 (i32.const 67108864)) ;; 64MB, 1024 page
;; --------------------------------------------------------
;; Stack/Heap/Memory helper functions
;; --------------------------------------------------------
;; Copyright 2023 The Wa Authors. All rights reserved.
(func $$waGetStackPtr (result i32)
(global.get $__stack_ptr)
)
(func $$waSetStackPtr (param $sp i32)
local.get $sp
global.set $__stack_ptr
......@@ -47,19 +29,6 @@ const modBaseWat_wasi = `
global.get $__heap_max
)
;; --------------------------------------------------------
;; math
;; --------------------------------------------------------
(func $$waSqrtF64 (param $x f64) (result f64)
local.get $x
f64.sqrt
)
;; --------------------------------------------------------
;; heap alloc/free
;; --------------------------------------------------------
(func $$waHeapAlloc (param $nbytes i32) (result i32) ;;result = ptr
(local $ptr i32)
......@@ -130,6 +99,7 @@ const modBaseWat_wasi = `
;; i32.const 10
;; call $$runtime.waPrintRune
)
(func $$waHeapFree (param $ptr i32)
local.get $ptr
call $runtime.free
......@@ -143,128 +113,126 @@ const modBaseWat_wasi = `
;; call $$runtime.waPrintRune
)
;; --------------------------------------------------------
(func $$wa.RT.Block.Init (param $ptr i32) (param $item_count i32) (param $release_func i32) (param $item_size i32) (result i32) ;;result = ptr
local.get $ptr
local.get $ptr
local.get $ptr
if
local.get $ptr
i32.const 1
i32.store offset=0 align=1
local.get $ptr
if
local.get $ptr
i32.const 1
i32.store offset=0 align=1
local.get $ptr
local.get $item_count
i32.store offset=4 align=1
local.get $ptr
local.get $item_count
i32.store offset=4 align=1
local.get $ptr
local.get $release_func
i32.store offset=8 align=1
local.get $ptr
local.get $release_func
i32.store offset=8 align=1
local.get $ptr
local.get $item_size
i32.store offset=12 align=1
end
local.get $ptr
local.get $item_size
i32.store offset=12 align=1
end
)
(func $$wa.RT.DupWatStack (param $p i32) (result i32) (result i32) ;;result0 = result1 = p
local.get $p
local.get $p
local.get $p
local.get $p
)
(func $$wa.RT.Block.Retain (param $ptr i32) (result i32) ;;result = ptr
local.get $ptr
local.get $ptr
if
local.get $ptr
local.get $ptr
i32.load offset=0 align=1
i32.const 1
i32.add
i32.store offset=0 align=1
end
local.get $ptr
local.get $ptr
if
local.get $ptr
local.get $ptr
i32.load offset=0 align=1
i32.const 1
i32.add
i32.store offset=0 align=1
end
)
(func $$wa.RT.Block.Release (param $ptr i32)
;;Todo
(local $ref_count i32)
(local $item_count i32)
(local $free_func i32)
(local $item_size i32)
(local $data_ptr i32)
local.get $ptr
i32.const 0
i32.eq
if
return
end
local.get $ptr
i32.load offset=0 align=1
i32.const 1
i32.sub
local.set $ref_count
local.get $ref_count
if
local.get $ptr
local.get $ref_count
i32.store offset=0 align=1
else ;;ref_count == 0
local.get $ptr
i32.load offset=8 align=1
local.set $free_func
local.get $free_func
if ;;free_func != 0
local.get $ptr
i32.load offset=4 align=1
local.set $item_count
local.get $item_count
if ;;item_count > 0
local.get $ptr
i32.load offset=12 align=1
local.set $item_size
local.get $ptr
i32.const 16
i32.add
local.set $data_ptr
loop $free_next
;; onFree(data_ptr)
local.get $data_ptr
local.get $free_func
call_indirect (type $$onFree)
;; item_count--
local.get $item_count
i32.const 1
i32.sub
local.set $item_count
local.get $item_count
if ;;while item_count>0
;; data_ptr += item_size
local.get $data_ptr
local.get $item_size
i32.add
local.set $data_ptr
br $free_next ;;continue
end ;;while item_count>0
end ;;loop $free_next
end ;;if item_count > 0
end ;;free_func != 0
;;Todo
(local $ref_count i32)
(local $item_count i32)
(local $free_func i32)
(local $item_size i32)
(local $data_ptr i32)
local.get $ptr
call $$waHeapFree
end ;;ref_count == 0
i32.const 0
i32.eq
if
return
end
local.get $ptr
i32.load offset=0 align=1
i32.const 1
i32.sub
local.set $ref_count
local.get $ref_count
if
local.get $ptr
local.get $ref_count
i32.store offset=0 align=1
else ;;ref_count == 0
local.get $ptr
i32.load offset=8 align=1
local.set $free_func
local.get $free_func
if ;;free_func != 0
local.get $ptr
i32.load offset=4 align=1
local.set $item_count
local.get $item_count
if ;;item_count > 0
local.get $ptr
i32.load offset=12 align=1
local.set $item_size
local.get $ptr
i32.const 16
i32.add
local.set $data_ptr
loop $free_next
;; onFree(data_ptr)
local.get $data_ptr
local.get $free_func
call_indirect (type $$onFree)
;; item_count--
local.get $item_count
i32.const 1
i32.sub
local.set $item_count
local.get $item_count
if ;;while item_count>0
;; data_ptr += item_size
local.get $data_ptr
local.get $item_size
i32.add
local.set $data_ptr
br $free_next ;;continue
end ;;while item_count>0
end ;;loop $free_next
end ;;if item_count > 0
end ;;free_func != 0
local.get $ptr
call $$waHeapFree
end ;;ref_count == 0
)
(func $$wa.RT.i32_ref_to_ptr (param $b i32) (param $d i32) (result i32) ;;result = ptr
......@@ -274,73 +242,3 @@ const modBaseWat_wasi = `
(func $$wa.RT.slice_to_ptr (param $b i32) (param $d i32) (param $l i32) (param $c i32) (result i32) ;;result = ptr
local.get $d
)
;; --------------------------------------------------------
;; 输出函数
;; --------------------------------------------------------
;; 打印字符串
(func $puts (param $str i32) (param $len i32)
;; {{$puts/body/begin}}
(local $sp i32)
(local $p_iov i32)
(local $p_nwritten i32)
(local $stdout i32)
;; 保存栈指针状态
(local.set $sp (global.get $__stack_ptr))
;; 分配 iov 结构体
(local.set $p_iov (call $$waStackAlloc (i32.const 8)))
;; 返回地址
(local.set $p_nwritten (call $$waStackAlloc (i32.const 4)))
;; 设置字符串指针和长度
(i32.store offset=0 align=1 (local.get $p_iov) (local.get $str))
(i32.store offset=4 align=1 (local.get $p_iov) (local.get $len))
;; 标准输出
(local.set $stdout (i32.const 1))
;; 输出字符串
(call $$runtime.fdWrite
(local.get $stdout)
(local.get $p_iov) (i32.const 1)
(local.get $p_nwritten)
)
;; 重置栈指针
(global.set $__stack_ptr (local.get $sp))
drop
;; {{$puts/body/end}}
)
;; 打印字符
(func $putchar (param $ch i32)
;; {{$putchar/body/begin}}
(local $sp i32)
(local $p_ch i32)
;; 保存栈指针状态
(local.set $sp (global.get $__stack_ptr))
;; 分配字符
(local.set $p_ch (call $$waStackAlloc (i32.const 4)))
(i32.store offset=0 align=1 (local.get $p_ch) (local.get $ch))
;; 输出字符
(call $puts (local.get $p_ch) (i32.const 1))
;; 重置栈指针
(global.set $__stack_ptr (local.get $sp))
;; {{$putchar/body/begin}}
)
;; --------------------------------------------------------
`
internal/waroot/_waroot/src/runtime/print.wat.ws 0 → 100644
浏览文件 @ dd2f0fe3
;; Copyright 2023 The Wa Authors. All rights reserved.
;; 打印字符串
(func $puts (param $str i32) (param $len i32)
;; {{$puts/body/begin}}
(local $sp i32)
(local $p_iov i32)
(local $p_nwritten i32)
(local $stdout i32)
;; 保存栈指针状态
(local.set $sp (global.get $__stack_ptr))
;; 分配 iov 结构体
(local.set $p_iov (call $$waStackAlloc (i32.const 8)))
;; 返回地址
(local.set $p_nwritten (call $$waStackAlloc (i32.const 4)))
;; 设置字符串指针和长度
(i32.store offset=0 align=1 (local.get $p_iov) (local.get $str))
(i32.store offset=4 align=1 (local.get $p_iov) (local.get $len))
;; 标准输出
(local.set $stdout (i32.const 1))
;; 输出字符串
(call $$runtime.fdWrite
(local.get $stdout)
(local.get $p_iov) (i32.const 1)
(local.get $p_nwritten)
)
;; 重置栈指针
(global.set $__stack_ptr (local.get $sp))
drop
;; {{$puts/body/end}}
)
;; 打印字符
(func $putchar (param $ch i32)
;; {{$putchar/body/begin}}
(local $sp i32)
(local $p_ch i32)
;; 保存栈指针状态
(local.set $sp (global.get $__stack_ptr))
;; 分配字符
(local.set $p_ch (call $$waStackAlloc (i32.const 4)))
(i32.store offset=0 align=1 (local.get $p_ch) (local.get $ch))
;; 输出字符
(call $puts (local.get $p_ch) (i32.const 1))
;; 重置栈指针
(global.set $__stack_ptr (local.get $sp))
;; {{$putchar/body/begin}}
)
internal/waroot/base.llvm.ws 0 → 100644
浏览文件 @ dd2f0fe3
; llir code
\ No newline at end of file
internal/waroot/base.wat.ws 0 → 100644
浏览文件 @ dd2f0fe3
;; Copyright 2023 The Wa Authors. All rights reserved.
(memory $memory 1024)
(export "memory" (memory $memory))
;; --------------------------------------------------------
;; | 0 <-- stack --> | <-- static-data --> | <-- heap --> |
;; --------------------------------------------------------
(global $__stack_ptr (mut i32) (i32.const 1024)) ;; index=0
(global $__heap_base (mut i32) (i32.const 2048)) ;; index=1
(global $__heap_max i32 (i32.const 67108864)) ;; 64MB, 1024 page
internal/waroot/waroot.go
浏览文件 @ dd2f0fe3
......@@ -6,11 +6,40 @@ import (
"embed"
"io/fs"
"strings"
"wa-lang.org/wa/internal/config"
)
//go:embed _waroot
var _warootFS embed.FS
//go:embed base.clang.ws
var baseWsFile_clang string
//go:embed base.llvm.ws
var baseWsFile_llvm string
//go:embed base.wat.ws
var baseWsFile_wat string
// 获取汇编基础代码
func GetBaseWsCode(backend string) string {
switch backend {
case config.WaBackend_clang:
return baseWsFile_clang
case config.WaBackend_llvm:
return baseWsFile_llvm
case config.WaBackend_wat:
return baseWsFile_wat
}
for _, s := range config.WaBackend_List {
if s == backend {
return ""
}
}
panic("unreachable")
}
func GetFS() fs.FS {
// embed.FS 均采用 Unix 风格路径
fs, err := fs.Sub(_warootFS, "_waroot/src")
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或