usr/src/lib/libshell/common/bltins/hist.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 #include        "defs.h"
  22 #include        <stak.h>
  23 #include        <ls.h>
  24 #include        <error.h>
  25 #include        "variables.h"
  26 #include        "io.h"
  27 #include        "name.h"
  28 #include        "history.h"
  29 #include        "builtins.h"
  30 #if SHOPT_HISTEXPAND
  31 #   include     "edit.h"
  32 #endif
  33
  34 #define HIST_RECURSE    5
  35
  36 static void hist_subst(const char*, int fd, char*);
  37
  38 #if 0
  39     /* for the benefit of the dictionary generator */
  40     int b_fc(int argc,char *argv[], void *extra){}
  41 #endif
  42 int     b_hist(int argc,char *argv[], void *extra)
  43 {
  44         register History_t *hp;
  45         register char *arg;
  46         register int flag,fdo;
  47         register Shell_t *shp = ((Shbltin_t*)extra)->shp;
  48         Sfio_t *outfile;
  49         char *fname;
  50         int range[2], incr, index2, indx= -1;
  51         char *edit = 0;         /* name of editor */
  52         char *replace = 0;              /* replace old=new */
  53         int lflag = 0, nflag = 0, rflag = 0;
  54 #if SHOPT_HISTEXPAND
  55         int pflag = 0;
  56 #endif
  57         Histloc_t location;
  58         NOT_USED(argc);
  59         if(!sh_histinit((void*)shp))
  60                 errormsg(SH_DICT,ERROR_system(1),e_histopen);
  61         hp = shp->hist_ptr;
  62         while((flag = optget(argv,sh_opthist))) switch(flag)
  63         {
  64             case 'e':
  65                 edit = opt_info.arg;
  66                 break;
  67             case 'n':
  68                 nflag++;
  69                 break;
  70             case 'l':
  71                 lflag++;
  72                 break;
  73             case 'r':
  74                 rflag++;
  75                 break;
  76             case 's':
  77                 edit = "-";
  78                 break;
  79 #if SHOPT_HISTEXPAND
  80             case 'p':
  81                 pflag++;
  82                 break;
  83 #endif
  84             case 'N':
  85                 if(indx<=0)
  86                 {
  87                         if((flag = hist_max(hp) - opt_info.num-1) < 0)
  88                                 flag = 1;
  89                         range[++indx] = flag;
  90                         break;
  91                 }
  92             case ':':
  93                 errormsg(SH_DICT,2, "%s", opt_info.arg);
  94                 break;
  95             case '?':
  96                 errormsg(SH_DICT,ERROR_usage(2), "%s", opt_info.arg);
  97                 break;
  98         }
  99         if(error_info.errors)
 100                 errormsg(SH_DICT,ERROR_usage(2),"%s",optusage((char*)0));
 101         argv += (opt_info.index-1);
 102 #if SHOPT_HISTEXPAND
 103         if(pflag)
 104         {
 105                 hist_cancel(hp);
 106                 pflag = 0;
 107                 while(arg=argv[1])
 108                 {
 109                         flag = hist_expand(arg,&replace);
 110                         if(!(flag & HIST_ERROR))
 111                                 sfputr(sfstdout, replace, '\n');
 112                         else
 113                                 pflag = 1;
 114                         free(replace);
 115                         argv++;
 116                 }
 117                 return pflag;
 118         }
 119 #endif
 120         flag = indx;
 121         while(flag<1 && (arg=argv[1]))
 122         {
 123                 /* look for old=new argument */
 124                 if(!replace && strchr(arg+1,'='))
 125                 {
 126                         replace = arg;
 127                         argv++;
 128                         continue;
 129                 }
 130                 else if(isdigit(*arg) || *arg == '-')
 131                 {
 132                         /* see if completely numeric */
 133                         do      arg++;
 134                         while(isdigit(*arg));
 135                         if(*arg==0)
 136                         {
 137                                 arg = argv[1];
 138                                 range[++flag] = (int)strtol(arg, (char**)0, 10);
 139                                 if(*arg == '-')
 140                                         range[flag] += (hist_max(hp)-1);
 141                                 argv++;
 142                                 continue;
 143                         }
 144                 }
 145                 /* search for last line starting with string */
 146                 location = hist_find(hp,argv[1],hist_max(hp)-1,0,-1);
 147                 if((range[++flag] = location.hist_command) < 0)
 148                         errormsg(SH_DICT,ERROR_exit(1),e_found,argv[1]);
 149                 argv++;
 150         }
 151         if(flag <0)
 152         {
 153                 /* set default starting range */
 154                 if(lflag)
 155                 {
 156                         flag = hist_max(hp)-16;
 157                         if(flag<1)
 158                                 flag = 1;
 159                 }
 160                 else
 161                         flag = hist_max(hp)-2;
 162                 range[0] = flag;
 163                 flag = 0;
 164         }
 165         index2 = hist_min(hp);
 166         if(range[0]<index2)
 167                 range[0] = index2;
 168         if(flag==0)
 169                 /* set default termination range */
 170                 range[1] = ((lflag && !edit)?hist_max(hp)-1:range[0]);
 171         if(range[1]>=(flag=(hist_max(hp) - !lflag)))
 172                 range[1] = flag;
 173         /* check for valid ranges */
 174         if(range[1]<index2 || range[0]>=flag)
 175                 errormsg(SH_DICT,ERROR_exit(1),e_badrange,range[0],range[1]);
 176         if(edit && *edit=='-' && range[0]!=range[1])
 177                 errormsg(SH_DICT,ERROR_exit(1),e_eneedsarg);
 178         /* now list commands from range[rflag] to range[1-rflag] */
 179         incr = 1;
 180         flag = rflag>0;
 181         if(range[1-flag] < range[flag])
 182                 incr = -1;
 183         if(lflag)
 184         {
 185                 outfile = sfstdout;
 186                 arg = "\n\t";
 187         }
 188         else
 189         {
 190                 if(!(fname=pathtmp(NIL(char*),0,0,NIL(int*))))
 191                         errormsg(SH_DICT,ERROR_exit(1),e_create,"");
 192                 if((fdo=open(fname,O_CREAT|O_RDWR,S_IRUSR|S_IWUSR)) < 0)
 193                         errormsg(SH_DICT,ERROR_system(1),e_create,fname);
 194                 outfile= sfnew(NIL(Sfio_t*),shp->outbuff,IOBSIZE,fdo,SF_WRITE);
 195                 arg = "\n";
 196                 nflag++;
 197         }
 198         while(1)
 199         {
 200                 if(nflag==0)
 201                         sfprintf(outfile,"%d\t",range[flag]);
 202                 else if(lflag)
 203                         sfputc(outfile,'\t');
 204                 hist_list(shp->hist_ptr,outfile,hist_tell(shp->hist_ptr,range[flag]),0,arg);
 205                 if(lflag)
 206                         sh_sigcheck();
 207                 if(range[flag] == range[1-flag])
 208                         break;
 209                 range[flag] += incr;
 210         }
 211         if(lflag)
 212                 return(0);
 213         sfclose(outfile);
 214         hist_eof(hp);
 215         arg = edit;
 216         if(!arg && !(arg=nv_getval(sh_scoped(shp,HISTEDIT))) && !(arg=nv_getval(sh_scoped(shp,FCEDNOD))))
 217                 arg = (char*)e_defedit;
 218 #ifdef apollo
 219         /*
 220          * Code to support the FC using the pad editor.
 221          * Exampled of how to use: HISTEDIT=pad
 222          */
 223         if (strcmp (arg, "pad") == 0)
 224         {
 225                 extern int pad_create(char*);
 226                 sh_close(fdo);
 227                 fdo = pad_create(fname);
 228                 pad_wait(fdo);
 229                 unlink(fname);
 230                 strcat(fname, ".bak");
 231                 unlink(fname);
 232                 lseek(fdo,(off_t)0,SEEK_SET);
 233         }
 234         else
 235         {
 236 #endif /* apollo */
 237         if(*arg != '-')
 238         {
 239                 char *com[3];
 240                 com[0] =  arg;
 241                 com[1] =  fname;
 242                 com[2] = 0;
 243                 error_info.errors = sh_eval(sh_sfeval(com),0);
 244         }
 245         fdo = sh_chkopen(fname);
 246         unlink(fname);
 247         free((void*)fname);
 248 #ifdef apollo
 249         }
 250 #endif /* apollo */
 251         /* don't history fc itself unless forked */
 252         error_info.flags |= ERROR_SILENT;
 253         if(!sh_isstate(SH_FORKED))
 254                 hist_cancel(hp);
 255         sh_onstate(SH_HISTORY);
 256         sh_onstate(SH_VERBOSE); /* echo lines as read */
 257         if(replace)
 258                 hist_subst(error_info.id,fdo,replace);
 259         else if(error_info.errors == 0)
 260         {
 261                 char buff[IOBSIZE+1];
 262                 Sfio_t *iop = sfnew(NIL(Sfio_t*),buff,IOBSIZE,fdo,SF_READ);
 263                 /* read in and run the command */
 264                 if(shp->hist_depth++ > HIST_RECURSE)
 265                         errormsg(SH_DICT,ERROR_exit(1),e_toodeep,"history");
 266                 sh_eval(iop,1);
 267                 shp->hist_depth--;
 268         }
 269         else
 270         {
 271                 sh_close(fdo);
 272                 if(!sh_isoption(SH_VERBOSE))
 273                         sh_offstate(SH_VERBOSE);
 274                 sh_offstate(SH_HISTORY);
 275         }
 276         return(shp->exitval);
 277 }
 278
 279
 280 /*
 281  * given a file containing a command and a string of the form old=new,
 282  * execute the command with the string old replaced by new
 283  */
 284
 285 static void hist_subst(const char *command,int fd,char *replace)
 286 {
 287         register char *newp=replace;
 288         register char *sp;
 289         register int c;
 290         off_t size;
 291         char *string;
 292         while(*++newp != '='); /* skip to '=' */
 293         if((size = lseek(fd,(off_t)0,SEEK_END)) < 0)
 294                 return;
 295         lseek(fd,(off_t)0,SEEK_SET);
 296         c =  (int)size;
 297         string = stakalloc(c+1);
 298         if(read(fd,string,c)!=c)
 299                 return;
 300         string[c] = 0;
 301         *newp++ =  0;
 302         if((sp=sh_substitute(string,replace,newp))==0)
 303                 errormsg(SH_DICT,ERROR_exit(1),e_subst,command);
 304         *(newp-1) =  '=';
 305         sh_eval(sfopen(NIL(Sfio_t*),sp,"s"),1);
 306 }
 307