86 lines
3.7 KiB
Java
86 lines
3.7 KiB
Java
package net.minecraft.network.chat;
|
|
|
|
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
|
|
import it.unimi.dsi.fastutil.objects.ObjectList;
|
|
import javax.annotation.Nullable;
|
|
|
|
public class LastSeenMessagesValidator {
|
|
private final int lastSeenCount;
|
|
private final ObjectList<LastSeenTrackedEntry> trackedMessages = new ObjectArrayList<>();
|
|
@Nullable
|
|
private MessageSignature lastPendingMessage;
|
|
|
|
public LastSeenMessagesValidator(int p_249951_) {
|
|
this.lastSeenCount = p_249951_;
|
|
|
|
for (int i = 0; i < p_249951_; i++) {
|
|
this.trackedMessages.add(null);
|
|
}
|
|
}
|
|
|
|
public void addPending(MessageSignature p_248841_) {
|
|
if (!p_248841_.equals(this.lastPendingMessage)) {
|
|
this.trackedMessages.add(new LastSeenTrackedEntry(p_248841_, true));
|
|
this.lastPendingMessage = p_248841_;
|
|
}
|
|
}
|
|
|
|
public int trackedMessagesCount() {
|
|
return this.trackedMessages.size();
|
|
}
|
|
|
|
public void applyOffset(int p_251273_) throws LastSeenMessagesValidator.ValidationException {
|
|
int i = this.trackedMessages.size() - this.lastSeenCount;
|
|
if (p_251273_ >= 0 && p_251273_ <= i) {
|
|
this.trackedMessages.removeElements(0, p_251273_);
|
|
} else {
|
|
throw new LastSeenMessagesValidator.ValidationException("Advanced last seen window by " + p_251273_ + " messages, but expected at most " + i);
|
|
}
|
|
}
|
|
|
|
public LastSeenMessages applyUpdate(LastSeenMessages.Update p_248868_) throws LastSeenMessagesValidator.ValidationException {
|
|
this.applyOffset(p_248868_.offset());
|
|
ObjectList<MessageSignature> objectlist = new ObjectArrayList<>(p_248868_.acknowledged().cardinality());
|
|
if (p_248868_.acknowledged().length() > this.lastSeenCount) {
|
|
throw new LastSeenMessagesValidator.ValidationException(
|
|
"Last seen update contained " + p_248868_.acknowledged().length() + " messages, but maximum window size is " + this.lastSeenCount
|
|
);
|
|
} else {
|
|
for (int i = 0; i < this.lastSeenCount; i++) {
|
|
boolean flag = p_248868_.acknowledged().get(i);
|
|
LastSeenTrackedEntry lastseentrackedentry = this.trackedMessages.get(i);
|
|
if (flag) {
|
|
if (lastseentrackedentry == null) {
|
|
throw new LastSeenMessagesValidator.ValidationException(
|
|
"Last seen update acknowledged unknown or previously ignored message at index " + i
|
|
);
|
|
}
|
|
|
|
this.trackedMessages.set(i, lastseentrackedentry.acknowledge());
|
|
objectlist.add(lastseentrackedentry.signature());
|
|
} else {
|
|
if (lastseentrackedentry != null && !lastseentrackedentry.pending()) {
|
|
throw new LastSeenMessagesValidator.ValidationException(
|
|
"Last seen update ignored previously acknowledged message at index " + i + " and signature " + lastseentrackedentry.signature()
|
|
);
|
|
}
|
|
|
|
this.trackedMessages.set(i, null);
|
|
}
|
|
}
|
|
|
|
LastSeenMessages lastseenmessages = new LastSeenMessages(objectlist);
|
|
if (!p_248868_.verifyChecksum(lastseenmessages)) {
|
|
throw new LastSeenMessagesValidator.ValidationException("Checksum mismatch on last seen update: the client and server must have desynced");
|
|
} else {
|
|
return lastseenmessages;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static class ValidationException extends Exception {
|
|
public ValidationException(String p_394588_) {
|
|
super(p_394588_);
|
|
}
|
|
}
|
|
} |