usr/utils/insmod.c

/* insmod.c: insert a module into the kernel.
    Copyright (C) 2001  Rusty Russell.
    Copyright (C) 2002  Rusty Russell, IBM Corporation.

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <errno.h>

#define streq(a,b) (strcmp((a),(b)) == 0)

/* This really needs to be in a header file... */
extern long init_module(void *, unsigned long, const char *);

static void print_usage(const char *progname)
{
	fprintf(stderr, "Usage: %s filename [args]\n", progname);
	exit(1);
}

/* We use error numbers in a loose translation... */
static const char *moderror(int err)
{
	switch (err) {
	case ENOEXEC:
		return "Invalid module format";
	case ENOENT:
		return "Unknown symbol in module";
	case ESRCH:
		return "Module has wrong symbol version";
	case EINVAL:
		return "Invalid parameters";
	default:
		return strerror(err);
	}
}

static void *grab_file(const char *filename, unsigned long *size)
{
	unsigned int max = 16384;
	int ret, fd;
	void *buffer = malloc(max);

	if (streq(filename, "-"))
		fd = dup(STDIN_FILENO);
	else
		fd = open(filename, O_RDONLY, 0);

	if (fd < 0)
		return NULL;

	*size = 0;
	while ((ret = read(fd, buffer + *size, max - *size)) > 0) {
		*size += ret;
		if (*size == max)
			buffer = realloc(buffer, max *= 2);
	}
	if (ret < 0) {
		free(buffer);
		buffer = NULL;
	}
	close(fd);
	return buffer;
}

int main(int argc, char *argv[])
{
	int i;
	long int ret;
	unsigned long len;
	void *file;
	char *filename, *options = strdup("");
	char *progname = argv[0];

	if (argv[1] && (streq(argv[1], "--version") || streq(argv[1], "-V"))) {
		puts("klibc insmod");
		exit(0);
	}

	/* Ignore old options, for backwards compat. */
	while (argv[1] && (streq(argv[1], "-p")
			   || streq(argv[1], "-s")
			   || streq(argv[1], "-f"))) {
		argv++;
		argc--;
	}

	filename = argv[1];
	if (!filename)
		print_usage(progname);

	/* Rest is options */
	for (i = 2; i < argc; i++) {
		options = realloc(options,
				  strlen(options) + 2 + strlen(argv[i]) + 2);
		/* Spaces handled by "" pairs, but no way of escaping
		   quotes */
		if (strchr(argv[i], ' '))
			strcat(options, "\"");
		strcat(options, argv[i]);
		if (strchr(argv[i], ' '))
			strcat(options, "\"");
		strcat(options, " ");
	}

	file = grab_file(filename, &len);
	if (!file) {
		fprintf(stderr, "insmod: can't read '%s': %s\n",
			filename, strerror(errno));
		exit(1);
	}

	ret = init_module(file, len, options);
	if (ret != 0) {
		fprintf(stderr, "insmod: error inserting '%s': %li %s\n",
			filename, ret, moderror(errno));
		exit(1);
	}
	exit(0);
}
1	/* insmod.c: insert a module into the kernel.
2	Copyright (C) 2001 Rusty Russell.
3	Copyright (C) 2002 Rusty Russell, IBM Corporation.
4
5	This program is free software; you can redistribute it and/or modify
6	it under the terms of the GNU General Public License as published by
7	the Free Software Foundation; either version 2 of the License, or
8	(at your option) any later version.
9
10	This program is distributed in the hope that it will be useful,
11	but WITHOUT ANY WARRANTY; without even the implied warranty of
12	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13	GNU General Public License for more details.
14
15	You should have received a copy of the GNU General Public License
16	along with this program; if not, write to the Free Software
17	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18	*/
19	#include <stdio.h>
20	#include <string.h>
21	#include <stdlib.h>
22	#include <unistd.h>
23	#include <fcntl.h>
24	#include <sys/mman.h>
25	#include <sys/types.h>
26	#include <sys/stat.h>
27	#include <unistd.h>
28	#include <errno.h>
29
30	#define streq(a,b) (strcmp((a),(b)) == 0)
31
32	/* This really needs to be in a header file... */
33	extern long init_module(void , unsigned long, const char );
34
35	static void print_usage(const char *progname)
36	{
37	fprintf(stderr, "Usage: %s filename [args]\n", progname);
38	exit(1);
39	}
40
41	/* We use error numbers in a loose translation... */
42	static const char *moderror(int err)
43	{
44	switch (err) {
45	case ENOEXEC:
46	return "Invalid module format";
47	case ENOENT:
48	return "Unknown symbol in module";
49	case ESRCH:
50	return "Module has wrong symbol version";
51	case EINVAL:
52	return "Invalid parameters";
53	default:
54	return strerror(err);
55	}
56	}
57
58	static void grab_file(const char filename, unsigned long *size)
59	{
60	unsigned int max = 16384;
61	int ret, fd;
62	void *buffer = malloc(max);
63
64	if (streq(filename, "-"))
65	fd = dup(STDIN_FILENO);
66	else
67	fd = open(filename, O_RDONLY, 0);
68
69	if (fd < 0)
70	return NULL;
71
72	*size = 0;
73	while ((ret = read(fd, buffer + size, max - size)) > 0) {
74	*size += ret;
75	if (*size == max)
76	buffer = realloc(buffer, max *= 2);
77	}
78	if (ret < 0) {
79	free(buffer);
80	buffer = NULL;
81	}
82	close(fd);
83	return buffer;
84	}
85
86	int main(int argc, char *argv[])
87	{
88	int i;
89	long int ret;
90	unsigned long len;
91	void *file;
92	char filename, options = strdup("");
93	char *progname = argv[0];
94
95	if (argv[1] && (streq(argv[1], "--version") \|\| streq(argv[1], "-V"))) {
96	puts("klibc insmod");
97	exit(0);
98	}
99
100	/* Ignore old options, for backwards compat. */
101	while (argv[1] && (streq(argv[1], "-p")
102	\|\| streq(argv[1], "-s")
103	\|\| streq(argv[1], "-f"))) {
104	argv++;
105	argc--;
106	}
107
108	filename = argv[1];
109	if (!filename)
110	print_usage(progname);
111
112	/* Rest is options */
113	for (i = 2; i < argc; i++) {
114	options = realloc(options,
115	strlen(options) + 2 + strlen(argv[i]) + 2);
116	/* Spaces handled by "" pairs, but no way of escaping
117	quotes */
118	if (strchr(argv[i], ' '))
119	strcat(options, "\"");
120	strcat(options, argv[i]);
121	if (strchr(argv[i], ' '))
122	strcat(options, "\"");
123	strcat(options, " ");
124	}
125
126	file = grab_file(filename, &len);
127	if (!file) {
128	fprintf(stderr, "insmod: can't read '%s': %s\n",
129	filename, strerror(errno));
130	exit(1);
131	}
132
133	ret = init_module(file, len, options);
134	if (ret != 0) {
135	fprintf(stderr, "insmod: error inserting '%s': %li %s\n",
136	filename, ret, moderror(errno));
137	exit(1);
138	}
139	exit(0);
140	}