libremanage

   1 #!/usr/bin/python3
   2
   3 """
   4 libremanage - Lightweight, free software for remote side-chanel server management
   5
   6 Copyright (C) 2018 Alyssa Rosenzweig
   7
   8 This program is free software: you can redistribute it and/or modify
   9 it under the terms of the GNU Affero General Public License as published by
  10 the Free Software Foundation, either version 3 of the License, or
  11 (at your option) any later version.
  12
  13 This program is distributed in the hope that it will be useful,
  14 but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 GNU Affero General Public License for more details.
  17
  18 You should have received a copy of the GNU Affero General Public License
  19 along with this program.  If not, see <https://www.gnu.org/licenses/>.
  20 """
  21
  22 USAGE = """
  23 Usage:
  24
  25     $ libremanage [server name] [command]
  26
  27 Example:
  28
  29     $ libremanage web2 reboot
  30
  31 Server names are defined in the accompanying config.py.
  32
  33 Valid commands are as follows:
  34
  35     - shutdown, reboot, poweron: Power management
  36     - tty: Open TTY in GNU Screen
  37     - sanity, sanity-sh: SSH sanity tests, ignore
  38
  39 Define a configuration file in ~/.libremanage.json. See the included
  40 config.json for an example. Servers correspond to managed servers; managers
  41 correspond to single-board computers connecting the servers. libremanage SSHs
  42 into the manager to access the server through the side-channel.
  43 """
  44
  45 import sys
  46 import json
  47 import functools
  48 import subprocess
  49 import time
  50 import os.path
  51
  52 def open_ssh(server, command, force_tty=False):
  53     config = server["ssh"]
  54     args = ["ssh"] + (["-t"] if force_tty else []) + [config["username"] + "@" + config["host"], "-p", str(config["port"]), command]
  55     subprocess.run(args)
  56
  57 def die_with_usage(message):
  58     print(message)
  59     print(USAGE)
  60     sys.exit(1)
  61
  62 def get_server_handle(name):
  63     try:
  64         server = CONFIG["servers"][name]
  65     except KeyError:
  66         die_with_usage("Unknown server, please configure")
  67
  68     # Associate manager configuration
  69     server["ssh"] = CONFIG["managers"][server["manager"]]
  70
  71     # Meta access
  72     server["name"] = name
  73
  74     return server
  75
  76 def gpio_export(server, pin, mode):
  77     if mode:
  78         open_ssh(server, "echo " + str(pin) + " > /sys/class/gpio/export")
  79         open_ssh(server, "echo out > /sys/class/gpio/gpio" + str(pin) + "/direction")
  80     else:
  81         open_ssh(server, "echo " + str(pin) + " > /sys/class/gpio/unexport")
  82
  83 def gpio_write(server, pin, value):
  84     open_ssh(server, "echo " + str(value) + " > /sys/class/gpio/gpio" + str(pin) + "/value")
  85
  86 def power_button(server, pin, state):
  87     # Hold down the power to force off (via the EC),
  88     # or just flick on to turn on
  89
  90     gpio_write(server, pin, 1)
  91     time.sleep(2 if state == POWER_OFF else 0.5)
  92     gpio_write(server, pin, 0)
  93
  94 POWER_OFF    = 0
  95 POWER_ON     = 1
  96 POWER_REBOOT = 2
  97
  98 def set_server_power(state, server):
  99     conf = server["power"]
 100
 101     # TODO: Invert
 102     # Export pin, configure, write value, unexport
 103     pin = conf["pin"]
 104
 105     gpio_export(server, pin, True)
 106
 107     # Act like a power button
 108
 109     if state == POWER_OFF or state == POWER_ON:
 110         power_button(server, pin, state)
 111     elif state == POWER_REBOOT:
 112         # Requires that we already be online.
 113         power_button(server, pin, POWER_OFF)
 114         power_button(server, pin, POWER_ON)
 115
 116     gpio_export(server, pin, False)
 117
 118 def open_tty(s):
 119     if s["tty"]["uncolor"]:
 120         # Broken serial port, workaround TTY garbage with libremanage-serial
 121         subprocess.run(["libremanage-serial", s["name"])
 122     else:
 123         # Use native GNU screen
 124         return open_ssh(s, "screen " + s["tty"]["file"] + " " + str(s["tty"]["baud"]), force_tty=True),
 125
 126 COMMANDS = {
 127         # Power managemment
 128
 129         "shutdown": functools.partial(set_server_power, POWER_OFF),
 130         "poweron": functools.partial(set_server_power, POWER_ON),
 131         "reboot": functools.partial(set_server_power, POWER_REBOOT),
 132
 133         # TTY access (or keyboard if wired as such)
 134
 135         "tty": open_tty,
 136         "tty-baud": lambda s: open_ssh(s, "stty -F "+ s["tty"]["file"] + " " + str(s["tty"]["baud"])),
 137         "tty-read": lambda s: open_ssh(s, "cat " + s["tty"]["file"], force_tty=True),
 138         "tty-write": lambda s: open_ssh(s, "stdbuf -o0 cat > " + s["tty"]["file"], force_tty=True),
 139
 140         # SSH sanity tests
 141
 142         "sanity": lambda s: open_ssh(s, "whoami"),
 143         "sanity-sh": lambda s: open_ssh(s, ""),
 144 }
 145
 146 def issue_command(server_name, command):
 147     server = get_server_handle(server_name)
 148
 149     try:
 150         callback = COMMANDS[command]
 151     except KeyError:
 152         die_with_usage("Invalid command supplied")
 153
 154     callback(server)
 155
 156 # Load configuration, get command, and go!
 157
 158 try:
 159     with open(os.path.expanduser("~/.libremanage.json")) as f:
 160         CONFIG = json.load(f)
 161 except FileNotFoundError:
 162     die_with_usage("Configuration file missing in ~/.libremanage.json")
 163
 164 if len(sys.argv) != 3:
 165     die_with_usage("Incorrect number of arguments")
 166
 167 issue_command(sys.argv[1], sys.argv[2])