mirror of
https://github.com/LLVMParty/llvm-nanobind
synced 2026-06-21 13:43:38 +00:00
Duncan Ogilvie
102 lines
3.3 KiB
C++
102 lines
3.3 KiB
C++
/**
|
|
* Test: test_module
|
|
* Tests LLVM Module creation and properties
|
|
*
|
|
* LLVM-C APIs covered:
|
|
* - LLVMModuleCreateWithNameInContext()
|
|
* - LLVMGetModuleIdentifier() / LLVMSetModuleIdentifier()
|
|
* - LLVMGetSourceFileName() / LLVMSetSourceFileName()
|
|
* - LLVMGetDataLayoutStr() / LLVMSetDataLayout()
|
|
* - LLVMGetTarget() / LLVMSetTarget()
|
|
* - LLVMPrintModuleToString()
|
|
* - LLVMCloneModule()
|
|
* - LLVMDisposeModule()
|
|
*/
|
|
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <llvm-c/Analysis.h>
|
|
#include <llvm-c/Core.h>
|
|
#include <string>
|
|
|
|
int main() {
|
|
LLVMContextRef ctx = LLVMContextCreate();
|
|
|
|
// Create module
|
|
LLVMModuleRef mod = LLVMModuleCreateWithNameInContext("test_module", ctx);
|
|
|
|
// Get initial module identifier (copy to string before modifying)
|
|
size_t id_len;
|
|
const char *id_ptr = LLVMGetModuleIdentifier(mod, &id_len);
|
|
std::string initial_id(id_ptr, id_len);
|
|
|
|
// Set new module identifier
|
|
LLVMSetModuleIdentifier(mod, "renamed_module", strlen("renamed_module"));
|
|
id_ptr = LLVMGetModuleIdentifier(mod, &id_len);
|
|
std::string new_id(id_ptr, id_len);
|
|
|
|
// Get/set source filename (copy to string before modifying)
|
|
size_t src_len;
|
|
const char *src_ptr = LLVMGetSourceFileName(mod, &src_len);
|
|
std::string initial_src(src_ptr, src_len);
|
|
LLVMSetSourceFileName(mod, "test_source.c", strlen("test_source.c"));
|
|
src_ptr = LLVMGetSourceFileName(mod, &src_len);
|
|
std::string new_src(src_ptr, src_len);
|
|
|
|
// Get/set data layout (copy to string before modifying)
|
|
std::string initial_layout = LLVMGetDataLayoutStr(mod);
|
|
LLVMSetDataLayout(
|
|
mod,
|
|
"e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128");
|
|
std::string new_layout = LLVMGetDataLayoutStr(mod);
|
|
|
|
// Get/set target triple (copy to string before modifying)
|
|
std::string initial_target = LLVMGetTarget(mod);
|
|
LLVMSetTarget(mod, "x86_64-unknown-linux-gnu");
|
|
std::string new_target = LLVMGetTarget(mod);
|
|
|
|
// Clone module
|
|
LLVMModuleRef cloned = LLVMCloneModule(mod);
|
|
id_ptr = LLVMGetModuleIdentifier(cloned, &id_len);
|
|
std::string cloned_id(id_ptr, id_len);
|
|
|
|
// Verify module
|
|
char *error = nullptr;
|
|
if (LLVMVerifyModule(mod, LLVMReturnStatusAction, &error)) {
|
|
fprintf(stderr, "; Verification failed: %s\n", error);
|
|
LLVMDisposeMessage(error);
|
|
LLVMDisposeModule(cloned);
|
|
LLVMDisposeModule(mod);
|
|
LLVMContextDispose(ctx);
|
|
return 1;
|
|
}
|
|
LLVMDisposeMessage(error);
|
|
|
|
// Print diagnostic comments
|
|
printf("; Test: test_module\n");
|
|
printf("; Initial module ID: %s\n", initial_id.c_str());
|
|
printf("; New module ID: %s\n", new_id.c_str());
|
|
printf("; Initial source filename: %s\n", initial_src.c_str());
|
|
printf("; New source filename: %s\n", new_src.c_str());
|
|
printf("; Initial data layout: %s\n",
|
|
initial_layout.empty() ? "(empty)" : initial_layout.c_str());
|
|
printf("; New data layout: %s\n", new_layout.c_str());
|
|
printf("; Initial target: %s\n",
|
|
initial_target.empty() ? "(empty)" : initial_target.c_str());
|
|
printf("; New target: %s\n", new_target.c_str());
|
|
printf("; Cloned module ID: %s\n", cloned_id.c_str());
|
|
printf("\n");
|
|
|
|
// Print module IR
|
|
char *ir = LLVMPrintModuleToString(mod);
|
|
printf("%s", ir);
|
|
LLVMDisposeMessage(ir);
|
|
|
|
// Cleanup
|
|
LLVMDisposeModule(cloned);
|
|
LLVMDisposeModule(mod);
|
|
LLVMContextDispose(ctx);
|
|
|
|
return 0;
|
|
}
|