1 // © 2016 and later: Unicode, Inc. and others.
2 // License & terms of use: http://www.unicode.org/copyright.html
3 /*
4 *******************************************************************************
5 *
6 *   Copyright (C) 1998-2012, International Business Machines
7 *   Corporation and others.  All Rights Reserved.
8 *
9 *******************************************************************************
10 *
11 * File ustr.c
12 *
13 * Modification History:
14 *
15 *   Date        Name        Description
16 *   05/28/99    stephen     Creation.
17 *******************************************************************************
18 */
19 
20 #include "ustr.h"
21 #include "cmemory.h"
22 #include "cstring.h"
23 #include "unicode/ustring.h"
24 #include "unicode/putil.h"
25 #include "unicode/utf16.h"
26 
27 /* Protos */
28 static void ustr_resize(struct UString *s, int32_t len, UErrorCode *status);
29 
30 /* Macros */
31 #define ALLOCATION(minSize) (minSize < 0x80 ? 0x80 : (2 * minSize + 0x80) & ~(0x80 - 1))
32 
33 U_CFUNC void
ustr_init(struct UString * s)34 ustr_init(struct UString *s)
35 {
36     s->fChars = 0;
37     s->fLength = s->fCapacity = 0;
38 }
39 
40 U_CFUNC void
ustr_initChars(struct UString * s,const char * source,int32_t length,UErrorCode * status)41 ustr_initChars(struct UString *s, const char* source, int32_t length, UErrorCode *status)
42 {
43     int i = 0;
44     if (U_FAILURE(*status)) return;
45     s->fChars = 0;
46     s->fLength = s->fCapacity = 0;
47     if (length == -1) {
48         length = (int32_t)uprv_strlen(source);
49     }
50     if(s->fCapacity < length) {
51       ustr_resize(s, ALLOCATION(length), status);
52       if(U_FAILURE(*status)) return;
53     }
54     for (; i < length; i++)
55     {
56       UChar charToAppend;
57       u_charsToUChars(source+i, &charToAppend, 1);
58       ustr_ucat(s, charToAppend, status);
59       /*
60 #if U_CHARSET_FAMILY==U_ASCII_FAMILY
61         ustr_ucat(s, (UChar)(uint8_t)(source[i]), status);
62 #elif U_CHARSET_FAMILY==U_EBCDIC_FAMILY
63         ustr_ucat(s, (UChar)asciiFromEbcdic[(uint8_t)(*cs++)], status);
64 #else
65 #   error U_CHARSET_FAMILY is not valid
66 #endif
67       */
68     }
69 }
70 
71 U_CFUNC void
ustr_deinit(struct UString * s)72 ustr_deinit(struct UString *s)
73 {
74     if (s) {
75         uprv_free(s->fChars);
76         s->fChars = 0;
77         s->fLength = s->fCapacity = 0;
78     }
79 }
80 
81 U_CFUNC void
ustr_cpy(struct UString * dst,const struct UString * src,UErrorCode * status)82 ustr_cpy(struct UString *dst,
83      const struct UString *src,
84      UErrorCode *status)
85 {
86     if(U_FAILURE(*status) || dst == src)
87         return;
88 
89     if(dst->fCapacity < src->fLength) {
90         ustr_resize(dst, ALLOCATION(src->fLength), status);
91         if(U_FAILURE(*status))
92             return;
93     }
94     if(src->fChars == NULL || dst->fChars == NULL){
95         return;
96     }
97     u_memcpy(dst->fChars, src->fChars, src->fLength);
98     dst->fLength = src->fLength;
99     dst->fChars[dst->fLength] = 0x0000;
100 }
101 
102 U_CFUNC void
ustr_setlen(struct UString * s,int32_t len,UErrorCode * status)103 ustr_setlen(struct UString *s,
104         int32_t len,
105         UErrorCode *status)
106 {
107     if(U_FAILURE(*status))
108         return;
109 
110     if(s->fCapacity < (len + 1)) {
111         ustr_resize(s, ALLOCATION(len), status);
112         if(U_FAILURE(*status))
113             return;
114     }
115 
116     s->fLength = len;
117     s->fChars[len] = 0x0000;
118 }
119 
120 U_CFUNC void
ustr_cat(struct UString * dst,const struct UString * src,UErrorCode * status)121 ustr_cat(struct UString *dst,
122      const struct UString *src,
123      UErrorCode *status)
124 {
125     ustr_ncat(dst, src, src->fLength, status);
126 }
127 
128 U_CFUNC void
ustr_ncat(struct UString * dst,const struct UString * src,int32_t n,UErrorCode * status)129 ustr_ncat(struct UString *dst,
130       const struct UString *src,
131       int32_t n,
132       UErrorCode *status)
133 {
134     if(U_FAILURE(*status) || dst == src)
135         return;
136 
137     if(dst->fCapacity < (dst->fLength + n)) {
138         ustr_resize(dst, ALLOCATION(dst->fLength + n), status);
139         if(U_FAILURE(*status))
140             return;
141     }
142 
143     uprv_memcpy(dst->fChars + dst->fLength, src->fChars,
144                 sizeof(UChar) * n);
145     dst->fLength += src->fLength;
146     dst->fChars[dst->fLength] = 0x0000;
147 }
148 
149 U_CFUNC void
ustr_ucat(struct UString * dst,UChar c,UErrorCode * status)150 ustr_ucat(struct UString *dst,
151       UChar c,
152       UErrorCode *status)
153 {
154     if(U_FAILURE(*status))
155         return;
156 
157     if(dst->fCapacity < (dst->fLength + 1)) {
158         ustr_resize(dst, ALLOCATION(dst->fLength + 1), status);
159         if(U_FAILURE(*status))
160             return;
161     }
162 
163     uprv_memcpy(dst->fChars + dst->fLength, &c,
164         sizeof(UChar) * 1);
165     dst->fLength += 1;
166     dst->fChars[dst->fLength] = 0x0000;
167 }
168 U_CFUNC void
ustr_u32cat(struct UString * dst,UChar32 c,UErrorCode * status)169 ustr_u32cat(struct UString *dst, UChar32 c, UErrorCode *status){
170     if(c > 0x10FFFF){
171         *status = U_ILLEGAL_CHAR_FOUND;
172         return;
173     }
174     if(c >0xFFFF){
175         ustr_ucat(dst, U16_LEAD(c), status);
176         ustr_ucat(dst, U16_TRAIL(c), status);
177     }else{
178         ustr_ucat(dst, (UChar) c, status);
179     }
180 }
181 U_CFUNC void
ustr_uscat(struct UString * dst,const UChar * src,int len,UErrorCode * status)182 ustr_uscat(struct UString *dst,
183       const UChar* src,int len,
184       UErrorCode *status)
185 {
186     if(U_FAILURE(*status))
187         return;
188 
189     if(dst->fCapacity < (dst->fLength + len)) {
190         ustr_resize(dst, ALLOCATION(dst->fLength + len), status);
191         if(U_FAILURE(*status))
192             return;
193     }
194 
195     uprv_memcpy(dst->fChars + dst->fLength, src,
196         sizeof(UChar) * len);
197     dst->fLength += len;
198     dst->fChars[dst->fLength] = 0x0000;
199 }
200 
201 /* Destroys data in the string */
202 static void
ustr_resize(struct UString * s,int32_t len,UErrorCode * status)203 ustr_resize(struct UString *s,
204         int32_t len,
205         UErrorCode *status)
206 {
207     if(U_FAILURE(*status))
208         return;
209 
210     /* +1 for trailing 0x0000 */
211     s->fChars = (UChar*) uprv_realloc(s->fChars, sizeof(UChar) * (len + 1));
212     if(s->fChars == 0) {
213         *status = U_MEMORY_ALLOCATION_ERROR;
214         s->fLength = s->fCapacity = 0;
215         return;
216     }
217 
218     s->fCapacity = len;
219 }
220