/*
 * This file is part of AirReceiver.
 *
 * AirReceiver is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.

 * AirReceiver is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with AirReceiver.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.dyndns.jkiddo.raop.server.airreceiver;

import javax.crypto.Cipher;
import javax.crypto.SecretKey;
import javax.crypto.spec.IvParameterSpec;

import org.jboss.netty.buffer.*;
import org.jboss.netty.channel.*;
import org.jboss.netty.handler.codec.oneone.OneToOneDecoder;

/**
 * De-crypt AES encoded audio data
 */
public class RaopRtpAudioDecryptionHandler extends OneToOneDecoder
{
	/**
	 * The AES cipher. We request no padding because RAOP/AirTunes only encrypts full block anyway and leaves the trailing byte unencrypted
	 */
	private final Cipher m_aesCipher = AirTunesCrytography.getCipher("AES/CBC/NoPadding");

	/**
	 * AES key
	 */
	private final SecretKey m_aesKey;

	/**
	 * AES initialization vector
	 */
	private final IvParameterSpec m_aesIv;

	public RaopRtpAudioDecryptionHandler(final SecretKey aesKey, final IvParameterSpec aesIv)
	{
		m_aesKey = aesKey;
		m_aesIv = aesIv;
	}

	@Override
	protected synchronized Object decode(final ChannelHandlerContext ctx, final Channel channel, final Object msg) throws Exception
	{
		if(msg instanceof RaopRtpPacket.Audio)
		{
			final RaopRtpPacket.Audio audioPacket = (RaopRtpPacket.Audio) msg;
			final ChannelBuffer audioPayload = audioPacket.getPayload();

			/*
			 * Cipher is restarted for every packet. We simply overwrite the encrypted data with the corresponding plain text
			 */
			m_aesCipher.init(Cipher.DECRYPT_MODE, m_aesKey, m_aesIv);
			for(int i = 0; (i + 16) <= audioPayload.capacity(); i += 16)
			{
				byte[] block = new byte[16];
				audioPayload.getBytes(i, block);
				block = m_aesCipher.update(block);
				audioPayload.setBytes(i, block);
			}
		}

		return msg;
	}
}