Question

我正在使用Cipher与基础socket和dataoutputstream进行连接。现在根据CipherOutputStream文档，为了让最后一个块脱离加密器，我们需要调用close()。因此，将它与try-with-resource一起使用也会关闭我不想要的基础流，因为我需要它们进行进一步的操作。

那么有没有任何方法或实现可以帮助我在没有cipheroutputstream和dataoutputstream的情况下关闭socket。我听说过FilterOutputStream，但我不知道如何使用它们。

我的代码：

public class ClientHandler implements Runnable {
Socket msock;
DataInputStream dis;
DataOutputStream dos;
String ip;
private miniClientHandler(Socket msock) {
            this.msock = msock;
        }
@Override
public void run() {
    try {
        dis = new DataInputStream(msock.getInputStream());
        dos = new DataOutputStream(msock.getOutputStream());

        ip = msock.getInetAddress().getHostAddress();

        String msg = dis.readLine();
        System.out.println(msg);

        if (msg.equals("Download")) {

                    String file2dl = dis.readLine(); //2
                    File file = new File(sharedDirectory.toString() + "\\" + file2dl);
                    dos.writeLong(file.length()); //3+

                    //AES-128 bit key initialization.
                    byte[] keyvalue = "AES128BitPasswd".getBytes();
                    SecretKey key = new SecretKeySpec(keyvalue, "AES");

                    //Initialize the Cipher.
                    Cipher encCipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
                    encCipher.init(Cipher.ENCRYPT_MODE, key);

                    //Get the IV from cipher.
                    IvParameterSpec spec = null;
                    try {
                        spec = encCipher.getParameters().getParameterSpec(IvParameterSpec.class);
                    } catch (InvalidParameterSpecException ex) {
                        Logger.getLogger(PeersController.class.getName()).log(Level.SEVERE, null, ex);
                    }

                    byte[] iv = spec.getIV();

                    dos.write(iv, 0, iv.length); //4+

                    //Encryption Mechanism.
                    try (FileInputStream fis = new FileInputStream(file)) {
                        try (CipherOutputStream cos = new CipherOutputStream(dos, encCipher)) {
                            int read;
                            byte[] buffer = new byte[1024 * 1024];
                            while ((read = fis.read(buffer)) != -1) {
                                cos.write(buffer, 0, read); //5+ due to dos as underlying parameter of cos.
                            }
                        }
                    }

                    String message = dis.readLine();
                    System.out.println(message);
                    if (message.equals("Fetching Done")) {
                        System.out.println("Fetching Done!");
                    } else if (message.equals("Fetching Drop")) {
                        System.out.println("Fetching Denied!");
                    }
                }
            } catch (IOException | NoSuchAlgorithmException | NoSuchPaddingException | InvalidKeyException e) {
                System.out.println(e.getMessage());
            }
        }

    }

现在，因为cos包含在try-with-resource中，所以当while循环结束时，它会关闭流和基础dos流，它们进一步包裹msock，导致整个套接字关闭，剩下的代码没用了。

剩余代码：

    String message = dis.readLine();
    System.out.println(message);
    if (message.equals("Fetching Done")) {
        System.out.println("Fetching Done!");
     } else if (message.equals("Fetching Drop")) {
                 System.out.println("Fetching Denied!");
     }

Answer 1

我在@EJP建议的帮助下以某种方式自行解决我的问题，通过update()＆amp;创建我自己的Cipher对象。 doFinal()。

的代码：

//Encryption Mechanism. try (FileInputStream fis = new FileInputStream(file)) { byte[] buffer = new byte[1024 * 1024]; byte[] cipherBlock = new byte[encCipher.getOutputSize(buffer.length)]; int cipherBytes; int read; while ((read = fis.read(buffer)) != -1) { cipherBytes = encCipher.update(buffer, 0, read, cipherBlock); dos.write(cipherBlock, 0, cipherBytes); } cipherBytes = encCipher.doFinal(cipherBlock, 0); //let out the last block out of the encryptor. dos.write(cipherBlock, 0, cipherBytes); } catch (ShortBufferException | IllegalBlockSizeException | BadPaddingException ex) { Logger.getLogger(PeersController.class.getName()).log(Level.SEVERE, null, ex); }

关闭没有DataOutputStream＆amp;的CipherOutputStream插座

1 个答案: