mirror of https://github.com/apache/cloudstack.git
109 lines
4.7 KiB
Java
109 lines
4.7 KiB
Java
/**
|
|
* Copyright (C) 2010 Cloud.com, Inc. All rights reserved.
|
|
*
|
|
* This software is licensed under the GNU General Public License v3 or later.
|
|
*
|
|
* It is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or any later version.
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
package com.cloud.async.executor;
|
|
|
|
import org.apache.log4j.Logger;
|
|
|
|
import com.cloud.api.BaseCmd;
|
|
import com.cloud.async.AsyncJobManager;
|
|
import com.cloud.async.AsyncJobResult;
|
|
import com.cloud.async.AsyncJobVO;
|
|
import com.cloud.async.BaseAsyncJobExecutor;
|
|
import com.cloud.exception.InternalErrorException;
|
|
import com.cloud.exception.InvalidParameterValueException;
|
|
import com.cloud.exception.NetworkRuleConflictException;
|
|
import com.cloud.exception.PermissionDeniedException;
|
|
import com.cloud.serializer.GsonHelper;
|
|
import com.cloud.server.ManagementServer;
|
|
import com.cloud.vm.DomainRouterVO;
|
|
import com.cloud.vm.UserVmVO;
|
|
import com.cloud.vm.dao.DomainRouterDao;
|
|
import com.cloud.vm.dao.UserVmDao;
|
|
import com.google.gson.Gson;
|
|
|
|
public class AssignSecurityGroupExecutor extends BaseAsyncJobExecutor {
|
|
public static final Logger s_logger = Logger.getLogger(AssignSecurityGroupExecutor.class.getName());
|
|
|
|
public boolean execute() {
|
|
Gson gson = GsonHelper.getBuilder().create();
|
|
AsyncJobManager asyncMgr = getAsyncJobMgr();
|
|
AsyncJobVO job = getJob();
|
|
ManagementServer managementServer = asyncMgr.getExecutorContext().getManagementServer();
|
|
SecurityGroupParam param = gson.fromJson(job.getCmdInfo(), SecurityGroupParam.class);
|
|
/*
|
|
if(getSyncSource() == null) {
|
|
DomainRouterVO router = getRouterSyncSource(param);
|
|
if(router == null) {
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED,
|
|
BaseCmd.NET_INVALID_PARAM_ERROR, "Unable to find router for user vm " + param.getInstanceId() + " when assigning security group");
|
|
} else {
|
|
asyncMgr.syncAsyncJobExecution(job.getId(), "Router", router.getId());
|
|
}
|
|
return true;
|
|
} else {
|
|
try {
|
|
managementServer.assignSecurityGroup(param.getUserId(), param.getSecurityGroupId(), param.getSecurityGroupIdList(), param.getPublicIp(), param.getInstanceId(), param.getEventId());
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_SUCCEEDED, 0,
|
|
"success");
|
|
} catch (PermissionDeniedException e) {
|
|
if(s_logger.isDebugEnabled())
|
|
s_logger.debug("Unable to assign security group : " + e.getMessage());
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED, BaseCmd.PARAM_ERROR,
|
|
e.getMessage());
|
|
} catch (NetworkRuleConflictException e) {
|
|
if(s_logger.isDebugEnabled())
|
|
s_logger.debug("Unable to assign security group : " + e.getMessage());
|
|
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED, BaseCmd.NET_CONFLICT_IPFW_RULE_ERROR,
|
|
e.getMessage());
|
|
} catch (InvalidParameterValueException e) {
|
|
if(s_logger.isDebugEnabled())
|
|
s_logger.debug("Unable to assign security group : " + e.getMessage());
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED, BaseCmd.PARAM_ERROR,
|
|
e.getMessage());
|
|
} catch (InternalErrorException e) {
|
|
if(s_logger.isDebugEnabled())
|
|
s_logger.debug("Unable to assign security group : " + e.getMessage());
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED, BaseCmd.INTERNAL_ERROR,
|
|
e.getMessage());
|
|
} catch(Exception e) {
|
|
s_logger.warn("Unable to assign security group : " + e.getMessage(), e);
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED, BaseCmd.INTERNAL_ERROR,
|
|
e.getMessage());
|
|
}
|
|
}
|
|
*/
|
|
return true;
|
|
}
|
|
|
|
private DomainRouterVO getRouterSyncSource(SecurityGroupParam param) {
|
|
UserVmDao userVmDao = getAsyncJobMgr().getExecutorContext().getVmDao();
|
|
DomainRouterDao routerDao = getAsyncJobMgr().getExecutorContext().getRouterDao();
|
|
|
|
UserVmVO userVm = userVmDao.findById(param.getInstanceId());
|
|
if(userVm == null)
|
|
return null;
|
|
|
|
if (userVm.getDomainRouterId() == null)
|
|
return null;
|
|
else
|
|
return routerDao.findById(userVm.getDomainRouterId());
|
|
}
|
|
}
|