use of org.spongepowered.api.network.channel.packet.RequestPacket in project Sponge by SpongePowered.
the class SpongePacketChannel method handleTransactionResponse.
@Override
protected void handleTransactionResponse(final EngineConnection connection, final Object stored, final TransactionResult result) {
if (result.isSuccess()) {
final ChannelBuf payload = result.getPayload();
final long typeAndValue = payload.readVarLong();
final int type = SpongePacketChannel.extractType(typeAndValue);
final int value = SpongePacketChannel.extractValue(typeAndValue);
if (type == SpongePacketChannel.TYPE_RESPONSE || type == SpongePacketChannel.TYPE_NO_RESPONSE || type == SpongePacketChannel.TYPE_DYNAMIC_RESPONSE) {
final TransactionData<RequestPacket<Packet>, Packet> transactionData = (TransactionData<RequestPacket<Packet>, Packet>) stored;
if (type == SpongePacketChannel.TYPE_RESPONSE) {
this.handleTransactionResponse(connection, transactionData, result, SpongePacketChannel.NO_DYNAMIC_OPCODE);
} else if (type == SpongePacketChannel.TYPE_NO_RESPONSE) {
this.handleTransactionResponse(connection, transactionData, TransactionResult.failure(new NoResponseException()), SpongePacketChannel.NO_DYNAMIC_OPCODE);
} else {
this.handleTransactionResponse(connection, transactionData, result, value);
}
} else {
this.handleException(connection, new ChannelIOException("Unknown packet type: " + type), null);
}
}
}
use of org.spongepowered.api.network.channel.packet.RequestPacket in project SpongeCommon by SpongePowered.
the class SpongeBasicPacketChannel method handleLoginRequestPayload.
@Override
protected void handleLoginRequestPayload(final EngineConnection connection, final int transactionId, final ChannelBuf payload) {
// Is currently always executed on the client,
// the server always expects a response
final int opcode = this.readOpcode(payload);
final SpongePacketBinding<Packet> binding = this.requireBinding(opcode);
final Packet packet = this.decodePayload(() -> binding.getPacketConstructor().get(), payload);
// A normal packet binding
if (binding instanceof SpongeHandlerPacketBinding) {
final SpongeHandlerPacketBinding<Packet> handlerBinding = (SpongeHandlerPacketBinding<Packet>) binding;
this.handle(connection, handlerBinding, packet);
// The server always expects a response
PacketSender.sendTo(connection, PacketUtil.createLoginPayloadResponse(null, transactionId));
} else {
// A transactional packet binding
final SpongeTransactionalPacketBinding<RequestPacket<Packet>, Packet> transactionalBinding = (SpongeTransactionalPacketBinding) binding;
final RequestPacketHandler<Packet, Packet, EngineConnection> handler = (RequestPacketHandler<Packet, Packet, EngineConnection>) transactionalBinding.getRequestHandler(connection);
boolean success = false;
if (handler != null) {
final SpongeRequestPacketResponse<Packet> response = new SpongeRequestPacketResponse<Packet>() {
@Override
protected void fail0(final ChannelException exception) {
final net.minecraft.network.protocol.Packet<?> mcPacket = PacketUtil.createLoginPayloadResponse(null, transactionId);
PacketSender.sendTo(connection, mcPacket);
}
@Override
protected void success0(final Packet response) {
try {
final ChannelBuf responsePayload = SpongeBasicPacketChannel.this.encodeLoginPayload(transactionalBinding.opcode(), response);
final net.minecraft.network.protocol.Packet<?> mcPacket = PacketUtil.createLoginPayloadResponse(responsePayload, transactionId);
PacketSender.sendTo(connection, mcPacket);
} catch (final Throwable ex) {
SpongeBasicPacketChannel.this.handleException(connection, new ChannelIOException("Failed to encode response packet", ex), null);
}
}
};
try {
handler.handleRequest(packet, connection, response);
success = true;
} catch (final Throwable ex) {
this.handleException(connection, new ChannelIOException("Failed to handle request packet", ex), null);
}
}
if (!success) {
final net.minecraft.network.protocol.Packet<?> mcPacket = PacketUtil.createLoginPayloadResponse(null, transactionId);
PacketSender.sendTo(connection, mcPacket);
}
}
}
use of org.spongepowered.api.network.channel.packet.RequestPacket in project SpongeCommon by SpongePowered.
the class SpongePacketChannel method sendRequestPacketTo.
private <P extends RequestPacket<R>, R extends Packet> void sendRequestPacketTo(final EngineConnection connection, final P packet, final CompletableFuture<?> future, @Nullable final Consumer<R> response, @Nullable final Runnable sendSuccess) {
final SpongeTransactionalPacketBinding<P, R> binding = (SpongeTransactionalPacketBinding) this.requireBinding(packet.getClass());
final TransactionStore transactionStore = ConnectionUtil.getTransactionStore(connection);
final int transactionId = transactionStore.nextId();
final boolean isLoginPhase = ConnectionUtil.isLoginPhase(connection);
final EngineConnectionSide<?> side = connection.side();
final ChannelBuf payload = this.manager().getBufferAllocator().buffer();
final Supplier<net.minecraft.network.protocol.Packet<?>> mcPacketSupplier;
if (isLoginPhase) {
if (side == EngineConnectionSide.CLIENT) {
payload.writeString(this.key().formatted());
payload.writeVarLong(SpongePacketChannel.packTypeAndValue(SpongePacketChannel.TYPE_REQUEST, transactionId));
payload.writeVarInt(binding.opcode());
mcPacketSupplier = () -> PacketUtil.createLoginPayloadResponse(payload, Constants.Channels.LOGIN_PAYLOAD_TRANSACTION_ID);
} else {
payload.writeVarLong(SpongePacketChannel.packTypeAndValue(SpongePacketChannel.TYPE_REQUEST, binding.opcode()));
mcPacketSupplier = () -> PacketUtil.createLoginPayloadRequest(this.key(), payload, transactionId);
}
} else {
payload.writeVarLong(SpongePacketChannel.packTypeAndValue(SpongePacketChannel.TYPE_REQUEST, transactionId));
payload.writeVarInt(binding.opcode());
mcPacketSupplier = () -> PacketUtil.createPlayPayload(this.key(), payload, side);
}
try {
this.encodePayload(payload, packet);
} catch (final Throwable ex) {
this.handleException(connection, ex, future);
return;
}
if (response != null) {
final TransactionData<P, R> transactionData = new TransactionData<>(packet, binding, response, future);
transactionStore.put(transactionId, this, transactionData);
}
final net.minecraft.network.protocol.Packet<?> mcPacket = mcPacketSupplier.get();
PacketSender.sendTo(connection, mcPacket, sendFuture -> {
if (!sendFuture.isSuccess()) {
this.handleException(connection, sendFuture.cause(), future);
// and remove it from the store
if (response != null) {
transactionStore.remove(transactionId);
}
} else if (sendSuccess != null) {
sendSuccess.run();
}
});
}
use of org.spongepowered.api.network.channel.packet.RequestPacket in project SpongeCommon by SpongePowered.
the class SpongePacketChannel method handleTransactionResponse.
private <P extends RequestPacket<R>, R extends Packet> void handleTransactionResponse(final EngineConnection connection, final TransactionData<P, R> transactionData, final TransactionResult result, final int dynamicOpcode) {
if (result.isSuccess()) {
final ChannelBuf payload = result.getPayload();
SpongePacketBinding<R> responseBinding = null;
final Supplier<R> packetSupplier;
if (dynamicOpcode != SpongePacketChannel.NO_DYNAMIC_OPCODE) {
responseBinding = (SpongePacketBinding<R>) this.requireBinding(dynamicOpcode);
packetSupplier = responseBinding.getPacketConstructor();
} else if (transactionData.binding instanceof SpongeFixedTransactionalPacketBinding) {
packetSupplier = (Supplier<R>) ((SpongeFixedTransactionalPacketBinding<RequestPacket<Packet>, Packet>) transactionData.binding).getResponsePacketConstructor();
} else {
throw new ChannelException("A fixed response was send but no fixed response was bound to the request: " + transactionData.request.getClass());
}
final R packet;
try {
packet = this.decodePayload(packetSupplier, payload);
} catch (final Throwable ex) {
this.handleException(connection, new ChannelIOException("Failed to decode packet", ex), transactionData.future);
return;
}
if (responseBinding != null) {
this.handle(connection, (SpongeHandlerPacketBinding<Packet>) responseBinding, packet);
} else {
this.handleResponse(connection, transactionData.binding, transactionData.request, packet);
}
if (transactionData.success != null) {
transactionData.success.accept(packet);
}
} else {
this.handleException(connection, result.getCause(), transactionData.future);
this.handleResponseFailure(connection, transactionData.binding, transactionData.request, result.getCause());
}
}
use of org.spongepowered.api.network.channel.packet.RequestPacket in project Sponge by SpongePowered.
the class SpongeBasicPacketChannel method handleLoginRequestPayload.
@Override
protected void handleLoginRequestPayload(final EngineConnection connection, final int transactionId, final ChannelBuf payload) {
// Is currently always executed on the client,
// the server always expects a response
final int opcode = this.readOpcode(payload);
final SpongePacketBinding<Packet> binding = this.requireBinding(opcode);
final Packet packet = this.decodePayload(() -> binding.getPacketConstructor().get(), payload);
// A normal packet binding
if (binding instanceof SpongeHandlerPacketBinding) {
final SpongeHandlerPacketBinding<Packet> handlerBinding = (SpongeHandlerPacketBinding<Packet>) binding;
this.handle(connection, handlerBinding, packet);
// The server always expects a response
PacketSender.sendTo(connection, PacketUtil.createLoginPayloadResponse(null, transactionId));
} else {
// A transactional packet binding
final SpongeTransactionalPacketBinding<RequestPacket<Packet>, Packet> transactionalBinding = (SpongeTransactionalPacketBinding) binding;
final RequestPacketHandler<Packet, Packet, EngineConnection> handler = (RequestPacketHandler<Packet, Packet, EngineConnection>) transactionalBinding.getRequestHandler(connection);
boolean success = false;
if (handler != null) {
final SpongeRequestPacketResponse<Packet> response = new SpongeRequestPacketResponse<Packet>() {
@Override
protected void fail0(final ChannelException exception) {
final net.minecraft.network.protocol.Packet<?> mcPacket = PacketUtil.createLoginPayloadResponse(null, transactionId);
PacketSender.sendTo(connection, mcPacket);
}
@Override
protected void success0(final Packet response) {
try {
final ChannelBuf responsePayload = SpongeBasicPacketChannel.this.encodeLoginPayload(transactionalBinding.opcode(), response);
final net.minecraft.network.protocol.Packet<?> mcPacket = PacketUtil.createLoginPayloadResponse(responsePayload, transactionId);
PacketSender.sendTo(connection, mcPacket);
} catch (final Throwable ex) {
SpongeBasicPacketChannel.this.handleException(connection, new ChannelIOException("Failed to encode response packet", ex), null);
}
}
};
try {
handler.handleRequest(packet, connection, response);
success = true;
} catch (final Throwable ex) {
this.handleException(connection, new ChannelIOException("Failed to handle request packet", ex), null);
}
}
if (!success) {
final net.minecraft.network.protocol.Packet<?> mcPacket = PacketUtil.createLoginPayloadResponse(null, transactionId);
PacketSender.sendTo(connection, mcPacket);
}
}
}
Aggregations