Quoting Tvrtko Ursulin (2021-01-21 18:10:05) > +static struct clients *init_clients(const char *drm_card) > +{ > + struct clients *clients = malloc(sizeof(*clients)); > + const char *slash; > + ssize_t ret; > + > + memset(clients, 0, sizeof(*clients)); > + > + if (drm_card) { > + slash = rindex(drm_card, '/'); > + assert(slash); > + } else { > + slash = "card0"; > + } > + > + ret = snprintf(clients->sysfs_root, sizeof(clients->sysfs_root), > + "/sys/class/drm/%s/clients/", slash); > + assert(ret > 0 && ret < sizeof(clients->sysfs_root)); Afaict, igt_device_card_match_pci et al are not guaranteed to fill in dev->drm_card and may leave it blank instead of finding the corresponding /dev/dri/cardN. > + > + return clients; > +} > + > +static struct client * > +find_client(struct clients *clients, enum client_status status, unsigned int id) > +{ > + struct client *c; > + int tmp; > + > + for_each_client(clients, c, tmp) { > + if ((status == FREE && c->status == FREE) || > + (status == c->status && c->id == id)) > + return c; if (status != c->status) continue; if (status == FREE || c->id == id) return c; > + } > + > + return NULL; > +} > + > +static void update_client(struct client *c, unsigned int pid, char *name) > +{ > + uint64_t val[c->clients->num_classes]; > + unsigned int i; > + > + if (c->pid != pid) > + c->pid = pid; > + > + if (strcmp(c->name, name)) > + strncpy(c->name, name, sizeof(c->name) - 1); That answers my question about what happens to clients that change their name after new contexts are created. You don't trust that (pid, name) are tied together? Right, either may change during the read of the pair. > + > + for (i = 0; i < c->clients->num_classes; i++) > + val[i] = read_client_busy(c, c->clients->class[i].class); > + > + c->total = 0; > + > + for (i = 0; i < c->clients->num_classes; i++) { > + if (val[i] < c->last[i]) > + continue; /* It will catch up soon. */ > + > + c->val[i] = val[i] - c->last[i]; > + c->total += c->val[i]; > + c->last[i] = val[i]; > + } > + > + c->samples++; > + c->status = ALIVE; > +} > + > +static void > +add_client(struct clients *clients, unsigned int id, unsigned int pid, > + char *name, int sysfs_root) > +{ > + struct client *c; > + > + if (find_client(clients, ALIVE, id)) > + return; > + > + c = find_client(clients, FREE, 0); > + if (!c) { > + unsigned int idx = clients->num_clients; > + > + clients->num_clients += (clients->num_clients + 2) / 2; > + clients->client = realloc(clients->client, > + clients->num_clients * sizeof(*c)); > + assert(clients->client); > + > + c = &clients->client[idx]; > + memset(c, 0, (clients->num_clients - idx) * sizeof(*c)); Hence must be FREE==zero. And for_each_client() walks an [] not list. > + } > + > + c->sysfs_root = sysfs_root; > + c->id = id; > + c->clients = clients; > + c->val = calloc(clients->num_classes, sizeof(c->val)); > + c->last = calloc(clients->num_classes, sizeof(c->last)); > + assert(c->val && c->last); > + > + update_client(c, pid, name); > +} > + > +static void free_client(struct client *c) > +{ > + if (c->sysfs_root >= 0) > + close(c->sysfs_root); > + free(c->val); > + free(c->last); > + memset(c, 0, sizeof(*c)); Worth compacting the array? I expect the array will trend to become quite sparse rather than auto-defragment. > +} > + > +static int > +read_client_sysfs(char *buf, int bufsize, const char *sysfs_root, > + unsigned int id, const char *field, int *client_root) > +{ > + ssize_t ret; > + > + if (*client_root < 0) { > + char namebuf[256]; > + > + ret = snprintf(namebuf, sizeof(namebuf), "%s/%u", > + sysfs_root, id); > + assert(ret > 0 && ret < sizeof(namebuf)); > + if (ret <= 0 || ret == sizeof(namebuf)) > + return -1; > + > + *client_root = open(namebuf, O_RDONLY | O_DIRECTORY); > + } > + > + if (*client_root < 0) > + return -1; > + > + return __read_client_field(*client_root, field, buf, bufsize); > +} > + > +static void scan_clients(struct clients *clients) > +{ > + struct dirent *dent; > + struct client *c; > + unsigned int id; > + int tmp; > + DIR *d; > + > + if (!clients) > + return; > + > + for_each_client(clients, c, tmp) { > + if (c->status == ALIVE) > + c->status = PROBE; > + } > + > + d = opendir(clients->sysfs_root); > + if (!d) > + return; > + > + while ((dent = readdir(d)) != NULL) { > + char name[256], pid[256]; > + int ret, root = -1, *pr; > + > + if (dent->d_type != DT_DIR) > + continue; > + if (!isdigit(dent->d_name[0])) > + continue; > + > + id = atoi(dent->d_name); > + > + c = find_client(clients, PROBE, id); > + > + if (c) > + pr = &c->sysfs_root; > + else > + pr = &root; > + > + ret = read_client_sysfs(name, sizeof(name), clients->sysfs_root, > + id, "name", pr); > + ret |= read_client_sysfs(pid, sizeof(pid), clients->sysfs_root, > + id, "pid", pr); > + if (!ret) { > + if (!c) > + add_client(clients, id, atoi(pid), name, root); > + else > + update_client(c, atoi(pid), name); > + } else if (c) { > + c->status = PROBE; /* Will be deleted below. */ > + } > + } > + > + closedir(d); > + > + for_each_client(clients, c, tmp) { > + if (c->status == PROBE) > + free_client(c); > + } > +} > + > +static int cmp(const void *_a, const void *_b) > +{ > + const struct client *a = _a; > + const struct client *b = _b; > + long tot_a = a->total; > + long tot_b = b->total; > + > + tot_a *= a->status == ALIVE && a->samples > 1; > + tot_b *= b->status == ALIVE && b->samples > 1; > + > + tot_b -= tot_a; > + > + if (!tot_b) > + return (int)b->id - a->id; If both active and identical runtimes, sort in ascending id, which is the client/<id> (chronological order of creation) return (int)(b->id - a->id); > + > + while (tot_b > INT_MAX || tot_b < INT_MIN) > + tot_b /= 2; This is just if (tot_b > 0) return 1; if (tot_b < 0) return -1; return 0; Right? Sort in order of runtimes, if active. If inactive, age of creation. > + > + return tot_b; > +} > + > static const char *bars[] = { " ", "▏", "▎", "▍", "▌", "▋", "▊", "▉", "█" }; > > +static void n_spaces(const unsigned int n) > +{ > + unsigned int i; > + > + for (i = 0; i < n; i++) > + putchar(' '); > +} > + > static void > print_percentage_bar(double percent, int max_len) > { > - int bar_len = percent * (8 * (max_len - 2)) / 100.0; > - int i; > + int bar_len, i, len = max_len - 2; > + const int w = 8; > + > + assert(max_len > 0); > + > + bar_len = percent * len / 100.0; > + if (bar_len > len) > + bar_len = len; > + bar_len *= w; > > putchar('|'); > > - for (i = bar_len; i >= 8; i -= 8) > - printf("%s", bars[8]); > + for (i = bar_len; i >= w; i -= w) > + printf("%s", bars[w]); > if (i) > printf("%s", bars[i]); > > - for (i = 0; i < (max_len - 2 - (bar_len + 7) / 8); i++) > - putchar(' '); > + len -= (bar_len + (w - 1)) / w; > + n_spaces(len); > > putchar('|'); > } > @@ -744,6 +1077,18 @@ json_close_struct(void) > fflush(stdout); > } > > +static void > +__json_add_member(const char *key, const char *val) > +{ > + assert(json_indent_level < ARRAY_SIZE(json_indent)); > + > + fprintf(out, "%s%s\"%s\": \"%s\"", > + json_struct_members ? ",\n" : "", > + json_indent[json_indent_level], key, val); > + > + json_struct_members++; > +} > + > static unsigned int > json_add_member(const struct cnt_group *parent, struct cnt_item *item, > unsigned int headers) > @@ -1046,8 +1391,6 @@ print_header(const struct igt_device_card *card, > memmove(&groups[0], &groups[1], > sizeof(groups) - sizeof(groups[0])); > > - pops->open_struct(NULL); > - > *consumed = print_groups(groups); > > if (output_mode == INTERACTIVE) { > @@ -1204,7 +1547,7 @@ print_engine(struct engines *engines, unsigned int i, double t, > engine->display_name, engine_items[0].buf); > > val = pmu_calc(&engine->busy.val, 1e9, t, 100); > - print_percentage_bar(val, max_w - len); > + print_percentage_bar(val, max_w > len ? max_w - len : 0); > > printf("%s\n", buf); > > @@ -1219,7 +1562,6 @@ print_engines_footer(struct engines *engines, double t, > int lines, int con_w, int con_h) > { > pops->close_struct(); > - pops->close_struct(); > > if (output_mode == INTERACTIVE) { > if (lines++ < con_h) > @@ -1243,6 +1585,9 @@ static void init_engine_classes(struct engines *engines) > unsigned int i, num; > int max = -1; > > + if (engines->num_classes) > + return; > + > for (i = 0; i < engines->num_engines; i++) { > struct engine *engine = engine_ptr(engines, i); > > @@ -1404,6 +1749,148 @@ print_engines(struct engines *engines, double t, int lines, int w, int h) > return lines; > } > > +static int > +print_clients_header(struct clients *clients, int lines, > + int con_w, int con_h, int *class_w) > +{ > + if (output_mode == INTERACTIVE) { > + const char *pidname = " PID NAME "; > + unsigned int num_active = 0; > + int len = strlen(pidname); > + > + if (lines++ >= con_h) > + return lines; > + > + printf("\033[7m"); > + printf("%s", pidname); > + > + if (lines++ >= con_h || len >= con_w) > + return lines; > + > + if (clients->num_classes) { > + unsigned int i; > + int width; > + > + for (i = 0; i < clients->num_classes; i++) { > + if (clients->class[i].num_engines) > + num_active++; > + } > + > + *class_w = width = (con_w - len) / num_active; > + > + for (i = 0; i < clients->num_classes; i++) { > + const char *name = clients->class[i].name; > + int name_len = strlen(name); > + int pad = (width - name_len) / 2; > + int spaces = width - pad - name_len; > + > + if (!clients->class[i].num_engines) > + continue; /* Assert in the ideal world. */ > + > + if (pad < 0 || spaces < 0) > + continue; > + > + n_spaces(pad); > + printf("%s", name); Should we sanitize the names and remove ansi control codes or only show printable char? > + n_spaces(spaces); > + len += pad + name_len + spaces; > + } > + } > + > + n_spaces(con_w - len); > + printf("\033[0m\n"); > + } else { > + if (clients->num_classes) > + pops->open_struct("clients"); > + } > + > + return lines; > +} > + > +static int > +print_client(struct client *c, struct engines *engines, double t, int lines, > + int con_w, int con_h, unsigned int period_us, int *class_w) > +{ > + struct clients *clients = c->clients; > + unsigned int i; > + > + if (output_mode == INTERACTIVE) { > + printf("%6u %17s ", c->pid, c->name); > + > + for (i = 0; i < clients->num_classes; i++) { > + double pct; > + > + if (!clients->class[i].num_engines) > + continue; /* Assert in the ideal world. */ > + > + pct = (double)c->val[i] / period_us / 1e3 * 100 / > + clients->class[i].num_engines; > + > + /* > + * Guard against possible time-drift between sampling > + * client data and time we obtained our time-delta from > + * PMU. > + */ > + if (pct > 100.0) > + pct = 100.0; > + > + print_percentage_bar(pct, *class_w); > + > + lines++; > + } > + > + putchar('\n'); > + } else if (output_mode == JSON) { > + char buf[64]; > + > + snprintf(buf, sizeof(buf), "%u", c->id); > + pops->open_struct(buf); > + > + __json_add_member("name", c->name); > + > + snprintf(buf, sizeof(buf), "%u", c->pid); > + __json_add_member("pid", buf); > + > + pops->open_struct("engine-classes"); > + > + for (i = 0; i < clients->num_classes; i++) { > + double pct; > + > + snprintf(buf, sizeof(buf), "%s", > + clients->class[i].name); > + pops->open_struct(buf); > + > + pct = (double)c->val[i] / period_us / 1e3 * 100; > + snprintf(buf, sizeof(buf), "%f", pct); > + __json_add_member("busy", buf); > + > + __json_add_member("unit", "%"); > + > + pops->close_struct(); > + } > + > + pops->close_struct(); > + pops->close_struct(); > + } > + > + return lines; > +} > + > +static int > +print_clients_footer(struct clients *clients, double t, > + int lines, int con_w, int con_h) > +{ > + if (output_mode == INTERACTIVE) { > + if (lines++ < con_h) > + printf("\n"); > + } else { > + if (clients->num_classes) > + pops->close_struct(); > + } > + > + return lines; > +} > + > static bool stop_top; > > static void sigint_handler(int sig) > @@ -1492,6 +1979,7 @@ static void process_stdin(unsigned int timeout_us) > int main(int argc, char **argv) > { > unsigned int period_us = DEFAULT_PERIOD_MS * 1000; > + struct clients *clients = NULL; > int con_w = -1, con_h = -1; > char *output_path = NULL; > struct engines *engines; > @@ -1625,13 +2113,20 @@ int main(int argc, char **argv) > > ret = EXIT_SUCCESS; > > + clients = init_clients(card.pci_slot_name[0] ? card.card : NULL); > + init_engine_classes(engines); > + clients->num_classes = engines->num_classes; > + clients->class = engines->class; > + > pmu_sample(engines); > + scan_clients(clients); > codename = igt_device_get_pretty_name(&card, false); > > while (!stop_top) { > bool consumed = false; > - int lines = 0; > + int j, lines = 0; > struct winsize ws; > + struct client *c; > double t; > > /* Update terminal size. */ > @@ -1650,10 +2145,18 @@ int main(int argc, char **argv) > pmu_sample(engines); > t = (double)(engines->ts.cur - engines->ts.prev) / 1e9; > > + scan_clients(clients); > + if (clients) { > + qsort(clients->client, clients->num_clients, > + sizeof(*clients->client), cmp); > + } > + > if (stop_top) > break; > > while (!consumed) { > + pops->open_struct(NULL); > + > lines = print_header(&card, codename, engines, > t, lines, con_w, con_h, > &consumed); > @@ -1661,6 +2164,36 @@ int main(int argc, char **argv) > lines = print_imc(engines, t, lines, con_w, con_h); > > lines = print_engines(engines, t, lines, con_w, con_h); > + > + if (clients) { > + int class_w; > + > + lines = print_clients_header(clients, lines, > + con_w, con_h, > + &class_w); > + > + for_each_client(clients, c, j) { > + if (lines++ > con_h) > + break; > + > + assert(c->status != PROBE); > + if (c->status != ALIVE) > + break; > + > + if (c->samples < 2) > + continue; > + > + lines = print_client(c, engines, t, > + lines, con_w, > + con_h, period_us, > + &class_w); > + } > + > + lines = print_clients_footer(clients, t, lines, > + con_w, con_h); > + } > + > + pops->close_struct(); > } > > if (stop_top) > -- > 2.27.0 > > _______________________________________________ > Intel-gfx mailing list > Intel-gfx@xxxxxxxxxxxxxxxxxxxxx > https://lists.freedesktop.org/mailman/listinfo/intel-gfx _______________________________________________ Intel-gfx mailing list Intel-gfx@xxxxxxxxxxxxxxxxxxxxx https://lists.freedesktop.org/mailman/listinfo/intel-gfx