diff --git a/tests/kernel-module-test.pl b/tests/kernel-module-test.pl index 1ba61a960..d0e33e9a8 100755 --- a/tests/kernel-module-test.pl +++ b/tests/kernel-module-test.pl @@ -6,6 +6,8 @@ use Socket; use Socket6; my %cmds = (noop => 1, add => 2, delete => 3, update => 4); +my %ciphers = ('null' => 1, 'aes-cm' => 2, 'aes-f8' => 3); +my %hmacs = ('null' => 1, 'hmac-sha1' => 2); $| = 1; open(F, "> /proc/mediaproxy/0/control") or die; @@ -19,31 +21,42 @@ sub mp_address { my ($fam, $addr, $port) = @_; if ($fam eq 'inet') { - return pack('i a4 a12 S S', 2, inet_aton($addr), '', $port, 0); + return pack('V a4 a12 v v', 2, inet_aton($addr), '', $port, 0); } if ($fam eq 'inet6') { - return pack('i a16 S S', 10, inet_pton(AF_INET6, $addr), $port, 0); + return pack('V a16 v v', 10, inet_pton(AF_INET6, $addr), $port, 0); } if ($fam eq '') { - return pack('i a16 S S', 0, '', 0, 0); + return pack('V a16 v v', 0, '', 0, 0); } die; } +sub mp_srtp { + my ($h) = @_; + no warnings; + return pack('VV a16 a16 QQ VV', $ciphers{$$h{cipher}}, $hmacs{$$h{hmac}}, + @$h{qw(master_key master_salt mki last_index auth_tag_len mki_len)}); + use warnings; +} sub mediaproxy_message { my ($cmd, $target_port, $src_addr_family, $src_addr_addr, $src_addr_port, $dst_addr_family, $dst_addr_addr, $dst_addr_port, $mirror_addr_family, $mirror_addr_addr, $mirror_addr_port, - $tos) = @_; + $tos, $decrypt, $encrypt) = @_; my $ret = ''; - $ret .= pack('I SS', $cmds{$cmd}, $target_port, 0); + # amd64 alignment + $ret .= pack('VV vv', $cmds{$cmd}, 0, $target_port, 0); $ret .= mp_address($src_addr_family, $src_addr_addr, $src_addr_port); $ret .= mp_address($dst_addr_family, $dst_addr_addr, $dst_addr_port); $ret .= mp_address($mirror_addr_family, $mirror_addr_addr, $mirror_addr_port); - $ret .= pack('C CS', $tos, 0, 0); + $ret .= pack('V', 0); + $ret .= mp_srtp($decrypt); + $ret .= mp_srtp($encrypt); + $ret .= pack('C CSI', $tos, 0, 0, 0); } my $sleep = 5; @@ -52,41 +65,43 @@ my $sleep = 5; my @src = qw(inet6 2a00:4600:1:0:a00:27ff:feb0:f7fe); my @dst = qw(inet6 2a00:4600:1:0:6884:adff:fe98:6ac5); my @nul = ('', '', ''); +my $dec = {cipher => 'null', hmac => 'null'}; +my $enc = {cipher => 'null', hmac => 'null'}; print("add 9876 -> 1234/6543\n"); -syswrite(F, mediaproxy_message('add', 9876, @src, 1234, @dst, 6543, @nul, 184)); +syswrite(F, mediaproxy_message('add', 9876, @src, 1234, @dst, 6543, @nul, 184, $dec, $enc)); sleep($sleep); print("add fail\n"); -syswrite(F, mediaproxy_message('add', 9876, @src, 1234, @dst, 6543, @dst, 6789, 184)); +syswrite(F, mediaproxy_message('add', 9876, @src, 1234, @dst, 6543, @dst, 6789, 184, $dec, $enc)); sleep($sleep); print("update 9876 -> 1234/6543 & 6789\n"); -syswrite(F, mediaproxy_message('update', 9876, @src, 1234, @dst, 6543, @dst, 6789, 184)); +syswrite(F, mediaproxy_message('update', 9876, @src, 1234, @dst, 6543, @dst, 6789, 184, $dec, $enc)); sleep($sleep); print("update 9876 -> 2345/7890 & 4321\n"); -syswrite(F, mediaproxy_message('update', 9876, @src, 2345, @dst, 7890, @dst, 4321, 184)); +syswrite(F, mediaproxy_message('update', 9876, @src, 2345, @dst, 7890, @dst, 4321, 184, $dec, $enc)); sleep($sleep); print("add fail\n"); -syswrite(F, mediaproxy_message('add', 9876, @src, 1234, @dst, 6543, @dst, 6789, 184)); +syswrite(F, mediaproxy_message('add', 9876, @src, 1234, @dst, 6543, @dst, 6789, 184, $dec, $enc)); sleep($sleep); print("update 9876 -> 1234/6543\n"); -syswrite(F, mediaproxy_message('update', 9876, @src, 1234, @dst, 6543, @nul, 184)); +syswrite(F, mediaproxy_message('update', 9876, @src, 1234, @dst, 6543, @nul, 184, $dec, $enc)); sleep($sleep); print("delete\n"); -syswrite(F, mediaproxy_message('delete', 9876, @nul, @nul, @nul, 0)); +syswrite(F, mediaproxy_message('delete', 9876, @nul, @nul, @nul, 0, $dec, $enc)); sleep($sleep); print("delete fail\n"); -syswrite(F, mediaproxy_message('delete', 9876, @nul, @nul, @nul, 0)); +syswrite(F, mediaproxy_message('delete', 9876, @nul, @nul, @nul, 0, $dec, $enc)); sleep($sleep); print("update fail\n"); -syswrite(F, mediaproxy_message('update', 9876, @src, 1234, @dst, 6543, @nul, 184)); +syswrite(F, mediaproxy_message('update', 9876, @src, 1234, @dst, 6543, @nul, 184, $dec, $enc)); sleep($sleep); close(F);