package services; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.PrintStream; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.util.concurrent.Executors; import java.util.concurrent.TimeUnit; import java.util.concurrent.atomic.AtomicInteger; import network.packets.Packet; import network.packets.soe.DataChannelA; import network.packets.soe.MultiPacket; import network.packets.swg.SWGPacket; import network.packets.swg.zone.baselines.Baseline; import intents.InboundPacketIntent; import intents.OutboundPacketIntent; import intents.ServerManagementIntent; import intents.chat.ChatBroadcastIntent; import intents.chat.ChatBroadcastIntent.BroadcastType; import resources.Galaxy; import resources.Galaxy.GalaxyStatus; import resources.config.ConfigFile; import resources.control.Intent; import resources.control.Manager; import resources.services.Config; import services.galaxy.GalacticManager; public class CoreManager extends Manager { private static final int galaxyId = 1; private static final boolean debugOutput = false; private static PrintStream packetOutput; static { try { packetOutput = new PrintStream(new FileOutputStream("packets.txt", false)); } catch (FileNotFoundException e) { e.printStackTrace(); packetOutput = System.out; } } private EngineManager engineManager; private GalacticManager galacticManager; private Galaxy galaxy; private long startTime; private boolean shutdownRequested; public CoreManager() { shutdownRequested = false; galaxy = getGalaxy(); if (galaxy != null) { engineManager = new EngineManager(galaxy); galacticManager = new GalacticManager(galaxy); addChildService(engineManager); addChildService(galacticManager); } } /** * Determines whether or not the core is operational * @return TRUE if the core is operational, FALSE otherwise */ public boolean isOperational() { return true; } public boolean isShutdownRequested() { return shutdownRequested; } @Override public boolean initialize() { startTime = System.nanoTime(); registerForIntent(InboundPacketIntent.TYPE); registerForIntent(OutboundPacketIntent.TYPE); registerForIntent(ServerManagementIntent.TYPE); return galaxy != null && super.initialize(); } @Override public void onIntentReceived(Intent i) { if (debugOutput) { if (i instanceof InboundPacketIntent) { InboundPacketIntent in = (InboundPacketIntent) i; packetOutput.println("IN " + in.getNetworkId() + ":" + in.getServerType()); outputPacket(1, in.getPacket()); } else if (i instanceof OutboundPacketIntent) { OutboundPacketIntent out = (OutboundPacketIntent) i; packetOutput.println("OUT " + out.getNetworkId()); outputPacket(1, out.getPacket()); } } if (i instanceof ServerManagementIntent) handleServerManagementIntent((ServerManagementIntent) i); } private void handleServerManagementIntent(ServerManagementIntent i) { switch(i.getEvent()) { case SHUTDOWN: initiateShutdownSequence(true); break; default: break; } } private void initiateShutdownSequence(boolean notifyPlayers) { System.out.println("Beginning server shutdown sequence..."); if (notifyPlayers) createNotificationThread(); else shutdownRequested = true; } private void createNotificationThread() { final AtomicInteger runCount = new AtomicInteger(); Executors.newSingleThreadScheduledExecutor().scheduleAtFixedRate(new Runnable() { public void run() { switch(runCount.getAndIncrement()) { case 0: new ChatBroadcastIntent("The server will be shutting down in 15 minutes.", BroadcastType.GALAXY).broadcast(); System.out.println("ProjectSWG: Shutdown in 15 minutes."); break; case 1: new ChatBroadcastIntent("The server will be shutting down in 10 minutes.", BroadcastType.GALAXY).broadcast(); System.out.println("ProjectSWG: Shutdown in 10 minutes."); break; case 2: new ChatBroadcastIntent("The server will be shutting down in 5 minutes.", BroadcastType.GALAXY).broadcast(); System.out.println("ProjectSWG: Shutdown in 5 minutes."); break; case 3: new ChatBroadcastIntent("The server will be shutting down in 1 minute.", BroadcastType.GALAXY).broadcast(); System.out.println("ProjectSWG: Shutdown in 1 minute."); try { Thread.sleep(60000); } catch (InterruptedException e) { e.printStackTrace(); } shutdownRequested = true; throw new RuntimeException("Limit reached."); // no "clean" ways to cancel the runnable I can think of } } }, 0, 5, TimeUnit.MINUTES); } public GalaxyStatus getGalaxyStatus() { return galaxy.getStatus(); } private void outputPacket(int indent, Packet packet) { if (packet instanceof DataChannelA) { for (SWGPacket p : ((DataChannelA) packet).getPackets()) { for (int i = 0; i < indent; i++) packetOutput.print(" "); outputSWG(p); } } else if (packet instanceof MultiPacket) { for (Packet p : ((MultiPacket) packet).getPackets()) { for (int i = 0; i < indent; i++) packetOutput.print(" "); if (p instanceof SWGPacket) outputSWG((SWGPacket) p); if (p instanceof DataChannelA) outputPacket(indent+1, p); } } else if (packet instanceof SWGPacket) { for (int i = 0; i < indent; i++) packetOutput.print(" "); outputSWG((SWGPacket) packet); } else { for (int i = 0; i < indent; i++) packetOutput.print(" "); packetOutput.println(packet.getClass().getSimpleName()); } } private void outputSWG(SWGPacket p) { if (p instanceof Baseline) packetOutput.println("Baseline " + ((Baseline)p).getType() + " " + ((Baseline)p).getNum()); else packetOutput.println(p.getClass().getSimpleName()); } /** * Returns the time in milliseconds since the server started initialization * @return the core time represented as a double */ public double getCoreTime() { return (System.nanoTime()-startTime)/1E6; } private Galaxy getGalaxy() { PreparedStatement getGalaxy = getLocalDatabase().prepareStatement("SELECT * FROM galaxies WHERE id = ?"); Config c = getConfig(ConfigFile.PRIMARY); try { getGalaxy.setInt(1, galaxyId); ResultSet set = getGalaxy.executeQuery(); if (!set.next()) { System.err.println("CoreManager: No such galaxy exists with ID " + galaxyId + "!"); return null; } Galaxy g = new Galaxy(); g.setId(set.getInt("id")); g.setName(set.getString("name")); g.setAddress(set.getString("address")); g.setPopulation(set.getInt("population")); g.setTimeZone(set.getInt("timezone")); g.setZonePort(set.getInt("zone_port")); g.setPingPort(set.getInt("ping_port")); g.setStatus(set.getInt("status")); g.setMaxCharacters(c.getInt("GALAXY-MAX-CHARACTERS", 2)); g.setOnlinePlayerLimit(c.getInt("GALAXY-MAX-ONLINE", 3000)); g.setOnlineFreeTrialLimit(c.getInt("GALAXY-MAX-ONLINE", 3000)); g.setRecommended(true); return g; } catch (SQLException e) { e.printStackTrace(); } return null; } }