summaryrefslogtreecommitdiff
path: root/kpov_judge/tasks/snmp_agent_uptime/task.py
blob: 13abf92cc8fb849253acf1be077b7e1be9d87d60 (plain)
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
 #!/usr/bin/env python
# -*- coding: utf-8 -*-

# kpov_random_helpers should be imported by add_assignment.py
# TODO: finish this.

instructions = {
    'si':u"""
Naloga: Postavi tri navidezne racunalnike: 
- SimpleArbiter s sliko diska simpleArbiterDhcp,
- SNMPServer in 
- SNMPClient. 
Napiši program upminutes, ki bo izpisal v minutah koliko casa je racunalnik 
vklopljen.

Postavi ga na SNMPClient v domaci imenik uporabnika test z geslom test. 

Poskrbi, da bo SNMP strežnik prek SNMP pod 
NET-SNMP-EXTEND-MIB::nsExtendOutpucd k	t2Table sporocal, koliko casa je vklopljen 
v sekundah.

Napiši skripto, poimenovano beri.sh, ki prek SNMP prebere vrednost s strežnika
 SNMPServer na OID 1.3.6.1.4.1.8072.1.3.2.4.1.4. 
Postavi jo na SNMP klienta, v domaci imenik uporabnika test z geslom test.

Navodila za ostale metode pa so obarvana rdece oz. komentirana z #

""",
    'en':u"""
Set up three virtual computers, SimpleArbiter with the disk image simpleArbiterDhcp,
SNMPServer and SNMPClient. Write a program called upminutes. This program should output
the uptime of the computer in minutes. Set it up on SNMPClient in the home directory
of the user test with the password test.

Make sure that the SNMP server reports it's uptime in seconds over SNMP under 
NET-SNMP-EXTEND-MIB::nsExtendOutput2Table .

Write a script called beri.sh that reads the value from the OID 1.3.6.1.4.1.8072.1.3.2.4.1.4. 
on SNMPServer. Set it up on SNMPClient in the home directory of the user test.
"""

}


computers = {
    'SNMPClient': {
        'disks': [
            {   'name': 'SNMPClient',

            },
            #{   'name': 'CDROM',
            #    'options':{'readonly': True},
            #    'parts': [],# no parts, no mounting.
            #}
        ],
        'network_interfaces': [{'network': 'net1'}],
        'flavor': 'm1.tiny',
        'config_drive': False
    },
    'SNMPServer': {
        'disks': [
            {   'name': 'SNMPServer',
            },
            #{   'name': 'CDROM',
            #    'options':{'readonly': True},
            #    'parts': [],# no parts, no mounting.
            #}
        ],
        'network_interfaces': [{'network': 'net1'}],
        'flavor': 'm1.tiny',
        'config_drive': False

    },
 
    'SimpleArbiter': {
        'disks': [
            {   'name': 'simpleArbiterDhcp',
                # attempt automount
            },
            #{   'name': 'CDROM',
            #    'options': {'readonly': True},
            #    'parts': [{'dev': 'b1', 'path': '/cdrom'}],
            #},
        ],
        'network_interfaces': [{'network': 'net1'}, {'network': 'test-net'}],
        'flavor': 'm1.tiny',
        'config_drive': False
    }
}

networks = { 'net1': {'public': False}, 'test-net': {'public': True} }

params_meta = {'SNMP_VALUE': {'descriptions': {'si': 'Vrednost, dostopna prek SNMP'}, 'w': False, 'public':True, 'type': 'number', 'generated': True}, 'SNMP_OID_ON_SERVER': {'descriptions': {'si': 'OID, na katerem je vrednost dostopna'}, 'w': False, 'public':True, 'type': 'str', 'generated': True}, 'SNMP_OID_ON_CLIENT': {'descriptions': {'si': 'OID, ki naj ga klient bere'}, 'w': False, 'public':True, 'type': 'OID', 'generated': True}, }

