// © 2016 and later: Unicode, Inc. and others. // License & terms of use: http://www.unicode.org/copyright.html /* ****************************************************************************** * * Copyright (C) 1998-2015, International Business Machines * Corporation and others. All Rights Reserved. * ****************************************************************************** * * File ufile.cpp * * Modification History: * * Date Name Description * 11/19/98 stephen Creation. * 03/12/99 stephen Modified for new C API. * 06/16/99 stephen Changed T_LocaleBundle to u_locbund * 07/19/99 stephen Fixed to use ucnv's default codepage. ****************************************************************************** */ #include "unicode/platform.h" #if defined(__GNUC__) && !defined(__clang__) && defined(__STRICT_ANSI__) // g++, fileno isn't defined if __STRICT_ANSI__ is defined. // clang fails to compile the header unless __STRICT_ANSI__ is defined. // __GNUC__ is set by both gcc and clang. #undef __STRICT_ANSI__ #endif #include "locmap.h" #include "unicode/ustdio.h" #if !UCONFIG_NO_CONVERSION #include #include "ufile.h" #include "unicode/uloc.h" #include "unicode/ures.h" #include "unicode/ucnv.h" #include "unicode/ustring.h" #include "unicode/unistr.h" #include "cstring.h" #include "cmemory.h" #if U_PLATFORM_USES_ONLY_WIN32_API && !defined(fileno) /* Windows likes to rename Unix-like functions */ #define fileno _fileno #endif static UFILE* finit_owner(FILE *f, const char *locale, const char *codepage, UBool takeOwnership ) { UErrorCode status = U_ZERO_ERROR; UFILE *result; if(f == NULL) { return 0; } result = (UFILE*) uprv_malloc(sizeof(UFILE)); if(result == NULL) { return 0; } uprv_memset(result, 0, sizeof(UFILE)); result->fFileno = fileno(f); result->fFile = f; result->str.fBuffer = result->fUCBuffer; result->str.fPos = result->fUCBuffer; result->str.fLimit = result->fUCBuffer; #if !UCONFIG_NO_FORMATTING /* if locale is 0, use the default */ if(u_locbund_init(&result->str.fBundle, locale) == 0) { /* DO NOT FCLOSE HERE! */ uprv_free(result); return 0; } #endif /* If the codepage is not "" use the ucnv_open default behavior */ if(codepage == NULL || *codepage != '\0') { result->fConverter = ucnv_open(codepage, &status); } /* else result->fConverter is already memset'd to NULL. */ if(U_SUCCESS(status)) { result->fOwnFile = takeOwnership; } else { #if !UCONFIG_NO_FORMATTING u_locbund_close(&result->str.fBundle); #endif /* DO NOT fclose here!!!!!! */ uprv_free(result); result = NULL; } return result; } U_CAPI UFILE* U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_finit(FILE *f, const char *locale, const char *codepage) { return finit_owner(f, locale, codepage, FALSE); } U_CAPI UFILE* U_EXPORT2 u_fadopt(FILE *f, const char *locale, const char *codepage) { return finit_owner(f, locale, codepage, TRUE); } U_CAPI UFILE* U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fopen(const char *filename, const char *perm, const char *locale, const char *codepage) { UFILE *result; FILE *systemFile = fopen(filename, perm); if(systemFile == 0) { return 0; } result = finit_owner(systemFile, locale, codepage, TRUE); if (!result) { /* Something bad happened. Maybe the converter couldn't be opened. */ fclose(systemFile); } return result; /* not a file leak */ } // FILENAME_BUF_MAX represents the largest size that we are willing to use for a // stack-allocated buffer to contain a file name or path. If PATH_MAX (POSIX) or MAX_PATH // (Windows) are defined and are smaller than this we will use their defined value; // otherwise, we will use FILENAME_BUF_MAX for the stack-allocated buffer, and dynamically // allocate a buffer for any file name or path that is that length or longer. #define FILENAME_BUF_MAX 296 #if defined PATH_MAX && PATH_MAX < FILENAME_BUF_MAX #define FILENAME_BUF_CAPACITY PATH_MAX #elif defined MAX_PATH && MAX_PATH < FILENAME_BUF_MAX #define FILENAME_BUF_CAPACITY MAX_PATH #else #define FILENAME_BUF_CAPACITY FILENAME_BUF_MAX #endif U_CAPI UFILE* U_EXPORT2 u_fopen_u(const UChar *filename, const char *perm, const char *locale, const char *codepage) { UFILE *result; char buffer[FILENAME_BUF_CAPACITY]; char *filenameBuffer = buffer; icu::UnicodeString filenameString(true, filename, -1); // readonly aliasing, does not allocate memory // extract with conversion to platform default codepage, return full length (not including 0 termination) int32_t filenameLength = filenameString.extract(0, filenameString.length(), filenameBuffer, FILENAME_BUF_CAPACITY); if (filenameLength >= FILENAME_BUF_CAPACITY) { // could not fit (with zero termination) in buffer filenameBuffer = static_cast(uprv_malloc(++filenameLength)); // add one for zero termination if (!filenameBuffer) { return nullptr; } filenameString.extract(0, filenameString.length(), filenameBuffer, filenameLength); } result = u_fopen(filenameBuffer, perm, locale, codepage); #if U_PLATFORM_USES_ONLY_WIN32_API /* Try Windows API _wfopen if the above fails. */ if (!result) { // TODO: test this code path, including wperm. wchar_t wperm[40] = {}; size_t retVal; mbstowcs_s(&retVal, wperm, UPRV_LENGTHOF(wperm), perm, _TRUNCATE); FILE *systemFile = _wfopen(reinterpret_cast(filename), wperm); // may return NULL for long filename if (systemFile) { result = finit_owner(systemFile, locale, codepage, TRUE); } if (!result && systemFile) { /* Something bad happened. Maybe the converter couldn't be opened. Bu do not fclose(systemFile) if systemFile is NULL. */ fclose(systemFile); } } #endif if (filenameBuffer != buffer) { uprv_free(filenameBuffer); } return result; /* not a file leak */ } U_CAPI UFILE* U_EXPORT2 u_fstropen(UChar *stringBuf, int32_t capacity, const char *locale) { UFILE *result; if (capacity < 0) { return NULL; } result = (UFILE*) uprv_malloc(sizeof(UFILE)); /* Null pointer test */ if (result == NULL) { return NULL; /* Just get out. */ } uprv_memset(result, 0, sizeof(UFILE)); result->str.fBuffer = stringBuf; result->str.fPos = stringBuf; result->str.fLimit = stringBuf+capacity; #if !UCONFIG_NO_FORMATTING /* if locale is 0, use the default */ if(u_locbund_init(&result->str.fBundle, locale) == 0) { /* DO NOT FCLOSE HERE! */ uprv_free(result); return 0; } #endif return result; } U_CAPI UBool U_EXPORT2 u_feof(UFILE *f) { UBool endOfBuffer; if (f == NULL) { return TRUE; } endOfBuffer = (UBool)(f->str.fPos >= f->str.fLimit); if (f->fFile != NULL) { return endOfBuffer && feof(f->fFile); } return endOfBuffer; } U_CAPI void U_EXPORT2 u_fflush(UFILE *file) { ufile_flush_translit(file); ufile_flush_io(file); if (file->fFile) { fflush(file->fFile); } else if (file->str.fPos < file->str.fLimit) { *(file->str.fPos++) = 0; } /* TODO: flush input */ } U_CAPI void u_frewind(UFILE *file) { u_fflush(file); ucnv_reset(file->fConverter); if (file->fFile) { rewind(file->fFile); file->str.fLimit = file->fUCBuffer; file->str.fPos = file->fUCBuffer; } else { file->str.fPos = file->str.fBuffer; } } U_CAPI void U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fclose(UFILE *file) { if (file) { u_fflush(file); ufile_close_translit(file); if(file->fOwnFile) fclose(file->fFile); #if !UCONFIG_NO_FORMATTING u_locbund_close(&file->str.fBundle); #endif ucnv_close(file->fConverter); uprv_free(file); } } U_CAPI FILE* U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fgetfile( UFILE *f) { return f->fFile; } #if !UCONFIG_NO_FORMATTING U_CAPI const char* U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fgetlocale( UFILE *file) { return file->str.fBundle.fLocale; } U_CAPI int32_t U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fsetlocale(UFILE *file, const char *locale) { u_locbund_close(&file->str.fBundle); return u_locbund_init(&file->str.fBundle, locale) == 0 ? -1 : 0; } #endif U_CAPI const char* U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fgetcodepage(UFILE *file) { UErrorCode status = U_ZERO_ERROR; const char *codepage = NULL; if (file->fConverter) { codepage = ucnv_getName(file->fConverter, &status); if(U_FAILURE(status)) return 0; } return codepage; } U_CAPI int32_t U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fsetcodepage( const char *codepage, UFILE *file) { UErrorCode status = U_ZERO_ERROR; int32_t retVal = -1; /* We use the normal default codepage for this system, and not the one for the locale. */ if ((file->str.fPos == file->str.fBuffer) && (file->str.fLimit == file->str.fBuffer)) { ucnv_close(file->fConverter); file->fConverter = ucnv_open(codepage, &status); if(U_SUCCESS(status)) { retVal = 0; } } return retVal; } U_CAPI UConverter * U_EXPORT2 /* U_CAPI ... U_EXPORT2 added by Peter Kirk 17 Nov 2001 */ u_fgetConverter(UFILE *file) { return file->fConverter; } #if !UCONFIG_NO_FORMATTING U_CAPI const UNumberFormat* U_EXPORT2 u_fgetNumberFormat(UFILE *file) { return u_locbund_getNumberFormat(&file->str.fBundle, UNUM_DECIMAL); } #endif #endif