libexec/ld.elf_so/expand.c

   1 /*      $NetBSD: expand.c,v 1.5 2008/04/28 20:23:03 martin Exp $        */
   2
   3 /*-
   4  * Copyright (c) 2007 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Christos Zoulas.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31 #include <sys/cdefs.h>
  32 #ifndef lint
  33 __RCSID("$NetBSD: expand.c,v 1.5 2008/04/28 20:23:03 martin Exp $");
  34 #endif /* not lint */
  35
  36 #include <ctype.h>
  37 #include <string.h>
  38 #include <sys/sysctl.h>
  39
  40 #ifdef DEBUG_EXPAND
  41 #include <stdio.h>
  42 #include <err.h>
  43 #define xwarn warn
  44 #define xerr err
  45 size_t _rtld_expand_path(char *, size_t, const char *, const char *,
  46     const char *);
  47 #else
  48 #include <sys/stat.h>
  49 #include "rtld.h"
  50 #endif
  51
  52 static const struct {
  53         const char *name;
  54         size_t namelen;
  55 } bltn[] = {
  56 #define ADD(a)  { #a, sizeof(#a) - 1 },
  57         ADD(HWCAP)      /* SSE, MMX, etc */
  58         ADD(ISALIST)    /* XXX */
  59         ADD(ORIGIN)     /* dirname argv[0] */
  60         ADD(OSNAME)     /* uname -s */
  61         ADD(OSREL)      /* uname -r */
  62         ADD(PLATFORM)   /* uname -p */
  63 };
  64
  65 #ifndef __minix
  66 static int mib[3][2] = {
  67         { CTL_KERN, KERN_OSTYPE },
  68         { CTL_KERN, KERN_OSRELEASE },
  69         { CTL_HW, HW_MACHINE_ARCH },
  70 };
  71 #endif
  72
  73 static size_t
  74 expand(char *buf, const char *execname, int what, size_t bl)
  75 {
  76         const char *p, *ep;
  77         char *bp = buf;
  78         size_t len;
  79         char name[32];
  80
  81         switch (what) {
  82         case 0: /* HWCAP XXX: Not yet */
  83         case 1: /* ISALIST XXX: Not yet */
  84                 return 0;
  85
  86         case 2: /* ORIGIN */
  87                 if (execname == NULL)
  88                         xerr(1, "execname not specified in AUX vector");
  89                 if ((ep = strrchr(p = execname, '/')) == NULL)
  90                         xerr(1, "bad execname `%s' in AUX vector", execname);
  91                 break;
  92
  93 #ifndef __minix
  94         case 3: /* OSNAME */
  95         case 4: /* OSREL */
  96         case 5: /* PLATFORM */
  97                 len = sizeof(name);
  98                 if (sysctl(mib[what - 3], 2, name, &len, NULL, 0) == -1) {
  99                         xwarn("sysctl");
 100                         return 0;
 101                 }
 102                 ep = (p = name) + len - 1;
 103                 break;
 104 #endif
 105         default:
 106                 return 0;
 107         }
 108
 109         while (p != ep && bl)
 110                 *bp++ = *p++, bl--;
 111
 112         return bp - buf;
 113 }
 114
 115
 116 size_t
 117 _rtld_expand_path(char *buf, size_t bufsize, const char *execname,
 118     const char *bp, const char *ep)
 119 {
 120         size_t i, ds = bufsize;
 121         char *dp = buf;
 122         const char *p;
 123         int br;
 124
 125         for (p = bp; p < ep;) {
 126                 if (*p == '$') {
 127                         br = *++p == '{';
 128
 129                         if (br)
 130                                 p++;
 131
 132                         for (i = 0; i < sizeof(bltn) / sizeof(bltn[0]); i++) {
 133                                 size_t s = bltn[i].namelen;
 134                                 const char *es = p + s;
 135
 136                                 if ((br && *es != '}') ||
 137                                     (!br && (es != ep &&
 138                                         isalpha((unsigned char)*es))))
 139                                         continue;
 140
 141                                 if (strncmp(bltn[i].name, p, s) == 0) {
 142                                         size_t ls = expand(dp, execname, i, ds);
 143                                         if (ls >= ds)
 144                                                 return bufsize;
 145                                         ds -= ls;
 146                                         dp += ls;
 147                                         p = es + br;
 148                                         goto done;
 149                                 }
 150                         }
 151                         p -= br + 1;
 152
 153                 }
 154                 *dp++ = *p++;
 155                 ds--;
 156 done:;
 157         }
 158         *dp = '\0';
 159         return dp - buf;
 160 }
 161
 162 #ifdef DEBUG_EXPAND
 163 int
 164 main(int argc, char *argv[])
 165 {
 166         char buf[1024];
 167         size_t i;
 168
 169         for (i = 1; i < argc; i++) {
 170                 char *p = argv[i], *ep = argv[i] + strlen(p);
 171                 size_t n = _rtld_expand_path(buf, sizeof(buf), argv[0], p, ep);
 172                 printf("%s\n", buf);
 173         }
 174         return 0;
 175 }
 176 #endif