starpc 0.49.8 → 0.49.10

package/README.md

@@ -151,20 +151,20 @@ Add the dependencies to your `Cargo.toml`:
 
 ```toml
 [dependencies]
-starpc = "0.1"
-prost = "0.13"
+starpc = "0.49"
+prost = "0.14"
 tokio = { version = "1", features = ["rt", "macros"] }
 
 [build-dependencies]
-starpc-build = "0.1"
-prost-build = "0.13"
+starpc = { version = "0.49", features = ["build"] }
+prost-build = "0.14"
 ```
 
 Create a `build.rs` to generate code from your proto files:
 
 ```rust
 fn main() -> Result<(), Box<dyn std::error::Error>> {
-    starpc_build::configure()
+    starpc::build::configure()
         .compile_protos(&["proto/echo.proto"], &["proto"])?;
     Ok(())
 }

package/dist/srpc/packet.d.ts

@@ -7,14 +7,14 @@ export declare function uint32LEDecode(data: Uint8ArrayList): number;
 export declare namespace uint32LEDecode {
     var bytes: number;
 }
-export declare function uint32LEEncode(value: number): Uint8ArrayList;
+export declare function uint32LEEncode(value: number): Uint8ArrayList<ArrayBuffer>;
 export declare namespace uint32LEEncode {
     var bytes: number;
 }
-export declare function lengthPrefixEncode(source: Source<Uint8Array | Uint8ArrayList>, lengthEncoder: typeof uint32LEEncode): AsyncGenerator<Uint8ArrayList, void, unknown>;
-export declare function lengthPrefixDecode(source: Source<Uint8Array | Uint8ArrayList>, lengthDecoder: typeof uint32LEDecode): AsyncGenerator<Uint8ArrayList, void, unknown>;
+export declare function lengthPrefixEncode(source: Source<Uint8Array | Uint8ArrayList>, lengthEncoder: typeof uint32LEEncode): AsyncGenerator<Uint8ArrayList<ArrayBufferLike>, void, unknown>;
+export declare function lengthPrefixDecode(source: Source<Uint8Array | Uint8ArrayList>, lengthDecoder: typeof uint32LEDecode): AsyncGenerator<Uint8ArrayList<ArrayBufferLike>, void, unknown>;
 export declare function prependLengthPrefixTransform(lengthEncoder?: {
-    (value: number): Uint8ArrayList;
+    (value: number): Uint8ArrayList<ArrayBuffer>;
     bytes: number;
 }): Transform<Source<Uint8Array | Uint8ArrayList>, AsyncGenerator<Uint8ArrayList, void, undefined> | Generator<Uint8ArrayList, void, undefined>>;
 export declare function parseLengthPrefixTransform(lengthDecoder?: {

package/go.mod

@@ -5,7 +5,7 @@ go 1.25.0
 require (
 	github.com/aperturerobotics/common v0.33.0 // latest
 	github.com/aperturerobotics/protobuf-go-lite v0.13.0 // latest
-	github.com/aperturerobotics/util v1.34.3 // latest
+	github.com/aperturerobotics/util v1.34.5-0.20260516103104-cbfc6d6a0589 // latest
 )
 
 require (
@@ -21,7 +21,7 @@ require (
 require (
 	github.com/libp2p/go-yamux/v4 v4.0.2 // latest
 	github.com/pkg/errors v0.9.1 // latest
-	github.com/sirupsen/logrus v1.9.5-0.20260426203557-6878cb36b029 // latest
+	github.com/sirupsen/logrus v1.9.5-0.20260508084601-d4a50659cfd6 // latest
 )
 
 require (
@@ -29,5 +29,5 @@ require (
 	github.com/tetratelabs/wazero v1.11.0 // indirect
 	github.com/xrash/smetrics v0.0.0-20250705151800-55b8f293f342 // indirect
 	golang.org/x/mod v0.35.0 // indirect
-	golang.org/x/sys v0.43.0 // indirect
+	golang.org/x/sys v0.44.0 // indirect
 )

package/go.sum

@@ -16,8 +16,10 @@ github.com/aperturerobotics/protobuf v0.0.0-20260203024654-8201686529c4 h1:4Dy3B
 github.com/aperturerobotics/protobuf v0.0.0-20260203024654-8201686529c4/go.mod h1:tMgO7y6SJo/d9ZcvrpNqIQtdYT9de+QmYaHOZ4KnhOg=
 github.com/aperturerobotics/protobuf-go-lite v0.13.0 h1:jEvCJhHaJEikDY/va2AUnS0DOb/0n82aISLAqxSh4Sk=
 github.com/aperturerobotics/protobuf-go-lite v0.13.0/go.mod h1:lGH3s5ArCTXKI4wJdlNpaybUtwSjfAG0vdWjxOfMcF8=
-github.com/aperturerobotics/util v1.34.3 h1:9lFYJovGlAHYX66aWKVzfoVzYox13P054d/Dy8T/GRg=
-github.com/aperturerobotics/util v1.34.3/go.mod h1:xtE2hwpgKGaW0TLx01+9xLsG+rYvhygQ+JCuQeAbvME=
+github.com/aperturerobotics/util v1.34.5-0.20260515183346-68f9eac1d69f h1:xISFLs00h441uZcMVxhZbLIZsMRcjOM5Yont18i7WjA=
+github.com/aperturerobotics/util v1.34.5-0.20260515183346-68f9eac1d69f/go.mod h1:mDe7WnncVuV7yjeeVSsagyfrw4xfncu7d+f0+d70niY=
+github.com/aperturerobotics/util v1.34.5-0.20260516103104-cbfc6d6a0589 h1:8B9O13He1sz8Spr2pc+RL3hBzAMveLgUCXT7BpAfvEY=
+github.com/aperturerobotics/util v1.34.5-0.20260516103104-cbfc6d6a0589/go.mod h1:mDe7WnncVuV7yjeeVSsagyfrw4xfncu7d+f0+d70niY=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
@@ -30,8 +32,8 @@ github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/sirupsen/logrus v1.9.5-0.20260426203557-6878cb36b029 h1:24s81vaIm1uch3GUIIZxneOyFRfBbHRdymMRBdoDuk0=
-github.com/sirupsen/logrus v1.9.5-0.20260426203557-6878cb36b029/go.mod h1:FXZFonkDAnFozmO+5hGAFvB0Yg9/j2SIhA/QuIkP180=
+github.com/sirupsen/logrus v1.9.5-0.20260508084601-d4a50659cfd6 h1:D6qewLO/pJ+JvUwuri1dmCbc32d/eO+xyqg+p3N+9kA=
+github.com/sirupsen/logrus v1.9.5-0.20260508084601-d4a50659cfd6/go.mod h1:FXZFonkDAnFozmO+5hGAFvB0Yg9/j2SIhA/QuIkP180=
 github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
 github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
 github.com/tetratelabs/wazero v1.11.0 h1:+gKemEuKCTevU4d7ZTzlsvgd1uaToIDtlQlmNbwqYhA=
@@ -40,8 +42,8 @@ github.com/xrash/smetrics v0.0.0-20250705151800-55b8f293f342 h1:FnBeRrxr7OU4VvAz
 github.com/xrash/smetrics v0.0.0-20250705151800-55b8f293f342/go.mod h1:Ohn+xnUBiLI6FVj/9LpzZWtj1/D6lUovWYBkxHVV3aM=
 golang.org/x/mod v0.35.0 h1:Ww1D637e6Pg+Zb2KrWfHQUnH2dQRLBQyAtpr/haaJeM=
 golang.org/x/mod v0.35.0/go.mod h1:+GwiRhIInF8wPm+4AoT6L0FA1QWAad3OMdTRx4tFYlU=
-golang.org/x/sys v0.43.0 h1:Rlag2XtaFTxp19wS8MXlJwTvoh8ArU6ezoyFsMyCTNI=
-golang.org/x/sys v0.43.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
+golang.org/x/sys v0.44.0 h1:ildZl3J4uzeKP07r2F++Op7E9B29JRUy+a27EibtBTQ=
+golang.org/x/sys v0.44.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
 google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
 google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "starpc",
-  "version": "0.49.8",
+  "version": "0.49.10",
   "description": "Streaming protobuf RPC service protocol over any two-way channel.",
   "license": "MIT",
   "author": {
@@ -137,7 +137,7 @@
     "it-pipe": "^3.0.1",
     "it-pushable": "^3.2.3",
     "it-stream-types": "^2.0.4",
-    "uint8arraylist": "^2.4.9",
+    "uint8arraylist": "^3.0.0",
     "ws": "^8.20.0"
   }
 }

package/srpc/build.rs

@@ -0,0 +1,593 @@
+//! Build-script helpers for generating Starpc Rust service bindings.
+//!
+//! Enable the `build` feature and call [`configure`] from a downstream
+//! `build.rs` to install the Starpc service generator into `prost-build`.
+
+/// Configures `prost-build` to generate Starpc service bindings.
+///
+/// The returned config generates normal Prost message types and appends Starpc
+/// client/server traits, client implementations, stream wrappers, and handler
+/// glue for every protobuf service.
+pub fn configure() -> prost_build::Config {
+    let mut config = prost_build::Config::new();
+    if let Ok(protoc) = protoc_bin_vendored::protoc_bin_path() {
+        config.protoc_executable(protoc);
+    }
+    if let Ok(include) = protoc_bin_vendored::include_path() {
+        config.protoc_arg(format!("--proto_path={}", include.display()));
+    }
+    config.extern_path(".rpcstream", "::starpc::rpcstream");
+    config.service_generator(Box::new(StarpcServiceGenerator::default()));
+    config
+}
+
+/// StarpcServiceGenerator emits Starpc client and server glue for Prost services.
+#[derive(Debug, Default)]
+pub struct StarpcServiceGenerator;
+
+impl prost_build::ServiceGenerator for StarpcServiceGenerator {
+    fn generate(&mut self, service: prost_build::Service, buf: &mut String) {
+        let mut gen = Generator::new(service, buf);
+        gen.generate();
+    }
+}
+
+struct Generator<'a> {
+    service: prost_build::Service,
+    buf: &'a mut String,
+}
+
+impl<'a> Generator<'a> {
+    fn new(service: prost_build::Service, buf: &'a mut String) -> Self {
+        Self { service, buf }
+    }
+
+    fn generate(&mut self) {
+        let service_id = self.service_id();
+        let service_name = self.service.name.clone();
+
+        self.line("");
+        self.line("#[allow(unused_imports)]");
+        self.line("use starpc::StreamExt;");
+        self.line("");
+        self.line(&format!("/// Service ID for {}.", service_name));
+        self.line(&format!(
+            "pub const {}: &str = {:?};",
+            service_id_const(&service_name),
+            service_id
+        ));
+        self.line("");
+
+        self.generate_stream_traits();
+        self.generate_client_trait();
+        self.generate_client_impl();
+        self.generate_stream_impls();
+        self.generate_server_trait();
+        self.generate_handler();
+    }
+
+    fn generate_stream_traits(&mut self) {
+        let service_name = self.service.name.clone();
+        for method in self.service.methods.clone() {
+            if !method.client_streaming && !method.server_streaming {
+                continue;
+            }
+
+            let stream_trait = stream_trait_name(&service_name, &method);
+
+            self.line(&format!(
+                "/// Stream trait for {}.{}.",
+                service_name, method.proto_name
+            ));
+            self.line("#[starpc::async_trait]");
+            self.line(&format!("pub trait {}: Send + Sync {{", stream_trait));
+            self.line("    /// Returns the context for this stream.");
+            self.line("    fn context(&self) -> &starpc::Context;");
+
+            if method.client_streaming {
+                self.line("    /// Sends a message on the stream.");
+                self.line(&format!(
+                    "    async fn send(&self, msg: &{}) -> starpc::Result<()>;",
+                    method.input_type
+                ));
+            }
+
+            if method.server_streaming {
+                self.line("    /// Receives a message from the stream.");
+                self.line(&format!(
+                    "    async fn recv(&self) -> starpc::Result<{}>;",
+                    method.output_type
+                ));
+            }
+
+            if method.client_streaming && !method.server_streaming {
+                self.line("    /// Closes the send side and receives the response.");
+                self.line(&format!(
+                    "    async fn close_and_recv(&self) -> starpc::Result<{}>;",
+                    method.output_type
+                ));
+            } else {
+                self.line("    /// Closes the stream.");
+                self.line("    async fn close(&self) -> starpc::Result<()>;");
+            }
+
+            self.line("}");
+            self.line("");
+        }
+    }
+
+    fn generate_client_trait(&mut self) {
+        let service_name = self.service.name.clone();
+        self.line(&format!("/// Client trait for {}.", service_name));
+        self.line("#[starpc::async_trait]");
+        self.line(&format!("pub trait {}Client: Send + Sync {{", service_name));
+
+        for method in self.service.methods.clone() {
+            self.line(&format!("    /// {}.", method.proto_name));
+            self.line(&format!(
+                "    {};",
+                client_trait_method(&service_name, &method)
+            ));
+        }
+
+        self.line("}");
+        self.line("");
+    }
+
+    fn generate_client_impl(&mut self) {
+        let service_id = self.service_id();
+        let service_name = self.service.name.clone();
+
+        self.line(&format!("/// Client implementation for {}.", service_name));
+        self.line(&format!("pub struct {}ClientImpl<C> {{", service_name));
+        self.line("    client: C,");
+        self.line("}");
+        self.line("");
+        self.line(&format!(
+            "impl<C: starpc::Client> {}ClientImpl<C> {{",
+            service_name
+        ));
+        self.line("    /// Creates a new client.");
+        self.line("    pub fn new(client: C) -> Self {");
+        self.line("        Self { client }");
+        self.line("    }");
+        self.line("}");
+        self.line("");
+        self.line("#[starpc::async_trait]");
+        self.line(&format!(
+            "impl<C: starpc::Client + 'static> {}Client for {}ClientImpl<C> {{",
+            service_name, service_name
+        ));
+
+        for method in self.service.methods.clone() {
+            let method_name = &method.name;
+            if method.client_streaming && method.server_streaming {
+                self.line(&format!(
+                    "    async fn {}(&self) -> starpc::Result<Box<dyn {}>> {{",
+                    method_name,
+                    stream_trait_name(&service_name, &method)
+                ));
+                self.line(&format!(
+                    "        let stream = self.client.new_stream({:?}, {:?}, None).await?;",
+                    service_id, method.proto_name
+                ));
+                self.line(&format!(
+                    "        Ok(Box::new({}Impl {{ stream }}))",
+                    stream_trait_name(&service_name, &method)
+                ));
+                self.line("    }");
+            } else if method.server_streaming {
+                self.line(&format!(
+                    "    async fn {}(&self, request: &{}) -> starpc::Result<Box<dyn {}>> {{",
+                    method_name,
+                    method.input_type,
+                    stream_trait_name(&service_name, &method)
+                ));
+                self.line("        use starpc::ProstMessage;");
+                self.line("        let data = request.encode_to_vec();");
+                self.line(&format!(
+                    "        let stream = self.client.new_stream({:?}, {:?}, Some(&data)).await?;",
+                    service_id, method.proto_name
+                ));
+                self.line("        stream.close_send().await?;");
+                self.line(&format!(
+                    "        Ok(Box::new({}Impl {{ stream }}))",
+                    stream_trait_name(&service_name, &method)
+                ));
+                self.line("    }");
+            } else if method.client_streaming {
+                self.line(&format!(
+                    "    async fn {}(&self) -> starpc::Result<Box<dyn {}>> {{",
+                    method_name,
+                    stream_trait_name(&service_name, &method)
+                ));
+                self.line(&format!(
+                    "        let stream = self.client.new_stream({:?}, {:?}, None).await?;",
+                    service_id, method.proto_name
+                ));
+                self.line(&format!(
+                    "        Ok(Box::new({}Impl {{ stream }}))",
+                    stream_trait_name(&service_name, &method)
+                ));
+                self.line("    }");
+            } else {
+                self.line(&format!(
+                    "    async fn {}(&self, request: &{}) -> starpc::Result<{}> {{",
+                    method_name, method.input_type, method.output_type
+                ));
+                self.line(&format!(
+                    "        self.client.exec_call({:?}, {:?}, request).await",
+                    service_id, method.proto_name
+                ));
+                self.line("    }");
+            }
+        }
+
+        self.line("}");
+        self.line("");
+    }
+
+    fn generate_stream_impls(&mut self) {
+        let service_name = self.service.name.clone();
+        for method in self.service.methods.clone() {
+            if !method.client_streaming && !method.server_streaming {
+                continue;
+            }
+
+            let stream_trait = stream_trait_name(&service_name, &method);
+
+            self.line(&format!("struct {}Impl {{", stream_trait));
+            self.line("    stream: Box<dyn starpc::Stream>,");
+            self.line("}");
+            self.line("");
+            self.line("#[starpc::async_trait]");
+            self.line(&format!(
+                "impl {} for {}Impl {{",
+                stream_trait, stream_trait
+            ));
+            self.line("    fn context(&self) -> &starpc::Context {");
+            self.line("        self.stream.context()");
+            self.line("    }");
+
+            if method.client_streaming {
+                self.line(&format!(
+                    "    async fn send(&self, msg: &{}) -> starpc::Result<()> {{",
+                    method.input_type
+                ));
+                self.line("        self.stream.msg_send(msg).await");
+                self.line("    }");
+            }
+
+            if method.server_streaming {
+                self.line(&format!(
+                    "    async fn recv(&self) -> starpc::Result<{}> {{",
+                    method.output_type
+                ));
+                self.line("        self.stream.msg_recv().await");
+                self.line("    }");
+            }
+
+            if method.client_streaming && !method.server_streaming {
+                self.line(&format!(
+                    "    async fn close_and_recv(&self) -> starpc::Result<{}> {{",
+                    method.output_type
+                ));
+                self.line("        self.stream.close_send().await?;");
+                self.line("        self.stream.msg_recv().await");
+                self.line("    }");
+            } else {
+                self.line("    async fn close(&self) -> starpc::Result<()> {");
+                self.line("        self.stream.close().await");
+                self.line("    }");
+            }
+
+            self.line("}");
+            self.line("");
+        }
+    }
+
+    fn generate_server_trait(&mut self) {
+        let service_name = self.service.name.clone();
+
+        self.line(&format!("/// Server trait for {}.", service_name));
+        self.line("#[starpc::async_trait]");
+        self.line(&format!("pub trait {}Server: Send + Sync {{", service_name));
+
+        for method in self.service.methods.clone() {
+            self.line(&format!("    /// {}.", method.proto_name));
+            if method.client_streaming && method.server_streaming {
+                self.line(&format!(
+                    "    async fn {}(&self, stream: Box<dyn starpc::Stream>) -> starpc::Result<()>;",
+                    method.name
+                ));
+            } else if method.server_streaming {
+                self.line(&format!(
+                    "    async fn {}(&self, request: {}, stream: Box<dyn starpc::Stream>) -> starpc::Result<()>;",
+                    method.name, method.input_type
+                ));
+            } else if method.client_streaming {
+                self.line(&format!(
+                    "    async fn {}(&self, stream: &dyn starpc::Stream) -> starpc::Result<{}>;",
+                    method.name, method.output_type
+                ));
+            } else {
+                self.line(&format!(
+                    "    async fn {}(&self, request: {}) -> starpc::Result<{}>;",
+                    method.name, method.input_type, method.output_type
+                ));
+            }
+        }
+
+        self.line("}");
+        self.line("");
+    }
+
+    fn generate_handler(&mut self) {
+        let service_id = self.service_id();
+        let service_name = self.service.name.clone();
+        let methods = self.service.methods.clone();
+        let method_ids = method_ids_const(&service_name);
+
+        self.line(&format!("const {}: &[&str] = &[", method_ids));
+        for method in &methods {
+            self.line(&format!("    {:?},", method.proto_name));
+        }
+        self.line("];");
+        self.line("");
+        self.line(&format!("/// Handler for {}.", service_name));
+        self.line(&format!(
+            "pub struct {}Handler<S: {}Server> {{",
+            service_name, service_name
+        ));
+        self.line("    server: std::sync::Arc<S>,");
+        self.line("}");
+        self.line("");
+        self.line(&format!(
+            "impl<S: {}Server + 'static> {}Handler<S> {{",
+            service_name, service_name
+        ));
+        self.line("    /// Creates a new handler wrapping the server implementation.");
+        self.line("    pub fn new(server: S) -> Self {");
+        self.line("        Self { server: std::sync::Arc::new(server) }");
+        self.line("    }");
+        self.line("");
+        self.line("    /// Creates a new handler with a shared server.");
+        self.line("    pub fn with_arc(server: std::sync::Arc<S>) -> Self {");
+        self.line("        Self { server }");
+        self.line("    }");
+        self.line("}");
+        self.line("");
+        self.line("#[starpc::async_trait]");
+        self.line(&format!(
+            "impl<S: {}Server + 'static> starpc::Invoker for {}Handler<S> {{",
+            service_name, service_name
+        ));
+        self.line("    async fn invoke_method(");
+        self.line("        &self,");
+        self.line("        _service_id: &str,");
+        self.line("        method_id: &str,");
+        self.line("        stream: Box<dyn starpc::Stream>,");
+        self.line("    ) -> (bool, starpc::Result<()>) {");
+        self.line("        match method_id {");
+
+        for method in &methods {
+            self.line(&format!("            {:?} => {{", method.proto_name));
+            if method.client_streaming && method.server_streaming {
+                self.line(&format!(
+                    "                (true, self.server.{}(stream).await)",
+                    method.name
+                ));
+            } else if method.server_streaming {
+                self.line(&format!(
+                    "                let request: {} = match stream.msg_recv().await {{",
+                    method.input_type
+                ));
+                self.line("                    Ok(r) => r,");
+                self.line("                    Err(e) => return (true, Err(e)),");
+                self.line("                };");
+                self.line(&format!(
+                    "                (true, self.server.{}(request, stream).await)",
+                    method.name
+                ));
+            } else if method.client_streaming {
+                self.line(&format!(
+                    "                match self.server.{}(stream.as_ref()).await {{",
+                    method.name
+                ));
+                self.line("                    Ok(response) => {");
+                self.line(
+                    "                        if let Err(e) = stream.msg_send(&response).await {",
+                );
+                self.line("                            return (true, Err(e));");
+                self.line("                        }");
+                self.line("                        (true, Ok(()))");
+                self.line("                    }");
+                self.line("                    Err(e) => (true, Err(e)),");
+                self.line("                }");
+            } else {
+                self.line(&format!(
+                    "                let request: {} = match stream.msg_recv().await {{",
+                    method.input_type
+                ));
+                self.line("                    Ok(r) => r,");
+                self.line("                    Err(e) => return (true, Err(e)),");
+                self.line("                };");
+                self.line(&format!(
+                    "                match self.server.{}(request).await {{",
+                    method.name
+                ));
+                self.line("                    Ok(response) => {");
+                self.line(
+                    "                        if let Err(e) = stream.msg_send(&response).await {",
+                );
+                self.line("                            return (true, Err(e));");
+                self.line("                        }");
+                self.line("                        (true, Ok(()))");
+                self.line("                    }");
+                self.line("                    Err(e) => (true, Err(e)),");
+                self.line("                }");
+            }
+            self.line("            }");
+        }
+
+        self.line("            _ => (false, Err(starpc::Error::Unimplemented)),");
+        self.line("        }");
+        self.line("    }");
+        self.line("}");
+        self.line("");
+        self.line(&format!(
+            "impl<S: {}Server + 'static> starpc::Handler for {}Handler<S> {{",
+            service_name, service_name
+        ));
+        self.line("    fn service_id(&self) -> &'static str {");
+        self.line(&format!("        {:?}", service_id));
+        self.line("    }");
+        self.line("");
+        self.line("    fn method_ids(&self) -> &'static [&'static str] {");
+        self.line(&format!("        {}", method_ids));
+        self.line("    }");
+        self.line("}");
+        self.line("");
+    }
+
+    fn service_id(&self) -> String {
+        if self.service.package.is_empty() {
+            self.service.proto_name.clone()
+        } else {
+            format!("{}.{}", self.service.package, self.service.proto_name)
+        }
+    }
+
+    fn line(&mut self, line: &str) {
+        self.buf.push_str(line);
+        self.buf.push('\n');
+    }
+}
+
+fn client_trait_method(service_name: &str, method: &prost_build::Method) -> String {
+    let stream_trait = stream_trait_name(service_name, method);
+    if method.client_streaming && method.server_streaming {
+        format!(
+            "async fn {}(&self) -> starpc::Result<Box<dyn {}>>",
+            method.name, stream_trait
+        )
+    } else if method.server_streaming {
+        format!(
+            "async fn {}(&self, request: &{}) -> starpc::Result<Box<dyn {}>>",
+            method.name, method.input_type, stream_trait
+        )
+    } else if method.client_streaming {
+        format!(
+            "async fn {}(&self) -> starpc::Result<Box<dyn {}>>",
+            method.name, stream_trait
+        )
+    } else {
+        format!(
+            "async fn {}(&self, request: &{}) -> starpc::Result<{}>",
+            method.name, method.input_type, method.output_type
+        )
+    }
+}
+
+fn stream_trait_name(service_name: &str, method: &prost_build::Method) -> String {
+    format!(
+        "{}{}Stream",
+        service_name,
+        upper_camel_from_snake(&method.name)
+    )
+}
+
+fn service_id_const(service_name: &str) -> String {
+    format!("{}_SERVICE_ID", screaming_snake(service_name))
+}
+
+fn method_ids_const(service_name: &str) -> String {
+    format!("{}_METHOD_IDS", screaming_snake(service_name))
+}
+
+fn screaming_snake(name: &str) -> String {
+    let mut out = String::new();
+    for (idx, ch) in name.chars().enumerate() {
+        if ch.is_uppercase() && idx != 0 {
+            out.push('_');
+        }
+        out.extend(ch.to_uppercase());
+    }
+    out
+}
+
+fn upper_camel_from_snake(name: &str) -> String {
+    let mut out = String::new();
+    let mut upper = true;
+    for ch in name.chars() {
+        if ch == '_' {
+            upper = true;
+            continue;
+        }
+        if upper {
+            out.extend(ch.to_uppercase());
+            upper = false;
+        } else {
+            out.push(ch);
+        }
+    }
+    out
+}
+
+#[cfg(test)]
+mod tests {
+    use std::fs;
+
+    #[test]
+    fn configure_generates_starpc_service_glue() {
+        let root = std::env::temp_dir().join(format!(
+            "starpc-codegen-test-{}-{}",
+            std::process::id(),
+            std::time::SystemTime::now()
+                .duration_since(std::time::UNIX_EPOCH)
+                .unwrap()
+                .as_nanos()
+        ));
+        let proto_dir = root.join("proto");
+        let out_dir = root.join("out");
+        let _ = fs::remove_dir_all(&root);
+        fs::create_dir_all(&proto_dir).unwrap();
+        fs::create_dir_all(&out_dir).unwrap();
+
+        let proto = proto_dir.join("test.proto");
+        fs::write(
+            &proto,
+            r#"syntax = "proto3";
+package fixture;
+
+service TestService {
+  rpc Unary(TestMsg) returns (TestMsg);
+  rpc ServerStream(TestMsg) returns (stream TestMsg);
+  rpc ClientStream(stream TestMsg) returns (TestMsg);
+  rpc Bidi(stream TestMsg) returns (stream TestMsg);
+}
+
+message TestMsg {
+  string body = 1;
+}
+"#,
+        )
+        .unwrap();
+
+        let mut config = super::configure();
+        config.out_dir(&out_dir);
+        config.compile_protos(&[proto], &[proto_dir]).unwrap();
+
+        let generated = fs::read_to_string(out_dir.join("fixture.rs")).unwrap();
+        assert!(generated.contains("pub const TEST_SERVICE_SERVICE_ID"));
+        assert!(generated.contains("pub trait TestServiceClient"));
+        assert!(generated.contains("pub struct TestServiceClientImpl"));
+        assert!(generated.contains("pub trait TestServiceServer"));
+        assert!(generated.contains("pub struct TestServiceHandler"));
+        assert!(generated.contains("TestServiceServerStreamStream"));
+        assert!(generated.contains("TestServiceClientStreamStream"));
+        assert!(generated.contains("TestServiceBidiStream"));
+
+        let _ = fs::remove_dir_all(&root);
+    }
+}

package/srpc/common-rpc.go

@@ -146,15 +146,22 @@ func (c *commonRPC) WriteCallData(data []byte, dataIsZero, complete bool, err er
 
 // HandleStreamClose handles the incoming stream closing w/ optional error.
 func (c *commonRPC) HandleStreamClose(closeErr error) {
+	var writer PacketWriter
 	c.bcast.HoldLock(func(broadcast func(), getWaitCh func() <-chan struct{}) {
+		if c.dataClosed {
+			return
+		}
 		if closeErr != nil && c.remoteErr == nil {
 			c.remoteErr = closeErr
 		}
 		c.dataClosed = true
 		c.ctxCancel()
-		_ = c.writer.Close()
+		writer = c.writer
 		broadcast()
 	})
+	if writer != nil {
+		_ = writer.Close()
+	}
 }
 
 // HandleCallCancel handles the call cancel packet.
@@ -210,6 +217,9 @@ func (c *commonRPC) WriteCallCancel() error {
 
 // closeLocked releases resources held by the RPC.
 func (c *commonRPC) closeLocked(broadcast func()) {
+	if c.dataClosed {
+		return
+	}
 	c.dataClosed = true
 	c.localCompleted.Store(true)
 	if c.remoteErr == nil {

package/srpc/common-rpc_test.go

@@ -0,0 +1,65 @@
+package srpc
+
+import (
+	"context"
+	"io"
+	"sync/atomic"
+	"testing"
+)
+
+type closeCountingPacketWriter struct {
+	closed atomic.Int32
+}
+
+type closeCallbackPacketWriter struct {
+	closeFn func()
+}
+
+func (w *closeCountingPacketWriter) WritePacket(*Packet) error {
+	return nil
+}
+
+func (w *closeCountingPacketWriter) Close() error {
+	w.closed.Add(1)
+	return nil
+}
+
+func (w *closeCallbackPacketWriter) WritePacket(*Packet) error {
+	return nil
+}
+
+func (w *closeCallbackPacketWriter) Close() error {
+	if w.closeFn != nil {
+		w.closeFn()
+	}
+	return nil
+}
+
+func TestCommonRPCHandleStreamCloseIdempotent(t *testing.T) {
+	writer := &closeCountingPacketWriter{}
+	rpc := NewServerRPC(context.Background(), InvokerFunc(nil), writer)
+
+	rpc.HandleStreamClose(io.EOF)
+	rpc.HandleStreamClose(context.Canceled)
+
+	if got := writer.closed.Load(); got != 1 {
+		t.Fatalf("expected writer closed once, got %d", got)
+	}
+}
+
+func TestCommonRPCHandleStreamCloseClosesWriterOutsideBroadcastLock(t *testing.T) {
+	var rpc *ServerRPC
+	writerClosedOutsideLock := false
+	writer := &closeCallbackPacketWriter{
+		closeFn: func() {
+			writerClosedOutsideLock = rpc.bcast.TryHoldLock(func(func(), func() <-chan struct{}) {})
+		},
+	}
+	rpc = NewServerRPC(context.Background(), InvokerFunc(nil), writer)
+
+	rpc.HandleStreamClose(io.EOF)
+
+	if !writerClosedOutsideLock {
+		t.Fatal("expected writer close outside broadcast lock")
+	}
+}

package/srpc/lib.rs

@@ -9,7 +9,7 @@
 //! - **Wire-compatible** with the Go and TypeScript implementations
 //! - **Streaming support** for all RPC patterns
 //! - **Transport agnostic** - works with TCP, WebSocket, or any AsyncRead/AsyncWrite
-//! - **Code generation** via starpc-build crate
+//! - **Code generation** via the optional `build` feature
 //!
 //! # Quick Start
 //!
@@ -73,6 +73,9 @@ pub mod stream;
 pub mod testing;
 pub mod transport;
 
+#[cfg(feature = "build")]
+pub mod build;
+
 // Re-exports for convenience.
 pub use client::{BoxClient, Client, OpenStream, SrpcClient};
 pub use codec::{PacketCodec, MAX_MESSAGE_SIZE};

package/srpc/packet-rw.go

@@ -5,14 +5,36 @@ import (
 	"context"
 	"encoding/binary"
 	"io"
-	"math"
 	"sync"
 
 	"github.com/pkg/errors"
 )
 
-// maxMessageSize is the max message size in bytes
-var maxMessageSize = 1e7
+const (
+	// maxMessageSize is the max message size in bytes.
+	maxMessageSize = 10_000_000
+	// readBufferSize is the packet read scratch buffer size.
+	readBufferSize = 2048
+	// pooledWriteBufferMaxSize is the largest outbound frame buffer to pool.
+	pooledWriteBufferMaxSize = 64 * 1024
+)
+
+var (
+	readBufferPool = sync.Pool{
+		New: func() any {
+			return new([readBufferSize]byte)
+		},
+	}
+	writeBufferPool = sync.Pool{
+		New: func() any {
+			return new(writeBuffer)
+		},
+	}
+)
+
+type writeBuffer struct {
+	data []byte
+}
 
 // PacketReadWriter reads and writes packets from a io.ReadWriter.
 // Uses a LittleEndian uint32 length prefix.
@@ -43,13 +65,13 @@ func (r *PacketReadWriter) WritePacket(p *Packet) error {
 	defer r.writeMtx.Unlock()
 
 	msgSize := p.SizeVT()
-
-	// G115: integer overflow conversion int -> uint32 (gosec)
-	if uint64(msgSize) > uint64(math.MaxUint32) {
-		return errors.New("message size exceeds maximum uint32 value")
+	if msgSize < 0 || msgSize > maxMessageSize {
+		return errors.Errorf("message size %v greater than maximum %v", msgSize, maxMessageSize)
 	}
 
-	data := make([]byte, 4+msgSize)
+	writeBuf := getWriteBuffer(4 + msgSize)
+	defer putWriteBuffer(writeBuf)
+	data := writeBuf.data
 	binary.LittleEndian.PutUint32(data, uint32(msgSize)) //nolint:gosec
 
 	_, err := p.MarshalToSizedBufferVT(data[4:])
@@ -59,10 +81,13 @@ func (r *PacketReadWriter) WritePacket(p *Packet) error {
 
 	var written, n int
 	for written < len(data) {
-		n, err = r.rw.Write(data)
+		n, err = r.rw.Write(data[written:])
 		if err != nil {
 			return err
 		}
+		if n == 0 {
+			return io.ErrShortWrite
+		}
 		written += n
 	}
 
@@ -84,7 +109,9 @@ func (r *PacketReadWriter) ReadPump(cb PacketDataHandler, closed CloseHandler) {
 // Does not handle closing the stream, use ReadPump instead.
 func (r *PacketReadWriter) ReadToHandler(cb PacketDataHandler) error {
 	var currLen uint32
-	buf := make([]byte, 2048)
+	bufPtr := readBufferPool.Get().(*[readBufferSize]byte)
+	defer readBufferPool.Put(bufPtr)
+	buf := bufPtr[:]
 	isOpen := true
 
 	for isOpen {
@@ -153,5 +180,25 @@ func (r *PacketReadWriter) readLengthPrefix(b []byte) uint32 {
 	return binary.LittleEndian.Uint32(b)
 }
 
+func getWriteBuffer(size int) *writeBuffer {
+	if size > pooledWriteBufferMaxSize {
+		return &writeBuffer{data: make([]byte, size)}
+	}
+	buf := writeBufferPool.Get().(*writeBuffer)
+	if cap(buf.data) < size {
+		buf.data = make([]byte, size)
+	}
+	buf.data = buf.data[:size]
+	return buf
+}
+
+func putWriteBuffer(buf *writeBuffer) {
+	if cap(buf.data) <= pooledWriteBufferMaxSize {
+		clear(buf.data)
+		buf.data = buf.data[:0]
+		writeBufferPool.Put(buf)
+	}
+}
+
 // _ is a type assertion
 var _ PacketWriter = (*PacketReadWriter)(nil)

package/srpc/packet-rw_test.go

@@ -0,0 +1,68 @@
+package srpc
+
+import (
+	"bytes"
+	"encoding/binary"
+	"io"
+	"testing"
+)
+
+type chunkedReadWriteCloser struct {
+	bytes.Buffer
+	maxWrite int
+}
+
+func (c *chunkedReadWriteCloser) Read([]byte) (int, error) {
+	return 0, io.EOF
+}
+
+func (c *chunkedReadWriteCloser) Write(p []byte) (int, error) {
+	if len(p) > c.maxWrite {
+		p = p[:c.maxWrite]
+	}
+	return c.Buffer.Write(p)
+}
+
+func (c *chunkedReadWriteCloser) Close() error {
+	return nil
+}
+
+func TestPacketReadWriterWritePacketHandlesShortWrites(t *testing.T) {
+	pkt := NewCallDataPacket([]byte("packet payload"), false, true, nil)
+	size := pkt.SizeVT()
+	want := make([]byte, 4+size)
+	binary.LittleEndian.PutUint32(want, uint32(size)) //nolint:gosec
+	if _, err := pkt.MarshalToSizedBufferVT(want[4:]); err != nil {
+		t.Fatal(err)
+	}
+
+	rwc := &chunkedReadWriteCloser{maxWrite: 3}
+	if err := NewPacketReadWriter(rwc).WritePacket(pkt); err != nil {
+		t.Fatal(err)
+	}
+	if got := rwc.Bytes(); !bytes.Equal(got, want) {
+		t.Fatalf("written packet mismatch:\ngot  %x\nwant %x", got, want)
+	}
+}
+
+func TestPacketUnmarshalCopiesByteFields(t *testing.T) {
+	want := []byte("stable data")
+	srcPkt := NewCallDataPacket(want, false, true, nil)
+	data, err := srcPkt.MarshalVT()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	var pkt Packet
+	if err := pkt.UnmarshalVT(data); err != nil {
+		t.Fatal(err)
+	}
+	for i := range data {
+		data[i] = 0xff
+	}
+
+	got := pkt.GetCallData().GetData()
+	if !bytes.Equal(got, want) {
+		t.Fatalf("unmarshal retained source bytes: got %q want %q", got, want)
+	}
+}