Creating JwksFetcher interface and impl v2

bazel/repositories.bzl

@@ -464,6 +464,10 @@ def _com_google_absl():
         name = "abseil_symbolize",
         actual = "@com_google_absl//absl/debugging:symbolize",
     )
+    native.bind(
+        name = "abseil_time",
+        actual = "@com_google_absl//absl/time:time",
+    )
 
 def _com_google_protobuf():
     _repository_impl("com_google_protobuf")

bazel/repository_locations.bzl

@@ -70,7 +70,7 @@ REPOSITORY_LOCATIONS = dict(
         remote = "https://github.com/google/googleapis",
     ),
     com_github_google_jwt_verify = dict(
-        commit = "4eb9e96485b71e00d43acc7207501caafb085b4a",
+        commit = "66792a057ec54e4b75c6a2eeda4e98220bd12a9a",
         remote = "https://github.com/google/jwt_verify_lib",
     ),
     com_github_nodejs_http_parser = dict(

source/extensions/filters/http/common/BUILD

@@ -23,3 +23,17 @@ envoy_cc_library(
         "//include/envoy/server:filter_config_interface",
     ],
 )
+
+envoy_cc_library(
+    name = "jwks_fetcher_lib",
+    srcs = ["jwks_fetcher.cc"],
+    hdrs = ["jwks_fetcher.h"],
+    external_deps = [
+        "jwt_verify_lib",
+    ],
+    deps = [
+        "//include/envoy/upstream:cluster_manager_interface",
+        "//source/common/http:utility_lib",
+        "@envoy_api//envoy/api/v2/core:http_uri_cc",
+    ],
+)

source/extensions/filters/http/common/jwks_fetcher.cc

@@ -0,0 +1,91 @@
+#include "extensions/filters/http/common/jwks_fetcher.h"
+
+#include "common/common/enum_to_int.h"
+#include "common/http/headers.h"
+#include "common/http/utility.h"
+
+#include "jwt_verify_lib/status.h"
+
+namespace Envoy {
+namespace Extensions {
+namespace HttpFilters {
+namespace Common {
+namespace {
+class JwksFetcherImpl : public JwksFetcher,
+                        public Logger::Loggable<Logger::Id::filter>,
+                        public Http::AsyncClient::Callbacks {
+private:
+  Upstream::ClusterManager& cm_;
+  JwksFetcher::JwksReceiver* receiver_ = nullptr;
+  const ::envoy::api::v2::core::HttpUri* uri_ = nullptr;
+  Http::AsyncClient::Request* request_ = nullptr;
+
+public:
+  JwksFetcherImpl(Upstream::ClusterManager& cm) : cm_(cm) { ENVOY_LOG(trace, "{}", __func__); }
+
+  void cancel() {
+    if (request_) {
+      request_->cancel();
+      request_ = nullptr;
+      ENVOY_LOG(debug, "fetch pubkey [uri = {}]: canceled", uri_->uri());
+    }
+  }
+
+  void fetch(const ::envoy::api::v2::core::HttpUri& uri, JwksFetcher::JwksReceiver* receiver) {
+    ENVOY_LOG(trace, "{}", __func__);
+    receiver_ = receiver;
+    uri_ = &uri;
+    Http::MessagePtr message = Http::Utility::prepareHeaders(uri);
+    message->headers().insertMethod().value().setReference(Http::Headers::get().MethodValues.Get);
+    ENVOY_LOG(debug, "fetch pubkey from [uri = {}]: start", uri_->uri());
+    request_ =
+        cm_.httpAsyncClientForCluster(uri.cluster())
+            .send(std::move(message), *this,
+                  std::chrono::milliseconds(DurationUtil::durationToMilliseconds(uri.timeout())));
+  }
+
+  // HTTP async receive methods
+  void onSuccess(Http::MessagePtr&& response) {
+    ENVOY_LOG(trace, "{}", __func__);
+    request_ = nullptr;
+    const uint64_t status_code = Http::Utility::getResponseStatus(response->headers());
+    if (status_code == enumToInt(Http::Code::OK)) {
+      ENVOY_LOG(debug, "{}: fetch pubkey [uri = {}]: success", __func__, uri_->uri());
+      if (response->body()) {
+        const auto len = response->body()->length();
+        const auto body = std::string(static_cast<char*>(response->body()->linearize(len)), len);
+        auto jwks =
+            google::jwt_verify::Jwks::createFrom(body, google::jwt_verify::Jwks::Type::JWKS);
+        if (jwks->getStatus() == google::jwt_verify::Status::Ok) {
+          ENVOY_LOG(debug, "{}: fetch pubkey [uri = {}]: succeeded", __func__, uri_->uri());
+          receiver_->onJwksSuccess(std::move(jwks));
+        } else {
+          ENVOY_LOG(debug, "{}: fetch pubkey [uri = {}]: invalid jwks", __func__, uri_->uri());
+          receiver_->onJwksError(JwksFetcher::JwksReceiver::Failure::invalid_jwks);
+        }
+      } else {
+        ENVOY_LOG(debug, "{}: fetch pubkey [uri = {}]: body is empty", __func__, uri_->uri());
+        receiver_->onJwksError(JwksFetcher::JwksReceiver::Failure::network);
+      }
+    } else {
+      ENVOY_LOG(debug, "{}: fetch pubkey [uri = {}]: response status code {}", __func__,
+                uri_->uri(), status_code);
+      receiver_->onJwksError(JwksFetcher::JwksReceiver::Failure::network);
+    }
+  }
+
+  void onFailure(Http::AsyncClient::FailureReason reason) {
+    ENVOY_LOG(debug, "{}: fetch pubkey [uri = {}]: network error {}", __func__, uri_->uri(),
+              enumToInt(reason));
+    receiver_->onJwksError(JwksFetcher::JwksReceiver::Failure::network);
+  }
+};
+} // namespace
+
+JwksFetcherPtr JwksFetcher::create(Upstream::ClusterManager& cm) {
+  return JwksFetcherPtr(new JwksFetcherImpl(cm));
+}
+} // namespace Common
+} // namespace HttpFilters
+} // namespace Extensions
+} // namespace Envoy

