obirvalger/distrobuilder
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Move check for root to Go

Browse Source 
Signed-off-by: Stéphane Graber <stgraber@ubuntu.com>
This commit is contained in:
Stéphane Graber 2018-02-20 02:14:23 -05:00
parent ce823c03c1
commit 2538b49e07
No known key found for this signature in database
GPG Key ID: C638974D64792D67
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
distrobuilder/main.go
View File

@ -16,8 +16,7 @@ __attribute__((constructor)) void init(void) {
int ret; int ret;
if (geteuid() != 0) { if (geteuid() != 0) {
fprintf(stderr, "Need to run as root\n"); return;
_exit(1);
} }
// Unshare a new mntns so our mounts don't leak // Unshare a new mntns so our mounts don't leak
@ -130,6 +129,11 @@ func run(c *cli.Context) error {
arch string arch string
) )
// Sanity checks
if os.Geteuid() != 0 {
return fmt.Errorf("You must be root to run this tool")
}
os.RemoveAll(c.GlobalString("cache-dir")) os.RemoveAll(c.GlobalString("cache-dir"))
os.MkdirAll(c.GlobalString("cache-dir"), 0755) os.MkdirAll(c.GlobalString("cache-dir"), 0755)