当前位置:首页 > 技术分析 > 正文内容

注释竟然还有特殊用途?一文解惑 //go:linkname 指令

ruisui883个月前 (02-07)技术分析10

大家好,我是站长 polarisxu。

我之前写过一篇文章:为什么 Go 标准库中有些函数只有签名,没有函数体?,其中有一点就是 //go:linkname 这个指令。

Go 中类似的指令挺多的,比如 Go1.16 中的 //go:embed。前些天有人问我,为什么它用 //go:embed 不起作用?我一看,它是这么写的:// go:embed,不知道你看到问题了没有?是的,指令是通过注释的方式,但有三点要求,要特别注意:

  • // 后不能有空格。有些人可能习惯 // 后不加空格。但一般认为,// 后应该加一个空格。不过 go 指令却要求不能有空格,这是一个小“坑”,得注意。所以上面那位朋友就是加了空格,导致出问题。(程序并不会报错,只是没有得到自己想要的结果)
  • 代码和指令之间不能有空行或其他注释。这一点应该还好,很多人不会用错吧;
  • 一般来说,使用指令需要导入相应的包。比如 //go:linkname 指令要求导入 unsafe 包,一般会 import _ "unsafe”//go:embed 指令,要求导入 embed 包。

有另外一位 Go 朋友「橘中秘士」微信私聊我:

大佬好,能不能写一篇 linkname 的文章。目前已经有了一些初步概念,但是尚有一些疑团不是特别清晰。

//go:linkname localname remotename,其中 local 作为占位符 remote 作为实现者或者 local 作为实现者 remote 作为占位符都是可以的。目前理解的就是给 Symbol 添加了一个 Linkname,查找 Symbo l的时候用 remote。

譬如 //go:linkname runtimeNano runtime.nanotime,runtimeNano 作为占位符 runtime.nanotime 提供实现,任何调用 runtimeNano 的地方实际替换为对 runtime.nanotime 的调用,这种场景比较容易接受。

譬如 //go:linkname runtime_cmpstring runtime.cmpstring,runtime_cmpstring 提供实现 runtime.cmpstring作为占位符,是不是这时符号表里不存在 runtime_cmpstring 只有 runtime.cmpstring?

经过简单沟通,他写了一篇文章解决自己的困惑。希望对各位有帮助。以下是他写的关于 //go:linkname 的文章(我做了一些调整)。


01 格式

//go:linkname?local?remote

remote 可以没有,此时 remote 使用 local 的值,效果就是 local 被导出。

02 local 和 remote 同时为函数

local 作为占位符,remote 作为实现者

标准库中的例子:

//?来自?time?包
//go:linkname?runtimeNano?runtime.nanotime
func?runtimeNano()?int64

//?来自?runtime?包
//go:nosplit
func?nanotime()?int64?{
?return?nanotime1()
}

此时二进制文件中并没有runtimeNano,直接转化为对runtime.nanotime的调用。

local 作为实现者,remote 作为占位符

同样来自标准库。这里存在函数没有函数体,但是被反向引用。

//?在标准库的一个?internal?中
//go:linkname?runtime_cmpstring?runtime.cmpstring
func?runtime_cmpstring(a,?b?string)?int?{
?l?:=?len(a)
?if?len(b)??c2?{
???return?+1
??}
?}
?if?len(a)??len(b)?{
??return?+1
?}
?return?0
}

//?来自?runtime
func?cmpstring(string,?string)?int

此时二进制文件中并没有runtime_cmpstring,对应的函数已经被命名为runtime.cmpstring。也就是说,实现在 internal 包,但最终通过 runtime.cmpstring 来引用。

一个占位符+一个汇编函数

//?在标准库的一个?internal?中
//go:linkname?abigen_runtime_memequal?runtime.memequal
func?abigen_runtime_memequal(a,?b?unsafe.Pointer,?size?uintptr)?bool

注意runtime.memequal的实现并不在runtime包中,使用汇编实现的话并不要求必须在相应的包中。

# memequal(a, b unsafe.Pointer, size uintptr) bool
TEXT runtime·memequal(SB),NOSPLIT,$0-25
	MOVQ	a+0(FP), SI
	MOVQ	b+8(FP), DI
	CMPQ	SI, DI
	JEQ	eq
	MOVQ	size+16(FP), BX
	LEAQ	ret+24(FP), AX
	JMP	memeqbody<>(SB)
eq:
	MOVB	$1, ret+24(FP)
	RET

03 local 和 remote 同时为变量

两个常规变量

//go:linkname?overflowError?runtime.overflowError
var?overflowError?error

//go:linkname?divideError?runtime.divideError
var?divideError?error

//go:linkname?zeroVal?runtime.zeroVal
var?zeroVal?[maxZero]byte

//go:linkname?_iscgo?runtime.iscgo
var?_iscgo?bool?=?true

//go:cgo_import_static?x_cgo_setenv
//go:linkname?x_cgo_setenv?x_cgo_setenv
//go:linkname?_cgo_setenv?runtime._cgo_setenv
var?x_cgo_setenv?byte
var?_cgo_setenv?=?&x_cgo_setenv

//go:cgo_import_static?x_cgo_unsetenv
//go:linkname?x_cgo_unsetenv?x_cgo_unsetenv
//go:linkname?_cgo_unsetenv?runtime._cgo_unsetenv
var?x_cgo_unsetenv?byte
var?_cgo_unsetenv?=?&x_cgo_unsetenv

一个占位符+一个伪符号

//go:linkname?runtime_inittask?runtime..inittask
var?runtime_inittask?initTask

