Fix --remove-modules command

This commit is contained in:
topjohnwu
2020-04-30 01:26:50 -07:00
parent 03c1053871
commit 5fd574a14f
5 changed files with 46 additions and 59 deletions

View File

@@ -56,6 +56,14 @@ static void *request_handler(void *args) {
close(client);
return nullptr;
}
break;
case REMOVE_MODULES:
if (credential.uid != UID_SHELL && credential.uid != UID_ROOT) {
write_int(client, 1);
close(client);
return nullptr;
}
break;
default:
break;
}
@@ -88,12 +96,8 @@ static void *request_handler(void *args) {
exec_sql(client);
break;
case REMOVE_MODULES:
if (credential.uid == UID_SHELL || credential.uid == UID_ROOT) {
remove_modules();
write_int(client, 0);
} else {
write_int(client, 1);
}
remove_modules();
write_int(client, 0);
close(client);
break;
case GET_PATH: