/src/icu/source/common/uenum.cpp
Line  | Count  | Source (jump to first uncovered line)  | 
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) 2002-2012, International Business Machines  | 
7  |  | *   Corporation and others.  All Rights Reserved.  | 
8  |  | *  | 
9  |  | *******************************************************************************  | 
10  |  | *   file name:  uenum.c  | 
11  |  | *   encoding:   UTF-8  | 
12  |  | *   tab size:   8 (not used)  | 
13  |  | *   indentation:2  | 
14  |  | *  | 
15  |  | *   created on: 2002jul08  | 
16  |  | *   created by: Vladimir Weinstein  | 
17  |  | */  | 
18  |  |  | 
19  |  | #include "unicode/putil.h"  | 
20  |  | #include "uenumimp.h"  | 
21  |  | #include "cmemory.h"  | 
22  |  |  | 
23  |  | /* Layout of the baseContext buffer. */  | 
24  |  | typedef struct { | 
25  |  |     int32_t len;  /* number of bytes available starting at 'data' */  | 
26  |  |     char    data; /* actual data starts here */  | 
27  |  | } _UEnumBuffer;  | 
28  |  |  | 
29  |  | /* Extra bytes to allocate in the baseContext buffer. */  | 
30  |  | static const int32_t PAD = 8;  | 
31  |  |  | 
32  |  | /* Return a pointer to the baseContext buffer, possibly allocating  | 
33  |  |    or reallocating it if at least 'capacity' bytes are not available. */  | 
34  | 0  | static void* _getBuffer(UEnumeration* en, int32_t capacity) { | 
35  |  | 
  | 
36  | 0  |     if (en->baseContext != NULL) { | 
37  | 0  |         if (((_UEnumBuffer*) en->baseContext)->len < capacity) { | 
38  | 0  |             capacity += PAD;  | 
39  | 0  |             en->baseContext = uprv_realloc(en->baseContext,  | 
40  | 0  |                                            sizeof(int32_t) + capacity);  | 
41  | 0  |             if (en->baseContext == NULL) { | 
42  | 0  |                 return NULL;  | 
43  | 0  |             }  | 
44  | 0  |             ((_UEnumBuffer*) en->baseContext)->len = capacity;  | 
45  | 0  |         }  | 
46  | 0  |     } else { | 
47  | 0  |         capacity += PAD;  | 
48  | 0  |         en->baseContext = uprv_malloc(sizeof(int32_t) + capacity);  | 
49  | 0  |         if (en->baseContext == NULL) { | 
50  | 0  |             return NULL;  | 
51  | 0  |         }  | 
52  | 0  |         ((_UEnumBuffer*) en->baseContext)->len = capacity;  | 
53  | 0  |     }  | 
54  |  |       | 
55  | 0  |     return (void*) & ((_UEnumBuffer*) en->baseContext)->data;  | 
56  | 0  | }  | 
57  |  |  | 
58  |  | U_CAPI void U_EXPORT2  | 
59  |  | uenum_close(UEnumeration* en)  | 
60  | 0  | { | 
61  | 0  |     if (en) { | 
62  | 0  |         if (en->close != NULL) { | 
63  | 0  |             if (en->baseContext) { | 
64  | 0  |                 uprv_free(en->baseContext);  | 
65  | 0  |             }  | 
66  | 0  |             en->close(en);  | 
67  | 0  |         } else { /* this seems dangerous, but we better kill the object */ | 
68  | 0  |             uprv_free(en);  | 
69  | 0  |         }  | 
70  | 0  |     }  | 
71  | 0  | }  | 
72  |  |  | 
73  |  | U_CAPI int32_t U_EXPORT2  | 
74  |  | uenum_count(UEnumeration* en, UErrorCode* status)  | 
75  | 0  | { | 
76  | 0  |     if (!en || U_FAILURE(*status)) { | 
77  | 0  |         return -1;  | 
78  | 0  |     }  | 
79  | 0  |     if (en->count != NULL) { | 
80  | 0  |         return en->count(en, status);  | 
81  | 0  |     } else { | 
82  | 0  |         *status = U_UNSUPPORTED_ERROR;  | 
83  | 0  |         return -1;  | 
84  | 0  |     }  | 
85  | 0  | }  | 
86  |  |  | 
87  |  | /* Don't call this directly. Only uenum_unext should be calling this. */  | 
88  |  | U_CAPI const UChar* U_EXPORT2  | 
89  |  | uenum_unextDefault(UEnumeration* en,  | 
90  |  |             int32_t* resultLength,  | 
91  |  |             UErrorCode* status)  | 
92  | 0  | { | 
93  | 0  |     UChar *ustr = NULL;  | 
94  | 0  |     int32_t len = 0;  | 
95  | 0  |     if (en->next != NULL) { | 
96  | 0  |         const char *cstr = en->next(en, &len, status);  | 
97  | 0  |         if (cstr != NULL) { | 
98  | 0  |             ustr = (UChar*) _getBuffer(en, (len+1) * sizeof(UChar));  | 
99  | 0  |             if (ustr == NULL) { | 
100  | 0  |                 *status = U_MEMORY_ALLOCATION_ERROR;  | 
101  | 0  |             } else { | 
102  | 0  |                 u_charsToUChars(cstr, ustr, len+1);  | 
103  | 0  |             }  | 
104  | 0  |         }  | 
105  | 0  |     } else { | 
106  | 0  |         *status = U_UNSUPPORTED_ERROR;  | 
107  | 0  |     }  | 
108  | 0  |     if (resultLength) { | 
109  | 0  |         *resultLength = len;  | 
110  | 0  |     }  | 
111  | 0  |     return ustr;  | 
112  | 0  | }  | 
113  |  |  | 
114  |  | /* Don't call this directly. Only uenum_next should be calling this. */  | 
115  |  | U_CAPI const char* U_EXPORT2  | 
116  |  | uenum_nextDefault(UEnumeration* en,  | 
117  |  |             int32_t* resultLength,  | 
118  |  |             UErrorCode* status)  | 
119  | 0  | { | 
120  | 0  |     if (en->uNext != NULL) { | 
121  | 0  |         char *tempCharVal;  | 
122  | 0  |         const UChar *tempUCharVal = en->uNext(en, resultLength, status);  | 
123  | 0  |         if (tempUCharVal == NULL) { | 
124  | 0  |             return NULL;  | 
125  | 0  |         }  | 
126  | 0  |         tempCharVal = (char*)  | 
127  | 0  |             _getBuffer(en, (*resultLength+1) * sizeof(char));  | 
128  | 0  |         if (!tempCharVal) { | 
129  | 0  |             *status = U_MEMORY_ALLOCATION_ERROR;  | 
130  | 0  |             return NULL;  | 
131  | 0  |         }  | 
132  | 0  |         u_UCharsToChars(tempUCharVal, tempCharVal, *resultLength + 1);  | 
133  | 0  |         return tempCharVal;  | 
134  | 0  |     } else { | 
135  | 0  |         *status = U_UNSUPPORTED_ERROR;  | 
136  | 0  |         return NULL;  | 
137  | 0  |     }  | 
138  | 0  | }  | 
139  |  |  | 
140  |  | U_CAPI const UChar* U_EXPORT2  | 
141  |  | uenum_unext(UEnumeration* en,  | 
142  |  |             int32_t* resultLength,  | 
143  |  |             UErrorCode* status)  | 
144  | 0  | { | 
145  | 0  |     if (!en || U_FAILURE(*status)) { | 
146  | 0  |         return NULL;  | 
147  | 0  |     }  | 
148  | 0  |     if (en->uNext != NULL) { | 
149  | 0  |         return en->uNext(en, resultLength, status);  | 
150  | 0  |     } else { | 
151  | 0  |         *status = U_UNSUPPORTED_ERROR;  | 
152  | 0  |         return NULL;  | 
153  | 0  |     }  | 
154  | 0  | }  | 
155  |  |  | 
156  |  | U_CAPI const char* U_EXPORT2  | 
157  |  | uenum_next(UEnumeration* en,  | 
158  |  |           int32_t* resultLength,  | 
159  |  |           UErrorCode* status)  | 
160  | 0  | { | 
161  | 0  |     if (!en || U_FAILURE(*status)) { | 
162  | 0  |         return NULL;  | 
163  | 0  |     }  | 
164  | 0  |     if (en->next != NULL) { | 
165  | 0  |         if (resultLength != NULL) { | 
166  | 0  |             return en->next(en, resultLength, status);  | 
167  | 0  |         }  | 
168  | 0  |         else { | 
169  | 0  |             int32_t dummyLength=0;  | 
170  | 0  |             return en->next(en, &dummyLength, status);  | 
171  | 0  |         }  | 
172  | 0  |     } else { | 
173  | 0  |         *status = U_UNSUPPORTED_ERROR;  | 
174  | 0  |         return NULL;  | 
175  | 0  |     }  | 
176  | 0  | }  | 
177  |  |  | 
178  |  | U_CAPI void U_EXPORT2  | 
179  |  | uenum_reset(UEnumeration* en, UErrorCode* status)  | 
180  | 0  | { | 
181  | 0  |     if (!en || U_FAILURE(*status)) { | 
182  | 0  |         return;  | 
183  | 0  |     }  | 
184  | 0  |     if (en->reset != NULL) { | 
185  | 0  |         en->reset(en, status);  | 
186  | 0  |     } else { | 
187  | 0  |         *status = U_UNSUPPORTED_ERROR;  | 
188  | 0  |     }  | 
189  | 0  | }  |