aion-server 4.8

package com.aionemu.gameserver.services.player;

import java.time.Duration;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.aionemu.gameserver.configs.main.SecurityConfig;
import com.aionemu.gameserver.configs.main.SecurityConfig.MultiClientingRestrictionMode;
import com.aionemu.gameserver.model.Race;
import com.aionemu.gameserver.model.gameobjects.player.Player;
import com.aionemu.gameserver.network.aion.AionConnection;
import com.aionemu.gameserver.world.World;

public class MultiClientingService {

	private static final Logger log = LoggerFactory.getLogger(MultiClientingService.class);
	private static final Map<Integer, AccountSession> sessionsByAccountId = new ConcurrentHashMap<>();

	public static boolean tryEnterWorld(Player player, AionConnection con) {
		if (SecurityConfig.MULTI_CLIENTING_RESTRICTION_MODE == MultiClientingRestrictionMode.FULL && !SecurityConfig.MULTI_CLIENTING_IGNORED_MAC_ADDRESSES.contains(con.getMacAddress())) {
			String mac = con.getMacAddress();
			String hdd = con.getHddSerial();
			String ip = con.getIP();
			for (Player onlinePlayer : World.getInstance().getAllPlayers()) {
				boolean sameIp = ip.equals(onlinePlayer.getClientConnection().getIP());
				boolean sameMac = !mac.isEmpty() && mac.equals(onlinePlayer.getClientConnection().getMacAddress());
				boolean sameHdd = !hdd.isEmpty() && hdd.equals(onlinePlayer.getClientConnection().getHddSerial());
				if (sameIp && (sameMac || sameHdd)) {
					log.info("Blocked {} from logging on (multi-clienting on {} with {})", player, sameMac ? "MAC address " + mac : "HDD " + hdd, onlinePlayer);
					return false;
				}
			}
		} else if (SecurityConfig.MULTI_CLIENTING_RESTRICTION_MODE == MultiClientingRestrictionMode.SAME_FACTION) {
			sessionsByAccountId.values().removeIf(AccountSession::isExpired);
			synchronized (sessionsByAccountId) {
				Integer matchedAccountId = checkForFactionSwitchCooldownTime(player.getRace(), con);
				if (matchedAccountId != null) {
					log.info("Blocked {} from logging on (faction switch cooldown from account ID {})", player, matchedAccountId);
					return false;
				}
				AccountSession accountSession = sessionsByAccountId.computeIfAbsent(player.getAccount().getId(), AccountSession::new);
				accountSession.putIdentifiers(con);
				accountSession.enterWorld(player);
			}
		}
		return true;
	}

	public static void onLeaveWorld(Player player) {
		AccountSession session = sessionsByAccountId.get(player.getAccount().getId());
		if (session != null)
			session.leaveWorld(player);
	}

	public static Integer checkForFactionSwitchCooldownTime(Race race, AionConnection con) {
		if (SecurityConfig.MULTI_CLIENTING_IGNORED_MAC_ADDRESSES.contains(con.getMacAddress()))
			return null;
		Race oppositeRace = race == Race.ELYOS ? Race.ASMODIANS : Race.ELYOS;
		long minLastOnlineMillis = System.currentTimeMillis() - Duration.ofMinutes(SecurityConfig.MULTI_CLIENTING_FACTION_SWITCH_COOLDOWN_MINUTES).toMillis();
		return sessionsByAccountId.values().stream()
			.filter(s -> !s.isIgnored() && s.wasPlayingOnSameIpOrMac(oppositeRace, minLastOnlineMillis, con))
			.findAny()
			.map(s -> s.accountId)
			.orElse(null);
	}

	private static class AccountSession {

		private final int accountId;
		private final Map<Race, Long> lastCharOnlineTimeMillis = new ConcurrentHashMap<>();
		private final List<Identifiers> identifiers = new LinkedList<>();

		public AccountSession(int accountId) {
			this.accountId = accountId;
		}

		synchronized boolean isIgnored() {
			return !identifiers.isEmpty() && SecurityConfig.MULTI_CLIENTING_IGNORED_MAC_ADDRESSES.contains(identifiers.getFirst().mac);
		}

		synchronized void putIdentifiers(AionConnection connection) {
			Identifiers ids = new Identifiers(connection.getIP(), connection.getMacAddress());
			if (!identifiers.contains(ids)) {
				identifiers.addFirst(ids);
				while (identifiers.size() > 3)
					identifiers.removeLast();
			}
		}

		synchronized boolean hasAny(String ip, String mac) {
			return identifiers.stream().anyMatch(identifiers -> identifiers.ip.equals(ip) || identifiers.mac.equals(mac));
		}

		boolean isExpired() {
			long minLastOnline = System.currentTimeMillis() - Duration.ofMinutes(SecurityConfig.MULTI_CLIENTING_FACTION_SWITCH_COOLDOWN_MINUTES).toMillis();
			return lastCharOnlineTimeMillis.values().stream().noneMatch(t -> t > minLastOnline);
		}

		void enterWorld(Player player) {
			lastCharOnlineTimeMillis.put(player.getRace(), Long.MAX_VALUE);
		}

		void leaveWorld(Player player) {
			lastCharOnlineTimeMillis.put(player.getRace(), System.currentTimeMillis());
		}

		boolean wasPlayingOnSameIpOrMac(Race race, long minLastOnlineMillis, AionConnection con) {
			Long lastOnlineMillis = lastCharOnlineTimeMillis.get(race);
			return lastOnlineMillis != null && lastOnlineMillis > minLastOnlineMillis && hasAny(con.getIP(), con.getMacAddress());
		}
	}

	private record Identifiers(String ip, String mac) {}
}