diff options
Diffstat (limited to '')
-rw-r--r-- | src/test/base64.cc | 102 |
1 files changed, 102 insertions, 0 deletions
diff --git a/src/test/base64.cc b/src/test/base64.cc new file mode 100644 index 000000000..48035118a --- /dev/null +++ b/src/test/base64.cc @@ -0,0 +1,102 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2011 Dreamhost + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ + +#include "common/armor.h" +#include "common/config.h" +#include "include/buffer.h" +#include "include/encoding.h" + +#include "gtest/gtest.h" + +using namespace std; + +TEST(RoundTrip, SimpleRoundTrip) { + static const int OUT_LEN = 4096; + const char * const original = "abracadabra"; + const char * const correctly_encoded = "YWJyYWNhZGFicmE="; + char out[OUT_LEN]; + memset(out, 0, sizeof(out)); + int alen = ceph_armor(out, out + OUT_LEN, original, original + strlen(original)); + ASSERT_STREQ(correctly_encoded, out); + + char out2[OUT_LEN]; + memset(out2, 0, sizeof(out2)); + ceph_unarmor(out2, out2 + OUT_LEN, out, out + alen); + ASSERT_STREQ(original, out2); +} + +TEST(RoundTrip, RandomRoundTrips) { + static const int IN_MAX = 1024; + static const int OUT_MAX = 4096; + static const int ITERS = 1000; + for (int i = 0; i < ITERS; ++i) { + unsigned int seed = i; + int in_len = rand_r(&seed) % IN_MAX; + + char in[IN_MAX]; + memset(in, 0, sizeof(in)); + for (int j = 0; j < in_len; ++j) { + in[j] = rand_r(&seed) % 0xff; + } + char out[OUT_MAX]; + memset(out, 0, sizeof(out)); + int alen = ceph_armor(out, out + OUT_MAX, in, in + in_len); + ASSERT_GE(alen, 0); + + char decoded[IN_MAX]; + memset(decoded, 0, sizeof(decoded)); + int blen = ceph_unarmor(decoded, decoded + IN_MAX, out, out + alen); + ASSERT_GE(blen, 0); + + ASSERT_EQ(memcmp(in, decoded, in_len), 0); + } +} + +TEST(EdgeCase, EndsInNewline) { + static const int OUT_MAX = 4096; + + char b64[] = + "aaaa\n"; + + char decoded[OUT_MAX]; + memset(decoded, 0, sizeof(decoded)); + int blen = ceph_unarmor(decoded, decoded + OUT_MAX, b64, b64 + sizeof(b64)-1); + ASSERT_GE(blen, 0); +} + +TEST(FuzzEncoding, BadDecode1) { + static const int OUT_LEN = 4096; + const char * const bad_encoded = "FAKEBASE64 foo"; + char out[OUT_LEN]; + memset(out, 0, sizeof(out)); + int alen = ceph_unarmor(out, out + OUT_LEN, bad_encoded, bad_encoded + strlen(bad_encoded)); + ASSERT_LT(alen, 0); +} + +TEST(FuzzEncoding, BadDecode2) { + string str("FAKEBASE64 foo"); + bool failed = false; + try { + bufferlist bl; + bl.append(str); + + bufferlist cl; + cl.decode_base64(bl); + cl.hexdump(std::cerr); + } + catch (const buffer::error &err) { + failed = true; + } + ASSERT_EQ(failed, true); +} |