Examples with SSLEngineResult javax.net.ssl.SSLEngineResult used on opensource projects

Example 61 with SSLEngineResult

use of javax.net.ssl.SSLEngineResult in project jdk8u_jdk by JetBrains.

the class RehandshakeFinished method runTest.

/*
     * Run the test.
     *
     * Sit in a tight loop, both engines calling wrap/unwrap regardless
     * of whether data is available or not.  We do this until both engines
     * report back they are closed.
     *
     * The main loop handles all of the I/O phases of the SSLEngine's
     * lifetime:
     *
     *     initial handshaking
     *     application data transfer
     *     engine closing
     *
     * One could easily separate these phases into separate
     * sections of code.
     */
private SSLSession runTest() throws Exception {
    boolean dataDone = false;
    createSSLEngines();
    createBuffers();
    // results from client's last operation
    SSLEngineResult clientResult;
    // results from server's last operation
    SSLEngineResult serverResult;
    /*
         * Examining the SSLEngineResults could be much more involved,
         * and may alter the overall flow of the application.
         *
         * For example, if we received a BUFFER_OVERFLOW when trying
         * to write to the output pipe, we could reallocate a larger
         * pipe, but instead we wait for the peer to drain it.
         */
    while (!isEngineClosed(clientEngine) || !isEngineClosed(serverEngine)) {
        log("================");
        clientResult = clientEngine.wrap(clientOut, cTOs);
        log("client wrap: ", clientResult);
        runDelegatedTasks(clientResult, clientEngine);
        serverResult = serverEngine.wrap(serverOut, sTOc);
        log("server wrap: ", serverResult);
        runDelegatedTasks(serverResult, serverEngine);
        cTOs.flip();
        sTOc.flip();
        log("----");
        clientResult = clientEngine.unwrap(sTOc, clientIn);
        log("client unwrap: ", clientResult);
        runDelegatedTasks(clientResult, clientEngine);
        serverResult = serverEngine.unwrap(cTOs, serverIn);
        log("server unwrap: ", serverResult);
        runDelegatedTasks(serverResult, serverEngine);
        cTOs.compact();
        sTOc.compact();
        /*
             * After we've transfered all application data between the client
             * and server, we close the clientEngine's outbound stream.
             * This generates a close_notify handshake message, which the
             * server engine receives and responds by closing itself.
             */
        if (!dataDone && (clientOut.limit() == serverIn.position()) && (serverOut.limit() == clientIn.position())) {
            /*
                 * A sanity check to ensure we got what was sent.
                 */
            checkTransfer(serverOut, clientIn);
            checkTransfer(clientOut, serverIn);
            log("\tClosing clientEngine's *OUTBOUND*...");
            clientEngine.closeOutbound();
            dataDone = true;
        }
    }
    return clientEngine.getSession();
}

Example 62 with SSLEngineResult

use of javax.net.ssl.SSLEngineResult in project jdk8u_jdk by JetBrains.

the class SSLEngineDeadlock method runTest.

/*
     * Run the test.
     *
     * Sit in a tight loop, both engines calling wrap/unwrap regardless
     * of whether data is available or not.  We do this until both engines
     * report back they are closed.
     *
     * The main loop handles all of the I/O phases of the SSLEngine's
     * lifetime:
     *
     *     initial handshaking
     *     application data transfer
     *     engine closing
     *
     * One could easily separate these phases into separate
     * sections of code.
     */
private void runTest() throws Exception {
    boolean dataDone = false;
    createSSLEngines();
    createBuffers();
    // results from client's last operation
    SSLEngineResult clientResult;
    // results from server's last operation
    SSLEngineResult serverResult;
    new Thread("SSLEngine Task Dispatcher") {

        public void run() {
            try {
                doTask();
            } catch (Exception e) {
                System.err.println("Task thread died...test will hang");
            }
        }
    }.start();
    /*
         * Examining the SSLEngineResults could be much more involved,
         * and may alter the overall flow of the application.
         *
         * For example, if we received a BUFFER_OVERFLOW when trying
         * to write to the output pipe, we could reallocate a larger
         * pipe, but instead we wait for the peer to drain it.
         */
    while (!isEngineClosed(clientEngine) || !isEngineClosed(serverEngine)) {
        log("================");
        clientResult = clientEngine.wrap(clientOut, cTOs);
        log("client wrap: ", clientResult);
        serverResult = serverEngine.wrap(serverOut, sTOc);
        log("server wrap: ", serverResult);
        cTOs.flip();
        sTOc.flip();
        log("----");
        clientResult = clientEngine.unwrap(sTOc, clientIn);
        log("client unwrap: ", clientResult);
        serverResult = serverEngine.unwrap(cTOs, serverIn);
        log("server unwrap: ", serverResult);
        cTOs.compact();
        sTOc.compact();
        /*
             * After we've transfered all application data between the client
             * and server, we close the clientEngine's outbound stream.
             * This generates a close_notify handshake message, which the
             * server engine receives and responds by closing itself.
             */
        if (!dataDone && (clientOut.limit() == serverIn.position()) && (serverOut.limit() == clientIn.position())) {
            /*
                 * A sanity check to ensure we got what was sent.
                 */
            checkTransfer(serverOut, clientIn);
            checkTransfer(clientOut, serverIn);
            log("\tClosing clientEngine's *OUTBOUND*...");
            clientEngine.closeOutbound();
            dataDone = true;
        }
    }
    testDone = true;
}

Example 63 with SSLEngineResult

