ICU-57149.0.1.tar.gz

[apple/icu.git] / icuSources / io / ustdio.c

diff --git a/icuSources/io/ustdio.c b/icuSources/io/ustdio.c
index 04dd2f95fca248a3e2513ff57e085bd6ef7444fd..6c910baa9a7290703605a42a1fed98d119b3bfbc 100644 (file)
--- a/icuSources/io/ustdio.c
+++ b/icuSources/io/ustdio.c
@@ -1,23 +1,26 @@
 /*
 /*

-******************************************************************************
-*
-*   Copyright (C) 1998-2006, International Business Machines
-*   Corporation and others.  All Rights Reserved.
-*
-******************************************************************************
-*
-* File ustdio.c
-*
-* Modification History:
-*
-*   Date        Name        Description
-*   11/18/98    stephen     Creation.
-*   03/12/99    stephen     Modified for new C API.
-*   07/19/99    stephen     Fixed read() and gets()
-******************************************************************************
-*/
+ ******************************************************************************
+ *
+ *   Copyright (C) 1998-2016, International Business Machines
+ *   Corporation and others.  All Rights Reserved.
+ *
+ ******************************************************************************
+ *
+ * File ustdio.c
+ *
+ * Modification History:
+ *
+ *   Date        Name        Description
+ *   11/18/98    stephen     Creation.
+ *   03/12/99    stephen     Modified for new C API.
+ *   07/19/99    stephen     Fixed read() and gets()
+ ******************************************************************************
+ */

 
 #include "unicode/ustdio.h"
 
 #include "unicode/ustdio.h"

+
+#if !UCONFIG_NO_CONVERSION
+

 #include "unicode/putil.h"
 #include "cmemory.h"
 #include "cstring.h"
 #include "unicode/putil.h"
 #include "cmemory.h"
 #include "cstring.h"


@@ -37,7 +40,7 @@
 #define DELIM_PS 0x2029
 
 /* TODO: is this correct for all codepages? Should we just use \n and let the converter handle it? */
 #define DELIM_PS 0x2029
 
 /* TODO: is this correct for all codepages? Should we just use \n and let the converter handle it? */

-#ifdef U_WINDOWS
+#if U_PLATFORM_USES_ONLY_WIN32_API

 static const UChar DELIMITERS [] = { DELIM_CR, DELIM_LF, 0x0000 };
 static const uint32_t DELIMITERS_LEN = 2;
 /* TODO: Default newline writing should be detected based upon the converter being used. */
 static const UChar DELIMITERS [] = { DELIM_CR, DELIM_LF, 0x0000 };
 static const uint32_t DELIMITERS_LEN = 2;
 /* TODO: Default newline writing should be detected based upon the converter being used. */


@@ -163,6 +166,10 @@ static const UChar * u_file_translit(UFILE *f, const UChar *src, int32_t *count,
         {
             f->fTranslit->buffer = (UChar*)uprv_realloc(f->fTranslit->buffer, newlen * sizeof(UChar));
         }
         {
             f->fTranslit->buffer = (UChar*)uprv_realloc(f->fTranslit->buffer, newlen * sizeof(UChar));
         }

+        /* Check for malloc/realloc failure. */
+        if (f->fTranslit->buffer == NULL) {
+               return NULL;
+        }

         f->fTranslit->capacity = newlen;
     }
 
         f->fTranslit->capacity = newlen;
     }
 


@@ -234,6 +241,17 @@ ufile_flush_translit(UFILE *f)
 }
 
 
 }
 
 

