apache/cloudstack
1
0
Fork 0
mirror of https://github.com/apache/cloudstack.git synced 2025-10-26 08:42:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Removed Base64 file and use the apache version

Browse Source
This commit is contained in:
Alex Huang 2012-08-07 15:38:59 -07:00
parent 1d3a443237
commit a6a864a1a8
9 changed files with 1998 additions and 3756 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
awsapi/.classpath
View File

@ -1,22 +1,4 @@
<?xml version="1.0" encoding="UTF-8"?> <?xml version="1.0" encoding="UTF-8"?>
<!--
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing,
software distributed under the License is distributed on an
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
-->
<classpath> <classpath>
<classpathentry kind="src" path="src"/> <classpathentry kind="src" path="src"/>
<classpathentry kind="src" path="test"/> <classpathentry kind="src" path="test"/>

2
build/developer.xml
View File

@ -241,7 +241,7 @@
<available file="${setup.db.dir}/override/templates.sql" /> <available file="${setup.db.dir}/override/templates.sql" />
</condition> </condition>
<echo message="deploydb ${server-setup.file} ${templates.file} ${DBROOTPW}" /> <echo message="${db.scripts.dir}\\deploy-db-dev.sh ${server-setup.file} ${templates.file} ${DBROOTPW}" />
<exec dir="${db.scripts.dir}" executable="bash" failonerror="true"> <exec dir="${db.scripts.dir}" executable="bash" failonerror="true">
<arg value="deploy-db-dev.sh" /> <arg value="deploy-db-dev.sh" />
<arg value="${server-setup.file}" /> <arg value="${server-setup.file}" />

126
server/src/com/cloud/api/ApiServer.java
View File

