SOCKET實現ICAP協議通訊

• 首先基于springboot搭一個socket即時通訊
• • 注意事項：

首先基于springboot搭一個socket即時通訊

@PostMapping("/upload")public String upload(@RequestParam("file") MultipartFile file) {if (file.isEmpty()) {return "上傳失敗，請選擇文件";}String fileName = file.getOriginalFilename();try {byte[] bytes = file.getBytes();ICAPClientUtil socket = new ICAPClientUtil("*.*.*.*", 1111, "icap服務端", fileName, 3000, 3000, InetAddress.getLocalHost().toString(), LOGGER);String icap = socket.scanFile(bytes, fileName);int i = socket.resultCode(icap);return icap + "\n" + "====resultCode==" + i+"服務器端口==="+port;} catch (Exception e) {return e.toString();}}

ICAPClientUtil工具類

private String serverAddress;private int serverPort;private String serviceName;private String profileName;private int conTimeOut;private int soTimeOut;private java.io.BufferedReader input = null;private java.io.DataOutputStream output = null;private java.net.Socket socket = null;private String[] results;private String clientIP;private Logger logger;public ICAPClientUtil(String serverAddress, int serverPort, String serviceName,String profileName, int conTimeOut, int soTimeOut, String clientIP, Logger log) {this.serverAddress = serverAddress;this.serverPort = serverPort;this.serviceName = serviceName;this.profileName = profileName;this.conTimeOut = conTimeOut;this.soTimeOut = soTimeOut;this.clientIP = clientIP;this.logger = log;}private int connect(String serverAddress, int serverPort) {int retValue = 0;if (this.socket == null) {try {InetAddress ip_address = InetAddress.getByName(serverAddress);this.socket = new Socket();socket.connect(new InetSocketAddress(ip_address, serverPort), conTimeOut);this.socket.setSoTimeout(soTimeOut);this.output = new DataOutputStream(this.socket.getOutputStream());this.input = new java.io.BufferedReader(new InputStreamReader(this.socket.getInputStream()));} catch (Exception e) {retValue = -1;logger.error("ICAP:Error:connect(" + serverAddress + ":" + serverPort + "):" + e);}}return retValue;}/*** disconnect socket**/private void disconnect() {try {this.output.flush();this.output.close();this.input.close();this.socket.close();this.socket = null;} catch (Exception e) {logger.error("ICAP:Error:disconnect():" + e);}}/*** receive response from ICAP server*/private String receive() {String value = " ";String returnString = new String();this.results = new String[10240];int array_location = 0;try {while ((value = this.input.readLine()) != null && value.length() != 0) {logger.info("ICAP:stdOut:" + value);returnString = returnString + value + "\n";this.results[array_location] = value;array_location++;}} catch (Exception e) {logger.error("ICAP:Error:receive():" + e);returnString = "Error:receive():" + e;}return returnString;}/*** enter here if you only want integer result codes*/public int resultCode(String returnString) {int retValue = 0;//if (returnString.startsWith("Error") || returnString.startsWith("Undefined") || "".equals(returnString)) {// retValue = -1;//}if (returnString.startsWith("Infected") || returnString.startsWith("Blocked")) {if (returnString.contains(":")) {try {String[] responses = returnString.split(":", 3);retValue = Integer.parseInt(responses[1]);} catch (Exception e) {logger.error("ICAP:Error:scanfiletext:parsereturnString:" + returnString + ":" + e);retValue = 0;}}}if (returnString.startsWith("Clean")) {retValue = 0;}return retValue;}/*** enter here if you want text results* 入口**/public String scanFile(byte[] downloadByte, String fileMd5) {int retValue = 0;long fileLen = downloadByte.length;String serverResponse = "";if (fileLen > 1 && (fileLen > Long.MAX_VALUE)) {retValue = -1;serverResponse = "Error:scanfiletext:fileio:too_large:(" + fileLen + "):\"" + fileMd5 + "\"";}if (retValue < 0) {logger.error("ICAP:" + serverResponse);return serverResponse;}retValue = connect(this.serverAddress, this.serverPort);ICAPResponse icapResponse = new ICAPResponse();java.text.DateFormat dateFormat = new java.text.SimpleDateFormat("yyyyMMddHHmmss");// where file is sent to be scanned.while (retValue == 0) {int req_header = 0;int res_header;int res_body;String requestHeader = "GET http://" + this.clientIP + "/" + dateFormat.format(new java.util.Date()) + "/" + fileMd5 + " HTTP/1.1\r\n" + "Host: " + this.clientIP + "\r\n\r\n";String responseHeader = "HTTP/1.1 200 OK\r\n" + "Transfer-Encoding: chunked\r\n\r\n";res_header = requestHeader.getBytes().length;res_body = res_header + (responseHeader.length());String profile = "";if (!"".equals(this.profileName)) {profile = "?profile=" + this.profileName;}String icapRequest = "RESPMOD icap://" + this.serverAddress + ":" + this.serverPort + "/" + this.serviceName + profile + " ICAP/1.0\r\n" + "Allow: 204\r\n"// added connection: close+ "Connection: close\r\n" + "Encapsulated: req-hdr=" + req_header + " res-hdr=" + res_header + " res-body=" + res_body + "\r\n" + "Host: " + this.serverAddress + "\r\n"// + "Preview: " + Long.toString(this.previewBytes) + "\r\n"+ "User-Agent: JavaICAPClient\r\n" + "X-Client-IP: " + this.clientIP + "\r\n\r\n";logger.info("監聽請求值===" + icapRequest);retValue = send(icapRequest);logger.warn("寫入返回值=====" + retValue);if (retValue == -1) {break;}retValue = send(requestHeader);if (retValue == -1) {break;}retValue = send(responseHeader);if (retValue == -1) {break;}try {retValue = send(Long.toHexString(fileLen) + "\r\n");this.output.write(downloadByte);retValue = send("\r\n");retValue = send("0\r\n\r\n");this.output.flush();serverResponse = icapResponse.request(receive());} catch (Exception e) {logger.error("ICAP:Error:filesend:" + e);return "false";}disconnect();break;}return serverResponse;}private int send(String str) {int retValue = 0;try {this.output.write(str.getBytes());this.output.flush();} catch (Exception e) {logger.error("ICAP:Error:send(" + str + "):" + e);retValue = -1;}return retValue;}

配置屬性類

public boolean continue_check(String server_response) {boolean retValue = true;if (server_response.length() != 12) {server_response = server_response.substring(0, 12);}if (server_response.startsWith("ICAP/1.0 100")) {retValue = true;} else {retValue = false;}return retValue;}public String request(String serverResponse) {String returnString = "";String status = "Undefined";String blockResult = "";String virusName = ";";String[] lines = serverResponse.split("\n");for (int i = 0; i < lines.length; i++) {if (lines[i].startsWith("ICAP/1.0 204")) {status = "Clean:";blockResult = "0";virusName = "No malware detected;";break;}if (lines[i].startsWith("ICAP/1.0 200")) {status = "Blocked:";blockResult = "200";continue;}if (lines[i].startsWith("X-Virus-Name:")) {status = "Infected:";if ((lines[i].substring(14)).startsWith(virusName.substring(0, virusName.length() - 1))) {virusName = lines[i].substring(14);} else {virusName += lines[i].substring(14);}virusName += ";";continue;}if (lines[i].startsWith("X-WWBlockResult:")) {blockResult = lines[i].substring(17);switch (Integer.parseInt(blockResult)) {case 90:virusName += "Policy:Unwanted Unsigned Content;";break;case 81:virusName += "Policy:Authorization needed;";break;case 45:virusName += "Policy:Macros unwanted;";break;case 44:virusName += "Policy:Office document unreadable;";break;case 43:virusName += "Policy:Encrypted document unwanted;";break;case 42:virusName += "Policy:ActiveX unwanted;";break;case 41:virusName += "Policy:Document Inspector;";break;case 40:virusName += "Policy:Text Categorization unwanted;";break;case 34:virusName += "Policy:Encrypted archive unwanted;";break;case 33:virusName += "Policy:Archive recursion level exceeded;";break;case 32:virusName += "Policy:Mailbomb unwanted;";break;case 31:virusName += "Policy:Corrupted archive unwanted;";break;case 30:virusName += "Policy:Multipart archive unwanted;";break;case 23:virusName += "Policy:Generic Body Filter;";break;case 22:virusName += "Policy:Media type blacklisted;";break;case 21:virusName += "Policy:Media type mismatch;";break;default:status = "Infected:";break;}continue;}if (lines[i].startsWith("ICAP/1.0 ")) {status = "Error:" + lines[i];}}if (virusName.endsWith(";"))virusName = virusName.substring(0, virusName.length() - 1);virusName.trim();if ("".equals(virusName))virusName = "Unknown";returnString = status + blockResult + ":\"" + virusName + "\"";return returnString;}

注意事項：

1.icap有自己的通訊協議格式，在百度文檔里面有詳細介紹

按照這個模板來寫也是沒什么問題的。存個檔，萬一以后又遇到怎么辦。

總結

以上是生活随笔為你收集整理的springboot配合socket实现icap协议通讯的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。