@SuppressWarnings("unchecked")
private int getRessiueBlocks(DlmsRequest req, DlmsRequest dr) {
int offset = 0;
List<Integer> ressiueBlocks = null;
if (req.containsKey(DlmsUpgradeAssisant.REISSUE_LIST)) {
ressiueBlocks = (List<Integer>) req.getAppendParam(DlmsUpgradeAssisant.REISSUE_LIST);
} else {
ressiueBlocks = new ArrayList<Integer>();
}
if (DlmsUpgradeAssisant.UPGRADE_03.equals(req.getOperator())
|| DlmsUpgradeAssisant.UPGRADE_RESSIUE.equals(req.getOperator())) {
return offset;
}
ASN1BitString bitString = req.getParams()[0].resultData.getBitString();
byte[] bitValue = bitString.getValue();
// 获得有效位数
int blockCount = (bitValue.length - 1) * 8 - bitValue[0];
getRessiueBlocks(bitValue, blockCount, ressiueBlocks, offset + 1);
if (req.containsKey(DlmsUpgradeAssisant.CURRENT_INDEX)) {
offset = (Integer) req.getAppendParam(DlmsUpgradeAssisant.CURRENT_INDEX);
} else {
offset = 0;
}
dr.addAppendParam(DlmsUpgradeAssisant.REISSUE_LIST, ressiueBlocks);
return offset;
}
/**
* 升级第4步 检查升级包是否传输完毕，首先客户端读取image_transferred_blocks_status（image
* transfer对象的属性3）获取各个imageBlock是否已接收的bitmap，如发现有遗漏的block，则补发相应的block，直至所有的block都被表计接收；当(image_transferred_blocks/8)>
* ImageBlockSize时，客户端需采用选择性读（选择性参数=entry_descriptor）来实现分组读取所有image_transferred_blocks的bitmap，如ImageBlockSize=100，image_transferred_blocks=1800，则客户端需要分三次来读取bitmap(第一次的entry_descriptor中From_entry=0，to_entry=99；第二次entry_descriptor中From_entry=100，to_entry=199；第三次entry_descriptor中From_entry=200，to_entry=00。Entry_descriptor的使用详见《DLMS/COSEM应用层标准》)；
*
* @param req
* @param dr
*/
private void upgradeStep_04(DlmsRequest req, DlmsRequest dr) {
setRequestParam(dr, new ObisDescription(18, "0.0.44.0.0.255", 3));
dr.setOpType(DLMS_OP_TYPE.OP_GET);
dr.removeAppendParam(DlmsUpgradeAssisant.CURRENT_RESSIUE_NUM);
int i_maxSize = (Integer) dr.getAppendParam(DlmsUpgradeAssisant.MAX_SIZE);
int blockCount = (Integer) dr.getAppendParam(DlmsUpgradeAssisant.BLOCK_COUNT);
if (blockCount / 8
> i_maxSize) { // 按照新的标准:如果(image_transferred_blocks/8)> ImageBlockSize,选择性读，否则，直接读
int fromIndex = getRessiueBlocks(req, dr);
int nextIndex = fromIndex + i_maxSize - 1; // 假如读3次的话[0,i_maxSize-1],[i_maxSize,199]
if (nextIndex > (blockCount / 8)) { // 最后一次读，读完之后，转移到第4步
nextIndex = 0;
dr.setOperator(DlmsUpgradeAssisant.UPGRADE_04);
} else { // 继续读
dr.setOperator(DlmsUpgradeAssisant.UPGRADE_READMAP);
dr.addAppendParam(DlmsUpgradeAssisant.CURRENT_INDEX, nextIndex + 1);
}
// 分多次读
SelectiveAccessDescriptor sad = new SelectiveAccessDescriptor();
sad.selectByIndex(fromIndex, nextIndex);
dr.getParams()[0].accessSelector = 2;
dr.getParams()[0].data.assignValue(sad.getParameter());
} else { // 直接读，读完之后转移到第4步
dr.setOperator(DlmsUpgradeAssisant.UPGRADE_04);
}
}
/**
* 软件升级断点续传初始化
*
* @param req
*/
public void upgradeReissueInit(DlmsRequest req) {
// 软件升级补发
req.setOperator(DlmsUpgradeAssisant.UPGRADE_03);
byte[] content = DlmsUpgradeAssisant.getInstance().getFtpFileContent(req);
Integer currentBlockNum = (Integer) req.getAppendParam(DlmsUpgradeAssisant.CURRENT_BLOCK_NUM);
Integer i_maxSize = (Integer) req.getAppendParam(DlmsUpgradeAssisant.MAX_SIZE);
byte[] currentBlock =
DlmsUpgradeHandler.getInstance().getCurUpgradeBlock(content, currentBlockNum, i_maxSize);
int blockCount = content.length / i_maxSize;
req.addAppendParam(DlmsUpgradeAssisant.BLOCK_COUNT, blockCount);
if (currentBlockNum == blockCount
|| content.length % i_maxSize == 0 && currentBlockNum - 1 == blockCount) {
req.addAppendParam(DlmsUpgradeAssisant.IS_TRANSFER_FINISHED, true);
}
DlmsUpgradeHandler.getInstance().blockTransferRequest(req, currentBlockNum, currentBlock);
req.addAppendParam(DlmsUpgradeAssisant.UPGRADE_CONTENT, content);
req.setOpType(DLMS_OP_TYPE.OP_ACTION);
}
/**
* 补发漏点块
*
* @param req
* @param dr
*/
@SuppressWarnings("unchecked")
private void reissueBlock(DlmsRequest dr) {
dr.setOperator(DlmsUpgradeAssisant.UPGRADE_RESSIUE);
dr.setOpType(DLMS_OP_TYPE.OP_ACTION);
int i_maxSize = (Integer) dr.getAppendParam(DlmsUpgradeAssisant.MAX_SIZE);
byte[] content = (byte[]) dr.getAppendParam(DlmsUpgradeAssisant.UPGRADE_CONTENT);
// 获得补发列表
List<Integer> reissueList = (List<Integer>) dr.getAppendParam(DlmsUpgradeAssisant.REISSUE_LIST);
// 从列表头开始发送
Integer reissueBlockNum = reissueList.remove(0);
dr.addAppendParam(DlmsUpgradeAssisant.CURRENT_RESSIUE_NUM, reissueBlockNum);
dr.addAppendParam(DlmsUpgradeAssisant.REISSUE_LIST, reissueList);
// 如果是最后一个
byte[] reissueBlock = getCurUpgradeBlock(content, reissueBlockNum, i_maxSize);
if (reissueList.size() == 0) {
dr.addAppendParam(DlmsUpgradeAssisant.IS_TRANSFER_FINISHED, true);
}
blockTransferRequest(dr, reissueBlockNum, reissueBlock);
}
/**
* 升级过程传输数据块 判断是否传输完毕，如果传输完毕直接进入第4步,否则继续传输
*
* @param req
* @param dr
* @return
*/
private boolean blockTransfering(DlmsRequest req, DlmsRequest dr) {
if (req.getParams()[0].resultCode != 0) return false;
if (dr.containsKey(DlmsUpgradeAssisant.IS_TRANSFER_FINISHED)) { // 发送完毕了,检查升级包是否完毕
boolean isTransferFinished =
(Boolean) dr.getAppendParam(DlmsUpgradeAssisant.IS_TRANSFER_FINISHED);
if (isTransferFinished) {
upgradeStep_04(req, dr);
} else {
reissueBlock(dr);
}
} else {
upgradeStep_03(req, dr);
}
return true;
}
@SuppressWarnings("unchecked")
private boolean checkBitMap(DlmsRequest req, DlmsRequest dr) {
getRessiueBlocks(req, dr);
List<Integer> ressiueBlocks =
(List<Integer>) dr.getAppendParam(DlmsUpgradeAssisant.REISSUE_LIST);
if (ressiueBlocks.size() != 0) { // 进入补发
dr.addAppendParam(DlmsUpgradeAssisant.REISSUE_LIST, ressiueBlocks);
dr.addAppendParam(DlmsUpgradeAssisant.IS_TRANSFER_FINISHED, false);
reissueBlock(dr);
return false;
}
return true;
}
public byte[] fileHeadConstruct(DlmsRequest req) {
String strFileHead = (String) req.getAppendParam(DlmsUpgradeAssisant.FILE_HEAD);
if (tracer.isEnabled()) tracer.trace("MeterId:" + req.getMeterId() + "FileHead:" + strFileHead);
byte[] fileHead = HexDump.toArray(strFileHead);
if (fileHead[1] == 1) { // 加密
ByteBuffer iv = ByteBuffer.allocate(12);
DlmsContext context = contextManager.getContext(req.getMeterId());
byte[] verfication = new byte[16];
try {
System.arraycopy(fileHead, 23, verfication, 0, 16);
byte[] encryptVerfiy = AESGcm128.encrypt(context.encryptKey, iv.array(), verfication, null);
System.arraycopy(encryptVerfiy, 0, fileHead, 23, 16);
} catch (InvalidCipherTextException e) {
log.error(StringUtil.getExceptionDetailInfo(e));
}
}
return fileHead;
}
/**
* DLMS升级第三步 传输升级包文件，客户端按照Step 1中获取的imageBlockSize分帧传送升级包文件，升级包的传输通过action操作image
* transfer对象的方法2完成；
*
* @param req
* @param dr
*/
private boolean upgradeStep_03(DlmsRequest req, DlmsRequest dr) {
if (req.getParams()[0].resultCode != 0) return false;
dr.setOperator(DlmsUpgradeAssisant.UPGRADE_03);
dr.setOpType(DLMS_OP_TYPE.OP_ACTION);
int i_maxSize = (Integer) dr.getAppendParam(DlmsUpgradeAssisant.MAX_SIZE);
byte[] content = (byte[]) dr.getAppendParam(DlmsUpgradeAssisant.UPGRADE_CONTENT);
int currentBlockNum = 0;
if (!dr.containsKey(DlmsUpgradeAssisant.CURRENT_BLOCK_NUM)) currentBlockNum = 0;
else currentBlockNum = (Integer) dr.getAppendParam(DlmsUpgradeAssisant.CURRENT_BLOCK_NUM) + 1;
dr.addAppendParam(DlmsUpgradeAssisant.CURRENT_BLOCK_NUM, currentBlockNum);
if (tracer.isEnabled()) {
tracer.trace(
"MeterId:"
+ dr.getMeterId()
+ ",BlockCount:"
+ (Integer) dr.getAppendParam(DlmsUpgradeAssisant.BLOCK_COUNT)
+ ",CurrentBlockNum:"
+ currentBlockNum);
}
if (currentBlockNum == (Integer) dr.getAppendParam(DlmsUpgradeAssisant.BLOCK_COUNT)
|| (content.length % i_maxSize == 0
&& currentBlockNum + 1
== (Integer) dr.getAppendParam(DlmsUpgradeAssisant.BLOCK_COUNT))) {
// 当前发送是最后一帧
dr.addAppendParam(DlmsUpgradeAssisant.IS_TRANSFER_FINISHED, true);
}
byte[] currentSendBlock = getCurUpgradeBlock(content, currentBlockNum, i_maxSize);
String logicAddr = req.getMeterId();
int tn = 0;
if (req.getRelayParam() != null) {
logicAddr = req.getRelayParam().getDcLogicalAddress();
tn = req.getRelayParam().getMeasurePoint();
}
if (tracer.isEnabled())
tracer.trace(
"meterId:"
+ logicAddr
+ ",tn:"
+ tn
+ ",currentBlockNum:"
+ currentBlockNum
+ ",blockContent:"
+ HexDump.toHex(currentSendBlock));
blockTransferRequest(dr, currentBlockNum, currentSendBlock);
return true;
}
/**
* step7 写入执行时间 新版本软件生效，通过action操作image transfer对象的方法4激活之前步骤中传输的升级包文件。 软件升级时间的格式为array[1] {
* structure include { time; octet-string[4] date; octet-string[5] } }
*
* @param req
* @param dr
*/
private void upgradeStep_07(DlmsRequest req, DlmsRequest dr) {
// 写执行时间
String strEffcDate = (String) dr.getAppendParam(DlmsUpgradeAssisant.EFFECTIVE_TIME);
String[] dateTime = strEffcDate.split(" ");
DlmsData[] effcDate = new DlmsData[2];
effcDate[0] = new DlmsData();
effcDate[0].setDlmsTime(dateTime[1]);
effcDate[1] = new DlmsData();
effcDate[1].setDlmsDate(dateTime[0]);
DlmsData[] array = new DlmsData[] {new DlmsData()};
ASN1SequenceOf struct = new ASN1SequenceOf(effcDate);
array[0].setStructure(struct);
DlmsData data = new DlmsData();
try {
data.setArray(array);
} catch (IOException e) {
log.error(StringUtil.getExceptionDetailInfo(e));
}
setRequestParam(dr, new ObisDescription(22, "0.0.15.0.2.255", 4));
dr.getParams()[0].data = data;
dr.setOpType(DLMS_OP_TYPE.OP_SET);
dr.setOperator(DlmsUpgradeAssisant.UPGRADE_07);
}
public void handleUpgrade(DlmsEventProcessor processor, DlmsRequest req, DlmsContext context) {
try {
boolean isFinished = false; // 是否升级结束
boolean isSuccess = true; // 命令是否成功
DlmsRequest dr = new DlmsRequest();
dr.setRelayParam(req.getRelayParam());
dr.setMeterId(context.meterId);
dr.addAllAppendParmas(req.getAllParam());
dr.setDestAddr(req.getDestAddr());
if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_00)) {
isSuccess = updateStep_01_01(req, dr);
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_01_01)) {
isSuccess = upgradeStep_01(req, dr);
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_01)) {
isSuccess = upgradeStep_02(req, dr);
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_02)) {
isSuccess = upgradeStep_03(req, dr);
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_03)
|| req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_RESSIUE)) {
isSuccess = blockTransfering(req, dr);
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_READMAP)) {
upgradeStep_04(req, dr);
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_04)) {
if (checkBitMap(req, dr)) { // 如果没有漏点，进入第五步
upgradeStep_05(req, dr);
}
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_05)) { // 检测验证是否成功
if (req.getParams()[0].resultCode == 2) { // 如果返回的是temporary-failure,休息一会再读
try {
Thread.sleep(sleepTimeStay06Step * 1000);
} catch (InterruptedException e) {
}
}
readTransferStatus(dr); // 这里无论返回什么，都去读升级状态
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_READ_STATUS)) {
int verfiyFlg = req.getParams()[0].resultData.getEnum();
if (verfiyFlg == 3) {
upgradeStep_06(req, dr);
} else if (verfiyFlg == 2) {
try {
Thread.sleep(sleepTimeStay06Step * 1000);
} catch (InterruptedException e) {
}
readTransferStatus(dr);
} else { // 除了上面两种情况，都是失败
isSuccess = false;
}
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_06)) {
// 读到的数据与主站下发做比较
// 如果成功，执行第7步
{
upgradeStep_07(req, dr);
}
} else if (req.getOperator().equals(DlmsUpgradeAssisant.UPGRADE_07)) {
// 如果结束 设置finished=true
if (req.getParams()[0].resultCode == 0) isFinished = true;
else {
isSuccess = false;
}
}
// 查询待发送列表中，是否有要升级的请求，如果有，则当前请求判定为失败
if (context.webReqList.size() > 0) {
DlmsEvent evt = (DlmsEvent) context.webReqList.get(0);
DlmsRequest request = (DlmsRequest) evt.getRequest();
if (request.getOperator() != null && request.getOperator().contains("UPGRADE")) {
isSuccess = false;
log.error(
"meterid="
+ context.meterId
+ ".New upgrade request is comming,current upgrade set fail");
}
}
if (!isSuccess) {
DlmsUpgradeAssisant.getInstance().updateUpgradeInfo(req, UpgradeInfo.FAIL); // 设置为失败
DlmsUpgradeAssisant.getInstance().updateUpgradeStatus(req, "00");
if (tracer.isEnabled())
tracer.trace(
"Upgrade Failed. MeterId=" + req.getMeterId() + ",Operator=" + req.getOperator());
return;
}
if (!isFinished) {
DlmsUpgradeAssisant.getInstance().onUpgradeSuccess(dr);
processor.postWebRequest(dr, null);
} else {
// 收尾阶段
req.addAppendParam(
DlmsUpgradeAssisant.CURRENT_BLOCK_NUM,
(Integer) req.getAppendParam(DlmsUpgradeAssisant.BLOCK_COUNT));
DlmsUpgradeAssisant.getInstance().updateUpgradeInfo(req, UpgradeInfo.SUCCESS); // 设置为成功
DlmsUpgradeAssisant.getInstance().updateUpgradeStatus(req, "02");
}
} catch (Exception e) {
log.error(StringUtil.getExceptionDetailInfo(e));
DlmsUpgradeAssisant.getInstance().updateUpgradeInfo(req, UpgradeInfo.FAIL); // 设置为失败
DlmsUpgradeAssisant.getInstance().updateUpgradeStatus(req, "00");
}
}
/**
* 收到请求，升级处理
*
* @param req
* @return
*/
public boolean upgradeProcesser(DlmsRequest req) {
String logicAddress = "";
int measurePoint = 0;
if (req.getRelayParam() != null) {
RelayParam relayParam = req.getRelayParam();
logicAddress = relayParam.getDcLogicalAddress();
measurePoint = relayParam.getMeasurePoint();
} else {
logicAddress = req.getMeterId();
measurePoint = 0;
}
try {
List<UpgradeInfo> infos =
dbService.getUpgradeInfo(
new Long((String) req.getAppendParam(DlmsUpgradeAssisant.UPGRADE_ID)));
if (infos.size() != 1) {
log.error(
"Upgrade Request Fail:Info.size="
+ infos.size()
+ ".MeterId="
+ logicAddress
+ ".Tn="
+ measurePoint);
if (tracer.isEnabled())
tracer.trace(
"Upgrade Request Fail:Info.size="
+ infos.size()
+ ".MeterId="
+ logicAddress
+ ".Tn="
+ measurePoint);
}
req.addAppendParam(DlmsUpgradeAssisant.UPGRADE_ID, infos.get(0).getSoftUpgradeID());
UpgradeInfo info = infos.get(0);
int status = info.getStatus();
DlmsUpgradeAssisant.getInstance().addRequestAppendParams(info, req);
String fileHead = (String) req.getAppendParam(DlmsUpgradeAssisant.FILE_HEAD);
String fileType = fileHead.substring(0, 2);
// 如果是给模块升级，设置dstAddr为 0x0070
// 根据地址文件头信息获得升级状态
// req.setDestAddr(0x0070);
if ("02".equals(fileType)) { // 判断是否是对模块升级
req.setDestAddr(DlmsMessage.DstAddrToModule);
}
boolean isRessiueSuccess = true;
switch (status) {
case UpgradeInfo.WAIT_UPGRADE:
upgradeEnable(req);
break;
case UpgradeInfo.WAIT_UPGARDEINIT:
upgradeInit(req, req, (Integer) req.getAppendParam(DlmsUpgradeAssisant.MAX_SIZE));
break;
case UpgradeInfo.UPGRADE_PAUSE:
upgradeReissueInit(req);
break;
case UpgradeInfo.RESSIUE_PAUSE:
req.addAppendParam(DlmsUpgradeAssisant.IS_TRANSFER_FINISHED, false);
reissueBlock(req);
break;
case UpgradeInfo.CHECK_MAP_FAIL:
req.setOperator(DlmsUpgradeAssisant.UPGRADE_RESSIUE);
upgradeStep_04(req, req);
break;
case UpgradeInfo.READ_STATUS_FAIL:
readTransferStatus(req);
break;
case UpgradeInfo.VERFIY_FILE_FAIL:
upgradeStep_05(req, req);
break;
case UpgradeInfo.CHECK_FILE_FAIL:
upgradeStep_06(req, req);
break;
case UpgradeInfo.SET_EFFECTTIME_FAIL:
upgradeStep_07(req, req);
break;
default:
isRessiueSuccess = false;
log.error(
"Upgrade File.Status:"
+ status
+ " .MeterId="
+ logicAddress
+ " .Tn="
+ measurePoint);
break;
}
if (isRessiueSuccess) {
DlmsUpgradeAssisant.getInstance().updateUpgradeStatus(req, "01");
return true;
}
return false;
} catch (Exception e) {
// 设置升级失败
log.error("update stoped.", e);
DlmsUpgradeAssisant.getInstance().updateUpgradeInfo(req, UpgradeInfo.FAIL); // 设置为失败
DlmsUpgradeAssisant.getInstance().updateUpgradeStatus(req, "00");
return false;
}
}