source/extensions/filters/http/common/jwks_fetcher.h

@@ -0,0 +1,69 @@
+#pragma once
+
+#include "envoy/api/v2/core/http_uri.pb.h"
+#include "envoy/common/pure.h"
+#include "envoy/upstream/cluster_manager.h"
+
+#include "jwt_verify_lib/jwks.h"
+
+namespace Envoy {
+namespace Extensions {
+namespace HttpFilters {
+namespace Common {
+
+class JwksFetcher;
+typedef std::unique_ptr<JwksFetcher> JwksFetcherPtr;
+/**
+ * JwksFetcher interface can be used to retrieve remote JWKS
+ * (https://tools.ietf.org/html/rfc7517) data structures returning a concrete,
+ * type-safe representation. An instance of this interface is designed to
+ * retrieve a single JWKS and should not be re-used to fetch further instances.
+ */
+class JwksFetcher {
+public:
+  class JwksReceiver {
+  public:
+    enum class Failure {
+      unknown,
+      network,
+      invalid_jwks,
+    };
+
+    virtual ~JwksReceiver(){};
+    /*
+     * Successful retrieval callback.
+     * of the returned JWKS object.
+     * @param jwks the JWKS object retrieved.
+     */
+    virtual void onJwksSuccess(google::jwt_verify::JwksPtr&& jwks) PURE;
+    /*
+     * Retrieval error callback.
+     * * @param reason the failure reason.
+     */
+    virtual void onJwksError(Failure reason) PURE;
+  };
+
+  virtual ~JwksFetcher(){};
+
+  /*
+   * Cancel any inflight request.
+   */
+  virtual void cancel() PURE;
+
+  /*
+   * Retrieve a JWKS resource from a remote HTTP host.
+   * @param uri the uri to retrieve the jwks from.
+   */
+  virtual void fetch(const ::envoy::api::v2::core::HttpUri& uri, JwksReceiver* receiver) PURE;
+
+  /*
+   * Factory method for creating a JwksFetcher.
+   * @param cm the cluster manager to use during Jwks retrieval
+   * @return a JwksFetcher instance
+   */
+  static JwksFetcherPtr create(Upstream::ClusterManager& cm);
+};
+} // namespace Common
+} // namespace HttpFilters
+} // namespace Extensions
+} // namespace Envoy

source/extensions/filters/http/jwt_authn/BUILD

@@ -46,6 +46,7 @@ envoy_cc_library(
         "//include/envoy/server:filter_config_interface",
         "//include/envoy/stats:stats_macros",
         "//source/common/http:message_lib",
+        "//source/extensions/filters/http/common:jwks_fetcher_lib",
     ],
 )