Merge lp:~tangent-org/libmemcached/1.0-build/ Build: jenkins-Libmemcached-192
[awesomized/libmemcached] / libtest / vchar.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Data Differential YATL (i.e. libtest) library
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 *
35 */
36
37 #include "libtest/yatlcon.h"
38 #include <libtest/common.h>
39
40 /* Use this for string generation */
41 static const char ALPHANUMERICS[]=
42 "0123456789ABCDEFGHIJKLMNOPQRSTWXYZabcdefghijklmnopqrstuvwxyz";
43
44 #define ALPHANUMERICS_SIZE (sizeof(ALPHANUMERICS)-1)
45
46 static size_t get_alpha_num(void)
47 {
48 return (size_t)random() % ALPHANUMERICS_SIZE;
49 }
50
51 namespace libtest {
52
53 static std::string printer(const char *str, size_t length)
54 {
55 std::ostringstream buf;
56 for (size_t x= 0; x < length; x++)
57 {
58 if (isprint(str[x]))
59 {
60 buf << str[x];
61 }
62 else
63 {
64 buf << "(" << int(str[x]) << ")";
65 }
66 }
67
68 return buf.str();
69 }
70
71 namespace vchar {
72
73 int compare(libtest::vchar_t& arg, const char *str, size_t length)
74 {
75 if (arg.size() == length and (memcmp(&arg[0], str, length) == 0))
76 {
77 return 0;
78 }
79 else if (arg.size() > length)
80 {
81 return 1;
82 }
83
84 return -1;
85 }
86
87 void make(libtest::vchar_t& arg)
88 {
89 size_t length= rand() % 1024;
90 make(arg, length);
91 }
92
93 void make(libtest::vchar_t& arg, size_t length)
94 {
95 arg.reserve(length);
96 for (uint32_t x= 0; x < length; ++x)
97 {
98 arg.push_back(ALPHANUMERICS[get_alpha_num()]);
99 }
100 }
101
102 void chomp(libtest::vchar_t& arg)
103 {
104 while(arg.size())
105 {
106 if (arg.back() == 0)
107 {
108 arg.pop_back();
109 }
110 else
111 {
112 break;
113 }
114 }
115 }
116
117 void append(libtest::vchar_ptr_t& arg, const char* ptr)
118 {
119 if (ptr)
120 {
121 char* new_ptr= strdup(ptr);
122 if (new_ptr == NULL)
123 {
124 fatal_message("UNABLE to allocate %s(%p)", ptr, ptr);
125 }
126
127 arg.push_back(new_ptr);
128 }
129 }
130
131 void append(libtest::vchar_t& arg, const char* ptr)
132 {
133 if (ptr)
134 {
135 size_t length= strlen(ptr);
136 ASSERT_TRUE(length);
137 arg.reserve(length);
138 do
139 {
140 arg.push_back(*ptr);
141 ++ptr;
142 } while (*ptr);
143
144 arg.push_back(0);
145 }
146 }
147
148 } // namespace vchar
149
150 void make_vector(libtest::vchar_t& arg, const char *str, size_t length)
151 {
152 arg.resize(length);
153 memcpy(&arg[0], str, length);
154 }
155
156 std::ostream& operator<<(std::ostream& output, const libtest::vchar_t& arg)
157 {
158 std::string tmp= libtest::printer(&arg[0], arg.size());
159 output << tmp << "[" << arg.size() << "]";
160
161 return output;
162 }
163
164 } // namespace libtest