Add the method DeriveBits to AlgorithmImplementation.

And wire it up to WebCryptoImplementation.

BUG=399093

Review URL: https://codereview.chromium.org/744993002

Cr-Commit-Position: refs/heads/master@{#305863}

1 files changed, 45 insertions, 0 deletions

diff --git a/content/child/webcrypto/webcrypto_impl.cc b/content/child/webcrypto/webcrypto_impl.cc
index 5b033a5..8efd9de 100644
--- a/content/child/webcrypto/webcrypto_impl.cc
+++ b/content/child/webcrypto/webcrypto_impl.cc
@@ -333,6 +333,23 @@ struct UnwrapKeyState : public BaseState {
   blink::WebCryptoKey unwrapped_key;
 };
 
+struct DeriveBitsState : public BaseState {
+  DeriveBitsState(const blink::WebCryptoAlgorithm& algorithm,
+                  const blink::WebCryptoKey& base_key,
+                  unsigned int length_bits,
+                  const blink::WebCryptoResult& result)
+      : BaseState(result),
+        algorithm(algorithm),
+        base_key(base_key),
+        length_bits(length_bits) {}
+
+  const blink::WebCryptoAlgorithm algorithm;
+  const blink::WebCryptoKey base_key;
+  const unsigned int length_bits;
+
+  std::vector<uint8_t> derived_bytes;
+};
+
 // --------------------------------------------------------------------
 // Wrapper functions
 // --------------------------------------------------------------------
@@ -519,6 +536,22 @@ void DoUnwrapKey(scoped_ptr<UnwrapKeyState> passed_state) {
       FROM_HERE, base::Bind(DoUnwrapKeyReply, Passed(&passed_state)));
 }
 
+void DoDeriveBitsReply(scoped_ptr<DeriveBitsState> state) {
+  CompleteWithBufferOrError(state->status, state->derived_bytes,
+                            &state->result);
+}
+
+void DoDeriveBits(scoped_ptr<DeriveBitsState> passed_state) {
+  DeriveBitsState* state = passed_state.get();
+  if (state->cancelled())
+    return;
+  state->status =
+      webcrypto::DeriveBits(state->algorithm, state->base_key,
+                            state->length_bits, &state->derived_bytes);
+  state->origin_thread->PostTask(
+      FROM_HERE, base::Bind(DoDeriveBitsReply, Passed(&passed_state)));
+}
+
 }  // namespace
 
 WebCryptoImpl::WebCryptoImpl() {
@@ -670,6 +703,18 @@ void WebCryptoImpl::unwrapKey(
   }
 }
 
+void WebCryptoImpl::deriveBits(const blink::WebCryptoAlgorithm& algorithm,
+                               const blink::WebCryptoKey& base_key,
+                               unsigned int length_bits,
+                               blink::WebCryptoResult result) {
+  scoped_ptr<DeriveBitsState> state(
+      new DeriveBitsState(algorithm, base_key, length_bits, result));
+  if (!CryptoThreadPool::PostTask(FROM_HERE,
+                                  base::Bind(DoDeriveBits, Passed(&state)))) {
+    CompleteWithThreadPoolError(&result);
+  }
+}
+
 blink::WebCryptoDigestor* WebCryptoImpl::createDigestor(
     blink::WebCryptoAlgorithmId algorithm_id) {
   return webcrypto::CreateDigestor(algorithm_id).release();