//go:build windows

package daemon

import (
	"bytes"
	"log"
	"net"
	"net/rpc"
	"os"
	"os/signal"
	"strconv"
	"time"

	"github.com/Microsoft/go-winio"
	"github.com/rwinkhart/go-boilerplate/security"
	"github.com/rwinkhart/peercred-mini"
	"golang.org/x/sys/windows"
)

const (
	PROCESS_QUERY_LIMITED_INFORMATION = 0x1000
)

// Start is the entry point for the RPC server responsible for
// returning decrypted data to authenticated clients.
func Start(password []byte) {
	// store password to be referenced by DecryptRequest method
	globalPassword = password

	// register RCWService with the RPC package
	err := rpc.Register(&RCWService{})
	if err != nil {
		log.Fatalf("Error registering RPC service: %v", err)
	}

	// store the hash of the daemon binary
	daemonHash, err = getFileHash(binPath)
	if err != nil {
		log.Fatalf("Error hashing daemon binary: %v", err)
	}

	// configure the named pipe
	pipeConfig := &winio.PipeConfig{
		SecurityDescriptor: "", // Default security
		MessageMode:        true,
		InputBufferSize:    65536,
		OutputBufferSize:   65536,
	}

	// create the named pipe listener
	listener, err := winio.ListenPipe(socketPath, pipeConfig)
	if err != nil {
		log.Fatalf("Failed to listen on named pipe %s: %v", socketPath, err)
	}
	defer listener.Close()
	log.Printf("RPC daemon listening on %s", socketPath)

	// capture termination signals to ensure listener is closed
	sigChan := make(chan os.Signal, 1)
	signal.Notify(sigChan, os.Interrupt)
	// create inactivity timer
	timer := time.NewTimer(time.Duration(Timeout) * time.Second)
	killTimer := make(chan struct{})
	go func() {
		select {
		case <-timer.C:
			log.Println(strconv.Itoa(Timeout) + " seconds have passed without any connections. Exiting...")
			listener.Close()
			security.ZeroizeBytes(globalPassword)
			os.Exit(0)
		case <-killTimer:
			return
		case <-sigChan:
			listener.Close()
			security.ZeroizeBytes(globalPassword)
			os.Exit(0)
		}
	}()

	// accept connections
	for {
		conn, err := listener.Accept()
		if err != nil {
			log.Printf("Accept error: %v", err)
			close(killTimer)
			continue
		}

		// reset timer after connection is accepted
		timer.Reset(3 * time.Minute)

		// use a goroutine to check the client's identity
		go handleConn(conn, sigChan)
	}
}

// handleConn verifies the identity of the client.
// It gets the PID of the client process and verifies it's running the same binary
func handleConn(conn net.Conn, sigChan chan os.Signal) {
	ucred := peercred.Get(conn)

	// get server SID (UID)
	var token windows.Token
	windows.OpenProcessToken(windows.CurrentProcess(), windows.TOKEN_QUERY, &token)
	defer token.Close()
	user, _ := token.GetTokenUser()

	// check if the RPC call is coming from an identical binary and from the same user
	callingBinPath := pidToPath(uint32(ucred.PID))
	callingBinHash, err := getFileHash(callingBinPath)
	if err != nil {
		// calling binary hash failure
		conn.Close()
		log.Printf("Failed to hash calling binary: PID(%d), UID(%s), Path(%s) - %v", ucred.PID, ucred.UID, callingBinPath, err)
		sigChan <- os.Interrupt // this zeroizes globalPassword and triggers os.Exit(0)
		return                  // explicitly return to avoid race
	}
	if ucred.UID == user.User.Sid.String() && bytes.Equal(callingBinHash, daemonHash) {
		rpc.ServeConn(conn)
	} else {
		// invalid client; close the connection w/o a response,
		// log the client's path, and kill the daemon
		conn.Close()
		log.Printf("Request received from invalid client: PID(%d), UID(%s), Path(%s)", ucred.PID, ucred.UID, callingBinPath)
		sigChan <- os.Interrupt // this zeroizes globalPassword and triggers os.Exit(0)
	}
}