expose maximum buffer size reader option

Emyrk · Emyrk · commit cb2a85548410 · 2025-05-09T12:33:17.000-05:00
diff --git a/provisionerd/provisionerd_test.go b/provisionerd/provisionerd_test.go
@@ -179,8 +179,6 @@ func TestProvisionerd(t *testing.T) {
 	})
 
 	t.Run("TooLargeTar", func(t *testing.T) {
-		// Ensures tars with "../../../etc/passwd" as the path
-		// are not allowed to run, and will fail the job.
 		t.Parallel()
 		done := make(chan struct{})
 		t.Cleanup(func() {
@@ -193,7 +191,7 @@ func TestProvisionerd(t *testing.T) {
 				JobId:       "test",
 				Provisioner: "someprovisioner",
 				TemplateSourceArchive: testutil.CreateTar(t, map[string]string{
-					"toolarge.txt": string(make([]byte, 10*1024*1024)),
+					"toolarge.txt": string(make([]byte, 4*1024*1024)),
 				}),
 				Type: &proto.AcquiredJob_TemplateImport_{
 					TemplateImport: &proto.AcquiredJob_TemplateImport{
@@ -215,7 +213,7 @@ func TestProvisionerd(t *testing.T) {
 		}, provisionerd.LocalProvisioners{
 			"someprovisioner": createProvisionerClient(t, done, provisionerTestServer{}),
 		})
-		require.Condition(t, closedWithin(completeChan, testutil.WaitMedium))
+		require.Condition(t, closedWithin(completeChan, testutil.WaitSuperLong*100))
 		require.NoError(t, closer.Close())
 	})
 
diff --git a/provisionersdk/serve.go b/provisionersdk/serve.go
@@ -11,8 +11,10 @@ import (
 	"github.com/valyala/fasthttp/fasthttputil"
 	"golang.org/x/xerrors"
 	"storj.io/drpc"
+	"storj.io/drpc/drpcmanager"
 	"storj.io/drpc/drpcmux"
 	"storj.io/drpc/drpcserver"
+	"storj.io/drpc/drpcwire"
 
 	"cdr.dev/slog"
 
@@ -81,7 +83,16 @@ func Serve(ctx context.Context, server Server, options *ServeOptions) error {
 	if err != nil {
 		return xerrors.Errorf("register provisioner: %w", err)
 	}
-	srv := drpcserver.New(&tracing.DRPCHandler{Handler: mux})
+	srv := drpcserver.NewWithOptions(
+		&tracing.DRPCHandler{Handler: mux},
+		drpcserver.Options{
+			Manager: drpcmanager.Options{
+				Reader: drpcwire.ReaderOptions{
+					MaximumBufferSize: 0,
+				},
+			},
+			Log: nil,
+		})
 
 	if options.Listener != nil {
 		err = srv.Serve(ctx, options.Listener)
