mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-08 13:18:52 +00:00
Make the MIKEY message and payload objects miniobjects so that they have a GType and are refcounted. We can reuse the dispose method to clear our payload objects. Add some annotations. Implement a copy function for the MIKEY message. Fix the unit test. Fixes https://bugzilla.gnome.org/show_bug.cgi?id=732589
222 lines
7.8 KiB
C
222 lines
7.8 KiB
C
/* GStreamer unit tests for the MIKEY support library
|
|
*
|
|
* Copyright (C) 2014 Wim Taymans <wim.taymans@gmail.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
|
* Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <gst/check/gstcheck.h>
|
|
|
|
#include <gst/sdp/gstmikey.h>
|
|
|
|
GST_START_TEST (create_common)
|
|
{
|
|
GstMIKEYMessage *msg;
|
|
const guint8 test_data[] =
|
|
{ 0x01, 0x00, 0x00, 0x00, 0x12, 0x34, 0x56, 0x78, 0x00, 0x00 };
|
|
const guint8 test_data2[] =
|
|
{ 0x01, 0x12, 0x34, 0x56, 0x78, 0x00, 0x00, 0x00, 0x00,
|
|
0x02, 0x23, 0x45, 0x67, 0x89, 0x00, 0x00, 0x00, 0x01
|
|
};
|
|
GBytes *bytes;
|
|
const guint8 *data;
|
|
gsize size;
|
|
const GstMIKEYMapSRTP *mi;
|
|
GstMIKEYMapSRTP srtp;
|
|
|
|
msg = gst_mikey_message_new ();
|
|
fail_unless (msg != NULL);
|
|
|
|
fail_unless (gst_mikey_message_set_info (msg, 1, GST_MIKEY_TYPE_PSK_INIT,
|
|
FALSE, GST_MIKEY_PRF_MIKEY_1, 0x12345678, GST_MIKEY_MAP_TYPE_SRTP));
|
|
fail_unless (gst_mikey_message_get_n_cs (msg) == 0);
|
|
|
|
fail_unless (msg->version == 1);
|
|
fail_unless (msg->type == GST_MIKEY_TYPE_PSK_INIT);
|
|
fail_unless (msg->V == FALSE);
|
|
fail_unless (msg->prf_func == GST_MIKEY_PRF_MIKEY_1);
|
|
fail_unless (msg->CSB_id == 0x12345678);
|
|
fail_unless (msg->map_type == GST_MIKEY_MAP_TYPE_SRTP);
|
|
|
|
bytes = gst_mikey_message_to_bytes (msg, NULL, NULL);
|
|
data = g_bytes_get_data (bytes, &size);
|
|
fail_unless (data != NULL);
|
|
fail_unless (size == 10);
|
|
fail_unless (memcmp (data, test_data, 10) == 0);
|
|
g_bytes_unref (bytes);
|
|
|
|
fail_unless (gst_mikey_message_add_cs_srtp (msg, 1, 0x12345678, 0));
|
|
fail_unless (gst_mikey_message_get_n_cs (msg) == 1);
|
|
fail_unless (gst_mikey_message_add_cs_srtp (msg, 2, 0x23456789, 1));
|
|
fail_unless (gst_mikey_message_get_n_cs (msg) == 2);
|
|
|
|
bytes = gst_mikey_message_to_bytes (msg, NULL, NULL);
|
|
data = g_bytes_get_data (bytes, &size);
|
|
fail_unless (size == 28);
|
|
fail_unless (memcmp (data + 10, test_data2, 18) == 0);
|
|
g_bytes_unref (bytes);
|
|
|
|
fail_unless ((mi = gst_mikey_message_get_cs_srtp (msg, 0)) != NULL);
|
|
fail_unless (mi->policy == 1);
|
|
fail_unless (mi->ssrc == 0x12345678);
|
|
fail_unless (mi->roc == 0);
|
|
fail_unless ((mi = gst_mikey_message_get_cs_srtp (msg, 1)) != NULL);
|
|
fail_unless (mi->policy == 2);
|
|
fail_unless (mi->ssrc == 0x23456789);
|
|
fail_unless (mi->roc == 1);
|
|
|
|
fail_unless (gst_mikey_message_remove_cs_srtp (msg, 0));
|
|
fail_unless (gst_mikey_message_get_n_cs (msg) == 1);
|
|
fail_unless ((mi = gst_mikey_message_get_cs_srtp (msg, 0)) != NULL);
|
|
fail_unless (mi->policy == 2);
|
|
fail_unless (mi->ssrc == 0x23456789);
|
|
fail_unless (mi->roc == 1);
|
|
srtp.policy = 1;
|
|
srtp.ssrc = 0x12345678;
|
|
srtp.roc = 0;
|
|
fail_unless (gst_mikey_message_insert_cs_srtp (msg, 0, &srtp));
|
|
fail_unless ((mi = gst_mikey_message_get_cs_srtp (msg, 0)) != NULL);
|
|
fail_unless (mi->policy == 1);
|
|
fail_unless (mi->ssrc == 0x12345678);
|
|
fail_unless (mi->roc == 0);
|
|
fail_unless ((mi = gst_mikey_message_get_cs_srtp (msg, 1)) != NULL);
|
|
fail_unless (mi->policy == 2);
|
|
fail_unless (mi->ssrc == 0x23456789);
|
|
fail_unless (mi->roc == 1);
|
|
|
|
fail_unless (gst_mikey_message_remove_cs_srtp (msg, 1));
|
|
fail_unless (gst_mikey_message_get_n_cs (msg) == 1);
|
|
fail_unless (gst_mikey_message_remove_cs_srtp (msg, 0));
|
|
fail_unless (gst_mikey_message_get_n_cs (msg) == 0);
|
|
|
|
gst_mikey_message_unref (msg);
|
|
}
|
|
|
|
GST_END_TEST
|
|
GST_START_TEST (create_payloads)
|
|
{
|
|
GstMIKEYMessage *msg;
|
|
GstMIKEYPayload *payload, *kp;
|
|
const GstMIKEYPayload *cp, *cp2;
|
|
const GstMIKEYPayloadKEMAC *p;
|
|
const GstMIKEYPayloadT *pt;
|
|
const GstMIKEYPayloadKeyData *pkd;
|
|
const guint8 ntp_data[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 };
|
|
const guint8 edata[] = { 0x10, 0x20, 0x30, 0x40, 0x50, 0x60, 0x70, 0x80,
|
|
0x90, 0xa0, 0xb0, 0xc0, 0xd0, 0xe0, 0xf0, 0x10
|
|
};
|
|
GBytes *bytes;
|
|
const guint8 *data;
|
|
gsize size;
|
|
|
|
msg = gst_mikey_message_new ();
|
|
fail_unless (msg != NULL);
|
|
|
|
fail_unless (gst_mikey_message_set_info (msg, 1, GST_MIKEY_TYPE_PSK_INIT,
|
|
FALSE, GST_MIKEY_PRF_MIKEY_1, 0x12345678, GST_MIKEY_MAP_TYPE_SRTP));
|
|
fail_unless (gst_mikey_message_get_n_cs (msg) == 0);
|
|
|
|
fail_unless (gst_mikey_message_get_n_payloads (msg) == 0);
|
|
|
|
payload = gst_mikey_payload_new (GST_MIKEY_PT_T);
|
|
fail_unless (payload->type == GST_MIKEY_PT_T);
|
|
fail_unless (payload->len == sizeof (GstMIKEYPayloadT));
|
|
fail_unless (gst_mikey_payload_t_set (payload, GST_MIKEY_TS_TYPE_NTP,
|
|
ntp_data));
|
|
pt = (GstMIKEYPayloadT *) payload;
|
|
fail_unless (pt->type == GST_MIKEY_TS_TYPE_NTP);
|
|
fail_unless (memcmp (pt->ts_value, ntp_data, 8) == 0);
|
|
|
|
fail_unless (gst_mikey_message_add_payload (msg, payload));
|
|
fail_unless (payload->type == GST_MIKEY_PT_T);
|
|
fail_unless (gst_mikey_message_get_n_payloads (msg) == 1);
|
|
|
|
bytes = gst_mikey_message_to_bytes (msg, NULL, NULL);
|
|
data = g_bytes_get_data (bytes, &size);
|
|
fail_unless (data != NULL);
|
|
fail_unless (size == 20);
|
|
g_bytes_unref (bytes);
|
|
|
|
payload = gst_mikey_payload_new (GST_MIKEY_PT_KEMAC);
|
|
fail_unless (gst_mikey_payload_kemac_set (payload, GST_MIKEY_ENC_NULL,
|
|
GST_MIKEY_MAC_NULL));
|
|
/* add the edata as a key payload */
|
|
kp = gst_mikey_payload_new (GST_MIKEY_PT_KEY_DATA);
|
|
gst_mikey_payload_key_data_set_key (kp, GST_MIKEY_KD_TEK,
|
|
sizeof (edata), edata);
|
|
fail_unless (gst_mikey_payload_kemac_add_sub (payload, kp));
|
|
fail_unless (gst_mikey_message_add_payload (msg, payload));
|
|
fail_unless (gst_mikey_message_get_n_payloads (msg) == 2);
|
|
|
|
p = (GstMIKEYPayloadKEMAC *) gst_mikey_message_get_payload (msg, 1);
|
|
fail_unless (p->enc_alg == GST_MIKEY_ENC_NULL);
|
|
fail_unless (p->mac_alg == GST_MIKEY_MAC_NULL);
|
|
fail_unless (gst_mikey_payload_kemac_get_n_sub (&p->pt) == 1);
|
|
|
|
fail_unless ((cp = gst_mikey_message_get_payload (msg, 0)) != NULL);
|
|
fail_unless (cp->type == GST_MIKEY_PT_T);
|
|
fail_unless ((cp = gst_mikey_message_get_payload (msg, 1)) != NULL);
|
|
fail_unless (cp->type == GST_MIKEY_PT_KEMAC);
|
|
|
|
bytes = gst_mikey_message_to_bytes (msg, NULL, NULL);
|
|
gst_mikey_message_unref (msg);
|
|
|
|
msg = gst_mikey_message_new_from_bytes (bytes, NULL, NULL);
|
|
fail_unless (msg != NULL);
|
|
g_bytes_unref (bytes);
|
|
fail_unless (gst_mikey_message_get_n_payloads (msg) == 2);
|
|
fail_unless ((cp = gst_mikey_message_get_payload (msg, 0)) != NULL);
|
|
fail_unless (cp->type == GST_MIKEY_PT_T);
|
|
fail_unless ((cp = gst_mikey_message_get_payload (msg, 1)) != NULL);
|
|
fail_unless (cp->type == GST_MIKEY_PT_KEMAC);
|
|
|
|
fail_unless ((cp2 = gst_mikey_payload_kemac_get_sub (cp, 0)) != NULL);
|
|
fail_unless (cp2->type == GST_MIKEY_PT_KEY_DATA);
|
|
pkd = (GstMIKEYPayloadKeyData *) cp2;
|
|
|
|
fail_unless (pkd->key_type == GST_MIKEY_KD_TEK);
|
|
fail_unless (pkd->key_len == sizeof (edata));
|
|
fail_unless (memcmp (pkd->key_data, edata, sizeof (edata)) == 0);
|
|
fail_unless (pkd->salt_len == 0);
|
|
fail_unless (pkd->salt_data == 0);
|
|
fail_unless (pkd->kv_type == GST_MIKEY_KV_NULL);
|
|
|
|
|
|
gst_mikey_message_unref (msg);
|
|
}
|
|
|
|
GST_END_TEST
|
|
/*
|
|
* End of test cases
|
|
*/
|
|
static Suite *
|
|
mikey_suite (void)
|
|
{
|
|
Suite *s = suite_create ("mikey");
|
|
TCase *tc_chain = tcase_create ("mikey");
|
|
|
|
suite_add_tcase (s, tc_chain);
|
|
tcase_add_test (tc_chain, create_common);
|
|
tcase_add_test (tc_chain, create_payloads);
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (mikey);
|