[120] | 1 | #Licensed to the Apache Software Foundation (ASF) under one |
---|
| 2 | #or more contributor license agreements. See the NOTICE file |
---|
| 3 | #distributed with this work for additional information |
---|
| 4 | #regarding copyright ownership. The ASF licenses this file |
---|
| 5 | #to you under the Apache License, Version 2.0 (the |
---|
| 6 | #"License"); you may not use this file except in compliance |
---|
| 7 | #with the License. You may obtain a copy of the License at |
---|
| 8 | |
---|
| 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
---|
| 10 | |
---|
| 11 | #Unless required by applicable law or agreed to in writing, software |
---|
| 12 | #distributed under the License is distributed on an "AS IS" BASIS, |
---|
| 13 | #WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
---|
| 14 | #See the License for the specific language governing permissions and |
---|
| 15 | #limitations under the License. |
---|
| 16 | """Gold Allocation Manager Implementation""" |
---|
| 17 | # -*- python -*- |
---|
| 18 | |
---|
| 19 | import sys, httplib |
---|
| 20 | import sha, base64, hmac |
---|
| 21 | import xml.dom.minidom |
---|
| 22 | |
---|
| 23 | from hodlib.Common.util import * |
---|
| 24 | |
---|
| 25 | class goldAllocationManager: |
---|
| 26 | def __init__(self, cfg, log): |
---|
| 27 | self.__GOLD_SECRET_KEY_FILE = cfg['auth-file'] |
---|
| 28 | (self.__goldHost, self.__goldPort) = (cfg['allocation-manager-address'][0], |
---|
| 29 | cfg['allocation-manager-address'][1]) |
---|
| 30 | self.cfg = cfg |
---|
| 31 | self.log = log |
---|
| 32 | |
---|
| 33 | def getQuote(self, user, project, ignoreErrors=True): |
---|
| 34 | # Get Secret Key from File |
---|
| 35 | secret = '' |
---|
| 36 | try: |
---|
| 37 | secretFile = open(self.__GOLD_SECRET_KEY_FILE) |
---|
| 38 | secret = secretFile.readline() |
---|
| 39 | except Exception, e: |
---|
| 40 | self.log.error("Unable to open file %s" % self.__GOLD_SECRET_KEY_FILE) |
---|
| 41 | self.log.debug(get_exception_string()) |
---|
| 42 | return (ignoreErrors or False) |
---|
| 43 | secretFile.close() |
---|
| 44 | secret = secret.rstrip() |
---|
| 45 | |
---|
| 46 | # construct the SSRMAP request body |
---|
| 47 | body = '<Body><Request action="Quote" actor="hod"><Object>Job</Object><Data><Job><ProjectId>%s</ProjectId><UserId>%s</UserId><WallDuration>10</WallDuration></Job></Data></Request></Body>' % (project, user) |
---|
| 48 | |
---|
| 49 | # compute digest |
---|
| 50 | message = sha.new() |
---|
| 51 | message.update(body) |
---|
| 52 | digest = message.digest() |
---|
| 53 | digestStr = base64.b64encode(digest) |
---|
| 54 | |
---|
| 55 | # compute signature |
---|
| 56 | message = hmac.new(secret, digest, sha) |
---|
| 57 | signatureStr = base64.b64encode(message.digest()) |
---|
| 58 | |
---|
| 59 | # construct the SSSRMAP Message |
---|
| 60 | sssrmapRequest = '<?xml version="1.0" encoding="UTF-8"?>\ |
---|
| 61 | <Envelope>%s<Signature><DigestValue>%s</DigestValue><SignatureValue>%s</SignatureValue><SecurityToken type="Symmetric"></SecurityToken></Signature></Envelope>' % (body, digestStr, signatureStr) |
---|
| 62 | self.log.info('sssrmapRequest: %s' % sssrmapRequest) |
---|
| 63 | |
---|
| 64 | try: |
---|
| 65 | # post message to GOLD server |
---|
| 66 | webservice = httplib.HTTP(self.__goldHost, self.__goldPort) |
---|
| 67 | webservice.putrequest("POST", "/SSSRMAP3 HTTP/1.1") |
---|
| 68 | webservice.putheader("Content-Type", "text/xml; charset=\"utf-8\"") |
---|
| 69 | webservice.putheader("Transfer-Encoding", "chunked") |
---|
| 70 | webservice.endheaders() |
---|
| 71 | webservice.send("%X" % len(sssrmapRequest) + "\r\n" + sssrmapRequest + '0\r\n') |
---|
| 72 | |
---|
| 73 | # handle the response |
---|
| 74 | statusCode, statusmessage, header = webservice.getreply() |
---|
| 75 | responseStr = webservice.getfile().read() |
---|
| 76 | self.log.debug("httpStatusCode: %d" % statusCode) |
---|
| 77 | self.log.info('responseStr: %s' % responseStr) |
---|
| 78 | |
---|
| 79 | # parse XML response |
---|
| 80 | if (statusCode == 200): |
---|
| 81 | responseArr = responseStr.split("\n") |
---|
| 82 | responseBody = responseArr[2] |
---|
| 83 | try: |
---|
| 84 | doc = xml.dom.minidom.parseString(responseBody) |
---|
| 85 | responseVal = doc.getElementsByTagName("Value")[0].firstChild.nodeValue |
---|
| 86 | self.log.info("responseVal: %s" % responseVal) |
---|
| 87 | if (responseVal == 'Success'): |
---|
| 88 | return True |
---|
| 89 | else: |
---|
| 90 | return False |
---|
| 91 | except Exception, e: |
---|
| 92 | self.log.error("Unable to parse GOLD responseBody XML \"(%s)\" to get responseVal" % (responseBody)) |
---|
| 93 | self.log.debug(get_exception_string()) |
---|
| 94 | return (ignoreErrors or False) |
---|
| 95 | else: |
---|
| 96 | self.log.error("Invalid HTTP statusCode %d" % statusCode) |
---|
| 97 | except Exception, e: |
---|
| 98 | self.log.error("Unable to POST message to GOLD server (%s, %d)" % |
---|
| 99 | (self.__goldHost, self.__goldPort)) |
---|
| 100 | self.log.debug(get_exception_string()) |
---|
| 101 | return (ignoreErrors or False) |
---|
| 102 | |
---|
| 103 | return True |
---|
| 104 | |
---|