ZahraSadatMirvakili/HW-10-Socket-Programming
1
0
Fork 0
forked from AdvancedProgramming1404/HW-10-Socket-Programming
Code Pull Requests Activity

Merge PR 'Develop' (#1) from develop into main

Browse Source 
Full Mark
No Bonus
This commit was merged in pull request #1.
This commit is contained in:
MehrdadShirvani
2026-07-10 17:51:17 +00:00
parent 240eac0504 559f91fbf0
commit 1bbd59905c
4 changed files with 173 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/university/chat/Client/ServerListener.java
+28 -8
View File
@@ -1,18 +1,38 @@
package com.university.chat.Client;
import com.university.chat.Common.ChatMessage;
import com.university.chat.Common.FileMessage;
import com.university.chat.Common.MessageType;
import java.io.ObjectInputStream;
public class ServerListener implements Runnable{
// TODO: store the ObjectInputStream from the user socket
// (this should be the same input stream the
// chatClient created when connecting)
private final ObjectInputStream in;
public ServerListener(ObjectInputStream in) {
this.in = in;
}
@Override
public void run() {
try {
// TODO: In an infinite loop read objects from the server
// - if it's a ChatMessage -> print "<sender>: <content>"
// - if it's a FileMessage -> print that a file was received
// (filename + sender), it's already
// saved to disk by the server.
while (true){
Object msg = in.readObject();
if(msg instanceof ChatMessage chat){
if (chat.getType() == MessageType.LOGIN_SUCCESS) {
System.out.println(" Login successful!");
} else if (chat.getType() == MessageType.LOGIN_FAILED) {
System.out.println(" Login failed: " + chat.getContent());
} else if (chat.getSender().equals("Server")) {
System.out.println("[Server]: " + chat.getContent());
} else {
System.out.println(chat.getSender() + ": " + chat.getContent());
}
}else if(msg instanceof FileMessage file){
System.out.println("File received: " + file.getFilename() + " from " + file.getSender());
}
}
} catch (Exception e){
System.out.println("Disconnected from server");
}
src/main/java/com/university/chat/Client/chatClient.java
+61 -22
View File
@@ -1,29 +1,68 @@
package com.university.chat.Client;
public class chatClient {
public static void main() {
// TODO: Connecting to the server
// 1. Create a socket and connect to the server
// 2. Create an ObjectOutputStream (out) and ObjectInputStream (in)
// from the socket's streams — output FIRST, then input.
// 2. Get the username, and send a LOGIN ChatMessage with that username
// 3. Start a new Thread running a ServerListener(in) so incoming
// messages are handled concurrently.
import com.university.chat.Common.ChatMessage;
import com.university.chat.Common.FileMessage;
import com.university.chat.Common.MessageType;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.Socket;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Scanner;
while (true){
try {
// TODO: Program loop — read a line from the console and act on it:
// - "/msg <user> <text>" -> build & send a PRIVATE_MESSAGE
// - "/users" -> build & send a USER_LIST request
// - "/sendfile <user> <path>" -> read the file into a byte[]
// (you can use TransferProgress
// to show progress)
// and send it as a FileMessage
// - anything else -> send a PUBLIC_MESSAGE
// Remember to flush() the output stream after writeObject().
} catch (Exception e){
System.out.println("command failed: " + e.getMessage());
public class chatClient {
public static void main(String[] args) {
try (Socket socket = new Socket("localhost", 12345)) {
Scanner scanner = new Scanner(System.in);
System.out.println("Connected to server");
ObjectOutputStream out = new ObjectOutputStream(socket.getOutputStream());
out.flush();
ObjectInputStream in = new ObjectInputStream(socket.getInputStream());
System.out.println("Enter username: ");
String username = scanner.nextLine().trim();
Thread listenerThread = new Thread(new ServerListener(in));
listenerThread.setDaemon(true);
listenerThread.start();
out.writeObject(new ChatMessage(MessageType.LOGIN, username, null, username));
out.flush();
Thread.sleep(300);
System.out.println("Login successful " + username);
while (true) {
String line = scanner.nextLine();
if (line.startsWith("/msg ")){
String [] parts = line.split(" ", 3);
if (parts.length == 3){
out.writeObject(new ChatMessage(MessageType.PRIVATE_MESSAGE, username, parts[1], parts[2]));
out.flush();
}
}else if (line.equals("/users")){
out.writeObject(new ChatMessage(MessageType.USER_LIST, username, null, null));
out.flush();
}else if (line.startsWith("/sendfile ")){
String [] parts = line.split(" ", 3);
if (parts.length == 3){
byte[] data = Files.readAllBytes(Paths.get(parts[2]));
out.writeObject(new FileMessage(username, parts[1], Paths.get(parts[2]).getFileName().toString(), data));
out.flush();
}
}else{
out.writeObject(new ChatMessage(MessageType.PUBLIC_MESSAGE, username, null, line));
out.flush();
}
}
} catch (Exception e){
System.out.println("command failed: " + e.getMessage());
}
}
}
src/main/java/com/university/chat/Server/ChatServer.java
+13 -8
View File
@@ -1,15 +1,20 @@
package com.university.chat.Server;
import java.net.ServerSocket;
import java.net.Socket;
public class ChatServer {
// TODO: declare a single shared UserManager instance (static final)
// This MUST be shared by all ClientSession threads so that
// broadcasting and private messaging work correctly.
public static final UserManager userManager = new UserManager();
public static void main(String[] args) {
// TODO: Create a ServerSocket
// TODO: In an infinite loop:
// accept an incoming client connection
// make a new thread running ClientSession for each user.
try(ServerSocket serverSocket = new ServerSocket(12345)){
System.out.println("Server started on port 12345");
while (true){
Socket clientSocket = serverSocket.accept();
new Thread(new ClientSession(clientSocket, userManager)).start();
}
}catch (Exception e){
e.printStackTrace();
}
}
}
src/main/java/com/university/chat/Server/ClientSession.java
+71 -26
View File
@@ -2,70 +2,115 @@ package com.university.chat.Server;
import com.university.chat.Common.ChatMessage;
import com.university.chat.Common.FileMessage;
import com.university.chat.Common.MessageType;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.Socket;
import java.nio.file.Files;
public class ClientSession implements Runnable {
private String username;
private final Socket socket;
private final UserManager userManager;
private ObjectOutputStream out;
private ObjectInputStream in;
private boolean isLogeIn = false;
public ClientSession(Socket socket, UserManager userManager) {
// TODO : Create an ObjectOutputStream from socket.getOutputStream()
// and an ObjectInputStream from socket.getInputStream().
this.socket = socket;
this.userManager = userManager;
try{
this.out = new ObjectOutputStream(socket.getOutputStream());
this.in = new ObjectInputStream(socket.getInputStream());
}catch (IOException e){
e.printStackTrace();
}
}
@Override
public void run() {
try {
ChatMessage loginMsg = (ChatMessage) in.readObject();
if(loginMsg.getType() == MessageType.LOGIN){
this.username = loginMsg.getContent();
if (userManager.addUser(username, this)){
FileManager.createUserFolders(username);
out.writeObject(new ChatMessage(MessageType.LOGIN_SUCCESS, "Server", null, "Welcome"));
out.flush();
isLogeIn = true;
System.out.println("User logged in: " + username);
}else{
out.writeObject(new ChatMessage(MessageType.LOGIN_FAILED, "Server", null, "User taken"));
out.flush();
socket.close();
return;
}
}
// TODO: Welcome the user (login step)
// 1. Read the first object sent by the client.
// 2. Check it's a ChatMessage with type LOGIN.
// 3. Extract the username.
// 4. Try to register the user via userManager.addUser(...).
// 5. If the username is taken, send back LOGIN_FAILED and close the socket.
// 6. Otherwise, create the user's folders with FileManager.createUserFolders(...)
// and send back LOGIN_SUCCESS.
// TODO: Main message loop
// In a loop, call in.readObject(), you can separate messages by their type:
// - if it's a ChatMessage -> call handleChatMessage(msg)
// - if it's a FileMessage -> call handleFileMessage(fileMsg)
// Keep looping until the connection is closed (an exception will be thrown).
while (true) {
Object obj = in.readObject();
if (obj instanceof ChatMessage) handleChatMessage((ChatMessage) obj);
else if (obj instanceof FileMessage) handleFileMessage((FileMessage) obj);
}
} catch (Exception e) {
System.out.println("Disconnected: " + username);
} finally {
// TODO: Remove the user from UserManager so they no longer
// receive broadcasts or appear in users list
}
if (username != null && userManager.getUser(username) == this){
userManager.removeUser(username);
System.out.println("User removed: " + username);
}
try {
if (socket != null && socket.isClosed()){
socket.close();
}
} catch (IOException e) {}
}
}
private void handleChatMessage(ChatMessage msg) throws IOException {
switch (msg.getType()) {
case PUBLIC_MESSAGE -> {
// TODO: Broadcast this message to every connected client.
System.out.println("Broadcasting from " + msg.getSender());
for (ClientSession s : userManager.getAllSessions()){
if(s != this && s.isLogeIn) s.sendMessage(msg);
}
}
case PRIVATE_MESSAGE -> {
// TODO: Forward this message to the receiver user.
System.out.println("Private message from " + msg.getSender() + "to " + msg.getReceiver());
ClientSession target = userManager.getUser(msg.getReceiver());
if(target != null && target != this && target.isLogeIn) {
target.sendMessage(msg);
}else if(target == this){
sendMessage(new ChatMessage(MessageType.PRIVATE_MESSAGE, "Server" , username, "You cannot message yourself"));
}else {
sendMessage(new ChatMessage(MessageType.PRIVATE_MESSAGE, "Server" , username, "User not found: " +msg.getReceiver()));
}
}
case USER_LIST -> {
// TODO: Reply to the requester with the list of online users.
String usersList = String.join(", ", userManager.listUsers());
sendMessage(new ChatMessage(MessageType.USER_LIST, "Server", username, usersList));
}
}
}
private void handleFileMessage(FileMessage fileMsg) throws IOException {
// Storing the file
var sentPath = FileManager.getSentPath(fileMsg.getSender(), fileMsg.getFilename());
var recvPath = FileManager.getReceivedPath(fileMsg.getReceiver(), fileMsg.getFilename());
Files.write(sentPath, fileMsg.getData());
Files.write(recvPath, fileMsg.getData());
// TODO: Forward the received file-message to the destination user.
ClientSession target = userManager.getUser(fileMsg.getReceiver());
if (target != null && target.isLogeIn) target.sendMessage(fileMsg);
}
public void sendMessage(Object msg) throws IOException{
if (out != null){
out.writeObject(msg);
out.flush();
}
}
}