chore: expose ProxySetBufferBytes correctly & implement high level api for setting request and response body

Author: Wim Spaargaren

proxytest/proxytest.go

@@ -116,6 +116,11 @@ func (h *hostEmulator) ProxyGetBufferBytes(bt types.BufferType, start int, maxSi
 	}
 }
 
+func (h *hostEmulator) ProxySetBufferBytes(bt types.BufferType, start int, maxSize int, bufferData *byte, bufferSize int) types.Status {
+	// TODO: implement host emulator set buffer bytes
+	return types.StatusOK
+}
+
 // impl rawhostcall.ProxyWASMHost
 func (h *hostEmulator) ProxyGetHeaderMapValue(mapType types.MapType, keyData *byte,
 	keySize int, returnValueData **byte, returnValueSize *int) types.Status {

proxywasm/hostcall.go

@@ -128,6 +128,15 @@ func GetHttpRequestBody(start, maxSize int) ([]byte, error) {
 	return ret, types.StatusToError(st)
 }
 
+func SetHttpRequestBody(body []byte) error {
+	if len(body) == 0 {
+		return types.StatusToError(types.StatusBadArgument)
+	}
+	bufferData := &body[0]
+	st := rawhostcall.ProxySetBufferBytes(types.BufferTypeHttpRequestBody, 0, len(body), bufferData, len(body))
+	return types.StatusToError(st)
+}
+
 func GetHttpRequestTrailers() ([][2]string, error) {
 	ret, st := getMap(types.MapTypeHttpRequestTrailers)
 	return ret, types.StatusToError(st)
@@ -189,6 +198,15 @@ func GetHttpResponseBody(start, maxSize int) ([]byte, error) {
 	return ret, types.StatusToError(st)
 }
 
+func SetHttpResponseBody(body []byte) error {
+	if len(body) == 0 {
+		return types.StatusToError(types.StatusBadArgument)
+	}
+	bufferData := &body[0]
+	st := rawhostcall.ProxySetBufferBytes(types.BufferTypeHttpResponseBody, 0, len(body), bufferData, len(body))
+	return types.StatusToError(st)
+}
+
 func GetHttpResponseTrailers() ([][2]string, error) {
 	ret, st := getMap(types.MapTypeHttpResponseTrailers)
 	return ret, types.StatusToError(st)

proxywasm/rawhostcall/rawhostcall_mock.go

@@ -44,6 +44,7 @@ type ProxyWASMHost interface {
 	ProxyGetHeaderMapPairs(mapType types.MapType, returnValueData **byte, returnValueSize *int) types.Status
 	ProxySetHeaderMapPairs(mapType types.MapType, mapData *byte, mapSize int) types.Status
 	ProxyGetBufferBytes(bt types.BufferType, start int, maxSize int, returnBufferData **byte, returnBufferSize *int) types.Status
+	ProxySetBufferBytes(bt types.BufferType, start int, maxSize int, bufferData *byte, bufferSize int) types.Status
 	ProxyHttpCall(upstreamData *byte, upstreamSize int, headerData *byte, headerSize int, bodyData *byte, bodySize int, trailersData *byte, trailersSize int, timeout uint32, calloutIDPtr *uint32) types.Status
 	ProxySetTickPeriodMilliseconds(period uint32) types.Status
 	ProxyGetCurrentTimeNanoseconds(returnTime *int64) types.Status
@@ -112,6 +113,9 @@ func (d DefaultProxyWAMSHost) ProxySetHeaderMapPairs(mapType types.MapType, mapD
 func (d DefaultProxyWAMSHost) ProxyGetBufferBytes(bt types.BufferType, start int, maxSize int, returnBufferData **byte, returnBufferSize *int) types.Status {
 	return 0
 }
+func (d DefaultProxyWAMSHost) ProxySetBufferBytes(bt types.BufferType, start int, maxSize int, bufferData *byte, bufferSize int) types.Status {
+	return 0
+}
 func (d DefaultProxyWAMSHost) ProxyHttpCall(upstreamData *byte, upstreamSize int, headerData *byte, headerSize int, bodyData *byte, bodySize int, trailersData *byte, trailersSize int, timeout uint32, calloutIDPtr *uint32) types.Status {
 	return 0
 }
@@ -209,6 +213,10 @@ func ProxyGetBufferBytes(bt types.BufferType, start int, maxSize int, returnBuff
 	return currentHost.ProxyGetBufferBytes(bt, start, maxSize, returnBufferData, returnBufferSize)
 }
 
+func ProxySetBufferBytes(bt types.BufferType, start int, maxSize int, bufferData *byte, bufferSize int) types.Status {
+	return currentHost.ProxySetBufferBytes(bt, start, maxSize, bufferData, bufferSize)
+}
+
 func ProxyHttpCall(upstreamData *byte, upstreamSize int, headerData *byte, headerSize int, bodyData *byte,
 	bodySize int, trailersData *byte, trailersSize int, timeout uint32, calloutIDPtr *uint32) types.Status {
 	return currentHost.ProxyHttpCall(upstreamData, upstreamSize,