You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
1.9 KiB
79 lines
1.9 KiB
#!/usr/bin/env python3
|
|
import sys
|
|
import re
|
|
import os
|
|
from hashlib import sha256
|
|
from typing import Iterable, Union, AnyStr
|
|
|
|
import requests
|
|
|
|
API_AUTH_SECRET = 'your api key'
|
|
|
|
SERVER_DOMAIN = 'http://127.0.0.1:8000'
|
|
|
|
IP_ADDR_REGEX = (
|
|
'^(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.'
|
|
'(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.'
|
|
'(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.'
|
|
'(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$'
|
|
)
|
|
|
|
|
|
def calc_hash(data):
|
|
if type(data) is str:
|
|
result_data = data.encode('utf-8')
|
|
else:
|
|
result_data = bytes(data)
|
|
return sha256(result_data).hexdigest()
|
|
|
|
|
|
def check_sign(get_list: Iterable, sign_str: str):
|
|
hashed = '_'.join(get_list)
|
|
my_sign = calc_hash(hashed)
|
|
return sign_str == my_sign
|
|
|
|
|
|
def validate(regexp: Union[bytes, str], string: AnyStr):
|
|
if not re.match(regexp, string):
|
|
raise ValueError
|
|
return string
|
|
|
|
|
|
def validate_status(text: str):
|
|
if text not in ('UP', 'DOWN', 'UNREACHABLE'):
|
|
raise ValueError
|
|
return text
|
|
|
|
|
|
def send_request(ip_addr, stat, sign_hash):
|
|
r = requests.get(
|
|
os.path.join(SERVER_DOMAIN, 'dev', 'on_device_event'),
|
|
params={
|
|
'ip_addr': ip_addr,
|
|
'status': stat,
|
|
'sign': sign_hash
|
|
})
|
|
if r.status_code == 200:
|
|
print(r.json())
|
|
else:
|
|
print('Status:', r.status_code, r.text)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) < 3:
|
|
print('You forget parameters, example of usage:\n'
|
|
'$ python3 ./monitoring_agent.py a2:c3:12:46:1f:92 DOWN|UP|UNREACHABLE')
|
|
exit(0)
|
|
|
|
if API_AUTH_SECRET == 'your api key':
|
|
raise NotImplementedError('You must specified secret api key')
|
|
|
|
dev_ip = validate(IP_ADDR_REGEX, sys.argv[1])
|
|
status = validate_status(sys.argv[2])
|
|
|
|
vars_to_hash = [dev_ip, status]
|
|
vars_to_hash.sort()
|
|
vars_to_hash.append(API_AUTH_SECRET)
|
|
sign = calc_hash('_'.join(vars_to_hash))
|
|
|
|
send_request(dev_ip, status, sign)
|