-
Notifications
You must be signed in to change notification settings - Fork 0
/
NSFileManagerAdditions.m
113 lines (94 loc) · 2.64 KB
/
NSFileManagerAdditions.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
//
// NSFileManagerAdditions.m
// TRKit
//
#import "NSFileManagerAdditions.h"
#import <Carbon/Carbon.h>
#import <Security/Security.h>
#import "stdlib.h"
#import "time.h"
static AuthorizationRef authorizationRef = NULL;
@implementation NSFileManager (TRAdditions)
- (NSString *)newTmpFilePath
{
NSString * tmpDirectory = NSTemporaryDirectory();
NSString * identifier = @"Hosts";
if (tmpDirectory == nil || identifier == nil) return nil;
for (;;)
{
srand(time(nil));
int random = rand() % 100000;
NSString * tmpFileName = [NSString stringWithFormat:@"%@_%i",identifier,random];
NSString * path = [tmpDirectory stringByAppendingPathComponent:tmpFileName];
path = [path stringByStandardizingPath];
if ([self fileExistsAtPath:path])
{
continue;
}
else
{
// 'Touch' a file here so that it guarentees that another won't be created (rare chance).
//[self createFileAtPath:path contents:nil attributes:nil];
return path;
}
}
}
- (BOOL)authorizedMovePath:(NSString *)source toPath:(NSString *)destination
{
//NSBundle * trkitBundle = [NSBundle bundleForClass:NSClassFromString(@"TRIntegration")];
//NSString * trkitResourcePath = [[NSBundle mainBundle] resourcePath];
//NSString * trkitMoveUtilityPath = [trkitResourcePath stringByAppendingPathComponent:@"move"];
NSString * trkitMoveUtilityPath = @"/bin/mv";
if (![self fileExistsAtPath:trkitMoveUtilityPath])
{
NSLog(@"Cannot find move utility.");
return NO;
}
/* The move utlity exists, we can now procede. */
OSStatus status;
if (authorizationRef == NULL)
{
// Get Authorization.
status = AuthorizationCreate(NULL,
kAuthorizationEmptyEnvironment,
kAuthorizationFlagDefaults,
&authorizationRef);
}
else
{
status = noErr;
}
// Make sure we have authorization.
if (status != noErr)
{
NSLog(@"Could not get authorization, failing.");
return NO;
}
// Set up the arguments.
char * args[4];
args[0] = "-f";
args[1] = (char *)[[source stringByStandardizingPath] UTF8String];
args[2] = (char *)[[destination stringByStandardizingPath] UTF8String];
args[3] = NULL;
status = AuthorizationExecuteWithPrivileges(authorizationRef,
[[trkitMoveUtilityPath stringByStandardizingPath] UTF8String],
0, args, NULL);
if (status != noErr)
{
NSLog(@"Could not move file.");
return NO;
}
else
{
return YES;
}
return NO;
}
- (BOOL)authorizedCopyPath:(NSString *)source toPath:(NSString *)destination
{
NSString * tempFile = [self newTmpFilePath];
[self copyItemAtPath:source toPath:tempFile error:nil];
[self authorizedMovePath:tempFile toPath:destination];
return YES;
}
@end