From e6172fa941f9d2e06fffc09bd4acb2ceada8372c Mon Sep 17 00:00:00 2001
From: Palapeli <26661008+mkwcat@users.noreply.github.com>
Date: Sun, 4 Jan 2026 01:14:19 -0500
Subject: [PATCH] NAS: Support multiple Stage1 versions

---
 nas/payload.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/nas/payload.go b/nas/payload.go
index 36ba5cc..ff79997 100644
--- a/nas/payload.go
+++ b/nas/payload.go
@@ -9,17 +9,21 @@ import (
 	"crypto/x509"
 	"encoding/hex"
 	"encoding/pem"
-	"github.com/logrusorgru/aurora/v3"
 	"net/http"
 	"net/url"
 	"os"
 	"strconv"
 	"wwfc/logging"
+
+	"github.com/logrusorgru/aurora/v3"
 )
 
 func downloadStage1(w http.ResponseWriter, stage1Ver int) {
-	// TODO: Actually use the stage 1 version
-	dat, err := os.ReadFile("payload/stage1.bin")
+	path := "payload/stage1.bin"
+	if stage1Ver != 0 {
+		path = "payload/stage1v" + strconv.Itoa(stage1Ver) + ".bin"
+	}
+	dat, err := os.ReadFile(path)
 	if err != nil {
 		panic(err)
 	}