Windows 클라이언트-서버 유틸리티 기능을 사용하여 소프트웨어 작성, 02부

Windows 콘솔 유틸리티의 사용자 정의 구현에 관한 지속적인 기사 시리즈를 계속하면서 간단한 파일 전송 프로토콜인 TFTP(Trivial File Transfer Protocol)에 대해 다루지 않을 수 없습니다.

지난번과 마찬가지로 간단히 이론을 살펴보고, 필요한 기능과 유사한 기능을 구현하는 코드를 보고 분석해 보겠습니다. 자세한 내용 - 컷 아래

전통적으로 기사 끝부분에서 찾을 수 있는 링크인 참조 정보를 복사하여 붙여넣지는 않겠습니다. 핵심적으로 TFTP는 액세스 제어 설정이 있는 FTP 프로토콜의 단순화된 변형이라고만 말씀드리겠습니다. 제거되었으며 실제로 파일 수신 및 전송 명령 외에는 아무것도 없습니다. 그러나 구현을 좀 더 우아하게 만들고 현재 코드 작성 원칙에 맞게 적용하기 위해 구문이 약간 변경되었습니다. 이로 인해 작동 원칙이 변경되지는 않지만 인터페이스 IMHO는 좀 더 논리적이고 FTP와 TFTP의 긍정적인 측면을 결합합니다.

특히, 시작되면 클라이언트는 서버의 IP 주소와 사용자 정의 TFTP가 열려 있는 포트를 요청합니다(표준 프로토콜과의 비호환성으로 인해 사용자가 포트를 선택할 수 있도록 하는 것이 적절하다고 생각함). 연결이 발생하면 클라이언트는 get 또는 put 명령 중 하나를 보내 서버에 파일을 받거나 보낼 수 있습니다. 모든 파일은 논리를 단순화하기 위해 바이너리 모드로 전송됩니다.

프로토콜을 구현하기 위해 전통적으로 4가지 클래스를 사용했습니다.

• TFTP클라이언트
• TFTP서버
• TFTP클라이언트테스터
• TFTP서버테스터

테스트 클래스는 주요 클래스 디버깅을 위해서만 존재하기 때문에 분석하지는 않겠지만 코드는 저장소에 있을 것이며 이에 대한 링크는 기사 끝부분에서 찾을 수 있습니다. 이제 주요 수업을 살펴보겠습니다.

TFTP클라이언트

이 클래스의 임무는 IP와 포트 번호로 원격 서버에 연결하고, 입력 스트림(이 경우 키보드)에서 명령을 읽고, 구문 분석하고, 서버로 전송하는 것입니다. 파일을 보내거나 받고, 전송하거나 받아야 합니다.

서버에 연결하고 입력 스트림의 명령을 기다리기 위해 클라이언트를 시작하는 코드는 다음과 같습니다. 여기에 사용되는 여러 전역 변수는 기사 외부의 프로그램 전체 텍스트에 설명되어 있습니다. 그 사소함 때문에 기사에 과부하가 걸리지 않도록 인용하지 않습니다.

 public void run(String ip, int port)
    {
        this.ip = ip;
        this.port = port;
        try {
            inicialization();
            Scanner keyboard = new Scanner(System.in);
            while (isRunning) {
                getAndParseInput(keyboard);
                sendCommand();
                selector();
                }
            }
        catch (Exception e) {
            System.out.println(e.getMessage());
        }
    }

이 코드 블록에서 호출되는 메서드를 살펴보겠습니다.

