mirror of
https://github.com/yv1ing/ShotRDP.git
synced 2025-09-16 15:10:57 +08:00
新增动态链接库调用
This commit is contained in:
34
mainRPC.go
Normal file
34
mainRPC.go
Normal file
@@ -0,0 +1,34 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"ShotRDP/client"
|
||||
"context"
|
||||
"google.golang.org/grpc"
|
||||
"net"
|
||||
|
||||
"ShotRDP/pb"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
pb.UnimplementedGetScreenServer
|
||||
}
|
||||
|
||||
func (server *Server) GetScreen(ctx context.Context, in *pb.Request) (*pb.Response, error) {
|
||||
imgBytes, err := client.RealGetScreen(in.GetHost())
|
||||
return &pb.Response{ImageBytes: imgBytes}, err
|
||||
}
|
||||
|
||||
func main() {
|
||||
listener, err := net.Listen("tcp", ":8972")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
server := grpc.NewServer()
|
||||
pb.RegisterGetScreenServer(server, &Server{})
|
||||
|
||||
err = server.Serve(listener)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user