| OLD | NEW |
|---|
| (Empty) |
| 1 /* | |
| 2 ********************************************************************** | |
| 3 * Copyright (c) 2004,2011 International Business Machines | |
| 4 * Corporation and others. All Rights Reserved. | |
| 5 ********************************************************************** | |
| 6 * Author: Alan Liu | |
| 7 * Created: March 19 2004 | |
| 8 * Since: ICU 3.0 | |
| 9 ********************************************************************** | |
| 10 */ | |
| 11 #include "textfile.h" | |
| 12 #include "cmemory.h" | |
| 13 #include "cstring.h" | |
| 14 #include "intltest.h" | |
| 15 #include "util.h" | |
| 16 | |
| 17 // If the symbol CCP is defined, then the 'name' and 'encoding' | |
| 18 // constructor parameters are copied. Otherwise they are aliased. | |
| 19 // #define CCP | |
| 20 | |
| 21 TextFile::TextFile(const char* _name, const char* _encoding, UErrorCode& ec) : | |
| 22 file(0), | |
| 23 name(0), encoding(0), | |
| 24 buffer(0), | |
| 25 capacity(0), | |
| 26 lineNo(0) | |
| 27 { | |
| 28 if (U_FAILURE(ec) || _name == 0 || _encoding == 0) { | |
| 29 if (U_SUCCESS(ec)) { | |
| 30 ec = U_ILLEGAL_ARGUMENT_ERROR; | |
| 31 } | |
| 32 return; | |
| 33 } | |
| 34 | |
| 35 #ifdef CCP | |
| 36 name = uprv_malloc(uprv_strlen(_name) + 1); | |
| 37 encoding = uprv_malloc(uprv_strlen(_encoding) + 1); | |
| 38 if (name == 0 || encoding == 0) { | |
| 39 ec = U_MEMORY_ALLOCATION_ERROR; | |
| 40 return; | |
| 41 } | |
| 42 uprv_strcpy(name, _name); | |
| 43 uprv_strcpy(encoding, _encoding); | |
| 44 #else | |
| 45 name = (char*) _name; | |
| 46 encoding = (char*) _encoding; | |
| 47 #endif | |
| 48 | |
| 49 const char* testDir = IntlTest::getSourceTestData(ec); | |
| 50 if (U_FAILURE(ec)) { | |
| 51 return; | |
| 52 } | |
| 53 if (!ensureCapacity((int32_t)(uprv_strlen(testDir) + uprv_strlen(name) + 1))
) { | |
| 54 ec = U_MEMORY_ALLOCATION_ERROR; | |
| 55 return; | |
| 56 } | |
| 57 uprv_strcpy(buffer, testDir); | |
| 58 uprv_strcat(buffer, name); | |
| 59 | |
| 60 file = T_FileStream_open(buffer, "rb"); | |
| 61 if (file == 0) { | |
| 62 ec = U_ILLEGAL_ARGUMENT_ERROR; | |
| 63 return; | |
| 64 } | |
| 65 } | |
| 66 | |
| 67 TextFile::~TextFile() { | |
| 68 if (file != 0) T_FileStream_close(file); | |
| 69 if (buffer != 0) uprv_free(buffer); | |
| 70 #ifdef CCP | |
| 71 uprv_free(name); | |
| 72 uprv_free(encoding); | |
| 73 #endif | |
| 74 } | |
| 75 | |
| 76 UBool TextFile::readLine(UnicodeString& line, UErrorCode& ec) { | |
| 77 if (T_FileStream_eof(file)) { | |
| 78 return FALSE; | |
| 79 } | |
| 80 // Note: 'buffer' may change after ensureCapacity() is called, | |
| 81 // so don't use | |
| 82 // p=buffer; *p++=c; | |
| 83 // but rather | |
| 84 // i=; buffer[i++]=c; | |
| 85 int32_t n = 0; | |
| 86 for (;;) { | |
| 87 int c = T_FileStream_getc(file); // sic: int, not int32_t | |
| 88 if (c < 0 || c == 0xD || c == 0xA) { | |
| 89 // consume 0xA following 0xD | |
| 90 if (c == 0xD) { | |
| 91 c = T_FileStream_getc(file); | |
| 92 if (c != 0xA && c >= 0) { | |
| 93 T_FileStream_ungetc(c, file); | |
| 94 } | |
| 95 } | |
| 96 break; | |
| 97 } | |
| 98 if (!setBuffer(n++, c, ec)) return FALSE; | |
| 99 } | |
| 100 if (!setBuffer(n++, 0, ec)) return FALSE; | |
| 101 UnicodeString str(buffer, encoding); | |
| 102 // Remove BOM in first line, if present | |
| 103 if (lineNo == 0 && str[0] == 0xFEFF) { | |
| 104 str.remove(0, 1); | |
| 105 } | |
| 106 ++lineNo; | |
| 107 line = str.unescape(); | |
| 108 return TRUE; | |
| 109 } | |
| 110 | |
| 111 UBool TextFile::readLineSkippingComments(UnicodeString& line, UErrorCode& ec, | |
| 112 UBool trim) { | |
| 113 for (;;) { | |
| 114 if (!readLine(line, ec)) return FALSE; | |
| 115 // Skip over white space | |
| 116 int32_t pos = 0; | |
| 117 ICU_Utility::skipWhitespace(line, pos, TRUE); | |
| 118 // Ignore blank lines and comment lines | |
| 119 if (pos == line.length() || line.charAt(pos) == 0x23/*'#'*/) { | |
| 120 continue; | |
| 121 } | |
| 122 // Process line | |
| 123 if (trim) line.remove(0, pos); | |
| 124 return TRUE; | |
| 125 } | |
| 126 } | |
| 127 | |
| 128 /** | |
| 129 * Set buffer[index] to c, growing buffer if necessary. Return TRUE if | |
| 130 * successful. | |
| 131 */ | |
| 132 UBool TextFile::setBuffer(int32_t index, char c, UErrorCode& ec) { | |
| 133 if (capacity <= index) { | |
| 134 if (!ensureCapacity(index+1)) { | |
| 135 ec = U_MEMORY_ALLOCATION_ERROR; | |
| 136 return FALSE; | |
| 137 } | |
| 138 } | |
| 139 buffer[index] = c; | |
| 140 return TRUE; | |
| 141 } | |
| 142 | |
| 143 /** | |
| 144 * Make sure that 'buffer' has at least 'mincapacity' bytes. | |
| 145 * Return TRUE upon success. Upon return, 'buffer' may change | |
| 146 * value. In any case, previous contents are preserved. | |
| 147 */ | |
| 148 #define LOWEST_MIN_CAPACITY 64 | |
| 149 UBool TextFile::ensureCapacity(int32_t mincapacity) { | |
| 150 if (capacity >= mincapacity) { | |
| 151 return TRUE; | |
| 152 } | |
| 153 | |
| 154 // Grow by factor of 2 to prevent frequent allocation | |
| 155 // Note: 'capacity' may be 0 | |
| 156 int32_t i = (capacity < LOWEST_MIN_CAPACITY)? LOWEST_MIN_CAPACITY: capacity; | |
| 157 while (i < mincapacity) { | |
| 158 i <<= 1; | |
| 159 if (i < 0) { | |
| 160 i = 0x7FFFFFFF; | |
| 161 break; | |
| 162 } | |
| 163 } | |
| 164 mincapacity = i; | |
| 165 | |
| 166 // Simple realloc() no good; contents not preserved | |
| 167 // Note: 'buffer' may be 0 | |
| 168 char* newbuffer = (char*) uprv_malloc(mincapacity); | |
| 169 if (newbuffer == 0) { | |
| 170 return FALSE; | |
| 171 } | |
| 172 if (buffer != 0) { | |
| 173 uprv_strncpy(newbuffer, buffer, capacity); | |
| 174 uprv_free(buffer); | |
| 175 } | |
| 176 buffer = newbuffer; | |
| 177 capacity = mincapacity; | |
| 178 return TRUE; | |
| 179 } | |
| 180 | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2435373002:
Delete source/test (Closed)
