com.google.ipc.invalidation.common.DigestFunction Java Examples

/**
 * Deserializes a Ticl state blob. Returns either the parsed state or {@code null}
 * if it could not be parsed.
 */
public static PersistentTiclState deserializeState(Logger logger, byte[] stateBlobBytes,
    DigestFunction digestFn) {
  PersistentStateBlob stateBlob;
  try {
    // Try parsing the envelope protocol buffer.
    stateBlob = PersistentStateBlob.parseFrom(stateBlobBytes);
  } catch (ValidationException exception) {
    logger.severe("Failed deserializing Ticl state: %s", exception.getMessage());
    return null;
  }

  // Check the mac in the envelope against the recomputed mac from the state.
  PersistentTiclState ticlState = stateBlob.getTiclState();
  Bytes mac = generateMac(ticlState, digestFn);
  if (!TypedUtil.<Bytes>equals(mac, stateBlob.getAuthenticationCode())) {
    logger.warning("Ticl state failed MAC check: computed %s vs %s", mac,
        stateBlob.getAuthenticationCode());
    return null;
  }
  return ticlState;
}
 

public RegistrationManager(Logger logger, Statistics statistics, DigestFunction digestFn,
    RegistrationManagerStateP registrationManagerState) {
  this.logger = logger;
  this.statistics = statistics;
  this.desiredRegistrations = new SimpleRegistrationStore(digestFn);

  if (registrationManagerState == null) {
    // Initialize the server summary with a 0 size and the digest corresponding
    // to it.  Using defaultInstance would wrong since the server digest will
    // not match unnecessarily and result in an info message being sent.
    this.lastKnownServerSummary = ProtoWrapper.of(getRegistrationSummary());
  } else {
    this.lastKnownServerSummary =
      ProtoWrapper.of(registrationManagerState.getLastKnownServerSummary());
    desiredRegistrations.add(registrationManagerState.getRegistrationsList());
    for (RegistrationP regOp : registrationManagerState.getPendingOperationsList()) {
      pendingOperations.put(ProtoWrapper.of(regOp.getObjectId()), regOp.getOpType());
    }
  }
}
 

/**
 * Deserializes a Ticl state blob. Returns either the parsed state or {@code null}
 * if it could not be parsed.
 */
public static PersistentTiclState deserializeState(Logger logger, byte[] stateBlobBytes,
    DigestFunction digestFn) {
  PersistentStateBlob stateBlob;
  try {
    // Try parsing the envelope protocol buffer.
    stateBlob = PersistentStateBlob.parseFrom(stateBlobBytes);
  } catch (InvalidProtocolBufferException exception) {
    logger.severe("Failed deserializing Ticl state: %s", exception.getMessage());
    return null;
  }

  // Check the mac in the envelope against the recomputed mac from the state.
  PersistentTiclState ticlState = stateBlob.getTiclState();
  ByteString mac = generateMac(ticlState, digestFn);
  if (!TypedUtil.<ByteString>equals(mac, stateBlob.getAuthenticationCode())) {
    logger.warning("Ticl state failed MAC check: computed %s vs %s", mac,
        stateBlob.getAuthenticationCode());
    return null;
  }
  return ticlState;
}
 

public RegistrationManager(Logger logger, Statistics statistics, DigestFunction digestFn,
    RegistrationManagerStateP registrationManagerState) {
  this.logger = logger;
  this.statistics = statistics;
  this.desiredRegistrations = new SimpleRegistrationStore(digestFn);

  if (registrationManagerState == null) {
    // Initialize the server summary with a 0 size and the digest corresponding
    // to it.  Using defaultInstance would wrong since the server digest will
    // not match unnecessarily and result in an info message being sent.
    this.lastKnownServerSummary = ProtoWrapper.of(getRegistrationSummary());
  } else {
    this.lastKnownServerSummary =
      ProtoWrapper.of(registrationManagerState.getLastKnownServerSummary());
    desiredRegistrations.add(registrationManagerState.getRegistrationsList());
    for (RegistrationP regOp : registrationManagerState.getPendingOperationsList()) {
      pendingOperations.put(ProtoWrapper.of(regOp.getObjectId()), regOp.getOpType());
    }
  }
}
 

/**
 * Deserializes a Ticl state blob. Returns either the parsed state or {@code null}
 * if it could not be parsed.
 */
public static PersistentTiclState deserializeState(Logger logger, byte[] stateBlobBytes,
    DigestFunction digestFn) {
  PersistentStateBlob stateBlob;
  try {
    // Try parsing the envelope protocol buffer.
    stateBlob = PersistentStateBlob.parseFrom(stateBlobBytes);
  } catch (InvalidProtocolBufferException exception) {
    logger.severe("Failed deserializing Ticl state: %s", exception.getMessage());
    return null;
  }

  // Check the mac in the envelope against the recomputed mac from the state.
  PersistentTiclState ticlState = stateBlob.getTiclState();
  ByteString mac = generateMac(ticlState, digestFn);
  if (!TypedUtil.<ByteString>equals(mac, stateBlob.getAuthenticationCode())) {
    logger.warning("Ticl state failed MAC check: computed %s vs %s", mac,
        stateBlob.getAuthenticationCode());
    return null;
  }
  return ticlState;
}
 

