github.com/HXSecurity/DongTai-agent-go@v0.4.2/core/grpc/clientConn/install.go (about) 1 package clientConn 2 3 import ( 4 "fmt" 5 "github.com/HXSecurity/DongTai-agent-go/model" 6 "github.com/brahma-adshonor/gohook" 7 "google.golang.org/grpc" 8 ) 9 10 func init() { 11 model.HookMap["grpcClientConn"] = new(GrpcClientConn) 12 } 13 14 type GrpcClientConn struct { 15 } 16 17 func (h *GrpcClientConn) Hook() { 18 cl := &grpc.ClientConn{} 19 err := gohook.HookMethod(cl, "Invoke", Invoke, InvokeT) 20 if err != nil { 21 fmt.Println(err, "GrpcClientConn") 22 } else { 23 fmt.Println("GrpcClientConn") 24 } 25 } 26 27 func (h *GrpcClientConn) UnHook() { 28 cl := &grpc.ClientConn{} 29 gohook.UnHookMethod(cl, "Invoke") 30 }