如何在 Python 中调用 .so 文件中的方法

#include <stdio.h>

// 一个简单的加法函数
int add(int a, int b) {
    return a + b;
}

// 一个打印问候语的函数
void greet(const char* name) {
    printf("Hello, %s!\n", name);
}

gcc -shared -o mylib.so mylib.c

import ctypes
import os

# 1. 加载 .so 文件
# 确保 mylib.so 文件与你的 Python 脚本在同一个目录下，
# 或者提供完整的路径。
current_dir = os.path.dirname(os.path.abspath(__file__))
lib_path = os.path.join(current_dir, 'mylib.so')

try:
    my_lib = ctypes.CDLL(lib_path)
except OSError as e:
    print(f"Error loading library: {e}")
    print("Please ensure 'mylib.so' is in the same directory as this script or provide the full path.")
    exit()

# 2. 调用 add 函数
# 默认情况下，ctypes 假定函数返回 int 类型，并且参数是 int 类型。
# 但是，为了明确和避免潜在错误，最好显式指定参数类型和返回类型。

# 设置 add 函数的参数类型和返回类型
my_lib.add.argtypes = [ctypes.c_int, ctypes.c_int]
my_lib.add.restype = ctypes.c_int

result = my_lib.add(5, 3)
print(f"5 + 3 = {result}")  # 输出: 5 + 3 = 8

# 3. 调用 greet 函数
# greet 函数接受一个字符串（C 中的 const char*）作为参数，并且没有返回值 (void)。

# 设置 greet 函数的参数类型和返回类型
# 对于字符串参数，使用 ctypes.c_char_p
# 对于 void 返回值，restype 不需要设置或设置为 None
my_lib.greet.argtypes = [ctypes.c_char_p]
my_lib.greet.restype = None  # 或者不设置这行

# Python 字符串需要编码为字节串 (bytes) 才能传递给 C 函数
name = "Pythonista"
my_lib.greet(name.encode('utf-8')) # 输出: Hello, Pythonista!

# 尝试调用不存在的函数（会报错）
# try:
#     my_lib.non_existent_function()
# except AttributeError as e:
#     print(f"\nError: {e}")