public RegistrationManager(Logger logger, Statistics statistics, DigestFunction digestFn,
    RegistrationManagerStateP registrationManagerState) {
  this.logger = logger;
  this.statistics = statistics;
  this.desiredRegistrations = new SimpleRegistrationStore(digestFn);

  if (registrationManagerState == null) {
    // Initialize the server summary with a 0 size and the digest corresponding
    // to it.  Using defaultInstance would wrong since the server digest will
    // not match unnecessarily and result in an info message being sent.
    this.lastKnownServerSummary = getRegistrationSummary();
  } else {
    this.lastKnownServerSummary = registrationManagerState.getNullableLastKnownServerSummary();
    if (this.lastKnownServerSummary == null) {
      // If no server summary is set, use a default with size 0.
      this.lastKnownServerSummary = getRegistrationSummary();
    }
    desiredRegistrations.add(registrationManagerState.getRegistrations());
    for (RegistrationP regOp : registrationManagerState.getPendingOperations()) {
      pendingOperations.put(regOp.getObjectId(), regOp.getOpType());
    }
  }
}
 

/**
 * Returns a copy of the registration manager's state
 * <p>
 * Direct test code MUST not call this method on a random thread. It must be called on the
 * InvalidationClientImpl's internal thread.
 */

RegistrationManagerState getRegistrationManagerStateCopyForTest(DigestFunction digestFunction) {
  List<ObjectIdP> registeredObjects = new ArrayList<ObjectIdP>();
  for (ObjectIdP oid : desiredRegistrations.getElements(EMPTY_PREFIX, 0)) {
    registeredObjects.add(oid);
  }
  return new RegistrationManagerState(
      RegistrationSummary.newBuilder(getRegistrationSummary()).build(),
      RegistrationSummary.newBuilder(lastKnownServerSummary.getProto()).build(),
      registeredObjects);
}
 

/**
 * Returns a copy of the registration manager's state
 * <p>
 * Direct test code MUST not call this method on a random thread. It must be called on the
 * InvalidationClientImpl's internal thread.
 */

RegistrationManagerState getRegistrationManagerStateCopyForTest(DigestFunction digestFunction) {
  List<ObjectIdP> registeredObjects = new ArrayList<ObjectIdP>();
  for (ObjectIdP oid : desiredRegistrations.getElements(EMPTY_PREFIX, 0)) {
    registeredObjects.add(oid);
  }
  return new RegistrationManagerState(
      RegistrationSummary.newBuilder(getRegistrationSummary()).build(),
      RegistrationSummary.newBuilder(lastKnownServerSummary.getProto()).build(),
      registeredObjects);
}
 

/** Returns the digest function used for computing digests for object registrations. */
DigestFunction getDigestFunctionForTest();
 

/** Returns the digest function used for computing digests for object registrations. */
DigestFunction getDigestFunctionForTest();
 

SimpleRegistrationStore(DigestFunction digestFunction) {
  this.digestFunction = digestFunction;
  recomputeDigest();
}
 

/** Serializes a Ticl state blob. */
public static byte[] serializeState(
    PersistentTiclState state, DigestFunction digestFn) {
  Bytes mac = generateMac(state, digestFn);
  return PersistentStateBlob.create(state, mac).toByteArray();
}
 

@Override

public DigestFunction getDigestFunctionForTest() {
  return this.digestFn;
}
 

/** Returns a message authentication code over {@code state}. */
private static ByteString generateMac(PersistentTiclState state, DigestFunction digestFn) {
  digestFn.reset();
  digestFn.update(state.toByteArray());
  return ByteString.copyFrom(digestFn.getDigest());
}
 

/** Returns a message authentication code over {@code state}. */
private static Bytes generateMac(PersistentTiclState state, DigestFunction digestFn) {
  digestFn.reset();
  digestFn.update(state.toByteArray());
  return new Bytes(digestFn.getDigest());
}
 

/** Serializes a Ticl state blob. */

public static byte[] serializeState(PersistentTiclState state, DigestFunction digestFn) {
  ByteString mac = generateMac(state, digestFn);
  return CommonProtos2.newPersistentStateBlob(state, mac).toByteArray();
}
 

SimpleRegistrationStore(DigestFunction digestFunction) {
  this.digestFunction = digestFunction;
  recomputeDigest();
}
 

SimpleRegistrationStore(DigestFunction digestFunction) {
  this.digestFunction = digestFunction;
  recomputeDigest();
}
 

@Override

public DigestFunction getDigestFunctionForTest() {
  return this.digestFn;
}
 

@Override

public DigestFunction getDigestFunctionForTest() {
  return this.digestFn;
}
 

/** Returns a message authentication code over {@code state}. */
private static ByteString generateMac(PersistentTiclState state, DigestFunction digestFn) {
  digestFn.reset();
  digestFn.update(state.toByteArray());
  return ByteString.copyFrom(digestFn.getDigest());
}
 

/** Serializes a Ticl state blob. */

public static byte[] serializeState(PersistentTiclState state, DigestFunction digestFn) {
  ByteString mac = generateMac(state, digestFn);
  return CommonProtos2.newPersistentStateBlob(state, mac).toByteArray();
}
 

/** Returns the digest function used for computing digests for object registrations. */
DigestFunction getDigestFunctionForTest();