여기에서 파일이 전송됩니다. 스캐너를 사용하여 파일 내용을 바이트 배열로 표시합니다. 이를 소켓에 하나씩 쓴 다음 파일을 닫았다가 다시 엽니다(가장 확실한 해결책은 아니지만 이는 리소스 해제를 보장합니다. 그 후 성공적인 전송에 대한 메시지가 표시됩니다.

private  void put(String sourcePath, String destPath)
    {

        File src = new File(sourcePath);
        try {

            InputStream scanner = new FileInputStream(src);
            byte[] bytes = scanner.readAllBytes();
            for (byte b : bytes)
                sout.write(b);
            sout.close();
            inicialization();
            System.out.println("nDonen");
            }

        catch (Exception e) {
            System.out.println(e.getMessage());
        }
    }

이 코드 단편은 서버에서 데이터를 검색하는 방법을 설명합니다. 모든 것이 다시 사소하며 첫 번째 코드 블록만 관심이 있습니다. 소켓에서 읽어야 하는 바이트 수를 정확히 이해하려면 전송된 파일의 무게를 알아야 합니다. 서버의 파일 크기는 긴 정수로 표시되므로 여기서는 4바이트가 허용되며 이후에는 하나의 숫자로 변환됩니다. 이는 Java 접근 방식이 아니며 SI와 유사하지만 문제를 해결합니다.

그러면 모든 것이 간단해집니다. 소켓에서 알려진 바이트 수를 수신하여 파일에 쓴 후 성공 메시지를 표시합니다.

   private void get(String sourcePath, String destPath){
        long sizeOfFile = 0;
        try {


            byte[] sizeBytes = new byte[Long.SIZE];
           for (int i =0; i< Long.SIZE/Byte.SIZE; i++)
           {
               sizeBytes[i] = (byte)sin.read();
               sizeOfFile*=256;
               sizeOfFile+=sizeBytes[i];
           }

           FileOutputStream writer = new FileOutputStream(new File(destPath));
           for (int i =0; i < sizeOfFile; i++)
           {
               writer.write(sin.read());
           }
           writer.close();
           System.out.println("nDONEn");
       }
       catch (Exception e){
            System.out.println(e.getMessage());
       }
    }

클라이언트 창에 get 또는 put 이외의 명령을 입력한 경우 showErrorMessage 함수가 호출되어 입력이 잘못되었음을 나타냅니다. 사소해서 인용하지 않겠습니다. 좀 더 흥미로운 점은 입력 문자열을 수신하고 분할하는 기능입니다. 스캐너를 여기에 전달하면 두 개의 공백으로 구분되고 명령, 소스 주소 및 대상 주소가 포함된 줄을 받을 것으로 예상됩니다.

    private void getAndParseInput(Scanner scanner)
    {
        try {

            input = scanner.nextLine().split(" ");
            typeOfCommand = input[0];
            sourcePath = input[1];
            destPath = input[2];
        }
        catch (Exception e) {
            System.out.println("Bad input");
        }
    }

명령 보내기 - 스캐너에서 입력한 명령을 소켓으로 전송하고 강제로 전송합니다.

    private void sendCommand()
    {
        try {

            for (String str : input) {
                for (char ch : str.toCharArray()) {
                    sout.write(ch);
                }
                sout.write(' ');
            }
            sout.write('n');
        }
        catch (Exception e) {
            System.out.print(e.getMessage());
        }
    }

선택기는 입력된 문자열에 따라 프로그램의 동작을 결정하는 기능입니다. 여기에 있는 모든 것은 별로 아름답지 않으며 사용된 트릭은 코드 블록 외부에서 강제 종료하는 가장 좋은 트릭은 아니지만, 그 주된 이유는 Java에 C#의 대리자, C++의 함수 포인터 또는 다음과 같은 것들이 없기 때문입니다. 최소한 끔찍하고 끔찍한 goto를 사용하면 이를 아름답게 구현할 수 있습니다. 코드를 좀 더 우아하게 만드는 방법을 알고 계시다면 댓글로 비판을 환영합니다. 여기에는 문자열 위임 사전이 필요한 것 같은데 위임자가 없습니다...

    private void selector()
    {
        do{
            if (typeOfCommand.equals("get")){
                get(sourcePath, destPath);
                break;
            }
            if (typeOfCommand.equals("put")){
                put(sourcePath, destPath);
                break;
            }
            showErrorMessage();
        }
        while (false);
    }
}

