aboutsummaryrefslogtreecommitdiffstats
path: root/got.py
blob: 850792e497f7fe8bc84b15a27c6ffe6d5f3a7827 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#!/usr/bin/env python3
'''
gates-of-tartaros – Minimal replacement for agetty with SSH support

Copyright © 2013  Mattias Andrée (maandree@member.fsf.org)

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
'''


import os
import sys
import time
from subprocess import Popen


for arg in sys.argv[1:]:
    if "=" in arg:
        os.putenv(arg.split("=")[0], "=".join(arg.split("=")[1:]))

def spawn(cmd):
    Popen(cmd, stdin = sys.stdin, stdout = sys.stdout, stderr = sys.stderr).wait()


user = input()
if "@" in user:
    os.putenv("GOT_COMMAND", "ssh " + user)
    spawn(["login", "-p", "-f", "sshlogin"])
else:
    if user.startswith("-"):
        spawn(["stty", "-icanon", "-echo", "-isig", "-ixon", "-ixoff", "-ixany"])
        print("\033[01;31mNice try! Sleeping for 10 minutes...\033[00m")
        for _ in range(10):
            time.sleep(60 * 1000)
    else:
        spawn(["login", user])