Implement authentication middlewares

hnet/handler.go

@@ -193,10 +193,10 @@ func handleUserRelationshipRemove(stream *common.IOStream, player *Player) error
 }
 
 func init() {
-	Handlers[CLIENT_LOGIN] = handleLogin
-	Handlers[CLIENT_LOGIN_RECONNECT] = handleReconnect
-	Handlers[CLIENT_CHANGE_STATUS] = handleStatusChange
-	Handlers[CLIENT_REQUEST_STATS] = handleRequestStats
-	Handlers[CLIENT_RELATIONSHIP_ADD] = handleUserRelationshipAdd
-	Handlers[CLIENT_RELATIONSHIP_REMOVE] = handleUserRelationshipRemove
+	Handlers[CLIENT_LOGIN] = ensureUnauthenticated(handleLogin)
+	Handlers[CLIENT_LOGIN_RECONNECT] = ensureUnauthenticated(handleReconnect)
+	Handlers[CLIENT_CHANGE_STATUS] = ensureAuthentication(handleStatusChange)
+	Handlers[CLIENT_REQUEST_STATS] = ensureAuthentication(handleRequestStats)
+	Handlers[CLIENT_RELATIONSHIP_ADD] = ensureAuthentication(handleUserRelationshipAdd)
+	Handlers[CLIENT_RELATIONSHIP_REMOVE] = ensureAuthentication(handleUserRelationshipRemove)
 }

hnet/middlewares.go

@@ -0,0 +1,29 @@
+package hnet
+
+import (
+	"fmt"
+
+	"github.com/hexis-revival/hexagon/common"
+)
+
+func ensureAuthentication(handler func(*common.IOStream, *Player) error) func(*common.IOStream, *Player) error {
+	return func(stream *common.IOStream, player *Player) error {
+		if !player.IsAuthenticated() {
+			player.CloseConnection()
+			return fmt.Errorf("unauthenticated player")
+		}
+
+		return handler(stream, player)
+	}
+}
+
+func ensureUnauthenticated(handler func(*common.IOStream, *Player) error) func(*common.IOStream, *Player) error {
+	return func(stream *common.IOStream, player *Player) error {
+		if player.IsAuthenticated() {
+			player.CloseConnection()
+			return fmt.Errorf("already authenticated")
+		}
+
+		return handler(stream, player)
+	}
+}

hnet/player.go

@@ -76,6 +76,10 @@ func (player *Player) SendPacket(packetId uint32, packet Serializable) error {
 	return player.SendPacketData(packetId, stream.Get())
 }
 
+func (player *Player) IsAuthenticated() bool {
+	return player.Info.Id != 0 || player.Info.Name != ""
+}
+
 func (player *Player) OnLoginSuccess(responsePassword string, userObject *common.User) error {
 	otherUser := player.Server.Players.ByID(uint32(userObject.Id))