def task(SNMP_VALUE, SNMP_OID_ON_SERVER, SNMP_OID_ON_CLIENT):
	#<== Aleksander Fujs 6310020 ==>
	# TODO popravi IPje
	import netsnmp
	from paramiko import SSHClient
	return_results = {}

	client = SSHClient()
	client.load_system_host_keys()
	client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

	client.connect('10.173.146.199', username='root', password='kaboom')
	stdin, stdout, stderr = client.exec_command('uptime=$(</proc/uptime); uptime=${uptime%%.*}; echo $uptime')
	return_results['server_uptime'] = stdout.readlines()

	try:
		session = netsnmp.Session(DestHost='10.173.146.199', Version=2, Community='studentje')
		results_objs = netsnmp.VarList(netsnmp.Varbind(SNMP_OID_ON_SERVER)) #117.112.116.105.109.101  <-uptime
		session.walk(results_objs)
		for result in results_objs:
		    	return_results['server_OID'] = ('%s.%s: %s') % (result.tag, result.iid, result.val)
	except Exception as exception_error:
	    	# Check for errors and print out results
		print ('ERROR: Occurred during SNMPget for OID %s from %s: '
	       		'(%s)') % (SNMP_OID_ON_SERVER, '10.173.146.194', exception_error)
		sys.exit(2)
	
	
	client.connect('10.173.146.194', username='test', password='test')
	stdin, stdout, stderr = client.exec_command('/bin/bash /home/test/upminutes.sh')
	#TODO preverit da ni v skripti hardcodan
	return_results['client_script'] = stdout.readlines()
	stdin, stdout, stderr = client.exec_command('uptime=$(</proc/uptime); uptime=${uptime%%.*}; echo $(( uptime/60 ))')
	return_results['client_uptime'] = stdout.readlines()

	#TODO add 3 part of assigement
    	return return_results


	#<== Aleksander Fujs 6310020 ==>

    # TODO (polz) start working on this. If gen_params doesn't yet work, change it to set the parameter values manually
    # run cpu temp script on client
    # run SNMP read SNMP_OID_ON_CLIENT script on client
    # read data on SNMP_OID_ON_CLIENT
    # read data from SNMP server
	#import netsnmp
	#session = netsnmp.Session(DestHost=SNMP_VALUE['type'],Version=2,Community='public')
    #Dobimo vrednost casa
	#vars=netsnmp.Varbind(SNMP_OID_ON_SERVER['type'])
    #Shranimo vrednost casa v imenik od SNMP_OID_ON_SERVER
	#var=netsnmp.Varbind(SNMP_OID_ON_SERVER['decdscriptions'['si'],'0',session.get(vars),SNMP_OID_ON_SERVER['type'])
	#netsnmp.snmpset(var,Version=2,DestHost=SNMP_VALUE['type'],Community='public')
	 

    	#return session.get(vars)
	return 0
    
def gen_params(user_id, params_meta):
	import random
	params = dict()
	# TODO (polz) this is wrong. Try something like:
	r = random.Random(user_id)
	#params['SNMP_VALUE'] = r.randint(0, 255)
	#params_meta['SNMP_OID_ON_SERVER'] = '1.3.6.1.4.1.8072.1.3.2.4.1.4'
	#params_meta['SNMP_OID_ON_CLIENT'] = '1.3.6.1.4.1.8072.1.3.2.4.1.4.'
	# You can also create an OID creation function in kpov_random_helpers.
	# this should probably return params_meta

        #<== Aleksander Fujs 6310020 ==>
	#TODO better random value
	params['SNMP_VALUE'] = str(r.randint(0, 255))
	params['SNMP_OID_ON_SERVER'] = 'NET-SNMP-EXTEND-MIB::nsExtendOutput2Table'
	#TODO possible upgrade for last number to bi random between 0 and 255 
	params['SNMP_OID_ON_CLIENT'] = '1.3.6.1.4.1.8072.1.3.2.4.1.4'
	#<== Aleksander Fujs 6310020 ==>

	return params

def task_check(results, params):
	#TODO improve regex
	import re
	score = 0
	print results
	if re.search(results['client_script'][0].strip(), results['client_uptime'][0].strip()):
		score += 3
	#TODO +- second or two of allowed error time 
	if re.search(results['server_OID'].split(':')[1].strip(), results['server_uptime'][0].strip()):
		score += 3

	print results['server_uptime'][0].strip()
	print results['server_OID'].split(':')[1].strip()
	print score
	return score

def prepare_disks(templates, params):
#    d = templates['simpleArbiterDhcp']
    pass