← Back to branch summary

~sseman/juju-chaos-monkey/add-shutdown

« back to all changes in this revision

Viewing changes to chaos/net.py

  • Committer: seman.said at canonical
  • Date: 2015-04-30 22:55:37 UTC
  • mfrom: (2.1.3 logging)
  • Revision ID: seman.said@canonical.com-20150430225537-ckemarzocgcjwga2
Added logging and timestamp.

Show diffs side-by-side

added added

removed removed

Lines of Context:
chaos/net.py
 
1
import logging
 
2
 
1
3
from chaos_monkey_base import (
2
4
    Chaos,
3
5
    ChaosMonkeyBase,
4
6
)
5
7
from utility import (
6
 
    log,
7
8
    run_shell_command,
8
9
)
9
10
 
22
23
        return cls()
23
24
 
24
25
    def reset(self):
25
 
        log("Net.reset ")
 
26
        logging.info("Net.reset ")
26
27
        cmd = 'ufw reset'
27
28
        run_shell_command(cmd)
28
29
 
29
30
    def default_deny(self):
30
 
        log("Net.default_deny")
 
31
        logging.info("Net.default_deny")
31
32
        cmd = "ufw default deny"
32
33
        run_shell_command(cmd)
33
34
 
34
35
    def default_allow(self):
35
 
        log("Net.default_allow")
 
36
        logging.info("Net.default_allow")
36
37
        cmd = "ufw default allow"
37
38
        run_shell_command(cmd)
38
39
 
39
40
    def allow_ssh(self):
40
 
        log("Net.allow_ssh")
 
41
        logging.info("Net.allow_ssh")
41
42
        cmd = 'ufw allow ssh'
42
43
        run_shell_command(cmd)
43
44
 
44
45
    def deny_ssh(self):
45
 
        log("Net.deny_ssh")
 
46
        logging.info("Net.deny_ssh")
46
47
        cmd = 'ufw deny ssh'
47
48
        run_shell_command(cmd)
48
49
 
49
50
    def deny_all_incoming_and_outgoing_except_ssh(self):
50
 
        log("Net.deny_all_incoming_and_outgoing_except_ssh")
 
51
        logging.info("Net.deny_all_incoming_and_outgoing_except_ssh")
51
52
        self.deny_all_incoming_except_ssh()
52
53
        self.deny_all_outgoing_except_ssh()
53
54
 
54
55
    def allow_all_incoming_and_outgoing(self):
55
 
        log("Net.allow_all_incoming_and_outgoing")
 
56
        logging.info("Net.allow_all_incoming_and_outgoing")
56
57
        self.allow_all_incoming()
57
58
        self.allow_all_outgoing()
58
59
 
59
60
    def deny_all_incoming_except_ssh(self):
60
 
        log("Net.deny_all_incoming_except_ssh")
 
61
        logging.info("Net.deny_all_incoming_except_ssh")
61
62
        self.allow_ssh()
62
63
        self.default_deny()
63
64
 
64
65
    def allow_all_incoming(self):
65
 
        log("Net.allow_all_incoming")
 
66
        logging.info("Net.allow_all_incoming")
66
67
        self.default_allow()
67
68
 
68
69
    def deny_all_outgoing_except_ssh(self):
69
 
        log("Net.deny_all_outgoing_except_ssh")
 
70
        logging.info("Net.deny_all_outgoing_except_ssh")
70
71
        self.allow_ssh()
71
72
        cmd = 'ufw deny out to any'
72
73
        run_shell_command(cmd)
73
74
 
74
75
    def allow_all_outgoing(self):
75
 
        log("Net.allow_all_outgoing")
 
76
        logging.info("Net.allow_all_outgoing")
76
77
        cmd = 'ufw delete deny out to any'
77
78
        run_shell_command(cmd)
78
79
 
79
80
    def deny_port(self, port=8080):
80
 
        log("Net.deny_port port=%s" % port)
 
81
        logging.info("Net.deny_port port=%s" % port)
81
82
        cmd = 'ufw deny ' + str(port)
82
83
        run_shell_command(cmd)
83
84