/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set et sw=2 ts=2: */
/***************************************************************************
 *            test_srtp.cc
 *
 *  Mon Sep  2 14:02:16 CEST 2013
 *  Copyright 2013 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of lrtp.
 *
 * lrtp is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * lrtp 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with lrtp; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */
#include <cppunit/extensions/HelperMacros.h>

#include <stdio.h>

#include "../src/rtp.h"
#include "../src/srtp.h"

#define KEY "123456789012345678901234567890123456789012345678901234567890"
#define SSRC 1234567890

class test_srtp_class : public CppUnit::TestFixture
{
  CPPUNIT_TEST_SUITE(test_srtp_class);
	CPPUNIT_TEST(test_srtp);
	CPPUNIT_TEST_SUITE_END();

public:
	void setUp() {}
	void tearDown() {}

	void test_srtp() {
    RTP rtp;
    rtp.setSSrc(SSRC);

    char payload[] = { 0xde, 0xad, 0xbe, 0xef };
    rtp.setPayload(payload, sizeof(payload));

    char buf[MAX_RTP_PACKET_SIZE];
    size_t sz = rtp.packet(buf, sizeof(buf));

    //dump("Vanilla", buf, sz);

    {
      SRTP srtp(KEY, rtp.SSrc());
      sz = srtp.encrypt(buf, sz);
    }

    //dump("Encrypted", buf, sz);

    {
      SRTP srtp(KEY, rtp.SSrc());
      sz = srtp.decrypt(buf, sz);
    }

    //dump("Decrypted", buf, sz);

    //printf("Compare:\n");
    char buf0[MAX_RTP_PACKET_SIZE];
    size_t sz0 = rtp.packet(buf0, sizeof(buf0));

    CPPUNIT_ASSERT_EQUAL(sz, sz0);

    int err = 0;
    for(size_t i = 0; i < sz0; i++) {
      err += abs(buf0[i] - buf[i]);
    }

    CPPUNIT_ASSERT_EQUAL(0, err);
	}
};

// Registers the fixture into the 'registry'
CPPUNIT_TEST_SUITE_REGISTRATION(test_srtp_class);