org.snf4j.tls.extension.ExtensionValidator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of snf4j-tls Show documentation
The SNF4J module for TLS protocol
The newest version!
/*
 * -------------------------------- MIT License --------------------------------
 * 
 * Copyright (c) 2022-2023 SNF4J contributors
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 * -----------------------------------------------------------------------------
 */
package org.snf4j.tls.extension;

import java.util.Arrays;

import org.snf4j.tls.handshake.HandshakeType;

public class ExtensionValidator implements IExtensionValidator {

	public final static IExtensionValidator DEFAULT = new ExtensionValidator();
	
	private final static int CH = HandshakeType.CLIENT_HELLO.value();
	
	private final static int SH = HandshakeType.SERVER_HELLO.value();
	
	private final static int EE = HandshakeType.ENCRYPTED_EXTENSIONS.value();
	
	private final static int CT = HandshakeType.CERTIFICATE.value();
	
	private final static int CR = HandshakeType.CERTIFICATE_REQUEST.value();
	
	private final static int NST = HandshakeType.NEW_SESSION_TICKET.value();
	
	private final static int HRR = 6;
	
	private final static int[] MAPPING = new int[ExtensionType.KEY_SHARE.value() + 1];
	
	private final static boolean[] KNOWN_HANDSHAKES = new boolean[32];
	
	private static void addMapping(ExtensionType extensionType, int... handshakeTypes) {
		int bits = 0;
		
		for (int handshakeType: handshakeTypes) {
			bits |= 1 << handshakeType;
		}
		MAPPING[extensionType.value()] = bits;
		for (int i=0; i=0 && type < MAPPING.length) {
			if (handshakeValue >= 0 && handshakeValue < 31 && KNOWN_HANDSHAKES[handshakeValue]) {
				return (MAPPING[type] & (1 << handshakeValue)) != 0; 
			}
			else {
				return true;
			}
		}
		return true;
	}
	
	@Override
	public boolean isAllowed(ExtensionType extensionType, HandshakeType handshakeType) {
		return isAllowed(extensionType, handshakeType.value());
	}

	@Override
	public boolean isAllowedInHelloRetryRequest(ExtensionType extensionType) {
		return isAllowed(extensionType, HRR);
	}
}