MD5ChannelOptions.java

/*
 * Copyright (c) 2013 Robert Varga. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.tcpmd5.nio;

import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Sets;

import java.io.IOException;
import java.net.SocketOption;
import java.nio.channels.NetworkChannel;
import java.util.Set;

import org.opendaylight.tcpmd5.api.KeyAccess;
import org.opendaylight.tcpmd5.api.KeyAccessFactory;
import org.opendaylight.tcpmd5.api.KeyMapping;
import org.opendaylight.tcpmd5.api.MD5SocketOptions;

/**
 * Channel option wrapper which unifies the underlying NetworkChannel with the TCP MD5 Signature option if the channel
 * is supported by the JNI library.
 */
final class MD5ChannelOptions {
    private static final Set<SocketOption<?>> MD5_OPTIONS = ImmutableSet.<SocketOption<?>> of(MD5SocketOptions.TCP_MD5SIG);
    private final NetworkChannel ch;
    private final KeyAccess access;

    private MD5ChannelOptions(final NetworkChannel ch, final KeyAccess access) {
        this.ch = Preconditions.checkNotNull(ch);
        this.access = access;
    }

    static MD5ChannelOptions create(final KeyAccessFactory keyAccessFactory, final NetworkChannel ch) {
        final KeyAccess access = keyAccessFactory.getKeyAccess(Preconditions.checkNotNull(ch));
        return new MD5ChannelOptions(ch, access);
    }

    @SuppressWarnings("unchecked")
    public <T> T getOption(final SocketOption<T> name) throws IOException {
        if (access != null && name.equals(MD5SocketOptions.TCP_MD5SIG)) {
            return (T) access.getKeys();
        }

        return ch.getOption(name);
    }

    public <T> void setOption(final SocketOption<T> name, final T value) throws IOException {
        if (access != null && name.equals(MD5SocketOptions.TCP_MD5SIG)) {
            access.setKeys((KeyMapping) value);
        } else {
            ch.setOption(name, value);
        }
    }

    public Set<SocketOption<?>> supportedOptions() {
        if (access != null) {
            return Sets.union(MD5_OPTIONS, ch.supportedOptions());
        } else {
            return ch.supportedOptions();
        }
    }
}