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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
#!/bin/python3
import serial
import subprocess
import sys
import os
import time
USAGE = f"USAGE: {sys.argv[0]} vm_guest gate ip mask ssh_key"
def send_cmd(ser, delay, cmd):
ser.write(f"{cmd}\n".encode("utf-8"))
time.sleep(delay)
def send_cmds(ser, cmds):
for cmd in cmds:
send_cmd(ser, cmd[0], cmd[1])
def main():
COM = "/dev/"
BAUD = 115200
TIMEOUT = 1
if len(sys.argv) != 7:
sys.stderr.write(USAGE)
sys.exit(1)
GUEST = "vm-tmp"
HOST = sys.argv[1]
cmd = f"vmctl show | grep {GUEST} | tr -s ' ' | cut -d ' ' -f7"
_buffer = subprocess.check_output(cmd, shell=True).decode().rstrip()
print(_buffer)
if _buffer == "":
sys.exit(1)
COM += _buffer
IP = sys.argv[2]
GATE = sys.argv[3]
MASK = sys.argv[4]
DNS = sys.argv[5]
SSHKEY = sys.argv[6]
ser = serial.Serial(COM, BAUD, timeout=TIMEOUT)
send_cmd(ser, 1, "root")
# virtual interface
send_cmds(ser, [
[1, "setup-interfaces"],
[1, ""],
[1, f"{IP}"],
[1, f"{MASK}"],
[1, f"{GATE}"],
[1, "no"],
[1, "ifdown -a"],
[10, "ifup -a"]
])
# dns
send_cmds(ser, [
[1, "setup-dns"],
[1, f"{HOST}"],
[1, f"{DNS}"]
])
# ssh
send_cmds(ser, [
[5, "apk add openssh"],
[1, "mkdir /root/.ssh"],
[1, f"echo '{SSHKEY}' > /root/.ssh/authorized_keys"],
[1, f"echo 'PermitRootLogin yes' >> /etc/ssh/sshd_config"],
[1, "/etc/init.d/sshd restart"]
])
ser.close()
print(COM)
if __name__ == "__main__":
main()
|
