hackedteam/core-macos

/*
 *  RCSMLogger.m
 *  RCSMac
 *
 *
 *  Created by revenge on 2/2/11.
 *  Copyright (C) HT srl 2011. All rights reserved
 *
 */

#import "RCSMLogger.h"

#ifdef ENABLE_LOGGING

static __m_MLogger *sharedLogger = nil;
static NSString *gComponent     = nil;
static BOOL gIsProcNameEnabled  = NO;

@implementation __m_MLogger

@synthesize mLevel;

+ (__m_MLogger *)sharedInstance
{
  @synchronized(self)
    {
      if (sharedLogger == nil)
        {
          //
          // Assignment is not done here
          //
          [[self alloc] init];
        }
    }
  
  return sharedLogger;
}

+ (id)allocWithZone: (NSZone *)aZone
{
  @synchronized(self)
    {
      if (sharedLogger == nil)
        {
          sharedLogger = [super allocWithZone: aZone];
          
          //
          // Assignment and return on first allocation
          //
          return sharedLogger;
        }
      }
  
  // On subsequent allocation attemps return nil
  return nil;
}

- (id)copyWithZone: (NSZone *)aZone
{
  return self;
}

- (id)init
{
  Class myClass = [self class];
  
  @synchronized(myClass)
    {
      if (sharedLogger != nil)
        {
          self = [super init];
          
          if (self != nil)
            {
              sharedLogger = self;
              
              if (gComponent == nil)
                {
                  gComponent = @"";
                }
              
              NSAutoreleasePool *outerPool = [[NSAutoreleasePool alloc] init];

              NSDate *date = [[NSDate alloc] init];
              NSDateFormatter *dateFormat = [[NSDateFormatter alloc] init];
              [dateFormat setDateFormat: @"dd-MM-yyyy"];
              
              NSString *dateString = [dateFormat stringFromDate: date];
              [dateFormat release];
              
              NSMutableString *logName = [NSMutableString stringWithFormat:
                                          @"%@/__m__%@_%@.log",
                                          NSHomeDirectory(),
                                          gComponent,
                                          dateString];
              mLogName = [[NSString alloc] initWithString: logName];
              
              if ([[NSFileManager defaultManager] fileExistsAtPath: mLogName] == NO)
                {
                  [@"" writeToFile: mLogName
                        atomically: YES
                          encoding: NSUTF8StringEncoding
                             error: nil];
                }
              
              mLevel = kErrLevel;
              [outerPool release];
            }
        }
    }
  
  return sharedLogger;
}

- (id)retain
{
  return self;
}

- (unsigned)retainCount
{
  // Denotes an object that cannot be released
  return UINT_MAX;
}

- (void)release
{
  // Do nothing
}

- (id)autorelease
{
  return self;
}

+ (void)setComponent: (NSString *)aComponent
{
  if (gComponent != aComponent)
    {
      [gComponent release];
      gComponent = [aComponent copy];
    }
}

+ (void)enableProcessNameVisualization: (BOOL)aFlag
{
  gIsProcNameEnabled = aFlag;
}

- (void)log: (const char *)aCaller
       line: (int)aLineNumber
      level: (int)aLogLevel
     string: (NSString *)aFormat, ...
{
  NSAutoreleasePool *outerPool = [[NSAutoreleasePool alloc] init];

  va_list argList;
  NSString *logString;
  NSString *entry;
  NSString *level;
  
  if (aLogLevel > mLevel)
    {
      [outerPool release];
      return;
    }
  
  va_start(argList, aFormat);
  logString = [[NSString alloc] initWithFormat: aFormat arguments: argList];
  va_end(argList);
  
  NSDate *date = [[NSDate alloc] init];
  NSDateFormatter *dateFormat = [[NSDateFormatter alloc] init];
  [dateFormat setDateFormat: @"HH:mm:ss"];
  NSString *dateString = [dateFormat stringFromDate: date];
  [dateFormat release];
  
  switch (aLogLevel)
    {
    case kInfoLevel:
      level = @"[INFO] ";
      break;
    case kWarnLevel:
      level = @"[WARN] ";
      break;
    case kErrLevel:
      level = @"[ERR]  ";
      break;
    case kVerboseLevel:
      level = @"[VERB] ";
      break;
    default:
      level = @"[INFO] ";
      break;
    }
  
  NSThread *thread     = [NSThread currentThread];
  NSString *threadDesc = [thread description];
  int threadNo         = [[threadDesc substringWithRange:
                           NSMakeRange([threadDesc length] - 2, 1)] intValue];
  
  if (gIsProcNameEnabled)
   {
     entry = [[NSString alloc] initWithFormat: @"[%@][%@]%@[TID:%d]%s:%d - %@",
                                               [[[NSBundle mainBundle] executablePath] lastPathComponent],
                                               dateString,
                                               level,
                                               threadNo,
                                               aCaller,
                                               aLineNumber,
                                               logString];
   }
  else
   {
     entry = [[NSString alloc] initWithFormat: @"[%@]%@[TID:%d]%s:%d - %@",
                                               dateString,
                                               level,
                                               threadNo,
                                               aCaller,
                                               aLineNumber,
                                               logString];
   }
  
  NSMutableData *entryData = [NSMutableData dataWithData:
                              [entry dataUsingEncoding: NSUTF8StringEncoding]];
  char newline = '\n';
  [entryData appendBytes: &newline
                  length: sizeof(newline)];
  
  mLogHandle = [NSFileHandle fileHandleForUpdatingAtPath: mLogName];
  //[mLogHandle retain];
    if(mLogHandle != nil)  // jo added condition
    {
        [mLogHandle seekToEndOfFile];
        [mLogHandle writeData: entryData];
        [mLogHandle closeFile];
    }
  [entry release];
  [date release];
  [logString release];
  [outerPool release];
}

@end

#endif