Search in sources :

Example 1 with Status

use of org.xnio.IoFuture.Status in project ysoserial by frohoff.

the class JBoss method getChannel.

private static Channel getChannel(ConnectionProviderContextImpl context, ConnectionHandler ch, OptionMap options) throws IOException {
    Channel c;
    FutureResult<Channel> chResult = new FutureResult<Channel>(context.getExecutor());
    ch.open("jmx", chResult, options);
    IoFuture<Channel> cFuture = chResult.getIoFuture();
    Status s2 = cFuture.await();
    if (s2 == Status.FAILED) {
        System.err.println("Cannot connect");
        if (cFuture.getException() != null) {
            throw new IOException("Connect failed", cFuture.getException());
        }
    } else if (s2 != Status.DONE) {
        cFuture.cancel();
        throw new IOException("Connect timeout");
    }
    c = cFuture.get();
    return c;
}
Also used : Status(org.xnio.IoFuture.Status) FutureResult(org.xnio.FutureResult) Channel(org.jboss.remoting3.Channel) IOException(java.io.IOException)

Example 2 with Status

use of org.xnio.IoFuture.Status in project ysoserial by frohoff.

the class JBoss method getConnection.

private static ConnectionHandlerFactory getConnection(SocketAddress destination, final String username, final String password, ConnectionProviderContextImpl context, ConnectionProvider instance, OptionMap options) throws IOException, InterruptedException, KeyManagementException, NoSuchProviderException, NoSuchAlgorithmException {
    XnioSsl xnioSsl = new JsseXnioSsl(context.getXnio(), options);
    FutureResult<ConnectionHandlerFactory> result = new FutureResult<ConnectionHandlerFactory>();
    instance.connect(null, destination, options, result, new CallbackHandler() {

        public void handle(Callback[] callbacks) throws IOException, UnsupportedCallbackException {
            for (Callback cb : callbacks) {
                if (cb instanceof NameCallback) {
                    ((NameCallback) cb).setName(username);
                } else if (cb instanceof PasswordCallback) {
                    ((PasswordCallback) cb).setPassword(password != null ? password.toCharArray() : new char[0]);
                } else if (!(cb instanceof RealmCallback)) {
                    System.err.println(cb);
                    throw new UnsupportedCallbackException(cb);
                }
            }
        }
    }, xnioSsl);
    System.err.println("waiting for connection");
    IoFuture<ConnectionHandlerFactory> ioFuture = result.getIoFuture();
    Status s = ioFuture.await(5, TimeUnit.SECONDS);
    if (s == Status.FAILED) {
        System.err.println("Cannot connect");
        if (ioFuture.getException() != null) {
            ioFuture.getException().printStackTrace(System.err);
        }
    } else if (s != Status.DONE) {
        ioFuture.cancel();
        System.err.println("Connect timeout");
        System.exit(-1);
    }
    ConnectionHandlerFactory chf = ioFuture.getInterruptibly();
    return chf;
}
Also used : Status(org.xnio.IoFuture.Status) CallbackHandler(javax.security.auth.callback.CallbackHandler) XnioSsl(org.xnio.ssl.XnioSsl) JsseXnioSsl(org.xnio.ssl.JsseXnioSsl) ConnectionHandlerFactory(org.jboss.remoting3.spi.ConnectionHandlerFactory) IOException(java.io.IOException) PasswordCallback(javax.security.auth.callback.PasswordCallback) Callback(javax.security.auth.callback.Callback) RealmCallback(javax.security.sasl.RealmCallback) NameCallback(javax.security.auth.callback.NameCallback) NameCallback(javax.security.auth.callback.NameCallback) FutureResult(org.xnio.FutureResult) JsseXnioSsl(org.xnio.ssl.JsseXnioSsl) PasswordCallback(javax.security.auth.callback.PasswordCallback) UnsupportedCallbackException(javax.security.auth.callback.UnsupportedCallbackException) RealmCallback(javax.security.sasl.RealmCallback)

Aggregations

IOException (java.io.IOException)2 FutureResult (org.xnio.FutureResult)2 Status (org.xnio.IoFuture.Status)2 Callback (javax.security.auth.callback.Callback)1 CallbackHandler (javax.security.auth.callback.CallbackHandler)1 NameCallback (javax.security.auth.callback.NameCallback)1 PasswordCallback (javax.security.auth.callback.PasswordCallback)1 UnsupportedCallbackException (javax.security.auth.callback.UnsupportedCallbackException)1 RealmCallback (javax.security.sasl.RealmCallback)1 Channel (org.jboss.remoting3.Channel)1 ConnectionHandlerFactory (org.jboss.remoting3.spi.ConnectionHandlerFactory)1 JsseXnioSsl (org.xnio.ssl.JsseXnioSsl)1 XnioSsl (org.xnio.ssl.XnioSsl)1