Search in sources :

Example 1 with GenericResponse

use of org.firebirdsql.gds.ng.wire.GenericResponse in project jaybird by FirebirdSQL.

the class TestV10WireOperations method testProcessResponse_noException.

/**
 * Test if processResponse does not throw an exception if the response does
 * not contain an exception.
 */
@Test
public void testProcessResponse_noException() throws Exception {
    AbstractWireOperations wire = createDummyWireOperations();
    GenericResponse genericResponse = new GenericResponse(-1, -1, null, null);
    wire.processResponse(genericResponse);
}
Also used : AbstractWireOperations(org.firebirdsql.gds.ng.wire.AbstractWireOperations) GenericResponse(org.firebirdsql.gds.ng.wire.GenericResponse) Test(org.junit.Test)

Example 2 with GenericResponse

use of org.firebirdsql.gds.ng.wire.GenericResponse in project jaybird by FirebirdSQL.

the class V13WireOperations method authReceiveResponse.

@Override
public void authReceiveResponse(FbWireAttachment.AcceptPacket acceptPacket, FbWireOperations.ProcessAttachCallback processAttachCallback) throws SQLException, IOException {
    assert acceptPacket == null || acceptPacket.operation == op_cond_accept : "Unexpected operation in AcceptPacket";
    final XdrInputStream xdrIn = getXdrIn();
    final XdrOutputStream xdrOut = getXdrOut();
    final ClientAuthBlock clientAuthBlock = getClientAuthBlock();
    final Encoding encoding = getEncoding();
    while (true) {
        String pluginName;
        byte[] data;
        if (acceptPacket != null) {
            data = acceptPacket.p_acpt_data;
            pluginName = acceptPacket.p_acpt_plugin;
            addServerKeys(acceptPacket.p_acpt_keys);
            log.debug(String.format("authReceiveResponse: cond_accept data=%d pluginName=%d '%s'", data.length, pluginName != null ? pluginName.length() : null, pluginName));
            // TODO handle compression
            acceptPacket = null;
        } else {
            int operation = readNextOperation();
            switch(operation) {
                case op_trusted_auth:
                    // p_trau_data
                    xdrIn.readBuffer();
                    throw new FbExceptionBuilder().nonTransientConnectionException(JaybirdErrorCodes.jb_receiveTrustedAuth_NotSupported).toFlatSQLException();
                case op_cont_auth:
                    // p_data
                    data = xdrIn.readBuffer();
                    // p_name
                    pluginName = xdrIn.readString(encoding);
                    // p_list (ignore?)
                    xdrIn.readBuffer();
                    // p_keys
                    addServerKeys(xdrIn.readBuffer());
                    log.debug(String.format("authReceiveResponse: cont_auth data=%d pluginName=%d '%s'", data.length, pluginName.length(), pluginName));
                    break;
                case op_cond_accept:
                    // Note this is the equivalent of handling the acceptPacket != null above
                    // p_acpt_version
                    xdrIn.readInt();
                    // p_acpt_architecture
                    xdrIn.readInt();
                    // p_acpt_type
                    xdrIn.readInt();
                    // p_acpt_data
                    data = xdrIn.readBuffer();
                    // p_acpt_plugin
                    pluginName = xdrIn.readString(encoding);
                    // p_acpt_authenticated
                    xdrIn.readInt();
                    // p_acpt_keys
                    addServerKeys(xdrIn.readBuffer());
                    log.debug(String.format("authReceiveResponse: cond_accept data=%d pluginName=%d '%s'", data.length, pluginName.length(), pluginName));
                    // TODO handle compression
                    break;
                case op_response:
                    GenericResponse response = (GenericResponse) readOperationResponse(operation, null);
                    boolean wasAuthComplete = clientAuthBlock.isAuthComplete();
                    clientAuthBlock.setAuthComplete(true);
                    processAttachCallback.processAttachResponse(response);
                    addServerKeys(response.getData());
                    WireCrypt wireCrypt = getAttachProperties().getWireCrypt();
                    if (!wasAuthComplete && wireCrypt != WireCrypt.DISABLED) {
                        tryKnownServerKeys();
                    }
                    return;
                default:
                    throw new SQLException(String.format("Unsupported operation code: %d", operation));
            }
        }
        if (pluginName != null && pluginName.length() > 0 && Objects.equals(pluginName, clientAuthBlock.getCurrentPluginName())) {
            pluginName = null;
        }
        if (pluginName != null && pluginName.length() > 0) {
            if (!clientAuthBlock.switchPlugin(pluginName)) {
                break;
            }
        }
        if (!clientAuthBlock.hasPlugin()) {
            break;
        }
        clientAuthBlock.setServerData(data);
        log.debug(String.format("receiveResponse: authenticate(%s)", clientAuthBlock.getCurrentPluginName()));
        clientAuthBlock.authenticate();
        xdrOut.writeInt(op_cont_auth);
        // TODO Move to ClientAuthBlock?
        // p_data
        xdrOut.writeBuffer(clientAuthBlock.getClientData());
        // p_name
        xdrOut.writeString(clientAuthBlock.getCurrentPluginName(), encoding);
        if (clientAuthBlock.isFirstTime()) {
            // p_list
            xdrOut.writeString(clientAuthBlock.getPluginNames(), encoding);
            clientAuthBlock.setFirstTime(false);
        } else {
            // p_list
            xdrOut.writeBuffer(null);
        }
        // p_keys
        xdrOut.writeBuffer(null);
        xdrOut.flush();
    }
    // If we have exited from the cycle, this mean auth failed
    throw new FbExceptionBuilder().exception(ISCConstants.isc_login).toFlatSQLException();
}
Also used : XdrInputStream(org.firebirdsql.gds.impl.wire.XdrInputStream) GenericResponse(org.firebirdsql.gds.ng.wire.GenericResponse) SQLException(java.sql.SQLException) FbExceptionBuilder(org.firebirdsql.gds.ng.FbExceptionBuilder) XdrOutputStream(org.firebirdsql.gds.impl.wire.XdrOutputStream) Encoding(org.firebirdsql.encodings.Encoding) WireCrypt(org.firebirdsql.gds.ng.WireCrypt) ClientAuthBlock(org.firebirdsql.gds.ng.wire.auth.ClientAuthBlock)

