src/Utils/StdioFileT.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2006 - Stefan Kueng
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #include "stdafx.h"
  20 #include "StdioFileT.h"
  21
  22 CStdioFileT::CStdioFileT() : CStdioFile()
  23 {
  24 }
  25
  26 CStdioFileT::CStdioFileT(LPCTSTR lpszFileName, UINT nOpenFlags)
  27 {
  28         ASSERT(lpszFileName != NULL);
  29         ASSERT(AfxIsValidString(lpszFileName));
  30
  31         CFileException e;
  32         if (!Open(lpszFileName, nOpenFlags, &e))
  33                 AfxThrowFileException(e.m_cause, e.m_lOsError, e.m_strFileName);
  34 }
  35
  36
  37 void CStdioFileT::WriteString(LPCSTR lpsz)
  38 {
  39         ASSERT(lpsz != NULL);
  40         ASSERT(m_pStream != NULL);
  41
  42         if (lpsz == NULL)
  43         {
  44                 AfxThrowInvalidArgException();
  45         }
  46
  47         if (fputs(lpsz, m_pStream) == EOF)
  48                 AfxThrowFileException(CFileException::diskFull, _doserrno, m_strFileName);
  49 }
  50
  51
  52 void CStdioFileT::WriteString(LPCWSTR lpsz)
  53 {
  54         ASSERT(lpsz != NULL);
  55         ASSERT(m_pStream != NULL);
  56
  57         if (lpsz == NULL)
  58         {
  59                 AfxThrowInvalidArgException();
  60         }
  61
  62         if (fputws(lpsz, m_pStream) == EOF)
  63                 AfxThrowFileException(CFileException::diskFull, _doserrno, m_strFileName);
  64 }
  65
  66 BOOL CStdioFileT::ReadString(CStringA& rString)
  67 {
  68         ASSERT_VALID(this);
  69
  70 #ifndef afxChNil
  71         static TCHAR afxChNil = '\0';
  72 #endif
  73
  74         rString = &afxChNil;    // empty string without deallocating
  75         const int nMaxSize = 128;
  76         LPSTR lpsz = rString.GetBuffer(nMaxSize);
  77         LPSTR lpszResult;
  78         int nLen = 0;
  79         for (;;)
  80         {
  81                 lpszResult = fgets(lpsz, nMaxSize+1, m_pStream);
  82                 rString.ReleaseBuffer();
  83
  84                 // handle error/eof case
  85                 if (lpszResult == NULL && !feof(m_pStream))
  86                 {
  87                         clearerr(m_pStream);
  88                         AfxThrowFileException(CFileException::genericException, _doserrno,
  89                                 m_strFileName);
  90                 }
  91
  92                 // if string is read completely or EOF
  93                 if (lpszResult == NULL ||
  94                         (nLen = (int)strlen(lpsz)) < nMaxSize ||
  95                         lpsz[nLen-1] == '\n')
  96                         break;
  97
  98                 nLen = rString.GetLength();
  99                 lpsz = rString.GetBuffer(nMaxSize + nLen) + nLen;
 100         }
 101
 102         // remove '\n' from end of string if present
 103         lpsz = rString.GetBuffer(0);
 104         nLen = rString.GetLength();
 105         if (nLen != 0 && lpsz[nLen-1] == '\n')
 106                 rString.GetBufferSetLength(nLen-1);
 107
 108         return lpszResult != NULL;
 109 }