nikdoof/oblogout
1
0
Fork 0
mirror of https://github.com/nikdoof/oblogout.git synced 2025-12-23 07:19:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed dbus auth functions, now takes a more brute force attempt if

Browse Source 
PolicyKit object is acting up (bug?)
This commit is contained in:
Andrew Williams
2009-03-29 16:05:01 +01:00
parent f3412e1eba
commit 704e07ba95
1 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
oblogout/dbushandler.py
View File

@@ -31,30 +31,36 @@ class DbusController (object):
def __check_perms(self, id):
""" Check if we have permissions for a action """
self.logger.debug('Checking permissions for %s' % id)
kit = dbus.SystemBus().get_object('org.freedesktop.PolicyKit','/')
if(kit == None):
print("Error: Could not get PolicyKit D-Bus Interface\n")
return False
if kit.IsProcessAuthorized(id, dbus.UInt32(os.getpid()), False):
try:
res = kit.IsProcessAuthorized(id, dbus.UInt32(os.getpid()), False)
except:
return False
if res:
self.logging.debug("Authorised to use %s" % id)
return True
def __auth_perms(self, id):
""" Check if we have permissions for a action, if not, try to obtain them via PolicyKit """
if self.__check_perms(id):
return True
else:
kit = dbus.SystemBus().get_object('org.freedesktop.PolicyKit','/')
kit = dbus.SessionBus().get_object('org.freedesktop.PolicyKit.AuthenticationAgent','/')
if(kit == None):
print("Error: Could not get PolicyKit D-Bus Interface\n")
return False
return kit.ObtainAuthorization(id, (dbus.UInt32)(xid), (dbus.UInt32)(os.getpid()))
return kit.ObtainAuthorization(id, dbus.UInt32(0), dbus.UInt32(os.getpid()))
def __get_sessions(self):
""" Using DBus and ConsoleKit, get the number of sessions. This is used by PolicyKit to dictate the
@@ -98,7 +104,7 @@ class DbusController (object):
if self.__auth_perms("org.freedesktop.hal.power-management.reboot-multiple-sessions"):
dbus_hal = dbus.SystemBus().get_object("org.freedesktop.Hal", "/org/freedesktop/Hal/devices/computer")
pm = dbus.Interface(dbus_hal, "org.freedesktop.Hal.Device.SystemPowerManagement")
return pm.Reboot()
#return pm.Reboot()
else:
return False
@@ -110,7 +116,7 @@ class DbusController (object):
dbus_hal = dbus.SystemBus().get_object("org.freedesktop.Hal", "/org/freedesktop/Hal/devices/computer")
pm = dbus.Interface(dbus_hal, "org.freedesktop.Hal.Device.SystemPowerManagement")
return pm.Shutdown()
#return pm.Shutdown()
else:
return False
@@ -130,9 +136,8 @@ if __name__ == "__main__":
logging.basicConfig(level=logging.DEBUG)
t = DbusController()
t.restart()
print t.restart()