diff options
Diffstat (limited to 'AuditronAppDelegate.m')
| -rw-r--r-- | AuditronAppDelegate.m | 401 |
1 files changed, 401 insertions, 0 deletions
diff --git a/AuditronAppDelegate.m b/AuditronAppDelegate.m new file mode 100644 index 0000000..c9d2ebd --- /dev/null +++ b/AuditronAppDelegate.m @@ -0,0 +1,401 @@ +// +// AuditronAppDelegate.m +// Auditron +// +// Created by Filipp Lepalaan on 9.7.2011. +// Copyright 2011 __MyCompanyName__. All rights reserved. +// + +#import "AuditronAppDelegate.h" + +@implementation AuditronAppDelegate + +@synthesize window; + +-(void) savePanelDidEnd:(NSSavePanel *)savePanel returnCode:(int)returnCode contextInfo:(void *)contextInfo { + + if (returnCode == NSOKButton) { + + NSFileManager *fm = [NSFileManager defaultManager]; + [fm moveItemAtPath:[self reportFile] toPath:[savePanel filename] error:NULL]; + + } + +} + +-(void) saveResultsPanelDidEnd:(NSSavePanel *)savePanel returnCode:(int)returnCode contextInfo:(void *)contextInfo +{ + if (returnCode == NSOKButton) + { + NSString *filePath = [savePanel filename]; + NSArray *data = [self comparisonData]; + NSUInteger i, count = [data count]; + NSString *csv = [NSString stringWithString:@""]; + + for (i = 0; i < count; i++) + { + NSDictionary *obj = [data objectAtIndex:i]; + NSString *csvRow = [NSString stringWithString:@"\""]; + csvRow = [csvRow stringByAppendingString:[[obj allValues] componentsJoinedByString:@"\";\""]]; + csv = [csv stringByAppendingString:[csvRow stringByAppendingString:@"\"\n"]]; + } + + [csv writeToFile:filePath atomically:YES + encoding:NSUnicodeStringEncoding + error:NULL]; + } + +} + +-(void) sheetDidEnd:(NSWindow *)sheet returnCode:(int)returnCode contextInfo:(void *)contextInfo { + + + +} + +-(void) sendmail +{ + NSDictionary *errorDict; + NSAppleEventDescriptor* returnDescriptor = NULL; + + NSString *subject = [NSString stringWithFormat:@"System Profiler Report for %@", hostName]; + NSString *recipientName = [NSString stringWithString:@"Filipp Lepalaan"]; + NSString *recipientEmail = [NSString stringWithString:@"filipp@mcare.fi"]; + + NSString *source; + + if ([mailer isEqualToString:@"com.apple.mail"]) + { + source = [NSString stringWithFormat: + @"tell application \"Mail\"\n\ + set theMessage to make new outgoing message with properties {visible:true, subject:\"%@\", content:\"\"}\n\ + tell theMessage\n\ + make new to recipient at end of to recipients with properties {name:\"%@\", address:\"%@\"}\n\ + end tell\n\ + tell content of theMessage\n\ + make new attachment with properties {file name:\"%@\"}\n\ + end tell\n\ + activate\n\ + end tell", subject, recipientName, recipientEmail, reportFile]; + } + else if ([mailer isEqualToString:@"com.microsoft.entourage"]) + { + source = [NSString stringWithFormat: + @"tell application \"Microsoft Entourage\"\n\ + set theRecipients to {{address:{display name:\"%@\", address:\"%@\"}, recipient type:to recipient}}\n\ + set theMessage to make new outgoing message with properties {recipient:theRecipients, subject:\"%@\", attachment:{file:(POSIX file \"%@\")}}\n\ + open theMessage\n\ + activate\n\ + end tell", recipientName, recipientEmail, subject, reportFile]; + } else { + + NSSavePanel *panel = [NSSavePanel savePanel]; + [panel setCanCreateDirectories:YES]; + NSString *fileName = [NSString stringWithFormat:@"%@.spx.gz", hostName]; + [panel setNameFieldStringValue:fileName]; + [panel beginSheetForDirectory:nil + file:fileName + modalForWindow:window + modalDelegate:self didEndSelector:@selector(savePanelDidEnd:returnCode:contextInfo:) + contextInfo:NULL]; + } + + NSAppleScript *as = [[NSAppleScript alloc] initWithSource:source]; + + returnDescriptor = [as executeAndReturnError: &errorDict]; + [as release]; + + if (returnDescriptor != NULL) + { + // successful execution + if (kAENullEvent != [returnDescriptor descriptorType]) + { + // script returned an AppleScript result + if (cAEList == [returnDescriptor descriptorType]) + { + // result is a list of other descriptors + } + else + { + // coerce the result to the appropriate ObjC type + } + } + } + else + { + // no script result, handle error here + NSLog(@"AppleScript failed: %@", [errorDict description]); + } +} + +-(NSString *)reportFile +{ + return reportFile; +} + +-(NSMutableArray *)comparisonData +{ + return comparisonData; +} + +- (IBAction)build_report:(id)sender +{ + NSString *path = [NSString stringWithFormat:@"%@/%@.spx.gz", NSTemporaryDirectory(), hostName]; + + [self setReportFile:path]; + + NSArray *args = [NSArray arrayWithObjects: + @"-c", [NSString stringWithFormat: + @"/usr/sbin/system_profiler -xml -detailLevel full | /usr/bin/gzip > %@", + path], + nil]; + + NSTask *t = [NSTask launchedTaskWithLaunchPath:@"/bin/bash" arguments:args]; + [progress startAnimation:sender]; + + [NSApp beginSheet:sheet modalForWindow:window modalDelegate:self + didEndSelector:@selector(sheetDidEnd:returnCode:contextInfo:) contextInfo:NULL]; + + [t waitUntilExit]; + [sheet orderOut:sender]; + [NSApp endSheet:sheet]; + + [self sendmail]; + +} + +-(void) openPanelDidEnd:(NSOpenPanel *) openPanel returnCode:(int)returnCode contextInfo:(void *)x +{ + NSString *path; + path = [openPanel filename]; + + NSMutableArray *outData = [NSMutableArray arrayWithCapacity:100]; + NSFileManager *fm = [NSFileManager defaultManager]; + NSDirectoryEnumerator *dirEnum = [fm enumeratorAtPath:path]; + NSString *file; + + while (file = [dirEnum nextObject]) + { + if ([[file pathExtension] isEqualToString:@"gz"]) + { + NSMutableDictionary *spec = [NSMutableDictionary dictionaryWithCapacity:100]; + + // uncompress and read plist + NSTask *t = [[[NSTask alloc] init] autorelease]; + NSPipe *p = [[[NSPipe alloc] init] autorelease]; + NSFileHandle *rh = [p fileHandleForReading]; + + [t setLaunchPath:@"/usr/bin/gzcat"]; + [t setArguments:[NSArray arrayWithObject:[path stringByAppendingPathComponent:file]]]; + [t setStandardOutput:p]; + [t launch]; + + NSData *inData = nil; + NSMutableData *totalData = [[[NSMutableData alloc] init] autorelease]; + + while ((inData = [rh availableData]) && [inData length]) { + [totalData appendData:inData]; + } + + [t waitUntilExit]; + NSString *plistError = nil; + NSPropertyListSerialization *profile = [NSPropertyListSerialization + propertyListFromData:totalData mutabilityOption:0 + format:NULL + errorDescription:&plistError]; + + // Interesting stuff in SPHardwareDataType: + // serial_number, physical_memory, machine_model, bus_speed, current_processor_speed, cpu_type, l2_cache + NSUInteger i, count = [profile count]; + for (i = 0; i < count; i++) + { + NSDictionary *obj = [profile objectAtIndex:i]; + if ([[obj valueForKey:@"_dataType"] isEqualToString:@"SPHardwareDataType"]) + { + NSArray *items = [obj objectForKey:@"_items"]; + NSUInteger i, count = [items count]; + for (i = 0; i < count; i++) + { + NSDictionary *obj = [items objectAtIndex:i]; + [spec setObject:[obj valueForKey:@"serial_number"] forKey:@"Serial Number"]; + [spec setObject:[obj valueForKey:@"physical_memory"] forKey:@"RAM"]; + [spec setObject:[obj valueForKey:@"machine_model"] forKey:@"Model"]; + [spec setObject:[obj valueForKey:@"bus_speed"] forKey:@"Bus Speed"]; + [spec setObject:[obj valueForKey:@"current_processor_speed"] forKey:@"CPU Speed"]; + [spec setObject:[obj valueForKey:@"cpu_type"] forKey:@"CPU Type"]; + [spec setObject:[obj valueForKey:@"l2_cache"] forKey:@"CPU Cache"]; + } + } + + // Interesting stuff in SPNetworkDataType: + // Ethernet/MAC Address + /* + if ([[obj valueForKey:@"_dataType"] isEqualToString:@"SPNetworkDataType"]) + { + NSArray *items = [obj objectForKey:@"_items"]; + NSUInteger i, count = [items count]; + for (i = 0; i < count; i++) { + NSDictionary *obj = [items objectAtIndex:i]; + [spec setObject:[obj valueForKey:@"serial_number"] forKey:@"Serial Number"]; + [spec setObject:[obj valueForKey:@"physical_memory"] forKey:@"RAM"]; + [spec setObject:[obj valueForKey:@"machine_model"] forKey:@"Model"]; + [spec setObject:[obj valueForKey:@"bus_speed"] forKey:@"Bus Speed"]; + [spec setObject:[obj valueForKey:@"current_processor_speed"] forKey:@"CPU Speed"]; + [spec setObject:[obj valueForKey:@"cpu_type"] forKey:@"CPU Type"]; + [spec setObject:[obj valueForKey:@"l2_cache"] forKey:@"CPU Cache"]; + } + } + */ + + // Interesting stuff in SPSoftwareDataType: + // 64bit_kernel_and_kexts, os_version, user_name, uptime + if ([[obj valueForKey:@"_dataType"] isEqualToString:@"SPSoftwareDataType"]) + { + NSArray *items = [obj objectForKey:@"_items"]; + NSUInteger i, count = [items count]; + for (i = 0; i < count; i++) { + NSDictionary *obj = [items objectAtIndex:i]; + [spec setObject:[obj valueForKey:@"64bit_kernel_and_kexts"] forKey:@"64bit kernel"]; + [spec setObject:[obj valueForKey:@"os_version"] forKey:@"OS Version"]; + [spec setObject:[obj valueForKey:@"user_name"] forKey:@"Current User"]; + [spec setObject:[obj valueForKey:@"uptime"] forKey:@"Uptime"]; + } + } + + } + + [outData addObject:[NSDictionary dictionaryWithDictionary:spec]]; + + } + + } + + [self setComparisonData:outData]; + [resultsTable setDataSource:self]; + [fm release]; + + md = [NSApp beginModalSessionForWindow:resultsWindow]; + +} + +- (int)numberOfRowsInTableView:(NSTableView *)aTableView +{ + return [[self comparisonData] count]; +} + +- (IBAction)export_results:(id)sender +{ + NSSavePanel *panel = [NSSavePanel savePanel]; + [panel setCanCreateDirectories:YES]; + NSString *fileName = [NSString stringWithFormat:@"%@.csv", @"Audit Results"]; + [panel setNameFieldStringValue:fileName]; + [panel beginSheetForDirectory:nil + file:fileName + modalForWindow:resultsWindow + modalDelegate:self didEndSelector:@selector(saveResultsPanelDidEnd:returnCode:contextInfo:) + contextInfo:NULL]; + +} + +-(id)tableView:(NSTableView *)aTableView objectValueForTableColumn:(NSTableColumn *)tableColumn row:(int)rowIndex +{ + NSDictionary *dataRecord = [[self comparisonData] objectAtIndex:rowIndex]; + return [dataRecord objectForKey:[tableColumn identifier]]; +} + +- (IBAction)compare_results:(id)sender +{ + NSOpenPanel *panel = [NSOpenPanel openPanel]; + [panel setCanChooseFiles:NO]; + [panel setCanChooseDirectories:YES]; + + [panel beginSheetForDirectory:nil + file:nil modalForWindow:window + modalDelegate:self + didEndSelector:@selector(openPanelDidEnd:returnCode:contextInfo:) + contextInfo:NULL]; +} + +- (void)applicationDidFinishLaunching:(NSNotification *)aNotification { + + [window center]; + + NSProcessInfo *pi = [NSProcessInfo processInfo]; + hostName = [pi hostName]; + + supportedMailers = [NSArray arrayWithObjects:@"com.apple.mail", @"com.microsoft.entourage", nil]; + + NSArray *dataTypes = [NSArray arrayWithObjects:@"SPHardwareDataType", @"SPNetworkDataType", + @"SPSoftwareDataType", @"SPAudioDataType", @"SPBluetoothDataType", + @"SPDiagnosticsDataType", @"SPEthernetDataType", @"SPFireWireDataType", + @"SPDisplaysDataType", @"SPPrintersDataType", @"SPSerialATADataType", + @"SPUSBDataType", @"SPAirPortDataType", @"SPFirewallDataType", + @"SPNetworkLocationDataType", @"SPNetworkVolumeDataType", + @"SPApplicationsDataType", @"SPExtensionsDataType", nil]; + + id helper; + NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults]; + + [defaults addSuiteNamed:@"com.apple.launchservices"]; + NSArray *helpers = [defaults objectForKey:@"LSHandlers"]; + NSEnumerator *helperEnum = [helpers objectEnumerator]; + mailer = [[[NSString alloc] init] autorelease]; + + while (helper = [helperEnum nextObject]){ + NSString *key = [helper valueForKey:@"LSHandlerURLScheme"]; + if ([key isEqualToString:@"mailto"]) { + mailer = [helper valueForKey:@"LSHandlerRoleAll"]; + } + } + + // Run a minimal system profile to get some basic info... + NSTask *t = [[[NSTask alloc] init] autorelease]; + NSArray *args = [NSArray arrayWithObjects:@"SPHardwareDataType", @"-xml", @"-detailLevel", @"full", nil]; + NSPipe *p = [[[NSPipe alloc] init] autorelease]; + + [t setLaunchPath:@"/usr/sbin/system_profiler"]; + [t setStandardOutput:p]; + [t setArguments:args]; + [t launch]; + + NSData *inData = nil; + NSFileHandle *rh = [p fileHandleForReading]; + NSMutableData *totalData = [[[NSMutableData alloc] init] autorelease]; + + while ((inData = [rh availableData]) && [inData length]) { + [totalData appendData:inData]; + } + + [t waitUntilExit]; + NSString *plistError = nil; + NSPropertyListSerialization *profile = [NSPropertyListSerialization + propertyListFromData:totalData + mutabilityOption:0 + format:NULL + errorDescription:&plistError]; + + NSArray *info = [profile lastObject]; + NSDictionary *items = [[info objectForKey:@"_items"] firstObject]; + + [label_ram setStringValue:[items valueForKey:@"physical_memory"]]; + [label_machine setStringValue:[items valueForKey:@"machine_name"]]; + [label_serial setStringValue:[items valueForKey:@"serial_number"]]; + [label_os setStringValue:[pi operatingSystemVersionString]]; + +} + +-(void) setReportFile:(NSString *)aFile +{ + aFile = [aFile copy]; + [reportFile release]; + reportFile = aFile; +} + +-(void) setComparisonData:(NSMutableArray *)someData +{ + someData = [someData copy]; + [comparisonData release]; + comparisonData = someData; +} + +@end |