@ -53,6 +53,7 @@ import javax.crypto.spec.SecretKeySpec;
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession; import javax.servlet.http.HttpSession;
import org.apache.commons.codec.binary.Base64;
import org.apache.http.ConnectionClosedException; import org.apache.http.ConnectionClosedException;
import org.apache.http.HttpException; import org.apache.http.HttpException;
import org.apache.http.HttpRequest; import org.apache.http.HttpRequest;
@ -112,7 +113,6 @@ import com.cloud.utils.component.PluggableService;
import com.cloud.utils.concurrency.NamedThreadFactory; import com.cloud.utils.concurrency.NamedThreadFactory;
import com.cloud.utils.db.SearchCriteria; import com.cloud.utils.db.SearchCriteria;
import com.cloud.utils.db.Transaction; import com.cloud.utils.db.Transaction;
import com.cloud.utils.encoding.Base64;
import com.cloud.utils.exception.CSExceptionErrorCode; import com.cloud.utils.exception.CSExceptionErrorCode;
import com.cloud.uuididentity.dao.IdentityDao; import com.cloud.uuididentity.dao.IdentityDao;
@ -432,31 +432,31 @@ public class ApiServer implements HttpRequestHandler {
} }
} catch (Exception ex) { } catch (Exception ex) {
if (ex instanceof InvalidParameterValueException) { if (ex instanceof InvalidParameterValueException) {
InvalidParameterValueException ref = (InvalidParameterValueException)ex; InvalidParameterValueException ref = (InvalidParameterValueException)ex;
ServerApiException e = new ServerApiException(BaseCmd.PARAM_ERROR, ex.getMessage()); ServerApiException e = new ServerApiException(BaseCmd.PARAM_ERROR, ex.getMessage());
// copy over the IdentityProxy information as well and throw the serverapiexception. // copy over the IdentityProxy information as well and throw the serverapiexception.
ArrayList<IdentityProxy> idList = ref.getIdProxyList(); ArrayList<IdentityProxy> idList = ref.getIdProxyList();
if (idList != null) { if (idList != null) {
// Iterate through entire arraylist and copy over each proxy id. // Iterate through entire arraylist and copy over each proxy id.
for (int i = 0 ; i < idList.size(); i++) { for (int i = 0 ; i < idList.size(); i++) {
IdentityProxy obj = idList.get(i); IdentityProxy obj = idList.get(i);
e.addProxyObject(obj.getTableName(), obj.getValue(), obj.getidFieldName()); e.addProxyObject(obj.getTableName(), obj.getValue(), obj.getidFieldName());
} }
} }
// Also copy over the cserror code and the function/layer in which it was thrown. // Also copy over the cserror code and the function/layer in which it was thrown.
e.setCSErrorCode(ref.getCSErrorCode()); e.setCSErrorCode(ref.getCSErrorCode());
throw e; throw e;
} else if (ex instanceof PermissionDeniedException) { } else if (ex instanceof PermissionDeniedException) {
PermissionDeniedException ref = (PermissionDeniedException)ex; PermissionDeniedException ref = (PermissionDeniedException)ex;
ServerApiException e = new ServerApiException(BaseCmd.ACCOUNT_ERROR, ex.getMessage()); ServerApiException e = new ServerApiException(BaseCmd.ACCOUNT_ERROR, ex.getMessage());
// copy over the IdentityProxy information as well and throw the serverapiexception. // copy over the IdentityProxy information as well and throw the serverapiexception.
ArrayList<IdentityProxy> idList = ref.getIdProxyList(); ArrayList<IdentityProxy> idList = ref.getIdProxyList();
if (idList != null) { if (idList != null) {
// Iterate through entire arraylist and copy over each proxy id. // Iterate through entire arraylist and copy over each proxy id.
for (int i = 0 ; i < idList.size(); i++) { for (int i = 0 ; i < idList.size(); i++) {
IdentityProxy obj = idList.get(i); IdentityProxy obj = idList.get(i);
e.addProxyObject(obj.getTableName(), obj.getValue(), obj.getidFieldName()); e.addProxyObject(obj.getTableName(), obj.getValue(), obj.getidFieldName());
} }
} }
e.setCSErrorCode(ref.getCSErrorCode()); e.setCSErrorCode(ref.getCSErrorCode());
throw e; throw e;
@ -751,7 +751,7 @@ public class ApiServer implements HttpRequestHandler {
mac.init(keySpec); mac.init(keySpec);
mac.update(unsignedRequest.getBytes()); mac.update(unsignedRequest.getBytes());
byte[] encryptedBytes = mac.doFinal(); byte[] encryptedBytes = mac.doFinal();
String computedSignature = Base64.encodeBytes(encryptedBytes); String computedSignature = Base64.encodeBase64URLSafeString(encryptedBytes);
boolean equalSig = signature.equals(computedSignature); boolean equalSig = signature.equals(computedSignature);
if (!equalSig) { if (!equalSig) {
s_logger.info("User signature: " + signature + " is not equaled to computed signature: " + computedSignature); s_logger.info("User signature: " + signature + " is not equaled to computed signature: " + computedSignature);
@ -765,7 +765,7 @@ public class ApiServer implements HttpRequestHandler {
} }
return false; return false;
} }
public Long fetchDomainId(String domainUUID){ public Long fetchDomainId(String domainUUID){
ComponentLocator locator = ComponentLocator.getLocator(ManagementServer.Name); ComponentLocator locator = ComponentLocator.getLocator(ManagementServer.Name);
IdentityDao identityDao = locator.getDao(IdentityDao.class); IdentityDao identityDao = locator.getDao(IdentityDao.class);
@ -816,19 +816,19 @@ public class ApiServer implements HttpRequestHandler {
if(user.getUuid() != null){ if(user.getUuid() != null){
session.setAttribute("user_UUID", user.getUuid()); session.setAttribute("user_UUID", user.getUuid());
} }
session.setAttribute("username", userAcct.getUsername()); session.setAttribute("username", userAcct.getUsername());
session.setAttribute("firstname", userAcct.getFirstname()); session.setAttribute("firstname", userAcct.getFirstname());
session.setAttribute("lastname", userAcct.getLastname()); session.setAttribute("lastname", userAcct.getLastname());
session.setAttribute("accountobj", account); session.setAttribute("accountobj", account);
session.setAttribute("account", account.getAccountName()); session.setAttribute("account", account.getAccountName());
session.setAttribute("domainid", account.getDomainId()); session.setAttribute("domainid", account.getDomainId());
DomainVO domain = (DomainVO) _domainMgr.getDomain(account.getDomainId()); DomainVO domain = (DomainVO) _domainMgr.getDomain(account.getDomainId());
if(domain.getUuid() != null){ if(domain.getUuid() != null){
session.setAttribute("domain_UUID", domain.getUuid()); session.setAttribute("domain_UUID", domain.getUuid());
} }
session.setAttribute("type", Short.valueOf(account.getType()).toString()); session.setAttribute("type", Short.valueOf(account.getType()).toString());
session.setAttribute("registrationtoken", userAcct.getRegistrationToken()); session.setAttribute("registrationtoken", userAcct.getRegistrationToken());
session.setAttribute("registered", new Boolean(userAcct.isRegistered()).toString()); session.setAttribute("registered", new Boolean(userAcct.isRegistered()).toString());
@ -843,7 +843,7 @@ public class ApiServer implements HttpRequestHandler {
SecureRandom sesssionKeyRandom = new SecureRandom(); SecureRandom sesssionKeyRandom = new SecureRandom();
byte sessionKeyBytes[] = new byte[20]; byte sessionKeyBytes[] = new byte[20];
sesssionKeyRandom.nextBytes(sessionKeyBytes); sesssionKeyRandom.nextBytes(sessionKeyBytes);
String sessionKey = Base64.encodeBytes(sessionKeyBytes); String sessionKey = Base64.encodeBase64URLSafeString(sessionKeyBytes);
session.setAttribute("sessionkey", sessionKey); session.setAttribute("sessionkey", sessionKey);
return; return;
@ -938,8 +938,8 @@ public class ApiServer implements HttpRequestHandler {
_params = new BasicHttpParams(); _params = new BasicHttpParams();
_params.setIntParameter(CoreConnectionPNames.SO_TIMEOUT, 30000).setIntParameter(CoreConnectionPNames.SOCKET_BUFFER_SIZE, 8 * 1024) _params.setIntParameter(CoreConnectionPNames.SO_TIMEOUT, 30000).setIntParameter(CoreConnectionPNames.SOCKET_BUFFER_SIZE, 8 * 1024)
.setBooleanParameter(CoreConnectionPNames.STALE_CONNECTION_CHECK, false).setBooleanParameter(CoreConnectionPNames.TCP_NODELAY, true) .setBooleanParameter(CoreConnectionPNames.STALE_CONNECTION_CHECK, false).setBooleanParameter(CoreConnectionPNames.TCP_NODELAY, true)
.setParameter(CoreProtocolPNames.ORIGIN_SERVER, "HttpComponents/1.1"); .setParameter(CoreProtocolPNames.ORIGIN_SERVER, "HttpComponents/1.1");
// Set up the HTTP protocol processor // Set up the HTTP protocol processor
BasicHttpProcessor httpproc = new BasicHttpProcessor(); BasicHttpProcessor httpproc = new BasicHttpProcessor();
@ -1053,44 +1053,44 @@ public class ApiServer implements HttpRequestHandler {
// Exception. When invoked from ApiServlet's processRequest(), this can be // Exception. When invoked from ApiServlet's processRequest(), this can be
// a standard exception like NumberFormatException. We'll leave the standard ones alone. // a standard exception like NumberFormatException. We'll leave the standard ones alone.
if (ex != null) { if (ex != null) {
if (ex instanceof ServerApiException || ex instanceof PermissionDeniedException if (ex instanceof ServerApiException || ex instanceof PermissionDeniedException
|| ex instanceof InvalidParameterValueException) { || ex instanceof InvalidParameterValueException) {
// Cast the exception appropriately and retrieve the IdentityProxy // Cast the exception appropriately and retrieve the IdentityProxy
if (ex instanceof ServerApiException) { if (ex instanceof ServerApiException) {
ServerApiException ref = (ServerApiException) ex; ServerApiException ref = (ServerApiException) ex;
ArrayList<IdentityProxy> idList = ref.getIdProxyList(); ArrayList<IdentityProxy> idList = ref.getIdProxyList();
if (idList != null) { if (idList != null) {
for (int i=0; i < idList.size(); i++) { for (int i=0; i < idList.size(); i++) {
IdentityProxy id = idList.get(i); IdentityProxy id = idList.get(i);
apiResponse.addProxyObject(id.getTableName(), id.getValue(), id.getidFieldName()); apiResponse.addProxyObject(id.getTableName(), id.getValue(), id.getidFieldName());
} }
} }
// Also copy over the cserror code and the function/layer in which it was thrown. // Also copy over the cserror code and the function/layer in which it was thrown.
apiResponse.setCSErrorCode(ref.getCSErrorCode()); apiResponse.setCSErrorCode(ref.getCSErrorCode());
} else if (ex instanceof PermissionDeniedException) { } else if (ex instanceof PermissionDeniedException) {
PermissionDeniedException ref = (PermissionDeniedException) ex; PermissionDeniedException ref = (PermissionDeniedException) ex;
ArrayList<IdentityProxy> idList = ref.getIdProxyList(); ArrayList<IdentityProxy> idList = ref.getIdProxyList();
if (idList != null) { if (idList != null) {
for (int i=0; i < idList.size(); i++) { for (int i=0; i < idList.size(); i++) {
IdentityProxy id = idList.get(i); IdentityProxy id = idList.get(i);
apiResponse.addProxyObject(id.getTableName(), id.getValue(), id.getidFieldName()); apiResponse.addProxyObject(id.getTableName(), id.getValue(), id.getidFieldName());
} }
} }
// Also copy over the cserror code and the function/layer in which it was thrown. // Also copy over the cserror code and the function/layer in which it was thrown.
apiResponse.setCSErrorCode(ref.getCSErrorCode()); apiResponse.setCSErrorCode(ref.getCSErrorCode());
} else if (ex instanceof InvalidParameterValueException) { } else if (ex instanceof InvalidParameterValueException) {
InvalidParameterValueException ref = (InvalidParameterValueException) ex; InvalidParameterValueException ref = (InvalidParameterValueException) ex;
ArrayList<IdentityProxy> idList = ref.getIdProxyList(); ArrayList<IdentityProxy> idList = ref.getIdProxyList();
if (idList != null) { if (idList != null) {
for (int i=0; i < idList.size(); i++) { for (int i=0; i < idList.size(); i++) {
IdentityProxy id = idList.get(i); IdentityProxy id = idList.get(i);
apiResponse.addProxyObject(id.getTableName(), id.getValue(), id.getidFieldName()); apiResponse.addProxyObject(id.getTableName(), id.getValue(), id.getidFieldName());
} }
} }
// Also copy over the cserror code and the function/layer in which it was thrown. // Also copy over the cserror code and the function/layer in which it was thrown.
apiResponse.setCSErrorCode(ref.getCSErrorCode()); apiResponse.setCSErrorCode(ref.getCSErrorCode());
} }
} }
} }
SerializationContext.current().setUuidTranslation(true); SerializationContext.current().setUuidTranslation(true);
responseText = ApiResponseSerializer.toSerializedString(apiResponse, responseType); responseText = ApiResponseSerializer.toSerializedString(apiResponse, responseType);

788
server/src/com/cloud/servlet/ConsoleProxyServlet.java
View File

@ -58,61 +58,61 @@ import com.cloud.vm.VirtualMachineManager;
* Authentication : /console?cmd=auth&vm=xxx&sid=xxx * Authentication : /console?cmd=auth&vm=xxx&sid=xxx
*/ */
public class ConsoleProxyServlet extends HttpServlet { public class ConsoleProxyServlet extends HttpServlet {
private static final long serialVersionUID = -5515382620323808168L; private static final long serialVersionUID = -5515382620323808168L;
public static final Logger s_logger = Logger.getLogger(ConsoleProxyServlet.class.getName()); public static final Logger s_logger = Logger.getLogger(ConsoleProxyServlet.class.getName());
private static final int DEFAULT_THUMBNAIL_WIDTH = 144; private static final int DEFAULT_THUMBNAIL_WIDTH = 144;
private static final int DEFAULT_THUMBNAIL_HEIGHT = 110; private static final int DEFAULT_THUMBNAIL_HEIGHT = 110;
private final static AccountManager _accountMgr = ComponentLocator.getLocator(ManagementServer.Name).getManager(AccountManager.class); private final static AccountManager _accountMgr = ComponentLocator.getLocator(ManagementServer.Name).getManager(AccountManager.class);
private final static VirtualMachineManager _vmMgr = ComponentLocator.getLocator(ManagementServer.Name).getManager(VirtualMachineManager.class); private final static VirtualMachineManager _vmMgr = ComponentLocator.getLocator(ManagementServer.Name).getManager(VirtualMachineManager.class);
private final static ManagementServer _ms = (ManagementServer)ComponentLocator.getComponent(ManagementServer.Name); private final static ManagementServer _ms = (ManagementServer)ComponentLocator.getComponent(ManagementServer.Name);
private final static IdentityService _identityService = (IdentityService)ComponentLocator.getLocator(ManagementServer.Name).getManager(IdentityService.class); private final static IdentityService _identityService = ComponentLocator.getLocator(ManagementServer.Name).getManager(IdentityService.class);
@Override @Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) { protected void doPost(HttpServletRequest req, HttpServletResponse resp) {
doGet(req, resp); doGet(req, resp);
} }
@Override @Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) { protected void doGet(HttpServletRequest req, HttpServletResponse resp) {
try { try {
if(_accountMgr == null || _vmMgr == null || _ms == null) { if(_accountMgr == null || _vmMgr == null || _ms == null) {
sendResponse(resp, "Service is not ready"); sendResponse(resp, "Service is not ready");
return; return;
} }
if(_ms.getHashKey() == null) { if(_ms.getHashKey() == null) {
s_logger.debug("Console/thumbnail access denied. Ticket service is not ready yet"); s_logger.debug("Console/thumbnail access denied. Ticket service is not ready yet");
sendResponse(resp, "Service is not ready"); sendResponse(resp, "Service is not ready");
return; return;
} }
String userId = null; String userId = null;
String account = null; String account = null;
Account accountObj = null; Account accountObj = null;
Map<String, Object[]> params = new HashMap<String, Object[]>(); Map<String, Object[]> params = new HashMap<String, Object[]>();
params.putAll(req.getParameterMap()); params.putAll(req.getParameterMap());
HttpSession session = req.getSession(false); HttpSession session = req.getSession(false);
if(session == null) { if(session == null) {
if(verifyRequest(params)) { if(verifyRequest(params)) {
userId = (String)params.get("userid")[0]; userId = (String)params.get("userid")[0];
account = (String)params.get("account")[0]; account = (String)params.get("account")[0];
accountObj = (Account)params.get("accountobj")[0]; accountObj = (Account)params.get("accountobj")[0];
} else { } else {
s_logger.debug("Invalid web session or API key in request, reject console/thumbnail access"); s_logger.debug("Invalid web session or API key in request, reject console/thumbnail access");
sendResponse(resp, "Access denied. Invalid web session or API key in request"); sendResponse(resp, "Access denied. Invalid web session or API key in request");
return; return;
} }
} else { } else {
// adjust to latest API refactoring changes // adjust to latest API refactoring changes
if(session.getAttribute("userid") != null) { if(session.getAttribute("userid") != null) {
userId = ((Long)session.getAttribute("userid")).toString(); userId = ((Long)session.getAttribute("userid")).toString();
} }
accountObj = (Account)session.getAttribute("accountobj"); accountObj = (Account)session.getAttribute("accountobj");
if(accountObj != null) { if(accountObj != null) {
account = "" + accountObj.getId(); account = "" + accountObj.getId();
} }
@ -120,381 +120,381 @@ public class ConsoleProxyServlet extends HttpServlet {
// Do a sanity check here to make sure the user hasn't already been deleted // Do a sanity check here to make sure the user hasn't already been deleted
if ((userId == null) || (account == null) || (accountObj == null) || !verifyUser(Long.valueOf(userId))) { if ((userId == null) || (account == null) || (accountObj == null) || !verifyUser(Long.valueOf(userId))) {
s_logger.debug("Invalid user/account, reject console/thumbnail access"); s_logger.debug("Invalid user/account, reject console/thumbnail access");
sendResponse(resp, "Access denied. Invalid or inconsistent account is found"); sendResponse(resp, "Access denied. Invalid or inconsistent account is found");
return; return;
} }
String cmd = req.getParameter("cmd"); String cmd = req.getParameter("cmd");
if(cmd == null || !isValidCmd(cmd)) { if(cmd == null || !isValidCmd(cmd)) {
s_logger.debug("invalid console servlet command: " + cmd); s_logger.debug("invalid console servlet command: " + cmd);
sendResponse(resp, ""); sendResponse(resp, "");
return; return;
} }
String vmIdString = req.getParameter("vm"); String vmIdString = req.getParameter("vm");
Long vmId = _identityService.getIdentityId("vm_instance", vmIdString); Long vmId = _identityService.getIdentityId("vm_instance", vmIdString);
if(vmId == null) { if(vmId == null) {
s_logger.info("invalid console servlet command parameter: " + vmIdString); s_logger.info("invalid console servlet command parameter: " + vmIdString);
sendResponse(resp, ""); sendResponse(resp, "");
return; return;
} }
if(!checkSessionPermision(req, vmId, accountObj)) { if(!checkSessionPermision(req, vmId, accountObj)) {
sendResponse(resp, "Permission denied"); sendResponse(resp, "Permission denied");
return; return;
} }
if(cmd.equalsIgnoreCase("thumbnail")) { if(cmd.equalsIgnoreCase("thumbnail")) {
handleThumbnailRequest(req, resp, vmId); handleThumbnailRequest(req, resp, vmId);
} else if(cmd.equalsIgnoreCase("access")) { } else if(cmd.equalsIgnoreCase("access")) {
handleAccessRequest(req, resp, vmId); handleAccessRequest(req, resp, vmId);
} else { } else {
handleAuthRequest(req, resp, vmId); handleAuthRequest(req, resp, vmId);
} }
} catch (Throwable e) { } catch (Throwable e) {
s_logger.error("Unexepected exception in ConsoleProxyServlet", e); s_logger.error("Unexepected exception in ConsoleProxyServlet", e);
sendResponse(resp, "Server Internal Error"); sendResponse(resp, "Server Internal Error");
} }
} }
private void handleThumbnailRequest(HttpServletRequest req, HttpServletResponse resp, long vmId) { private void handleThumbnailRequest(HttpServletRequest req, HttpServletResponse resp, long vmId) {
VMInstanceVO vm = _vmMgr.findById(vmId); VMInstanceVO vm = _vmMgr.findById(vmId);
if(vm == null) { if(vm == null) {
s_logger.warn("VM " + vmId + " does not exist, sending blank response for thumbnail request"); s_logger.warn("VM " + vmId + " does not exist, sending blank response for thumbnail request");
sendResponse(resp, ""); sendResponse(resp, "");
return; return;
} }
if(vm.getHostId() == null) { if(vm.getHostId() == null) {
s_logger.warn("VM " + vmId + " lost host info, sending blank response for thumbnail request"); s_logger.warn("VM " + vmId + " lost host info, sending blank response for thumbnail request");
sendResponse(resp, ""); sendResponse(resp, "");
return; return;
} }
HostVO host = _ms.getHostBy(vm.getHostId()); HostVO host = _ms.getHostBy(vm.getHostId());
if(host == null) { if(host == null) {
s_logger.warn("VM " + vmId + "'s host does not exist, sending blank response for thumbnail request"); s_logger.warn("VM " + vmId + "'s host does not exist, sending blank response for thumbnail request");
sendResponse(resp, ""); sendResponse(resp, "");
return; return;
} }
String rootUrl = _ms.getConsoleAccessUrlRoot(vmId); String rootUrl = _ms.getConsoleAccessUrlRoot(vmId);
if(rootUrl == null) { if(rootUrl == null) {
sendResponse(resp, ""); sendResponse(resp, "");
return; return;
} }
int w = DEFAULT_THUMBNAIL_WIDTH; int w = DEFAULT_THUMBNAIL_WIDTH;
int h = DEFAULT_THUMBNAIL_HEIGHT; int h = DEFAULT_THUMBNAIL_HEIGHT;
String value = req.getParameter("w"); String value = req.getParameter("w");
try { try {
w = Integer.parseInt(value); w = Integer.parseInt(value);
} catch(NumberFormatException e) { } catch(NumberFormatException e) {
} }
value = req.getParameter("h"); value = req.getParameter("h");
try { try {
h = Integer.parseInt(value); h = Integer.parseInt(value);
} catch(NumberFormatException e) { } catch(NumberFormatException e) {
} }
try { try {
resp.sendRedirect(composeThumbnailUrl(rootUrl, vm, host, w, h)); resp.sendRedirect(composeThumbnailUrl(rootUrl, vm, host, w, h));
} catch (IOException e) { } catch (IOException e) {
if(s_logger.isInfoEnabled()) { if(s_logger.isInfoEnabled()) {
s_logger.info("Client may already close the connection"); s_logger.info("Client may already close the connection");
} }
}
}
private void handleAccessRequest(HttpServletRequest req, HttpServletResponse resp, long vmId) {
VMInstanceVO vm = _vmMgr.findById(vmId);
if(vm == null) {
s_logger.warn("VM " + vmId + " does not exist, sending blank response for console access request");
sendResponse(resp, "");
return;
}
if(vm.getHostId() == null) {
s_logger.warn("VM " + vmId + " lost host info, sending blank response for console access request");
sendResponse(resp, "");
return;
}
HostVO host = _ms.getHostBy(vm.getHostId());
if(host == null) {
s_logger.warn("VM " + vmId + "'s host does not exist, sending blank response for console access request");
sendResponse(resp, "");
return;
}
String rootUrl = _ms.getConsoleAccessUrlRoot(vmId);
if(rootUrl == null) {
sendResponse(resp, "<html><body><p>Console access will be ready in a few minutes. Please try it again later.</p></body></html>");
return;
}
String vmName = vm.getHostName();
if(vm.getType() == VirtualMachine.Type.User) {
UserVm userVm = (UserVm)_vmMgr.findByIdAndType(VirtualMachine.Type.User, vmId);
String displayName = userVm.getDisplayName();
if(displayName != null && !displayName.isEmpty() && !displayName.equals(vmName)) {
vmName += "(" + displayName + ")";
}
}
StringBuffer sb = new StringBuffer();
sb.append("<html><title>").append(escapeHTML(vmName)).append("</title><frameset><frame src=\"").append(composeConsoleAccessUrl(rootUrl, vm, host));
sb.append("\"></frame></frameset></html>");
s_logger.debug("the console url is :: " + sb.toString());
sendResponse(resp, sb.toString());
}
private void handleAuthRequest(HttpServletRequest req, HttpServletResponse resp, long vmId) {
// TODO authentication channel between console proxy VM and management server needs to be secured,
// the data is now being sent through private network, but this is apparently not enough
VMInstanceVO vm = _vmMgr.findById(vmId);
if(vm == null) {
s_logger.warn("VM " + vmId + " does not exist, sending failed response for authentication request from console proxy");
sendResponse(resp, "failed");
return;
}
if(vm.getHostId() == null) {
s_logger.warn("VM " + vmId + " lost host info, failed response for authentication request from console proxy");
sendResponse(resp, "failed");
return;
}
HostVO host = _ms.getHostBy(vm.getHostId());
if(host == null) {
s_logger.warn("VM " + vmId + "'s host does not exist, sending failed response for authentication request from console proxy");
sendResponse(resp, "failed");
return;
}
String sid = req.getParameter("sid");
if(sid == null || !sid.equals(vm.getVncPassword())) {
s_logger.warn("sid " + sid + " in url does not match stored sid " + vm.getVncPassword());
sendResponse(resp, "failed");
return;
}
sendResponse(resp, "success");
}
// put the ugly stuff here
static public Ternary<String, String, String> parseHostInfo(String hostInfo) {
String host = null;
String tunnelUrl = null;
String tunnelSession = null;
s_logger.info("Parse host info returned from executing GetVNCPortCommand. host info: " + hostInfo);
if(hostInfo != null && hostInfo.startsWith("consoleurl")) {
String tokens[] = hostInfo.split("&");
if(hostInfo.length() > 19 && hostInfo.indexOf('/', 19) > 19) {
host = hostInfo.substring(19, hostInfo.indexOf('/', 19)).trim();
tunnelUrl = tokens[0].substring("consoleurl=".length());
tunnelSession = tokens[1].split("=")[1];
} else {
host = "";
}
} else {
host = hostInfo;
} }
}
private void handleAccessRequest(HttpServletRequest req, HttpServletResponse resp, long vmId) {
VMInstanceVO vm = _vmMgr.findById(vmId);
if(vm == null) {
s_logger.warn("VM " + vmId + " does not exist, sending blank response for console access request");
sendResponse(resp, "");
return;
}
if(vm.getHostId() == null) {
s_logger.warn("VM " + vmId + " lost host info, sending blank response for console access request");
sendResponse(resp, "");
return;
}
HostVO host = _ms.getHostBy(vm.getHostId());
if(host == null) {
s_logger.warn("VM " + vmId + "'s host does not exist, sending blank response for console access request");
sendResponse(resp, "");
return;
}
String rootUrl = _ms.getConsoleAccessUrlRoot(vmId);
if(rootUrl == null) {
sendResponse(resp, "<html><body><p>Console access will be ready in a few minutes. Please try it again later.</p></body></html>");
return;
}
String vmName = vm.getHostName();
if(vm.getType() == VirtualMachine.Type.User) {
UserVm userVm = (UserVm)_vmMgr.findByIdAndType(VirtualMachine.Type.User, vmId);
String displayName = userVm.getDisplayName();
if(displayName != null && !displayName.isEmpty() && !displayName.equals(vmName)) {
vmName += "(" + displayName + ")";
}
}
StringBuffer sb = new StringBuffer();
sb.append("<html><title>").append(escapeHTML(vmName)).append("</title><frameset><frame src=\"").append(composeConsoleAccessUrl(rootUrl, vm, host));
sb.append("\"></frame></frameset></html>");
s_logger.debug("the console url is :: " + sb.toString());
sendResponse(resp, sb.toString());
}
private void handleAuthRequest(HttpServletRequest req, HttpServletResponse resp, long vmId) {
// TODO authentication channel between console proxy VM and management server needs to be secured,
// the data is now being sent through private network, but this is apparently not enough
VMInstanceVO vm = _vmMgr.findById(vmId);
if(vm == null) {
s_logger.warn("VM " + vmId + " does not exist, sending failed response for authentication request from console proxy");
sendResponse(resp, "failed");
return;
}
if(vm.getHostId() == null) {
s_logger.warn("VM " + vmId + " lost host info, failed response for authentication request from console proxy");
sendResponse(resp, "failed");
return;
}
HostVO host = _ms.getHostBy(vm.getHostId());
if(host == null) {
s_logger.warn("VM " + vmId + "'s host does not exist, sending failed response for authentication request from console proxy");
sendResponse(resp, "failed");
return;
}
String sid = req.getParameter("sid");
if(sid == null || !sid.equals(vm.getVncPassword())) {
s_logger.warn("sid " + sid + " in url does not match stored sid " + vm.getVncPassword());
sendResponse(resp, "failed");
return;
}
sendResponse(resp, "success");
}
// put the ugly stuff here
static public Ternary<String, String, String> parseHostInfo(String hostInfo) {
String host = null;
String tunnelUrl = null;
String tunnelSession = null;
s_logger.info("Parse host info returned from executing GetVNCPortCommand. host info: " + hostInfo);
if(hostInfo != null && hostInfo.startsWith("consoleurl")) {
String tokens[] = hostInfo.split("&");
if(hostInfo.length() > 19 && hostInfo.indexOf('/', 19) > 19) {
host = hostInfo.substring(19, hostInfo.indexOf('/', 19)).trim();
tunnelUrl = tokens[0].substring("consoleurl=".length());
tunnelSession = tokens[1].split("=")[1];
} else {
host = "";
}
} else {
host = hostInfo;
}
return new Ternary<String, String, String>(host, tunnelUrl, tunnelSession); return new Ternary<String, String, String>(host, tunnelUrl, tunnelSession);
} }
private String composeThumbnailUrl(String rootUrl, VMInstanceVO vm, HostVO hostVo, int w, int h) {
StringBuffer sb = new StringBuffer(rootUrl);
String host = hostVo.getPrivateIpAddress(); private String composeThumbnailUrl(String rootUrl, VMInstanceVO vm, HostVO hostVo, int w, int h) {
StringBuffer sb = new StringBuffer(rootUrl);
Pair<String, Integer> portInfo = _ms.getVncPort(vm);
Ternary<String, String, String> parsedHostInfo = parseHostInfo(portInfo.first());
String sid = vm.getVncPassword();
String tag = String.valueOf(vm.getId());
tag = _identityService.getIdentityUuid("vm_instance", tag);
String ticket = genAccessTicket(host, String.valueOf(portInfo.second()), sid, tag);
ConsoleProxyPasswordBasedEncryptor encryptor = new ConsoleProxyPasswordBasedEncryptor(_ms.getHashKey()); String host = hostVo.getPrivateIpAddress();
ConsoleProxyClientParam param = new ConsoleProxyClientParam();
param.setClientHostAddress(parsedHostInfo.first()); Pair<String, Integer> portInfo = _ms.getVncPort(vm);
param.setClientHostPort(portInfo.second()); Ternary<String, String, String> parsedHostInfo = parseHostInfo(portInfo.first());
param.setClientHostPassword(sid);
param.setClientTag(tag); String sid = vm.getVncPassword();
param.setTicket(ticket); String tag = String.valueOf(vm.getId());
if(parsedHostInfo.second() != null && parsedHostInfo.third() != null) { tag = _identityService.getIdentityUuid("vm_instance", tag);
param.setClientTunnelUrl(parsedHostInfo.second()); String ticket = genAccessTicket(host, String.valueOf(portInfo.second()), sid, tag);
param.setClientTunnelSession(parsedHostInfo.third());
} ConsoleProxyPasswordBasedEncryptor encryptor = new ConsoleProxyPasswordBasedEncryptor(_ms.getHashKey());
ConsoleProxyClientParam param = new ConsoleProxyClientParam();
sb.append("/ajax?token=" + encryptor.encryptObject(ConsoleProxyClientParam.class, param)); param.setClientHostAddress(parsedHostInfo.first());
sb.append("&w=").append(w).append("&h=").append(h); param.setClientHostPort(portInfo.second());
param.setClientHostPassword(sid);
if(s_logger.isDebugEnabled()) { param.setClientTag(tag);
param.setTicket(ticket);
if(parsedHostInfo.second() != null && parsedHostInfo.third() != null) {
param.setClientTunnelUrl(parsedHostInfo.second());
param.setClientTunnelSession(parsedHostInfo.third());
}
sb.append("/ajax?token=" + encryptor.encryptObject(ConsoleProxyClientParam.class, param));
sb.append("&w=").append(w).append("&h=").append(h);
if(s_logger.isDebugEnabled()) {
s_logger.debug("Compose thumbnail url: " + sb.toString()); s_logger.debug("Compose thumbnail url: " + sb.toString());
} }
return sb.toString(); return sb.toString();
} }
private String composeConsoleAccessUrl(String rootUrl, VMInstanceVO vm, HostVO hostVo) {
StringBuffer sb = new StringBuffer(rootUrl);
String host = hostVo.getPrivateIpAddress();
Pair<String, Integer> portInfo = _ms.getVncPort(vm);
if(s_logger.isDebugEnabled())
s_logger.debug("Port info " + portInfo.first());
Ternary<String, String, String> parsedHostInfo = parseHostInfo(portInfo.first()); private String composeConsoleAccessUrl(String rootUrl, VMInstanceVO vm, HostVO hostVo) {
StringBuffer sb = new StringBuffer(rootUrl);
String host = hostVo.getPrivateIpAddress();
String sid = vm.getVncPassword(); Pair<String, Integer> portInfo = _ms.getVncPort(vm);
String tag = String.valueOf(vm.getId()); if(s_logger.isDebugEnabled())
tag = _identityService.getIdentityUuid("vm_instance", tag); s_logger.debug("Port info " + portInfo.first());
String ticket = genAccessTicket(host, String.valueOf(portInfo.second()), sid, tag);
ConsoleProxyPasswordBasedEncryptor encryptor = new ConsoleProxyPasswordBasedEncryptor(_ms.getHashKey()); Ternary<String, String, String> parsedHostInfo = parseHostInfo(portInfo.first());
ConsoleProxyClientParam param = new ConsoleProxyClientParam();
param.setClientHostAddress(parsedHostInfo.first()); String sid = vm.getVncPassword();
param.setClientHostPort(portInfo.second()); String tag = String.valueOf(vm.getId());
param.setClientHostPassword(sid); tag = _identityService.getIdentityUuid("vm_instance", tag);
param.setClientTag(tag); String ticket = genAccessTicket(host, String.valueOf(portInfo.second()), sid, tag);
param.setTicket(ticket); ConsoleProxyPasswordBasedEncryptor encryptor = new ConsoleProxyPasswordBasedEncryptor(_ms.getHashKey());
if(parsedHostInfo.second() != null && parsedHostInfo.third() != null) { ConsoleProxyClientParam param = new ConsoleProxyClientParam();
param.setClientTunnelUrl(parsedHostInfo.second()); param.setClientHostAddress(parsedHostInfo.first());
param.setClientTunnelSession(parsedHostInfo.third()); param.setClientHostPort(portInfo.second());
} param.setClientHostPassword(sid);
param.setClientTag(tag);
sb.append("/ajax?token=" + encryptor.encryptObject(ConsoleProxyClientParam.class, param)); param.setTicket(ticket);
if(parsedHostInfo.second() != null && parsedHostInfo.third() != null) {
// for console access, we need guest OS type to help implement keyboard param.setClientTunnelUrl(parsedHostInfo.second());
long guestOs = vm.getGuestOSId(); param.setClientTunnelSession(parsedHostInfo.third());
GuestOSVO guestOsVo = _ms.getGuestOs(guestOs); }
if(guestOsVo.getCategoryId() == 6)
sb.append("&guest=windows"); sb.append("/ajax?token=" + encryptor.encryptObject(ConsoleProxyClientParam.class, param));
if(s_logger.isDebugEnabled()) { // for console access, we need guest OS type to help implement keyboard
long guestOs = vm.getGuestOSId();
GuestOSVO guestOsVo = _ms.getGuestOs(guestOs);
if(guestOsVo.getCategoryId() == 6)
sb.append("&guest=windows");
if(s_logger.isDebugEnabled()) {
s_logger.debug("Compose console url: " + sb.toString()); s_logger.debug("Compose console url: " + sb.toString());
} }
return sb.toString(); return sb.toString();
} }
public static String genAccessTicket(String host, String port, String sid, String tag) { public static String genAccessTicket(String host, String port, String sid, String tag) {
return genAccessTicket(host, port, sid, tag, new Date()); return genAccessTicket(host, port, sid, tag, new Date());
} }
public static String genAccessTicket(String host, String port, String sid, String tag, Date normalizedHashTime) { public static String genAccessTicket(String host, String port, String sid, String tag, Date normalizedHashTime) {
String params = "host=" + host + "&port=" + port + "&sid=" + sid + "&tag=" + tag; String params = "host=" + host + "&port=" + port + "&sid=" + sid + "&tag=" + tag;
try { try {
Mac mac = Mac.getInstance("HmacSHA1"); Mac mac = Mac.getInstance("HmacSHA1");
long ts = normalizedHashTime.getTime(); long ts = normalizedHashTime.getTime();
ts = ts/60000; // round up to 1 minute ts = ts/60000; // round up to 1 minute
String secretKey = _ms.getHashKey(); String secretKey = _ms.getHashKey();
SecretKeySpec keySpec = new SecretKeySpec(secretKey.getBytes(), "HmacSHA1"); SecretKeySpec keySpec = new SecretKeySpec(secretKey.getBytes(), "HmacSHA1");
mac.init(keySpec); mac.init(keySpec);
mac.update(params.getBytes()); mac.update(params.getBytes());
mac.update(String.valueOf(ts).getBytes()); mac.update(String.valueOf(ts).getBytes());
byte[] encryptedBytes = mac.doFinal(); byte[] encryptedBytes = mac.doFinal();
return Base64.encodeBase64URLSafeString(encryptedBytes); return Base64.encodeBase64URLSafeString(encryptedBytes);
} catch(Exception e) { } catch(Exception e) {
s_logger.error("Unexpected exception ", e); s_logger.error("Unexpected exception ", e);
} }
return ""; return "";
} }
private void sendResponse(HttpServletResponse resp, String content) { private void sendResponse(HttpServletResponse resp, String content) {
try { try {
resp.setContentType("text/html"); resp.setContentType("text/html");
resp.getWriter().print(content); resp.getWriter().print(content);
} catch(IOException e) { } catch(IOException e) {
if(s_logger.isInfoEnabled()) { if(s_logger.isInfoEnabled()) {
s_logger.info("Client may already close the connection"); s_logger.info("Client may already close the connection");
} }
} }
} }
private boolean checkSessionPermision(HttpServletRequest req, long vmId, Account accountObj) { private boolean checkSessionPermision(HttpServletRequest req, long vmId, Account accountObj) {
VMInstanceVO vm = _vmMgr.findById(vmId); VMInstanceVO vm = _vmMgr.findById(vmId);
if(vm == null) { if(vm == null) {
s_logger.debug("Console/thumbnail access denied. VM " + vmId + " does not exist in system any more"); s_logger.debug("Console/thumbnail access denied. VM " + vmId + " does not exist in system any more");
return false; return false;
} }
// root admin can access anything // root admin can access anything
if(accountObj.getType() == Account.ACCOUNT_TYPE_ADMIN) if(accountObj.getType() == Account.ACCOUNT_TYPE_ADMIN)
return true; return true;
switch(vm.getType()) switch(vm.getType())
{ {
case User : case User :
try { try {
_accountMgr.checkAccess(accountObj, null, true, vm); _accountMgr.checkAccess(accountObj, null, true, vm);
} catch (PermissionDeniedException ex) { } catch (PermissionDeniedException ex) {
if (accountObj.getType() == Account.ACCOUNT_TYPE_NORMAL) { if (accountObj.getType() == Account.ACCOUNT_TYPE_NORMAL) {
if (s_logger.isDebugEnabled()) { if (s_logger.isDebugEnabled()) {
s_logger.debug("VM access is denied. VM owner account " + vm.getAccountId() s_logger.debug("VM access is denied. VM owner account " + vm.getAccountId()
+ " does not match the account id in session " + accountObj.getId() + " and caller is a normal user"); + " does not match the account id in session " + accountObj.getId() + " and caller is a normal user");
} }
} else if(accountObj.getType() == Account.ACCOUNT_TYPE_DOMAIN_ADMIN || accountObj.getType() == Account.ACCOUNT_TYPE_READ_ONLY_ADMIN) { } else if(accountObj.getType() == Account.ACCOUNT_TYPE_DOMAIN_ADMIN || accountObj.getType() == Account.ACCOUNT_TYPE_READ_ONLY_ADMIN) {
if(s_logger.isDebugEnabled()) { if(s_logger.isDebugEnabled()) {
s_logger.debug("VM access is denied. VM owner account " + vm.getAccountId() s_logger.debug("VM access is denied. VM owner account " + vm.getAccountId()
+ " does not match the account id in session " + accountObj.getId() + " and the domain-admin caller does not manage the target domain"); + " does not match the account id in session " + accountObj.getId() + " and the domain-admin caller does not manage the target domain");
} }
} }
return false; return false;
} }
break; break;
case DomainRouter: case DomainRouter:
case ConsoleProxy : case ConsoleProxy :
case SecondaryStorageVm: case SecondaryStorageVm:
return false; return false;
default : default :
s_logger.warn("Unrecoginized virtual machine type, deny access by default. type: " + vm.getType()); s_logger.warn("Unrecoginized virtual machine type, deny access by default. type: " + vm.getType());
return false; return false;
} }
return true; return true;
} }
private boolean isValidCmd(String cmd) { private boolean isValidCmd(String cmd) {
if(cmd.equalsIgnoreCase("thumbnail") || cmd.equalsIgnoreCase("access") || cmd.equalsIgnoreCase("auth")) { if(cmd.equalsIgnoreCase("thumbnail") || cmd.equalsIgnoreCase("access") || cmd.equalsIgnoreCase("auth")) {
return true; return true;
} }
return false;
}
public boolean verifyUser(Long userId) {
// copy from ApiServer.java, a bit ugly here
User user = _accountMgr.getUserIncludingRemoved(userId);
Account account = null;
if (user != null) {
account = _accountMgr.getAccount(user.getAccountId());
}
if ((user == null) || (user.getRemoved() != null) || !user.getState().equals(Account.State.enabled) return false;
|| (account == null) || !account.getState().equals(Account.State.enabled)) {
s_logger.warn("Deleted/Disabled/Locked user with id=" + userId + " attempting to access public API");
return false;
}
return true;
} }
// copied and modified from ApiServer.java. public boolean verifyUser(Long userId) {
// copy from ApiServer.java, a bit ugly here
User user = _accountMgr.getUserIncludingRemoved(userId);
Account account = null;
if (user != null) {
account = _accountMgr.getAccount(user.getAccountId());
}
if ((user == null) || (user.getRemoved() != null) || !user.getState().equals(Account.State.enabled)
|| (account == null) || !account.getState().equals(Account.State.enabled)) {
s_logger.warn("Deleted/Disabled/Locked user with id=" + userId + " attempting to access public API");
return false;
}
return true;
}
// copied and modified from ApiServer.java.
// TODO need to replace the whole servlet with a API command // TODO need to replace the whole servlet with a API command
private boolean verifyRequest(Map<String, Object[]> requestParameters) { private boolean verifyRequest(Map<String, Object[]> requestParameters) {
try { try {
@ -516,7 +516,7 @@ public class ConsoleProxyServlet extends HttpServlet {
for (String paramName : parameterNames) { for (String paramName : parameterNames) {
// parameters come as name/value pairs in the form String/String[] // parameters come as name/value pairs in the form String/String[]
String paramValue = ((String[])requestParameters.get(paramName))[0]; String paramValue = ((String[])requestParameters.get(paramName))[0];
if ("signature".equalsIgnoreCase(paramName)) { if ("signature".equalsIgnoreCase(paramName)) {
signature = paramValue; signature = paramValue;
} else { } else {
@ -531,7 +531,7 @@ public class ConsoleProxyServlet extends HttpServlet {
} }
} }
} }
// if api/secret key are passed to the parameters // if api/secret key are passed to the parameters
if ((signature == null) || (apiKey == null)) { if ((signature == null) || (apiKey == null)) {
@ -573,16 +573,16 @@ public class ConsoleProxyServlet extends HttpServlet {
mac.init(keySpec); mac.init(keySpec);
mac.update(unsignedRequest.getBytes()); mac.update(unsignedRequest.getBytes());
byte[] encryptedBytes = mac.doFinal(); byte[] encryptedBytes = mac.doFinal();
String computedSignature = com.cloud.utils.encoding.Base64.encodeBytes(encryptedBytes); String computedSignature = Base64.encodeBase64URLSafeString(encryptedBytes);
boolean equalSig = signature.equals(computedSignature); boolean equalSig = signature.equals(computedSignature);
if (!equalSig) { if (!equalSig) {
s_logger.debug("User signature: " + signature + " is not equaled to computed signature: " + computedSignature); s_logger.debug("User signature: " + signature + " is not equaled to computed signature: " + computedSignature);
} }
if(equalSig) { if(equalSig) {
requestParameters.put("userid", new Object[] {String.valueOf(user.getId())}); requestParameters.put("userid", new Object[] {String.valueOf(user.getId())});
requestParameters.put("account", new Object[] {account.getAccountName()}); requestParameters.put("account", new Object[] {account.getAccountName()});
requestParameters.put("accountobj", new Object[] { account }); requestParameters.put("accountobj", new Object[] { account });
} }
return equalSig; return equalSig;
} catch (Exception ex) { } catch (Exception ex) {
@ -590,23 +590,23 @@ public class ConsoleProxyServlet extends HttpServlet {
} }
return false; return false;
} }
public static final String escapeHTML(String content){ public static final String escapeHTML(String content){
if(content == null || content.isEmpty()) if(content == null || content.isEmpty())
return content; return content;
StringBuffer sb = new StringBuffer(); StringBuffer sb = new StringBuffer();
for (int i = 0; i < content.length(); i++) { for (int i = 0; i < content.length(); i++) {
char c = content.charAt(i); char c = content.charAt(i);
switch (c) { switch (c) {
case '<': sb.append("&lt;"); break; case '<': sb.append("&lt;"); break;
case '>': sb.append("&gt;"); break; case '>': sb.append("&gt;"); break;
case '&': sb.append("&amp;"); break; case '&': sb.append("&amp;"); break;
case '"': sb.append("&quot;"); break; case '"': sb.append("&quot;"); break;
case ' ': sb.append("&nbsp;");break; case ' ': sb.append("&nbsp;");break;
default: sb.append(c); break; default: sb.append(c); break;
} }
} }
return sb.toString(); return sb.toString();
} }
} }

3
test/src/com/cloud/sample/UserCloudAPIExecutor.java
View File

@ -31,10 +31,9 @@ import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpMethod; import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.methods.GetMethod; import org.apache.commons.httpclient.methods.GetMethod;
import com.cloud.utils.encoding.Base64;
/** /**
* *
* *
* *
* *

2630
test/src/com/cloud/test/stress/StressTestDirectAttach.java
View File

File diff suppressed because it is too large Load Diff

64
test/src/com/cloud/test/stress/TestClientWithAPI.java
View File

@ -49,7 +49,6 @@ import org.w3c.dom.Element;
import org.w3c.dom.Node; import org.w3c.dom.Node;
import org.w3c.dom.NodeList; import org.w3c.dom.NodeList;
import com.cloud.utils.encoding.Base64;
import com.cloud.utils.exception.CloudRuntimeException; import com.cloud.utils.exception.CloudRuntimeException;
import com.trilead.ssh2.ChannelCondition; import com.trilead.ssh2.ChannelCondition;
import com.trilead.ssh2.Connection; import com.trilead.ssh2.Connection;
@ -182,7 +181,7 @@ public class TestClientWithAPI {
if (arg.equals("-no")) { if (arg.equals("-no")) {
networkOfferingId = iter.next(); networkOfferingId = iter.next();
} }
if (arg.equals("-pass")) { if (arg.equals("-pass")) {
vmPassword = iter.next(); vmPassword = iter.next();
} }
@ -210,6 +209,7 @@ public class TestClientWithAPI {
for (int i = 0; i < numThreads; i++) { for (int i = 0; i < numThreads; i++) {
new Thread(new Runnable() { new Thread(new Runnable() {
@Override
public void run() { public void run() {
do { do {
String username = null; String username = null;
@ -440,7 +440,7 @@ public class TestClientWithAPI {
for (int j = 0; j < childNodes.getLength(); j++) { for (int j = 0; j < childNodes.getLength(); j++) {
Node n = childNodes.item(j); Node n = childNodes.item(j);
if ("id".equals(n.getNodeName())) { if ("id".equals(n.getNodeName())) {
// if ("ipaddress".equals(n.getNodeName())) { // if ("ipaddress".equals(n.getNodeName())) {
ipAddress = n.getTextContent(); ipAddress = n.getTextContent();
} else if ("issourcenat".equals(n.getNodeName())) { } else if ("issourcenat".equals(n.getNodeName())) {
isSourceNat = Boolean.parseBoolean(n.getTextContent()); isSourceNat = Boolean.parseBoolean(n.getTextContent());
@ -478,7 +478,7 @@ public class TestClientWithAPI {
} }
else if("ipaddress".equals(n.getNodeName())) else if("ipaddress".equals(n.getNodeName()))
{ {
ipAddress = n.getTextContent(); ipAddress = n.getTextContent();
} }
else if ("issourcenat".equals(n.getNodeName())) { else if ("issourcenat".equals(n.getNodeName())) {
isSourceNat = Boolean.parseBoolean(n.getTextContent()); isSourceNat = Boolean.parseBoolean(n.getTextContent());
@ -609,7 +609,7 @@ public class TestClientWithAPI {
s_logger.error("Create virtual network failed for account " + username + " with error code :" + responseCode + ", aborting deployment test. The command was sent with url " + url); s_logger.error("Create virtual network failed for account " + username + " with error code :" + responseCode + ", aborting deployment test. The command was sent with url " + url);
return -1; return -1;
} }
/* /*
// --------------------------------- // ---------------------------------
// CREATE DIRECT NETWORK // CREATE DIRECT NETWORK
// --------------------------------- // ---------------------------------
@ -629,9 +629,9 @@ public class TestClientWithAPI {
s_logger.error("Create direct network failed for account " + username + " with error code :" + responseCode + ", aborting deployment test. The command was sent with url " + url); s_logger.error("Create direct network failed for account " + username + " with error code :" + responseCode + ", aborting deployment test. The command was sent with url " + url);
return -1; return -1;
} }
*/ */
// --------------------------------- // ---------------------------------
// DEPLOY LINUX VM // DEPLOY LINUX VM
// --------------------------------- // ---------------------------------
@ -706,7 +706,7 @@ public class TestClientWithAPI {
} }
else else
{ {
s_logger.info("Associate IP Address response code: " + responseCode); s_logger.info("Associate IP Address response code: " + responseCode);
long publicIpId = Long.parseLong(values.get("id")); long publicIpId = Long.parseLong(values.get("id"));
s_logger.info("Associate IP's Id: " + publicIpId); s_logger.info("Associate IP's Id: " + publicIpId);
_publicIpId.set(values.get("id")); _publicIpId.set(values.get("id"));
@ -715,13 +715,13 @@ public class TestClientWithAPI {
s_logger.error("associate ip address for windows vm failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("associate ip address for windows vm failed with error code: " + responseCode + ". Following URL was sent: " + url);
return responseCode; return responseCode;
} }
String encodedPublicIpId = URLEncoder.encode(_publicIpId.get(), "UTF-8"); String encodedPublicIpId = URLEncoder.encode(_publicIpId.get(), "UTF-8");
requestToSign = "apikey=" + encodedApiKey + "&command=listPublicIpAddresses"+"&id="+ encodedPublicIpId; requestToSign = "apikey=" + encodedApiKey + "&command=listPublicIpAddresses"+"&id="+ encodedPublicIpId;
requestToSign = requestToSign.toLowerCase(); requestToSign = requestToSign.toLowerCase();
signature = signRequest(requestToSign, _secretKey.get()); signature = signRequest(requestToSign, _secretKey.get());
encodedSignature = URLEncoder.encode(signature, "UTF-8"); encodedSignature = URLEncoder.encode(signature, "UTF-8");
url = developerServer + "?command=listPublicIpAddresses&apikey=" + encodedApiKey + "&id=" + encodedPublicIpId + "&signature=" + encodedSignature; url = developerServer + "?command=listPublicIpAddresses&apikey=" + encodedApiKey + "&id=" + encodedPublicIpId + "&signature=" + encodedSignature;
client = new HttpClient(); client = new HttpClient();
method = new GetMethod(url); method = new GetMethod(url);
@ -730,9 +730,9 @@ public class TestClientWithAPI {
s_logger.info("list ip addresses for user " + userId + " response code: " + responseCode); s_logger.info("list ip addresses for user " + userId + " response code: " + responseCode);
if (responseCode == 200) { if (responseCode == 200) {
InputStream is = method.getResponseBodyAsStream(); InputStream is = method.getResponseBodyAsStream();
// InputStream ips = method.getResponseBodyAsStream(); // InputStream ips = method.getResponseBodyAsStream();
List<String> ipAddressValues = getIPs(is, false); List<String> ipAddressValues = getIPs(is, false);
// List<String> ipAddressVals = getIPs(is, false, true); // List<String> ipAddressVals = getIPs(is, false, true);
if ((ipAddressValues != null) && !ipAddressValues.isEmpty()) { if ((ipAddressValues != null) && !ipAddressValues.isEmpty()) {
_windowsIpId.set(ipAddressValues.get(0)); _windowsIpId.set(ipAddressValues.get(0));
_windowsIP.set(ipAddressValues.get(1)); _windowsIP.set(ipAddressValues.get(1));
@ -743,7 +743,7 @@ public class TestClientWithAPI {
s_logger.error("list ip addresses failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("list ip addresses failed with error code: " + responseCode + ". Following URL was sent: " + url);
return responseCode; return responseCode;
} }
// --------------------------------- // ---------------------------------
// Use the SourceNat IP for linux // Use the SourceNat IP for linux
// --------------------------------- // ---------------------------------
@ -776,17 +776,17 @@ public class TestClientWithAPI {
return responseCode; return responseCode;
} }
} }
//-------------------------------------------- //--------------------------------------------
// Enable Static NAT for the Source NAT Ip // Enable Static NAT for the Source NAT Ip
//-------------------------------------------- //--------------------------------------------
String encodedSourceNatPublicIpId = URLEncoder.encode(_linuxIpId.get(), "UTF-8"); String encodedSourceNatPublicIpId = URLEncoder.encode(_linuxIpId.get(), "UTF-8");
/* requestToSign = "apikey=" + encodedApiKey + "&command=enableStaticNat"+"&id=" + encodedSourceNatPublicIpId + "&virtualMachineId=" + encodedVmId;; /* requestToSign = "apikey=" + encodedApiKey + "&command=enableStaticNat"+"&id=" + encodedSourceNatPublicIpId + "&virtualMachineId=" + encodedVmId;;
requestToSign = requestToSign.toLowerCase(); requestToSign = requestToSign.toLowerCase();
signature = signRequest(requestToSign, _secretKey.get()); signature = signRequest(requestToSign, _secretKey.get());
encodedSignature = URLEncoder.encode(signature, "UTF-8"); encodedSignature = URLEncoder.encode(signature, "UTF-8");
url = developerServer + "?command=enableStaticNat&apikey=" + encodedApiKey + "&signature=" + encodedSignature + "&id=" + encodedSourceNatPublicIpId + "&virtualMachineId=" + encodedVmId; url = developerServer + "?command=enableStaticNat&apikey=" + encodedApiKey + "&signature=" + encodedSignature + "&id=" + encodedSourceNatPublicIpId + "&virtualMachineId=" + encodedVmId;
client = new HttpClient(); client = new HttpClient();
method = new GetMethod(url); method = new GetMethod(url);
@ -801,7 +801,7 @@ public class TestClientWithAPI {
s_logger.error("Enable Static NAT failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("Enable Static NAT failed with error code: " + responseCode + ". Following URL was sent: " + url);
return responseCode; return responseCode;
} }
*/ */
// ------------------------------------------------------------- // -------------------------------------------------------------
// CREATE IP FORWARDING RULE -- Linux VM // CREATE IP FORWARDING RULE -- Linux VM
// ------------------------------------------------------------- // -------------------------------------------------------------
@ -825,7 +825,7 @@ public class TestClientWithAPI {
long ipfwdid = Long.parseLong(values.get("id")); long ipfwdid = Long.parseLong(values.get("id"));
s_logger.info("got Port Forwarding Rule's Id:" + ipfwdid); s_logger.info("got Port Forwarding Rule's Id:" + ipfwdid);
_linipfwdid.set(values.get("id")); _linipfwdid.set(values.get("id"));
} else { } else {
s_logger.error("Port forwarding rule creation failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("Port forwarding rule creation failed with error code: " + responseCode + ". Following URL was sent: " + url);
return responseCode; return responseCode;
@ -882,7 +882,7 @@ public class TestClientWithAPI {
String encodedTemplateId = URLEncoder.encode("" + templateId, "UTF-8"); String encodedTemplateId = URLEncoder.encode("" + templateId, "UTF-8");
encodedApiKey = URLEncoder.encode(_apiKey.get(), "UTF-8"); encodedApiKey = URLEncoder.encode(_apiKey.get(), "UTF-8");
String encodedNetworkIds = URLEncoder.encode(_networkId.get()+",206","UTF-8"); String encodedNetworkIds = URLEncoder.encode(_networkId.get()+",206","UTF-8");
requestToSign = "apikey=" + encodedApiKey + "&command=deployVirtualMachine&diskofferingid=" + diskOfferingId + "&networkids=" + encodedNetworkIds + "&serviceofferingid=" + encodedServiceOfferingId + "&templateid=" + encodedTemplateId requestToSign = "apikey=" + encodedApiKey + "&command=deployVirtualMachine&diskofferingid=" + diskOfferingId + "&networkids=" + encodedNetworkIds + "&serviceofferingid=" + encodedServiceOfferingId + "&templateid=" + encodedTemplateId
+ "&zoneid=" + encodedZoneId; + "&zoneid=" + encodedZoneId;
requestToSign = requestToSign.toLowerCase(); requestToSign = requestToSign.toLowerCase();
@ -918,14 +918,14 @@ public class TestClientWithAPI {
//-------------------------------------------- //--------------------------------------------
// Enable Static NAT for the Non Source NAT Ip // Enable Static NAT for the Non Source NAT Ip
//-------------------------------------------- //--------------------------------------------
encodedVmId = URLEncoder.encode(_windowsVmId.get(), "UTF-8"); encodedVmId = URLEncoder.encode(_windowsVmId.get(), "UTF-8");
encodedPublicIpId = URLEncoder.encode(_publicIpId.get(), "UTF-8"); encodedPublicIpId = URLEncoder.encode(_publicIpId.get(), "UTF-8");
requestToSign = "apikey=" + encodedApiKey + "&command=enableStaticNat"+"&ipaddressid="+ encodedPublicIpId + "&virtualMachineId=" + encodedVmId; requestToSign = "apikey=" + encodedApiKey + "&command=enableStaticNat"+"&ipaddressid="+ encodedPublicIpId + "&virtualMachineId=" + encodedVmId;
requestToSign = requestToSign.toLowerCase(); requestToSign = requestToSign.toLowerCase();
signature = signRequest(requestToSign, _secretKey.get()); signature = signRequest(requestToSign, _secretKey.get());
encodedSignature = URLEncoder.encode(signature, "UTF-8"); encodedSignature = URLEncoder.encode(signature, "UTF-8");
url = developerServer + "?command=enableStaticNat&apikey=" + encodedApiKey + "&ipaddressid=" + encodedPublicIpId + "&signature=" + encodedSignature + "&virtualMachineId=" + encodedVmId; url = developerServer + "?command=enableStaticNat&apikey=" + encodedApiKey + "&ipaddressid=" + encodedPublicIpId + "&signature=" + encodedSignature + "&virtualMachineId=" + encodedVmId;
client = new HttpClient(); client = new HttpClient();
method = new GetMethod(url); method = new GetMethod(url);
@ -941,7 +941,7 @@ public class TestClientWithAPI {
return responseCode; return responseCode;
} }
// ------------------------------------------------------------- // -------------------------------------------------------------
// CREATE IP FORWARDING RULE -- Windows VM // CREATE IP FORWARDING RULE -- Windows VM
// ------------------------------------------------------------- // -------------------------------------------------------------
@ -972,7 +972,7 @@ public class TestClientWithAPI {
s_logger.error("Port forwarding rule creation failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("Port forwarding rule creation failed with error code: " + responseCode + ". Following URL was sent: " + url);
return responseCode; return responseCode;
} }
} }
} }
return responseCode; return responseCode;
} }
@ -1178,7 +1178,7 @@ public class TestClientWithAPI {
} }
// Create volume from the snapshot created on the previous step and attach it to the running vm // Create volume from the snapshot created on the previous step and attach it to the running vm
/* encodedApiKey = URLEncoder.encode(_apiKey.get(), "UTF-8"); /* encodedApiKey = URLEncoder.encode(_apiKey.get(), "UTF-8");
requestToSign = "apikey=" + encodedApiKey + "&command=createVolume&name=" + _account.get() + "&snapshotid=" + _snapshot.get(); requestToSign = "apikey=" + encodedApiKey + "&command=createVolume&name=" + _account.get() + "&snapshotid=" + _snapshot.get();
requestToSign = requestToSign.toLowerCase(); requestToSign = requestToSign.toLowerCase();
signature = signRequest(requestToSign, _secretKey.get()); signature = signRequest(requestToSign, _secretKey.get());
@ -1222,7 +1222,7 @@ public class TestClientWithAPI {
return responseCode; return responseCode;
} }
} }
*/ */
// ----------------------------- // -----------------------------
// Execute reboot/stop/start commands for the VMs before deleting the account - made to exercise xen // Execute reboot/stop/start commands for the VMs before deleting the account - made to exercise xen
// ----------------------------- // -----------------------------
@ -1896,7 +1896,7 @@ public class TestClientWithAPI {
InputStream input = method.getResponseBodyAsStream(); InputStream input = method.getResponseBodyAsStream();
Element el = queryAsyncJobResult(server, input); Element el = queryAsyncJobResult(server, input);
s_logger.info("IP forwarding rule was successfully deleted"); s_logger.info("IP forwarding rule was successfully deleted");
} else { } else {
s_logger.error("IP forwarding rule creation failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("IP forwarding rule creation failed with error code: " + responseCode + ". Following URL was sent: " + url);
return responseCode; return responseCode;
@ -1911,7 +1911,7 @@ public class TestClientWithAPI {
requestToSign = requestToSign.toLowerCase(); requestToSign = requestToSign.toLowerCase();
signature = signRequest(requestToSign, _secretKey.get()); signature = signRequest(requestToSign, _secretKey.get());
encodedSignature = URLEncoder.encode(signature, "UTF-8"); encodedSignature = URLEncoder.encode(signature, "UTF-8");
url = developerServer + "?command=disableStaticNat&apikey=" + encodedApiKey + "&id=" + encodedPublicIpId + "&signature=" + encodedSignature ; url = developerServer + "?command=disableStaticNat&apikey=" + encodedApiKey + "&id=" + encodedPublicIpId + "&signature=" + encodedSignature ;
client = new HttpClient(); client = new HttpClient();
method = new GetMethod(url); method = new GetMethod(url);
@ -1926,7 +1926,7 @@ public class TestClientWithAPI {
s_logger.error("Disable Static NAT failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("Disable Static NAT failed with error code: " + responseCode + ". Following URL was sent: " + url);
return responseCode; return responseCode;
} }
// ----------------------------------------- // -----------------------------------------
// DISASSOCIATE IP ADDRESSES // DISASSOCIATE IP ADDRESSES
// ----------------------------------------- // -----------------------------------------
@ -1946,7 +1946,7 @@ public class TestClientWithAPI {
InputStream input = method.getResponseBodyAsStream(); InputStream input = method.getResponseBodyAsStream();
Element disassocipel = queryAsyncJobResult(server, input); Element disassocipel = queryAsyncJobResult(server, input);
Map<String, String> success = getSingleValueFromXML(disassocipel, new String[] {"success"}); Map<String, String> success = getSingleValueFromXML(disassocipel, new String[] {"success"});
// Map<String, String> success = getSingleValueFromXML(input, new String[] { "success" }); // Map<String, String> success = getSingleValueFromXML(input, new String[] { "success" });
s_logger.info("disassociate ip address..success? " + success.get("success")); s_logger.info("disassociate ip address..success? " + success.get("success"));
} else { } else {
s_logger.error("disassociate ip address failed with error code: " + responseCode + ". Following URL was sent: " + url); s_logger.error("disassociate ip address failed with error code: " + responseCode + ". Following URL was sent: " + url);
@ -1977,7 +1977,7 @@ public class TestClientWithAPI {
mac.init(keySpec); mac.init(keySpec);
mac.update(request.getBytes()); mac.update(request.getBytes());
byte[] encryptedBytes = mac.doFinal(); byte[] encryptedBytes = mac.doFinal();
return Base64.encodeBytes(encryptedBytes); return org.apache.commons.codec.binary.Base64.encodeBase64URLSafeString(encryptedBytes);
} catch (Exception ex) { } catch (Exception ex) {
s_logger.error("unable to sign request", ex); s_logger.error("unable to sign request", ex);
} }

384
test/src/com/cloud/test/utils/UtilsForTest.java
View File

@ -30,208 +30,208 @@ import javax.crypto.spec.SecretKeySpec;
import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.parsers.DocumentBuilderFactory;
import org.apache.commons.codec.binary.Base64;
import org.w3c.dom.Document; import org.w3c.dom.Document;
import org.w3c.dom.Element; import org.w3c.dom.Element;
import org.w3c.dom.Node; import org.w3c.dom.Node;
import org.w3c.dom.NodeList; import org.w3c.dom.NodeList;
import com.cloud.utils.encoding.Base64;
import com.cloud.utils.exception.CloudRuntimeException; import com.cloud.utils.exception.CloudRuntimeException;
public class UtilsForTest { public class UtilsForTest {
private static DocumentBuilderFactory factory = DocumentBuilderFactory
.newInstance();
public static boolean verifyTags (Map<String, String> params) {
boolean result = true;
for (String value : params.keySet()) {
if (params.get(value) == null) {
result=false;
}
}
return result;
}
public static boolean verifyTagValues (Map<String, String> params, Map<String, String> pattern) {
boolean result = true;
if (pattern != null) {
for (String value : pattern.keySet()) {
if (!pattern.get(value).equals(params.get(value))) {
result=false;
System.out.println("Tag " + value + " has " + params.get(value) + " while expected value is: " + pattern.get(value));
}
}
}
return result;
}
public static Map<String, String> parseXML(InputStream is,
String[] tagNames) {
Map<String, String> returnValues = new HashMap<String, String>();
try {
DocumentBuilder docBuilder = factory.newDocumentBuilder();
Document doc = docBuilder.parse(is);
Element rootElement = doc.getDocumentElement();
for (int i = 0; i < tagNames.length; i++) { private static DocumentBuilderFactory factory = DocumentBuilderFactory
NodeList targetNodes = rootElement .newInstance();
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in the response");
returnValues.put(tagNames[i], null);
} else {
returnValues.put(tagNames[i], targetNodes.item(0)
.getTextContent());
}
}
} catch (Exception ex) {
System.out.println("error processing XML");
ex.printStackTrace();
}
return returnValues;
}
public static ArrayList<HashMap<String, String>> parseMulXML (InputStream is, String[] tagNames){
ArrayList<HashMap<String, String>> returnValues = new ArrayList<HashMap <String, String>>();
try {
DocumentBuilder docBuilder = factory.newDocumentBuilder();
Document doc = docBuilder.parse(is);
Element rootElement = doc.getDocumentElement();
for (int i = 0; i < tagNames.length; i++) {
NodeList targetNodes = rootElement
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in XML response...returning null");
} else {
for (int j = 0; j < targetNodes.getLength(); j++) {
HashMap<String, String> valueList = new HashMap<String,String> ();
Node node = targetNodes.item(j);
//parse child nodes
NodeList child = node.getChildNodes();
for (int c=0; c<node.getChildNodes().getLength(); c++){
child.item(c).getNodeName();
valueList.put(child.item(c).getNodeName(), child.item(c).getTextContent());
}
returnValues.add(valueList);
}
}
}
} catch (Exception ex) {
System.out.println(ex);
}
return returnValues;
}
public static String createMD5String(String password) {
MessageDigest md5;
try {
md5 = MessageDigest.getInstance("MD5");
} catch (NoSuchAlgorithmException e) {
throw new CloudRuntimeException("Error", e);
}
md5.reset(); public static boolean verifyTags (Map<String, String> params) {
BigInteger pwInt = new BigInteger(1, md5.digest(password.getBytes())); boolean result = true;
for (String value : params.keySet()) {
if (params.get(value) == null) {
result=false;
}
}
return result;
}
// make sure our MD5 hash value is 32 digits long... public static boolean verifyTagValues (Map<String, String> params, Map<String, String> pattern) {
StringBuffer sb = new StringBuffer(); boolean result = true;
String pwStr = pwInt.toString(16);
int padding = 32 - pwStr.length(); if (pattern != null) {
for (int i = 0; i < padding; i++) { for (String value : pattern.keySet()) {
sb.append('0'); if (!pattern.get(value).equals(params.get(value))) {
} result=false;
sb.append(pwStr); System.out.println("Tag " + value + " has " + params.get(value) + " while expected value is: " + pattern.get(value));
return sb.toString(); }
} }
}
return result;
}
public static Map<String, String> parseXML(InputStream is,
String[] tagNames) {
public static Map<String, String> getSingleValueFromXML(InputStream is, Map<String, String> returnValues = new HashMap<String, String>();
String[] tagNames) { try {
Map<String, String> returnValues = new HashMap<String, String>(); DocumentBuilder docBuilder = factory.newDocumentBuilder();
try { Document doc = docBuilder.parse(is);
DocumentBuilder docBuilder = factory.newDocumentBuilder(); Element rootElement = doc.getDocumentElement();
Document doc = docBuilder.parse(is);
Element rootElement = doc.getDocumentElement(); for (int i = 0; i < tagNames.length; i++) {
NodeList targetNodes = rootElement
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in the response");
returnValues.put(tagNames[i], null);
} else {
returnValues.put(tagNames[i], targetNodes.item(0)
.getTextContent());
}
}
} catch (Exception ex) {
System.out.println("error processing XML");
ex.printStackTrace();
}
return returnValues;
}
public static ArrayList<HashMap<String, String>> parseMulXML (InputStream is, String[] tagNames){
ArrayList<HashMap<String, String>> returnValues = new ArrayList<HashMap <String, String>>();
try {
DocumentBuilder docBuilder = factory.newDocumentBuilder();
Document doc = docBuilder.parse(is);
Element rootElement = doc.getDocumentElement();
for (int i = 0; i < tagNames.length; i++) {
NodeList targetNodes = rootElement
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in XML response...returning null");
} else {
for (int j = 0; j < targetNodes.getLength(); j++) {
HashMap<String, String> valueList = new HashMap<String,String> ();
Node node = targetNodes.item(j);
//parse child nodes
NodeList child = node.getChildNodes();
for (int c=0; c<node.getChildNodes().getLength(); c++){
child.item(c).getNodeName();
valueList.put(child.item(c).getNodeName(), child.item(c).getTextContent());
}
returnValues.add(valueList);
}
}
}
} catch (Exception ex) {
System.out.println(ex);
}
return returnValues;
}
public static String createMD5String(String password) {
MessageDigest md5;
try {
md5 = MessageDigest.getInstance("MD5");
} catch (NoSuchAlgorithmException e) {
throw new CloudRuntimeException("Error", e);
}
md5.reset();
BigInteger pwInt = new BigInteger(1, md5.digest(password.getBytes()));
// make sure our MD5 hash value is 32 digits long...
StringBuffer sb = new StringBuffer();
String pwStr = pwInt.toString(16);
int padding = 32 - pwStr.length();
for (int i = 0; i < padding; i++) {
sb.append('0');
}
sb.append(pwStr);
return sb.toString();
}
public static Map<String, String> getSingleValueFromXML(InputStream is,
String[] tagNames) {
Map<String, String> returnValues = new HashMap<String, String>();
try {
DocumentBuilder docBuilder = factory.newDocumentBuilder();
Document doc = docBuilder.parse(is);
Element rootElement = doc.getDocumentElement();
for (int i = 0; i < tagNames.length; i++) {
NodeList targetNodes = rootElement
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in XML response...returning null");
} else {
returnValues.put(tagNames[i], targetNodes.item(0)
.getTextContent());
}
}
} catch (Exception ex) {
System.out.println("error processing XML");
ex.printStackTrace();
}
return returnValues;
}
public static Map<String, List<String>> getMultipleValuesFromXML(
InputStream is, String[] tagNames) {
Map<String, List<String>> returnValues = new HashMap<String, List<String>>();
try {
DocumentBuilder docBuilder = factory.newDocumentBuilder();
Document doc = docBuilder.parse(is);
Element rootElement = doc.getDocumentElement();
for (int i = 0; i < tagNames.length; i++) {
NodeList targetNodes = rootElement
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in XML response...returning null");
} else {
List<String> valueList = new ArrayList<String>();
for (int j = 0; j < targetNodes.getLength(); j++) {
Node node = targetNodes.item(j);
valueList.add(node.getTextContent());
}
returnValues.put(tagNames[i], valueList);
}
}
} catch (Exception ex) {
System.out.println(ex);
}
return returnValues;
}
public static String signRequest(String request, String key) {
try {
Mac mac = Mac.getInstance("HmacSHA1");
SecretKeySpec keySpec = new SecretKeySpec(key.getBytes(),
"HmacSHA1");
mac.init(keySpec);
mac.update(request.getBytes());
byte[] encryptedBytes = mac.doFinal();
//System.out.println("HmacSHA1 hash: " + encryptedBytes);
return Base64.encodeBase64URLSafeString(encryptedBytes);
} catch (Exception ex) {
System.out.println("unable to sign request");
ex.printStackTrace();
}
return null;
}
for (int i = 0; i < tagNames.length; i++) {
NodeList targetNodes = rootElement
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in XML response...returning null");
} else {
returnValues.put(tagNames[i], targetNodes.item(0)
.getTextContent());
}
}
} catch (Exception ex) {
System.out.println("error processing XML");
ex.printStackTrace();
}
return returnValues;
}
public static Map<String, List<String>> getMultipleValuesFromXML(
InputStream is, String[] tagNames) {
Map<String, List<String>> returnValues = new HashMap<String, List<String>>();
try {
DocumentBuilder docBuilder = factory.newDocumentBuilder();
Document doc = docBuilder.parse(is);
Element rootElement = doc.getDocumentElement();
for (int i = 0; i < tagNames.length; i++) {
NodeList targetNodes = rootElement
.getElementsByTagName(tagNames[i]);
if (targetNodes.getLength() <= 0) {
System.out.println("no " + tagNames[i]
+ " tag in XML response...returning null");
} else {
List<String> valueList = new ArrayList<String>();
for (int j = 0; j < targetNodes.getLength(); j++) {
Node node = targetNodes.item(j);
valueList.add(node.getTextContent());
}
returnValues.put(tagNames[i], valueList);
}
}
} catch (Exception ex) {
System.out.println(ex);
}
return returnValues;
}
public static String signRequest(String request, String key) {
try {
Mac mac = Mac.getInstance("HmacSHA1");
SecretKeySpec keySpec = new SecretKeySpec(key.getBytes(),
"HmacSHA1");
mac.init(keySpec);
mac.update(request.getBytes());
byte[] encryptedBytes = mac.doFinal();
//System.out.println("HmacSHA1 hash: " + encryptedBytes);
return Base64.encodeBytes(encryptedBytes);
} catch (Exception ex) {
System.out.println("unable to sign request");
ex.printStackTrace();
}
return null;
}
} }

1739
utils/src/com/cloud/utils/encoding/Base64.java
View File

File diff suppressed because it is too large Load Diff