HeartbeatModeConverter.java

/**
 * TLS-Attacker - A Modular Penetration Testing Framework for TLS
 *
 * Copyright 2014-2016 Ruhr University Bochum / Hackmanit GmbH
 *
 * Licensed under Apache License 2.0
 * http://www.apache.org/licenses/LICENSE-2.0
 */
package de.rub.nds.tlsattacker.tls.config.converters;

import com.beust.jcommander.IStringConverter;
import com.beust.jcommander.ParameterException;
import de.rub.nds.tlsattacker.tls.constants.HeartbeatMode;
import java.util.Arrays;

/**
 * Converts a heartbeat mode string to a HeartbeatMode (for command line
 * purposes).
 * 
 * @author Juraj Somorovsky <juraj.somorovsky@rub.de>
 */
public class HeartbeatModeConverter implements IStringConverter<HeartbeatMode> {

    @Override
    public HeartbeatMode convert(String value) {

	try {
	    return HeartbeatMode.valueOf(value);
	} catch (IllegalArgumentException e) {
	    throw new ParameterException("Value " + value + " cannot be converted to a HeartbeatMode. "
		    + "Available values are: " + Arrays.toString(HeartbeatMode.values()));
	}
    }
}