Uploaded sockets package

Author: chen-robert

src/sockets/AdvancedConnection.java

@@ -0,0 +1,74 @@
+package sockets;
+
+import java.io.PrintStream;
+import java.net.InetAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.util.Scanner;
+
+public class AdvancedConnection {
+    static final int PORT = 8080;
+    static Scanner in;
+    public static void main(String[] args) throws Exception {
+        InetAddress inetAddress = InetAddress.getLocalHost();
+        System.out.println("IP Address: " + inetAddress.getHostAddress());
+        System.out.println("Host Name: " + inetAddress.getHostName());
+
+        System.out.println("What do you want to do");
+        System.out.println("1. Host a server");
+        System.out.println("2. Connect to a server");
+
+        in = new Scanner(System.in);
+
+        switch(in.nextInt()){
+            case 1:
+                hostServer();
+                break;
+            case 2:
+                System.out.print("Give me an ip to connect to: ");
+                connect(in.next());
+                break;
+            default:
+                System.out.println("Not an option :/");
+                break;
+        }
+    }
+
+    public static void hostServer() throws Exception {
+        InetAddress inetAddress = InetAddress.getLocalHost();
+        System.out.println("Started a new server on "+ inetAddress.getHostAddress());
+        ServerSocket ss = new ServerSocket(PORT);
+
+        Socket s = ss.accept();
+
+        System.out.println("Connection from " + s.getRemoteSocketAddress());
+        System.out.println();
+
+        PrintStream sout = new PrintStream(s.getOutputStream());
+
+        // Clear the newline from the previous call to in.nextInt();
+        in.nextLine();
+        while(true){
+            System.out.print("> ");
+            String line = in.nextLine();
+
+            sout.println(line);
+        }
+
+    }
+
+    public static void connect(String ip) throws Exception{
+        Socket s = new Socket(ip, PORT);
+
+        System.out.println("Connected to " + s.getRemoteSocketAddress());
+        System.out.println();
+
+        Scanner sin = new Scanner(s.getInputStream());
+        while(true){
+
+            String line = sin.nextLine();
+
+            System.out.println("Server > " + line);
+        }
+    }
+}

src/sockets/BasicConnection.java

@@ -0,0 +1,50 @@
+package sockets;
+
+import java.net.InetAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.util.Scanner;
+
+public class BasicConnection {
+    static final int PORT = 8080;
+    public static void main(String[] args) throws Exception {
+        InetAddress inetAddress = InetAddress.getLocalHost();
+        System.out.println("IP Address: " + inetAddress.getHostAddress());
+        System.out.println("Host Name: " + inetAddress.getHostName());
+
+        System.out.println("What do you want to do");
+        System.out.println("1. Host a server");
+        System.out.println("2. Connect to a server");
+
+        Scanner in = new Scanner(System.in);
+
+        switch(in.nextInt()){
+            case 1:
+                hostServer();
+                break;
+            case 2:
+                System.out.println("Give me an ip to connect to: ");
+                connect(in.next());
+                break;
+            default:
+                System.out.println("Not an option :/");
+                break;
+        }
+    }
+
+    public static void hostServer() throws Exception {
+        InetAddress inetAddress = InetAddress.getLocalHost();
+        System.out.println("Started a new server on "+ inetAddress.getHostAddress());
+        ServerSocket ss = new ServerSocket(PORT);
+
+        Socket s = ss.accept();
+
+        System.out.println("Connection from " + s.getRemoteSocketAddress());
+    }
+
+    public static void connect(String ip) throws Exception{
+        Socket s = new Socket(ip, PORT);
+
+        System.out.println("Connected to " + s.getRemoteSocketAddress());
+    }
+}

src/sockets/Challenge.java

@@ -0,0 +1,93 @@
+package sockets;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.io.PrintStream;
+import java.net.InetAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.util.Random;
+import java.util.Scanner;
+
+public class Challenge {
+    static final int PORT = 8080;
+    static Random rand = new Random();
+    static final int superSecureSecret = getSecureSecret(18);
+
+    public static void main(String[] args) throws Exception{
+        InetAddress inetAddress = InetAddress.getLocalHost();
+        System.out.println("Started a new server on "+ inetAddress.getHostAddress());
+        ServerSocket ss = new ServerSocket(PORT);
+
+        while(true){
+            Socket s = ss.accept();
+
+            new Thread(() -> process(s)).start();
+        }
+    }
+
+    public static void process(Socket s) {
+        try{
+            Scanner sin = new Scanner(s.getInputStream());
+            PrintStream sout = new PrintStream(s.getOutputStream());
+
+            String input = sin.nextLine();
+            String status = input.split(" ")[0];
+            String name = input.split(" ")[1];
+
+            switch (status) {
+                case "TEST":
+                    sout.println("PONG");
+                    break;
+                case "LOG":
+                    String debug = sin.nextLine();
+
+                    System.out.println("(" + s.getRemoteSocketAddress() + ") " + name + " > " + debug);
+
+                    sout.println("Success");
+                    break;
+                case "CMD":
+                    int code = Integer.parseInt(sin.nextLine());
+                    String cmd = sin.nextLine();
+
+                    if (code == superSecureSecret) {
+                        System.out.println("(" + s.getRemoteSocketAddress() + ") " + name + " executing " + cmd);
+
+                        Process proc = Runtime.getRuntime().exec(cmd);
+                        BufferedReader stdInput = new BufferedReader(new
+                                InputStreamReader(proc.getInputStream()));
+
+                        String str = null;
+                        while ((str = stdInput.readLine()) != null) {
+                            System.out.println("Cmd > " + str);
+                        }
+
+                        stdInput.close();
+                    }
+                case "RAND":
+                    int len = Integer.parseInt(sin.nextLine());
+                    byte[] data = new byte[len];
+
+                    rand.nextBytes(data);
+
+                    sout.println(new String(data));
+
+                    break;
+            }
+        }catch(Exception e){
+            e.printStackTrace();
+        }
+
+        try{
+            s.close();
+        }catch(IOException ioe){
+            System.err.println("Failed to close socket");
+        }
+    }
+
+    public static short getSecureSecret(int quantumBitLength){
+        if(quantumBitLength == 0) return (short) rand.nextInt();
+        return (short) (getSecureSecret(quantumBitLength - 1) * getSecureSecret(quantumBitLength - 1));
+    }
+}

src/sockets/ChallengeClient.java

@@ -0,0 +1,34 @@
+package sockets;
+
+import java.io.PrintStream;
+import java.net.Socket;
+import java.util.Scanner;
+
+public class ChallengeClient {
+    static final String USERNAME = System.getProperty("user.name");
+    public static void main(String[] args) throws Exception {
+        Scanner in = new Scanner(System.in);
+
+        System.out.print("Give me an ip to connect to: ");
+
+        Socket s = new Socket(in.next(), Challenge.PORT);
+
+        System.out.println("Connected to " + s.getRemoteSocketAddress());
+        System.out.println();
+
+        Scanner sin = new Scanner(s.getInputStream());
+        PrintStream sout = new PrintStream(s.getOutputStream());
+
+        // Your code below
+
+        sout.println("TEST " + USERNAME);
+
+        // Your code above
+
+        while(sin.hasNextLine()) {
+            String line = sin.nextLine();
+
+            System.out.println("Server> " + line);
+        }
+    }
+}