PostMark: Implement the postmark attribute in the cc_binary rule.

This is not available in Bazel at the moment.

PiperOrigin-RevId: 825562421
Change-Id: Ib77c80eca0f76c4177e5ed7741846e65dacf46f7

Author: bigelephant29

cc/private/rules_impl/BUILD

@@ -40,6 +40,7 @@ bzl_library(
         "cc_binary.bzl",
         "cc_import.bzl",
         "cc_library.bzl",
+        "cc_postmark.bzl",
         "cc_shared_library.bzl",
         "cc_static_library.bzl",
         "cc_test.bzl",

cc/private/rules_impl/attrs.bzl

@@ -17,6 +17,7 @@
 
 load("//cc/common:cc_info.bzl", "CcInfo")
 load("//cc/common:semantics.bzl", "semantics")
+load("//cc/private/rules_impl:cc_postmark.bzl", "postmark")
 load(":cc_shared_library.bzl", "dynamic_deps_attrs")
 
 visibility("private")
@@ -293,7 +294,7 @@ Whether to encode build information into the binary. Possible values:
 """ + semantics.stamp_extra_docs
 
 # buildifier: disable=attr-licenses
-cc_binary_attrs = common_attrs | {
+cc_binary_attrs = common_attrs | postmark.get_attrs() | {
     "deps": attr.label_list(
         allow_files = semantics.ALLOWED_FILES_IN_DEPS,
         allow_rules = semantics.ALLOWED_RULES_IN_DEPS + semantics.ALLOWED_RULES_WITH_WARNINGS_IN_DEPS,

cc/private/rules_impl/cc_binary.bzl

@@ -22,6 +22,7 @@ load("//cc/common:cc_info.bzl", "CcInfo")
 load("//cc/common:debug_package_info.bzl", "DebugPackageInfo")
 load("//cc/common:semantics.bzl", "semantics")
 load(":attrs.bzl", "cc_binary_attrs")
+load(":cc_postmark.bzl", "postmark")
 load(":cc_shared_library.bzl", "GraphNodeInfo", "add_unused_dynamic_deps", "build_exports_map_from_only_dynamic_deps", "build_link_once_static_libs_map", "dynamic_deps_initializer", "merge_cc_shared_library_infos", "separate_static_and_dynamic_link_libraries", "sort_linker_inputs", "throw_linked_but_not_exported_errors")
 
 _CcLauncherInfo = cc_common.launcher_provider
@@ -307,7 +308,8 @@ def _create_transitive_linking_actions(
         additional_linkopts,
         additional_make_variable_substitutions,
         link_variables,
-        additional_outputs):
+        additional_outputs,
+        stamp):
     cc_compilation_outputs_with_only_objects = cc_common.create_compilation_outputs(objects = None, pic_objects = None)
     deps_cc_info = CcInfo(linking_context = deps_cc_linking_context)
     libraries_for_current_cc_linking_context = []
@@ -381,7 +383,7 @@ def _create_transitive_linking_actions(
         feature_configuration = feature_configuration,
         cc_toolchain = cc_toolchain,
         compilation_outputs = cc_compilation_outputs_with_only_objects,
-        stamp = cc_helper.is_stamping_enabled(ctx),
+        stamp = int(stamp),
         additional_inputs = additional_linker_inputs,
         linking_contexts = [cc_linking_context],
         name = ctx.label.name,
@@ -633,6 +635,16 @@ def cc_binary_impl(ctx, additional_linkopts, force_linkstatic = False):
     if extra_link_time_libraries != None:
         linker_inputs_extra, runtime_libraries_extra = cc_common.build_extra_link_time_libraries(extra_libraries = extra_link_time_libraries, ctx = ctx, static_mode = linking_mode != linker_mode.LINKING_DYNAMIC, for_dynamic_library = _is_link_shared(ctx))
 
+    use_postmark = postmark.get_use_postmark(ctx)
+    output_binary_for_linking = binary
+    link_stamp = cc_helper.is_stamping_enabled(ctx)
+    if use_postmark:
+        output_binary_for_linking = ctx.actions.declare_file(
+            "unstamped_" + binary.basename,
+            sibling = binary,
+        )
+        link_stamp = False
+
     cc_linking_outputs_binary, cc_launcher_info, deps_cc_linking_context = _create_transitive_linking_actions(
         ctx,
         cc_toolchain,
@@ -641,7 +653,7 @@ def cc_binary_impl(ctx, additional_linkopts, force_linkstatic = False):
         cc_compilation_outputs,
         additional_linker_inputs,
         cc_linking_outputs,
-        binary,
+        output_binary_for_linking,
         deps_cc_linking_context,
         linker_inputs_extra,
         link_compile_output_separately,
@@ -651,8 +663,17 @@ def cc_binary_impl(ctx, additional_linkopts, force_linkstatic = False):
         additional_make_variable_substitutions,
         link_variables,
         additional_linker_outputs,
+        stamp = link_stamp,
     )
 
+    if use_postmark:
+        postmark.add_action(
+            ctx,
+            cc_toolchain,
+            binary,
+            output_binary_for_linking,
+        )
+
     cc_linking_outputs_binary_library = cc_linking_outputs_binary.library_to_link
     libraries = []
     if _is_link_shared(ctx) and cc_linking_outputs_binary_library != None:
@@ -809,6 +830,10 @@ ALLOWED_SRC_FILES.extend(cc_helper.extensions.SHARED_LIBRARY)
 ALLOWED_SRC_FILES.extend(cc_helper.extensions.OBJECT_FILE)
 ALLOWED_SRC_FILES.extend(cc_helper.extensions.PIC_OBJECT_FILE)
 
+def _cc_binary_initializer(**kwargs):
+    kwargs = postmark.initializer(**kwargs)
+    return dynamic_deps_initializer(**kwargs)
+
 def _impl(ctx):
     binary_info, providers = cc_binary_impl(ctx, [])
 
@@ -831,7 +856,7 @@ def _impl(ctx):
 
 cc_binary = rule(
     implementation = _impl,
-    initializer = dynamic_deps_initializer,
+    initializer = _cc_binary_initializer,
     doc = """
 <p>It produces an executable binary.</p>
 

cc/private/rules_impl/cc_postmark.bzl

@@ -0,0 +1,38 @@
+# Copyright 2025 The Bazel Authors. All rights reserved.
+#
+# 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.
+
+"""PostMark is not implemented in Bazel."""
+
+def _get_postmark_attrs():
+    return {}
+
+def _postmark_initializer(**kwargs):
+    return kwargs
+
+def _add_postmark_action(
+        ctx,  # @unused
+        cc_toolchain,  # @unused
+        binary,  # @unused
+        output_binary_for_linking):  # @unused
+    pass
+
+def _get_use_postmark(ctx):  # @unused
+    return False
+
+postmark = struct(
+    get_attrs = _get_postmark_attrs,
+    add_action = _add_postmark_action,
+    get_use_postmark = _get_use_postmark,
+    initializer = _postmark_initializer,
+)

cc/private/rules_impl/cc_test.bzl

@@ -20,6 +20,7 @@ load("//cc/common:cc_info.bzl", "CcInfo")
 load("//cc/common:semantics.bzl", "semantics")
 load(":attrs.bzl", "cc_binary_attrs", "linkstatic_doc", "stamp_doc")
 load(":cc_binary.bzl", "cc_binary_impl")
+load(":cc_postmark.bzl", "postmark")
 load(":cc_shared_library.bzl", "dynamic_deps_initializer")
 
 _CC_TEST_TOOLCHAIN_TYPE = "@bazel_tools//tools/cpp:test_runner_toolchain_type"
@@ -105,10 +106,14 @@ def cc_test_initializer(**kwargs):
 
     Args:
         **kwargs: Arguments suitable for cc_test.
+
+    Returns:
+        Arguments suitable for cc_test.
     """
 
     if "linkstatic" not in kwargs:
         kwargs["linkstatic"] = semantics.get_linkstatic_default_for_test()
+    kwargs = postmark.initializer(**kwargs)
 
     return dynamic_deps_initializer(**kwargs)