Java 网络编程(二)

class PicClient {
    public static void main(String[] args) throws IOException {
        /*
         * 一系列判断
         */
        if(args.length != 1) {
            System.out.println("请选择一个jpg格式的图片");
            return;
        }
        File file = new File(args[0]);
        if(!(file.exists() && file.isFile())) {
            System.out.println("该文件有问题，要么不存在，要么不是文件");
            return;
        }
        if(!file.getName().endsWith(".jpg")) {
            System.out.println("图片格式错误，请重新选择");
            return;
        }
        if(file.length() > 1024*1024*10) {
            System.out.println("文件过大，没安好心");
            return;
        }
        
        
        Socket s = new Socket("10.48.62.209", 10007);
        
        FileInputStream fis = new FileInputStream(file);
        
        OutputStream out = s.getOutputStream();
        byte[] buf = new byte[1024];
        int len = 0;
        while((len = fis.read(buf)) != -1) {
            out.write(buf, 0, len);
        }
        
        //告诉服务端数据已写完
        s.shutdownOutput();
        
        InputStream in = s.getInputStream();
        byte[] bufIn = new byte[1024];
        int num = in.read(bufIn);
        System.out.println(new String(bufIn, 0, num));
        fis.close();
        s.close();
    }
}

class PicThread implements Runnable {
    private Socket s;
    PicThread(Socket s) {
        this.s = s;
    }
    @Override
    public void run() {
        int count = 1;
        String ip = s.getInetAddress().getHostAddress();
        try {
            
            System.out.println(ip+"......connected.");
        
            
            InputStream in = s.getInputStream();
            
            File file = new File(ip+"("+(count)+")"+".jpg");//10.48.62.209(1).jpg
            while(file.exists()) {
                file = new File(ip+"("+(count++)+")"+".jpg");
            }
            
            
            FileOutputStream fos = new FileOutputStream(file);
            
            
            byte[] buf = new byte[1024];
            int len = 0;
            while((len = in.read(buf)) != -1) {
                fos.write(buf, 0, len);
            }
            OutputStream out = s.getOutputStream();
            out.write("上传成功".getBytes());
            fos.close();
            s.close();
        } catch (IOException e) {
            throw new RuntimeException(ip+"上传失败");
        }
    }
    
}

class PicServer {
    public static void main(String[] args) throws IOException {
        ServerSocket ss = new ServerSocket(10007);
        while(true) {
            Socket s = ss.accept();
            
            new Thread(new PicThread(s)).start();
        }
        //ss.close();
    }
    
}

class LoginClient {
    public static void main(String[] args) throws IOException {
        Socket s = new Socket("10.48.62.209", 10008);
        
        BufferedReader bufr = 
                new BufferedReader(new InputStreamReader(System.in));
        
        PrintWriter out = new PrintWriter(s.getOutputStream(), true);
        
        BufferedReader bufIn = 
                new BufferedReader(new InputStreamReader(s.getInputStream()));
        
        for(int x = 0; x < 3; x++) {
            /*
             * 在键盘敲Ctrl+C键，即代表结束录入
             * read()方法返回-1，readLine()方法返回null
             */
            String line = bufr.readLine();//Ctrl+C，那么就是-1，readLine()返回null
            if(line == null) 
                break;
            out.println(line);
            String info = bufIn.readLine();
            System.out.println("info:"+info);
            if(info.contains("欢迎"))
                break;
        }
        bufr.close();
        s.close();//socket流中加了一个-1，相当于给流中加入一个结束标记，即-1
    }
}

class UserThread implements Runnable {
    private Socket s;
    UserThread(Socket s) {
        this.s = s;
    }
    @Override
    public void run() {
        String ip = s.getInetAddress().getHostAddress();
        System.out.println(ip+"........connected.");
        try {
            for(int x = 0; x < 3; x++) {
                BufferedReader bufIn = new BufferedReader(
                        new InputStreamReader(s.getInputStream()));
                /*
                 * 客户端那边敲入Ctrl+C键键时，
                 * read()返回-1，readLine()返回null。
                 * 所以name得加一个判断
                 */
                String name = bufIn.readLine();
                if(name == null) 
                    break;
                
                /*
                 * 校验
                 */
                BufferedReader bufr = new BufferedReader(new FileReader("user.txt"));
                
                PrintWriter out = new PrintWriter(s.getOutputStream(), true);
                
                String line = null;
                
                boolean flag = false;//定义标记
                while((line = bufr.readLine()) != null) {
                    if(line.equals(name)) {
                        flag = true;
                        break;
                    }
                }
                if(flag) {
                    System.out.println(name+"，已登录");
                    out.println(name+"，欢迎光临");
                    break;
                } else {
                    System.out.println(name+"尝试登录");
                    out.println(name+"，用户名不存在");
                }
            }
            s.close();
        } catch (IOException e) {
            throw new RuntimeException(ip+"校验失败");
        }
    }
    
}

class LoginServer {
    public static void main(String[] args) throws IOException {
        ServerSocket ss = new ServerSocket(10008);
        while(true) {
            Socket s = ss.accept();
            new Thread(new UserThread(s)).start();
        }
    }
    
}