From 240e775401f33ff3b6f4637dfc257c5993bb487d Mon Sep 17 00:00:00 2001
From: buptczq <9695911+buptczq@users.noreply.github.com>
Date: Sun, 16 May 2021 10:42:24 +0800
Subject: [PATCH] feature: pageant debug log

---
 app/pageant.go   |  7 ++++++-
 utils/pageant.go | 28 ++++++++++++++++++++++++++--
 2 files changed, 32 insertions(+), 3 deletions(-)

diff --git a/app/pageant.go b/app/pageant.go
index 0b712c7..4f2ffd0 100644
--- a/app/pageant.go
+++ b/app/pageant.go
@@ -4,13 +4,18 @@ import (
 	"context"
 	"github.com/buptczq/WinCryptSSHAgent/utils"
 	"io"
+	"os"
 	"sync"
 )
 
 type Pageant struct{}
 
 func (*Pageant) Run(ctx context.Context, handler func(conn io.ReadWriteCloser)) error {
-	win, err := utils.NewPageant()
+	debug := false
+	if os.Getenv("WCSA_DEBUG") == "1" {
+		debug = true
+	}
+	win, err := utils.NewPageant(debug)
 	if err != nil {
 		return err
 	}
diff --git a/utils/pageant.go b/utils/pageant.go
index 7bf8641..627a46e 100644
--- a/utils/pageant.go
+++ b/utils/pageant.go
@@ -51,16 +51,19 @@ type pageantWindow struct {
 	class     *wndClassEx
 	window    windows.Handle
 	requestCh chan request
+	debug     bool
 }
 
-func NewPageant() (*pageantWindow, error) {
+func NewPageant(debug bool) (*pageantWindow, error) {
 
 	classNamePtr, err := syscall.UTF16PtrFromString(className)
 	if err != nil {
 		return nil, err
 	}
 
-	win := new(pageantWindow)
+	win := &pageantWindow{
+		debug: debug,
+	}
 
 	wcex := &wndClassEx{
 		WndProc:   windows.NewCallback(win.wndProc),
@@ -163,10 +166,22 @@ func (s *pageantWindow) wndProc(hWnd windows.Handle, message uint32, wParam, lPa
 	}
 	copyData := (*copyDataStruct)(unsafe.Pointer(lParam))
 	if copyData.dwData != agentCopydataId {
+		if s.debug {
+			println("Pageant: invalid copy data id", copyData.dwData)
+		}
 		return 0
 	}
+	if s.debug {
+		h := [3]uintptr{copyData.lpData, uintptr(copyData.cbData), uintptr(copyData.cbData)}
+		mapName := *(*[]byte)(unsafe.Pointer(&h))
+		if mapName[len(mapName)-1] == 0 {
+			mapName = mapName[:len(mapName)-1]
+		}
+		println("Pageant: OpenFileMapping", copyData.lpData, copyData.cbData, string(mapName))
+	}
 	fileMap, err := OpenFileMapping(fileMapAllAccess, 0, copyData.lpData)
 	if err != nil {
+		println("Pageant: OpenFileMapping error", err.Error())
 		return
 	}
 	defer func() {
@@ -175,22 +190,30 @@ func (s *pageantWindow) wndProc(hWnd windows.Handle, message uint32, wParam, lPa
 	// check security
 	ourself, err := GetUserSID()
 	if err != nil {
+		println("Pageant: GetUserSID error", err.Error())
 		return
 	}
 	ourself2, err := GetDefaultSID()
 	if err != nil {
+		println("Pageant: GetDefaultSID error", err.Error())
 		return
 	}
 	mapOwner, err := GetHandleSID(fileMap)
 	if err != nil {
+		println("Pageant: GetHandleSID error", err.Error())
 		return
 	}
+	if s.debug {
+		println("Pageant: ourSID:", ourself.String(), "ourSID2:", ourself2.String(), "mapOwnerSID:", mapOwner.String())
+	}
 	if !windows.EqualSid(mapOwner, ourself) && !windows.EqualSid(mapOwner, ourself2) {
+		println("Pageant: wrong owning SID of file mapping")
 		return
 	}
 	// get map view
 	sharedMemory, err := windows.MapViewOfFile(fileMap, fileMapWrite, 0, 0, 0)
 	if err != nil {
+		println("Pageant: MapViewOfFile error", err.Error())
 		return
 	}
 	defer windows.UnmapViewOfFile(sharedMemory)
@@ -199,6 +222,7 @@ func (s *pageantWindow) wndProc(hWnd windows.Handle, message uint32, wParam, lPa
 	size := binary.BigEndian.Uint32(sharedMemoryArray[:4])
 	size += 4
 	if size > agentMaxMsglen {
+		println("Pageant: invalid message length", size)
 		return
 	}