import java.net.*; import java.io.*; import java.util.*; /** * @author Nicolás A. Ortega * @license GNU GPLv3 * @year 2014 * * For details on the copyright, look at the COPYRIGHT file that came with * this program. * */ public class Server implements Runnable { private String version = "v1.0"; private ArrayList clients = new ArrayList(); private ServerSocket sSocket = null; private Thread thread = null; private int clientCount = 0; private String passwd = "admin"; public static void main(String[] args) { if(args.length != 1) { System.out.println("You are using the program incorrectly."); } else { Server server = new Server(Integer.parseInt(args[0])); } } public Server(int port) { System.out.println("ConsoleChat server " + version + " Copyright (C) 2014 Nicolás A. Ortega\n" + "This program comes with ABSOLUTELY NO WARRANTY; details in WARRANTY file.\n" + "This is free software, and you are welcome to redistribute it\n" + "under certain conditions; details in LICENSE file.\n"); try { System.out.println("Binding to port " + port + "..."); // Try to open a port on the specified port number sSocket = new ServerSocket(port); System.out.println("Server started: " + sSocket); start(); } catch(IOException e) { System.out.println(e); } } public void run() { while(thread != null) { try { System.out.println("Waiting for clients..."); addThread(sSocket.accept()); } catch(IOException e) { System.out.println("Acceptance error: " + e); stop(); } } } public void start() { if(thread == null) { thread = new Thread(this); thread.start(); } } public void stop() { if(thread != null) { thread.interrupt(); thread = null; } for(int i = 0; i < clients.size(); i++) { try { clients.get(i).close(); } catch(IOException e) { System.out.println("Error closing thread: " + e); } } } public int findClient(int id) { for(int i = 0; i < clientCount; i++) { if(clients.get(i).getID() == id) { return i; } } return -1; } public synchronized void handle(int id, String username, String input) { if(input.startsWith("/")) { if(input.equals("/quit")) { clients.get(findClient(id)).send("/quit"); remove(id); } else if(input.equals("/list")) { int pos = findClient(id); for(int i = 0; i < clients.size(); i++) { String list = Integer.toString(clients.get(i).getID()) + " - " + clients.get(i).getUsername() + " |"; if(pos == i) { list += " (you)"; } if(clients.get(i).isAdmin()) { list += " (admin)"; } clients.get(pos).send(list); } clients.get(pos).send("\n"); } else if(input.substring(0, 3).equals("/pm")) { int pos = findClient(Integer.parseInt(input.substring(4, 9))); clients.get(pos).send("<" + username + ">: " + input.substring(10)); } else if(input.length() > 12 && input.substring(0, 5).equals("/kick")) { if(clients.get(findClient(id)).isAdmin()) { int kID = Integer.parseInt(input.substring(6, 11)); String reason = input.substring(12); clients.get(findClient(kID)).send("Kick: " + reason); remove(kID); } else { clients.get(findClient(id)).send("You are not admin."); } } else if(input.length() > 10 && input.substring(0, 10).equals("/giveadmin")) { if(clients.get(findClient(id)).isAdmin()) { int aID = Integer.parseInt(input.substring(11)); clients.get(findClient(aID)).setAdmin(true); clients.get(findClient(aID)).send("You have been made admin.\n"); } else { clients.get(findClient(id)).send("You are not admin."); } } else { clients.get(findClient(id)).runCommand(input); } } else { for(int i = 0; i < clientCount; i++) { if(clients.get(i).getID() != id) { clients.get(i).send(username + ": " + input); } } } } public synchronized void remove(int id) { int pos = findClient(id); if(pos >= 0) { ServerThread toTerminate = clients.get(pos); System.out.println("Remove client thread: " + id + " at " + pos); try { clients.get(pos).close(); } catch(IOException e) { System.out.println("Error closing thread: " + e); } clients.remove(pos); clientCount--; try { toTerminate.close(); } catch(IOException e) { System.out.println("Error closing thread: " + e); } toTerminate.interrupt(); } } public void addThread(Socket socket) { clients.add(new ServerThread(this, socket)); try { clients.get(clientCount).open(); clients.get(clientCount).start(); clientCount++; } catch(IOException e) { System.out.println("Error opening thread: " + e); } } public String getPasswd() { return passwd; } public void setPasswd(String npasswd) { this.passwd = npasswd; } }