| Message ID | a0f9f8c1030c66305c9b921057c3d483064d5529.1651220820.git.bristot@kernel.org (mailing list archive) |
|---|---|
| State | Superseded |
| Headers | show |
| Series | [v2] rtla: Remove procps-ng dependency | expand |
On Fri, 29 Apr 2022, Daniel Bristot de Oliveira wrote: > Daniel Wagner reported to me that readproc.h got deprecated. Also, > while the procps-ng library was available on Fedora, it was not available > on RHEL, which is a piece of evidence that it was not that used. I am for removing this dependency, but I feel I have to correct your description. procps-ng is available on RHEL. The development library however is just available at boot-time, but only because no other packages depend on it. If we were to add a package that depended on it, it would then be available at runtime. I would remove any references to RHEL in your description, and just say you are reducing the dependency on this to simply building rtla. > > rtla uses procps-ng only to find the PID of the tracers' workload. > > I used the procps-ng library to avoid reinventing the wheel. But in this > case, reinventing the wheel took me less time than the time we already > took trying to work around problems. > > Implement a function that reads /proc/ entries, checking if: > - the entry is a directory > - the directory name is composed only of digits (PID) > - the directory contains the comm file > - the comm file contains a comm that matches the tracers' > workload prefix. > - then return true; otherwise, return false. > > And use it instead of procps-ng. > > Changes from V1: > - Use a single buffer for comm and comm_path > - Cause an error in case of a too long command prefix > - Do a close_dir() > - Improve log messages > > Cc: John Kacur <jkacur@redhat.com> > Cc: Steven Rostedt <rostedt@goodmis.org> > Fixes: b1696371d865 ("rtla: Helper functions for rtla") > Reported-by: Daniel Wagner <dwagner@suse.de> > Signed-off-by: Daniel Bristot de Oliveira <bristot@kernel.org> > --- > tools/tracing/rtla/Makefile | 2 +- > tools/tracing/rtla/README.txt | 1 - > tools/tracing/rtla/src/utils.c | 102 ++++++++++++++++++++++++++------- > tools/tracing/rtla/src/utils.h | 3 +- > 4 files changed, 84 insertions(+), 24 deletions(-) > > diff --git a/tools/tracing/rtla/Makefile b/tools/tracing/rtla/Makefile > index 5a3226e436ef..523f0a8c38c2 100644 > --- a/tools/tracing/rtla/Makefile > +++ b/tools/tracing/rtla/Makefile > @@ -32,7 +32,7 @@ TRACEFS_HEADERS := $$($(PKG_CONFIG) --cflags libtracefs) > > CFLAGS := -O -g -DVERSION=\"$(VERSION)\" $(FOPTS) $(MOPTS) $(WOPTS) $(TRACEFS_HEADERS) > LDFLAGS := -ggdb > -LIBS := $$($(PKG_CONFIG) --libs libtracefs) -lprocps > +LIBS := $$($(PKG_CONFIG) --libs libtracefs) > > SRC := $(wildcard src/*.c) > HDR := $(wildcard src/*.h) > diff --git a/tools/tracing/rtla/README.txt b/tools/tracing/rtla/README.txt > index 0fbad2640b8c..4af3fd40f171 100644 > --- a/tools/tracing/rtla/README.txt > +++ b/tools/tracing/rtla/README.txt > @@ -11,7 +11,6 @@ RTLA depends on the following libraries and tools: > > - libtracefs > - libtraceevent > - - procps > > It also depends on python3-docutils to compile man pages. > > diff --git a/tools/tracing/rtla/src/utils.c b/tools/tracing/rtla/src/utils.c > index 3bd6f64780cf..f3d562f7f14c 100644 > --- a/tools/tracing/rtla/src/utils.c > +++ b/tools/tracing/rtla/src/utils.c > @@ -3,7 +3,7 @@ > * Copyright (C) 2021 Red Hat Inc, Daniel Bristot de Oliveira <bristot@kernel.org> > */ > > -#include <proc/readproc.h> > +#include <dirent.h> > #include <stdarg.h> > #include <stdlib.h> > #include <string.h> > @@ -262,43 +262,103 @@ int __set_sched_attr(int pid, struct sched_attr *attr) > > return 0; > } > + > /* > - * set_comm_sched_attr - set sched params to threads starting with char *comm > + * procfs_is_workload_pid - check if a procfs entry contains a comm_prefix* comm > + * > + * Check if the procfs entry is a directory of a process, and then check if the > + * process has a comm with the prefix set in char *comm_prefix. As the > + * current users of this function only check for kernel threads, there is no > + * need to check for the threads for the process. > * > - * This function uses procps to list the currently running threads and then > - * set the sched_attr *attr to the threads that start with char *comm. It is > + * Return: True if the proc_entry contains a comm file with comm_prefix*. > + * Otherwise returns false. > + */ > +static int procfs_is_workload_pid(const char *comm_prefix, struct dirent *proc_entry) > +{ > + char buffer[MAX_PATH]; > + int comm_fd, retval; > + char*t_name; > + > + if (proc_entry->d_type != DT_DIR) > + return 0; > + > + if (*proc_entry->d_name == '.') > + return 0; > + > + /* check if the string is a pid */ > + for (t_name = proc_entry->d_name; t_name; t_name++) { > + if (!isdigit(*t_name)) > + break; > + } > + > + if (*t_name != '\0') > + return 0; > + > + snprintf(buffer, MAX_PATH, "/proc/%s/comm", proc_entry->d_name); > + comm_fd = open(buffer, O_RDONLY); > + if (comm_fd < 0) > + return 0; > + > + memset(buffer, 0, MAX_PATH); > + retval = read(comm_fd, buffer, MAX_PATH); > + > + close(comm_fd); > + > + if (retval <= 0) > + return 0; > + > + retval = strncmp(comm_prefix, buffer, strlen(comm_prefix)); > + if (retval) > + return 0; > + > + /* comm already have \n */ > + debug_msg("Found workload pid:%s comm:%s", proc_entry->d_name, buffer); > + > + return 1; > +} > + > +/* > + * set_comm_sched_attr - set sched params to threads starting with char *comm_prefix > + * > + * This function uses procps to list the currently running threads and then set the > + * sched_attr *attr to the threads that start with char *comm_prefix. It is > * mainly used to set the priority to the kernel threads created by the > * tracers. > */ > -int set_comm_sched_attr(const char *comm, struct sched_attr *attr) > +int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr) > { > - int flags = PROC_FILLCOM | PROC_FILLSTAT; > - PROCTAB *ptp; > - proc_t task; > + struct dirent *proc_entry; > + DIR *procfs; > int retval; > > - ptp = openproc(flags); > - if (!ptp) { > - err_msg("error openproc()\n"); > - return -ENOENT; > + if (strlen(comm_prefix) >= MAX_PATH) { > + err_msg("Command prefix is too long: %d < strlen(%s)\n", > + MAX_PATH, comm_prefix); > + exit(EINVAL); > } > > - memset(&task, 0, sizeof(task)); > + procfs = opendir("/proc"); > + > + while ((proc_entry = readdir(procfs))) { > > - while (readproc(ptp, &task)) { > - retval = strncmp(comm, task.cmd, strlen(comm)); > - if (retval) > + retval = procfs_is_workload_pid(comm_prefix, proc_entry); > + if (!retval) > continue; > - retval = __set_sched_attr(task.tid, attr); > - if (retval) > + > + /* procfs_is_workload_pid confirmed it is a pid */ > + retval = __set_sched_attr(atoi(proc_entry->d_name), attr); > + if (retval) { > + err_msg("Error setting sched attributes for pid:%s\n", proc_entry->d_name); > goto out_err; > - } > + } > > - closeproc(ptp); > + debug_msg("Set sched attributes for pid:%s\n", proc_entry->d_name); > + } > return 0; > > out_err: > - closeproc(ptp); > + closedir(procfs); > return 1; > } > > diff --git a/tools/tracing/rtla/src/utils.h b/tools/tracing/rtla/src/utils.h > index fa08e374870a..5571afd3b549 100644 > --- a/tools/tracing/rtla/src/utils.h > +++ b/tools/tracing/rtla/src/utils.h > @@ -6,6 +6,7 @@ > * '18446744073709551615\0' > */ > #define BUFF_U64_STR_SIZE 24 > +#define MAX_PATH 1024 > > #define container_of(ptr, type, member)({ \ > const typeof(((type *)0)->member) *__mptr = (ptr); \ > @@ -53,5 +54,5 @@ struct sched_attr { > }; > > int parse_prio(char *arg, struct sched_attr *sched_param); > -int set_comm_sched_attr(const char *comm, struct sched_attr *attr); > +int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr); > int set_cpu_dma_latency(int32_t latency); > -- > 2.32.0 > >
On 5/5/22 18:16, John Kacur wrote: > If we were to add a package that depended on it, it would then be > available at runtime. I would remove any references to RHEL in your > description, and just say you are reducing the dependency on this to > simply building rtla. If I try to install today, it is not there. I am not saying it would not be there. It is an argument that it it not that used, and it is correct. -- Daniel
On Fri, Apr 29, 2022 at 03:01:48PM +0200, Daniel Bristot de Oliveira wrote: > Daniel Wagner reported to me that readproc.h got deprecated. Also, > while the procps-ng library was available on Fedora, it was not available > on RHEL, which is a piece of evidence that it was not that used. > > rtla uses procps-ng only to find the PID of the tracers' workload. > > I used the procps-ng library to avoid reinventing the wheel. But in this > case, reinventing the wheel took me less time than the time we already > took trying to work around problems. > > Implement a function that reads /proc/ entries, checking if: > - the entry is a directory > - the directory name is composed only of digits (PID) > - the directory contains the comm file > - the comm file contains a comm that matches the tracers' > workload prefix. > - then return true; otherwise, return false. > > And use it instead of procps-ng. > > Changes from V1: > - Use a single buffer for comm and comm_path > - Cause an error in case of a too long command prefix > - Do a close_dir() > - Improve log messages > > Cc: John Kacur <jkacur@redhat.com> > Cc: Steven Rostedt <rostedt@goodmis.org> > Fixes: b1696371d865 ("rtla: Helper functions for rtla") > Reported-by: Daniel Wagner <dwagner@suse.de> > Signed-off-by: Daniel Bristot de Oliveira <bristot@kernel.org> > --- > tools/tracing/rtla/Makefile | 2 +- > tools/tracing/rtla/README.txt | 1 - > tools/tracing/rtla/src/utils.c | 102 ++++++++++++++++++++++++++------- > tools/tracing/rtla/src/utils.h | 3 +- > 4 files changed, 84 insertions(+), 24 deletions(-) > > diff --git a/tools/tracing/rtla/Makefile b/tools/tracing/rtla/Makefile > index 5a3226e436ef..523f0a8c38c2 100644 > --- a/tools/tracing/rtla/Makefile > +++ b/tools/tracing/rtla/Makefile > @@ -32,7 +32,7 @@ TRACEFS_HEADERS := $$($(PKG_CONFIG) --cflags libtracefs) > > CFLAGS := -O -g -DVERSION=\"$(VERSION)\" $(FOPTS) $(MOPTS) $(WOPTS) $(TRACEFS_HEADERS) > LDFLAGS := -ggdb > -LIBS := $$($(PKG_CONFIG) --libs libtracefs) -lprocps > +LIBS := $$($(PKG_CONFIG) --libs libtracefs) > > SRC := $(wildcard src/*.c) > HDR := $(wildcard src/*.h) > diff --git a/tools/tracing/rtla/README.txt b/tools/tracing/rtla/README.txt > index 0fbad2640b8c..4af3fd40f171 100644 > --- a/tools/tracing/rtla/README.txt > +++ b/tools/tracing/rtla/README.txt > @@ -11,7 +11,6 @@ RTLA depends on the following libraries and tools: > > - libtracefs > - libtraceevent > - - procps > > It also depends on python3-docutils to compile man pages. > > diff --git a/tools/tracing/rtla/src/utils.c b/tools/tracing/rtla/src/utils.c > index 3bd6f64780cf..f3d562f7f14c 100644 > --- a/tools/tracing/rtla/src/utils.c > +++ b/tools/tracing/rtla/src/utils.c > @@ -3,7 +3,7 @@ > * Copyright (C) 2021 Red Hat Inc, Daniel Bristot de Oliveira <bristot@kernel.org> > */ > > -#include <proc/readproc.h> > +#include <dirent.h> > #include <stdarg.h> > #include <stdlib.h> > #include <string.h> > @@ -262,43 +262,103 @@ int __set_sched_attr(int pid, struct sched_attr *attr) > > return 0; > } > + > /* > - * set_comm_sched_attr - set sched params to threads starting with char *comm > + * procfs_is_workload_pid - check if a procfs entry contains a comm_prefix* comm > + * > + * Check if the procfs entry is a directory of a process, and then check if the > + * process has a comm with the prefix set in char *comm_prefix. As the > + * current users of this function only check for kernel threads, there is no > + * need to check for the threads for the process. > * > - * This function uses procps to list the currently running threads and then > - * set the sched_attr *attr to the threads that start with char *comm. It is > + * Return: True if the proc_entry contains a comm file with comm_prefix*. > + * Otherwise returns false. > + */ > +static int procfs_is_workload_pid(const char *comm_prefix, struct dirent *proc_entry) > +{ > + char buffer[MAX_PATH]; > + int comm_fd, retval; > + char*t_name; Need a blank.. char *t_name; > + > + if (proc_entry->d_type != DT_DIR) > + return 0; > + > + if (*proc_entry->d_name == '.') > + return 0; > + > + /* check if the string is a pid */ > + for (t_name = proc_entry->d_name; t_name; t_name++) { > + if (!isdigit(*t_name)) > + break; > + } > + > + if (*t_name != '\0') > + return 0; > + > + snprintf(buffer, MAX_PATH, "/proc/%s/comm", proc_entry->d_name); > + comm_fd = open(buffer, O_RDONLY); > + if (comm_fd < 0) > + return 0; > + > + memset(buffer, 0, MAX_PATH); > + retval = read(comm_fd, buffer, MAX_PATH); > + > + close(comm_fd); > + > + if (retval <= 0) > + return 0; > + > + retval = strncmp(comm_prefix, buffer, strlen(comm_prefix)); > + if (retval) > + return 0; Confused. For example: comm_prefix is "osnoise/", buffer is "osnoise\n"(as said by comment below), strlen_prefix is 8. The return value of strncmp() is 1 and not set sched attr. Or use "osnoise" as the comm_prefix. Or am I miss something here. > + /* comm already have \n */ > + debug_msg("Found workload pid:%s comm:%s", proc_entry->d_name, buffer); > + > + return 1; > +} > + > +/* > + * set_comm_sched_attr - set sched params to threads starting with char *comm_prefix > + * > + * This function uses procps to list the currently running threads and then set the s/procps/procfs/ Thanks, Tao > + * sched_attr *attr to the threads that start with char *comm_prefix. It is > * mainly used to set the priority to the kernel threads created by the > * tracers. > */ > -int set_comm_sched_attr(const char *comm, struct sched_attr *attr) > +int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr) > { > - int flags = PROC_FILLCOM | PROC_FILLSTAT; > - PROCTAB *ptp; > - proc_t task; > + struct dirent *proc_entry; > + DIR *procfs; > int retval; > > - ptp = openproc(flags); > - if (!ptp) { > - err_msg("error openproc()\n"); > - return -ENOENT; > + if (strlen(comm_prefix) >= MAX_PATH) { > + err_msg("Command prefix is too long: %d < strlen(%s)\n", > + MAX_PATH, comm_prefix); > + exit(EINVAL); > } > > - memset(&task, 0, sizeof(task)); > + procfs = opendir("/proc"); > + > + while ((proc_entry = readdir(procfs))) { > > - while (readproc(ptp, &task)) { > - retval = strncmp(comm, task.cmd, strlen(comm)); > - if (retval) > + retval = procfs_is_workload_pid(comm_prefix, proc_entry); > + if (!retval) > continue; > - retval = __set_sched_attr(task.tid, attr); > - if (retval) > + > + /* procfs_is_workload_pid confirmed it is a pid */ > + retval = __set_sched_attr(atoi(proc_entry->d_name), attr); > + if (retval) { > + err_msg("Error setting sched attributes for pid:%s\n", proc_entry->d_name); > goto out_err; > - } > + } > > - closeproc(ptp); > + debug_msg("Set sched attributes for pid:%s\n", proc_entry->d_name); > + } > return 0; > > out_err: > - closeproc(ptp); > + closedir(procfs); > return 1; > } > > diff --git a/tools/tracing/rtla/src/utils.h b/tools/tracing/rtla/src/utils.h > index fa08e374870a..5571afd3b549 100644 > --- a/tools/tracing/rtla/src/utils.h > +++ b/tools/tracing/rtla/src/utils.h > @@ -6,6 +6,7 @@ > * '18446744073709551615\0' > */ > #define BUFF_U64_STR_SIZE 24 > +#define MAX_PATH 1024 > > #define container_of(ptr, type, member)({ \ > const typeof(((type *)0)->member) *__mptr = (ptr); \ > @@ -53,5 +54,5 @@ struct sched_attr { > }; > > int parse_prio(char *arg, struct sched_attr *sched_param); > -int set_comm_sched_attr(const char *comm, struct sched_attr *attr); > +int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr); > int set_cpu_dma_latency(int32_t latency); > -- > 2.32.0 >
On 5/6/22 17:06, Tao Zhou wrote: > On Fri, Apr 29, 2022 at 03:01:48PM +0200, > Daniel Bristot de Oliveira wrote: > >> Daniel Wagner reported to me that readproc.h got deprecated. Also, >> while the procps-ng library was available on Fedora, it was not available >> on RHEL, which is a piece of evidence that it was not that used. >> >> rtla uses procps-ng only to find the PID of the tracers' workload. >> >> I used the procps-ng library to avoid reinventing the wheel. But in this >> case, reinventing the wheel took me less time than the time we already >> took trying to work around problems. >> >> Implement a function that reads /proc/ entries, checking if: >> - the entry is a directory >> - the directory name is composed only of digits (PID) >> - the directory contains the comm file >> - the comm file contains a comm that matches the tracers' >> workload prefix. >> - then return true; otherwise, return false. >> >> And use it instead of procps-ng. >> >> Changes from V1: >> - Use a single buffer for comm and comm_path >> - Cause an error in case of a too long command prefix >> - Do a close_dir() >> - Improve log messages >> >> Cc: John Kacur <jkacur@redhat.com> >> Cc: Steven Rostedt <rostedt@goodmis.org> >> Fixes: b1696371d865 ("rtla: Helper functions for rtla") >> Reported-by: Daniel Wagner <dwagner@suse.de> >> Signed-off-by: Daniel Bristot de Oliveira <bristot@kernel.org> [...] >> + >> /* >> - * set_comm_sched_attr - set sched params to threads starting with char *comm >> + * procfs_is_workload_pid - check if a procfs entry contains a comm_prefix* comm >> + * >> + * Check if the procfs entry is a directory of a process, and then check if the >> + * process has a comm with the prefix set in char *comm_prefix. As the >> + * current users of this function only check for kernel threads, there is no >> + * need to check for the threads for the process. >> * >> - * This function uses procps to list the currently running threads and then >> - * set the sched_attr *attr to the threads that start with char *comm. It is >> + * Return: True if the proc_entry contains a comm file with comm_prefix*. >> + * Otherwise returns false. >> + */ >> +static int procfs_is_workload_pid(const char *comm_prefix, struct dirent *proc_entry) >> +{ >> + char buffer[MAX_PATH]; >> + int comm_fd, retval; >> + char*t_name; > > Need a blank.. > char *t_name; right >> + >> + if (proc_entry->d_type != DT_DIR) >> + return 0; >> + >> + if (*proc_entry->d_name == '.') >> + return 0; >> + >> + /* check if the string is a pid */ >> + for (t_name = proc_entry->d_name; t_name; t_name++) { >> + if (!isdigit(*t_name)) >> + break; >> + } >> + >> + if (*t_name != '\0') >> + return 0; >> + >> + snprintf(buffer, MAX_PATH, "/proc/%s/comm", proc_entry->d_name); >> + comm_fd = open(buffer, O_RDONLY); >> + if (comm_fd < 0) >> + return 0; >> + >> + memset(buffer, 0, MAX_PATH); >> + retval = read(comm_fd, buffer, MAX_PATH); >> + >> + close(comm_fd); >> + >> + if (retval <= 0) >> + return 0; >> + >> + retval = strncmp(comm_prefix, buffer, strlen(comm_prefix)); >> + if (retval) >> + return 0; > > > Confused. > > For example: > comm_prefix is "osnoise/", buffer is "osnoise\n"(as said by comment below), > strlen_prefix is 8. The return value of strncmp() is 1 and not set sched attr. it should not set "osnoise" thread priority... so it is correct. > Or use "osnoise" as the comm_prefix. Or am I miss something here. I am looking for "osnoise/$CPU" threads, not "osnoise" threads. Run rtla osnoise with log messages (-D) you will see it finding the expected threads. >> + /* comm already have \n */ >> + debug_msg("Found workload pid:%s comm:%s", proc_entry->d_name, buffer); >> + >> + return 1; >> +} >> + >> +/* >> + * set_comm_sched_attr - set sched params to threads starting with char *comm_prefix >> + * >> + * This function uses procps to list the currently running threads and then set the > > s/procps/procfs/ right Thanks -- Daniel > > Thanks, > Tao
diff --git a/tools/tracing/rtla/Makefile b/tools/tracing/rtla/Makefile index 5a3226e436ef..523f0a8c38c2 100644 --- a/tools/tracing/rtla/Makefile +++ b/tools/tracing/rtla/Makefile @@ -32,7 +32,7 @@ TRACEFS_HEADERS := $$($(PKG_CONFIG) --cflags libtracefs) CFLAGS := -O -g -DVERSION=\"$(VERSION)\" $(FOPTS) $(MOPTS) $(WOPTS) $(TRACEFS_HEADERS) LDFLAGS := -ggdb -LIBS := $$($(PKG_CONFIG) --libs libtracefs) -lprocps +LIBS := $$($(PKG_CONFIG) --libs libtracefs) SRC := $(wildcard src/*.c) HDR := $(wildcard src/*.h) diff --git a/tools/tracing/rtla/README.txt b/tools/tracing/rtla/README.txt index 0fbad2640b8c..4af3fd40f171 100644 --- a/tools/tracing/rtla/README.txt +++ b/tools/tracing/rtla/README.txt @@ -11,7 +11,6 @@ RTLA depends on the following libraries and tools: - libtracefs - libtraceevent - - procps It also depends on python3-docutils to compile man pages. diff --git a/tools/tracing/rtla/src/utils.c b/tools/tracing/rtla/src/utils.c index 3bd6f64780cf..f3d562f7f14c 100644 --- a/tools/tracing/rtla/src/utils.c +++ b/tools/tracing/rtla/src/utils.c @@ -3,7 +3,7 @@ * Copyright (C) 2021 Red Hat Inc, Daniel Bristot de Oliveira <bristot@kernel.org> */ -#include <proc/readproc.h> +#include <dirent.h> #include <stdarg.h> #include <stdlib.h> #include <string.h> @@ -262,43 +262,103 @@ int __set_sched_attr(int pid, struct sched_attr *attr) return 0; } + /* - * set_comm_sched_attr - set sched params to threads starting with char *comm + * procfs_is_workload_pid - check if a procfs entry contains a comm_prefix* comm + * + * Check if the procfs entry is a directory of a process, and then check if the + * process has a comm with the prefix set in char *comm_prefix. As the + * current users of this function only check for kernel threads, there is no + * need to check for the threads for the process. * - * This function uses procps to list the currently running threads and then - * set the sched_attr *attr to the threads that start with char *comm. It is + * Return: True if the proc_entry contains a comm file with comm_prefix*. + * Otherwise returns false. + */ +static int procfs_is_workload_pid(const char *comm_prefix, struct dirent *proc_entry) +{ + char buffer[MAX_PATH]; + int comm_fd, retval; + char*t_name; + + if (proc_entry->d_type != DT_DIR) + return 0; + + if (*proc_entry->d_name == '.') + return 0; + + /* check if the string is a pid */ + for (t_name = proc_entry->d_name; t_name; t_name++) { + if (!isdigit(*t_name)) + break; + } + + if (*t_name != '\0') + return 0; + + snprintf(buffer, MAX_PATH, "/proc/%s/comm", proc_entry->d_name); + comm_fd = open(buffer, O_RDONLY); + if (comm_fd < 0) + return 0; + + memset(buffer, 0, MAX_PATH); + retval = read(comm_fd, buffer, MAX_PATH); + + close(comm_fd); + + if (retval <= 0) + return 0; + + retval = strncmp(comm_prefix, buffer, strlen(comm_prefix)); + if (retval) + return 0; + + /* comm already have \n */ + debug_msg("Found workload pid:%s comm:%s", proc_entry->d_name, buffer); + + return 1; +} + +/* + * set_comm_sched_attr - set sched params to threads starting with char *comm_prefix + * + * This function uses procps to list the currently running threads and then set the + * sched_attr *attr to the threads that start with char *comm_prefix. It is * mainly used to set the priority to the kernel threads created by the * tracers. */ -int set_comm_sched_attr(const char *comm, struct sched_attr *attr) +int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr) { - int flags = PROC_FILLCOM | PROC_FILLSTAT; - PROCTAB *ptp; - proc_t task; + struct dirent *proc_entry; + DIR *procfs; int retval; - ptp = openproc(flags); - if (!ptp) { - err_msg("error openproc()\n"); - return -ENOENT; + if (strlen(comm_prefix) >= MAX_PATH) { + err_msg("Command prefix is too long: %d < strlen(%s)\n", + MAX_PATH, comm_prefix); + exit(EINVAL); } - memset(&task, 0, sizeof(task)); + procfs = opendir("/proc"); + + while ((proc_entry = readdir(procfs))) { - while (readproc(ptp, &task)) { - retval = strncmp(comm, task.cmd, strlen(comm)); - if (retval) + retval = procfs_is_workload_pid(comm_prefix, proc_entry); + if (!retval) continue; - retval = __set_sched_attr(task.tid, attr); - if (retval) + + /* procfs_is_workload_pid confirmed it is a pid */ + retval = __set_sched_attr(atoi(proc_entry->d_name), attr); + if (retval) { + err_msg("Error setting sched attributes for pid:%s\n", proc_entry->d_name); goto out_err; - } + } - closeproc(ptp); + debug_msg("Set sched attributes for pid:%s\n", proc_entry->d_name); + } return 0; out_err: - closeproc(ptp); + closedir(procfs); return 1; } diff --git a/tools/tracing/rtla/src/utils.h b/tools/tracing/rtla/src/utils.h index fa08e374870a..5571afd3b549 100644 --- a/tools/tracing/rtla/src/utils.h +++ b/tools/tracing/rtla/src/utils.h @@ -6,6 +6,7 @@ * '18446744073709551615\0' */ #define BUFF_U64_STR_SIZE 24 +#define MAX_PATH 1024 #define container_of(ptr, type, member)({ \ const typeof(((type *)0)->member) *__mptr = (ptr); \ @@ -53,5 +54,5 @@ struct sched_attr { }; int parse_prio(char *arg, struct sched_attr *sched_param); -int set_comm_sched_attr(const char *comm, struct sched_attr *attr); +int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr); int set_cpu_dma_latency(int32_t latency);
Daniel Wagner reported to me that readproc.h got deprecated. Also, while the procps-ng library was available on Fedora, it was not available on RHEL, which is a piece of evidence that it was not that used. rtla uses procps-ng only to find the PID of the tracers' workload. I used the procps-ng library to avoid reinventing the wheel. But in this case, reinventing the wheel took me less time than the time we already took trying to work around problems. Implement a function that reads /proc/ entries, checking if: - the entry is a directory - the directory name is composed only of digits (PID) - the directory contains the comm file - the comm file contains a comm that matches the tracers' workload prefix. - then return true; otherwise, return false. And use it instead of procps-ng. Changes from V1: - Use a single buffer for comm and comm_path - Cause an error in case of a too long command prefix - Do a close_dir() - Improve log messages Cc: John Kacur <jkacur@redhat.com> Cc: Steven Rostedt <rostedt@goodmis.org> Fixes: b1696371d865 ("rtla: Helper functions for rtla") Reported-by: Daniel Wagner <dwagner@suse.de> Signed-off-by: Daniel Bristot de Oliveira <bristot@kernel.org> --- tools/tracing/rtla/Makefile | 2 +- tools/tracing/rtla/README.txt | 1 - tools/tracing/rtla/src/utils.c | 102 ++++++++++++++++++++++++++------- tools/tracing/rtla/src/utils.h | 3 +- 4 files changed, 84 insertions(+), 24 deletions(-)