[llvm] [MC][RISCV] Check hasEmitNops before call shouldInsertExtraNopBytesForCodeAlign (PR #77236)

[Jinyang He via llvm-commits]

https://github.com/MQ-mengqing created https://github.com/llvm/llvm-project/pull/77236

The shouldInsertExtraNopBytesForCodeAlign() need STI to check whether relax is enabled or not. It is initialized when call setEmitNops. The setEmitNops may not be called in a section which has instructions but is not executable. In this case uninitialized STI will cause problems. Thus, check hasEmitNops before call it.

Fixes: https://github.com/llvm/llvm-project/pull/76552#issuecomment-1878952480

>From 3ae28f8b40dcd957180985db2dd687d393282af2 Mon Sep 17 00:00:00 2001
From: Jinyang He <hejinyang at loongson.cn>
Date: Sun, 7 Jan 2024 20:31:55 +0800
Subject: [PATCH] [MC][RISCV] Check hasEmitNops before call
 shouldInsertExtraNopBytesForCodeAlign

The shouldInsertExtraNopBytesForCodeAlign() need STI to check whether
relax is enabled or not. It is initialized when call setEmitNops. The
setEmitNops may not be called in a section which has instructions but
is not executable. In this case uninitialized STI will cause problems.
Thus, check hasEmitNops before call it.

Fixes: https://github.com/llvm/llvm-project/pull/76552#issuecomment-1878952480
---
 llvm/lib/MC/MCExpr.cpp                    |  2 +-
 llvm/test/MC/RISCV/align-non-executable.s | 29 +++++++++++++++++++++++
 2 files changed, 30 insertions(+), 1 deletion(-)
 create mode 100644 llvm/test/MC/RISCV/align-non-executable.s

diff --git a/llvm/lib/MC/MCExpr.cpp b/llvm/lib/MC/MCExpr.cpp
index 9dae026535ccfb..80def6dfc24b1a 100644
--- a/llvm/lib/MC/MCExpr.cpp
+++ b/llvm/lib/MC/MCExpr.cpp
@@ -708,7 +708,7 @@ static void AttemptToFoldSymbolOffsetDifference(
       if (DF) {
         Displacement += DF->getContents().size();
       } else if (auto *AF = dyn_cast<MCAlignFragment>(FI);
-                 AF && Layout &&
+                 AF && Layout && AF->hasEmitNops() &&
                  !Asm->getBackend().shouldInsertExtraNopBytesForCodeAlign(
                      *AF, Count)) {
         Displacement += Asm->computeFragmentSize(*Layout, *AF);
diff --git a/llvm/test/MC/RISCV/align-non-executable.s b/llvm/test/MC/RISCV/align-non-executable.s
new file mode 100644
index 00000000000000..a5bccf360729e1
--- /dev/null
+++ b/llvm/test/MC/RISCV/align-non-executable.s
@@ -0,0 +1,29 @@
+# RUN: llvm-mc --filetype=obj --triple=riscv64 --mattr=+relax %s \
+# RUN:     | llvm-readobj -r - | FileCheck --check-prefixes=CHECK,RELAX %s
+# RUN: llvm-mc --filetype=obj --triple=riscv64 --mattr=-relax %s \
+# RUN:     | llvm-readobj -r - | FileCheck %s
+
+.section ".dummy", "a"
+.L1:
+  call func
+.p2align 3
+.L2:
+.dword .L2 - .L1
+.word .L2 - .L1
+.half .L2 - .L1
+.byte .L2 - .L1
+
+# CHECK:       Relocations [
+# CHECK-NEXT:    Section ({{.*}}) .rela.dummy {
+# CHECK-NEXT:      0x0 R_RISCV_CALL_PLT func 0x0
+# RELAX-NEXT:      0x0 R_RISCV_RELAX - 0x0
+# CHECK-NEXT:      0x8 R_RISCV_ADD64 .L2 0x0
+# CHECK-NEXT:      0x8 R_RISCV_SUB64 .L1 0x0
+# CHECK-NEXT:      0x10 R_RISCV_ADD32 .L2 0x0
+# CHECK-NEXT:      0x10 R_RISCV_SUB32 .L1 0x0
+# CHECK-NEXT:      0x14 R_RISCV_ADD16 .L2 0x0
+# CHECK-NEXT:      0x14 R_RISCV_SUB16 .L1 0x0
+# CHECK-NEXT:      0x16 R_RISCV_ADD8 .L2 0x0
+# CHECK-NEXT:      0x16 R_RISCV_SUB8 .L1 0x0
+# CHECK-NEXT:    }
+# CHECK-NEXT:  ]