use of javax.net.ssl.SSLEngineResult in project jdk8u_jdk by JetBrains.

the class RehandshakeFinished method runRehandshake.

private SSLSession runRehandshake() throws Exception {
    log("\n\n==============================================");
    log("Staring actual test.");
    createSSLEngines();
    createBuffers();
    SSLEngineResult result;
    log("Client's ClientHello");
    checkResult(clientEngine, clientEngine.wrap(clientOut, cTOs), HandshakeStatus.NEED_UNWRAP, false, true);
    cTOs.flip();
    checkResult(serverEngine, serverEngine.unwrap(cTOs, serverIn), HandshakeStatus.NEED_WRAP, true, false);
    cTOs.compact();
    log("Server's ServerHello/ServerHelloDone");
    checkResult(serverEngine, serverEngine.wrap(serverOut, sTOc), HandshakeStatus.NEED_WRAP, false, true);
    sTOc.flip();
    checkResult(clientEngine, clientEngine.unwrap(sTOc, clientIn), HandshakeStatus.NEED_UNWRAP, true, false);
    sTOc.compact();
    log("Server's CCS");
    checkResult(serverEngine, serverEngine.wrap(serverOut, sTOc), HandshakeStatus.NEED_WRAP, false, true);
    sTOc.flip();
    checkResult(clientEngine, clientEngine.unwrap(sTOc, clientIn), HandshakeStatus.NEED_UNWRAP, true, false);
    sTOc.compact();
    log("Server's FINISHED");
    checkResult(serverEngine, serverEngine.wrap(serverOut, sTOc), HandshakeStatus.NEED_UNWRAP, false, true);
    sTOc.flip();
    checkResult(clientEngine, clientEngine.unwrap(sTOc, clientIn), HandshakeStatus.NEED_WRAP, true, false);
    sTOc.compact();
    log("Client's CCS");
    checkResult(clientEngine, clientEngine.wrap(clientOut, cTOs), HandshakeStatus.NEED_WRAP, false, true);
    cTOs.flip();
    checkResult(serverEngine, serverEngine.unwrap(cTOs, serverIn), HandshakeStatus.NEED_UNWRAP, true, false);
    cTOs.compact();
    log("Client's FINISHED should trigger FINISHED messages all around.");
    checkResult(clientEngine, clientEngine.wrap(clientOut, cTOs), HandshakeStatus.FINISHED, false, true);
    cTOs.flip();
    checkResult(serverEngine, serverEngine.unwrap(cTOs, serverIn), HandshakeStatus.FINISHED, true, false);
    cTOs.compact();
    return clientEngine.getSession();
}

Example 64 with SSLEngineResult

use of javax.net.ssl.SSLEngineResult in project tigervnc by TigerVNC.

the class SSLEngineManager method doHandshake.

public void doHandshake() throws Exception {
    // Begin handshake
    engine.beginHandshake();
    SSLEngineResult.HandshakeStatus hs = engine.getHandshakeStatus();
    // Process handshaking message
    SSLEngineResult res = null;
    int appBufSize = engine.getSession().getApplicationBufferSize();
    ByteBuffer peerAppData = ByteBuffer.allocate(appBufSize);
    ByteBuffer myAppData = ByteBuffer.allocate(appBufSize);
    while (hs != SSLEngineResult.HandshakeStatus.FINISHED && hs != SSLEngineResult.HandshakeStatus.NOT_HANDSHAKING) {
        switch(hs) {
            case NEED_UNWRAP:
                // Receive handshaking data from peer
                peerNetData.flip();
                res = engine.unwrap(peerNetData, peerAppData);
                peerNetData.compact();
                hs = res.getHandshakeStatus();
                // Check status
                switch(res.getStatus()) {
                    case BUFFER_UNDERFLOW:
                        int avail = in.check(1, peerNetData.remaining(), false);
                        in.readBytes(peerNetData, avail);
                        break;
                    case OK:
                        // Process incoming handshaking data
                        break;
                    case CLOSED:
                        engine.closeInbound();
                        break;
                }
                break;
            case NEED_WRAP:
                // Generate handshaking data
                res = engine.wrap(myAppData, myNetData);
                hs = res.getHandshakeStatus();
                // Check status
                switch(res.getStatus()) {
                    case OK:
                        myNetData.flip();
                        os.writeBytes(myNetData, myNetData.remaining());
                        os.flush();
                        myNetData.compact();
                        break;
                    case CLOSED:
                        engine.closeOutbound();
                        break;
                }
                break;
            case NEED_TASK:
                // Handle blocking tasks
                executeTasks();
                break;
        }
        hs = engine.getHandshakeStatus();
    }
}

Example 65 with SSLEngineResult

use of javax.net.ssl.SSLEngineResult in project tigervnc by TigerVNC.

the class SSLEngineManager method write.

public int write(ByteBuffer data, int length) throws IOException {
    int n = 0;
    while (data.hasRemaining()) {
        SSLEngineResult res = engine.wrap(data, myNetData);
        n += res.bytesConsumed();
        switch(res.getStatus()) {
            case OK:
                myNetData.flip();
                os.writeBytes(myNetData, myNetData.remaining());
                os.flush();
                myNetData.compact();
                break;
            case BUFFER_OVERFLOW:
                // Make room in the buffer by flushing the outstream
                myNetData.flip();
                os.writeBytes(myNetData, myNetData.remaining());
                myNetData.compact();
                break;
            case CLOSED:
                engine.closeOutbound();
                break;
        }
    }
    return n;
}