1 files changed, 87 insertions, 0 deletions
diff --git a/test/test_srtp.cc b/test/test_srtp.cc
new file mode 100644
index 0000000..962c18f
--- /dev/null
+++ b/test/test_srtp.cc
@@ -0,0 +1,87 @@
+/* -*- 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 <stdio.h>
+
+#include "../src/rtp.h"
+#include "../src/srtp.h"
+
+#define KEY "123456789012345678901234567890123456789012345678901234567890"
+#define SSRC 1234567890
+
+void dump(const char *title, const char *buf, size_t size)
+{
+  printf("%12s: ", title);
+  for(int i = 0; i < size; i++) {
+    if(i % 8 == 0) printf(" ");
+    printf("%02x ", (unsigned char)*buf++);
+  }
+  printf("\n");
+}
+
+int main()
+{
+  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));
+  if(sz0 != sz) printf("Sizes differ (%d %d)...\n", sz0, sz);
+  unsigned int sum = 0;
+  for(int i = 0; i < sz0; i++) {
+    sum += abs(buf0[i] - buf[i]);
+  }
+  if(sum) printf("NOT EQUAL! diff = %d\n", sum);
+  else printf("Vanilla == Decrypted\n");
+
+  return 0;
+}
+