Skip to content

T
teku
Commit 249eb9c2 authored by luxq's avatar luxq
Browse files
Options

fix error

parent 76fec17e
Hide whitespace changes
Inline Side-by-side
Showing with 64 additions and 208 deletions
attacker/client/src/main/java/tech/pegasys/teku/attacker/AttackService.java
View file @ 249eb9c2
package tech.pegasys.teku.attacker;
import com.googlecode.jsonrpc4j.JsonRpcHttpClient;
import tech.pegasys.teku.infrastructure.async.SafeFuture;
public class AttackService {
private static final String BLOCK_MODULE = "block";
private static final String ATTEST_MODULE = "attest";
/**
* Get new parent root for block.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param parentRoot The parent root
* @return A future that completes with the response
*/
private final JsonRpcHttpClient jsonRpcClient;
public AttackService() {
this.jsonRpcClient = JsonRpcClientUtil.getJsonRpcClient();
}
private SafeFuture<AttackerResponse> invokeRpc(String method, Object[] params) {
try {
AttackerResponse response = jsonRpcClient.invoke(method, params, AttackerResponse.class);
return SafeFuture.completedFuture(response);
} catch (Exception e) {
return SafeFuture.failedFuture(e);
} catch (Throwable e) {
throw new RuntimeException(e);
}
}
public SafeFuture<AttackerResponse> blockGetNewParentRoot(long slot, String pubkey, String parentRoot) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_getNewParentRoot",
new Object[] {slot,pubkey,parentRoot},
AttackerResponse.class);
}
/**
* Delay for receive block.
*
* @param slot The slot number
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_getNewParentRoot", new Object[]{slot, pubkey, parentRoot});
}
public SafeFuture<AttackerResponse> delayForReceiveBlock(long slot) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_delayForReceiveBlock",
new Object[] {slot},
AttackerResponse.class);
}
/**
* Before block broadcast.
*
* @param slot The slot number
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_delayForReceiveBlock", new Object[]{slot});
}
public SafeFuture<AttackerResponse> blockBeforeBroadcast(long slot) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_beforeBroadCast",
new Object[] {slot},
AttackerResponse.class);
}
/**
* After block broadcast.
*
* @param slot The slot number
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_beforeBroadCast", new Object[]{slot});
}
public SafeFuture<AttackerResponse> blockAfterBroadcast(long slot) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_afterBroadCast",
new Object[] {slot},
AttackerResponse.class);
}
/**
* Before sign block.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param blockDataBase64 The block data in base64 encoding
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_afterBroadCast", new Object[]{slot});
}
public SafeFuture<AttackerResponse> blockBeforeSign(long slot, String pubkey, String blockDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_beforeSign",
AttackerResponse.class,
new Object[] {slot,pubkey,blockDataBase64},
AttackerResponse.class);
}
/**
* After sign block.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param signedBlockDataBase64 The signed block data in base64 encoding
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_beforeSign", new Object[]{slot, pubkey, blockDataBase64});
}
public SafeFuture<AttackerResponse> blockAfterSign(long slot, String pubkey, String signedBlockDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_afterSign",
new Object[] {slot,pubkey,signedBlockDataBase64},
AttackerResponse.class);
}
/**
* Before propose block.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param signedBlockDataBase64 The signed block data in base64 encoding
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_afterSign", new Object[]{slot, pubkey, signedBlockDataBase64});
}
public SafeFuture<AttackerResponse> blockBeforePropose(long slot, String pubkey, String signedBlockDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_beforePropose",
new Object[] {slot,pubkey,signedBlockDataBase64},
AttackerResponse.class);
}
/**
* After propose block.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param signedBlockDataBase64 The signed block data in base64 encoding
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_beforePropose", new Object[]{slot, pubkey, signedBlockDataBase64});
}
public SafeFuture<AttackerResponse> blockAfterPropose(long slot, String pubkey, String signedBlockDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
BLOCK_MODULE + "_afterPropose",
new Object[] {slot,pubkey,signedBlockDataBase64},
AttackerResponse.class);
}
/**
* Before attestation broadcast.
*
* @param slot The slot number
* @return A future that completes with the response
*/
return invokeRpc(BLOCK_MODULE + "_afterPropose", new Object[]{slot, pubkey, signedBlockDataBase64});
}
public SafeFuture<AttackerResponse> attestBeforeBroadcast(long slot) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
ATTEST_MODULE + "_beforeBroadCast",
new Object[] {slot},
AttackerResponse.class);
}
/**
* After attestation broadcast.
*
* @param slot The slot number
* @return A future that completes with the response
*/
return invokeRpc(ATTEST_MODULE + "_beforeBroadCast", new Object[]{slot});
}
public SafeFuture<AttackerResponse> attestAfterBroadcast(long slot) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
ATTEST_MODULE + "_afterBroadCast",
new Object[] {slot},
AttackerResponse.class);
}
/**
* Before signing attestation.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param attestDataBase64 The attestation data in base64 encoding
* @return A future that completes with the response
*/
public SafeFuture<AttackerResponse> attestBeforeSign(
long slot,
String pubkey,
String attestDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
ATTEST_MODULE + "_beforeSign",
new Object[] {slot,pubkey,attestDataBase64},
AttackerResponse.class);
}
/**
* After signing attestation.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param signedAttestDataBase64 The signed attestation data in base64 encoding
* @return A future that completes with the response
*/
public SafeFuture<AttackerResponse> attestAfterSign(
long slot,
String pubkey,
String signedAttestDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
ATTEST_MODULE + "_afterSign",
new Object[] {slot,pubkey,signedAttestDataBase64},
AttackerResponse.class);
}
/**
* Before proposing attestation.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param signedAttestDataBase64 The signed attestation data in base64 encoding
* @return A future that completes with the response
*/
public SafeFuture<AttackerResponse> attestBeforePropose(
long slot,
String pubkey,
String signedAttestDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
ATTEST_MODULE + "_beforePropose",
new Object[] {slot,pubkey,signedAttestDataBase64},
AttackerResponse.class);
}
/**
* After proposing attestation.
*
* @param slot The slot number
* @param pubkey The validator public key
* @param signedAttestDataBase64 The signed attestation data in base64 encoding
* @return A future that completes with the response
*/
public SafeFuture<AttackerResponse> attestAfterPropose(
long slot,
String pubkey,
String signedAttestDataBase64) {
return JsonRpcClientUtil.getJsonRpcClient().invoke(
ATTEST_MODULE + "_afterPropose",
new Object[] {slot,pubkey,signedAttestDataBase64},
AttackerResponse.class);
}
}
return invokeRpc(ATTEST_MODULE + "_afterBroadCast", new Object[]{slot});
}
public SafeFuture<AttackerResponse> attestBeforeSign(long slot, String pubkey, String attestDataBase64) {
return invokeRpc(ATTEST_MODULE + "_beforeSign", new Object[]{slot, pubkey, attestDataBase64});
}
public SafeFuture<AttackerResponse> attestAfterSign(long slot, String pubkey, String signedAttestDataBase64) {
return invokeRpc(ATTEST_MODULE + "_afterSign", new Object[]{slot, pubkey, signedAttestDataBase64});
}
public SafeFuture<AttackerResponse> attestBeforePropose(long slot, String pubkey, String signedAttestDataBase64) {
return invokeRpc(ATTEST_MODULE + "_beforePropose", new Object[]{slot, pubkey, signedAttestDataBase64});
}
public SafeFuture<AttackerResponse> attestAfterPropose(long slot, String pubkey, String signedAttestDataBase64) {
return invokeRpc(ATTEST_MODULE + "_afterPropose", new Object[]{slot, pubkey, signedAttestDataBase64});
}
}
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment