Category

Development Platform
Home > SourceCode/Document > Internet-Socket-Network > Remote Control > View Code
Log.cpp
Package: vnc3326s.zip　[view]
Upload User: sbftbdw
Upload Date: 2007-01-03
Package Size: 379k
Code Size: 3k
Category:

Remote Control

Development Platform:

Visual C++

Log.cpp：Code Content
  1. // Log.cpp: implementation of the Log class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdhdrs.h"
  6. #include "Log.h"
  8. //////////////////////////////////////////////////////////////////////
  9. // Construction/Destruction
  10. //////////////////////////////////////////////////////////////////////
  12. const int Log::ToDebug   =  1;
  13. const int Log::ToFile    =  2;
  14. const int Log::ToConsole =  4;
  16. const static int LINE_BUFFER_SIZE = 1024;
  18. Log::Log(int mode, int level, LPSTR filename, bool append)
  19. {
  20. m_filename = NULL;
  21. m_append = false;
  22.     hlogfile = NULL;
  23.     m_todebug = false;
  24.     m_toconsole = false;
  25.     m_tofile = false;
  26.     SetFile(filename, append);
  27.     SetMode(mode);
  28. }
  30. void Log::SetMode(int mode) {
  31.     
  32.     if (mode & ToDebug)
  33.         m_todebug = true;
  34.     else
  35.         m_todebug = false;
  37.     if (mode & ToFile)  {
  38. if (!m_tofile)
  39. OpenFile();
  40. } else {
  41. CloseFile();
  42.         m_tofile = false;
  43.     }
  44.     
  45.     if (mode & ToConsole) {
  46.         if (!m_toconsole)
  47.             AllocConsole();
  48.         m_toconsole = true;
  49.     } else {
  50.         m_toconsole = false;
  51.     }
  52. }
  55. void Log::SetLevel(int level) {
  56.     m_level = level;
  57. }
  59. void Log::SetFile(LPSTR filename, bool append) 
  60. {
  61. CloseFile();
  62. if (m_filename != NULL)
  63. free(m_filename);
  64. m_filename = strdup(filename);
  65. m_append = append;
  66. if (m_tofile)
  67. OpenFile();
  68. }
  70. void Log::OpenFile()
  71. {
  72. // Is there a file-name?
  73. if (m_filename == NULL)
  74. {
  75.         m_todebug = true;
  76.         m_tofile = false;
  77.         Print(0, "Error opening log filen");
  78. return;
  79. }
  81.     m_tofile  = true;
  82.     
  83.     // If filename is NULL or invalid we should throw an exception here
  84.     hlogfile = CreateFile(
  85.         m_filename,  GENERIC_WRITE, FILE_SHARE_READ, NULL,
  86.         OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL  );
  87.     
  88.     if (hlogfile == NULL) {
  89.         // We should throw an exception here
  90.         m_todebug = true;
  91.         m_tofile = false;
  92.         Print(0, "Error opening log file %sn", m_filename);
  93.     }
  94.     if (m_append) {
  95.         SetFilePointer( hlogfile, 0, NULL, FILE_END );
  96.     } else {
  97.         SetEndOfFile( hlogfile );
  98.     }
  99. }
  101. // if a log file is open, close it now.
  102. void Log::CloseFile() {
  103.     if (hlogfile != NULL) {
  104.         CloseHandle(hlogfile);
  105.         hlogfile = NULL;
  106.     }
  107. }
  109. void Log::ReallyPrint(LPSTR format, va_list ap) 
  110. {
  111.     TCHAR line[LINE_BUFFER_SIZE];
  112.     vsprintf(line, format, ap);
  113.     if (m_todebug) OutputDebugString(line);
  114.     if (m_toconsole) {
  115.         DWORD byteswritten;
  116.         WriteConsole(GetStdHandle(STD_OUTPUT_HANDLE), line, strlen(line), &byteswritten, NULL); 
  117.     };
  118.     if (m_tofile && (hlogfile != NULL)) {
  119.         DWORD byteswritten;
  120.         WriteFile(hlogfile, line, strlen(line), &byteswritten, NULL); 
  121.     }
  122. }
  124. Log::~Log()
  125. {
  126. if (m_filename != NULL)
  127. free(m_filename);
  128.     CloseFile();
  129. }