1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
|
package com.nis.nmsclient.thread.socket;
import java.io.File;
import java.util.List;
import java.util.concurrent.Callable;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocket;
import javax.net.ssl.SSLSocketFactory;
import net.sf.json.JSONArray;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import com.nis.nmsclient.common.Contants;
import com.nis.nmsclient.model.ReturnFilePO;
import com.nis.nmsclient.thread.task.TaskReqHandle;
import com.nis.nmsclient.thread.task.TaskResultOper;
import com.nis.nmsclient.util.FileUtil;
import com.nis.nmsclient.util.Utils;
/**
* 安全通讯的客户端
**/
public class SSLClient extends CommonSocket implements Callable<Object>{
Logger logger = Logger.getLogger(SSLClient.class);
private String name;
private String reqCmd;
private Object obj;
private String serverHost;
public SSLClient(String name, String reqCmd, Object obj) {
this.name = name;
this.reqCmd = reqCmd;
this.obj = obj;
this.serverHost = Contants.SOCKET_SERVER_HOST;
}
public SSLClient(String name, String reqCmd, Object obj, String serverHost) {
this.name = name;
this.reqCmd = reqCmd;
this.obj = obj;
this.serverHost = serverHost;
}
/**
* 初始化客户端Socket
**/
public void init() throws Exception {
SSLContext ctx = SSLCertOper.getSSLContext();
SSLSocketFactory ssf = ctx.getSocketFactory();
socket = (SSLSocket) ssf.createSocket(serverHost,
Contants.SOCKET_SERVER_PORT);
logger.debug("create socket success.");
//2014-1-23 hyx 如果建立socket成功，但是startHandshake握手失败，且未设置超时时间时，则会一直阻塞
socket.setSoTimeout(1000 * 60 * Contants.SOCKET_TIMEOUT_MINUTES);
((SSLSocket) socket).startHandshake();
logger.debug("handshake success.");
}
@Override
public Object call() throws Exception {
// Thread.currentThread().setName(name + ">>通信 " + serverHost);
Thread.currentThread().setName(name + ">>Communication " + serverHost);
Object object = null;
try {
init();
if(socket!=null){
out = socket.getOutputStream();
in = socket.getInputStream();
//设置超时时间
socket.setSoTimeout(1000 * 60 * Contants.SOCKET_TIMEOUT_MINUTES);
object = toDo();
}
}catch (Exception e) {
// object = Contants.COMMON_MSG_FAIL + Contants.COMMON_MSG_SEPRATOR + "异常";
object = Contants.COMMON_MSG_FAIL + Contants.COMMON_MSG_SEPRATOR + "anomaly";
logger.error("Communication anomaly：" + Utils.printExceptionStack(e));
} finally {
logger.debug("关闭通信");
close();
}
return object;
}
protected Object toDo() throws Exception {
logger.debug("发送通信请求：" + reqCmd);
//-- 无效操作处理
if(StringUtils.isEmpty(reqCmd)){
return null;
}
boolean flag = false;
String msg = null;
String result = null;
//-- 命令判断
// 与Server通信
if(reqCmd.equals(REQ_HAND_SHAKE)){
flag = this.sendMessageByChar(reqCmd);
logger.debug("握手状态：" + (result = this.receiveMessageByChar()));
}
// 通知Server准备升级
if(reqCmd.equals(REQ_SERVER_UPGRADE)){
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
}
// 获取本机标志UUID
if(reqCmd.equals(REQ_LOCAL_UUID)){
flag = this.sendMessageByChar(reqCmd);
msg = this.receiveMessageByChar();
flag = this.sendMessageByChar(CommonSocket.SUCCESS);
logger.info("本机标志UUID：" + msg);
}
//注释 by jinsj 2012-0531 修改为DC主动获取NC时间
// 获取服务器系统时间
if(reqCmd.equals(REQ_SERVER_SYSTEMDATE)){
flag = this.sendMessageByChar(reqCmd);
msg = this.receiveMessageByChar();
logger.debug("服务器系统时间：" + msg);
flag = this.sendMessageByChar(CommonSocket.SUCCESS);
}
// 发送本机的变更信息（操作系统类型和IP）
if(reqCmd.equals(REQ_LOCAL_CHANGE)){
// 发送请求
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
// 发送信息: UUID$@$操作系统类型$@$LocalIP
flag = this.sendMessageByChar((String)obj);
// 接收变更结果: 0/1 $@$ 信息
msg = this.receiveMessageByChar();
flag = this.sendMessageByChar(CommonSocket.SUCCESS);
return msg;
}
// 初始化配置
if(reqCmd.equals(REQ_INIT_CONFIG)){
flag = this.sendMessageByChar(reqCmd);
msg = this.receiveMessageByChar();//接收配置信息
flag = this.sendMessageByChar(CommonSocket.SUCCESS);
}
// 初始化任务
if(reqCmd.equals(REQ_INIT_TASK)){
// 发送请求
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
// 发送本机唯一标识
flag = this.sendMessageByChar(Contants.AGENT_HOST_UUID + "");
msg = this.receiveMessageByChar();
flag = this.sendMessageByChar(CommonSocket.SUCCESS);
if (msg != null && !"".equals(msg)) {
JSONArray jsonArr = JSONArray.fromObject(msg);
//这里处理的任务，原则上不应含有”1 文件推送“类型的任务
for (int i = 0; i < jsonArr.size(); i++) {
new TaskReqHandle().taskHandle(jsonArr.get(i).toString());
}
}
logger.debug("初始化任务完成--" + msg);
return null;
}
// 发送主动告警信息
if(reqCmd.equals(REQ_ALARM)){
if(obj!=null){
// 主动告警请求
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
// 发送主动告警信息内容
flag = this.sendMessageByChar((String)obj);
logger.debug("主动告警信息：" + (String)obj);
result = this.receiveMessageByChar();
}else{
logger.debug("主动告警信息为空");
}
}
// 发送任务结果
if(reqCmd.equals(REQ_TASK_RESULT)){
if(obj!=null){
//发送任务结果请求
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
//发送任务结果内容
flag = this.sendMessageByChar((String)obj);
result = this.receiveMessageByChar();
}else{
logger.warn("Task result information is empty");
}
}
// 批量上传监测数据【数据收集方式改为DC主动后，此通信废弃】
if(reqCmd.equals(REQ_UPLOAD_DATAS)){
if(obj!=null && obj instanceof Object[]) {
Object[] objArr = (Object[])obj;
if (objArr != null && objArr.length > 1 && objArr[0] != null
&& objArr[1] != null && objArr[1] instanceof List) {
//发送上传数据请求
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
//上传数据
this.sendFileByBath((String) objArr[0], (List<File>) objArr[1]);
result = this.receiveMessageByChar();
}else{
logger.warn("Uploading the contents of the monitored data object is incorrect");
}
}else{
logger.warn("Uploading monitoring data objects is empty");
}
}
// 任务执行的回传文件：单个文件发送，断点续传【数据收集方式改为DC主动后，此类废弃】
if(reqCmd.equals(REQ_TASK_RETURNFILE)){
if(obj!=null && obj instanceof ReturnFilePO){
ReturnFilePO rfPo = (ReturnFilePO)obj;
//发送回传文件请求
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
//发送回传文件任务信息
this.sendMessageByChar(TaskResultOper.getTaskResultMsg(rfPo
.getTaskId(), rfPo.getTaskType(), null, null, null, rfPo
.getStartTime(), rfPo.getEndTime(), rfPo.getIsLoop()));
result = this.receiveMessageByChar();
//发送回传文件文件名称
this.sendMessageByChar(rfPo.getReturnFileName());
result = this.receiveMessageByChar();
//发送回传文件
flag = this.bpSendFile(Contants.localTaskReturnPath + File.separator + rfPo.getReturnFileName());
result = this.receiveMessageByChar();
}else{
logger.warn("The return file object is empty");
}
}
// 发送压缩文件，断点续传
if(reqCmd.equals(REQ_BP_UPLOAD_FIFE)){
if(obj!=null && obj instanceof String[]) {
String[] strArr = (String[])obj;
if (strArr != null && strArr.length > 1){
//打包上传文件请求
flag = this.sendMessageByChar(reqCmd + ":" + strArr[0]);
result = this.receiveMessageByChar();
//发送打包文件名
File file = new File(strArr[1]);
flag = this.sendMessageByChar(file.getName());
result = this.receiveMessageByChar();
//上传打包文件
flag = this.bpSendFile(strArr[1]);
result = this.receiveMessageByChar();
//上传成功后删除或移动文件
if(flag && SUCCESS.equalsIgnoreCase(result) && file.exists()){
String dataType = strArr[0];
if(BP_TYPE_DETECT_DATA.equalsIgnoreCase(dataType)){
FileUtil.moveFile(file, Contants.localDataDonePath, true);
}else if(BP_TYPE_TASK_RESULT.equalsIgnoreCase(dataType)){
FileUtil.moveFile(file, Contants.localTaskDonePath, true);
}else if(BP_TYPE_TASK_RETURN.equalsIgnoreCase(dataType)){
FileUtil.moveFile(file, Contants.localTaskDonePath, true);
}
}
}
}
}
// 向DC发送NC端异常信息
if(reqCmd.equals(REQ_ERROR_INFO)){
if(obj!=null){
flag = this.sendMessageByChar(reqCmd);
result = this.receiveMessageByChar();
//发送异常内容
flag = this.sendMessageByChar((String)obj);
result = this.receiveMessageByChar();
}else{
logger.warn("Abnormal information is empty");
}
}
if (flag && (SUCCESS.equalsIgnoreCase(result) || msg!=null)) {
msg = Contants.COMMON_MSG_SUCCESS + Contants.COMMON_MSG_SEPRATOR + (msg!=null ? msg : "成功");
} else {
// msg = Contants.COMMON_MSG_FAIL + Contants.COMMON_MSG_SEPRATOR + "失败";
msg = Contants.COMMON_MSG_FAIL + Contants.COMMON_MSG_SEPRATOR + "failed";
}
logger.debug("SSLClient toDo()---" + msg);
logger.debug("发送通信请求结束：" + reqCmd);
// -- 命令判断
return msg;
}
}
|
