16d756449SSean Bruno /*- 2910938f0SSean Bruno * Copyright (c) 2013-16, Stacey D. Son 36d756449SSean Bruno * All rights reserved. 46d756449SSean Bruno * 56d756449SSean Bruno * Redistribution and use in source and binary forms, with or without 66d756449SSean Bruno * modification, are permitted provided that the following conditions 76d756449SSean Bruno * are met: 86d756449SSean Bruno * 1. Redistributions of source code must retain the above copyright 96d756449SSean Bruno * notice, this list of conditions and the following disclaimer. 106d756449SSean Bruno * 2. Redistributions in binary form must reproduce the above copyright 116d756449SSean Bruno * notice, this list of conditions and the following disclaimer in the 126d756449SSean Bruno * documentation and/or other materials provided with the distribution. 136d756449SSean Bruno * 146d756449SSean Bruno * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 156d756449SSean Bruno * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 166d756449SSean Bruno * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 176d756449SSean Bruno * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 186d756449SSean Bruno * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 196d756449SSean Bruno * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 206d756449SSean Bruno * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 216d756449SSean Bruno * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 226d756449SSean Bruno * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 236d756449SSean Bruno * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 246d756449SSean Bruno * SUCH DAMAGE. 256d756449SSean Bruno */ 266d756449SSean Bruno 276d756449SSean Bruno #include <sys/cdefs.h> 286d756449SSean Bruno __FBSDID("$FreeBSD$"); 296d756449SSean Bruno 306d756449SSean Bruno #include <sys/param.h> 316d756449SSean Bruno #include <sys/ctype.h> 326d756449SSean Bruno #include <sys/exec.h> 336d756449SSean Bruno #include <sys/imgact.h> 346d756449SSean Bruno #include <sys/imgact_binmisc.h> 356d756449SSean Bruno #include <sys/kernel.h> 366d756449SSean Bruno #include <sys/lock.h> 376d756449SSean Bruno #include <sys/malloc.h> 386d756449SSean Bruno #include <sys/mutex.h> 39df69035dSKyle Evans #include <sys/sbuf.h> 406d756449SSean Bruno #include <sys/sysctl.h> 415f98711dSSean Bruno #include <sys/sx.h> 425f98711dSSean Bruno 435f98711dSSean Bruno #include <machine/atomic.h> 446d756449SSean Bruno 456d756449SSean Bruno /** 466d756449SSean Bruno * Miscellaneous binary interpreter image activator. 476d756449SSean Bruno * 486d756449SSean Bruno * If the given target executable's header matches 'xbe_magic' field in the 496d756449SSean Bruno * 'interpreter_list' then it will use the user-level interpreter specified in 506d756449SSean Bruno * the 'xbe_interpreter' field to execute the binary. The 'xbe_magic' field may 516d756449SSean Bruno * be adjusted to a given offset using the value in the 'xbe_moffset' field 526d756449SSean Bruno * and bits of the header may be masked using the 'xbe_mask' field. The 536d756449SSean Bruno * 'interpreter_list' entries are managed using sysctl(3) as described in the 546d756449SSean Bruno * <sys/imgact_binmisc.h> file. 556d756449SSean Bruno */ 566d756449SSean Bruno 576d756449SSean Bruno /* 586d756449SSean Bruno * Node of the interpreter list. 596d756449SSean Bruno */ 606d756449SSean Bruno typedef struct imgact_binmisc_entry { 616d756449SSean Bruno char *ibe_name; 626d756449SSean Bruno uint8_t *ibe_magic; 636d756449SSean Bruno uint32_t ibe_moffset; 646d756449SSean Bruno uint32_t ibe_msize; 656d756449SSean Bruno uint8_t *ibe_mask; 666d756449SSean Bruno uint8_t *ibe_interpreter; 676d756449SSean Bruno uint32_t ibe_interp_argcnt; 686d756449SSean Bruno uint32_t ibe_interp_length; 696d756449SSean Bruno uint32_t ibe_flags; 706d756449SSean Bruno SLIST_ENTRY(imgact_binmisc_entry) link; 716d756449SSean Bruno } imgact_binmisc_entry_t; 726d756449SSean Bruno 736d756449SSean Bruno /* 746d756449SSean Bruno * sysctl() commands. 756d756449SSean Bruno */ 766d756449SSean Bruno #define IBC_ADD 1 /* Add given entry. */ 776d756449SSean Bruno #define IBC_REMOVE 2 /* Remove entry for a given name. */ 786d756449SSean Bruno #define IBC_DISABLE 3 /* Disable entry for a given name. */ 796d756449SSean Bruno #define IBC_ENABLE 4 /* Enable entry for a given name. */ 806d756449SSean Bruno #define IBC_LOOKUP 5 /* Lookup and return entry for given name. */ 816d756449SSean Bruno #define IBC_LIST 6 /* Get a snapshot of the interpretor list. */ 826d756449SSean Bruno 836d756449SSean Bruno /* 846d756449SSean Bruno * Interpreter string macros. 856d756449SSean Bruno * 866d756449SSean Bruno * They all start with '#' followed by a single letter: 876d756449SSean Bruno */ 886d756449SSean Bruno #define ISM_POUND '#' /* "##" is the escape sequence for single #. */ 896d756449SSean Bruno #define ISM_OLD_ARGV0 'a' /* "#a" is replaced with the old argv0. */ 906d756449SSean Bruno 916d756449SSean Bruno MALLOC_DEFINE(M_BINMISC, KMOD_NAME, "misc binary image activator"); 926d756449SSean Bruno 936d756449SSean Bruno /* The interpreter list. */ 946d756449SSean Bruno static SLIST_HEAD(, imgact_binmisc_entry) interpreter_list = 956d756449SSean Bruno SLIST_HEAD_INITIALIZER(interpreter_list); 966d756449SSean Bruno 97df69035dSKyle Evans static int interp_list_entry_count; 98280b7169SSean Bruno 995f98711dSSean Bruno static struct sx interp_list_sx; 1006d756449SSean Bruno 1016d756449SSean Bruno /* 1026d756449SSean Bruno * Populate the entry with the information about the interpreter. 1036d756449SSean Bruno */ 1046d756449SSean Bruno static void 1056d756449SSean Bruno imgact_binmisc_populate_interp(char *str, imgact_binmisc_entry_t *ibe) 1066d756449SSean Bruno { 1076d756449SSean Bruno uint32_t len = 0, argc = 1; 1086d756449SSean Bruno char t[IBE_INTERP_LEN_MAX]; 1096d756449SSean Bruno char *sp, *tp; 1106d756449SSean Bruno 1115f98711dSSean Bruno memset(t, 0, sizeof(t)); 1126d756449SSean Bruno 1136d756449SSean Bruno /* 1146d756449SSean Bruno * Normalize interpreter string. Replace white space between args with 1156d756449SSean Bruno * single space. 1166d756449SSean Bruno */ 1176d756449SSean Bruno sp = str; tp = t; 1186d756449SSean Bruno while (*sp != '\0') { 1196d756449SSean Bruno if (*sp == ' ' || *sp == '\t') { 12026af6115SEd Maste if (++len >= IBE_INTERP_LEN_MAX) 1216d756449SSean Bruno break; 1226d756449SSean Bruno *tp++ = ' '; 1236d756449SSean Bruno argc++; 1246d756449SSean Bruno while (*sp == ' ' || *sp == '\t') 1256d756449SSean Bruno sp++; 1266d756449SSean Bruno continue; 1276d756449SSean Bruno } else { 1286d756449SSean Bruno *tp++ = *sp++; 1296d756449SSean Bruno len++; 1306d756449SSean Bruno } 1316d756449SSean Bruno } 1326d756449SSean Bruno *tp = '\0'; 1336d756449SSean Bruno len++; 1346d756449SSean Bruno 1356d756449SSean Bruno ibe->ibe_interpreter = malloc(len, M_BINMISC, M_WAITOK|M_ZERO); 1366d756449SSean Bruno 1376d756449SSean Bruno /* Populate all the ibe fields for the interpreter. */ 1386d756449SSean Bruno memcpy(ibe->ibe_interpreter, t, len); 1396d756449SSean Bruno ibe->ibe_interp_argcnt = argc; 1406d756449SSean Bruno ibe->ibe_interp_length = len; 1416d756449SSean Bruno } 1426d756449SSean Bruno 1436d756449SSean Bruno /* 1446d756449SSean Bruno * Allocate memory and populate a new entry for the interpreter table. 1456d756449SSean Bruno */ 1466d756449SSean Bruno static imgact_binmisc_entry_t * 1476d756449SSean Bruno imgact_binmisc_new_entry(ximgact_binmisc_entry_t *xbe) 1486d756449SSean Bruno { 1496d756449SSean Bruno imgact_binmisc_entry_t *ibe = NULL; 1506d756449SSean Bruno size_t namesz = min(strlen(xbe->xbe_name) + 1, IBE_NAME_MAX); 1516d756449SSean Bruno 1526d756449SSean Bruno ibe = malloc(sizeof(*ibe), M_BINMISC, M_WAITOK|M_ZERO); 1536d756449SSean Bruno 1546d756449SSean Bruno ibe->ibe_name = malloc(namesz, M_BINMISC, M_WAITOK|M_ZERO); 1556d756449SSean Bruno strlcpy(ibe->ibe_name, xbe->xbe_name, namesz); 1566d756449SSean Bruno 1576d756449SSean Bruno imgact_binmisc_populate_interp(xbe->xbe_interpreter, ibe); 1586d756449SSean Bruno 1596d756449SSean Bruno ibe->ibe_magic = malloc(xbe->xbe_msize, M_BINMISC, M_WAITOK|M_ZERO); 1606d756449SSean Bruno memcpy(ibe->ibe_magic, xbe->xbe_magic, xbe->xbe_msize); 1616d756449SSean Bruno 1626d756449SSean Bruno ibe->ibe_mask = malloc(xbe->xbe_msize, M_BINMISC, M_WAITOK|M_ZERO); 1636d756449SSean Bruno memcpy(ibe->ibe_mask, xbe->xbe_mask, xbe->xbe_msize); 1646d756449SSean Bruno 1656d756449SSean Bruno ibe->ibe_moffset = xbe->xbe_moffset; 1666d756449SSean Bruno ibe->ibe_msize = xbe->xbe_msize; 1676d756449SSean Bruno ibe->ibe_flags = xbe->xbe_flags; 1686d756449SSean Bruno 1696d756449SSean Bruno return (ibe); 1706d756449SSean Bruno } 1716d756449SSean Bruno 1726d756449SSean Bruno /* 1736d756449SSean Bruno * Free the allocated memory for a given list item. 1746d756449SSean Bruno */ 1756d756449SSean Bruno static void 1766d756449SSean Bruno imgact_binmisc_destroy_entry(imgact_binmisc_entry_t *ibe) 1776d756449SSean Bruno { 1786d756449SSean Bruno if (!ibe) 1796d756449SSean Bruno return; 180b888dae4SSean Bruno if (ibe->ibe_magic) 1816d756449SSean Bruno free(ibe->ibe_magic, M_BINMISC); 1826d756449SSean Bruno if (ibe->ibe_mask) 1836d756449SSean Bruno free(ibe->ibe_mask, M_BINMISC); 1846d756449SSean Bruno if (ibe->ibe_interpreter) 1856d756449SSean Bruno free(ibe->ibe_interpreter, M_BINMISC); 1866d756449SSean Bruno if (ibe->ibe_name) 1876d756449SSean Bruno free(ibe->ibe_name, M_BINMISC); 1886d756449SSean Bruno if (ibe) 1896d756449SSean Bruno free(ibe, M_BINMISC); 1906d756449SSean Bruno } 1916d756449SSean Bruno 1926d756449SSean Bruno /* 1936d756449SSean Bruno * Find the interpreter in the list by the given name. Return NULL if not 1946d756449SSean Bruno * found. 1956d756449SSean Bruno */ 1966d756449SSean Bruno static imgact_binmisc_entry_t * 1976d756449SSean Bruno imgact_binmisc_find_entry(char *name) 1986d756449SSean Bruno { 1996d756449SSean Bruno imgact_binmisc_entry_t *ibe; 2006d756449SSean Bruno 2015f98711dSSean Bruno sx_assert(&interp_list_sx, SA_LOCKED); 2026d756449SSean Bruno 2036d756449SSean Bruno SLIST_FOREACH(ibe, &interpreter_list, link) { 2046d756449SSean Bruno if (strncmp(name, ibe->ibe_name, IBE_NAME_MAX) == 0) 2056d756449SSean Bruno return (ibe); 2066d756449SSean Bruno } 2076d756449SSean Bruno 2086d756449SSean Bruno return (NULL); 2096d756449SSean Bruno } 2106d756449SSean Bruno 2116d756449SSean Bruno /* 2126d756449SSean Bruno * Add the given interpreter if it doesn't already exist. Return EEXIST 2136d756449SSean Bruno * if the name already exist in the interpreter list. 2146d756449SSean Bruno */ 2156d756449SSean Bruno static int 2166d756449SSean Bruno imgact_binmisc_add_entry(ximgact_binmisc_entry_t *xbe) 2176d756449SSean Bruno { 2186d756449SSean Bruno imgact_binmisc_entry_t *ibe; 2196d756449SSean Bruno char *p; 220910938f0SSean Bruno int cnt; 2216d756449SSean Bruno 2226d756449SSean Bruno if (xbe->xbe_msize > IBE_MAGIC_MAX) 2236d756449SSean Bruno return (EINVAL); 2246d756449SSean Bruno 225910938f0SSean Bruno for(cnt = 0, p = xbe->xbe_name; *p != 0; cnt++, p++) 226910938f0SSean Bruno if (cnt >= IBE_NAME_MAX || !isascii((int)*p)) 2276d756449SSean Bruno return (EINVAL); 2286d756449SSean Bruno 229910938f0SSean Bruno for(cnt = 0, p = xbe->xbe_interpreter; *p != 0; cnt++, p++) 230910938f0SSean Bruno if (cnt >= IBE_INTERP_LEN_MAX || !isascii((int)*p)) 2316d756449SSean Bruno return (EINVAL); 2326d756449SSean Bruno 2336d756449SSean Bruno /* Make sure we don't have any invalid #'s. */ 2346d756449SSean Bruno p = xbe->xbe_interpreter; 2356d756449SSean Bruno while (1) { 2366d756449SSean Bruno p = strchr(p, '#'); 2376d756449SSean Bruno if (!p) 2386d756449SSean Bruno break; 2396d756449SSean Bruno 2406d756449SSean Bruno p++; 2416d756449SSean Bruno switch(*p) { 2426d756449SSean Bruno case ISM_POUND: 2436d756449SSean Bruno /* "##" */ 2446d756449SSean Bruno p++; 2456d756449SSean Bruno break; 2466d756449SSean Bruno 2476d756449SSean Bruno case ISM_OLD_ARGV0: 2486d756449SSean Bruno /* "#a" */ 2496d756449SSean Bruno p++; 2506d756449SSean Bruno break; 2516d756449SSean Bruno 2526d756449SSean Bruno case 0: 2536d756449SSean Bruno default: 2546d756449SSean Bruno /* Anything besides the above is invalid. */ 2556d756449SSean Bruno return (EINVAL); 2566d756449SSean Bruno } 2576d756449SSean Bruno } 2586d756449SSean Bruno 2595f98711dSSean Bruno sx_xlock(&interp_list_sx); 260280b7169SSean Bruno if (imgact_binmisc_find_entry(xbe->xbe_name) != NULL) { 2615f98711dSSean Bruno sx_xunlock(&interp_list_sx); 262280b7169SSean Bruno return (EEXIST); 263280b7169SSean Bruno } 264280b7169SSean Bruno 2655f98711dSSean Bruno /* Preallocate a new entry. */ 2666d756449SSean Bruno ibe = imgact_binmisc_new_entry(xbe); 2676d756449SSean Bruno 2686d756449SSean Bruno SLIST_INSERT_HEAD(&interpreter_list, ibe, link); 2696d756449SSean Bruno interp_list_entry_count++; 2705f98711dSSean Bruno sx_xunlock(&interp_list_sx); 2716d756449SSean Bruno 2726d756449SSean Bruno return (0); 2736d756449SSean Bruno } 2746d756449SSean Bruno 2756d756449SSean Bruno /* 2766d756449SSean Bruno * Remove the interpreter in the list with the given name. Return ENOENT 2776d756449SSean Bruno * if not found. 2786d756449SSean Bruno */ 2796d756449SSean Bruno static int 2806d756449SSean Bruno imgact_binmisc_remove_entry(char *name) 2816d756449SSean Bruno { 2826d756449SSean Bruno imgact_binmisc_entry_t *ibe; 2836d756449SSean Bruno 2845f98711dSSean Bruno sx_xlock(&interp_list_sx); 2856d756449SSean Bruno if ((ibe = imgact_binmisc_find_entry(name)) == NULL) { 2865f98711dSSean Bruno sx_xunlock(&interp_list_sx); 2876d756449SSean Bruno return (ENOENT); 2886d756449SSean Bruno } 2896d756449SSean Bruno SLIST_REMOVE(&interpreter_list, ibe, imgact_binmisc_entry, link); 2906d756449SSean Bruno interp_list_entry_count--; 2915f98711dSSean Bruno sx_xunlock(&interp_list_sx); 2926d756449SSean Bruno 2936d756449SSean Bruno imgact_binmisc_destroy_entry(ibe); 2946d756449SSean Bruno 2956d756449SSean Bruno return (0); 2966d756449SSean Bruno } 2976d756449SSean Bruno 2986d756449SSean Bruno /* 2996d756449SSean Bruno * Disable the interpreter in the list with the given name. Return ENOENT 3006d756449SSean Bruno * if not found. 3016d756449SSean Bruno */ 3026d756449SSean Bruno static int 3036d756449SSean Bruno imgact_binmisc_disable_entry(char *name) 3046d756449SSean Bruno { 3056d756449SSean Bruno imgact_binmisc_entry_t *ibe; 3066d756449SSean Bruno 3074e83b32aSSean Bruno sx_xlock(&interp_list_sx); 3086d756449SSean Bruno if ((ibe = imgact_binmisc_find_entry(name)) == NULL) { 3094e83b32aSSean Bruno sx_xunlock(&interp_list_sx); 3106d756449SSean Bruno return (ENOENT); 3116d756449SSean Bruno } 3126d756449SSean Bruno 3134e83b32aSSean Bruno ibe->ibe_flags &= ~IBF_ENABLED; 3144e83b32aSSean Bruno sx_xunlock(&interp_list_sx); 3156d756449SSean Bruno 3166d756449SSean Bruno return (0); 3176d756449SSean Bruno } 3186d756449SSean Bruno 3196d756449SSean Bruno /* 3206d756449SSean Bruno * Enable the interpreter in the list with the given name. Return ENOENT 3216d756449SSean Bruno * if not found. 3226d756449SSean Bruno */ 3236d756449SSean Bruno static int 3246d756449SSean Bruno imgact_binmisc_enable_entry(char *name) 3256d756449SSean Bruno { 3266d756449SSean Bruno imgact_binmisc_entry_t *ibe; 3276d756449SSean Bruno 3284e83b32aSSean Bruno sx_xlock(&interp_list_sx); 3296d756449SSean Bruno if ((ibe = imgact_binmisc_find_entry(name)) == NULL) { 3304e83b32aSSean Bruno sx_xunlock(&interp_list_sx); 3316d756449SSean Bruno return (ENOENT); 3326d756449SSean Bruno } 3336d756449SSean Bruno 3344e83b32aSSean Bruno ibe->ibe_flags |= IBF_ENABLED; 3354e83b32aSSean Bruno sx_xunlock(&interp_list_sx); 3366d756449SSean Bruno 3376d756449SSean Bruno return (0); 3386d756449SSean Bruno } 3396d756449SSean Bruno 3406d756449SSean Bruno static int 3416d756449SSean Bruno imgact_binmisc_populate_xbe(ximgact_binmisc_entry_t *xbe, 3426d756449SSean Bruno imgact_binmisc_entry_t *ibe) 3436d756449SSean Bruno { 3446d756449SSean Bruno uint32_t i; 3456d756449SSean Bruno 3465f98711dSSean Bruno sx_assert(&interp_list_sx, SA_LOCKED); 3476d756449SSean Bruno 3485f98711dSSean Bruno memset(xbe, 0, sizeof(*xbe)); 3496d756449SSean Bruno strlcpy(xbe->xbe_name, ibe->ibe_name, IBE_NAME_MAX); 3506d756449SSean Bruno 3516d756449SSean Bruno /* Copy interpreter string. Replace NULL breaks with space. */ 3526d756449SSean Bruno memcpy(xbe->xbe_interpreter, ibe->ibe_interpreter, 3536d756449SSean Bruno ibe->ibe_interp_length); 3546d756449SSean Bruno for(i = 0; i < (ibe->ibe_interp_length - 1); i++) 3556d756449SSean Bruno if (xbe->xbe_interpreter[i] == '\0') 3566d756449SSean Bruno xbe->xbe_interpreter[i] = ' '; 3576d756449SSean Bruno 3586d756449SSean Bruno memcpy(xbe->xbe_magic, ibe->ibe_magic, ibe->ibe_msize); 3596d756449SSean Bruno memcpy(xbe->xbe_mask, ibe->ibe_mask, ibe->ibe_msize); 3606d756449SSean Bruno xbe->xbe_version = IBE_VERSION; 3616d756449SSean Bruno xbe->xbe_flags = ibe->ibe_flags; 3626d756449SSean Bruno xbe->xbe_moffset = ibe->ibe_moffset; 3636d756449SSean Bruno xbe->xbe_msize = ibe->ibe_msize; 3646d756449SSean Bruno 3656d756449SSean Bruno return (0); 3666d756449SSean Bruno } 3676d756449SSean Bruno 3686d756449SSean Bruno /* 3696d756449SSean Bruno * Retrieve the interpreter with the give name and populate the 3706d756449SSean Bruno * ximgact_binmisc_entry structure. Return ENOENT if not found. 3716d756449SSean Bruno */ 3726d756449SSean Bruno static int 3736d756449SSean Bruno imgact_binmisc_lookup_entry(char *name, ximgact_binmisc_entry_t *xbe) 3746d756449SSean Bruno { 3756d756449SSean Bruno imgact_binmisc_entry_t *ibe; 3766d756449SSean Bruno int error = 0; 3776d756449SSean Bruno 3785f98711dSSean Bruno sx_slock(&interp_list_sx); 3796d756449SSean Bruno if ((ibe = imgact_binmisc_find_entry(name)) == NULL) { 3805f98711dSSean Bruno sx_sunlock(&interp_list_sx); 3816d756449SSean Bruno return (ENOENT); 3826d756449SSean Bruno } 3836d756449SSean Bruno 3846d756449SSean Bruno error = imgact_binmisc_populate_xbe(xbe, ibe); 3855f98711dSSean Bruno sx_sunlock(&interp_list_sx); 3866d756449SSean Bruno 3876d756449SSean Bruno return (error); 3886d756449SSean Bruno } 3896d756449SSean Bruno 3906d756449SSean Bruno /* 3916d756449SSean Bruno * Get a snapshot of all the interpreter entries in the list. 3926d756449SSean Bruno */ 3936d756449SSean Bruno static int 3946d756449SSean Bruno imgact_binmisc_get_all_entries(struct sysctl_req *req) 3956d756449SSean Bruno { 3966d756449SSean Bruno ximgact_binmisc_entry_t *xbe, *xbep; 3976d756449SSean Bruno imgact_binmisc_entry_t *ibe; 3986d756449SSean Bruno int error = 0, count; 3996d756449SSean Bruno 4005f98711dSSean Bruno sx_slock(&interp_list_sx); 4016d756449SSean Bruno count = interp_list_entry_count; 402e0ae213fSSean Bruno xbe = malloc(sizeof(*xbe) * count, M_BINMISC, M_WAITOK|M_ZERO); 4036d756449SSean Bruno 4046d756449SSean Bruno xbep = xbe; 4056d756449SSean Bruno SLIST_FOREACH(ibe, &interpreter_list, link) { 4066d756449SSean Bruno error = imgact_binmisc_populate_xbe(xbep++, ibe); 4076d756449SSean Bruno if (error) 4086d756449SSean Bruno break; 4096d756449SSean Bruno } 4105f98711dSSean Bruno sx_sunlock(&interp_list_sx); 4116d756449SSean Bruno 4126d756449SSean Bruno if (!error) 4136d756449SSean Bruno error = SYSCTL_OUT(req, xbe, sizeof(*xbe) * count); 4146d756449SSean Bruno 4156d756449SSean Bruno free(xbe, M_BINMISC); 4166d756449SSean Bruno return (error); 4176d756449SSean Bruno } 4186d756449SSean Bruno 4196d756449SSean Bruno /* 4206d756449SSean Bruno * sysctl() handler for munipulating interpretor table. 4216d756449SSean Bruno * Not MP safe (locked by sysctl). 4226d756449SSean Bruno */ 4236d756449SSean Bruno static int 4246d756449SSean Bruno sysctl_kern_binmisc(SYSCTL_HANDLER_ARGS) 4256d756449SSean Bruno { 4266d756449SSean Bruno ximgact_binmisc_entry_t xbe; 4276d756449SSean Bruno int error = 0; 4286d756449SSean Bruno 4296d756449SSean Bruno switch(arg2) { 4306d756449SSean Bruno case IBC_ADD: 4316d756449SSean Bruno /* Add an entry. Limited to IBE_MAX_ENTRIES. */ 4326d756449SSean Bruno error = SYSCTL_IN(req, &xbe, sizeof(xbe)); 4336d756449SSean Bruno if (error) 4346d756449SSean Bruno return (error); 4356d756449SSean Bruno if (IBE_VERSION != xbe.xbe_version) 4366d756449SSean Bruno return (EINVAL); 437*7d3ed977SKyle Evans if ((xbe.xbe_flags & ~IBF_VALID_UFLAGS) != 0) 438*7d3ed977SKyle Evans return (EINVAL); 4396d756449SSean Bruno if (interp_list_entry_count == IBE_MAX_ENTRIES) 4406d756449SSean Bruno return (ENOSPC); 4416d756449SSean Bruno error = imgact_binmisc_add_entry(&xbe); 4426d756449SSean Bruno break; 4436d756449SSean Bruno 4446d756449SSean Bruno case IBC_REMOVE: 4456d756449SSean Bruno /* Remove an entry. */ 4466d756449SSean Bruno error = SYSCTL_IN(req, &xbe, sizeof(xbe)); 4476d756449SSean Bruno if (error) 4486d756449SSean Bruno return (error); 4496d756449SSean Bruno if (IBE_VERSION != xbe.xbe_version) 4506d756449SSean Bruno return (EINVAL); 4516d756449SSean Bruno error = imgact_binmisc_remove_entry(xbe.xbe_name); 4526d756449SSean Bruno break; 4536d756449SSean Bruno 4546d756449SSean Bruno case IBC_DISABLE: 4556d756449SSean Bruno /* Disable an entry. */ 4566d756449SSean Bruno error = SYSCTL_IN(req, &xbe, sizeof(xbe)); 4576d756449SSean Bruno if (error) 4586d756449SSean Bruno return (error); 4596d756449SSean Bruno if (IBE_VERSION != xbe.xbe_version) 4606d756449SSean Bruno return (EINVAL); 4616d756449SSean Bruno error = imgact_binmisc_disable_entry(xbe.xbe_name); 4626d756449SSean Bruno break; 4636d756449SSean Bruno 4646d756449SSean Bruno case IBC_ENABLE: 4656d756449SSean Bruno /* Enable an entry. */ 4666d756449SSean Bruno error = SYSCTL_IN(req, &xbe, sizeof(xbe)); 4676d756449SSean Bruno if (error) 4686d756449SSean Bruno return (error); 4696d756449SSean Bruno if (IBE_VERSION != xbe.xbe_version) 4706d756449SSean Bruno return (EINVAL); 4716d756449SSean Bruno error = imgact_binmisc_enable_entry(xbe.xbe_name); 4726d756449SSean Bruno break; 4736d756449SSean Bruno 4746d756449SSean Bruno case IBC_LOOKUP: 4756d756449SSean Bruno /* Lookup an entry. */ 4766d756449SSean Bruno error = SYSCTL_IN(req, &xbe, sizeof(xbe)); 4776d756449SSean Bruno if (error) 4786d756449SSean Bruno return (error); 4796d756449SSean Bruno if (IBE_VERSION != xbe.xbe_version) 4806d756449SSean Bruno return (EINVAL); 4816d756449SSean Bruno error = imgact_binmisc_lookup_entry(xbe.xbe_name, &xbe); 4826d756449SSean Bruno if (!error) 4836d756449SSean Bruno error = SYSCTL_OUT(req, &xbe, sizeof(xbe)); 4846d756449SSean Bruno break; 4856d756449SSean Bruno 4866d756449SSean Bruno case IBC_LIST: 4876d756449SSean Bruno /* Return a snapshot of the interpretor list. */ 4886d756449SSean Bruno 4896d756449SSean Bruno if (!req->oldptr) { 4906d756449SSean Bruno /* No pointer then just return the list size. */ 4916d756449SSean Bruno error = SYSCTL_OUT(req, 0, interp_list_entry_count * 4926d756449SSean Bruno sizeof(ximgact_binmisc_entry_t)); 4936d756449SSean Bruno return (error); 4946d756449SSean Bruno } else 4956d756449SSean Bruno if (!req->oldlen) 4966d756449SSean Bruno return (EINVAL); 4976d756449SSean Bruno 4986d756449SSean Bruno error = imgact_binmisc_get_all_entries(req); 4996d756449SSean Bruno break; 5006d756449SSean Bruno 5016d756449SSean Bruno default: 5026d756449SSean Bruno return (EINVAL); 5036d756449SSean Bruno } 5046d756449SSean Bruno 5056d756449SSean Bruno return (error); 5066d756449SSean Bruno } 5076d756449SSean Bruno 5087029da5cSPawel Biernacki SYSCTL_NODE(_kern, OID_AUTO, binmisc, CTLFLAG_RW | CTLFLAG_MPSAFE, 0, 5096d756449SSean Bruno "Image activator for miscellaneous binaries"); 5106d756449SSean Bruno 5116d756449SSean Bruno SYSCTL_PROC(_kern_binmisc, OID_AUTO, add, 5126d756449SSean Bruno CTLFLAG_MPSAFE|CTLTYPE_STRUCT|CTLFLAG_WR, NULL, IBC_ADD, 5136d756449SSean Bruno sysctl_kern_binmisc, "S,ximgact_binmisc_entry", 5146d756449SSean Bruno "Add an activator entry"); 5156d756449SSean Bruno 5166d756449SSean Bruno SYSCTL_PROC(_kern_binmisc, OID_AUTO, remove, 5176d756449SSean Bruno CTLFLAG_MPSAFE|CTLTYPE_STRUCT|CTLFLAG_WR, NULL, IBC_REMOVE, 5186d756449SSean Bruno sysctl_kern_binmisc, "S,ximgact_binmisc_entry", 5196d756449SSean Bruno "Remove an activator entry"); 5206d756449SSean Bruno 5216d756449SSean Bruno SYSCTL_PROC(_kern_binmisc, OID_AUTO, disable, 5226d756449SSean Bruno CTLFLAG_MPSAFE|CTLTYPE_STRUCT|CTLFLAG_WR, NULL, IBC_DISABLE, 5236d756449SSean Bruno sysctl_kern_binmisc, "S,ximgact_binmisc_entry", 5246d756449SSean Bruno "Disable an activator entry"); 5256d756449SSean Bruno 5266d756449SSean Bruno SYSCTL_PROC(_kern_binmisc, OID_AUTO, enable, 5276d756449SSean Bruno CTLFLAG_MPSAFE|CTLTYPE_STRUCT|CTLFLAG_WR, NULL, IBC_ENABLE, 5286d756449SSean Bruno sysctl_kern_binmisc, "S,ximgact_binmisc_entry", 5296d756449SSean Bruno "Enable an activator entry"); 5306d756449SSean Bruno 5316d756449SSean Bruno SYSCTL_PROC(_kern_binmisc, OID_AUTO, lookup, 5326d756449SSean Bruno CTLFLAG_MPSAFE|CTLTYPE_STRUCT|CTLFLAG_RW|CTLFLAG_ANYBODY, NULL, IBC_LOOKUP, 5336d756449SSean Bruno sysctl_kern_binmisc, "S,ximgact_binmisc_entry", 5346d756449SSean Bruno "Lookup an activator entry"); 5356d756449SSean Bruno 5366d756449SSean Bruno SYSCTL_PROC(_kern_binmisc, OID_AUTO, list, 5376d756449SSean Bruno CTLFLAG_MPSAFE|CTLTYPE_STRUCT|CTLFLAG_RD|CTLFLAG_ANYBODY, NULL, IBC_LIST, 5386d756449SSean Bruno sysctl_kern_binmisc, "S,ximgact_binmisc_entry", 5396d756449SSean Bruno "Get snapshot of all the activator entries"); 5406d756449SSean Bruno 5416d756449SSean Bruno static imgact_binmisc_entry_t * 5426d756449SSean Bruno imgact_binmisc_find_interpreter(const char *image_header) 5436d756449SSean Bruno { 5446d756449SSean Bruno imgact_binmisc_entry_t *ibe; 5456d756449SSean Bruno const char *p; 5466d756449SSean Bruno int i; 5476d756449SSean Bruno size_t sz; 5486d756449SSean Bruno 5495f98711dSSean Bruno sx_assert(&interp_list_sx, SA_LOCKED); 5506d756449SSean Bruno 5516d756449SSean Bruno SLIST_FOREACH(ibe, &interpreter_list, link) { 5526d756449SSean Bruno if (!(IBF_ENABLED & ibe->ibe_flags)) 5536d756449SSean Bruno continue; 5546d756449SSean Bruno 5556d756449SSean Bruno p = image_header + ibe->ibe_moffset; 5566d756449SSean Bruno sz = ibe->ibe_msize; 5576d756449SSean Bruno if (IBF_USE_MASK & ibe->ibe_flags) { 5586d756449SSean Bruno /* Compare using mask. */ 5596d756449SSean Bruno for (i = 0; i < sz; i++) 5606d756449SSean Bruno if ((*p++ ^ ibe->ibe_magic[i]) & 5616d756449SSean Bruno ibe->ibe_mask[i]) 5626d756449SSean Bruno break; 5636d756449SSean Bruno } else { 5646d756449SSean Bruno for (i = 0; i < sz; i++) 5656d756449SSean Bruno if (*p++ ^ ibe->ibe_magic[i]) 5666d756449SSean Bruno break; 5676d756449SSean Bruno } 5686d756449SSean Bruno if (i == ibe->ibe_msize) 5696d756449SSean Bruno return (ibe); 5706d756449SSean Bruno } 5716d756449SSean Bruno return (NULL); 5726d756449SSean Bruno } 5736d756449SSean Bruno 574945afa7cSSean Bruno static int 5756d756449SSean Bruno imgact_binmisc_exec(struct image_params *imgp) 5766d756449SSean Bruno { 5776d756449SSean Bruno const char *image_header = imgp->image_header; 5786d756449SSean Bruno const char *fname = NULL; 5796d756449SSean Bruno int error = 0; 5806d756449SSean Bruno size_t offset, l; 5816d756449SSean Bruno imgact_binmisc_entry_t *ibe; 5826d756449SSean Bruno struct sbuf *sname; 5836d756449SSean Bruno char *s, *d; 5846d756449SSean Bruno 58580083216SKyle Evans sname = NULL; 5866d756449SSean Bruno /* Do we have an interpreter for the given image header? */ 5875f98711dSSean Bruno sx_slock(&interp_list_sx); 5886d756449SSean Bruno if ((ibe = imgact_binmisc_find_interpreter(image_header)) == NULL) { 58980083216SKyle Evans error = -1; 59080083216SKyle Evans goto done; 5916d756449SSean Bruno } 5926d756449SSean Bruno 5936d756449SSean Bruno /* No interpreter nesting allowed. */ 59465f20a89SSean Bruno if (imgp->interpreted & IMGACT_BINMISC) { 59580083216SKyle Evans error = ENOEXEC; 59680083216SKyle Evans goto done; 5976d756449SSean Bruno } 5986d756449SSean Bruno 59965f20a89SSean Bruno imgp->interpreted |= IMGACT_BINMISC; 6006d756449SSean Bruno 6016d756449SSean Bruno if (imgp->args->fname != NULL) { 6026d756449SSean Bruno fname = imgp->args->fname; 6036d756449SSean Bruno } else { 6046d756449SSean Bruno /* Use the fdescfs(5) path for fexecve(2). */ 6056d756449SSean Bruno sname = sbuf_new_auto(); 6066d756449SSean Bruno sbuf_printf(sname, "/dev/fd/%d", imgp->args->fd); 6076d756449SSean Bruno sbuf_finish(sname); 6086d756449SSean Bruno fname = sbuf_data(sname); 6096d756449SSean Bruno } 6106d756449SSean Bruno 6116d756449SSean Bruno /* 6126d756449SSean Bruno * We need to "push" the interpreter in the arg[] list. To do this, 6136d756449SSean Bruno * we first shift all the other values in the `begin_argv' area to 6146d756449SSean Bruno * provide the exact amount of room for the values added. Set up 6156d756449SSean Bruno * `offset' as the number of bytes to be added to the `begin_argv' 6166d756449SSean Bruno * area. 6176d756449SSean Bruno */ 6186d756449SSean Bruno offset = ibe->ibe_interp_length; 6196d756449SSean Bruno 6206d756449SSean Bruno /* Adjust the offset for #'s. */ 6216d756449SSean Bruno s = ibe->ibe_interpreter; 6226d756449SSean Bruno while (1) { 6236d756449SSean Bruno s = strchr(s, '#'); 6246d756449SSean Bruno if (!s) 6256d756449SSean Bruno break; 6266d756449SSean Bruno 6276d756449SSean Bruno s++; 6286d756449SSean Bruno switch(*s) { 6296d756449SSean Bruno case ISM_POUND: 6306d756449SSean Bruno /* "##" -> "#": reduce offset by one. */ 6316d756449SSean Bruno offset--; 6326d756449SSean Bruno break; 6336d756449SSean Bruno 6346d756449SSean Bruno case ISM_OLD_ARGV0: 6356d756449SSean Bruno /* "#a" -> (old argv0): increase offset to fit fname */ 6366d756449SSean Bruno offset += strlen(fname) - 2; 6376d756449SSean Bruno break; 6386d756449SSean Bruno 6396d756449SSean Bruno default: 6406d756449SSean Bruno /* Hmm... This shouldn't happen. */ 6416d756449SSean Bruno printf("%s: Unknown macro #%c sequence in " 6426d756449SSean Bruno "interpreter string\n", KMOD_NAME, *(s + 1)); 6436d756449SSean Bruno error = EINVAL; 6446d756449SSean Bruno goto done; 6456d756449SSean Bruno } 6466d756449SSean Bruno s++; 6476d756449SSean Bruno } 6486d756449SSean Bruno 649f373437aSBrooks Davis /* Make room for the interpreter */ 650f373437aSBrooks Davis error = exec_args_adjust_args(imgp->args, 0, offset); 651f373437aSBrooks Davis if (error != 0) { 6526d756449SSean Bruno goto done; 6536d756449SSean Bruno } 6546d756449SSean Bruno 6556d756449SSean Bruno /* Add the new argument(s) in the count. */ 6566d756449SSean Bruno imgp->args->argc += ibe->ibe_interp_argcnt; 6576d756449SSean Bruno 6586d756449SSean Bruno /* 6596d756449SSean Bruno * The original arg[] list has been shifted appropriately. Copy in 6606d756449SSean Bruno * the interpreter path. 6616d756449SSean Bruno */ 6626d756449SSean Bruno s = ibe->ibe_interpreter; 6636d756449SSean Bruno d = imgp->args->begin_argv; 6646d756449SSean Bruno while(*s != '\0') { 6656d756449SSean Bruno switch (*s) { 6666d756449SSean Bruno case '#': 6676d756449SSean Bruno /* Handle "#" in interpreter string. */ 6686d756449SSean Bruno s++; 6696d756449SSean Bruno switch(*s) { 6706d756449SSean Bruno case ISM_POUND: 6716d756449SSean Bruno /* "##": Replace with a single '#' */ 6726d756449SSean Bruno *d++ = '#'; 6736d756449SSean Bruno break; 6746d756449SSean Bruno 6756d756449SSean Bruno case ISM_OLD_ARGV0: 6766d756449SSean Bruno /* "#a": Replace with old arg0 (fname). */ 6776d756449SSean Bruno if ((l = strlen(fname)) != 0) { 6786d756449SSean Bruno memcpy(d, fname, l); 6796d756449SSean Bruno d += l; 6806d756449SSean Bruno } 6816d756449SSean Bruno break; 6826d756449SSean Bruno 6836d756449SSean Bruno default: 6846d756449SSean Bruno /* Shouldn't happen but skip it if it does. */ 6856d756449SSean Bruno break; 6866d756449SSean Bruno } 6876d756449SSean Bruno break; 6886d756449SSean Bruno 6896d756449SSean Bruno case ' ': 690e3043798SPedro F. Giffuni /* Replace space with NUL to separate arguments. */ 6916d756449SSean Bruno *d++ = '\0'; 6926d756449SSean Bruno break; 6936d756449SSean Bruno 6946d756449SSean Bruno default: 6956d756449SSean Bruno *d++ = *s; 6966d756449SSean Bruno break; 6976d756449SSean Bruno } 6986d756449SSean Bruno s++; 6996d756449SSean Bruno } 7006d756449SSean Bruno *d = '\0'; 7016d756449SSean Bruno 7026d756449SSean Bruno imgp->interpreter_name = imgp->args->begin_argv; 7036d756449SSean Bruno 7046d756449SSean Bruno done: 70580083216SKyle Evans sx_sunlock(&interp_list_sx); 7066d756449SSean Bruno if (sname) 7076d756449SSean Bruno sbuf_delete(sname); 7086d756449SSean Bruno return (error); 7096d756449SSean Bruno } 7106d756449SSean Bruno 7116d756449SSean Bruno static void 7126d756449SSean Bruno imgact_binmisc_init(void *arg) 7136d756449SSean Bruno { 7146d756449SSean Bruno 7155f98711dSSean Bruno sx_init(&interp_list_sx, KMOD_NAME); 7166d756449SSean Bruno } 7176d756449SSean Bruno 7186d756449SSean Bruno static void 7196d756449SSean Bruno imgact_binmisc_fini(void *arg) 7206d756449SSean Bruno { 7216d756449SSean Bruno imgact_binmisc_entry_t *ibe, *ibe_tmp; 7226d756449SSean Bruno 7236d756449SSean Bruno /* Free all the interpreters. */ 7245f98711dSSean Bruno sx_xlock(&interp_list_sx); 7256d756449SSean Bruno SLIST_FOREACH_SAFE(ibe, &interpreter_list, link, ibe_tmp) { 7266d756449SSean Bruno SLIST_REMOVE(&interpreter_list, ibe, imgact_binmisc_entry, 7276d756449SSean Bruno link); 7286d756449SSean Bruno imgact_binmisc_destroy_entry(ibe); 7296d756449SSean Bruno } 7305f98711dSSean Bruno sx_xunlock(&interp_list_sx); 7316d756449SSean Bruno 7325f98711dSSean Bruno sx_destroy(&interp_list_sx); 7336d756449SSean Bruno } 7346d756449SSean Bruno 735891cf3edSEd Maste SYSINIT(imgact_binmisc, SI_SUB_EXEC, SI_ORDER_MIDDLE, imgact_binmisc_init, 736891cf3edSEd Maste NULL); 737891cf3edSEd Maste SYSUNINIT(imgact_binmisc, SI_SUB_EXEC, SI_ORDER_MIDDLE, imgact_binmisc_fini, 738891cf3edSEd Maste NULL); 7396d756449SSean Bruno 7406d756449SSean Bruno /* 7416d756449SSean Bruno * Tell kern_execve.c about it, with a little help from the linker. 7426d756449SSean Bruno */ 743b7feabf9SEd Maste static struct execsw imgact_binmisc_execsw = { 744b7feabf9SEd Maste .ex_imgact = imgact_binmisc_exec, 745b7feabf9SEd Maste .ex_name = KMOD_NAME 746b7feabf9SEd Maste }; 7476d756449SSean Bruno EXEC_SET(imgact_binmisc, imgact_binmisc_execsw); 748