org.bdware.irp.client.CdiIrpClient Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of doip-audit-tool Show documentation
Show all versions of doip-audit-tool Show documentation
doip audit tool developed by bdware
package org.bdware.irp.client;
import com.google.gson.Gson;
import com.google.gson.JsonObject;
import com.nimbusds.jose.jwk.JWK;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicNameValuePair;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.bdware.irp.crypto.CertUtils;
import org.bdware.irp.exception.IrpClientException;
import org.bdware.irp.irplib.core.IrpMessageSigner;
import org.bdware.irp.irplib.exception.IrpConnectException;
import org.bdware.irp.stateinfo.DoStateInfo;
import org.bdware.irp.stateinfo.StateInfoBase;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.*;
public class CdiIrpClient implements IrpClient {
JWK clientKeyPair;
String LHSUrl;
static Logger logger = LogManager.getLogger(CdiIrpClient.class);
String ClientID;
public CdiIrpClient(JWK jwk, String clientID, String LHSUrl) {
this.clientKeyPair = jwk;
this.ClientID = clientID;
this.LHSUrl = LHSUrl;
}
@Override
public StateInfoBase resolve(String handle) throws IrpClientException {
String response = resolveFromLHS(handle);
JsonObject resp = new Gson().fromJson(response, JsonObject.class);
StateInfoBase hr = new StateInfoBase();
if (resp == null || !resp.has("identifier")
|| (resp.has("status"))) {
throw new IrpClientException(response);
}
hr.identifier = resp.get("identifier").getAsString();
hr.handleValues = resp;
return hr;
}
@Override
public String register(StateInfoBase hr) throws IrpClientException {
Map map = new HashMap<>();
for (String keys : hr.handleValues.keySet()) {
map.put(keys, hr.handleValues.get(keys).getAsString());
}
map.remove("identifier");
map.put("action", "forceRegister");
String response = signAndSend(hr, map);
JsonObject resp = new Gson().fromJson(response, JsonObject.class);
if (resp == null || !resp.has("identifier")) {
throw new IrpClientException(response);
}
return resp.get("identifier").getAsString();
}
@Override
public String reRegister(StateInfoBase hr) throws IrpClientException {
if (hr.identifier == null) {
logger.warn("handle need to be set.");
}
Map map = new HashMap<>();
for (String keys : hr.handleValues.keySet()) {
map.put(keys, hr.handleValues.get(keys).getAsString());
}
map.put("identifier", hr.identifier);
map.put("action", "forceRegister");
String response = signAndSend(hr, map);
JsonObject resp = new Gson().fromJson(response, JsonObject.class);
if (resp == null || !resp.has("identifier")) {
throw new IrpClientException(response);
}
return resp.get("identifier").getAsString();
}
@Override
public String unRegister(String handle) {
Map map = new HashMap<>();
map.put("identifier", handle);
map.put("action", "forceDelete");
DoStateInfo hr = new DoStateInfo("", "");
hr.identifier = handle;
return signAndSend(hr, map);
}
@Override
public List batchRegister(StateInfoBase hr, int count) {
return null;
}
@Override
public void close() {
}
@Override
public void connect(String url) {
}
@Override
public void connect(String clientID, String LHSUrl, IrpMessageSigner signer) {
}
@Override
public void reconnect() throws IrpConnectException {
}
@Override
public boolean isConnected() {
return false;
}
private String signAndSend(StateInfoBase hr, Map map) {
map.put("dou", this.ClientID);
String reqMapStr = MapToString(map);
if (clientKeyPair != null) {
try {
String signature = CertUtils.Sign(reqMapStr.getBytes(), clientKeyPair);
map.put("signature", signature);
} catch (Exception e) {
e.printStackTrace();
}
}
return sendPost(map, LHSUrl + getSubUrl(hr));
}
public String MapToString(Map reqMap) {
Set keySet = reqMap.keySet();
keySet.remove("signature");
//升序
Set sortSet = new TreeSet<>();
sortSet.addAll(keySet);
String mapString = "";
for (String key : sortSet) {
if (reqMap.get(key) != null)
mapString += reqMap.get(key);
}
return mapString;
}
public String resolveFromLHS(String id) {
String result;
String resolveUrl = LHSUrl + "resolve?identifier=" + id;
HttpGet httpGet = new HttpGet(resolveUrl);
try {
CloseableHttpClient httpClient = HttpClients.createDefault();
HttpResponse response = httpClient.execute(httpGet);
InputStream in = response.getEntity().getContent();
BufferedReader br = new BufferedReader(new InputStreamReader(in, "utf-8"));
StringBuilder strber = new StringBuilder();
String line = null;
while ((line = br.readLine()) != null) {
strber.append(line + '\n');
}
br.close();
in.close();
result = strber.toString();
if (response.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
result = "服务器异常";
}
} catch (Exception e) {
System.out.println("请求异常");
throw new RuntimeException(e);
}
return result;
}
public String sendPost(Map reqMap, String url) {
String result;
HttpPost post = new HttpPost(url);
try {
CloseableHttpClient httpClient = HttpClients.createDefault();
List params = new ArrayList<>();
for (String keys : reqMap.keySet()) {
params.add(new BasicNameValuePair(keys, reqMap.get(keys)));
}
UrlEncodedFormEntity postForm = new UrlEncodedFormEntity(params, "utf-8");
post.setEntity(postForm);
HttpResponse response = httpClient.execute(post);
InputStream in = response.getEntity().getContent();
BufferedReader br = new BufferedReader(new InputStreamReader(in, "utf-8"));
StringBuilder strber = new StringBuilder();
String line = null;
while ((line = br.readLine()) != null) {
strber.append(line + '\n');
}
br.close();
in.close();
result = strber.toString();
if (response.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
result = "服务器异常";
}
} catch (Exception e) {
System.out.println("请求异常");
throw new RuntimeException(e);
} finally {
post.abort();
}
return result;
}
private String getSubUrl(StateInfoBase hr) {
String className = hr.getClass().getSimpleName();
String subUrl;
switch (className) {
case "DoHandleRecord":
subUrl = "do/";
break;
case "DoipServiceHandleRecord":
subUrl = "service/";
break;
case "UserHandleRecord":
subUrl = "dou/";
break;
default:
logger.error("use specific handle record class instead.");
subUrl = "";
break;
}
return subUrl;
}
}