TFTP서버

서버의 기능은 명령이 키보드가 아닌 소켓에서 전달된다는 점에서만 클라이언트의 기능과 대체로 다릅니다. 일부 방법은 일반적으로 동일하므로 인용하지 않고 차이점만 다루겠습니다.

시작하려면 포트를 입력으로 받고 소켓의 입력 데이터를 영원한 루프로 처리하는 run 메서드가 사용됩니다.

    public void run(int port) {
            this.port = port;
            incialization();
            while (true) {
                getAndParseInput();
                selector();
            }
    }

파일에 대한 쓰기 스트림을 열고 소켓의 모든 입력 바이트를 쓰는 writeToFileFromSocket 메서드를 래핑하는 put 메서드는 쓰기가 완료되면 전송이 성공적으로 완료되었음을 나타내는 메시지를 표시합니다.

    private  void put(String source, String dest){
            writeToFileFromSocket();
            System.out.print("nDonen");
    };
    private void writeToFileFromSocket()
    {
        try {
            FileOutputStream writer = new FileOutputStream(new File(destPath));
            byte[] bytes = sin.readAllBytes();
            for (byte b : bytes) {
                writer.write(b);
            }
            writer.close();
        }
        catch (Exception e){
            System.out.println(e.getMessage());
        }
    }

get 메소드는 서버 파일을 검색합니다. 프로그램의 클라이언트 측 섹션에서 이미 언급했듯이 파일을 성공적으로 전송하려면 긴 정수로 저장된 크기를 알아야 하므로 이를 4바이트 배열로 분할하여 바이트 단위로 전송합니다. 소켓에 연결한 다음 클라이언트에서 이를 받아 숫자로 다시 조합한 후 파일을 구성하는 모든 바이트를 파일의 입력 스트림에서 읽은 후 전송합니다.

 private  void get(String source, String dest){
        File sending = new File(source);
        try {
            FileInputStream readFromFile = new FileInputStream(sending);
            byte[] arr = readFromFile.readAllBytes();
            byte[] bytes = ByteBuffer.allocate(Long.SIZE / Byte.SIZE).putLong(sending.length()).array();
            for (int i = 0; i<Long.SIZE / Byte.SIZE; i++)
                sout.write(bytes[i]);
            sout.flush();
            for (byte b : arr)
                sout.write(b);
        }
        catch (Exception e){
            System.out.println(e.getMessage());
        }
    };

getAndParseInput 메소드는 클라이언트와 동일하며 유일한 차이점은 키보드가 아닌 소켓에서 데이터를 읽는다는 것입니다. 코드는 선택기와 마찬가지로 저장소에 있습니다.
이 경우 초기화는 별도의 코드 블록에 배치됩니다. 이 구현 내에서는 전송이 완료된 후 리소스가 해제되고 다시 점유되어 다시 메모리 누수에 대한 보호를 제공합니다.

    private void  incialization()
    {
        try {
            serverSocket = new ServerSocket(port);
            socket = serverSocket.accept();
            sin = socket.getInputStream();
            sout = socket.getOutputStream();
        }
        catch (Exception e) {
            System.out.print(e.getMessage());
        }
    }

요약 :

우리는 간단한 데이터 전송 프로토콜에 대한 변형을 작성하고 그것이 어떻게 작동해야 하는지 알아냈습니다. 원칙적으로 나는 여기서 미국을 발견하지 못했고 새로운 것을 많이 쓰지 않았지만 Habré에 대한 유사한 기사는 없었으며 cmd 유틸리티에 대한 일련의 기사를 작성하는 과정에서 그것을 건드리지 않는 것이 불가능했습니다.

링크 :

소스 코드 저장소
TFTP에 대해 간략히 설명
똑같은 일이지만 러시아어로

출처 : habr.com