Hook系统调用

可以用来拦截网络流量

  1. package hook
  3. import (
  4.     "net"
  5.     "runtime"
  6.     "syscall"
  7. )
  9. // GetCurrentGoRoutineID GetCurrentGoRoutineID
  10. func GetCurrentGoRoutineID() int64 {
  11.     return runtime.GetCurrentGoRoutineId()
  12. }
  14. // SetDelegatedFromGoRoutineID SetDelegatedFromGoRoutineID
  15. func SetDelegatedFromGoRoutineID(gID int64) {
  16.     runtime.SetDelegatedFromGoRoutineId(gID)
  17. }
  19. // RegisterOnConnect RegisterOnConnect
  20. func RegisterOnConnect(callback func(fd int, sa syscall.Sockaddr)) {
  21.     syscall.OnConnect = callback
  22. }
  24. // RegisterOnAccept RegisterOnAccept
  25. func RegisterOnAccept(callback func(serverSocketFD int, clientSocketFD int, sa syscall.Sockaddr)) {
  26.     syscall.OnAccept = callback
  27. }
  29. // RegisterOnRecv RegisterOnRecv
  30. func RegisterOnRecv(callback func(fd int, net string, raddr net.Addr, span []byte)) {
  31.     net.OnRead = callback
  32. }
  34. // RegisterOnSend RegisterOnSend
  35. func RegisterOnSend(callback func(fd int, net string, raddr net.Addr, span []byte)) {
  36.     net.OnWrite = callback
  37. }
  39. // RegisterOnClose RegisterOnClose
  40. func RegisterOnClose(callback func(fd int)) {
  41.     net.OnClose = callback
  42. }
  44. // RegisterOnGoRoutineExit RegisterOnGoRoutineExit
  45. func RegisterOnGoRoutineExit(callback func(goid int64)) {
  46.     runtime.OnGoRoutineExit = callback
  47. }

go:linkname

可以实现引导编译器将当前(私有)方法或者变量在编译时链接到指定的位置的方法或者变量，第一个参数表示当前方法或变量，第二个参数表示目标方法或变量，因为这关指令会破坏系统和包的模块化，因此在使用时必须导入unsafe

Prof

如果你的go程序不是web服务器，而是一个服务进程，那么你也可以选择使用net/http/pprof包，同样引入包net/http/pprof，然后在开启另外一个goroutine来开启端口监听。

  1. go func() {
  2.         log.Println(http.ListenAndServe("localhost:6060", nil)) 
  3. }()

调用go tool pprof http://localhost:3999/debug/pprof/profile来开始采集数据，
之后输入png就可以生成一个调用图
image.png
pprof现在支持加label可以观察时间都耗费在哪里