-
Notifications
You must be signed in to change notification settings - Fork 0
/
hardwarefaults.py
41 lines (30 loc) · 1.3 KB
/
hardwarefaults.py
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
import os
import json
import psutil
import slowoom as so
def execute(command, key, username, ip, fqdn, op, node):
if 'all' in command:
try:
parameter = ''
if op == 'oom' or op == 'stress':
parameter = command.split()[2]
print("%s %s %s" % (op, ip, parameter))
os.system("ssh -i %s %s@%s nohup 'bash -s' < ops/%s.sh %s &" % (key, username, ip, op, parameter))
except:
print()
elif fqdn in command or ip in command:
try:
parameter = ''
# get the parameters needed for oom and stress (memory and cpu utilization)
if op == 'oom' or 'stress' in op:
parameter = command.split()[2]
loop = 1
# if there is a speed parameter given and the operation is slowoom, set the speed
if command.split()[4].isdigit() and op == 'slowoom':
loop = command.split()[4]
print("%s %s %s %s" % (op, ip, parameter, loop))
# perform operation n times (higher speed = faster memory filling)
for i in range(int(loop)):
os.system("ssh -i %s %s@%s nohup 'bash -s' < ops/%s.sh %s &" % (key, username, ip, op, parameter))
except:
print()