Browse Source

Release ByteBuf

Closes gh-9661
pull/9717/head
kevin 5 years ago committed by Josh Cummings
parent
commit
32ac31c101
No known key found for this signature in database
GPG Key ID: 49EF60DD7FF83443
  1. 28
      rsocket/src/main/java/org/springframework/security/rsocket/authentication/AuthenticationPayloadExchangeConverter.java

28
rsocket/src/main/java/org/springframework/security/rsocket/authentication/AuthenticationPayloadExchangeConverter.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2019 the original author or authors. * Copyright 2019-2021 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -71,18 +71,24 @@ public class AuthenticationPayloadExchangeConverter implements PayloadExchangeAu
if (authenticationMetadata == null) { if (authenticationMetadata == null) {
return null; return null;
} }
ByteBuf rawAuthentication = ByteBufAllocator.DEFAULT.buffer().writeBytes(authenticationMetadata); ByteBuf rawAuthentication = ByteBufAllocator.DEFAULT.buffer();
if (!AuthMetadataCodec.isWellKnownAuthType(rawAuthentication)) { try {
return null; rawAuthentication.writeBytes(authenticationMetadata);
} if (!AuthMetadataCodec.isWellKnownAuthType(rawAuthentication)) {
WellKnownAuthType wellKnownAuthType = AuthMetadataCodec.readWellKnownAuthType(rawAuthentication); return null;
if (WellKnownAuthType.SIMPLE.equals(wellKnownAuthType)) { }
return simple(rawAuthentication); WellKnownAuthType wellKnownAuthType = AuthMetadataCodec.readWellKnownAuthType(rawAuthentication);
if (WellKnownAuthType.SIMPLE.equals(wellKnownAuthType)) {
return simple(rawAuthentication);
}
if (WellKnownAuthType.BEARER.equals(wellKnownAuthType)) {
return bearer(rawAuthentication);
}
throw new IllegalArgumentException("Unknown Mime Type " + wellKnownAuthType);
} }
if (WellKnownAuthType.BEARER.equals(wellKnownAuthType)) { finally {
return bearer(rawAuthentication); rawAuthentication.release();
} }
throw new IllegalArgumentException("Unknown Mime Type " + wellKnownAuthType);
} }
private Authentication simple(ByteBuf rawAuthentication) { private Authentication simple(ByteBuf rawAuthentication) {

Loading…
Cancel
Save