fix: handle accounts

Author: ItzSwirlz

globals/accounts.go

@@ -0,0 +1,63 @@
+package globals
+
+import (
+	"strconv"
+
+	"github.com/PretendoNetwork/nex-go/v2"
+	"github.com/PretendoNetwork/nex-go/v2/types"
+)
+
+var AuthenticationServerAccount *nex.Account
+var SecureServerAccount *nex.Account
+
+func InitAccounts() {
+	AuthenticationServerAccount = nex.NewAccount(types.NewPID(1), "Quazal Authentication", KerberosPassword)
+	SecureServerAccount = nex.NewAccount(types.NewPID(2), "Quazal Rendez-Vous", KerberosPassword)
+}
+
+func AccountDetailsByPID(pid *types.PID) (*nex.Account, *nex.Error) {
+	if pid.Equals(AuthenticationServerAccount.PID) {
+		return AuthenticationServerAccount, nil
+	}
+
+	if pid.Equals(SecureServerAccount.PID) {
+		return SecureServerAccount, nil
+	}
+
+	password, errorCode := PasswordFromPID(pid)
+	if errorCode != 0 {
+		return nil, nex.NewError(errorCode, "Failed to get password from PID")
+	}
+
+	account := nex.NewAccount(pid, strconv.Itoa(int(pid.LegacyValue())), password)
+
+	return account, nil
+}
+
+func AccountDetailsByUsername(username string) (*nex.Account, *nex.Error) {
+	if username == AuthenticationServerAccount.Username {
+		return AuthenticationServerAccount, nil
+	}
+
+	if username == SecureServerAccount.Username {
+		return SecureServerAccount, nil
+	}
+
+	pidInt, err := strconv.Atoi(username)
+	if err != nil {
+		Logger.Error(err.Error())
+		return nil, nex.NewError(nex.ResultCodes.RendezVous.InvalidUsername, "Invalid username")
+	}
+
+	pid := types.NewPID(uint64(pidInt))
+
+	password, errorCode := PasswordFromPID(pid)
+	if errorCode != 0 {
+		Logger.Errorf("Password err: %v", errorCode)
+		return nil, nex.NewError(errorCode, "Failed to get password from PID")
+	}
+
+	account := nex.NewAccount(pid, username, password)
+
+	return account, nil
+}

globals/globals.go

@@ -9,6 +9,7 @@ import (
 )
 
 var Logger = plogger.NewLogger()
+var KerberosPassword = "password" // * Default password
 var SecureServer *nex.PRUDPServer
 var SecureEndpoint *nex.PRUDPEndPoint
 var GRPCFriendsClientConnection *grpc.ClientConn

globals/password_from_pid.go

@@ -0,0 +1,23 @@
+package globals
+
+import (
+	"context"
+
+	"github.com/PretendoNetwork/nex-go/v2/types"
+
+	pb "github.com/PretendoNetwork/grpc-go/account"
+	"github.com/PretendoNetwork/nex-go/v2"
+	"google.golang.org/grpc/metadata"
+)
+
+func PasswordFromPID(pid *types.PID) (string, uint32) {
+	ctx := metadata.NewOutgoingContext(context.Background(), GRPCAccountCommonMetadata)
+
+	response, err := GRPCAccountClient.GetNEXPassword(ctx, &pb.GetNEXPasswordRequest{Pid: pid.LegacyValue()})
+	if err != nil {
+		Logger.Error(err.Error())
+		return "", nex.ResultCodes.RendezVous.InvalidUsername
+	}
+
+	return response.Password, 0
+}

init.go

@@ -24,10 +24,17 @@ func init() {
 		logger.Warning("Error loading .env file")
 	}
 
+	kerberosPassword := os.Getenv("PN_WIIU_CHAT_KERBEROS_PASSWORD")
 	friendsGRPCHost := os.Getenv("PN_WIIU_CHAT_FRIENDS_GRPC_HOST")
 	friendsGRPCPort := os.Getenv("PN_WIIU_CHAT_FRIENDS_GRPC_PORT")
 	friendsGRPCAPIKey := os.Getenv("PN_WIIU_CHAT_FRIENDS_GRPC_API_KEY")
 
+	if strings.TrimSpace(kerberosPassword) == "" {
+		globals.Logger.Warningf("PN_KERBEROS_PASSWORD environment variable not set. Using default password: %q", globals.KerberosPassword)
+	} else {
+		globals.KerberosPassword = kerberosPassword
+	}
+
 	if strings.TrimSpace(friendsGRPCHost) == "" {
 		globals.Logger.Error("PN_WIIU_CHAT_FRIENDS_GRPC_HOST environment variable not set")
 		os.Exit(0)
@@ -61,5 +68,6 @@ func init() {
 		"X-API-Key", friendsGRPCAPIKey,
 	)
 
+	globals.InitAccounts()
 	database.ConnectAll()
 }

nex/server.go

@@ -13,11 +13,15 @@ func StartNEXServer() {
 
 	globals.SecureEndpoint = nex.NewPRUDPEndPoint(1)
 	globals.SecureEndpoint.IsSecureEndPoint = true
-	globals.SecureEndpoint.SetAccessKey("e7a47214")
-	globals.SecureServer.LibraryVersions.SetDefault(nex.NewLibraryVersion(3, 4, 2))
+	globals.SecureEndpoint.AccountDetailsByPID = globals.AccountDetailsByPID
+	globals.SecureEndpoint.AccountDetailsByUsername = globals.AccountDetailsByUsername
+	globals.SecureEndpoint.ServerAccount = globals.SecureServerAccount
 	globals.SecureServer.BindPRUDPEndPoint(globals.SecureEndpoint)
 	// globals.SecureServer.SetPingTimeout(65535) // TODO: what to replace with
 
+	globals.SecureServer.LibraryVersions.SetDefault(nex.NewLibraryVersion(3, 4, 2))
+	globals.SecureEndpoint.SetAccessKey("e7a47214")
+
 	globals.SecureEndpoint.OnData(func(packet nex.PacketInterface) {
 		request := packet.RMCMessage()