コード例 #1
0
 @Override
 public void handle(NetworkContext context, MessageForgeHandshakeInOutAck message) {
   Session session = context.getSession();
   Attribute<ForgeServerHandshakePhase> phase =
       context.getChannel().attr(ForgeHandshakePhase.PHASE);
   switch (phase.get()) {
     case WAITING_ACK:
       if (!message.getPhase().equals(ForgeClientHandshakePhase.WAITING_SERVER_DATA)) {
         session.disconnect(
             "Retrieved unexpected forge handshake ack message. (Got "
                 + message.getPhase()
                 + ", expected "
                 + ForgeClientHandshakePhase.WAITING_SERVER_DATA
                 + ")");
       } else {
         List<MessageForgeHandshakeOutRegistryData.Entry> entries = Lists.newArrayList();
         entries.add(
             new MessageForgeHandshakeOutRegistryData.Entry(
                 "fml:items", Maps.newHashMap(), Lists.newArrayList()));
         entries.add(
             new MessageForgeHandshakeOutRegistryData.Entry(
                 "fml:blocks", Maps.newHashMap(), Lists.newArrayList()));
         session.send(new MessageForgeHandshakeOutRegistryData(entries));
         session.send(new MessageForgeHandshakeInOutAck(ForgeServerHandshakePhase.WAITING_ACK));
         phase.set(ForgeServerHandshakePhase.COMPLETE);
       }
       LanternGame.log()
           .info(
               "{}: Forge handshake -> Received ack (waitingServerData) message.",
               session.getGameProfile().getName());
       break;
     case COMPLETE:
       if (!message.getPhase().equals(ForgeClientHandshakePhase.WAITING_SERVER_COMPLETE)) {
         session.disconnect(
             "Retrieved unexpected forge handshake ack message. (Got "
                 + message.getPhase()
                 + ", expected "
                 + ForgeClientHandshakePhase.WAITING_SERVER_COMPLETE
                 + ")");
       } else {
         session.send(new MessageForgeHandshakeInOutAck(ForgeServerHandshakePhase.COMPLETE));
         phase.set(ForgeServerHandshakePhase.DONE);
       }
       LanternGame.log()
           .info(
               "{}: Forge handshake -> Received ack (waitingServerComplete) message.",
               session.getGameProfile().getName());
       break;
     case DONE:
       if (!message.getPhase().equals(ForgeClientHandshakePhase.PENDING_COMPLETE)
           && !message.getPhase().equals(ForgeClientHandshakePhase.COMPLETE)) {
         session.disconnect(
             "Retrieved unexpected forge handshake ack message. (Got "
                 + message.getPhase()
                 + ", expected "
                 + ForgeClientHandshakePhase.PENDING_COMPLETE
                 + " or "
                 + ForgeClientHandshakePhase.COMPLETE
                 + ")");
       } else {
         if (message.getPhase().equals(ForgeClientHandshakePhase.PENDING_COMPLETE)) {
           session.send(new MessageForgeHandshakeInOutAck(ForgeServerHandshakePhase.DONE));
           LanternGame.log()
               .info(
                   "{}: Forge handshake -> Received ack (pendingComplete) message.",
                   session.getGameProfile().getName());
         } else {
           session.setProtocolState(ProtocolState.PLAY);
           session.spawnPlayer();
           LanternGame.log()
               .info(
                   "{}: Forge handshake -> Received ack (complete) message.",
                   session.getGameProfile().getName());
         }
       }
       break;
     case ERROR:
       break;
     default:
       session.disconnect(
           "Retrieved unexpected forge handshake ack message. (Got " + message.getPhase() + ")");
   }
 }
コード例 #2
0
  @Override
  public void handle(NetworkContext context, MessageLoginInEncryptionResponse message) {
    Session session = context.getSession();
    PrivateKey privateKey = session.getServer().getKeyPair().getPrivate();

    // Create rsaCipher
    Cipher rsaCipher;
    try {
      rsaCipher = Cipher.getInstance("RSA");
    } catch (GeneralSecurityException e) {
      LanternGame.log().error("Could not initialize RSA cipher", e);
      session.disconnect("Unable to initialize RSA cipher.");
      return;
    }

    // Decrypt shared secret
    SecretKey sharedSecret;
    try {
      rsaCipher.init(Cipher.DECRYPT_MODE, privateKey);
      sharedSecret = new SecretKeySpec(rsaCipher.doFinal(message.getSharedSecret()), "AES");
    } catch (Exception e) {
      LanternGame.log().warn("Could not decrypt shared secret", e);
      session.disconnect("Unable to decrypt shared secret.");
      return;
    }

    // Decrypt verify token
    byte[] verifyToken;
    try {
      rsaCipher.init(Cipher.DECRYPT_MODE, privateKey);
      verifyToken = rsaCipher.doFinal(message.getVerifyToken());
    } catch (Exception e) {
      LanternGame.log().warn("Could not decrypt verify token", e);
      session.disconnect("Unable to decrypt verify token.");
      return;
    }

    // Check verify token
    if (!Arrays.equals(verifyToken, session.getVerifyToken())) {
      session.disconnect("Invalid verify token.");
      return;
    }

    // Initialize stream encryption
    session.setEncryption(sharedSecret);

    // Create hash for auth
    String hash;
    try {
      MessageDigest digest = MessageDigest.getInstance("SHA-1");
      String sessionId = context.getChannel().attr(HandlerLoginStart.SESSION_ID).getAndRemove();

      digest.update(sessionId.getBytes());
      digest.update(sharedSecret.getEncoded());
      digest.update(session.getServer().getKeyPair().getPublic().getEncoded());

      // BigInteger takes care of sign and leading zeroes
      hash = new BigInteger(digest.digest()).toString(16);
    } catch (NoSuchAlgorithmException e) {
      LanternGame.log().error("Unable to generate SHA-1 digest", e);
      session.disconnect("Failed to hash login data.");
      return;
    }

    // Start auth thread
    Thread clientAuthThread =
        new Thread(new ClientAuthRunnable(session, session.getVerifyUsername(), hash));
    clientAuthThread.setName("ClientAuth{" + session.getVerifyUsername() + "}");
    clientAuthThread.start();
  }