|
|
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 | |
|
|
101 | inline bool operator ==(const shstr_tmp &a, const shstr_tmp &b) |
|
|
102 | { |
|
|
103 | return a.s == b.s; |
|
|
104 | } |
|
|
105 | |
|
|
106 | inline bool operator !=(const shstr_tmp &a, const shstr_tmp &b) |
|
|
107 | { |
|
|
108 | return a.s != b.s; |
|
|
109 | } |
|
|
110 | |
|
|
111 | inline int strlen (const shstr_tmp &sh) |
|
|
112 | { |
|
|
113 | return sh.length (); |
|
|
114 | } |
|
|
115 | |
|
|
116 | // undefined external reference to catch people using strcmp when they shouldn't |
|
|
117 | int strcmp (const shstr_tmp &a, const shstr_tmp &b); |
|
|
118 | |
|
|
119 | static std::ostream &operator <<(std::ostream &o, shstr_tmp sh) |
|
|
120 | { |
|
|
121 | o.write (sh.s, sh.length ()); |
|
|
122 | |
|
|
123 | return o; |
|
|
124 | } |
|
|
125 | |
|
|
126 | struct shstr : shstr_tmp |
|
|
127 | { |
|
|
128 | static unsigned int &refcnt (const char *s) |
|
|
129 | { |
|
|
130 | return *((unsigned int *)s - 1); |
|
|
131 | } |
|
|
132 | |
|
|
133 | unsigned int &refcnt () const |
25 | { |
134 | { |
26 | return refcnt (s); |
135 | return refcnt (s); |
27 | } |
136 | } |
28 | |
137 | |
29 | int length () const |
138 | shstr () |
30 | { |
139 | { |
31 | return length (s); |
|
|
32 | } |
140 | } |
33 | |
141 | |
34 | static const char *find (const char *s); |
142 | static const char *find (const char *s); |
35 | static const char *intern (const char *s); |
143 | static const char *intern (const char *s); |
36 | |
144 | |
37 | static void gc (); // garbage collect a few strings |
145 | static void gc (); // garbage collect a few strings |
38 | |
146 | |
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) |
147 | shstr (const shstr &sh) |
51 | : s (sh.s) |
148 | : shstr_tmp (sh) |
52 | { |
149 | { |
53 | ++refcnt (); |
150 | ++refcnt (); |
54 | } |
151 | } |
55 | |
152 | |
|
|
153 | shstr (const shstr_tmp &sh) |
|
|
154 | : shstr_tmp (sh) |
|
|
155 | { |
|
|
156 | ++refcnt (); |
|
|
157 | } |
|
|
158 | |
56 | explicit shstr (const char *s) |
159 | explicit shstr (const char *str) |
57 | : s (intern (s)) |
|
|
58 | { |
160 | { |
|
|
161 | s = is_constant (str) && !str ? null () : intern (str); |
59 | } |
162 | } |
60 | |
163 | |
61 | ~shstr () |
164 | ~shstr () |
62 | { |
165 | { |
63 | --refcnt (); |
166 | --refcnt (); |
64 | } |
167 | } |
65 | |
168 | |
|
|
169 | using shstr_tmp::operator &; |
|
|
170 | using shstr_tmp::operator const char *; |
|
|
171 | |
|
|
172 | // (note: not the copy constructor) |
66 | const shstr &operator =(const shstr &sh) |
173 | shstr &operator =(const shstr_tmp &sh) |
67 | { |
174 | { |
68 | --refcnt (); |
175 | --refcnt (); |
69 | s = sh.s; |
176 | s = sh.s; |
70 | ++refcnt (); |
177 | ++refcnt (); |
71 | |
178 | |
72 | return *this; |
179 | return *this; |
73 | } |
180 | } |
74 | |
181 | |
|
|
182 | // here it comes |
|
|
183 | shstr &operator =(const shstr &sh) |
|
|
184 | { |
|
|
185 | return (*this) = (shstr_tmp)sh; |
|
|
186 | } |
|
|
187 | |
|
|
188 | // shstr_tmp doesn't have this one |
75 | const shstr &operator =(const char *str) |
189 | shstr &operator =(const char *str) |
76 | { |
190 | { |
77 | --refcnt (); |
191 | --refcnt (); |
78 | |
192 | 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 | |
193 | |
85 | return *this; |
194 | return *this; |
86 | } |
195 | } |
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 | }; |
196 | }; |
98 | |
197 | |
99 | inline int strlen (const shstr &sh) |
|
|
100 | { |
|
|
101 | return sh.length (); |
|
|
102 | } |
|
|
103 | |
|
|
104 | // only good for mass comparisons to shstr objects |
198 | // only good for mass comparisons to shstr objects, or very |
|
|
199 | // temporary passing, basically a non-refcounted shstr |
105 | struct shstr_cmp |
200 | struct shstr_cmp |
106 | { |
201 | { |
107 | const char *s; |
202 | const char *s; |
108 | |
203 | |
109 | explicit shstr_cmp (const char *s) |
204 | explicit shstr_cmp (const char *str) |
110 | : s (shstr::find (s)) |
205 | : s (shstr::find (str)) |
111 | { |
206 | { |
112 | } |
207 | } |
|
|
208 | |
|
|
209 | shstr_cmp (shstr_tmp sh) |
|
|
210 | : s (sh.s) |
|
|
211 | { |
|
|
212 | } |
|
|
213 | |
|
|
214 | // this is used for informational messages and the like |
|
|
215 | const char *operator &() const { return s; } |
113 | |
216 | |
114 | operator const char *() const { return s; } |
217 | operator const char *() const { return s; } |
115 | }; |
218 | }; |
116 | |
219 | |
117 | inline bool operator ==(const shstr_cmp &a, const shstr &b) |
220 | inline bool operator ==(const shstr_cmp &a, const shstr_tmp &b) |
118 | { |
221 | { |
119 | return a.s == b.s; |
222 | return a.s == b.s; |
120 | } |
223 | } |
121 | |
224 | |
122 | inline bool operator ==(const shstr &a, const shstr_cmp &b) |
225 | inline bool operator ==(const shstr_tmp &a, const shstr_cmp &b) |
123 | { |
226 | { |
124 | return b == a; |
227 | return a.s == b.s; |
125 | } |
228 | } |
126 | |
229 | |
127 | extern const shstr undead_name; /* Used in hit_player() in main.c */ |
230 | #define def(str) extern const shstr shstr_ ## str; |
|
|
231 | # include "shstrinc.h" |
|
|
232 | #undef def |
128 | |
233 | |
129 | #endif |
234 | #endif |
130 | |
235 | |