Fix bad if path
[m6w6/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 int random_alpha_num(void)
54 {
55 return ALPHANUMERICS[get_alpha_num()];
56 }
57
58 static std::string printer(const char *str, size_t length)
59 {
60 std::ostringstream buf;
61 for (size_t x= 0; x < length; x++)
62 {
63 if (isprint(str[x]))
64 {
65 buf << str[x];
66 }
67 else
68 {
69 buf << "(" << int(str[x]) << ")";
70 }
71 }
72
73 return buf.str();
74 }
75
76 namespace vchar {
77
78 int compare(libtest::vchar_t& arg, const char *str, size_t length)
79 {
80 if (arg.size() == length and (memcmp(&arg[0], str, length) == 0))
81 {
82 return 0;
83 }
84 else if (arg.size() > length)
85 {
86 return 1;
87 }
88
89 return -1;
90 }
91
92 void make(libtest::vchar_t& arg)
93 {
94 size_t length= rand() % 1024;
95 make(arg, length);
96 }
97
98 void make(libtest::vchar_t& arg, size_t length)
99 {
100 arg.reserve(length);
101 for (uint32_t x= 0; x < length; ++x)
102 {
103 arg.push_back(ALPHANUMERICS[get_alpha_num()]);
104 }
105 }
106
107 void chomp(libtest::vchar_t& arg)
108 {
109 while(arg.size())
110 {
111 if (arg.back() == 0)
112 {
113 arg.pop_back();
114 }
115 else
116 {
117 break;
118 }
119 }
120 }
121
122 void append(libtest::vchar_ptr_t& arg, const char* ptr)
123 {
124 if (ptr)
125 {
126 char* new_ptr= strdup(ptr);
127 if (new_ptr == NULL)
128 {
129 FATAL("UNABLE to allocate %s(%p)", ptr, ptr);
130 }
131
132 arg.push_back(new_ptr);
133 }
134 }
135
136 void append(libtest::vchar_t& arg, const char* ptr)
137 {
138 if (ptr)
139 {
140 size_t length= strlen(ptr);
141 ASSERT_TRUE(length);
142 arg.reserve(length);
143 do
144 {
145 arg.push_back(*ptr);
146 ++ptr;
147 } while (*ptr);
148
149 arg.push_back(0);
150 }
151 }
152
153 } // namespace vchar
154
155 void make_vector(libtest::vchar_t& arg, const char *str, size_t length)
156 {
157 arg.resize(length);
158 memcpy(&arg[0], str, length);
159 }
160
161 std::ostream& operator<<(std::ostream& output, const libtest::vchar_t& arg)
162 {
163 std::string tmp= libtest::printer(&arg[0], arg.size());
164 output << tmp << "[" << arg.size() << "]";
165
166 return output;
167 }
168
169 } // namespace libtest