diff --git a/tests/Makefile.am b/tests/Makefile.am index 8d390109..f8cfca3a 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -52,6 +52,7 @@ test_libguac_SOURCES = \ client/buffer_pool.c \ client/layer_pool.c \ protocol/suite.c \ + protocol/base64_decode.c \ protocol/instruction_read.c \ protocol/instruction_write.c \ protocol/nest_write.c \ diff --git a/tests/protocol/base64_decode.c b/tests/protocol/base64_decode.c new file mode 100644 index 00000000..eeb420f4 --- /dev/null +++ b/tests/protocol/base64_decode.c @@ -0,0 +1,79 @@ + +/* ***** BEGIN LICENSE BLOCK ***** + * Version: MPL 1.1/GPL 2.0/LGPL 2.1 + * + * The contents of this file are subject to the Mozilla Public License Version + * 1.1 (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * http://www.mozilla.org/MPL/ + * + * Software distributed under the License is distributed on an "AS IS" basis, + * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License + * for the specific language governing rights and limitations under the + * License. + * + * The Original Code is libguac. + * + * The Initial Developer of the Original Code is + * Michael Jumper. + * Portions created by the Initial Developer are Copyright (C) 2010 + * the Initial Developer. All Rights Reserved. + * + * Contributor(s): + * + * Alternatively, the contents of this file may be used under the terms of + * either the GNU General Public License Version 2 or later (the "GPL"), or + * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), + * in which case the provisions of the GPL or the LGPL are applicable instead + * of those above. If you wish to allow use of your version of this file only + * under the terms of either the GPL or the LGPL, and not to allow others to + * use your version of this file under the terms of the MPL, indicate your + * decision by deleting the provisions above and replace them with the notice + * and other provisions required by the GPL or the LGPL. If you do not delete + * the provisions above, a recipient may use your version of this file under + * the terms of any one of the MPL, the GPL or the LGPL. + * + * ***** END LICENSE BLOCK ***** */ + +#include +#include +#include +#include + +#include + +#include "suite.h" + + +void test_base64_decode() { + + /* Test strings */ + char test_HELLO[] = "SEVMTE8="; + char test_AVOCADO[] = "QVZPQ0FETw=="; + char test_GUACAMOLE[] = "R1VBQ0FNT0xF"; + + /* Invalid strings */ + char invalid1[] = "===="; + char invalid2[] = ""; + + /* Test one character of padding */ + CU_ASSERT_EQUAL(guac_protocol_decode_base64(test_HELLO), 5); + CU_ASSERT_STRING_EQUAL(test_HELLO, "HELLO"); + + /* Test two characters of padding */ + CU_ASSERT_EQUAL(guac_protocol_decode_base64(test_AVOCADO), 7); + CU_ASSERT_STRING_EQUAL(test_AVOCADO, "AVOCADO"); + + /* Test three characters of padding */ + CU_ASSERT_EQUAL(guac_protocol_decode_base64(test_GUACAMOLE), 9); + CU_ASSERT_STRING_EQUAL(test_GUACAMOLE, "GUACAMOLE"); + + /* Verify invalid strings stop early as expected */ + CU_ASSERT_EQUAL(guac_protocol_decode_base64(invalid1), 0); + CU_ASSERT_STRING_EQUAL(invalid1, "===="); + + CU_ASSERT_EQUAL(guac_protocol_decode_base64(invalid2), 0); + CU_ASSERT_STRING_EQUAL(invalid2, ""); + +} + diff --git a/tests/protocol/suite.c b/tests/protocol/suite.c index 925433c7..cc87d606 100644 --- a/tests/protocol/suite.c +++ b/tests/protocol/suite.c @@ -59,7 +59,8 @@ int register_protocol_suite() { /* Add tests */ if ( - CU_add_test(suite, "instruction-read", test_instruction_read) == NULL + CU_add_test(suite, "base64-decode", test_base64_decode) == NULL + || CU_add_test(suite, "instruction-read", test_instruction_read) == NULL || CU_add_test(suite, "instruction-write", test_instruction_write) == NULL || CU_add_test(suite, "nest-write", test_nest_write) == NULL ) { diff --git a/tests/protocol/suite.h b/tests/protocol/suite.h index f05d2c1f..90bbe1f2 100644 --- a/tests/protocol/suite.h +++ b/tests/protocol/suite.h @@ -48,6 +48,7 @@ int register_protocol_suite(); +void test_base64_decode(); void test_instruction_read(); void test_instruction_write(); void test_nest_write();