Chameleon Applications

Chameleon Applications Svn Source Tree

Root/trunk/ChameleonPrefPane/Sources/ChameleonPrefPane.mm

1//␊
2// StartupPrefPanePref.mm␊
3//␊
4// Created by Rekursor on 1/16/10.␊
5//␊
6␊
7#import "ChameleonPrefPane.h"␊
8#import "BootSetupController.h"␊
9#import "BootFlagsController.h"␊
10#import "PeripheralsController.h"␊
11#import "AdvancedSetupController.h"␊
12␊
13#include "ShellProcess.h"␊
14#import "ChameleonPropertyList.h"␊
15#include <string>␊
16␊
17//--------------------------------------------------------------------------␊
18// Constants␊
19//--------------------------------------------------------------------------␊
20static const char * const szBootPaths[]= {␊
21␉"/",␊
22␉"/Extra/",␊
23␉"/Volumes/EFI/Extra/",␊
24␉"/Volumes/Cham/Extra/",␊
25␉"/Volumes/BootLoaders/Extra/",␊
26␉"/Volumes/RX0/Extra/",␊
27␉"/Library/Preferences/SystemConfiguration/",␊
28␉NULL␊
29};␊
30␊
31static const char* const szPropFileName = "com.apple.Boot.plist";␊
32␊
33static const int CurrentPreferencesFileVersion = 0x02; // for future back compatibility␊
34//--------------------------------------------------------------------------␊
35// Static file variables␊
36//--------------------------------------------------------------------------␊
37static std::string sCurrentDefaultPartition;␊
38␊
39PartitionExtractor * partExtractor=NULL;␊
40␊
41static int currentRowSel = -1;␊
42static ChameleonPrefPane *gInstance = NULL;␊
43␊
44//--------------------------------------------------------------------------␊
45␊
46@implementation ChameleonPrefPane␊
47␊
48+ (ChameleonPrefPane *)instance { return(gInstance);}␊
49␊
50//--------------------------------------------------------------------------␊
51- (id) init␊
52{␊
53␉self = [super init];␊
54␉gInstance = self;␊
55␉␊
56␉// set the image to display the current file being played␊
57␉mMacOSXImage = [self getImageResource: @"MacOSX" ofType: @"png"];␊
58␉mWindowsImage = [self getImageResource: @"Windows" ofType: @"png"];␊
59␉mLinuxImage = [self getImageResource: @"Linux" ofType: @"png"];␊
60␉mCDROMImage = [self getImageResource: @"CDROM" ofType: @"png"];␊
61␉mUnknownImage = [self getImageResource: @"Chameleon" ofType: @"tiff"];␊
62␉␊
63␉␊
64 // create the process that will extract the diskutil list infos␊
65␉if(!partExtractor) partExtractor = new PartitionExtractor();␊
66␉␊
67␉// Retrieve the com.chameleon.prefPane.plist config␊
68␉return self;␊
69}␊
70- (void)dealloc␊
71{␊
72␉// release the colors␊
73␉[mOptionsDict release];␊
74␉[mPartitionsDict release];␊
75␉␊
76␉[super dealloc];␊
77}␊
78␊
79- (NSMutableDictionary*) preferencesFile {␉return mOptionsDict; }␊
80- (NSMutableDictionary*) preferencesParts {␉return mPartitionsDict; }␊
81- (NSTableView*) partitionsTable { return mPartitionsTable;}␊
82␊
83//--------------------------------------------------------------------------␊
84-(bool) savePreferences: (NSDictionary*) dict␊
85{␊
86␉std::string sPath = [kPreferencesFilePath UTF8String];␊
87␉␊
88␉if(dict==nil || sPath.length()==0) return false;␊
89␉␊
90␉AuthorizationRef auth = [self isUnlocked] ? [[authView authorization] authorizationRef] : NULL;␊
91␉␊
92␉PropertyList::chmodFile(sPath.c_str(), "0777", auth);␊
93␉[dict writeToFile:kPreferencesFilePath atomically:YES];␊
94␉PropertyList::chmodFile(sPath.c_str(), "0644", auth);␊
95␉return true;␊
96}␊
97␊
98//--------------------------------------------------------------------------␊
99// SFAuthorization implementation␊
100//--------------------------------------------------------------------------␊
101␊
102// SFAuthorization delegates␊
103- (void)authorizationViewDidAuthorize:(SFAuthorizationView *)view {␊
104 [self selectDefaultPartition];␊
105␉[self refreshLockStates];␊
106␊
107} ␊
108␊
109//--------------------------------------------------------------------------␊
110- (void)authorizationViewDidDeauthorize:(SFAuthorizationView *)view {␊
111 [self refreshLockStates];␊
112}␊
113␊
114//--------------------------------------------------------------------------␊
115// Setup security for changing boot options␊
116-(void) initAuthorization␊
117{␊
118 AuthorizationItem items = {kAuthorizationRightExecute, 0, NULL, 0};␊
119 AuthorizationRights rights = {1, &items};␊
120 ␊
121␉[authView setAuthorizationRights:&rights];␊
122 authView.delegate = self;␊
123 [authView updateStatus:nil];␊
124}␊
125//--------------------------------------------------------------------------␊
126- (␉AuthorizationRef) auth␊
127{␊
128␉return [self isUnlocked] ? [[authView authorization] authorizationRef] : NULL;␊
129␉␊
130}␊
131␊
132//--------------------------------------------------------------------------␊
133- (void) refreshLockStates␊
134{␊
135 [mPartitionsTable setEnabled:[self isUnlocked]];␊
136 [mStatusText setEnabled:[self isUnlocked]];␊
137␉␊
138␉// Refresh other panels␊
139␉[PreferencesControllerBase doForEachGroup: RefreshLockStates withOption: nil];␊
140}␊
141␊
142␊
143//--------------------------------------------------------------------------␊
144- (bool)isUnlocked ␊
145{␊
146 return [authView authorizationState] == SFAuthorizationViewUnlockedState;␊
147}␊
148␊
149//--------------------------------------------------------------------------␊
150- (id) getImageResource: (NSString *) str ofType: (NSString*) sType␊
151{␊
152␉NSImage * img=nil;␊
153␉if(!str) return nil;␊
154␉NSBundle * b = [NSBundle bundleForClass:[self class]];␊
155␉NSString* sRes = [b pathForResource: str ofType:sType ];␊
156␉img = [[NSImage alloc] initWithContentsOfFile: sRes];␊
157␉return img;␊
158}␊
159␊
160␊
161//--------------------------------------------------------------------------␊
162/** When called here, all outlets references are initialized */␊
163- (void)awakeFromNib␊
164{ // called more than once, we only need one resource init␊
165␉static bool ft=true;␊
166␉if(ft)␊
167␉{␊
168␉␉ft=false;␊
169␉␉[self␉loadPreferences];␊
170␉␉[self initBootConfig];␊
171␉}␊
172}␊
173␊
174- (IBAction)onAbout: (id)sender␊
175{␊
176␉[mAboutDlg makeKeyAndOrderFront: sender];␊
177}␊
178//--------------------------------------------------------------------------␊
179- (void) loadPreferences␊
180{␊
181//␉PrefsProp::instance().open([kPreferencesFilePath UTF8String], CreateIfNotExist,␊
182//␉␉␉␉␉␉␉␉␉␉␉ [self auth]);␊
183␉␊
184␉// test with preferences file already created and when no prefs exists␊
185␉␊
186␉id oldGlobalPreferences = [ [NSDictionary dictionaryWithContentsOfFile: ␊
187␉␉␉␉␉␉␉␉ kPreferencesFilePath ] retain];␊
188␉␊
189␉mPartitionsDict = [[NSMutableDictionary alloc] init];␊
190␉[mPartitionsDict retain];␊
191␉␊
192␉// Initialize bootConfig desc dict␊
193␉[PreferencesControllerBase doForEachGroup: AddOptionsDesc withOption: nil];␊
194␉␊
195␉if (oldGlobalPreferences!=nil)␊
196␉{␊
197␉␉mPreferenceFileVersion= [[oldGlobalPreferences objectForKey: keyPreferencesFileVersion] intValue ];␊
198␉␉[PreferencesControllerBase doForEachGroup: LoadPreferencesOptions withOption: oldGlobalPreferences];␊
199␉␉[mPartitionsDict addEntriesFromDictionary: [oldGlobalPreferences objectForKey: keyPartitionsList] ];␊
200␉}␊
201␉else ␊
202␉{ // Create a preference plist file with Defaults values␊
203␉␉oldGlobalPreferences = [[NSMutableDictionary alloc] init];␊
204␉␉[PreferencesControllerBase doForEachGroup: SetDefaultValues withOption: oldGlobalPreferences];␊
205␉␉␊
206␉␉// Initialize defaults␊
207␉␉[oldGlobalPreferences setObject: [[NSNumber alloc] initWithInt: CurrentPreferencesFileVersion] ␊
208␉␉␉␉␉␉␉␉ forKey: keyPreferencesFileVersion];␊
209␉␉[oldGlobalPreferences setObject: mPartitionsDict forKey: keyPartitionsList];␊
210␉␉␊
211␉␉// Save the preferences file␊
212␉␉[ self savePreferences:oldGlobalPreferences ];␊
213␉}␊
214␉␊
215␉mOptionsDict = [[NSMutableDictionary alloc] init];␊
216␉[mOptionsDict addEntriesFromDictionary:oldGlobalPreferences];␊
217␉if (mPartitionsDict!=nil) [mPartitionsDict retain];␊
218␉[oldGlobalPreferences release];␊
219}␊
220␊
221//--------------------------------------------------------------------------␊
222- (void) initBootConfig␊
223{␊
224␉static bool ft=true;␊
225␊
226␉[self initAuthorization];␊
227␉␊
228␉if (!BootProp::instance().isValid())␊
229␉{␊
230␉␉std::string sPath;␊
231␉␉CFStringRef errorString=NULL;␊
232␉␉bool cont =true;␊
233␉␉␊
234␉␉id sForcedPath = [mOptionsDict valueForKey: keyForceBootConfigPath];␊
235␉␉const char * szForcedPath = sForcedPath!=nil ? [sForcedPath UTF8String] : NULL;␊
236␉␉if (szForcedPath && *szForcedPath)␊
237␉␉{␊
238␉␉␉cont = !BootProp::instance().open(szForcedPath, [self auth]);␊
239␉␉}␊
240␉␉else {␊
241␉␉␉for(int i=0; szBootPaths[i] && cont; i++)␊
242␉␉␉{␊
243␉␉␉␉sPath = szBootPaths[i];␊
244␉␉␉␉sPath += szPropFileName;␊
245␉␉␉␉cont = !BootProp::instance().open(sPath.c_str(), [self auth]);␊
246␉␉␉}␊
247␉␉}␊
248␉␉if (cont)␊
249␉␉{␊
250␉␉␉if(!ft) return;␊
251␉␉␉ft=false;␊
252␉␉␉[mStatusText setTextColor: [NSColor redColor] ];␊
253␉␉␉if (errorString)␊
254␉␉␉␉[mStatusText setStringValue:[NSString stringWithFormat: @"Error while parsing com.apple.Boot.plist : %@",␊
255␉␉␉␉␉␉␉␉␉␉␉ errorString] ];␊
256␉␉␉else␊
257␉␉␉␉[mStatusText setStringValue: @"Error while searching for com.apple.Boot.plist"];␊
258␉␉␉␊
259␉␉}␊
260␉␉else␊
261␉␉{ ␊
262␉␉␉[mStatusText setTextColor: [NSColor grayColor] ];␊
263␉␉␉NSString* ns = [ [NSString alloc] initWithUTF8String:BootProp::instance().propFilePath() ];␊
264␉␉␉[mStatusText setStringValue: [NSString stringWithFormat: @"bootConfig: %@", ns] ];␊
265␉␉␉[[BootSetupController instance]->mBootConfigPathText setStringValue: ␊
266␉␉␉␉[[NSString alloc] initWithUTF8String: BootProp::instance().propFilePath()] ];␊
267␊
268␉␉}␊
269␉␉␊
270␉␉if (BootProp::instance().isValid())␊
271␉␉{␊
272␉␉␉// read options in the plist file␊
273␉␉␉␊
274␉␉␉partExtractor->hidePartitions(BootProp::instance().getStringForKey(kHidePartition));␊
275␉␉␉partExtractor->renamedPartitions(BootProp::instance().getStringForKey(kRenamePartition));␊
276␉␉␉␊
277␉␉␉// partExtractor->resetSwapping();␊
278␉␉␉id val = [mOptionsDict valueForKey: keySwapHD01];␊
279␉␉␉[[BootSetupController instance]->mSwapHD01 setIntValue: [val intValue] ];␊
280␉␉␉[[BootSetupController instance] doSwapHD: [val boolValue] save: false src:0 dst:1];␊
281␉␉␉␊
282␉␉␉val = [mOptionsDict valueForKey: keySwapHD02];␊
283␉␉␉[[BootSetupController instance]->mSwapHD02 setIntValue: [val intValue] ];␊
284␉␉␉[[BootSetupController instance] doSwapHD: [val boolValue] save: false src:0 dst:2];␊
285␊
286␉␉␉// Load all boot Options into the interface components␊
287␉␉␉[PreferencesControllerBase loadOptionsFromBootFile];␊
288␉␉␉[self selectDefaultPartition];␊
289␉␉␉␉ ␊
290␉␉}␊
291␉␉␊
292␉}␉␊
293}␊
294//--------------------------------------------------------------------------␊
295- (void) selectDefaultPartition␊
296{␊
297 if(!authView) return;␊
298␉␊
299␉[self refreshLockStates];␊
300␉␊
301␉// try to get the current default partition if any␊
302␉if(partExtractor && BootProp::instance().isValid())␊
303␉{␊
304␉␉const char *sdp = BootProp::instance().getStringForKey(kDefaultPartition);␊
305␉␉sCurrentDefaultPartition = sdp ? sdp : ""; ␊
306␉␉if (sCurrentDefaultPartition.size())␊
307␉␉{␊
308␉␉␉int index = partExtractor->getIndexFromHdStringSpec(sCurrentDefaultPartition.c_str());␊
309␉␉␉if (index>=0)␊
310␉␉␉{␊
311␉␉␉␉[mPartitionsTable selectRowIndexes:␊
312␉␉␉␉ [NSIndexSet indexSetWithIndex:index] byExtendingSelection:NO];␊
313␉␉␉␉currentRowSel = index;␊
314␉␉␉}␊
315␉␉}␊
316␉}␊
317␉␊
318}␊
319␊
320//--------------------------------------------------------------------------␊
321// following DieBuch recommendation : using applescript and system events (thanks!):␊
322- (IBAction)onRestart: (id)sender␊
323{␊
324␉NSInteger n = NSRunAlertPanel(@"Restarting OS X", ␊
325␉␉␉␉␉␉␉␉ @"Are you sure you want to restart your computer now ?",␊
326␉␉␉␉␉␉␉␉ @"OK", @"Cancel", nil);␊
327␉if (n==1)␊
328␉{␊
329␉␉AuthorizationRef auth = [[authView authorization] authorizationRef];␊
330␉␉executePrivilegedCmd(auth,"/usr/bin/osascript","-e 'tell app \"System Events\" to restart'");␊
331␉}␊
332␉␊
333}␊
334//--------------------------------------------------------------------------␊
335- (IBAction)onShutdown: (id)sender␊
336{␊
337␉NSInteger n = NSRunAlertPanel(@"Shutting Down OS X", ␊
338␉␉␉␉␉␉␉␉ @"Are you sure you want to shut down your computer now ?",␊
339␉␉␉␉␉␉␉␉ @"OK", @"Cancel", /*ThirdButtonHere:*/nil␊
340␉␉␉␉␉␉␉␉ /*, args for a printf-style msg go here */);␊
341␉if (n==1)␊
342␉{␊
343␉␉system("/usr/bin/osascript -e 'tell app \"System Events\" to shut down'");␊
344␉}␊
345}␊
346␊
347- (IBAction)onSleep: (id)sender␊
348{␊
349␉system("/usr/bin/osascript -e 'tell app \"System Events\" to sleep'");␊
350}␊
351␊
352//--------------------------------------------------------------------------␊
353- (void)tableViewSelectionDidChange:(NSNotification *)notification {␊
354␊
355␉NSTableView* tv= mPartitionsTable;␊
356␉if ([tv selectedRow] != currentRowSel)␊
357␉{␊
358␉␉currentRowSel = [tv selectedRow];␊
359␉␉if (currentRowSel>=0)␊
360␉␉{␊
361␉␉␉const std::vector<PartitionInfo>& partInfo = partExtractor->partList();␊
362␉␉␉char hd[7+1]="hd(n,m)";␊
363␉␉␉hd[3]= ('0'+partInfo[currentRowSel].disk());␊
364␉␉␉hd[5]= ('0'+partInfo[currentRowSel].partition());␉␊
365␉␉␉AuthorizationRef auth= [self isUnlocked] ? [[authView authorization] authorizationRef] : NULL;␊
366␉␉␉if(BootProp::instance().setStringForKey(kDefaultPartition, hd))␊
367␉␉␉ BootProp::instance().save(auth);␊
368␉␉␉[ PreferencesControllerBase loadOptionsFromBootFile];␊
369␉␉}␊
370␉␉else␊
371␉␉{ // no line selected␊
372␉␉␉BootProp::instance().removeKeyAndValue(kDefaultPartition);␊
373␉␉␉[ PreferencesControllerBase loadOptionsFromBootFile];␊
374␉␉␉␊
375␉␉}␊
376␉}␊
377␉␉␊
378}␊
379␊
380//--------------------------------------------------------------------------␊
381- (NSInteger) numberOfRowsInTableView:(NSTableView *)tableView␊
382{␊
383␉int size = partExtractor ? partExtractor->partList().size() : 0;␊
384␉return size;␊
385}␊
386␊
387//--------------------------------------------------------------------------␊
388- (id)tableView:(NSTableView *)tableView objectValueForTableColumn:(NSTableColumn *)tableColumn row:(NSInteger)row␊
389{␊
390␉id ret=nil;␊
391␉␊
392␉const std::vector<PartitionInfo>& partInfo = partExtractor->partList();␊
393␊
394␉if (tableColumn == mPartitionImgColumn) ␊
395␉{␊
396␉␉switch(partInfo[row].imageIndexFromFs())␊
397␉␉{␊
398␉␉␉case 0:␊
399␉␉␉␉ret = mMacOSXImage;␊
400␉␉␉␉break;␊
401␉␉␉case 1:␊
402␉␉␉␉ret = mWindowsImage;␊
403␉␉␉␉break;␊
404␉␉␉case 2:␊
405␉␉␉␉ret = mLinuxImage;␊
406␉␉␉␉break;␊
407␉␉␉default:␊
408␉␉␉␉ret = mUnknownImage;␊
409␉␉␉␉break;␊
410␉␉␉␉␊
411␉␉}␊
412␉}␊
413␉else if (tableColumn == mFileSystemColumn) ␊
414␉{␊
415␉␉ret = [NSString stringWithFormat: @"%s", partInfo[row].cfsType() ];␊
416␉}␊
417␉else if (tableColumn == mPartitionNameColumn) ␊
418␉{␊
419␉␉ret = [NSString stringWithFormat: @"%s", partInfo[row].clabel()␉];␊
420␉}␊
421␉else if (tableColumn == mPartitionIDColumn) ␊
422␉{␊
423␉␉ret= [NSString stringWithFormat: @"hd(%d,%d)", ␊
424␉␉␉␉partInfo[row].disk(),␊
425␉␉␉␉partInfo[row].partition()␊
426␉␉␉␉];␊
427␉} ␊
428␉␊
429␉return ret;␊
430}␊
431//--------------------------------------------------------------------------␊
432␊
433@end␊
434

Archive Download this file

Revision: 69