Method readFull

cmd/run/socket/proxy.go:665–690

readFull reads len(p) bytes from the client or server reader, enforcing a shared byte limit across both. Returns an error if the limit is exceeded. Safe for both the client and server to call concurrently.

(isClient bool, p []byte)

// byte limit across both. Returns an error if the limit is exceeded. Safe for both
// the client and server to call concurrently.
func (w *websocket) readFull(isClient bool, p []byte) (int, error) {
	l := uint64(len(p))

	for {
		remaining := w.n.Load()
		if remaining < l {
			return 0, fmt.Errorf("%w: limit=%d, have=%d, want=%d, msgsRead=%d", errWebsocketPayloadLimitExceeded, tracer.PayloadLimitBytes, remaining, l, len(w.getMessages()))
		}

		if w.n.CompareAndSwap(remaining, remaining-l) {
			var r io.Reader
			if isClient {
				r = w.cli
			} else {
				r = w.srv
			}

			n, err := io.ReadFull(r, p)
			if err != nil {
				w.n.Add(l - uint64(n))
				return n, err
			}
			return n, nil
		}
	}
}

func (w *websocket) readFrame(isClient bool) (*websocketFrame, error) {
	header := make([]byte, 2)