blob: 2a6f175026f3a1b11e53c0a6e62e6fa357697f37 [file] [log] [blame]
#!/usr/bin/env python
# TODO: openbmc/openbmc#2994 remove python 2 support
try: # python 2
import gobject
except ImportError: # python 3
from gi.repository import GObject as gobject
import dbus
import dbus.service
import dbus.mainloop.glib
from obmc.dbuslib.bindings import get_dbus, DbusProperties, DbusObjectManager
DBUS_NAME = 'org.openbmc.control.Chassis'
OBJ_NAME = '/org/openbmc/control/chassis0'
CONTROL_INTF = 'org.openbmc.Control'
MACHINE_ID = '/etc/machine-id'
POWER_OFF = 0
POWER_ON = 1
BOOTED = 100
class ChassisControlObject(DbusProperties, DbusObjectManager):
def getUuid(self):
uuid = ""
try:
with open(MACHINE_ID) as f:
data = f.readline().rstrip('\n')
if (len(data) == 32):
uuid = data
else:
print("ERROR: UUID is not formatted correctly: " + data)
except Exception:
print("ERROR: Unable to open uuid file: " + MACHINE_ID)
return uuid
def __init__(self, bus, name):
super(ChassisControlObject, self).__init__(
conn=bus,
object_path=name)
# load utilized objects
self.dbus_objects = {
'power_control': {
'bus_name': 'org.openbmc.control.Power',
'object_name': '/org/openbmc/control/power0',
'interface_name': 'org.openbmc.control.Power'
},
'host_services': {
'bus_name': 'org.openbmc.HostServices',
'object_name': '/org/openbmc/HostServices',
'interface_name': 'org.openbmc.HostServices'
},
'systemd': {
'bus_name': 'org.freedesktop.systemd1',
'object_name': '/org/freedesktop/systemd1',
'interface_name': 'org.freedesktop.systemd1.Manager'
},
}
# uuid
self.Set(DBUS_NAME, "uuid", self.getUuid())
self.Set(DBUS_NAME, "reboot", 0)
bus.add_signal_receiver(self.power_button_signal_handler,
dbus_interface="org.openbmc.Button",
signal_name="Released",
path="/org/openbmc/buttons/power0")
bus.add_signal_receiver(self.long_power_button_signal_handler,
dbus_interface="org.openbmc.Button",
signal_name="PressedLong",
path="/org/openbmc/buttons/power0")
bus.add_signal_receiver(self.softreset_button_signal_handler,
dbus_interface="org.openbmc.Button",
signal_name="Released",
path="/org/openbmc/buttons/reset0")
bus.add_signal_receiver(self.host_watchdog_signal_handler,
dbus_interface="org.openbmc.Watchdog",
signal_name="WatchdogError")
def getInterface(self, name):
o = self.dbus_objects[name]
obj = bus.get_object(o['bus_name'], o['object_name'], introspect=False)
return dbus.Interface(obj, o['interface_name'])
@dbus.service.method(DBUS_NAME,
in_signature='', out_signature='')
def powerOn(self):
print("Turn on power and boot")
self.Set(DBUS_NAME, "reboot", 0)
intf = self.getInterface('systemd')
f = getattr(intf, 'StartUnit')
f.call_async('obmc-host-start@0.target', 'replace')
return None
@dbus.service.method(DBUS_NAME,
in_signature='', out_signature='')
def powerOff(self):
print("Turn off power")
intf = self.getInterface('systemd')
f = getattr(intf, 'StartUnit')
f.call_async('obmc-chassis-hard-poweroff@0.target', 'replace')
return None
@dbus.service.method(DBUS_NAME,
in_signature='', out_signature='')
def softPowerOff(self):
print("Soft off power")
intf = self.getInterface('systemd')
f = getattr(intf, 'StartUnit')
f.call_async('obmc-host-shutdown@0.target', 'replace')
return None
@dbus.service.method(DBUS_NAME,
in_signature='', out_signature='')
def reboot(self):
print("Rebooting")
if self.getPowerState() == POWER_OFF:
self.powerOn()
else:
self.Set(DBUS_NAME, "reboot", 1)
self.powerOff()
return None
@dbus.service.method(DBUS_NAME,
in_signature='', out_signature='')
def softReboot(self):
print("Soft Rebooting")
if self.getPowerState() == POWER_OFF:
self.powerOn()
else:
self.Set(DBUS_NAME, "reboot", 1)
self.softPowerOff()
return None
@dbus.service.method(DBUS_NAME,
in_signature='', out_signature='')
def quiesce(self):
intf = self.getInterface('systemd')
f = getattr(intf, 'StartUnit')
f.call_async('obmc-host-quiesce@0.target', 'replace')
return None
@dbus.service.method(DBUS_NAME,
in_signature='', out_signature='i')
def getPowerState(self):
intf = self.getInterface('power_control')
return intf.getPowerState()
# Signal handler
def power_button_signal_handler(self):
# toggle power, power-on / soft-power-off
state = self.getPowerState()
if state == POWER_OFF:
self.powerOn()
elif state == POWER_ON:
self.softPowerOff()
def long_power_button_signal_handler(self):
print("Long-press button, hard power off")
self.powerOff()
def softreset_button_signal_handler(self):
self.softReboot()
def host_watchdog_signal_handler(self):
print("Watchdog Error, Going to quiesce")
self.quiesce()
if __name__ == '__main__':
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
bus = get_dbus()
obj = ChassisControlObject(bus, OBJ_NAME)
mainloop = gobject.MainLoop()
obj.unmask_signals()
name = dbus.service.BusName(DBUS_NAME, bus)
print("Running ChassisControlService")
mainloop.run()
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4