home *** CD-ROM | disk | FTP | other *** search
/ Tricks of the Windows Gam…ming Gurus (2nd Edition) / Disc2.iso / vc98 / mfc / src / filest.cpp < prev    next >
C/C++ Source or Header  |  1998-06-16  |  8KB  |  285 lines
  1. // This is a part of the Microsoft Foundation Classes C++ library.
  2. // Copyright (C) 1992-1998 Microsoft Corporation
  3. // All rights reserved.
  4. //
  5. // This source code is only intended as a supplement to the
  6. // Microsoft Foundation Classes Reference and related
  7. // electronic documentation provided with the library.
  8. // See these sources for detailed information regarding the
  9. // Microsoft Foundation Classes product.
  10.  
  11. #include "stdafx.h"
  12. #include <errno.h>
  13. #include <io.h>
  14. #include <sys\types.h>
  15. #include <sys\stat.h>
  16.  
  17. #ifdef AFX_CORE1_SEG
  18. #pragma code_seg(AFX_CORE1_SEG)
  19. #endif
  20.  
  21. #ifdef _DEBUG
  22. #undef THIS_FILE
  23. static char THIS_FILE[] = __FILE__;
  24. #endif
  25.  
  26. #define new DEBUG_NEW
  27.  
  28. ////////////////////////////////////////////////////////////////////////////
  29. // Status information for all file classes
  30. // In this file so everyone doesn't get the CTime package
  31.  
  32. /////////////////////////////////////////////////////////////////////////////
  33. // CFileStatus diagnostics
  34.  
  35. #ifdef _DEBUG
  36. void CFileStatus::Dump(CDumpContext& dc) const
  37. {
  38.     dc << "a CFileStatus at " << (void*)this;
  39.  
  40.     dc << "\nm_ctime = " << m_ctime;
  41.     dc << "\nm_mtime = " << m_mtime;
  42.     dc << "\nm_atime = " << m_atime;
  43.     dc << "\nm_size = " << m_size;
  44.     dc << "\nm_attribute = " << m_attribute;
  45.     dc << "\nm_szFullName = " << m_szFullName;
  46.  
  47.     dc << "\n";
  48. }
  49. #endif
  50.  
  51. /////////////////////////////////////////////////////////////////////////////
  52. // CFile name handlers
  53.  
  54. CString CFile::GetFileName() const
  55. {
  56.     ASSERT_VALID(this);
  57.  
  58.     CFileStatus status;
  59.     GetStatus(status);
  60.     CString strResult;
  61.     AfxGetFileName(status.m_szFullName, strResult.GetBuffer(_MAX_FNAME),
  62.         _MAX_FNAME);
  63.     strResult.ReleaseBuffer();
  64.     return strResult;
  65. }
  66.  
  67. CString CFile::GetFileTitle() const
  68. {
  69.     ASSERT_VALID(this);
  70.  
  71.     CFileStatus status;
  72.     GetStatus(status);
  73.     CString strResult;
  74.     AfxGetFileTitle(status.m_szFullName, strResult.GetBuffer(_MAX_FNAME),
  75.         _MAX_FNAME);
  76.     strResult.ReleaseBuffer();
  77.     return strResult;
  78. }
  79.  
  80. CString CFile::GetFilePath() const
  81. {
  82.     ASSERT_VALID(this);
  83.  
  84.     CFileStatus status;
  85.     GetStatus(status);
  86.     return status.m_szFullName;
  87. }
  88.  
  89. /////////////////////////////////////////////////////////////////////////////
  90. // CFile Status implementation
  91.  
  92. BOOL CFile::GetStatus(CFileStatus& rStatus) const
  93. {
  94.     ASSERT_VALID(this);
  95.  
  96.     memset(&rStatus, 0, sizeof(CFileStatus));
  97.  
  98.     // copy file name from cached m_strFileName
  99.     lstrcpyn(rStatus.m_szFullName, m_strFileName,
  100.         _countof(rStatus.m_szFullName));
  101.  
  102.     if (m_hFile != hFileNull)
  103.     {
  104.         // get time current file size
  105.         FILETIME ftCreate, ftAccess, ftModify;
  106.         if (!::GetFileTime((HANDLE)m_hFile, &ftCreate, &ftAccess, &ftModify))
  107.             return FALSE;
  108.  
  109.         if ((rStatus.m_size = ::GetFileSize((HANDLE)m_hFile, NULL)) == (DWORD)-1L)
  110.             return FALSE;
  111.  
  112.         if (m_strFileName.IsEmpty())
  113.             rStatus.m_attribute = 0;
  114.         else
  115.         {
  116.             DWORD dwAttribute = ::GetFileAttributes(m_strFileName);
  117.  
  118.             // don't return an error for this because previous versions of MFC didn't
  119.             if (dwAttribute == 0xFFFFFFFF)
  120.                 rStatus.m_attribute = 0;
  121.             else
  122.             {
  123.                 rStatus.m_attribute = (BYTE) dwAttribute;
  124. #ifdef _DEBUG
  125.                 // MFC BUG: m_attribute is only a BYTE wide
  126.                 if (dwAttribute & ~0xFF)
  127.                     TRACE0("Warning: CFile::GetStatus() returns m_attribute without high-order flags.\n");
  128. #endif
  129.             }
  130.         }
  131.  
  132.         // convert times as appropriate
  133.         rStatus.m_ctime = CTime(ftCreate);
  134.         rStatus.m_atime = CTime(ftAccess);
  135.         rStatus.m_mtime = CTime(ftModify);
  136.  
  137.         if (rStatus.m_ctime.GetTime() == 0)
  138.             rStatus.m_ctime = rStatus.m_mtime;
  139.  
  140.         if (rStatus.m_atime.GetTime() == 0)
  141.             rStatus.m_atime = rStatus.m_mtime;
  142.     }
  143.     return TRUE;
  144. }
  145.  
  146. BOOL PASCAL CFile::GetStatus(LPCTSTR lpszFileName, CFileStatus& rStatus)
  147. {
  148.     // attempt to fully qualify path first
  149.     if (!AfxFullPath(rStatus.m_szFullName, lpszFileName))
  150.     {
  151.         rStatus.m_szFullName[0] = '\0';
  152.         return FALSE;
  153.     }
  154.  
  155.     WIN32_FIND_DATA findFileData;
  156.     HANDLE hFind = FindFirstFile((LPTSTR)lpszFileName, &findFileData);
  157.     if (hFind == INVALID_HANDLE_VALUE)
  158.         return FALSE;
  159.     VERIFY(FindClose(hFind));
  160.  
  161.     // strip attribute of NORMAL bit, our API doesn't have a "normal" bit.
  162.     rStatus.m_attribute = (BYTE)
  163.         (findFileData.dwFileAttributes & ~FILE_ATTRIBUTE_NORMAL);
  164.  
  165.     // get just the low DWORD of the file size
  166.     ASSERT(findFileData.nFileSizeHigh == 0);
  167.     rStatus.m_size = (LONG)findFileData.nFileSizeLow;
  168.  
  169.     // convert times as appropriate
  170.     rStatus.m_ctime = CTime(findFileData.ftCreationTime);
  171.     rStatus.m_atime = CTime(findFileData.ftLastAccessTime);
  172.     rStatus.m_mtime = CTime(findFileData.ftLastWriteTime);
  173.  
  174.     if (rStatus.m_ctime.GetTime() == 0)
  175.         rStatus.m_ctime = rStatus.m_mtime;
  176.  
  177.     if (rStatus.m_atime.GetTime() == 0)
  178.         rStatus.m_atime = rStatus.m_mtime;
  179.  
  180.     return TRUE;
  181. }
  182.  
  183. void AFX_CDECL AfxTimeToFileTime(const CTime& time, LPFILETIME pFileTime)
  184. {
  185.     SYSTEMTIME sysTime;
  186.     sysTime.wYear = (WORD)time.GetYear();
  187.     sysTime.wMonth = (WORD)time.GetMonth();
  188.     sysTime.wDay = (WORD)time.GetDay();
  189.     sysTime.wHour = (WORD)time.GetHour();
  190.     sysTime.wMinute = (WORD)time.GetMinute();
  191.     sysTime.wSecond = (WORD)time.GetSecond();
  192.     sysTime.wMilliseconds = 0;
  193.  
  194.     // convert system time to local file time
  195.     FILETIME localTime;
  196.     if (!SystemTimeToFileTime((LPSYSTEMTIME)&sysTime, &localTime))
  197.         CFileException::ThrowOsError((LONG)::GetLastError());
  198.  
  199.     // convert local file time to UTC file time
  200.     if (!LocalFileTimeToFileTime(&localTime, pFileTime))
  201.         CFileException::ThrowOsError((LONG)::GetLastError());
  202. }
  203.  
  204. void PASCAL CFile::SetStatus(LPCTSTR lpszFileName, const CFileStatus& status)
  205. {
  206.     DWORD wAttr;
  207.     FILETIME creationTime;
  208.     FILETIME lastAccessTime;
  209.     FILETIME lastWriteTime;
  210.     LPFILETIME lpCreationTime = NULL;
  211.     LPFILETIME lpLastAccessTime = NULL;
  212.     LPFILETIME lpLastWriteTime = NULL;
  213.  
  214.     if ((wAttr = GetFileAttributes((LPTSTR)lpszFileName)) == (DWORD)-1L)
  215.         CFileException::ThrowOsError((LONG)GetLastError());
  216.  
  217.     if ((DWORD)status.m_attribute != wAttr && (wAttr & readOnly))
  218.     {
  219.         // Set file attribute, only if currently readonly.
  220.         // This way we will be able to modify the time assuming the
  221.         // caller changed the file from readonly.
  222.  
  223.         if (!SetFileAttributes((LPTSTR)lpszFileName, (DWORD)status.m_attribute))
  224.             CFileException::ThrowOsError((LONG)GetLastError());
  225.     }
  226.  
  227.     // last modification time
  228.     if (status.m_mtime.GetTime() != 0)
  229.     {
  230.         AfxTimeToFileTime(status.m_mtime, &lastWriteTime);
  231.         lpLastWriteTime = &lastWriteTime;
  232.  
  233.         // last access time
  234.         if (status.m_atime.GetTime() != 0)
  235.         {
  236.             AfxTimeToFileTime(status.m_atime, &lastAccessTime);
  237.             lpLastAccessTime = &lastAccessTime;
  238.         }
  239.  
  240.         // create time
  241.         if (status.m_ctime.GetTime() != 0)
  242.         {
  243.             AfxTimeToFileTime(status.m_ctime, &creationTime);
  244.             lpCreationTime = &creationTime;
  245.         }
  246.  
  247.         HANDLE hFile = ::CreateFile(lpszFileName, GENERIC_READ|GENERIC_WRITE,
  248.             FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL,
  249.             NULL);
  250.  
  251.         if (hFile == INVALID_HANDLE_VALUE)
  252.             CFileException::ThrowOsError((LONG)::GetLastError());
  253.  
  254.         if (!SetFileTime((HANDLE)hFile, lpCreationTime, lpLastAccessTime, lpLastWriteTime))
  255.             CFileException::ThrowOsError((LONG)::GetLastError());
  256.  
  257.         if (!::CloseHandle(hFile))
  258.             CFileException::ThrowOsError((LONG)::GetLastError());
  259.     }
  260.  
  261.     if ((DWORD)status.m_attribute != wAttr && !(wAttr & readOnly))
  262.     {
  263.         if (!SetFileAttributes((LPTSTR)lpszFileName, (DWORD)status.m_attribute))
  264.             CFileException::ThrowOsError((LONG)GetLastError());
  265.     }
  266. }
  267.  
  268. ///////////////////////////////////////////////////////////////////////////////
  269. // CMemFile::GetStatus implementation
  270.  
  271. BOOL CMemFile::GetStatus(CFileStatus& rStatus) const
  272. {
  273.     ASSERT_VALID(this);
  274.  
  275.     rStatus.m_ctime = 0;
  276.     rStatus.m_mtime = 0;
  277.     rStatus.m_atime = 0;
  278.     rStatus.m_size = m_nFileSize;
  279.     rStatus.m_attribute = normal;
  280.     rStatus.m_szFullName[0] = '\0';
  281.     return TRUE;
  282. }
  283.  
  284. /////////////////////////////////////////////////////////////////////////////
  285.