diff options
| -rw-r--r-- | src/kmodule.c | 50 |
1 files changed, 29 insertions, 21 deletions
diff --git a/src/kmodule.c b/src/kmodule.c index 5b0340ee..e2bcfb96 100644 --- a/src/kmodule.c +++ b/src/kmodule.c @@ -10,7 +10,7 @@ */ #include <ctype.h> -#include <ftw.h> +#include <fcntl.h> #include <stdio.h> #include <stdlib.h> #include <string.h> @@ -19,40 +19,48 @@ #include <popt.h> +#include <sys/mman.h> #include <sys/utsname.h> #include <kudzu/kudzu.h> - -/* HACK. This is so not thread-safe. */ -static char mod_name[100], cmod_name[100]; - -static int isModule(const char *filename, const struct stat *sb, int flag) { - char *fname = basename(filename); - if ((!strcmp(fname,mod_name) || !strcmp(fname,cmod_name))) { - return 1; +char *setupFile() +{ + struct stat sbuf; + char path[512]; + int fd; + struct utsname utsbuf; + char *buf = NULL; + + uname(&utsbuf); + snprintf(path,512,"/lib/modules/%s/modules.dep",utsbuf.release); + if (!stat(path,&sbuf)) { + fd = open(path,O_RDONLY); + buf = mmap(0,sbuf.st_size,PROT_READ,MAP_SHARED,fd,0); + close(fd); } - return 0; + return buf; } - int isAvailable(char *modulename) { - struct utsname utsbuf; - char path[512]; + char mod_name[100]; + static char *buf = NULL; - uname(&utsbuf); - snprintf(mod_name,100,"%s.ko",modulename); - snprintf(cmod_name,100,"%s.ko.gz",modulename); - snprintf(path,512,"/lib/modules/%s/kernel",utsbuf.release); - /* Do not set the third argument of this function to < 6. Blarg. */ - if (ftw(path,isModule,15) == 1) { - return 1; + if (!buf) { + buf = setupFile(); + if (!buf) + return 0; } + snprintf(mod_name,100,"/%s.ko:",modulename); + if (strstr(buf,mod_name)) + return 1; + snprintf(mod_name,100,"/%s.ko.gz:",modulename); + if (strstr(buf,mod_name)) + return 1; return 0; } - int main(int argc, char **argv) { char *bus = NULL, *class = NULL; |