Fix iovec usages

Author: nekohasekai

common/buf/buffer_unix.go

@@ -4,9 +4,9 @@ package buf
 
 import "golang.org/x/sys/unix"
 
-func (b *Buffer) Iovec() unix.Iovec {
+func (b *Buffer) Iovec(length int) unix.Iovec {
 	var iov unix.Iovec
 	iov.Base = &b.data[b.start]
-	iov.SetLen(b.capacity)
+	iov.SetLen(length)
 	return iov
 }

common/buf/buffer_windows.go

@@ -2,9 +2,9 @@ package buf
 
 import "golang.org/x/sys/windows"
 
-func (b *Buffer) Iovec() windows.WSABuf {
+func (b *Buffer) Iovec(length int) windows.WSABuf {
 	return windows.WSABuf{
 		Buf: &b.data[b.start],
-		Len: uint32(b.capacity),
+		Len: uint32(length),
 	}
 }

common/bufio/copy_direct_posix.go

@@ -114,7 +114,7 @@ func (w *vectorisedSyscallReadWaiter) InitializeReadWaiter(options N.ReadWaitOpt
 				buffer = w.options.NewBuffer()
 				w.buffers[i] = buffer
 			}
-			w.iovecList[i] = buffer.Iovec()
+			w.iovecList[i] = buffer.Iovec(buffer.Cap())
 		}
 		var (
 			readN     uintptr

common/bufio/copy_direct_windows.go

@@ -125,7 +125,7 @@ func (w *vectorisedSyscallReadWaiter) InitializeReadWaiter(options N.ReadWaitOpt
 				buffer = w.options.NewBuffer()
 				w.buffers[i] = buffer
 			}
-			w.iovecList[i] = buffer.Iovec()
+			w.iovecList[i] = buffer.Iovec(buffer.Cap())
 		}
 		var readN uint32
 		//nolint:staticcheck

common/bufio/vectorised_unix.go

@@ -24,7 +24,7 @@ func (w *SyscallVectorisedWriter) WriteVectorised(buffers []*buf.Buffer) error {
 	defer buf.ReleaseMulti(buffers)
 	iovecList := w.iovecList
 	for _, buffer := range buffers {
-		iovecList = append(iovecList, buffer.Iovec())
+		iovecList = append(iovecList, buffer.Iovec(buffer.Len()))
 	}
 	var innerErr unix.Errno
 	err := w.rawConn.Write(func(fd uintptr) (done bool) {
@@ -50,7 +50,7 @@ func (w *SyscallVectorisedPacketWriter) WriteVectorisedPacket(buffers []*buf.Buf
 	defer buf.ReleaseMulti(buffers)
 	iovecList := w.iovecList
 	for _, buffer := range buffers {
-		iovecList = append(iovecList, buffer.Iovec())
+		iovecList = append(iovecList, buffer.Iovec(buffer.Len()))
 	}
 	var innerErr error
 	err := w.rawConn.Write(func(fd uintptr) (done bool) {

common/bufio/vectorised_windows.go

@@ -20,7 +20,7 @@ func (w *SyscallVectorisedWriter) WriteVectorised(buffers []*buf.Buffer) error {
 	defer buf.ReleaseMulti(buffers)
 	iovecList := w.iovecList
 	for _, buffer := range buffers {
-		iovecList = append(iovecList, buffer.Iovec())
+		iovecList = append(iovecList, buffer.Iovec(buffer.Len()))
 	}
 	var n uint32
 	var innerErr error
@@ -46,7 +46,7 @@ func (w *SyscallVectorisedPacketWriter) WriteVectorisedPacket(buffers []*buf.Buf
 	defer buf.ReleaseMulti(buffers)
 	iovecList := w.iovecList
 	for _, buffer := range buffers {
-		iovecList = append(iovecList, buffer.Iovec())
+		iovecList = append(iovecList, buffer.Iovec(buffer.Len()))
 	}
 	var n uint32
 	var innerErr error