source: cpp/frams/util/sstringutils.cpp @ 785

Last change on this file since 785 was 785, checked in by Maciej Komosinski, 6 years ago

Very long string values are not printed shortened
  • Property svn:eol-style set to native
File size: 7.2 KB
Line 
1// This file is a part of Framsticks SDK.  http://www.framsticks.com/
2// Copyright (C) 1999-2015  Maciej Komosinski and Szymon Ulatowski.
3// See LICENSE.txt for details.
4
5#include "sstringutils.h"
6#include <common/virtfile/virtfile.h>
7#include <common/log.h>
8#include <common/nonstd.h>
9#include <common/Convert.h>
10
11int loadSString(const char* filename, SString& s, const char* framsgmodule, const char* error, bool remove_cr)
12{
13        VirtFILE *f;
14        int ret = 0;
15        if (f = Vfopen(filename, FOPEN_READ_BINARY))
16        {
17                loadSString(f, s, remove_cr);
18                ret = 1;
19                delete f;
20        }
21        else if (framsgmodule)
22                logPrintf(framsgmodule, "loadSString", LOG_WARN, error ? error : "Can't open file \"%s\"", filename);
23        return ret;
24}
25
26void loadSString(VirtFILE *f, SString& s, bool remove_cr)
27{
28        char buf[1024];
29        int len;
30        while (!f->Veof())
31        {
32                len = f->Vread(buf, 1, sizeof(buf));
33                s.append(buf, len);
34        }
35        if (remove_cr)
36                removeCR(s);
37}
38
39//load single line, discarding any \r or \n found at the end, return false if nothing could be loaded (error or eof)
40bool loadSStringLine(VirtFILE* f, SString& s)
41{
42        char buf[100];
43        bool eolfound = false;
44        bool ret = false;
45        s = SString::empty();
46        while (!eolfound)
47        {
48                char *r = f->Vgets(buf, sizeof(buf));
49                if (r == NULL) break;
50                ret = true;
51                int d = strlen(r);
52                if (d > 0)
53                {
54                        if (r[d - 1] == '\n') { d--; eolfound = true; }
55                        if (d > 0) if (r[d - 1] == '\r') d--;
56                        s += SString(r, d);
57                }
58        }
59        return ret;
60}
61
62//////////////////////////
63
64/** "x~xx~xxx" -> "x\~xx\~xxx"  */
65int quoteTilde(SString &target)
66{
67        const char* x = target.c_str();
68        SString tmp;
69        char *f;
70        while (1)
71        {
72                f = strchr((char*)x, '~');
73                if (f)
74                {
75                        tmp.append(x, f - x);
76                        tmp += "\\~";
77                        x = f + 1;
78                }
79                else
80                {
81                        if (tmp.len() == 0) return 0; // nothing was changed!
82                        tmp += x;
83                        target = tmp;
84                        return 1;
85                }
86        }
87}
88
89/** "x\~xx\~xxx" -> "x~xx~xxx"  */
90int unquoteTilde(SString &target)
91{
92        const char* x = target.c_str();
93        SString tmp;
94        char *f;
95        while (1)
96        {
97                f = strchr((char*)x, '\\');
98                if (f)
99                {
100                        tmp.append(x, f - x);
101                        if (f[1] == '~')
102                        {
103                                tmp += '~';
104                                x = f + 2;
105                        }
106                        else
107                        {
108                                tmp += "\\";
109                                x = f + 1;
110                        }
111                }
112                else
113                {
114                        if (tmp.len() == 0) return 0; // nothing was changed!
115                        tmp += x;
116                        target = tmp;
117                        return 1;
118                }
119        }
120}
121
122/////////////////
123
124bool strContainsOneOf(const char* str, const char* chars)
125{
126        while (*str)
127        {
128                if (strchr(chars, *str)) return 1;
129                str++;
130        }
131        return 0;
132}
133
134//////////////
135
136bool sstringQuote(SString& target)
137{
138        const char* x = target.c_str();
139        bool changed = 0;
140        SString tmp;
141        tmp.memoryHint(target.len());
142        while (*x)
143        {
144                switch (*x)
145                {
146                case '\n': tmp += "\\n"; changed = 1; break;
147                case '\r': tmp += "\\r"; changed = 1; break;
148                case '\t': tmp += "\\t"; changed = 1; break;
149                case '\"': tmp += "\\\""; changed = 1; break;
150                case '\\': tmp += "\\\\"; changed = 1; break;
151                default: tmp += *x;
152                }
153                x++;
154        }
155        if (changed) target = tmp;
156        return changed;
157}
158
159SString sstringShorten(const SString &in, int maxlen)
160{
161        SString out;
162        if (in.len() > maxlen)
163                out = in.substr(0, maxlen / 2) + "..." + in.substr(in.len() - maxlen + maxlen / 2) + SString::sprintf(" (length %d)", in.len());
164        else
165                out = in;
166        sstringQuote(out);
167        return out;
168}
169
170const char* skipQuoteString(const char* txt, const char* limit)
171{
172        while (*txt)
173        {
174                if (*txt == '\"') return txt;
175                if (*txt == '\\') txt++;
176                txt++;
177                if (txt == limit) break;
178        }
179        return txt;
180}
181
182int sstringUnquote(SString &target)
183{
184        const char* x = target.c_str();
185        SString tmp;
186        char *f;
187        while (1)
188        {
189                f = strchr((char*)x, '\\');
190                if (f)
191                {
192                        tmp.append(x, f - x);
193                        switch (f[1])
194                        {
195                        case 'n': tmp += '\n'; break;
196                        case 'r': tmp += '\r'; break;
197                        case 't': tmp += '\t'; break;
198                        case '\"': tmp += '\"'; break;
199                        default: tmp += f[1];
200                        }
201                        x = f + 2;
202                }
203                else
204                {
205                        if (tmp.len() == 0) return 0; // nothing was changed!
206                        tmp += x;
207                        target = tmp;
208                        return 1;
209                }
210        }
211}
212
213int strFindField(const SString& txt, const SString& name, int &end)
214{
215        const char* t = txt.c_str(), *n;
216        int pos = 0;
217        while (1)
218        {
219                n = strchr(t + pos, ',');
220                if ((!strncmp(t + pos, name.c_str(), name.len())) && (t[pos + name.len()] == '='))
221                {
222                        if (n) end = n - t; else end = txt.len();
223                        return pos;
224                }
225                if (n) pos = n - t + 1; else break;
226        }
227        return -1;
228}
229
230SString strGetField(const SString& txt, const SString& name)
231{
232        int p, e;
233        p = strFindField(txt, name, e);
234        if (p < 0) return SString();
235        p += name.len() + 1;
236        return SString(txt.substr(p, e - p));
237}
238
239void strSetField(SString& txt, const SString& name, const SString& value)
240{
241        int p, e;
242        p = strFindField(txt, name, e);
243        if (p < 0)
244        {
245                if (!value.len()) return;
246                char *t = txt.directAppend(1 + name.len() + value.len());
247                char *b = t;
248                if (txt.len()) *(t++) = ',';
249                strcpy(t, name.c_str()); t += name.len();
250                *(t++) = '=';
251                strcpy(t, value.c_str()); t += value.len();
252                txt.endAppend(t - b);
253        }
254        else
255        {
256                if (!value.len())
257                {
258                        if (p > 0) p--; else if (e < txt.len()) e++;
259                        char *t = txt.directWrite(0);
260                        memmove(t + p, t + e, txt.len() - e);
261                        txt.endWrite(txt.len() + value.len() - (e - p));
262                }
263                else
264                {
265                        p += name.len() + 1;
266                        char *t = txt.directWrite(txt.len() + value.len() - (e - p));
267                        memmove(t + p + value.len(), t + e, txt.len() - e);
268                        memmove(t + p, value.c_str(), value.len());
269                        txt.endWrite(txt.len() + value.len() - (e - p));
270                }
271        }
272}
273
274SString trim(const SString& s)
275{
276        const unsigned char*b = (const unsigned char*)s.c_str();
277        const unsigned char*e = b + s.len();
278        while ((b < e) && (*b <= ' ')) b++;
279        while ((b < e) && (e[-1] <= ' ')) e--;
280        if ((e - b) == s.len()) return s;
281        SString newstring;
282        char* t = newstring.directWrite(e - b);
283        memmove(t, b, e - b);
284        newstring.endWrite(e - b);
285        return newstring;
286}
287
288bool removeCR(SString& s)
289{
290        const char* p = s.c_str();
291        const char* cr = strchr(p, '\r');
292        if (!cr) return false;
293        char* begin = s.directWrite();
294        char* src = begin + (cr - p), *dst = src;
295        while (*src)
296                if (*src == '\r')
297                        src++;
298                else
299                        *(dst++) = *(src++);
300        s.endWrite(dst - begin);
301        return true;
302}
303
304bool matchWildcard(const SString& word, const SString& pattern)
305{
306        if (pattern.len() == 0)
307                return word.len() == 0;
308        int aster = pattern.indexOf('*');
309        if (aster >= 0)
310        {
311                SString before = pattern.substr(0, aster);
312                SString after = pattern.substr(aster + 1);
313                if (!word.len()) return false;
314                if (before.len()) if (!word.startsWith(before.c_str())) return false;
315                if (after.len())
316                        if ((word.len() < after.len())
317                                || (strcmp(after.c_str(), word.c_str() + word.len() - after.len())))
318                                return false;
319                return true;
320        }
321        else
322                return word == pattern;
323}
324
325bool matchWildcardList(const SString& word, const SString& patterns)
326{
327        if (patterns.len() == 0)
328                return word.len() == 0;
329        int pos = 0;
330        SString pattern;
331        while (patterns.getNextToken(pos, pattern, ','))
332                if (matchWildcard(word, pattern))
333                        return true;
334        return false;
335}
336
337SString getUIDString(uint64_t uid, char prefix)
338{
339        return SString::sprintf("%c" UINT64_FORMAT, prefix, uid);
340}
341
342bool parseUIDString(const char* str, char prefix, uint64_t &uid, bool err)
343{
344        if ((str[0] == prefix) && (isdigit(str[1])))
345        {
346                char* end;
347                uid = strtoull(str + 1, &end, 10);
348                if (end == (str + 1 + strlen(str + 1)))
349                        return true;
350        }
351        if (err)
352                logPrintf("SString", "parseUIDString", LOG_ERROR, "Invalid uid: '%s'", str);
353        return false;
354}
Note: See TracBrowser for help on using the repository browser.