+void
+ufile_flush_io(UFILE *f)
+{
+  if((!f) || (!f->fFile)) {
+    return; /* skip if no file */
+  }
+
+  u_file_write_flush(NULL, 0, f, TRUE, FALSE);
+}
+
+

 void
 ufile_close_translit(UFILE *f)
 {
 void
 ufile_close_translit(UFILE *f)
 {


@@ -278,7 +296,7 @@ u_fputc(UChar32      uc,
     int32_t idx = 0;
     UBool isError = FALSE;
 
     int32_t idx = 0;
     UBool isError = FALSE;
 

-    U16_APPEND(buf, idx, sizeof(buf)/sizeof(*buf), uc, isError);
+    U16_APPEND(buf, idx, UPRV_LENGTHOF(buf), uc, isError);

     if (isError) {
         return U_EOF;
     }
     if (isError) {
         return U_EOF;
     }


@@ -286,7 +304,7 @@ u_fputc(UChar32      uc,
 }
 
 
 }
 
 

-U_CAPI int32_t U_EXPORT2
+U_CFUNC int32_t U_EXPORT2

 u_file_write_flush(const UChar *chars,
                    int32_t     count,
                    UFILE       *f,
 u_file_write_flush(const UChar *chars,
                    int32_t     count,
                    UFILE       *f,


@@ -296,6 +314,7 @@ u_file_write_flush(const UChar *chars,
     /* Set up conversion parameters */
     UErrorCode  status       = U_ZERO_ERROR;
     const UChar *mySource    = chars;
     /* Set up conversion parameters */
     UErrorCode  status       = U_ZERO_ERROR;
     const UChar *mySource    = chars;

+    const UChar *mySourceBegin; 

     const UChar *mySourceEnd;
     char        charBuffer[UFILE_CHARBUFFER_SIZE];
     char        *myTarget   = charBuffer;
     const UChar *mySourceEnd;
     char        charBuffer[UFILE_CHARBUFFER_SIZE];
     char        *myTarget   = charBuffer;


@@ -330,6 +349,7 @@ u_file_write_flush(const UChar *chars,
 
     /* Perform the conversion in a loop */
     do {
 
     /* Perform the conversion in a loop */
     do {

+        mySourceBegin = mySource; /* beginning location for this loop */

         status     = U_ZERO_ERROR;
         if(f->fConverter != NULL) { /* We have a valid converter */
             ucnv_fromUnicode(f->fConverter,
         status     = U_ZERO_ERROR;
         if(f->fConverter != NULL) { /* We have a valid converter */
             ucnv_fromUnicode(f->fConverter,


@@ -341,8 +361,14 @@ u_file_write_flush(const UChar *chars,
                 flushIO,
                 &status);
         } else { /*weiv: do the invariant conversion */
                 flushIO,
                 &status);
         } else { /*weiv: do the invariant conversion */

-            u_UCharsToChars(mySource, myTarget, count);
-            myTarget += count;
+            int32_t convertChars = (int32_t) (mySourceEnd - mySource); 
+            if (convertChars > UFILE_CHARBUFFER_SIZE) { 
+                convertChars = UFILE_CHARBUFFER_SIZE; 
+                status = U_BUFFER_OVERFLOW_ERROR; 
+            } 
+            u_UCharsToChars(mySource, myTarget, convertChars); 
+            mySource += convertChars; 
+            myTarget += convertChars; 

         }
         numConverted = (int32_t)(myTarget - charBuffer);
 
         }
         numConverted = (int32_t)(myTarget - charBuffer);
 


@@ -353,7 +379,7 @@ u_file_write_flush(const UChar *chars,
                 numConverted,
                 f->fFile);
 
                 numConverted,
                 f->fFile);
 

-            written     += numConverted;
+            written     += (int32_t) (mySource - mySourceBegin);

         }
         myTarget     = charBuffer;
     }
         }
         myTarget     = charBuffer;
     }


@@ -402,7 +428,7 @@ ufile_fill_uchar_buffer(UFILE *f)
 
     /* shift the buffer if it isn't empty */
     if(dataSize != 0) {
 
     /* shift the buffer if it isn't empty */
     if(dataSize != 0) {

-        uprv_memmove(f->fUCBuffer, str->fPos, dataSize * sizeof(UChar));
+        u_memmove(f->fUCBuffer, str->fPos, dataSize); /* not accessing beyond memory */

     }
 
 
     }
 
 


@@ -569,7 +595,7 @@ ufile_getch(UFILE *f, UChar *ch)
         *ch = *(f->str.fPos)++;
         isValidChar = TRUE;
     }
         *ch = *(f->str.fPos)++;
         isValidChar = TRUE;
     }

-    else if (f) {
+    else {

         /* otherwise, fill the buffer and return the next character */
         if(f->str.fPos >= f->str.fLimit) {
             ufile_fill_uchar_buffer(f);
         /* otherwise, fill the buffer and return the next character */
         if(f->str.fPos >= f->str.fLimit) {
             ufile_fill_uchar_buffer(f);


@@ -701,3 +727,4 @@ u_file_read(    UChar        *chars,
 
     return read;
 }
 
     return read;
 }

+#endif