def log_plot(self, plot, name=None, footer=None):
if name == None:
name = "Scripting plot"
if not plot is None and hasattr(plot, 'pv'):
if footer is None:
if hasattr(plot, 'title'):
footer = plot.title
try:
self.logger.appendImageEntry(name,
plot.pv.getPlot().getImage(),
footer)
except:
print 'failed to send plot to notebook db'
elif not plot is None and hasattr(plot, 'cache'):
if footer is None:
if hasattr(plot, 'title'):
footer = plot.title
try:
from org.apache.commons.codec.binary import Base64
from java.lang import String
self.logger.appendImageEntry(
name,
String(Base64.encodeBase64(plot.cache.getImageCache())),
footer)
except:
print 'failed to send plot to notebook db'
def encryptPassword(password):
if password is not None and password is not ' ':
cipher.init(Cipher.ENCRYPT_MODE, skeySpec)
encrypted = cipher.doFinal(password)
encodeTxt = Base64.encodeBase64(encrypted)
return encodeTxt.tostring()
from java.lang import System, String
from java.util import Properties
from java.math import BigInteger
from java.security import *
from javax.crypto import KeyGenerator, SecretKey, Cipher
from javax.crypto.spec import SecretKeySpec
from org.apache.log4j import Logger
from org.apache.commons.codec.binary import Base64
log = Logger.getLogger('password_encrypter')
kgen = KeyGenerator.getInstance("AES")
kgen.init(128)
skey = kgen.generateKey()
raw = skey.getEncoded()
rawkey = Base64.encodeBase64(raw).tostring()
skeySpec = SecretKeySpec(raw, "AES")
cipher = Cipher.getInstance("AES")
def encryptPassword(password):
if password is not None and password is not ' ':
cipher.init(Cipher.ENCRYPT_MODE, skeySpec)
encrypted = cipher.doFinal(password)
encodeTxt = Base64.encodeBase64(encrypted)
return encodeTxt.tostring()
def encryptAllPasswords(configfile):
log.info("Encrypting Passwords in config file")
if (configfile is not None and not ''):
ntpipelineapiSet = mbo.getMboSet("NTPIPELINEAPI")
if ntpipelineapiSet.isEmpty():
service.error('ntpipeline', 'pipelineMap', None) #Break
pipelineapi = ntpipelineapiSet.moveFirst()
uri = host + '/os/'
uri = uri + pipelineapi.getString("INTOBJECTNAME")
uri = uri + '?lean=1&oslc.where=' #MAXDOMAINID='
uri = uri + pipelineapi.getString("UNIQUECOLUMN") + '='
uri = uri + str(mbo.getInt("FOREIGNKEY"))
uri = uri + '&oslc.select=*'
clientid = "maxadmin"
clientsecurity = "maxadmin"
# get authentication header
auth = clientid + ":" + clientsecurity
encodedAuth = String(Base64.encodeBase64(String.getBytes(auth, 'ISO-8859-1')),
"UTF-8")
authHeader = str(encodedAuth)
# get http parameters
params = BasicHttpParams()
paramsBean = HttpProtocolParamBean(params)
paramsBean.setVersion(HttpVersion.HTTP_1_1)
paramsBean.setContentCharset("UTF-8")
paramsBean.setUseExpectContinue(True)
# get http body entities
formparams = ArrayList()
#formparams.add(BasicNameValuePair("username", username))
entity = UrlEncodedFormEntity(formparams, "UTF-8")
def executeSSOTest():
reqInstant = dateFormat.format(GregorianCalendar(utctz).getTime())
reqId = UUID.randomUUID().toString()
authnRequest = props.get('shib.sp.authnreq') % (idpSSOEndpoint, reqId,
reqInstant, spId)
webClient = WebClient()
webClient.setUseInsecureSSL(True)
listreq = ArrayList()
listreq.add(
NameValuePair(
"SAMLRequest",
String(Base64.encodeBase64(String(authnRequest).getBytes()),
'US-ASCII')))
wrs = WebRequestSettings(URL(idpSSOEndpoint), HttpMethod.POST)
wrs.setRequestParameters(listreq)
if props.get('shib.idp.auth') == 'FORM':
webClient.setRedirectEnabled(True)
if debug: log("sending SSO request to: " + idpSSOEndpoint, stdout)
wr = webClient.loadWebResponse(wrs)
if debug: log("Response content: \n" + wr.getContentAsString(), stdout)
if wr.getStatusCode() != 200:
raise Exception('Invalid HTTP response code: ' +
str(wr.getStatusCode()))
wrs.setHttpMethod(HttpMethod.POST)
listauth = ArrayList()
listauth.add(NameValuePair("j_username", props.get('shib.user.name')))
listauth.add(NameValuePair("j_password", props.get('shib.user.pass')))
wrs.setRequestParameters(listauth)
wrs.setUrl(wr.getRequestUrl())
else:
# BASIC auth
webClient.setRedirectEnabled(False)
if debug: log("sending SSO request to: " + idpSSOEndpoint, stdout)
wr = webClient.loadWebResponse(wrs)
if debug: log("Response content: \n" + wr.getContentAsString(), stdout)
if wr.getStatusCode() != 302:
raise Exception('Invalid HTTP response code: ' +
str(wr.getStatusCode()))
wrs.setHttpMethod(HttpMethod.GET)
creds = DefaultCredentialsProvider()
creds.addCredentials(props.get('shib.user.name'),
props.get('shib.user.pass'), idpHost, idpPort,
props.get('shib.user.realm'))
wrs.setCredentialsProvider(creds)
if debug:
log(
"sending BASIC auth to: " +
wr.getResponseHeaderValue('Location'), stdout)
wrs.setUrl(URL(wr.getResponseHeaderValue('Location')))
if debug:
log("sending login request to: " + str(wr.getRequestUrl()), stdout)
wr = webClient.loadWebResponse(wrs)
if debug:
log("Request URL = " + wr.getRequestUrl().toString(), stdout)
log(wr.getContentAsString(), stdout)
if wr.getStatusCode() not in [200, 301, 302]:
if debug: log("error response: " + response.getText(), stdout)
raise Exception('Invalid HTTP response code: ' +
str(wr.getStatusCode()))
return wr.getContentAsString()
