L2LoginClient.java

/*
 * Copyright © 2004-2024 L2J Server
 * 
 * This file is part of L2J Server.
 * 
 * L2J Server is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * L2J Server is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package com.l2jserver.loginserver.network;

import java.io.IOException;
import java.net.InetAddress;
import java.nio.ByteBuffer;
import java.security.interfaces.RSAPrivateKey;
import java.util.HashMap;
import java.util.Map;

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

import com.l2jserver.commons.util.Rnd;
import com.l2jserver.loginserver.LoginController;
import com.l2jserver.loginserver.SessionKey;
import com.l2jserver.loginserver.network.serverpackets.L2LoginServerPacket;
import com.l2jserver.loginserver.network.serverpackets.LoginFail;
import com.l2jserver.loginserver.network.serverpackets.LoginFail.LoginFailReason;
import com.l2jserver.loginserver.network.serverpackets.PlayFail;
import com.l2jserver.loginserver.network.serverpackets.PlayFail.PlayFailReason;
import com.l2jserver.loginserver.security.ScrambledKeyPair;
import com.l2jserver.loginserver.security.crypt.LoginCrypt;
import com.l2jserver.mmocore.MMOClient;
import com.l2jserver.mmocore.MMOConnection;
import com.l2jserver.mmocore.SendablePacket;

/**
 * Represents a client connected into the LoginServer
 * @author KenM
 * @version 2.6.1.0
 */
public final class L2LoginClient extends MMOClient<MMOConnection<L2LoginClient>> {
	
	private static final Logger LOG = LoggerFactory.getLogger(L2LoginClient.class);
	
	public enum LoginClientState {
		CONNECTED,
		AUTHED_GG,
		AUTHED_LOGIN
	}
	
	private LoginClientState _state;
	
	private final LoginCrypt _loginCrypt;
	
	private final ScrambledKeyPair _scrambledPair;
	
	private final byte[] _blowfishKey;
	
	private String _account;
	
	private int _accessLevel;
	
	private int _lastServer;
	
	private SessionKey _sessionKey;
	
	private final int _sessionId;
	
	private boolean _joinedGS;
	
	private Map<Integer, Integer> _charsOnServers;
	
	private Map<Integer, long[]> _charsToDelete;
	
	private final long _connectionStartTime;
	
	public L2LoginClient(MMOConnection<L2LoginClient> con) {
		super(con);
		_state = LoginClientState.CONNECTED;
		_scrambledPair = LoginController.getInstance().getScrambledRSAKeyPair();
		_blowfishKey = LoginController.getInstance().getBlowfishKey();
		_sessionId = Rnd.nextInt();
		_connectionStartTime = System.currentTimeMillis();
		_loginCrypt = new LoginCrypt();
		_loginCrypt.setKey(_blowfishKey);
	}
	
	@Override
	public boolean decrypt(ByteBuffer buf, int size) {
		try {
			if (!_loginCrypt.decrypt(buf.array(), buf.position(), size)) {
				LOG.warn("Wrong checksum from client {}!", this);
				super.getConnection().close((SendablePacket<L2LoginClient>) null);
				return false;
			}
			return true;
		} catch (IOException ex) {
			LOG.warn("There has been an error decrypting message!", ex);
			super.getConnection().close((SendablePacket<L2LoginClient>) null);
			return false;
		}
	}
	
	@Override
	public boolean encrypt(ByteBuffer buf, int size) {
		final int offset = buf.position();
		try {
			size = _loginCrypt.encrypt(buf.array(), offset, size);
		} catch (IOException ex) {
			LOG.warn("There has been an error encrypting message!", ex);
			return false;
		}
		buf.position(offset + size);
		return true;
	}
	
	public LoginClientState getState() {
		return _state;
	}
	
	public void setState(LoginClientState state) {
		_state = state;
	}
	
	public byte[] getBlowfishKey() {
		return _blowfishKey;
	}
	
	public byte[] getScrambledModulus() {
		return _scrambledPair.getScrambledModulus();
	}
	
	public RSAPrivateKey getRSAPrivateKey() {
		return (RSAPrivateKey) _scrambledPair.getPair().getPrivate();
	}
	
	public String getAccount() {
		return _account;
	}
	
	public void setAccount(String account) {
		_account = account;
	}
	
	public void setAccessLevel(int accessLevel) {
		_accessLevel = accessLevel;
	}
	
	public int getAccessLevel() {
		return _accessLevel;
	}
	
	public void setLastServer(int lastServer) {
		_lastServer = lastServer;
	}
	
	public int getLastServer() {
		return _lastServer;
	}
	
	public int getSessionId() {
		return _sessionId;
	}
	
	public boolean hasJoinedGS() {
		return _joinedGS;
	}
	
	public void setJoinedGS(boolean val) {
		_joinedGS = val;
	}
	
	public void setSessionKey(SessionKey sessionKey) {
		_sessionKey = sessionKey;
	}
	
	public SessionKey getSessionKey() {
		return _sessionKey;
	}
	
	public long getConnectionStartTime() {
		return _connectionStartTime;
	}
	
	public void sendPacket(L2LoginServerPacket lsp) {
		getConnection().sendPacket(lsp);
	}
	
	public void close(LoginFailReason reason) {
		getConnection().close(new LoginFail(reason));
	}
	
	public void close(PlayFailReason reason) {
		getConnection().close(new PlayFail(reason));
	}
	
	public void close(L2LoginServerPacket lsp) {
		getConnection().close(lsp);
	}
	
	public void setCharsOnServ(int servId, int chars) {
		if (_charsOnServers == null) {
			_charsOnServers = new HashMap<>();
		}
		_charsOnServers.put(servId, chars);
	}
	
	public Map<Integer, Integer> getCharsOnServ() {
		return _charsOnServers;
	}
	
	public void serCharsWaitingDelOnServ(int servId, long[] charsToDel) {
		if (_charsToDelete == null) {
			_charsToDelete = new HashMap<>();
		}
		_charsToDelete.put(servId, charsToDel);
	}
	
	public Map<Integer, long[]> getCharsWaitingDelOnServ() {
		return _charsToDelete;
	}
	
	@Override
	public void onDisconnection() {
		if (!hasJoinedGS() || ((getConnectionStartTime() + LoginController.LOGIN_TIMEOUT) < System.currentTimeMillis())) {
			LoginController.getInstance().removeAuthedLoginClient(getAccount());
		}
	}
	
	@Override
	public String toString() {
		InetAddress address = getConnection().getInetAddress();
		if (getState() == LoginClientState.AUTHED_LOGIN) {
			return "[" + getAccount() + " (" + (address == null ? "disconnected" : address.getHostAddress()) + ")]";
		}
		return "[" + (address == null ? "disconnected" : address.getHostAddress()) + "]";
	}
	
	@Override
	protected void onForcedDisconnection() {
		// Empty
	}
}