|
|
1 | /* |
|
|
2 | * This file is part of Deliantra, the Roguelike Realtime MMORPG. |
|
|
3 | * |
|
|
4 | * Copyright (©) 2005,2006,2007,2008 Marc Alexander Lehmann / Robin Redeker / the Deliantra team |
|
|
5 | * |
|
|
6 | * Deliantra is free software: you can redistribute it and/or modify |
|
|
7 | * it under the terms of the GNU General Public License as published by |
|
|
8 | * the Free Software Foundation, either version 3 of the License, or |
|
|
9 | * (at your option) any later version. |
|
|
10 | * |
|
|
11 | * This program is distributed in the hope that it will be useful, |
|
|
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
14 | * GNU General Public License for more details. |
|
|
15 | * |
|
|
16 | * You should have received a copy of the GNU General Public License |
|
|
17 | * along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
18 | * |
|
|
19 | * The authors can be reached via e-mail to <support@deliantra.net> |
|
|
20 | */ |
|
|
21 | |
1 | #ifndef SHSTR_H__ |
22 | #ifndef SHSTR_H__ |
2 | #define SHSTR_H__ |
23 | #define SHSTR_H__ |
3 | |
24 | |
|
|
25 | #include <cstring> |
|
|
26 | #include <sstream> |
|
|
27 | |
4 | #include "util.h" |
28 | #include "traits.h" |
|
|
29 | |
|
|
30 | extern size_t shstr_alloc; |
5 | |
31 | |
6 | extern int buf_overflow (const char *buf1, const char *buf2, int bufsize); |
32 | extern int buf_overflow (const char *buf1, const char *buf2, int bufsize); |
7 | |
33 | |
|
|
34 | template<int size> |
8 | struct shstr |
35 | struct shstr_vec |
9 | { |
36 | { |
10 | static const char *null; |
37 | uint32_t len; |
|
|
38 | uint32_t refcnt; |
|
|
39 | char string[size]; |
|
|
40 | }; |
|
|
41 | |
|
|
42 | // this class is a non-refcounted shared string |
|
|
43 | // it cannot be used to create or store shared strings, but |
|
|
44 | // it can be used to apss shared strings around, i.e. as function arguments |
|
|
45 | // or return values. their lifetime must not span a gc () call, i.e. |
|
|
46 | // they are only valid as temporary values within the same server tick. |
|
|
47 | struct shstr_tmp |
|
|
48 | { |
|
|
49 | static shstr_vec<sizeof ("(null)")> nullvec; |
|
|
50 | static const char *null () { return nullvec.string; } // this is the null pointer value |
11 | |
51 | |
12 | const char *s; |
52 | const char *s; |
13 | |
53 | |
14 | static int &refcnt (const char *s) |
|
|
15 | { |
|
|
16 | return *((int *)s - 1); |
|
|
17 | } |
|
|
18 | |
|
|
19 | static int &length (const char *s) |
54 | static unsigned int &length (const char *s) |
20 | { |
55 | { |
21 | return *((int *)s - 2); |
56 | return *((unsigned int *)s - 2); |
22 | } |
57 | } |
23 | |
58 | |
24 | int &refcnt () const |
59 | int length () const |
|
|
60 | { |
|
|
61 | return length (s); |
|
|
62 | } |
|
|
63 | |
|
|
64 | // returns whether this shared string begins with the given prefix, |
|
|
65 | // used mainly for searches when users give only the start of a name. |
|
|
66 | bool starts_with (const char *prefix) const |
|
|
67 | { |
|
|
68 | int plen = strlen (prefix); |
|
|
69 | |
|
|
70 | return length () >= plen && !strncasecmp (s, prefix, plen); |
|
|
71 | } |
|
|
72 | |
|
|
73 | bool contains (const char *substring) const |
|
|
74 | { |
|
|
75 | return strstr (s, substring); |
|
|
76 | } |
|
|
77 | |
|
|
78 | shstr_tmp () |
|
|
79 | : s (null ()) |
|
|
80 | { |
|
|
81 | } |
|
|
82 | |
|
|
83 | shstr_tmp (const shstr_tmp &sh) |
|
|
84 | : s (sh.s) |
|
|
85 | { |
|
|
86 | } |
|
|
87 | |
|
|
88 | shstr_tmp operator =(const shstr_tmp &sh) |
|
|
89 | { |
|
|
90 | s = sh.s; |
|
|
91 | |
|
|
92 | return *this; |
|
|
93 | } |
|
|
94 | |
|
|
95 | // this is used for informational messages and the like |
|
|
96 | const char *operator &() const { return s; } |
|
|
97 | |
|
|
98 | operator const char *() const { return s == null () ? 0 : s; } |
|
|
99 | |
|
|
100 | protected: |
|
|
101 | // dummy is there so it isn't used as type converter accidentally |
|
|
102 | shstr_tmp (int dummy, const char *s) |
|
|
103 | : s(s) |
|
|
104 | { |
|
|
105 | } |
|
|
106 | }; |
|
|
107 | |
|
|
108 | inline bool operator ==(const shstr_tmp &a, const shstr_tmp &b) |
|
|
109 | { |
|
|
110 | return a.s == b.s; |
|
|
111 | } |
|
|
112 | |
|
|
113 | inline bool operator !=(const shstr_tmp &a, const shstr_tmp &b) |
|
|
114 | { |
|
|
115 | return a.s != b.s; |
|
|
116 | } |
|
|
117 | |
|
|
118 | inline int strlen (const shstr_tmp &sh) |
|
|
119 | { |
|
|
120 | return sh.length (); |
|
|
121 | } |
|
|
122 | |
|
|
123 | // undefined external reference to catch people using strcmp when they shouldn't |
|
|
124 | int strcmp (const shstr_tmp &a, const shstr_tmp &b); |
|
|
125 | |
|
|
126 | static std::ostream &operator <<(std::ostream &o, shstr_tmp sh) |
|
|
127 | { |
|
|
128 | o.write (sh.s, sh.length ()); |
|
|
129 | |
|
|
130 | return o; |
|
|
131 | } |
|
|
132 | |
|
|
133 | struct shstr : shstr_tmp |
|
|
134 | { |
|
|
135 | static unsigned int &refcnt (const char *s) |
|
|
136 | { |
|
|
137 | return *((unsigned int *)s - 1); |
|
|
138 | } |
|
|
139 | |
|
|
140 | unsigned int &refcnt () const |
25 | { |
141 | { |
26 | return refcnt (s); |
142 | return refcnt (s); |
27 | } |
143 | } |
28 | |
144 | |
29 | int length () const |
145 | shstr () |
30 | { |
146 | { |
31 | return length (s); |
|
|
32 | } |
147 | } |
33 | |
148 | |
34 | static const char *find (const char *s); |
149 | static const char *find (const char *s); |
35 | static const char *intern (const char *s); |
150 | static const char *intern (const char *s); |
36 | |
151 | |
37 | static void gc (); // garbage collect a few strings |
152 | static void gc (); // garbage collect a few strings |
38 | |
153 | |
39 | // this is used for informational messages and the like |
|
|
40 | const char *operator &() const { return s; } |
|
|
41 | |
|
|
42 | const char &operator [](int i) const { return s[i]; } |
|
|
43 | operator const char *() const { return s == null ? 0 : s; } |
|
|
44 | |
|
|
45 | shstr () |
|
|
46 | : s (null) |
|
|
47 | { |
|
|
48 | } |
|
|
49 | |
|
|
50 | shstr (const shstr &sh) |
154 | shstr (const shstr &sh) |
51 | : s (sh.s) |
155 | : shstr_tmp (sh) |
52 | { |
156 | { |
53 | ++refcnt (); |
157 | ++refcnt (); |
54 | } |
158 | } |
55 | |
159 | |
|
|
160 | shstr (const shstr_tmp &sh) |
|
|
161 | : shstr_tmp (sh) |
|
|
162 | { |
|
|
163 | ++refcnt (); |
|
|
164 | } |
|
|
165 | |
56 | explicit shstr (const char *s) |
166 | explicit shstr (const char *str) |
57 | : s (intern (s)) |
167 | : shstr_tmp (0, is_constant (str) && !str ? null () : intern (str)) |
58 | { |
168 | { |
59 | } |
169 | } |
60 | |
170 | |
61 | ~shstr () |
171 | ~shstr () |
62 | { |
172 | { |
63 | --refcnt (); |
173 | --refcnt (); |
64 | } |
174 | } |
65 | |
175 | |
|
|
176 | using shstr_tmp::operator &; |
|
|
177 | using shstr_tmp::operator const char *; |
|
|
178 | |
|
|
179 | // (note: not the copy constructor) |
66 | const shstr &operator =(const shstr &sh) |
180 | shstr &operator =(const shstr_tmp &sh) |
67 | { |
181 | { |
68 | --refcnt (); |
182 | --refcnt (); |
69 | s = sh.s; |
183 | s = sh.s; |
70 | ++refcnt (); |
184 | ++refcnt (); |
71 | |
185 | |
72 | return *this; |
186 | return *this; |
73 | } |
187 | } |
74 | |
188 | |
|
|
189 | // here it comes |
|
|
190 | shstr &operator =(const shstr &sh) |
|
|
191 | { |
|
|
192 | return (*this) = (shstr_tmp)sh; |
|
|
193 | } |
|
|
194 | |
|
|
195 | // shstr_tmp doesn't have this one |
75 | const shstr &operator =(const char *str) |
196 | shstr &operator =(const char *str) |
76 | { |
197 | { |
77 | --refcnt (); |
198 | --refcnt (); |
78 | |
199 | s = is_constant (str) && !str ? null () : intern (str); |
79 | // this optimises the important case of str == constant 0 |
|
|
80 | if (is_constant (str)) |
|
|
81 | s = str ? intern (str) : null; |
|
|
82 | else |
|
|
83 | s = intern (str); |
|
|
84 | |
200 | |
85 | return *this; |
201 | return *this; |
86 | } |
202 | } |
87 | |
|
|
88 | bool operator ==(const shstr &b) |
|
|
89 | { |
|
|
90 | return s == b.s; |
|
|
91 | } |
|
|
92 | |
|
|
93 | bool operator !=(const shstr &b) |
|
|
94 | { |
|
|
95 | return !(*this == b); |
|
|
96 | } |
|
|
97 | }; |
203 | }; |
98 | |
204 | |
99 | inline int strlen (const shstr &sh) |
|
|
100 | { |
|
|
101 | return sh.length (); |
|
|
102 | } |
|
|
103 | |
|
|
104 | // only good for mass comparisons to shstr objects |
205 | // only good for mass comparisons to shstr objects, or very |
|
|
206 | // temporary passing, basically a non-refcounted shstr |
105 | struct shstr_cmp |
207 | struct shstr_cmp |
106 | { |
208 | { |
107 | const char *s; |
209 | const char *s; |
108 | |
210 | |
109 | explicit shstr_cmp (const char *s) |
211 | explicit shstr_cmp (const char *str) |
110 | : s (shstr::find (s)) |
212 | : s (shstr::find (str)) |
111 | { |
213 | { |
112 | } |
214 | } |
|
|
215 | |
|
|
216 | shstr_cmp (shstr_tmp sh) |
|
|
217 | : s (sh.s) |
|
|
218 | { |
|
|
219 | } |
|
|
220 | |
|
|
221 | // this is used for informational messages and the like |
|
|
222 | const char *operator &() const { return s; } |
113 | |
223 | |
114 | operator const char *() const { return s; } |
224 | operator const char *() const { return s; } |
115 | }; |
225 | }; |
116 | |
226 | |
117 | inline bool operator ==(const shstr_cmp &a, const shstr &b) |
227 | inline bool operator ==(const shstr_cmp &a, const shstr_tmp &b) |
118 | { |
228 | { |
119 | return a.s == b.s; |
229 | return a.s == b.s; |
120 | } |
230 | } |
121 | |
231 | |
122 | inline bool operator ==(const shstr &a, const shstr_cmp &b) |
232 | inline bool operator ==(const shstr_tmp &a, const shstr_cmp &b) |
123 | { |
233 | { |
124 | return b == a; |
234 | return a.s == b.s; |
125 | } |
235 | } |
126 | |
236 | |
127 | extern const shstr undead_name; /* Used in hit_player() in main.c */ |
237 | struct shstr_const : shstr_tmp |
|
|
238 | { |
|
|
239 | shstr_const (const char *s) |
|
|
240 | : shstr_tmp (0, s) |
|
|
241 | { |
|
|
242 | } |
|
|
243 | }; |
|
|
244 | |
|
|
245 | #define def(str) extern const shstr_const shstr_ ## str; |
|
|
246 | # include "shstrinc.h" |
|
|
247 | #undef def |
128 | |
248 | |
129 | #endif |
249 | #endif |
130 | |
250 | |