87 lines
2.1 KiB
Java
87 lines
2.1 KiB
Java
package communication.udp;
|
|
|
|
import java.io.IOException;
|
|
import java.net.DatagramPacket;
|
|
import java.net.DatagramSocket;
|
|
import java.net.InetAddress;
|
|
import java.net.SocketException;
|
|
import main.Utilisateur;
|
|
import messages.*;
|
|
|
|
public class UDPServer extends Thread {
|
|
|
|
private DatagramSocket sockUDP;
|
|
private CommunicationUDP commUDP;
|
|
private byte[] buffer;
|
|
private boolean running;
|
|
|
|
public UDPServer(int port, CommunicationUDP commUDP) throws SocketException {
|
|
this.running = true;
|
|
this.commUDP = commUDP;
|
|
this.sockUDP = new DatagramSocket(port);
|
|
this.buffer = new byte[256];
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
while (this.running) {
|
|
|
|
try {
|
|
|
|
DatagramPacket inPacket = new DatagramPacket(buffer, buffer.length);
|
|
this.sockUDP.receive(inPacket);
|
|
String msgString = new String(inPacket.getData(), 0, inPacket.getLength());
|
|
Message msg = Message.stringToMessage(msgString);
|
|
|
|
if (inPacket.getAddress().equals(InetAddress.getLocalHost())) {
|
|
continue;
|
|
}
|
|
|
|
switch (msg.getTypeMessage()) {
|
|
case JE_SUIS_CONNECTE:
|
|
|
|
if (Utilisateur.getSelf() != null) {
|
|
|
|
this.commUDP.sendMessageInfoPseudo(inPacket.getAddress());
|
|
}
|
|
|
|
break;
|
|
|
|
case INFO_PSEUDO:
|
|
MessageSysteme m = (MessageSysteme) msg;
|
|
|
|
if (this.commUDP.containsUserFromID(m.getId())) {
|
|
this.commUDP.changePseudoUser(m.getId(), m.getPseudo(), inPacket.getAddress());
|
|
} else {
|
|
|
|
this.commUDP.addUser(m.getId(), m.getPseudo(), inPacket.getAddress());
|
|
System.out.println(m.getId() + ", " + m.getPseudo());
|
|
}
|
|
break;
|
|
|
|
case JE_SUIS_DECONNECTE:
|
|
MessageSysteme m2 = (MessageSysteme) msg;
|
|
this.commUDP.removeUser(m2.getId(), m2.getPseudo(), inPacket.getAddress());
|
|
break;
|
|
|
|
default: // Others types of messages are ignored because they are supposed to be
|
|
// transmitted by TCP and not UDP
|
|
}
|
|
|
|
} catch (IOException e) {
|
|
System.out.println("receive exception");
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void interrupt() {
|
|
// Stop the thread
|
|
this.running = false;
|
|
// Close the stream and the socket
|
|
this.sockUDP.close();
|
|
this.buffer = null;
|
|
this.commUDP = null;
|
|
}
|
|
}
|