-
Notifications
You must be signed in to change notification settings - Fork 2.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add OAuth support for Kafka client in native mode
- Loading branch information
Showing
2 changed files
with
125 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
94 changes: 94 additions & 0 deletions
94
...ient/runtime/src/main/java/io/quarkus/kafka/client/runtime/graal/SubjectSubstitution.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,94 @@ | ||
| package io.quarkus.kafka.client.runtime.graal; | ||
|
|
||
| import java.security.AccessControlContext; | ||
| import java.security.AccessController; | ||
| import java.util.Collections; | ||
| import java.util.Map; | ||
| import java.util.WeakHashMap; | ||
|
|
||
| import javax.security.auth.AuthPermission; | ||
| import javax.security.auth.Subject; | ||
| import javax.security.auth.SubjectDomainCombiner; | ||
|
|
||
| import com.oracle.svm.core.annotate.Substitute; | ||
| import com.oracle.svm.core.annotate.TargetClass; | ||
|
|
||
| @TargetClass(className = "javax.security.auth.Subject") | ||
| final class Target_javax_security_auth_Subject { | ||
|
|
||
| @Substitute | ||
| public static <T> T doAs(final Subject subject, | ||
| final java.security.PrivilegedExceptionAction<T> action) | ||
| throws java.security.PrivilegedActionException { | ||
|
|
||
| java.lang.SecurityManager sm = System.getSecurityManager(); | ||
| if (sm != null) { | ||
| sm.checkPermission(AuthPermissionHolder.DO_AS_PERMISSION); | ||
| } | ||
| if (action == null) { | ||
| throw new NullPointerException("Invalid null action provided"); | ||
| } | ||
|
|
||
| final AccessControlContext currentAcc = AccessController.getContext(); | ||
|
|
||
| SubjectHolder.subjects.put(currentAcc, subject); | ||
|
|
||
| return java.security.AccessController.doPrivileged(action, | ||
| createContext(subject, currentAcc)); | ||
| } | ||
|
|
||
| @Substitute | ||
| public static Subject getSubject(final AccessControlContext acc) { | ||
|
|
||
| java.lang.SecurityManager sm = System.getSecurityManager(); | ||
| if (sm != null) { | ||
| sm.checkPermission(AuthPermissionHolder.GET_SUBJECT_PERMISSION); | ||
| } | ||
|
|
||
| if (acc == null) { | ||
| throw new NullPointerException("Invalid null AccessControlContext provided"); | ||
| } | ||
|
|
||
| // return the Subject from the DomainCombiner of the provided context | ||
| return AccessController.doPrivileged(new java.security.PrivilegedAction<Subject>() { | ||
| public Subject run() { | ||
| return SubjectHolder.subjects.remove(acc); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| @Substitute | ||
| private static AccessControlContext createContext(final Subject subject, | ||
| final AccessControlContext acc) { | ||
|
|
||
| return java.security.AccessController.doPrivileged(new java.security.PrivilegedAction<AccessControlContext>() { | ||
| public AccessControlContext run() { | ||
| if (subject == null) | ||
| return new AccessControlContext(acc, null); | ||
| else | ||
| return new AccessControlContext(acc, | ||
| new SubjectDomainCombiner(subject)); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| final static class AuthPermissionHolder { | ||
| static final AuthPermission DO_AS_PERMISSION = new AuthPermission("doAs"); | ||
|
|
||
| static final AuthPermission DO_AS_PRIVILEGED_PERMISSION = new AuthPermission("doAsPrivileged"); | ||
|
|
||
| static final AuthPermission SET_READ_ONLY_PERMISSION = new AuthPermission("setReadOnly"); | ||
|
|
||
| static final AuthPermission GET_SUBJECT_PERMISSION = new AuthPermission("getSubject"); | ||
|
|
||
| static final AuthPermission MODIFY_PRINCIPALS_PERMISSION = new AuthPermission("modifyPrincipals"); | ||
|
|
||
| static final AuthPermission MODIFY_PUBLIC_CREDENTIALS_PERMISSION = new AuthPermission("modifyPublicCredentials"); | ||
|
|
||
| static final AuthPermission MODIFY_PRIVATE_CREDENTIALS_PERMISSION = new AuthPermission("modifyPrivateCredentials"); | ||
| } | ||
|
|
||
| final static class SubjectHolder { | ||
| static final Map<AccessControlContext, Subject> subjects = Collections.synchronizedMap(new WeakHashMap<>()); | ||
| } | ||
| } |