totalWritten += written;
            }

            // Update original buffer position
            buffer.position(buffer.position() + totalWritten);

            log.debug("TCP RDMA write completed: {} bytes", totalWritten);
            return totalWritten;

        } catch (IOException e) {
            state = RdmaConnectionState.ERROR;

import java.io.IOException
import okio.Buffer
import okio.ForwardingSink
import okio.Sink

/** A sink that never throws IOExceptions, even if the underlying sink does. */
internal open class FaultHidingSink(
  delegate: Sink,
  val onException: (IOException) -> Unit,
) : ForwardingSink(delegate) {
  private var hasErrors = false

  override fun write(
    source: Buffer,
    byteCount: Long,
  ) {

#define TF_CLEANUP_AND_RETURN_IF_ERROR(C_STATUS, BUFFER, __VA_ARGS__) \
  do {                                                                \
    ::tensorflow::Status _s(__VA_ARGS__);                             \
    if (TF_PREDICT_TRUE(!_s.ok())) {                                  \
      TF_DeleteStatus(C_STATUS);                                      \
      TF_DeleteBuffer(BUFFER);                                        \

    private static final Pattern INVALID_PATH_CHARS = Pattern.compile("[\\x00-\\x1f\"*:<>?|]");

    /**
     * Validates buffer size to prevent overflow
     *
     * @param size the buffer size to validate
     * @param maxSize the maximum allowed size
     * @param fieldName the field name for error reporting
     * @throws IllegalArgumentException if size is invalid
     */

        if (args == null || args.length == 0) {
            return "";
        }
        final StringBuilder buffer = new StringBuilder();
        for (final Object arg : args) {
            buffer.append(arg + ", ");
        }
        buffer.setLength(buffer.length() - ", ".length());
        return new String(buffer);
    }

    /**
     * Initializes the class.
     */

    /** Maximum number of bytes to buffer before reporting a flow control error. */
    private val maxByteCount: Long,
    /**
     * True if either side has cleanly shut down this stream. We will receive no more bytes beyond
     * those already in the buffer.
     */
    internal var finished: Boolean,
  ) : Source {
    /** Buffer to receive data from the network into. Only accessed by the reader thread. */

        encodeSockaddr(buffer, 20);
        
        return buffer;
    }
    
    private void encodeSockaddr(byte[] buffer, int offset) {
        if (ipv6) {
            // IPv6 sockaddr_in6 structure
            writeInt2(buffer, offset, 23);  // AF_INET6
            writeInt2(buffer, offset + 2, 445);  // Port
            writeInt4(buffer, offset + 4, 0);  // Flow info

  object : BufferedSocket {
    private val delegate = this@asBufferedSocket
    override val source = delegate.source.buffer()
    override val sink = delegate.sink.buffer()

    override fun cancel() {
      delegate.cancel()
    }

     *
     * @param config the SMB configuration
     * @param pipeName the name of the pipe to call
     * @param data the data buffer to write to the pipe
     * @param off the offset in the data buffer
     * @param len the length of data to write
     */
    public TransCallNamedPipe(final Configuration config, final String pipeName, final byte[] data, final int off, final int len) {

     * Constructs a transaction request for a named pipe operation.
     *
     * @param config the configuration to use
     * @param fid the file ID of the named pipe
     * @param data the data buffer to send
     * @param off the offset in the data buffer
     * @param len the length of data to send
     */
    public TransTransactNamedPipe(final Configuration config, final int fid, final byte[] data, final int off, final int len) {