RSA加密Socket传输文件、签名（二）

服务器端采用多线程方式并行处理传输请求以提高效率。

因为我想用JavaServiceWrapper将 程序作为windows服务运行，所以main方法中需接收一些参数，比如端口、密钥文件位置、文件存放路径等。

1.从main方法接受配置参数

for(String arg : args){ String argName = arg.substring(0,arg.indexOf(":")).trim(); String argValue = arg.substring(arg.indexOf(":")+1).trim(); if(argName.equals("rsa_private")){ privateKey = argValue; }else if(argName.equals("rsa_public")){ publicKey = argValue; }else if(argName.equals("port")){ port = Integer.parseInt(argValue); }else if(argName.equals("file_dir")){ file_dir = argValue; } }

2.初始化解密、签名类实例

RSASecurityCoder coder = new RSASecurityCoder(privateKey); RSASecuritySignature sign = new RSASecuritySignature(publicKey);

3.建立服务器

private static void runServer(String[] args){ long start = System.currentTimeMillis(); //获取配置参数 for(String arg : args){ String argName = arg.substring(0,arg.indexOf(":")).trim(); String argValue = arg.substring(arg.indexOf(":")+1).trim(); if(argName.equals("rsa_private")){ privateKey = argValue; }else if(argName.equals("rsa_public")){ publicKey = argValue; }else if(argName.equals("port")){ port = Integer.parseInt(argValue); }else if(argName.equals("file_dir")){ file_dir = argValue; } } //配置 coder = new RSASecurityCoder(privateKey); sign = new RSASecuritySignature(publicKey); //启动服务器 System.out.println("服务器启动中..."); try { ss = new ServerSocket(port); clientCount = 0; } catch (NumberFormatException e) { System.err.println("端口配置错误"); } catch (IOException e) { System.err.println("服务器在端口"+port+"启动失败"); } //文件存放目录 dir = new File(file_dir); if(!dir.exists()){ dir.mkdir(); } System.out.println("服务器已启动，端口： "+port); //计算启动时间 long end = System.currentTimeMillis(); System.out.println("共消耗 "+(end-start)+" ms."); //接收数据 while(true){ try { if(ss == null){ ss = new ServerSocket(port); clientCount = 0; } Socket socket = ss.accept(); clientCount += 1; System.out.println("有新的连接,当前总连接数："+clientCount); Thread th = new Thread(new ServerProcessor(socket)); th.start(); } catch (IOException e) { e.printStackTrace(); } } }

3.建立线程类ServerProcessor处理每个文件

static class ServerProcessor extends Thread{ private Socket socket; private InputStream in; private String filename; private long totalBytes; private File file; private FileOutputStream fos; public ServerProcessor(Socket socket){ this.socket = socket; } @Override public void run(){ try { in = socket.getInputStream(); DataInputStream dis = new DataInputStream(in); filename = dis.readUTF(); //读取文件名 totalBytes = dis.readLong(); //读取文件大小 file = new File(dir,filename); fos = new FileOutputStream(file); //FileOutputStream fos = new FileOutputStream(File.createTempFile(filename, null, dir)); //输出日志 //RSA加密以128 bytes位单位，一次最多加密117bytes. byte[] buf = new byte[128]; int available; while((available = in.read(buf)) != -1){ //读取签名 byte[] signature = buf; //读取数据 buf = new byte[128]; available = in.read(buf); byte[] availableBytes = null; if(available == buf.length){ availableBytes = buf; }else{ availableBytes = new byte[available]; for (int i = 0; i < available; i++) { availableBytes[i] = (Byte) buf[i]; } } //验证数据签名 boolean flag = sign.verifySignature(availableBytes, signature); //写入数据 if(flag){ //count += availableBytes.length; fos.write(coder.decrypt(availableBytes)); //System.out.println(""+count+" bytes received"); } } fos.close(); socket.shutdownInput(); clientCount -= 1; System.out.println("有连接断开，当前总连接数："+clientCount); System.out.println(new SimpleDateFormat("[yyyy-MM-DD HH:mm:ss ]").format(new Date()) +filename+" 接收完毕. 大小 "+(totalBytes/1024)+" kb，保存路径："+dir.getAbsolutePath()); } catch (NumberFormatException e) { System.err.println("端口配置错误"); e.printStackTrace(); } catch (FileNotFoundException e) { System.err.println("文件没找到，文件名："+filename+"，可能是因为读取socket数据失败"); e.printStackTrace(); } catch (IOException e) { System.err.println("文件读/写错误"); try { fos.close(); } catch (IOException e1) { e1.printStackTrace(); } file.delete(); e.printStackTrace(); } } }

main方法

public static void main(String[] args) { runServer(args); }