既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!
由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新
func TestReadFirstLineMock(t *testing.T) {
monkey.Patch(ReadFirstLine, func() string {
return “hi”
})
defer monkey.Unpatch(ReadFirstLine)
line := ReadFirstLine()
assert.Equal(t, line, “hi”)
}
复制代码
其中,Patch方法是我们关注的对象,其作用是更改 **第一个参数函数(例子中的ReadFirstLine)的地址,将其改为第二个参数函数的地址**,这样无论在什么环境下,我们调用ReadFirstLine()返回的都是"hi",而UnPatch就是将ReadFirstLine恢复为原来的地址。
运行`go test -v -run TestReadFirstLineMock$ read_file.go read_file_test.go`进行测试。
测试结果:
![](https://img-blog.csdnimg.cn/img_convert/8a0e823169c8f63f07d401867b06fac1.webp?x-oss-process=image/format,png)
可以看到,我们明明在test.txt中第一行为 *hello* ,但是测试中判断第一行为 *hi* 却通过了
## 3、Patch源码
我们点进Patch()源码:
func Patch(target, replacement interface{}) *PatchGuard {
t := reflect.ValueOf(target)
r := reflect.ValueOf(replacement)
patchValue(t, r)
return &PatchGuard{t, r}
}
复制代码
首先通过反射获取了target和replacement的函数地址,之后调用*patchValue()*将target地址改为replacement。我们点进*patchValue(t,r)*
func patchValue(target, replacement reflect.Value) {
lock.Lock()
defer lock.Unlock()
if target.Kind() != reflect.Func {
panic("target has to be a Func")
}
if replacement.Kind() != reflect.Func {
panic("replacement has to be a Func")
}
if target.Type() != replacement.Type() {
panic(fmt.Sprintf("target and replacement have to have the same type %s != %s", target.Type(), replacement.Type()))
}
if patch, ok := patches[target.Pointer()]; ok {
unpatch(target.Pointer(), patch)
}
bytes := replaceFunction(target.Pointer(), (uintptr)(getPtr(replacement)))
patches[target.Pointer()] = patch{bytes, &replacement}
}
复制代码
* 在这一步,首先做了加锁操作,保证一定并发的安全性。
* 接着,对各个参数进行校验。
* 再下来,查看这个函数是不是先前有patch过(这里维护了一个map即patches,存储已pacth的信息)。如果有,那么先unpatch,保证只patch一次。
* 之后的 *replaceFunction(target.Pointer(), (uintptr)(getPtr(replacement)))* 便是真正替代函数地址的方法,其返回的bytes包含了target的元信息。
* 最后将我们patch的原函数放入patches中,保存它的信息,以便之后unpatch时修改回来
我们再点进 *replaceFunction()*
func replaceFunction(from, to uintptr) (original []byte) {
jumpData := jmpToFunctionValue(to)
f := rawMemoryAccess(from, len(jumpData))
original = make([]byte, len(f))
copy(original, f)
copyToLocation(from, jumpData)
return
}
复制代码
* 这里的jumpData保存了to的jump信息
* *rawMemoryAccess()* 返回的是一个结构体f,包含了from(上文的target)的元信息
* original拷贝了f(即from)的元信息,也是返回值
* 之后将jumpData的信息复制到from(即将from的地址改为to的)
至于Unpatch就是从patches这个map中取到原来被替换的函数的信息,将其恢复,就不多赘述
## 4、tips
* 在测试时,有时候会有patch了但替换失败的情况出现,这是由于**函数内联**导致的。我们需要在go test中加上`-gcflags=-l`参数来禁止函数内联。
* 在部分面向安全的操作系统上,系统不允许一个内存页同时写和执行,这会使得Monkey不能运行
* Monkey不是绝对线程安全的
![img](https://img-blog.csdnimg.cn/img_convert/aaf2ef9390294797b9be1b59ae659588.png)
![img](https://img-blog.csdnimg.cn/img_convert/405c7ed467d1163c1d0c3b2132b22b60.png)
**网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。**
**[需要这份系统化的资料的朋友,可以添加戳这里获取](https://bbs.csdn.net/topics/618658159)**
**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**
s.csdn.net/topics/618658159)**
**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**