From bd775f017f77043aac5b6290c9899f7daf23e172 Mon Sep 17 00:00:00 2001
From: TitanSnow <sweeto@live.cn>
Date: Mon, 22 May 2017 19:37:24 +0800
Subject: add module privilege to manage root privilege

---
 xmake/core/base/privilege.lua | 92 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 xmake/core/base/privilege.lua

(limited to 'xmake/core/base/privilege.lua')

diff --git a/xmake/core/base/privilege.lua b/xmake/core/base/privilege.lua
new file mode 100644
index 000000000..5c66ba5f1
--- /dev/null
+++ b/xmake/core/base/privilege.lua
@@ -0,0 +1,92 @@
+--!The Make-like Build Utility based on Lua
+--
+-- Licensed to the Apache Software Foundation (ASF) under one
+-- or more contributor license agreements.  See the NOTICE file
+-- distributed with this work for additional information
+-- regarding copyright ownership.  The ASF licenses this file
+-- to you under the Apache License, Version 2.0 (the
+-- "License"); you may not use this file except in compliance
+-- with the License.  You may obtain a copy of the License at
+--
+--     http://www.apache.org/licenses/LICENSE-2.0
+--
+-- Unless required by applicable law or agreed to in writing, software
+-- distributed under the License is distributed on an "AS IS" BASIS,
+-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+-- See the License for the specific language governing permissions and
+-- limitations under the License.
+--
+-- Copyright (C) 2015 - 2017, TBOOX Open Source Group.
+--
+-- @author      TitanSnow
+-- @file        privilege.lua
+--
+
+-- define module
+local privilege = privilege or {}
+
+-- load modules
+local os = require("base/os")
+
+-- store privilege
+function privilege.store()
+    -- check if root
+    if not os.isroot() then
+        return false
+    end
+
+    -- find projectdir's owner
+    local projectdir = xmake._PROJECT_DIR
+    assert(projectdir)
+    local owner = os.getown(projectdir)
+    if not owner then
+        -- fallback to current dir
+        owner = os.getown(".")
+        if not owner then
+            return false
+        end
+    end
+
+    -- set gid
+    if os.gid(owner.gid).setegid_errno ~= 0 then
+        return false
+    end
+
+    -- set uid
+    if os.uid(owner.uid).seteuid_errno ~= 0 then
+        return false
+    end
+
+    -- set flag
+    privilege._HAS_PRIVILEGE = true
+
+    -- ok
+    return true
+end
+
+-- check if has stored privilege
+function privilege.has()
+    return privilege._HAS_PRIVILEGE or false
+end
+
+function privilege.get()
+    -- has?
+    if privilege._HAS_PRIVILEGE ~= true then
+        return false
+    end
+
+    -- set uid
+    if os.uid(0).seteuid_errno ~= 0 then
+        return false
+    end
+
+    -- set gid
+    if os.gid(0).setegid_errno ~= 0 then
+        return false
+    end
+
+    return true
+end
+
+-- return module
+return privilege
-- 
cgit v1.3.1


From 2cb7a91bf266680266de51636ce6620cbecc9362 Mon Sep 17 00:00:00 2001
From: TitanSnow <sweeto@live.cn>
Date: Tue, 23 May 2017 07:59:21 +0800
Subject: fix privilege

---
 xmake/core/base/privilege.lua | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'xmake/core/base/privilege.lua')

diff --git a/xmake/core/base/privilege.lua b/xmake/core/base/privilege.lua
index 5c66ba5f1..275a7e7b5 100644
--- a/xmake/core/base/privilege.lua
+++ b/xmake/core/base/privilege.lua
@@ -48,12 +48,12 @@ function privilege.store()
     end
 
     -- set gid
-    if os.gid(owner.gid).setegid_errno ~= 0 then
+    if os.gid(owner.gid).errno ~= 0 then
         return false
     end
 
     -- set uid
-    if os.uid(owner.uid).seteuid_errno ~= 0 then
+    if os.uid({["ruid"] = owner.uid}).errno ~= 0 or os.uid({["euid"] = owner.uid}).errno ~= 0 then
         return false
     end
 
@@ -76,12 +76,12 @@ function privilege.get()
     end
 
     -- set uid
-    if os.uid(0).seteuid_errno ~= 0 then
+    if os.uid({["euid"] = 0}).errno ~= 0 or os.uid({["ruid"] = 0}).errno ~= 0 then
         return false
     end
 
     -- set gid
-    if os.gid(0).setegid_errno ~= 0 then
+    if os.gid(0).errno ~= 0 then
         return false
     end
 
-- 
cgit v1.3.1