usr/src/lib/libshell/common/sh/shcomp.c

   1 /***********************************************************************
   2 *                                                                      *
   3 *               This software is part of the ast package               *
   4 *          Copyright (c) 1982-2010 AT&T Intellectual Property          *
   5 *                      and is licensed under the                       *
   6 *                  Common Public License, Version 1.0                  *
   7 *                    by AT&T Intellectual Property                     *
   8 *                                                                      *
   9 *                A copy of the License is available at                 *
  10 *            http://www.opensource.org/licenses/cpl1.0.txt             *
  11 *         (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9)         *
  12 *                                                                      *
  13 *              Information and Software Systems Research               *
  14 *                            AT&T Research                             *
  15 *                           Florham Park NJ                            *
  16 *                                                                      *
  17 *                  David Korn <dgk@research.att.com>                   *
  18 *                                                                      *
  19 ***********************************************************************/
  20 #pragma prototyped
  21 /*
  22  * David Korn
  23  * AT&T Labs
  24  *
  25  * shell script to shell binary converter
  26  *
  27  */
  28
  29 static const char usage[] =
  30 "[-?\n@(#)$Id: shcomp (AT&T Research) 2003-03-02 $\n]"
  31 USAGE_LICENSE
  32 "[+NAME?shcomp - compile a shell script]"
  33 "[+DESCRIPTION?Unless \b-D\b is specified, \bshcomp\b takes a shell script, "
  34         "\ainfile\a, and creates a binary format file, \aoutfile\a, that "
  35         "\bksh\b can read and execute with the same effect as the original "
  36         "script.]"
  37 "[+?Since aliases are processed as the script is read, alias definitions "
  38         "whose value requires variable expansion will not work correctly.]"
  39 "[+?If \b-D\b is specified, all double quoted strings that are preceded by "
  40         "\b$\b are output.  These are the messages that need to be "
  41         "translated to locale specific versions for internationalization.]"
  42 "[+?If \aoutfile\a is omitted, then the results will be written to "
  43         "standard output.  If \ainfile\a is also omitted, the shell script "
  44         "will be read from standard input.]"
  45 "[D:dictionary?Generate a list of strings that need to be placed in a message "
  46         "catalog for internationalization.]"
  47 "[n:noexec?Displays warning messages for obsolete or non-conforming "
  48         "constructs.] "
  49 "[v:verbose?Displays input from \ainfile\a onto standard error as it "
  50         "reads it.]"
  51 "\n"
  52 "\n[infile [outfile]]\n"
  53 "\n"
  54 "[+EXIT STATUS?]{"
  55         "[+0?Successful completion.]"
  56         "[+>0?An error occurred.]"
  57 "}"
  58 "[+SEE ALSO?\bksh\b(1)]"
  59 ;
  60
  61 #include        <shell.h>
  62 #include        "defs.h"
  63 #include        "shnodes.h"
  64 #include        "sys/stat.h"
  65
  66 #define CNTL(x) ((x)&037)
  67 #define VERSION 3
  68 static const char header[6] = { CNTL('k'),CNTL('s'),CNTL('h'),0,VERSION,0 };
  69
  70 int main(int argc, char *argv[])
  71 {
  72         Sfio_t *in, *out;
  73         Shell_t *shp;
  74         Namval_t *np;
  75         Shnode_t *t;
  76         char *cp;
  77         int n, nflag=0, vflag=0, dflag=0;
  78         error_info.id = argv[0];
  79         while(n = optget(argv, usage )) switch(n)
  80         {
  81             case 'D':
  82                 dflag=1;
  83                 break;
  84             case 'v':
  85                 vflag=1;
  86                 break;
  87             case 'n':
  88                 nflag=1;
  89                 break;
  90             case ':':
  91                 errormsg(SH_DICT,2,"%s",opt_info.arg);
  92                 break;
  93             case '?':
  94                 errormsg(SH_DICT,ERROR_usage(2),"%s",opt_info.arg);
  95                 break;
  96         }
  97         shp = sh_init(argc,argv,(Shinit_f)0);
  98         shp->shcomp = 1;
  99         argv += opt_info.index;
 100         argc -= opt_info.index;
 101         if(error_info.errors || argc>2)
 102                 errormsg(SH_DICT,ERROR_usage(2),"%s",optusage((char*)0));
 103         if(cp= *argv)
 104         {
 105                 argv++;
 106                 in = sh_pathopen(cp);
 107         }
 108         else
 109                 in = sfstdin;
 110         if(cp= *argv)
 111         {
 112                 struct stat statb;
 113                 if(!(out = sfopen((Sfio_t*)0,cp,"w")))
 114                         errormsg(SH_DICT,ERROR_system(1),"%s: cannot create",cp);
 115                 if(fstat(sffileno(out),&statb) >=0)
 116                         chmod(cp,(statb.st_mode&~S_IFMT)|S_IXUSR|S_IXGRP|S_IXOTH);
 117         }
 118         else
 119                 out = sfstdout;
 120         if(dflag)
 121         {
 122                 sh_onoption(SH_DICTIONARY);
 123                 sh_onoption(SH_NOEXEC);
 124         }
 125         if(nflag)
 126                 sh_onoption(SH_NOEXEC);
 127         if(vflag)
 128                 sh_onoption(SH_VERBOSE);
 129         if(!dflag)
 130                 sfwrite(out,header,sizeof(header));
 131         shp->inlineno = 1;
 132 #if SHOPT_BRACEPAT
 133         sh_onoption(SH_BRACEEXPAND);
 134 #endif
 135         while(1)
 136         {
 137                 stakset((char*)0,0);
 138                 if(t = (Shnode_t*)sh_parse(shp,in,0))
 139                 {
 140                         if((t->tre.tretyp&(COMMSK|COMSCAN))==0 && t->com.comnamp && strcmp(nv_name((Namval_t*)t->com.comnamp),"alias")==0)
 141                                 sh_exec(t,0);
 142                         if(!dflag && sh_tdump(out,t) < 0)
 143                                 errormsg(SH_DICT,ERROR_exit(1),"dump failed");
 144                 }
 145                 else if(sfeof(in))
 146                         break;
 147                 if(sferror(in))
 148                         errormsg(SH_DICT,ERROR_system(1),"I/O error");
 149                 if(t && ((t->tre.tretyp&COMMSK)==TCOM) && (np=t->com.comnamp) && (cp=nv_name(np)))
 150                 {
 151                         if(strcmp(cp,"exit")==0)
 152                                 break;
 153                         /* check for exec of a command */
 154                         if(strcmp(cp,"exec")==0)
 155                         {
 156                                 if(t->com.comtyp&COMSCAN)
 157                                 {
 158                                         if(t->com.comarg->argnxt.ap)
 159                                                 break;
 160                                 }
 161                                 else
 162                                 {
 163                                         struct dolnod *ap = (struct dolnod*)t->com.comarg;
 164                                         if(ap->dolnum>1)
 165                                                 break;
 166                                 }
 167                         }
 168                 }
 169         }
 170         /* copy any remaining input */
 171         sfmove(in,out,SF_UNBOUND,-1);
 172         if(in!=sfstdin)
 173                 sfclose(in);
 174         if(out!=sfstdout)
 175                 sfclose(out);
 176         return(0);
 177 }