deal with resource allocation bugs in arcmsr
[pv_ops_mirror.git] / fs / binfmt_misc.c
blobb53c7e5f41bbcffd8f936642117ecfce5f258a21
1 /*
2 * binfmt_misc.c
4 * Copyright (C) 1997 Richard Günther
6 * binfmt_misc detects binaries via a magic or filename extension and invokes
7 * a specified wrapper. This should obsolete binfmt_java, binfmt_em86 and
8 * binfmt_mz.
10 * 1997-04-25 first version
11 * [...]
12 * 1997-05-19 cleanup
13 * 1997-06-26 hpa: pass the real filename rather than argv[0]
14 * 1997-06-30 minor cleanup
15 * 1997-08-09 removed extension stripping, locking cleanup
16 * 2001-02-28 AV: rewritten into something that resembles C. Original didn't.
19 #include <linux/module.h>
20 #include <linux/init.h>
21 #include <linux/sched.h>
22 #include <linux/binfmts.h>
23 #include <linux/slab.h>
24 #include <linux/ctype.h>
25 #include <linux/file.h>
26 #include <linux/pagemap.h>
27 #include <linux/namei.h>
28 #include <linux/mount.h>
29 #include <linux/syscalls.h>
31 #include <asm/uaccess.h>
33 enum {
34 VERBOSE_STATUS = 1 /* make it zero to save 400 bytes kernel memory */
37 static LIST_HEAD(entries);
38 static int enabled = 1;
40 enum {Enabled, Magic};
41 #define MISC_FMT_PRESERVE_ARGV0 (1<<31)
42 #define MISC_FMT_OPEN_BINARY (1<<30)
43 #define MISC_FMT_CREDENTIALS (1<<29)
45 typedef struct {
46 struct list_head list;
47 unsigned long flags; /* type, status, etc. */
48 int offset; /* offset of magic */
49 int size; /* size of magic/mask */
50 char *magic; /* magic or filename extension */
51 char *mask; /* mask, NULL for exact match */
52 char *interpreter; /* filename of interpreter */
53 char *name;
54 struct dentry *dentry;
55 } Node;
57 static DEFINE_RWLOCK(entries_lock);
58 static struct file_system_type bm_fs_type;
59 static struct vfsmount *bm_mnt;
60 static int entry_count;
62 /*
63 * Check if we support the binfmt
64 * if we do, return the node, else NULL
65 * locking is done in load_misc_binary
67 static Node *check_file(struct linux_binprm *bprm)
69 char *p = strrchr(bprm->interp, '.');
70 struct list_head *l;
72 list_for_each(l, &entries) {
73 Node *e = list_entry(l, Node, list);
74 char *s;
75 int j;
77 if (!test_bit(Enabled, &e->flags))
78 continue;
80 if (!test_bit(Magic, &e->flags)) {
81 if (p && !strcmp(e->magic, p + 1))
82 return e;
83 continue;
86 s = bprm->buf + e->offset;
87 if (e->mask) {
88 for (j = 0; j < e->size; j++)
89 if ((*s++ ^ e->magic[j]) & e->mask[j])
90 break;
91 } else {
92 for (j = 0; j < e->size; j++)
93 if ((*s++ ^ e->magic[j]))
94 break;
96 if (j == e->size)
97 return e;
99 return NULL;
103 * the loader itself
105 static int load_misc_binary(struct linux_binprm *bprm, struct pt_regs *regs)
107 Node *fmt;
108 struct file * interp_file = NULL;
109 char iname[BINPRM_BUF_SIZE];
110 char *iname_addr = iname;
111 int retval;
112 int fd_binary = -1;
113 struct files_struct *files = NULL;
115 retval = -ENOEXEC;
116 if (!enabled)
117 goto _ret;
119 /* to keep locking time low, we copy the interpreter string */
120 read_lock(&entries_lock);
121 fmt = check_file(bprm);
122 if (fmt)
123 strlcpy(iname, fmt->interpreter, BINPRM_BUF_SIZE);
124 read_unlock(&entries_lock);
125 if (!fmt)
126 goto _ret;
128 if (!(fmt->flags & MISC_FMT_PRESERVE_ARGV0)) {
129 retval = remove_arg_zero(bprm);
130 if (retval)
131 goto _ret;
134 if (fmt->flags & MISC_FMT_OPEN_BINARY) {
136 files = current->files;
137 retval = unshare_files();
138 if (retval < 0)
139 goto _ret;
140 if (files == current->files) {
141 put_files_struct(files);
142 files = NULL;
144 /* if the binary should be opened on behalf of the
145 * interpreter than keep it open and assign descriptor
146 * to it */
147 fd_binary = get_unused_fd();
148 if (fd_binary < 0) {
149 retval = fd_binary;
150 goto _unshare;
152 fd_install(fd_binary, bprm->file);
154 /* if the binary is not readable than enforce mm->dumpable=0
155 regardless of the interpreter's permissions */
156 if (file_permission(bprm->file, MAY_READ))
157 bprm->interp_flags |= BINPRM_FLAGS_ENFORCE_NONDUMP;
159 allow_write_access(bprm->file);
160 bprm->file = NULL;
162 /* mark the bprm that fd should be passed to interp */
163 bprm->interp_flags |= BINPRM_FLAGS_EXECFD;
164 bprm->interp_data = fd_binary;
166 } else {
167 allow_write_access(bprm->file);
168 fput(bprm->file);
169 bprm->file = NULL;
171 /* make argv[1] be the path to the binary */
172 retval = copy_strings_kernel (1, &bprm->interp, bprm);
173 if (retval < 0)
174 goto _error;
175 bprm->argc++;
177 /* add the interp as argv[0] */
178 retval = copy_strings_kernel (1, &iname_addr, bprm);
179 if (retval < 0)
180 goto _error;
181 bprm->argc ++;
183 bprm->interp = iname; /* for binfmt_script */
185 interp_file = open_exec (iname);
186 retval = PTR_ERR (interp_file);
187 if (IS_ERR (interp_file))
188 goto _error;
190 bprm->file = interp_file;
191 if (fmt->flags & MISC_FMT_CREDENTIALS) {
193 * No need to call prepare_binprm(), it's already been
194 * done. bprm->buf is stale, update from interp_file.
196 memset(bprm->buf, 0, BINPRM_BUF_SIZE);
197 retval = kernel_read(bprm->file, 0, bprm->buf, BINPRM_BUF_SIZE);
198 } else
199 retval = prepare_binprm (bprm);
201 if (retval < 0)
202 goto _error;
204 retval = search_binary_handler (bprm, regs);
205 if (retval < 0)
206 goto _error;
208 if (files) {
209 put_files_struct(files);
210 files = NULL;
212 _ret:
213 return retval;
214 _error:
215 if (fd_binary > 0)
216 sys_close(fd_binary);
217 bprm->interp_flags = 0;
218 bprm->interp_data = 0;
219 _unshare:
220 if (files)
221 reset_files_struct(current, files);
222 goto _ret;
225 /* Command parsers */
228 * parses and copies one argument enclosed in del from *sp to *dp,
229 * recognising the \x special.
230 * returns pointer to the copied argument or NULL in case of an
231 * error (and sets err) or null argument length.
233 static char *scanarg(char *s, char del)
235 char c;
237 while ((c = *s++) != del) {
238 if (c == '\\' && *s == 'x') {
239 s++;
240 if (!isxdigit(*s++))
241 return NULL;
242 if (!isxdigit(*s++))
243 return NULL;
246 return s;
249 static int unquote(char *from)
251 char c = 0, *s = from, *p = from;
253 while ((c = *s++) != '\0') {
254 if (c == '\\' && *s == 'x') {
255 s++;
256 c = toupper(*s++);
257 *p = (c - (isdigit(c) ? '0' : 'A' - 10)) << 4;
258 c = toupper(*s++);
259 *p++ |= c - (isdigit(c) ? '0' : 'A' - 10);
260 continue;
262 *p++ = c;
264 return p - from;
267 static char * check_special_flags (char * sfs, Node * e)
269 char * p = sfs;
270 int cont = 1;
272 /* special flags */
273 while (cont) {
274 switch (*p) {
275 case 'P':
276 p++;
277 e->flags |= MISC_FMT_PRESERVE_ARGV0;
278 break;
279 case 'O':
280 p++;
281 e->flags |= MISC_FMT_OPEN_BINARY;
282 break;
283 case 'C':
284 p++;
285 /* this flags also implies the
286 open-binary flag */
287 e->flags |= (MISC_FMT_CREDENTIALS |
288 MISC_FMT_OPEN_BINARY);
289 break;
290 default:
291 cont = 0;
295 return p;
298 * This registers a new binary format, it recognises the syntax
299 * ':name:type:offset:magic:mask:interpreter:flags'
300 * where the ':' is the IFS, that can be chosen with the first char
302 static Node *create_entry(const char __user *buffer, size_t count)
304 Node *e;
305 int memsize, err;
306 char *buf, *p;
307 char del;
309 /* some sanity checks */
310 err = -EINVAL;
311 if ((count < 11) || (count > 256))
312 goto out;
314 err = -ENOMEM;
315 memsize = sizeof(Node) + count + 8;
316 e = kmalloc(memsize, GFP_USER);
317 if (!e)
318 goto out;
320 p = buf = (char *)e + sizeof(Node);
322 memset(e, 0, sizeof(Node));
323 if (copy_from_user(buf, buffer, count))
324 goto Efault;
326 del = *p++; /* delimeter */
328 memset(buf+count, del, 8);
330 e->name = p;
331 p = strchr(p, del);
332 if (!p)
333 goto Einval;
334 *p++ = '\0';
335 if (!e->name[0] ||
336 !strcmp(e->name, ".") ||
337 !strcmp(e->name, "..") ||
338 strchr(e->name, '/'))
339 goto Einval;
340 switch (*p++) {
341 case 'E': e->flags = 1<<Enabled; break;
342 case 'M': e->flags = (1<<Enabled) | (1<<Magic); break;
343 default: goto Einval;
345 if (*p++ != del)
346 goto Einval;
347 if (test_bit(Magic, &e->flags)) {
348 char *s = strchr(p, del);
349 if (!s)
350 goto Einval;
351 *s++ = '\0';
352 e->offset = simple_strtoul(p, &p, 10);
353 if (*p++)
354 goto Einval;
355 e->magic = p;
356 p = scanarg(p, del);
357 if (!p)
358 goto Einval;
359 p[-1] = '\0';
360 if (!e->magic[0])
361 goto Einval;
362 e->mask = p;
363 p = scanarg(p, del);
364 if (!p)
365 goto Einval;
366 p[-1] = '\0';
367 if (!e->mask[0])
368 e->mask = NULL;
369 e->size = unquote(e->magic);
370 if (e->mask && unquote(e->mask) != e->size)
371 goto Einval;
372 if (e->size + e->offset > BINPRM_BUF_SIZE)
373 goto Einval;
374 } else {
375 p = strchr(p, del);
376 if (!p)
377 goto Einval;
378 *p++ = '\0';
379 e->magic = p;
380 p = strchr(p, del);
381 if (!p)
382 goto Einval;
383 *p++ = '\0';
384 if (!e->magic[0] || strchr(e->magic, '/'))
385 goto Einval;
386 p = strchr(p, del);
387 if (!p)
388 goto Einval;
389 *p++ = '\0';
391 e->interpreter = p;
392 p = strchr(p, del);
393 if (!p)
394 goto Einval;
395 *p++ = '\0';
396 if (!e->interpreter[0])
397 goto Einval;
400 p = check_special_flags (p, e);
402 if (*p == '\n')
403 p++;
404 if (p != buf + count)
405 goto Einval;
406 return e;
408 out:
409 return ERR_PTR(err);
411 Efault:
412 kfree(e);
413 return ERR_PTR(-EFAULT);
414 Einval:
415 kfree(e);
416 return ERR_PTR(-EINVAL);
420 * Set status of entry/binfmt_misc:
421 * '1' enables, '0' disables and '-1' clears entry/binfmt_misc
423 static int parse_command(const char __user *buffer, size_t count)
425 char s[4];
427 if (!count)
428 return 0;
429 if (count > 3)
430 return -EINVAL;
431 if (copy_from_user(s, buffer, count))
432 return -EFAULT;
433 if (s[count-1] == '\n')
434 count--;
435 if (count == 1 && s[0] == '0')
436 return 1;
437 if (count == 1 && s[0] == '1')
438 return 2;
439 if (count == 2 && s[0] == '-' && s[1] == '1')
440 return 3;
441 return -EINVAL;
444 /* generic stuff */
446 static void entry_status(Node *e, char *page)
448 char *dp;
449 char *status = "disabled";
450 const char * flags = "flags: ";
452 if (test_bit(Enabled, &e->flags))
453 status = "enabled";
455 if (!VERBOSE_STATUS) {
456 sprintf(page, "%s\n", status);
457 return;
460 sprintf(page, "%s\ninterpreter %s\n", status, e->interpreter);
461 dp = page + strlen(page);
463 /* print the special flags */
464 sprintf (dp, "%s", flags);
465 dp += strlen (flags);
466 if (e->flags & MISC_FMT_PRESERVE_ARGV0) {
467 *dp ++ = 'P';
469 if (e->flags & MISC_FMT_OPEN_BINARY) {
470 *dp ++ = 'O';
472 if (e->flags & MISC_FMT_CREDENTIALS) {
473 *dp ++ = 'C';
475 *dp ++ = '\n';
478 if (!test_bit(Magic, &e->flags)) {
479 sprintf(dp, "extension .%s\n", e->magic);
480 } else {
481 int i;
483 sprintf(dp, "offset %i\nmagic ", e->offset);
484 dp = page + strlen(page);
485 for (i = 0; i < e->size; i++) {
486 sprintf(dp, "%02x", 0xff & (int) (e->magic[i]));
487 dp += 2;
489 if (e->mask) {
490 sprintf(dp, "\nmask ");
491 dp += 6;
492 for (i = 0; i < e->size; i++) {
493 sprintf(dp, "%02x", 0xff & (int) (e->mask[i]));
494 dp += 2;
497 *dp++ = '\n';
498 *dp = '\0';
502 static struct inode *bm_get_inode(struct super_block *sb, int mode)
504 struct inode * inode = new_inode(sb);
506 if (inode) {
507 inode->i_mode = mode;
508 inode->i_uid = 0;
509 inode->i_gid = 0;
510 inode->i_blocks = 0;
511 inode->i_atime = inode->i_mtime = inode->i_ctime =
512 current_fs_time(inode->i_sb);
514 return inode;
517 static void bm_clear_inode(struct inode *inode)
519 kfree(inode->i_private);
522 static void kill_node(Node *e)
524 struct dentry *dentry;
526 write_lock(&entries_lock);
527 dentry = e->dentry;
528 if (dentry) {
529 list_del_init(&e->list);
530 e->dentry = NULL;
532 write_unlock(&entries_lock);
534 if (dentry) {
535 dentry->d_inode->i_nlink--;
536 d_drop(dentry);
537 dput(dentry);
538 simple_release_fs(&bm_mnt, &entry_count);
542 /* /<entry> */
544 static ssize_t
545 bm_entry_read(struct file * file, char __user * buf, size_t nbytes, loff_t *ppos)
547 Node *e = file->f_path.dentry->d_inode->i_private;
548 loff_t pos = *ppos;
549 ssize_t res;
550 char *page;
551 int len;
553 if (!(page = (char*) __get_free_page(GFP_KERNEL)))
554 return -ENOMEM;
556 entry_status(e, page);
557 len = strlen(page);
559 res = -EINVAL;
560 if (pos < 0)
561 goto out;
562 res = 0;
563 if (pos >= len)
564 goto out;
565 if (len < pos + nbytes)
566 nbytes = len - pos;
567 res = -EFAULT;
568 if (copy_to_user(buf, page + pos, nbytes))
569 goto out;
570 *ppos = pos + nbytes;
571 res = nbytes;
572 out:
573 free_page((unsigned long) page);
574 return res;
577 static ssize_t bm_entry_write(struct file *file, const char __user *buffer,
578 size_t count, loff_t *ppos)
580 struct dentry *root;
581 Node *e = file->f_path.dentry->d_inode->i_private;
582 int res = parse_command(buffer, count);
584 switch (res) {
585 case 1: clear_bit(Enabled, &e->flags);
586 break;
587 case 2: set_bit(Enabled, &e->flags);
588 break;
589 case 3: root = dget(file->f_path.mnt->mnt_sb->s_root);
590 mutex_lock(&root->d_inode->i_mutex);
592 kill_node(e);
594 mutex_unlock(&root->d_inode->i_mutex);
595 dput(root);
596 break;
597 default: return res;
599 return count;
602 static const struct file_operations bm_entry_operations = {
603 .read = bm_entry_read,
604 .write = bm_entry_write,
607 /* /register */
609 static ssize_t bm_register_write(struct file *file, const char __user *buffer,
610 size_t count, loff_t *ppos)
612 Node *e;
613 struct inode *inode;
614 struct dentry *root, *dentry;
615 struct super_block *sb = file->f_path.mnt->mnt_sb;
616 int err = 0;
618 e = create_entry(buffer, count);
620 if (IS_ERR(e))
621 return PTR_ERR(e);
623 root = dget(sb->s_root);
624 mutex_lock(&root->d_inode->i_mutex);
625 dentry = lookup_one_len(e->name, root, strlen(e->name));
626 err = PTR_ERR(dentry);
627 if (IS_ERR(dentry))
628 goto out;
630 err = -EEXIST;
631 if (dentry->d_inode)
632 goto out2;
634 inode = bm_get_inode(sb, S_IFREG | 0644);
636 err = -ENOMEM;
637 if (!inode)
638 goto out2;
640 err = simple_pin_fs(&bm_fs_type, &bm_mnt, &entry_count);
641 if (err) {
642 iput(inode);
643 inode = NULL;
644 goto out2;
647 e->dentry = dget(dentry);
648 inode->i_private = e;
649 inode->i_fop = &bm_entry_operations;
651 d_instantiate(dentry, inode);
652 write_lock(&entries_lock);
653 list_add(&e->list, &entries);
654 write_unlock(&entries_lock);
656 err = 0;
657 out2:
658 dput(dentry);
659 out:
660 mutex_unlock(&root->d_inode->i_mutex);
661 dput(root);
663 if (err) {
664 kfree(e);
665 return -EINVAL;
667 return count;
670 static const struct file_operations bm_register_operations = {
671 .write = bm_register_write,
674 /* /status */
676 static ssize_t
677 bm_status_read(struct file *file, char __user *buf, size_t nbytes, loff_t *ppos)
679 char *s = enabled ? "enabled" : "disabled";
681 return simple_read_from_buffer(buf, nbytes, ppos, s, strlen(s));
684 static ssize_t bm_status_write(struct file * file, const char __user * buffer,
685 size_t count, loff_t *ppos)
687 int res = parse_command(buffer, count);
688 struct dentry *root;
690 switch (res) {
691 case 1: enabled = 0; break;
692 case 2: enabled = 1; break;
693 case 3: root = dget(file->f_path.mnt->mnt_sb->s_root);
694 mutex_lock(&root->d_inode->i_mutex);
696 while (!list_empty(&entries))
697 kill_node(list_entry(entries.next, Node, list));
699 mutex_unlock(&root->d_inode->i_mutex);
700 dput(root);
701 default: return res;
703 return count;
706 static const struct file_operations bm_status_operations = {
707 .read = bm_status_read,
708 .write = bm_status_write,
711 /* Superblock handling */
713 static const struct super_operations s_ops = {
714 .statfs = simple_statfs,
715 .clear_inode = bm_clear_inode,
718 static int bm_fill_super(struct super_block * sb, void * data, int silent)
720 static struct tree_descr bm_files[] = {
721 [2] = {"status", &bm_status_operations, S_IWUSR|S_IRUGO},
722 [3] = {"register", &bm_register_operations, S_IWUSR},
723 /* last one */ {""}
725 int err = simple_fill_super(sb, 0x42494e4d, bm_files);
726 if (!err)
727 sb->s_op = &s_ops;
728 return err;
731 static int bm_get_sb(struct file_system_type *fs_type,
732 int flags, const char *dev_name, void *data, struct vfsmount *mnt)
734 return get_sb_single(fs_type, flags, data, bm_fill_super, mnt);
737 static struct linux_binfmt misc_format = {
738 .module = THIS_MODULE,
739 .load_binary = load_misc_binary,
742 static struct file_system_type bm_fs_type = {
743 .owner = THIS_MODULE,
744 .name = "binfmt_misc",
745 .get_sb = bm_get_sb,
746 .kill_sb = kill_litter_super,
749 static int __init init_misc_binfmt(void)
751 int err = register_filesystem(&bm_fs_type);
752 if (!err) {
753 err = register_binfmt(&misc_format);
754 if (err)
755 unregister_filesystem(&bm_fs_type);
757 return err;
760 static void __exit exit_misc_binfmt(void)
762 unregister_binfmt(&misc_format);
763 unregister_filesystem(&bm_fs_type);
766 core_initcall(init_misc_binfmt);
767 module_exit(exit_misc_binfmt);
768 MODULE_LICENSE("GPL");