def __init__(
self,
debug=False,
proxy='',
url_base='',
info_keys=None,
cert=None
):
if info_keys is None:
info_keys = {}
self._debug = debug
self._proxy = proxy
self._url_base = url_base
self._cert = cert
logging.info('SSL certificate: %s', self._cert)
if isinstance(info_keys, dict):
self._path_keys = info_keys.get('path')
self._private_key = info_keys.get('private')
self._public_key = info_keys.get('public')
if self._proxy:
logging.info('Proxy selected: %s', self._proxy)
if self._url_base:
logging.info('URL base: %s', self._url_base)
else:
logging.critical('URL base not assigned!!!')
if self._path_keys:
logging.info('Path keys: %s', self._path_keys)
if self._private_key:
logging.info('Private key: %s', self._private_key)
if self._public_key:
logging.info('Public key: %s', self._public_key)
self._filename_pattern = '%s.%s' % (
utils.get_mfc_computer_name(),
utils.get_hardware_uuid()
)
def __init__(
self,
debug=False,
proxy='',
url_base='',
info_keys={},
cert=None
):
self._debug = debug
self._proxy = proxy
self._url_base = url_base
self._cert = cert
logging.info('SSL certificate: %s', self._cert)
if type(info_keys) is dict:
self._path_keys = info_keys.get('path')
self._private_key = info_keys.get('private')
self._public_key = info_keys.get('public')
if self._proxy:
logging.info('Proxy selected: %s', self._proxy)
if self._url_base:
logging.info('URL base: %s', self._url_base)
else:
logging.critical('URL base not assigned!!!')
if self._path_keys:
logging.info('Path keys: %s', self._path_keys)
if self._private_key:
logging.info('Private key: %s', self._private_key)
if self._public_key:
logging.info('Public key: %s', self._public_key)
self._filename_pattern = '%s.%s' % (
utils.get_mfc_computer_name(),
utils.get_hardware_uuid()
)
def run(
self,
cmd,
data='',
upload_file=None,
sign=True,
exit_on_error=True
):
logging.debug('URL base: %s', self._url_base)
logging.debug('URL command: %s', cmd)
logging.debug('URL data: %s', data)
logging.debug('URL upload file: %s', upload_file)
logging.debug('Sign request: %s', sign)
logging.debug('Exit on error: %s', exit_on_error)
if not os.path.exists(TMP_PATH):
try:
os.makedirs(TMP_PATH, 0777)
except:
_msg = 'Error creating %s directory' % TMP_PATH
logging.exception(_msg)
return {
'errmfs': {
'info': _msg,
'code': server_errors.GENERIC
}
}
# API changed in server 3.0
_filename = os.path.join(
TMP_PATH,
'%s.%s.%s' % (
utils.get_mfc_computer_name(),
utils.get_hardware_uuid(),
cmd
)
)
if self._debug:
print _filename
if sign:
secure.wrap(
_filename,
{cmd: data},
key=os.path.join(self._path_keys, self._private_key)
)
else:
secure.wrap(_filename, {cmd: data})
_post = [
('message', (pycurl.FORM_FILE, _filename))
]
if upload_file:
_post.append(('package', (pycurl.FORM_FILE, upload_file)))
logging.debug('Post data: %s', _post)
_curl = curl.Curl(
self._url_base,
_post,
proxy=self._proxy,
cert=self._cert,
)
_curl.run()
if not self._debug:
os.remove(_filename)
if _curl.error:
_msg = _('Curl error: %s') % _curl.error
logging.error(_msg)
print _msg
return {'errmfs': {'info': _msg, 'code': server_errors.GENERIC}}
#sys.exit(errno.EBADRQC)
if _curl.http_code >= 400:
print _('HTTP error code: %s') % _curl.http_code
if self._debug:
_file = '/tmp/response.%s.%s.html' % (_curl.http_code, cmd)
utils.write_file(_file, str(_curl.body))
print _file
return {
'errmfs': {
'info': str(_curl.body),
'code': server_errors.GENERIC
}
}
#sys.exit(errno.EBADRQC)
# evaluate response
_response = '%s.return' % _filename
utils.write_file(_response, str(_curl.body))
if sign:
_ret = secure.unwrap(
_response,
key=os.path.join(self._path_keys, self._public_key)
)
else:
_ret = secure.unwrap(_response)
if not self._debug:
os.remove(_response)
else:
print _response
if not type(_ret) is dict or not ('%s.return' % cmd) in _ret:
_msg = 'url_request unexpected response: %s. Expected: %s'
if self._debug:
print _msg % (_ret, '%s.return' % cmd)
logging.critical(_msg, _ret, '%s.return' % cmd)
sys.exit(errno.EACCES)
_ret = _ret['%s.return' % cmd] # unwrapping cmd response
if type(_ret) is dict and 'errmfs' in _ret:
if _ret['errmfs']['code'] != server_errors.ALL_OK:
_error = server_errors.error_info(_ret['errmfs']['code'])
if self._debug:
print _('Error: %s') % _error
if _ret['errmfs']['info']:
print _('Information: %s') % _ret['errmfs']['info']
logging.error(
'url_request server error response code: %s',
_error
)
logging.error(
'url_request server error response info: %s',
_ret['errmfs']['info']
)
if exit_on_error:
sys.exit(errno.EACCES)
return _ret
