1 #include <stdio.h>
2 #include <string.h>
3 #include <fcntl.h>
4 #include <unistd.h>
5 #include <malloc.h>
6 #include <errno.h>
7 #include <sys/param.h>
8 #include <sys/types.h>
9 #include <sys/stat.h>
10 #include <unistd.h>
11
12 extern int init_module(void *, unsigned long, const char *);
13
read_file(const char * filename,ssize_t * _size)14 static void *read_file(const char *filename, ssize_t *_size)
15 {
16 int ret, fd;
17 struct stat sb;
18 ssize_t size;
19 void *buffer = NULL;
20
21 /* open the file */
22 fd = open(filename, O_RDONLY);
23 if (fd < 0)
24 return NULL;
25
26 /* find out how big it is */
27 if (fstat(fd, &sb) < 0)
28 goto bail;
29 size = sb.st_size;
30
31 /* allocate memory for it to be read into */
32 buffer = malloc(size);
33 if (!buffer)
34 goto bail;
35
36 /* slurp it into our buffer */
37 ret = read(fd, buffer, size);
38 if (ret != size)
39 goto bail;
40
41 /* let the caller know how big it is */
42 *_size = size;
43
44 bail:
45 close(fd);
46 return buffer;
47 }
48
insmod_main(int argc,char ** argv)49 int insmod_main(int argc, char **argv)
50 {
51 void *file;
52 ssize_t size = 0;
53 char opts[1024];
54 int ret;
55
56 /* make sure we've got an argument */
57 if (argc < 2) {
58 fprintf(stderr, "usage: insmod <module.o>\n");
59 return -1;
60 }
61
62 /* read the file into memory */
63 file = read_file(argv[1], &size);
64 if (!file) {
65 fprintf(stderr, "insmod: can't open '%s'\n", argv[1]);
66 return -1;
67 }
68
69 opts[0] = '\0';
70 if (argc > 2) {
71 int i, len;
72 char *end = opts + sizeof(opts) - 1;
73 char *ptr = opts;
74
75 for (i = 2; (i < argc) && (ptr < end); i++) {
76 len = MIN(strlen(argv[i]), (size_t)(end - ptr));
77 memcpy(ptr, argv[i], len);
78 ptr += len;
79 *ptr++ = ' ';
80 }
81 *(ptr - 1) = '\0';
82 }
83
84 /* pass it to the kernel */
85 ret = init_module(file, size, opts);
86 if (ret != 0) {
87 fprintf(stderr,
88 "insmod: init_module '%s' failed (%s)\n",
89 argv[1], strerror(errno));
90 }
91
92 /* free the file buffer */
93 free(file);
94
95 return ret;
96 }
97
98