wfc-server/qr2/challenge.go

package qr2

import (
	"fmt"
	"net"
	"strconv"
	"strings"
	"time"
	"wwfc/common"
	"wwfc/logging"

	"github.com/logrusorgru/aurora/v3"
)

func sendChallenge(conn net.PacketConn, addr net.UDPAddr, session Session, lookupAddr uint64) {
	challenge := session.Challenge
	if challenge == "" {
		// Generate challenge
		addrString := strings.Split(addr.String(), ":")
		var hexIP string
		for _, i := range strings.Split(addrString[0], ".") {
			val, err := strconv.ParseUint(i, 10, 64)
			if err != nil {
				panic(err)
			}

			hexIP += fmt.Sprintf("%02X", val)
		}

		port, err := strconv.ParseUint(addrString[1], 10, 64)
		if err != nil {
			panic(err)
		}

		hexPort := fmt.Sprintf("%04X", port)

		challenge = common.RandomString(6) + "00" + hexIP + hexPort
		mutex.Lock()
		if sessionPtr := sessions[lookupAddr]; sessionPtr != nil {
			sessionPtr.Challenge = challenge
		} else {
			mutex.Unlock()
			return
		}
		mutex.Unlock()
	}

	response := createResponseHeader(ChallengeRequest, session.SessionID)
	response = append(response, []byte(challenge)...)
	response = append(response, 0)

	go func() {
		for i := 0; i < 5; i++ {
			_, err := conn.WriteTo(response, &addr)

			time.Sleep(1 * time.Second)

			if err != nil {
				continue
			}

			mutex.Lock()
			session, ok := sessions[lookupAddr]
			if !ok || session.Authenticated || session.LastKeepAlive < time.Now().UTC().Unix()-60 {
				mutex.Unlock()
				return
			}
			addr = session.Addr
			mutex.Unlock()
		}
		logging.Info("QR2", "Failed to send challenge to", aurora.Cyan(addr.String()))
		mutex.Lock()
		defer mutex.Unlock()
		removeSession(lookupAddr)
	}()
}