1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
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);
}
|
