Skip to content

Commit f8bdb0c

Browse files
committed
加入授权判断
1 parent 726fe95 commit f8bdb0c

File tree

2 files changed

+14
-9
lines changed

2 files changed

+14
-9
lines changed

mdserver/mdserver/AppDelegate.m

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ -(void)AuthorizeCmd:(NSString *)cmdPath
123123
#pragma mark 获取cmd执行特权
124124
#define ADMIN_PRIVILEGE "system.privilege.admin"
125125
#pragma mark 给执行文件授权
126-
-(void)AuthorizeCreate
126+
-(BOOL)AuthorizeCreate
127127
{
128128
NSString *app_dir = [NSCommon getAppDir];
129129
NSString *addhost = [NSString stringWithFormat:@"%@Contents/Resources/addhost", app_dir];
@@ -136,7 +136,7 @@ -(void)AuthorizeCreate
136136

137137
if (self->_authRef) {
138138
//NSLog(@"ok");
139-
return;
139+
return true;
140140
}else{
141141
UInt32 count = (UInt32)[list count];
142142
AuthorizationItem authItem[count];
@@ -160,11 +160,14 @@ -(void)AuthorizeCreate
160160
OSStatus status = AuthorizationCreate(&authRights, kAuthorizationEmptyEnvironment, flags, &self->_authRef);
161161
if(status != errAuthorizationSuccess){
162162
NSLog(@"AuthorizationCreate failed!");
163-
return;
164163
}else{
165164
NSLog(@"AuthorizationCreate ok!");
165+
return true;
166166
}
167167
}
168+
169+
[self userCenter:@"授权失败!"];
170+
return false;
168171
}
169172

170173

@@ -536,10 +539,11 @@ - (void)stopWebService
536539
#pragma mark - 重置服务 -
537540
-(IBAction)reloadSVC:(id)sender
538541
{
539-
[self AuthorizeCreate];
540-
NSString *rootDir = [NSCommon getRootDir];
541-
NSString *reloadSVC = [NSString stringWithFormat:@"%@bin/reloadSVC.sh", rootDir];
542-
[self AuthorizeExeCmd:reloadSVC];
542+
if ([self AuthorizeCreate]){
543+
NSString *rootDir = [NSCommon getRootDir];
544+
NSString *reloadSVC = [NSString stringWithFormat:@"%@bin/reloadSVC.sh", rootDir];
545+
[self AuthorizeExeCmd:reloadSVC];
546+
}
543547
}
544548

545549
#pragma mark 启动时清空内容
@@ -579,8 +583,9 @@ -(void)selfStart
579583

580584
#pragma mark - 按钮启动 -
581585
- (IBAction)start:(id)sender {
582-
[self AuthorizeCreate];
583-
[self selfStart];
586+
if ([self AuthorizeCreate]){
587+
[self selfStart];
588+
}
584589
}
585590

586591
#pragma mark 跳到开发地址

0 commit comments

Comments
 (0)