//go:linkname?main_inittask?main..inittask
var?main_inittask?initTask

注意是..inittask不是.inittask,而且.inittask只存在于编译阶段,任何包中都无法声明该变量。

这里额外解释下 ..inittask 为什么两个点。第一个点就是普通的 runtime. 这种调用方式,第二个点和 inittask 一起构成一个符号(变量)。注意,Go 中的变量是不允许以 . 开头的,所以,这个叫伪符号,只在不编译阶段存在。

04 一个例子

研究 //go:linkname 是因为如下的背景:

Java 里有 InheritableThreadLocal,SpringWeb 在 ServletActionContext 里使用它,达到在任何地方都能方便的获取HttpServletRequest。


Go 并没有提供类似的机制,即使通过 stack 找到 goroutine id(99% 的文章都是这么介绍的),再配合 sync.Map,也只是实现了一个比较粗糙的 ThreadLocal,在子协程里仍然获取不到父协程的内容。


g.label 虽然不是给这种场景准备的,但它具备了 InheritableThreadLocal 的一切要求,只要我们能够访问到 label 私有字段,我们就有了完整版的 InheritableThreadLocal。

下面这个例子是作者真实项目中用的。

在 runtime 和 runtime/pprof 包中有两个函数:runtime_setProfLabel 和 runtime_getProfLabel。其中,runtime 包中的提供了实现,而 pprof 中的没有提供实现。如果基于它们创建另外的函数,如下:

//go:linkname?SetPointer?runtime/pprof.runtime_setProfLabel
func?SetPointer(ptr?unsafe.Pointer)

//go:linkname?GetPointer?runtime/pprof.runtime_getProfLabel
func?GetPointer()?unsafe.Pointer

根据前面的分析,虽然runtime.runtime_setProfLabel/runtime.runtime_getProfLabel提供了函数实现,但是二进制文件中并不会出现(见下方代码),此时想要调用必须通过runtime/pprof.runtime_setProfLabel/runtime/pprof.runtime_getProfLabel,这也是上面linknamepprof而不是runtime的根本原因。

//?来自?runtime?包
//go:linkname?runtime_setProfLabel?runtime/pprof.runtime_setProfLabel
func?runtime_setProfLabel(labels?unsafe.Pointer)?{
?if?raceenabled?{
??racereleasemerge(unsafe.Pointer(&labelSync))
?}
?getg().labels?=?labels
}

//?来自?runtime/pprof?包
func?runtime_setProfLabel(labels?unsafe.Pointer)

//?来自?runtime?包
//go:linkname?runtime_getProfLabel?runtime/pprof.runtime_getProfLabel
func?runtime_getProfLabel()?unsafe.Pointer?{
?return?getg().labels
}

//?来自?runtime/pprof?包
func?runtime_getProfLabel()?unsafe.Pointer

05 总结

Go 中有不少指令,有些指令你可能不太需要关心,也不会用到。然而有些指令了解它们的意思,对阅读相关代码很有帮助。

这篇文章全面介绍了 //go:linkname 指令,不知道是否彻底解除了你的疑惑?欢迎留言交流!



扫描二维码推送至手机访问。

版权声明:本文由ruisui88发布,如需转载请注明出处。

本文链接:http://www.ruisui88.com/post/1760.html

标签: pprof
分享给朋友:

“注释竟然还有特殊用途?一文解惑 //go:linkname 指令” 的相关文章

学会使用Vue JSX,一车老干妈都是你的

作者:子君转发链接:https://mp.weixin.qq.com/s/eAOivpHeowLShfwPfW8-BA?君自前端来,应知前端事。需求时时变,bug改不完。?连续几篇文章,每篇都有女神,被老铁给吐槽了,今天不提了女神了,反正女神都是别人的(扎心了)。这两天小编看了腾讯与老干妈的事情,晚...

VIM配置整理

一、基本配色set number set showcmd set incsearch set expandtab set showcmd set history=400 set autoread set ffs=unix,mac,dos set hlsearch set shiftwidth=2 s...

深度解析!AI智能体在To B领域应用,汽车售后服务落地全攻略

在汽车售后服务领域,AI智能体的应用正带来一场效率和专业度的革命。本文深度解析了一个AI智能体在To B领域的实际应用案例,介绍了AI智能体如何通过提升服务顾问和维修技师的专业度及维修效率,优化汽车售后服务流程。上周我分享了AI智能体+AI小程序To C的AI应用场景《1000%增长!我仅用一个小时...

前端路由简介以及vue-router实现原理

作者:muwoo 来源:https://zhuanlan.zhihu.com/p/37730038后端路由简介路由这个概念最先是后端出现的。在以前用模板引擎开发页面时,经常会看到这样http://www.xxx.com/login 大致流程可以看成这样:浏览器发出请求服务器监听到80 端口(或443...

微信将研发“应用号”体系 或成为App大杂烩应用

Akiha发表于 2016-01-11 18:11微信正在探讨一种新的公众号形态,即在现有的订阅号、服务号之外,再研发新的“应用号”体系。让用户不必去装各种 App,在微信里就能实现各种服务、功能;关注后,“应用号”平时不向用户发送消息,需要时,用户再去点开“应用号”等。目前,微信钱包内已有的服务,...

Intel RST技术操作指南

一、RST技术介绍1、RST:Intel Rapid Storage Technology(Intel快速存储技术)的英文简称,功能是提高计算机系统性能。就是用一块固态硬盘(SSD)给机械硬盘(HDD)做缓存,提高系统存储的读写速度。2、优势:它具有的特点和优势是能够提升整体硬盘的速度,但是却不会降...