mirror of
https://github.com/yv1ing/ShotRDP.git
synced 2025-09-16 15:10:57 +08:00
35 lines
591 B
Go
35 lines
591 B
Go
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)
|
|
}
|
|
}
|