CringeStudios/ShareServer
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow FULL_SYNC and CHECKSUM broadcast

Browse Source
This commit is contained in:
MrLetsplay 2023-12-06 21:05:11 +01:00
parent 8fa23d9dff
commit 46738ab51a
Signed by: mr
SSH Key Fingerprint: SHA256:92jBH80vpXyaZHjaIl47pjRq+Yt7XGTArqQg1V7hSqg
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/me/mrletsplay/shareserver/ShareWSServer.java
View File

@ -84,6 +84,7 @@ public class ShareWSServer extends WebSocketServer {
SessionUser user = conn.getAttachment(); SessionUser user = conn.getAttachment();
Session session = user.session(); Session session = user.session();
boolean isHost = getHost(getPeers(session)).equals(conn);
switch(m.getType()) { switch(m.getType()) {
case CHANGE -> getPeers(session).forEach(peer -> send(peer, m)); case CHANGE -> getPeers(session).forEach(peer -> send(peer, m));
case REQUEST_FULL_SYNC, REQUEST_CHECKSUM -> { case REQUEST_FULL_SYNC, REQUEST_CHECKSUM -> {
@ -96,9 +97,18 @@ public class ShareWSServer extends WebSocketServer {
send(getHost(getPeers(session)), m); send(getHost(getPeers(session)), m);
} }
case FULL_SYNC, CHECKSUM -> { case FULL_SYNC, CHECKSUM -> {
if(!isHost) {
conn.close(CloseFrame.POLICY_VALIDATION, "Only host can send FULL_SYNC or CHECKSUM");
return;
}
AddressableMessage msg = (AddressableMessage) m; AddressableMessage msg = (AddressableMessage) m;
WebSocket peer = getPeer(getPeers(session), msg.siteID()); if(msg.siteID() == AddressableMessage.BROADCAST_SITE_ID) {
if(peer != null) send(peer, m); getPeers(session).forEach(peer -> send(peer, m));
}else {
WebSocket peer = getPeer(getPeers(session), msg.siteID());
if(peer != null) send(peer, m);
}
} }
default -> { default -> {
conn.close(CloseFrame.POLICY_VALIDATION, "Invalid message received"); conn.close(CloseFrame.POLICY_VALIDATION, "Invalid message received");