1 // SPDX-License-Identifier: GPL-2.0 2 #include "debug.h" 3 #include "dsos.h" 4 #include "dso.h" 5 #include "util.h" 6 #include "vdso.h" 7 #include "namespaces.h" 8 #include <errno.h> 9 #include <libgen.h> 10 #include <stdlib.h> 11 #include <string.h> 12 #include <symbol.h> // filename__read_build_id 13 #include <unistd.h> 14 15 void dsos__init(struct dsos *dsos) 16 { 17 init_rwsem(&dsos->lock); 18 19 dsos->cnt = 0; 20 dsos->allocated = 0; 21 dsos->dsos = NULL; 22 dsos->sorted = true; 23 } 24 25 static void dsos__purge(struct dsos *dsos) 26 { 27 down_write(&dsos->lock); 28 29 for (unsigned int i = 0; i < dsos->cnt; i++) { 30 struct dso *dso = dsos->dsos[i]; 31 32 dso__set_dsos(dso, NULL); 33 dso__put(dso); 34 } 35 36 zfree(&dsos->dsos); 37 dsos->cnt = 0; 38 dsos->allocated = 0; 39 dsos->sorted = true; 40 41 up_write(&dsos->lock); 42 } 43 44 void dsos__exit(struct dsos *dsos) 45 { 46 dsos__purge(dsos); 47 exit_rwsem(&dsos->lock); 48 } 49 50 51 static int __dsos__for_each_dso(struct dsos *dsos, 52 int (*cb)(struct dso *dso, void *data), 53 void *data) 54 { 55 for (unsigned int i = 0; i < dsos->cnt; i++) { 56 struct dso *dso = dsos->dsos[i]; 57 int err; 58 59 err = cb(dso, data); 60 if (err) 61 return err; 62 } 63 return 0; 64 } 65 66 struct dsos__read_build_ids_cb_args { 67 bool with_hits; 68 bool have_build_id; 69 }; 70 71 static int dsos__read_build_ids_cb(struct dso *dso, void *data) 72 { 73 struct dsos__read_build_ids_cb_args *args = data; 74 struct nscookie nsc; 75 76 if (args->with_hits && !dso__hit(dso) && !dso__is_vdso(dso)) 77 return 0; 78 if (dso__has_build_id(dso)) { 79 args->have_build_id = true; 80 return 0; 81 } 82 nsinfo__mountns_enter(dso__nsinfo(dso), &nsc); 83 if (filename__read_build_id(dso__long_name(dso), dso__bid(dso)) > 0) { 84 args->have_build_id = true; 85 dso__set_has_build_id(dso); 86 } else if (errno == ENOENT && dso__nsinfo(dso)) { 87 char *new_name = dso__filename_with_chroot(dso, dso__long_name(dso)); 88 89 if (new_name && filename__read_build_id(new_name, dso__bid(dso)) > 0) { 90 args->have_build_id = true; 91 dso__set_has_build_id(dso); 92 } 93 free(new_name); 94 } 95 nsinfo__mountns_exit(&nsc); 96 return 0; 97 } 98 99 bool dsos__read_build_ids(struct dsos *dsos, bool with_hits) 100 { 101 struct dsos__read_build_ids_cb_args args = { 102 .with_hits = with_hits, 103 .have_build_id = false, 104 }; 105 106 dsos__for_each_dso(dsos, dsos__read_build_ids_cb, &args); 107 return args.have_build_id; 108 } 109 110 static int __dso__cmp_long_name(const char *long_name, const struct dso_id *id, 111 const struct dso *b) 112 { 113 int rc = strcmp(long_name, dso__long_name(b)); 114 return rc ?: dso_id__cmp(id, dso__id_const(b)); 115 } 116 117 static int __dso__cmp_short_name(const char *short_name, const struct dso_id *id, 118 const struct dso *b) 119 { 120 int rc = strcmp(short_name, dso__short_name(b)); 121 return rc ?: dso_id__cmp(id, dso__id_const(b)); 122 } 123 124 static int dsos__cmp_long_name_id_short_name(const void *va, const void *vb) 125 { 126 const struct dso *a = *((const struct dso **)va); 127 const struct dso *b = *((const struct dso **)vb); 128 int rc = strcmp(dso__long_name(a), dso__long_name(b)); 129 130 if (!rc) { 131 rc = dso_id__cmp(dso__id_const(a), dso__id_const(b)); 132 if (!rc) 133 rc = strcmp(dso__short_name(a), dso__short_name(b)); 134 } 135 return rc; 136 } 137 138 struct dsos__key { 139 const char *long_name; 140 const struct dso_id *id; 141 }; 142 143 static int dsos__cmp_key_long_name_id(const void *vkey, const void *vdso) 144 { 145 const struct dsos__key *key = vkey; 146 const struct dso *dso = *((const struct dso **)vdso); 147 148 return __dso__cmp_long_name(key->long_name, key->id, dso); 149 } 150 151 /* 152 * Find a matching entry and/or link current entry to RB tree. 153 * Either one of the dso or name parameter must be non-NULL or the 154 * function will not work. 155 */ 156 static struct dso *__dsos__find_by_longname_id(struct dsos *dsos, 157 const char *name, 158 struct dso_id *id, 159 bool write_locked) 160 { 161 struct dsos__key key = { 162 .long_name = name, 163 .id = id, 164 }; 165 struct dso **res; 166 167 if (dsos->dsos == NULL) 168 return NULL; 169 170 if (!dsos->sorted) { 171 if (!write_locked) { 172 struct dso *dso; 173 174 up_read(&dsos->lock); 175 down_write(&dsos->lock); 176 dso = __dsos__find_by_longname_id(dsos, name, id, 177 /*write_locked=*/true); 178 up_write(&dsos->lock); 179 down_read(&dsos->lock); 180 return dso; 181 } 182 qsort(dsos->dsos, dsos->cnt, sizeof(struct dso *), 183 dsos__cmp_long_name_id_short_name); 184 dsos->sorted = true; 185 } 186 187 res = bsearch(&key, dsos->dsos, dsos->cnt, sizeof(struct dso *), 188 dsos__cmp_key_long_name_id); 189 if (!res) 190 return NULL; 191 192 return dso__get(*res); 193 } 194 195 int __dsos__add(struct dsos *dsos, struct dso *dso) 196 { 197 if (dsos->cnt == dsos->allocated) { 198 unsigned int to_allocate = 2; 199 struct dso **temp; 200 201 if (dsos->allocated > 0) 202 to_allocate = dsos->allocated * 2; 203 temp = realloc(dsos->dsos, sizeof(struct dso *) * to_allocate); 204 if (!temp) 205 return -ENOMEM; 206 dsos->dsos = temp; 207 dsos->allocated = to_allocate; 208 } 209 dsos->dsos[dsos->cnt++] = dso__get(dso); 210 if (dsos->cnt >= 2 && dsos->sorted) { 211 dsos->sorted = dsos__cmp_long_name_id_short_name(&dsos->dsos[dsos->cnt - 2], 212 &dsos->dsos[dsos->cnt - 1]) 213 <= 0; 214 } 215 dso__set_dsos(dso, dsos); 216 return 0; 217 } 218 219 int dsos__add(struct dsos *dsos, struct dso *dso) 220 { 221 int ret; 222 223 down_write(&dsos->lock); 224 ret = __dsos__add(dsos, dso); 225 up_write(&dsos->lock); 226 return ret; 227 } 228 229 struct dsos__find_id_cb_args { 230 const char *name; 231 struct dso_id *id; 232 struct dso *res; 233 }; 234 235 static int dsos__find_id_cb(struct dso *dso, void *data) 236 { 237 struct dsos__find_id_cb_args *args = data; 238 239 if (__dso__cmp_short_name(args->name, args->id, dso) == 0) { 240 args->res = dso__get(dso); 241 return 1; 242 } 243 return 0; 244 245 } 246 247 static struct dso *__dsos__find_id(struct dsos *dsos, const char *name, struct dso_id *id, 248 bool cmp_short, bool write_locked) 249 { 250 struct dso *res; 251 252 if (cmp_short) { 253 struct dsos__find_id_cb_args args = { 254 .name = name, 255 .id = id, 256 .res = NULL, 257 }; 258 259 __dsos__for_each_dso(dsos, dsos__find_id_cb, &args); 260 return args.res; 261 } 262 res = __dsos__find_by_longname_id(dsos, name, id, write_locked); 263 return res; 264 } 265 266 struct dso *dsos__find(struct dsos *dsos, const char *name, bool cmp_short) 267 { 268 struct dso *res; 269 270 down_read(&dsos->lock); 271 res = __dsos__find_id(dsos, name, NULL, cmp_short, /*write_locked=*/false); 272 up_read(&dsos->lock); 273 return res; 274 } 275 276 static void dso__set_basename(struct dso *dso) 277 { 278 char *base, *lname; 279 int tid; 280 281 if (perf_pid_map_tid(dso__long_name(dso), &tid)) { 282 if (asprintf(&base, "[JIT] tid %d", tid) < 0) 283 return; 284 } else { 285 /* 286 * basename() may modify path buffer, so we must pass 287 * a copy. 288 */ 289 lname = strdup(dso__long_name(dso)); 290 if (!lname) 291 return; 292 293 /* 294 * basename() may return a pointer to internal 295 * storage which is reused in subsequent calls 296 * so copy the result. 297 */ 298 base = strdup(basename(lname)); 299 300 free(lname); 301 302 if (!base) 303 return; 304 } 305 dso__set_short_name(dso, base, true); 306 } 307 308 static struct dso *__dsos__addnew_id(struct dsos *dsos, const char *name, struct dso_id *id) 309 { 310 struct dso *dso = dso__new_id(name, id); 311 312 if (dso != NULL) { 313 /* 314 * The dsos lock is held on entry, so rename the dso before 315 * adding it to avoid needing to take the dsos lock again to say 316 * the array isn't sorted. 317 */ 318 dso__set_basename(dso); 319 __dsos__add(dsos, dso); 320 } 321 return dso; 322 } 323 324 static struct dso *__dsos__findnew_id(struct dsos *dsos, const char *name, struct dso_id *id) 325 { 326 struct dso *dso = __dsos__find_id(dsos, name, id, false, /*write_locked=*/true); 327 328 if (dso && dso_id__empty(dso__id(dso)) && !dso_id__empty(id)) 329 __dso__inject_id(dso, id); 330 331 return dso ? dso : __dsos__addnew_id(dsos, name, id); 332 } 333 334 struct dso *dsos__findnew_id(struct dsos *dsos, const char *name, struct dso_id *id) 335 { 336 struct dso *dso; 337 down_write(&dsos->lock); 338 dso = __dsos__findnew_id(dsos, name, id); 339 up_write(&dsos->lock); 340 return dso; 341 } 342 343 struct dsos__fprintf_buildid_cb_args { 344 FILE *fp; 345 bool (*skip)(struct dso *dso, int parm); 346 int parm; 347 size_t ret; 348 }; 349 350 static int dsos__fprintf_buildid_cb(struct dso *dso, void *data) 351 { 352 struct dsos__fprintf_buildid_cb_args *args = data; 353 char sbuild_id[SBUILD_ID_SIZE]; 354 355 if (args->skip && args->skip(dso, args->parm)) 356 return 0; 357 build_id__sprintf(dso__bid(dso), sbuild_id); 358 args->ret += fprintf(args->fp, "%-40s %s\n", sbuild_id, dso__long_name(dso)); 359 return 0; 360 } 361 362 size_t dsos__fprintf_buildid(struct dsos *dsos, FILE *fp, 363 bool (*skip)(struct dso *dso, int parm), int parm) 364 { 365 struct dsos__fprintf_buildid_cb_args args = { 366 .fp = fp, 367 .skip = skip, 368 .parm = parm, 369 .ret = 0, 370 }; 371 372 dsos__for_each_dso(dsos, dsos__fprintf_buildid_cb, &args); 373 return args.ret; 374 } 375 376 struct dsos__fprintf_cb_args { 377 FILE *fp; 378 size_t ret; 379 }; 380 381 static int dsos__fprintf_cb(struct dso *dso, void *data) 382 { 383 struct dsos__fprintf_cb_args *args = data; 384 385 args->ret += dso__fprintf(dso, args->fp); 386 return 0; 387 } 388 389 size_t dsos__fprintf(struct dsos *dsos, FILE *fp) 390 { 391 struct dsos__fprintf_cb_args args = { 392 .fp = fp, 393 .ret = 0, 394 }; 395 396 dsos__for_each_dso(dsos, dsos__fprintf_cb, &args); 397 return args.ret; 398 } 399 400 static int dsos__hit_all_cb(struct dso *dso, void *data __maybe_unused) 401 { 402 dso__set_hit(dso); 403 return 0; 404 } 405 406 int dsos__hit_all(struct dsos *dsos) 407 { 408 return dsos__for_each_dso(dsos, dsos__hit_all_cb, NULL); 409 } 410 411 struct dso *dsos__findnew_module_dso(struct dsos *dsos, 412 struct machine *machine, 413 struct kmod_path *m, 414 const char *filename) 415 { 416 struct dso *dso; 417 418 down_write(&dsos->lock); 419 420 dso = __dsos__find_id(dsos, m->name, NULL, /*cmp_short=*/true, /*write_locked=*/true); 421 if (dso) { 422 up_write(&dsos->lock); 423 return dso; 424 } 425 /* 426 * Failed to find the dso so create it. Change the name before adding it 427 * to the array, to avoid unnecessary sorts and potential locking 428 * issues. 429 */ 430 dso = dso__new_id(m->name, /*id=*/NULL); 431 if (!dso) { 432 up_write(&dsos->lock); 433 return NULL; 434 } 435 dso__set_basename(dso); 436 dso__set_module_info(dso, m, machine); 437 dso__set_long_name(dso, strdup(filename), true); 438 dso__set_kernel(dso, DSO_SPACE__KERNEL); 439 __dsos__add(dsos, dso); 440 441 up_write(&dsos->lock); 442 return dso; 443 } 444 445 static int dsos__find_kernel_dso_cb(struct dso *dso, void *data) 446 { 447 struct dso **res = data; 448 /* 449 * The cpumode passed to is_kernel_module is not the cpumode of *this* 450 * event. If we insist on passing correct cpumode to is_kernel_module, 451 * we should record the cpumode when we adding this dso to the linked 452 * list. 453 * 454 * However we don't really need passing correct cpumode. We know the 455 * correct cpumode must be kernel mode (if not, we should not link it 456 * onto kernel_dsos list). 457 * 458 * Therefore, we pass PERF_RECORD_MISC_CPUMODE_UNKNOWN. 459 * is_kernel_module() treats it as a kernel cpumode. 460 */ 461 if (!dso__kernel(dso) || 462 is_kernel_module(dso__long_name(dso), PERF_RECORD_MISC_CPUMODE_UNKNOWN)) 463 return 0; 464 465 *res = dso__get(dso); 466 return 1; 467 } 468 469 struct dso *dsos__find_kernel_dso(struct dsos *dsos) 470 { 471 struct dso *res = NULL; 472 473 dsos__for_each_dso(dsos, dsos__find_kernel_dso_cb, &res); 474 return res; 475 } 476 477 int dsos__for_each_dso(struct dsos *dsos, int (*cb)(struct dso *dso, void *data), void *data) 478 { 479 int err; 480 481 down_read(&dsos->lock); 482 err = __dsos__for_each_dso(dsos, cb, data); 483 up_read(&dsos->lock); 484 return err; 485 } 486