Code:


import java.net.*;
import java.util.*;
import java.io.*;

abstract class SC {
    public void start(){}
    public void start(String address){}
}

class Client extends SC{

    public void start(String ipad){
        try{
            Socket link = new Socket(ipad, 7904);
            System.out.println("Konekcija uspostavljena");
            Scanner input = new Scanner(link.getInputStream());
            PrintWriter output = new PrintWriter(link.getOutputStream(),true);

            Scanner us = new Scanner(System.in);
            String poruka, odgovor;

            System.out.print("Unesi poruku : ");
            poruka = us.nextLine();
            output.println(poruka);
            odgovor = input.nextLine();
            System.out.println("\nSERVER> "+odgovor);
        }catch(Exception e){

        }
    }
}
class Server extends SC{


    public void start(String adresa){
        try{
            ServerSocket soc = new ServerSocket(7904, 2, InetAddress.getByName(adresa));
            Socket socket = soc.accept();
            Scanner input = new Scanner(socket.getInputStream());
            String inpu = input.nextLine();
            System.out.println(inpu);
            PrintWriter output = new PrintWriter(socket.getOutputStream(),true);
            output.println("Poruka primljena, zatvaram konekciju !");

            socket.close();
            System.out.println("Konekciaj zatvorena !");

        }catch(Exception e){

        }
    }
}
class Glavna {

    public static void main(String[] args){
        
        InetAddress inet=null;
        String address="";
        try{

            inet = InetAddress.getLocalHost();
            address = InetAddress.getLocalHost().getHostAddress();
        }catch(Exception e){
            
        }
        System.out.println("Tvoja IP Adresa je " +inet.toString());

        System.out.println("Unesi s za servera ili nesto drugo za klijenta");
        Scanner in = new Scanner(System.in);
        String inh = in.nextLine();

        SC chater;

        if(inh.equals("s")){
            chater = new Server();
            chater.start(address);
        }else{
            System.out.println("Unesi IP adresu ");
            Scanner ip = new Scanner(System.in);
            String ipad = in.nextLine();

            chater = new Client();
            chater.start(ipad);
        }
    }
}