From 8b3a46d2b023e3f4d5cd6778cdf17ecd3d5911d7 Mon Sep 17 00:00:00 2001 From: Karel Zak Date: Mon, 2 Jul 2007 13:48:15 +0200 Subject: sys-utils: added setarch command This patch merges setarch command to util-linux-ng code. Signed-off-by: Karel Zak --- sys-utils/setarch.c | 216 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 216 insertions(+) create mode 100644 sys-utils/setarch.c (limited to 'sys-utils/setarch.c') diff --git a/sys-utils/setarch.c b/sys-utils/setarch.c new file mode 100644 index 000000000..1af4c6340 --- /dev/null +++ b/sys-utils/setarch.c @@ -0,0 +1,216 @@ +/* Copyright (C) 2003-2006 Red Hat, Inc. + * Licensed under the terms of the GPL + * + * Written by Elliot Lee + * New personality options & code added by Jindrich Novy + * ADD_NO_RANDOMIZE flag added by Arjan van de Ven + * Help and MIPS support from Mike Frysinger (vapier@gentoo.org) + * Better error handling from Dmitry V. Levin (ldv@altlinux.org) + * + * based on ideas from the ppc32 util by Guy Streeter (2002-01), based on the + * sparc32 util by Jakub Jelinek (1998, 1999) + */ + +#ifndef _GNU_SOURCE +# define _GNU_SOURCE +#endif + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#define set_pers(pers) ((long)syscall(SYS_personality, pers)) + +struct { + char c; + const char *name; + unsigned int option; +} flags[] = { + {'R', "ADDR_NO_RANDOMIZE", 0x0040000}, + {'F', "FDPIC_FUNCPTRS", 0x0080000}, + {'Z', "MMAP_PAGE_ZERO", 0x0100000}, + {'L', "ADDR_COMPAT_LAYOUT", 0x0200000}, + {'X', "READ_IMPLIES_EXEC", 0x0400000}, + {'B', "ADDR_LIMIT_32BIT", 0x0800000}, + {'I', "SHORT_INODE", 0x1000000}, + {'S', "WHOLE_SECONDS", 0x2000000}, + {'T', "STICKY_TIMEOUTS", 0x4000000}, + {'3', "ADDR_LIMIT_3GB", 0x8000000} +}; + +static void __attribute__((__noreturn__)) +show_help(void) +{ + int f; + const char *p = program_invocation_short_name; + + if (!*p) + p = "setarch"; + + printf("Usage: %s%s [options] [program [program arguments]]\n\nOptions:\n", + p, !strcmp(p, "setarch") ? " " : ""); + + for (f = 0; f < sizeof(flags) / sizeof(flags[0]); f++) + printf("\t-%c\tEnable %s\n", flags[f].c, flags[f].name); + + printf("\nFor more information see setarch(8).\n"); + exit(EXIT_SUCCESS); +} + +static void __attribute__((__noreturn__)) +show_usage(const char *s) +{ + const char *p = program_invocation_short_name; + + if (!*p) + p = "setarch"; + + fprintf(stderr, "%s: %s\nTry `%s --help' for more information.\n", p, s, p); + exit(EXIT_FAILURE); +} + +int set_arch(const char *pers, unsigned long options) +{ + struct utsname un; + int i; + unsigned long pers_value, res; + + struct { + int perval; + const char *target_arch, *result_arch; + } transitions[] = { + {PER_LINUX32, "linux32", NULL}, + {PER_LINUX, "linux64", NULL}, +#if defined(__powerpc__) || defined(__powerpc64__) + {PER_LINUX32, "ppc32", "ppc"}, + {PER_LINUX32, "ppc", "ppc"}, + {PER_LINUX, "ppc64", "ppc64"}, + {PER_LINUX, "ppc64pseries", "ppc64"}, + {PER_LINUX, "ppc64iseries", "ppc64"}, +#endif +#if defined(__x86_64__) || defined(__i386__) || defined(__ia64__) + {PER_LINUX32, "i386", "i386"}, + {PER_LINUX32, "i486", "i386"}, + {PER_LINUX32, "i586", "i386"}, + {PER_LINUX32, "i686", "i386"}, + {PER_LINUX32, "athlon", "i386"}, +#endif +#if defined(__x86_64__) || defined(__i386__) + {PER_LINUX, "x86_64", "x86_64"}, +#endif +#if defined(__ia64__) || defined(__i386__) + {PER_LINUX, "ia64", "ia64"}, +#endif +#if defined(__s390x__) || defined(__s390__) + {PER_LINUX32, "s390", "s390"}, + {PER_LINUX, "s390x", "s390x"}, +#endif +#if defined(__sparc64__) || defined(__sparc__) + {PER_LINUX32, "sparc", "sparc"}, + {PER_LINUX32, "sparc32", "sparc"}, + {PER_LINUX, "sparc64", "sparc64"}, +#endif +#if defined(__mips64__) || defined(__mips__) + {PER_LINUX32, "mips32", "mips"}, + {PER_LINUX32, "mips", "mips"}, + {PER_LINUX, "mips64", "mips64"}, +#endif + {-1, NULL, NULL} + }; + + for(i = 0; transitions[i].perval >= 0; i++) + if(!strcmp(pers, transitions[i].target_arch)) + break; + + if(transitions[i].perval < 0) + error(EXIT_FAILURE, 0, "%s: Unrecognized architecture", pers); + + pers_value = transitions[i].perval | options; + res = set_pers(pers_value); + if(res == -EINVAL) + return 1; + + uname(&un); + if(transitions[i].result_arch && + strcmp(un.machine, transitions[i].result_arch)) + { + if(strcmp(transitions[i].result_arch, "i386") + || (strcmp(un.machine, "i486") + && strcmp(un.machine, "i586") + && strcmp(un.machine, "i686") + && strcmp(un.machine, "athlon"))) + error(EXIT_FAILURE, 0, "%s: Unrecognized architecture", pers); + } + + return 0; +} + +int main(int argc, char *argv[]) +{ + const char *p; + unsigned long options = 0; + int verbose = 0; + + if (argc < 1) + show_usage("Not enough arguments"); + + p = program_invocation_short_name; + if (!strcmp(p, "setarch")) { + argv++; + argc--; + if (argc < 1) + show_usage("Not enough arguments"); + p = argv[0]; + if (!strcmp(p, "-h") || !strcmp(p, "--help")) + show_help(); + } + for (argv++, argc--; argc && argv[0][0] == '-'; argv++, argc--) { + int n, unknown = 1; + const char *arg = argv[0]; + + if (!strcmp(arg, "--help")) + show_help(); + + for (n = 1; arg[n]; n++) { + int f; + + if (arg[n] == 'v') { + verbose = 1; + continue; + } + + if (arg[n] == 'h') + show_help(); + + for (f = 0; f < sizeof(flags) / sizeof(flags[0]); f++) { + if (arg[n] == flags[f].c) { + if (verbose) + fprintf(stderr, "Switching on %s.\n", flags[f].name); + options |= flags[f].option; + unknown = 0; + break; + } + } + if (unknown) + error(0, 0, "Unknown option `%c' ignored", arg[n]); + } + } + + if (set_arch(p, options)) + error(EXIT_FAILURE, errno, "Failed to set personality to %s", p); + + if (!argc) { + execl("/bin/sh", "-sh", NULL); + error(EXIT_FAILURE, errno, "/bin/sh"); + } + + execvp(argv[0], argv); + error(EXIT_FAILURE, errno, "%s", argv[0]); + return EXIT_FAILURE; +} -- cgit v1.2.3-55-g7522