dao-jun commented on code in PR #22760: URL: https://github.com/apache/pulsar/pull/22760#discussion_r1610502839
########## pulsar-common/src/main/java/org/apache/pulsar/common/protocol/ByteBufPair.java: ########## @@ -122,38 +123,22 @@ public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) // ByteBuf are automatically released after a write. If the ByteBufPair ref count is increased and it // gets written multiple times, the individual buffers refcount should be reflected as well. try { - ctx.write(b.getFirst().retainedDuplicate(), ctx.voidPromise()); - ctx.write(b.getSecond().retainedDuplicate(), promise); + ctx.write(readOnlyRetainedDuplicate(b.getFirst()), ctx.voidPromise()); + ctx.write(readOnlyRetainedDuplicate(b.getSecond()), promise); } finally { ReferenceCountUtil.safeRelease(b); } } else { ctx.write(msg, promise); } } - } - - @Sharable - @SuppressWarnings("checkstyle:JavadocType") - public static class CopyingEncoder extends ChannelOutboundHandlerAdapter { - @Override - public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception { - if (msg instanceof ByteBufPair) { - ByteBufPair b = (ByteBufPair) msg; - // Some handlers in the pipeline will modify the bytebufs passed in to them (i.e. SslHandler). - // For these handlers, we need to pass a copy of the buffers as the source buffers may be cached - // for multiple requests. - try { - ctx.write(b.getFirst().copy(), ctx.voidPromise()); - ctx.write(b.getSecond().copy(), promise); - } finally { - ReferenceCountUtil.safeRelease(b); - } - } else { - ctx.write(msg, promise); - } + // .asReadOnly() is needed to prevent SslHandler from modifying the input buffers. + private static ByteBuf readOnlyRetainedDuplicate(ByteBuf buf) { + // If the buffer is already read-only, .asReadOnly() will return the same buffer. + // That's why the additional .retainedDuplicate() is needed to ensure that the returned buffer + // has independent readIndex and writeIndex. + return buf.asReadOnly().retainedDuplicate(); Review Comment: This is just defensive programming, if we are sure that the input buf cannot be null, this line can be removed. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: commits-unsubscr...@pulsar.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org