rust-sdk: fix codegen for cross-crate proto extensions (#5646)

Author: dreveman

contrib/rust-sdk/Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 edition = "2024"
 name = "perfetto-sdk-protos-gpu"
-version = "0.3.7"
+version = "0.3.8"
 authors = ["David Reveman <reveman@meta.com>"]
 description = "Extra protobuf bindings for GPU events"
 readme = "README.md"

contrib/rust-sdk/perfetto-protos-gpu/Cargo.toml

@@ -0,0 +1,27 @@
+// Copyright (C) 2025 Rivos Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Autogenerated by the ProtoZero Rust compiler plugin.
+// Invoked by contrib/rust-sdk/tools/gen_rust_protos
+// DO NOT EDIT.
+
+use crate::pb_msg;
+use crate::pb_msg_ext;
+use crate::protos::trace::gpu::gpu_render_stage_event::*;
+use perfetto_sdk::protos::trace::interned_data::interned_data::*;
+
+pb_msg_ext!(InternedData {
+    compute_kernels: InternedComputeKernel, msg, 1000,
+    compute_arg_names: InternedComputeArgName, msg, 1001,
+});

contrib/rust-sdk/perfetto-protos-gpu/src/protos/trace/gpu/gpu_interned_data.pz.rs

@@ -19,6 +19,10 @@
 #[path = "gpu_counter_event.pz.rs"]
 pub mod gpu_counter_event;
 
+/// `gpu_interned_data` protos.
+#[path = "gpu_interned_data.pz.rs"]
+pub mod gpu_interned_data;
+
 /// `gpu_log` protos.
 #[path = "gpu_log.pz.rs"]
 pub mod gpu_log;

contrib/rust-sdk/perfetto-protos-gpu/src/protos/trace/gpu/mod.rs

@@ -80,6 +80,7 @@ SOURCE_FILES = [
             "protos/perfetto/config/gpu/gpu_renderstages_config.proto",
             "protos/perfetto/config/gpu/vulkan_memory_config.proto",
             "protos/perfetto/trace/gpu/gpu_counter_event.proto",
+            "protos/perfetto/trace/gpu/gpu_interned_data.proto",
             "protos/perfetto/trace/gpu/gpu_log.proto",
             "protos/perfetto/trace/gpu/gpu_render_stage_event.proto",
             "protos/perfetto/trace/gpu/gpu_track_event.proto",
@@ -263,7 +264,7 @@ def main():
               path_strip_prefix=sources["path_strip_prefix"],
               path_add_prefix=sources["path_add_prefix"],
               external_crate=sources.get("external_crate"),
-              local_files=all_files,
+              local_files=sources["files"],
           )
 
           tmpfilename = os.path.join(tmpdirname, transform_extension(source))

contrib/rust-sdk/tools/gen_rust_protos

@@ -376,7 +376,7 @@ class GeneratorJob {
     if (!enums_.empty()) {
       stub_rs_->Print("use crate::pb_enum;\n");
     }
-    if (!messages_.empty()) {
+    if (!messages_.empty() || !extensions_.empty()) {
       stub_rs_->Print("use crate::pb_msg;\n");
     }
     if (!extensions_.empty()) {