1 /* 2 * cpu_rmap.c: CPU affinity reverse-map support 3 * Copyright 2011 Solarflare Communications Inc. 4 * 5 * This program is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 as published 7 * by the Free Software Foundation, incorporated herein by reference. 8 */ 9 10 #include <linux/cpu_rmap.h> 11 #include <linux/interrupt.h> 12 #include <linux/export.h> 13 14 /* 15 * These functions maintain a mapping from CPUs to some ordered set of 16 * objects with CPU affinities. This can be seen as a reverse-map of 17 * CPU affinity. However, we do not assume that the object affinities 18 * cover all CPUs in the system. For those CPUs not directly covered 19 * by object affinities, we attempt to find a nearest object based on 20 * CPU topology. 21 */ 22 23 /** 24 * alloc_cpu_rmap - allocate CPU affinity reverse-map 25 * @size: Number of objects to be mapped 26 * @flags: Allocation flags e.g. %GFP_KERNEL 27 */ 28 struct cpu_rmap *alloc_cpu_rmap(unsigned int size, gfp_t flags) 29 { 30 struct cpu_rmap *rmap; 31 unsigned int cpu; 32 size_t obj_offset; 33 34 /* This is a silly number of objects, and we use u16 indices. */ 35 if (size > 0xffff) 36 return NULL; 37 38 /* Offset of object pointer array from base structure */ 39 obj_offset = ALIGN(offsetof(struct cpu_rmap, near[nr_cpu_ids]), 40 sizeof(void *)); 41 42 rmap = kzalloc(obj_offset + size * sizeof(rmap->obj[0]), flags); 43 if (!rmap) 44 return NULL; 45 46 kref_init(&rmap->refcount); 47 rmap->obj = (void **)((char *)rmap + obj_offset); 48 49 /* Initially assign CPUs to objects on a rota, since we have 50 * no idea where the objects are. Use infinite distance, so 51 * any object with known distance is preferable. Include the 52 * CPUs that are not present/online, since we definitely want 53 * any newly-hotplugged CPUs to have some object assigned. 54 */ 55 for_each_possible_cpu(cpu) { 56 rmap->near[cpu].index = cpu % size; 57 rmap->near[cpu].dist = CPU_RMAP_DIST_INF; 58 } 59 60 rmap->size = size; 61 return rmap; 62 } 63 EXPORT_SYMBOL(alloc_cpu_rmap); 64 65 /** 66 * cpu_rmap_release - internal reclaiming helper called from kref_put 67 * @ref: kref to struct cpu_rmap 68 */ 69 static void cpu_rmap_release(struct kref *ref) 70 { 71 struct cpu_rmap *rmap = container_of(ref, struct cpu_rmap, refcount); 72 kfree(rmap); 73 } 74 75 /** 76 * cpu_rmap_get - internal helper to get new ref on a cpu_rmap 77 * @rmap: reverse-map allocated with alloc_cpu_rmap() 78 */ 79 static inline void cpu_rmap_get(struct cpu_rmap *rmap) 80 { 81 kref_get(&rmap->refcount); 82 } 83 84 /** 85 * cpu_rmap_put - release ref on a cpu_rmap 86 * @rmap: reverse-map allocated with alloc_cpu_rmap() 87 */ 88 int cpu_rmap_put(struct cpu_rmap *rmap) 89 { 90 return kref_put(&rmap->refcount, cpu_rmap_release); 91 } 92 EXPORT_SYMBOL(cpu_rmap_put); 93 94 /* Reevaluate nearest object for given CPU, comparing with the given 95 * neighbours at the given distance. 96 */ 97 static bool cpu_rmap_copy_neigh(struct cpu_rmap *rmap, unsigned int cpu, 98 const struct cpumask *mask, u16 dist) 99 { 100 int neigh; 101 102 for_each_cpu(neigh, mask) { 103 if (rmap->near[cpu].dist > dist && 104 rmap->near[neigh].dist <= dist) { 105 rmap->near[cpu].index = rmap->near[neigh].index; 106 rmap->near[cpu].dist = dist; 107 return true; 108 } 109 } 110 return false; 111 } 112 113 #ifdef DEBUG 114 static void debug_print_rmap(const struct cpu_rmap *rmap, const char *prefix) 115 { 116 unsigned index; 117 unsigned int cpu; 118 119 pr_info("cpu_rmap %p, %s:\n", rmap, prefix); 120 121 for_each_possible_cpu(cpu) { 122 index = rmap->near[cpu].index; 123 pr_info("cpu %d -> obj %u (distance %u)\n", 124 cpu, index, rmap->near[cpu].dist); 125 } 126 } 127 #else 128 static inline void 129 debug_print_rmap(const struct cpu_rmap *rmap, const char *prefix) 130 { 131 } 132 #endif 133 134 /** 135 * cpu_rmap_add - add object to a rmap 136 * @rmap: CPU rmap allocated with alloc_cpu_rmap() 137 * @obj: Object to add to rmap 138 * 139 * Return index of object. 140 */ 141 int cpu_rmap_add(struct cpu_rmap *rmap, void *obj) 142 { 143 u16 index; 144 145 BUG_ON(rmap->used >= rmap->size); 146 index = rmap->used++; 147 rmap->obj[index] = obj; 148 return index; 149 } 150 EXPORT_SYMBOL(cpu_rmap_add); 151 152 /** 153 * cpu_rmap_update - update CPU rmap following a change of object affinity 154 * @rmap: CPU rmap to update 155 * @index: Index of object whose affinity changed 156 * @affinity: New CPU affinity of object 157 */ 158 int cpu_rmap_update(struct cpu_rmap *rmap, u16 index, 159 const struct cpumask *affinity) 160 { 161 cpumask_var_t update_mask; 162 unsigned int cpu; 163 164 if (unlikely(!zalloc_cpumask_var(&update_mask, GFP_KERNEL))) 165 return -ENOMEM; 166 167 /* Invalidate distance for all CPUs for which this used to be 168 * the nearest object. Mark those CPUs for update. 169 */ 170 for_each_online_cpu(cpu) { 171 if (rmap->near[cpu].index == index) { 172 rmap->near[cpu].dist = CPU_RMAP_DIST_INF; 173 cpumask_set_cpu(cpu, update_mask); 174 } 175 } 176 177 debug_print_rmap(rmap, "after invalidating old distances"); 178 179 /* Set distance to 0 for all CPUs in the new affinity mask. 180 * Mark all CPUs within their NUMA nodes for update. 181 */ 182 for_each_cpu(cpu, affinity) { 183 rmap->near[cpu].index = index; 184 rmap->near[cpu].dist = 0; 185 cpumask_or(update_mask, update_mask, 186 cpumask_of_node(cpu_to_node(cpu))); 187 } 188 189 debug_print_rmap(rmap, "after updating neighbours"); 190 191 /* Update distances based on topology */ 192 for_each_cpu(cpu, update_mask) { 193 if (cpu_rmap_copy_neigh(rmap, cpu, 194 topology_sibling_cpumask(cpu), 1)) 195 continue; 196 if (cpu_rmap_copy_neigh(rmap, cpu, 197 topology_core_cpumask(cpu), 2)) 198 continue; 199 if (cpu_rmap_copy_neigh(rmap, cpu, 200 cpumask_of_node(cpu_to_node(cpu)), 3)) 201 continue; 202 /* We could continue into NUMA node distances, but for now 203 * we give up. 204 */ 205 } 206 207 debug_print_rmap(rmap, "after copying neighbours"); 208 209 free_cpumask_var(update_mask); 210 return 0; 211 } 212 EXPORT_SYMBOL(cpu_rmap_update); 213 214 /* Glue between IRQ affinity notifiers and CPU rmaps */ 215 216 struct irq_glue { 217 struct irq_affinity_notify notify; 218 struct cpu_rmap *rmap; 219 u16 index; 220 }; 221 222 /** 223 * free_irq_cpu_rmap - free a CPU affinity reverse-map used for IRQs 224 * @rmap: Reverse-map allocated with alloc_irq_cpu_map(), or %NULL 225 * 226 * Must be called in process context, before freeing the IRQs. 227 */ 228 void free_irq_cpu_rmap(struct cpu_rmap *rmap) 229 { 230 struct irq_glue *glue; 231 u16 index; 232 233 if (!rmap) 234 return; 235 236 for (index = 0; index < rmap->used; index++) { 237 glue = rmap->obj[index]; 238 irq_set_affinity_notifier(glue->notify.irq, NULL); 239 } 240 241 cpu_rmap_put(rmap); 242 } 243 EXPORT_SYMBOL(free_irq_cpu_rmap); 244 245 /** 246 * irq_cpu_rmap_notify - callback for IRQ subsystem when IRQ affinity updated 247 * @notify: struct irq_affinity_notify passed by irq/manage.c 248 * @mask: cpu mask for new SMP affinity 249 * 250 * This is executed in workqueue context. 251 */ 252 static void 253 irq_cpu_rmap_notify(struct irq_affinity_notify *notify, const cpumask_t *mask) 254 { 255 struct irq_glue *glue = 256 container_of(notify, struct irq_glue, notify); 257 int rc; 258 259 rc = cpu_rmap_update(glue->rmap, glue->index, mask); 260 if (rc) 261 pr_warning("irq_cpu_rmap_notify: update failed: %d\n", rc); 262 } 263 264 /** 265 * irq_cpu_rmap_release - reclaiming callback for IRQ subsystem 266 * @ref: kref to struct irq_affinity_notify passed by irq/manage.c 267 */ 268 static void irq_cpu_rmap_release(struct kref *ref) 269 { 270 struct irq_glue *glue = 271 container_of(ref, struct irq_glue, notify.kref); 272 273 cpu_rmap_put(glue->rmap); 274 kfree(glue); 275 } 276 277 /** 278 * irq_cpu_rmap_add - add an IRQ to a CPU affinity reverse-map 279 * @rmap: The reverse-map 280 * @irq: The IRQ number 281 * 282 * This adds an IRQ affinity notifier that will update the reverse-map 283 * automatically. 284 * 285 * Must be called in process context, after the IRQ is allocated but 286 * before it is bound with request_irq(). 287 */ 288 int irq_cpu_rmap_add(struct cpu_rmap *rmap, int irq) 289 { 290 struct irq_glue *glue = kzalloc(sizeof(*glue), GFP_KERNEL); 291 int rc; 292 293 if (!glue) 294 return -ENOMEM; 295 glue->notify.notify = irq_cpu_rmap_notify; 296 glue->notify.release = irq_cpu_rmap_release; 297 glue->rmap = rmap; 298 cpu_rmap_get(rmap); 299 glue->index = cpu_rmap_add(rmap, glue); 300 rc = irq_set_affinity_notifier(irq, &glue->notify); 301 if (rc) { 302 cpu_rmap_put(glue->rmap); 303 kfree(glue); 304 } 305 return rc; 306 } 307 EXPORT_SYMBOL(irq_cpu_rmap_add); 308