Example 3 with GenericResponse

use of org.firebirdsql.gds.ng.wire.GenericResponse in project jaybird by FirebirdSQL.

the class TestV10WireOperations method testProcessResponseWarnings_exception.

/**
 * Test if no warning is registered with the callback if the response
 * contains an exception that is not a warning.
 */
@Test
public void testProcessResponseWarnings_exception() throws Exception {
    AbstractWireOperations wire = createDummyWireOperations();
    SQLException exception = new FbExceptionBuilder().exception(ISCConstants.isc_numeric_out_of_range).toSQLException();
    GenericResponse genericResponse = new GenericResponse(-1, -1, null, exception);
    wire.processResponseWarnings(genericResponse, null);
    List<SQLWarning> warnings = warningCallback.getWarnings();
    assertEquals("Expected no warnings to be registered", 0, warnings.size());
}
Also used : SQLWarning(java.sql.SQLWarning) AbstractWireOperations(org.firebirdsql.gds.ng.wire.AbstractWireOperations) SQLException(java.sql.SQLException) GenericResponse(org.firebirdsql.gds.ng.wire.GenericResponse) FbExceptionBuilder(org.firebirdsql.gds.ng.FbExceptionBuilder) Test(org.junit.Test)

Example 4 with GenericResponse

use of org.firebirdsql.gds.ng.wire.GenericResponse in project jaybird by FirebirdSQL.

the class TestV10WireOperations method testProcessResponseWarnings_warning.

/**
 * Test if a warning is registered with the callback if the response
 * contains an exception that is a warning.
 */
@Test
public void testProcessResponseWarnings_warning() throws Exception {
    AbstractWireOperations wire = createDummyWireOperations();
    SQLWarning warning = new FbExceptionBuilder().warning(ISCConstants.isc_numeric_out_of_range).toSQLException(SQLWarning.class);
    GenericResponse genericResponse = new GenericResponse(-1, -1, null, warning);
    wire.processResponseWarnings(genericResponse, null);
    List<SQLWarning> warnings = warningCallback.getWarnings();
    assertEquals("Unexpected warnings registered or no warnings registered", Collections.singletonList(warning), warnings);
}
Also used : SQLWarning(java.sql.SQLWarning) AbstractWireOperations(org.firebirdsql.gds.ng.wire.AbstractWireOperations) GenericResponse(org.firebirdsql.gds.ng.wire.GenericResponse) FbExceptionBuilder(org.firebirdsql.gds.ng.FbExceptionBuilder) Test(org.junit.Test)

Example 5 with GenericResponse

use of org.firebirdsql.gds.ng.wire.GenericResponse in project jaybird by FirebirdSQL.

the class TestV10WireOperations method testProcessResponseWarnings_noException.

/**
 * Test if no warning is registered with the callback if the response does
 * not contain an exception.
 */
@Test
public void testProcessResponseWarnings_noException() throws Exception {
    AbstractWireOperations wire = createDummyWireOperations();
    GenericResponse genericResponse = new GenericResponse(-1, -1, null, null);
    wire.processResponseWarnings(genericResponse, null);
    List<SQLWarning> warnings = warningCallback.getWarnings();
    assertEquals("Expected no warnings to be registered", 0, warnings.size());
}
Also used : SQLWarning(java.sql.SQLWarning) AbstractWireOperations(org.firebirdsql.gds.ng.wire.AbstractWireOperations) GenericResponse(org.firebirdsql.gds.ng.wire.GenericResponse) Test(org.junit.Test)

Aggregations

GenericResponse (org.firebirdsql.gds.ng.wire.GenericResponse)9 SQLException (java.sql.SQLException)6 FbExceptionBuilder (org.firebirdsql.gds.ng.FbExceptionBuilder)6 AbstractWireOperations (org.firebirdsql.gds.ng.wire.AbstractWireOperations)6 Test (org.junit.Test)6 SQLWarning (java.sql.SQLWarning)3 XdrOutputStream (org.firebirdsql.gds.impl.wire.XdrOutputStream)3 Encoding (org.firebirdsql.encodings.Encoding)2 XdrInputStream (org.firebirdsql.gds.impl.wire.XdrInputStream)2 WireCrypt (org.firebirdsql.gds.ng.WireCrypt)2 ClientAuthBlock (org.firebirdsql.gds.ng.wire.auth.ClientAuthBlock)2 IOException (java.io.IOException)1