Per-cpu变量动静态申请即使用

Per-cpu变量动静态申请即简单使用举例

PS:

• per-cpu变量，是一个专业名词，不是一个变量，更不是一个变量类型，指一类变量，就像自旋锁不是一个锁，而是指代这一类的锁。
• 当建立一个per-CPU变量时，系统中每个处理器都会拥有该变量的特有副本。
• 可避免一部分竞态，减少使用锁，但无法避免抢占造成的竞态。
• 经常见到void __percpu * my_percpu，此处__percpu只是告诉编译器用。

API：

• per-cpu相关的声明可见<linux/percpu.h>
• 静态:
  • 创建：
    • DEFINE_PER_CPU(type, name);
  • 使用：以下函数禁止了抢占
    • get_cpu_var(name);
    • put_cpu_var(name);
• 动态：
  • 创建：
    • void *alloc_percpu(type);
    • void *__alloc_percpu(size_t size, size_t align); /* 可字节对齐 */
  • 销毁：
    • free_percpu(void *per_cpu_var);
  • 使用：
    • per_cpu_ptr(void *per_cpu_var, int cpu_id);
• 变量导出给模块：
  • EXPORT_PER_CPU_SYMBOL(per_cpu_var);
  • EXPORT_PER_CPU_SYMBOL_GPL(per_cpu_var);

EXP:

• 静态：

#include <linux/percpu.h>
/* 每个CPU都拥有int sockets_in_use */
DEFINE_PER_CPU(int, sockets_in_use);	

/* 只获取并修改当前CPU的int sockets_in_use */
get_cpu_var(sockets_in_use)++;
put_cpu_var(sockets_in_use);

• 动态：

#include <linux/percpu.h>

int *A;
int *B;
/*
 * 每个CPU都拥有int大小的空间
 * A并不一定是当前CPU拥有的
 * int空间的地址，因此不直接调用。
 */
A = alloc_percpu(int);
if (!A)
	return -ENOMEM;
/* get_cpu() & put_cpu(),内部实现了禁止抢占 */
B = per_cpu_ptr(A, get_cpu());
*B = 1;
put_cpu();

/* 验证为什么不能直接使用A */
printk(KERN_INFO "A = %p\n", A);
printk(KERN_INFO "B = %p\n", B);

free_percpu(A);

• 结构体中包含per-cpu变量：

#include <linux/percpu.h>

typedef struct port_stats {
    ... ....
    long __percpu **stats;
}

long **stats;
struct port_stats pstats;

pstats.stats = alloc_percpu(long *);
if (!pstats.stats)
    return -ENOMEM;

stats = per_cpu_ptr(pstats->stats, get_cpu());

/* Read or write *stats, the per-cpu (long *) value for this cpu */

put_cpu();

free_percpu(pstats->stats);

IMG：