usr.bin/hexdump/display.c

   1 /*
   2  * Copyright (c) 1989, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  *
  33  * @(#)display.c        8.1 (Berkeley) 6/6/93
  34  * $FreeBSD: src/usr.bin/hexdump/display.c,v 1.4.2.2 2002/07/23 14:27:06 tjr Exp $
  35  * $DragonFly: src/usr.bin/hexdump/display.c,v 1.6 2005/04/10 20:55:38 drhodus Exp $
  36  */
  37
  38 #include <sys/param.h>
  39 #include <sys/stat.h>
  40
  41 #include <ctype.h>
  42 #include <err.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <unistd.h>
  47 #include "hexdump.h"
  48
  49 enum _vflag vflag = FIRST;
  50
  51 static off_t address;                   /* address/offset in stream */
  52 static off_t eaddress;                  /* end address */
  53
  54 static __inline void print(PR *, u_char *);
  55
  56 static __inline void
  57 print(PR *pr, u_char *bp)
  58 {
  59         long double ldbl;
  60            double f8;
  61             float f4;
  62           int16_t s2;
  63            int8_t s8;
  64           int32_t s4;
  65         u_int16_t u2;
  66         u_int32_t u4;
  67         u_int64_t u8;
  68
  69         switch(pr->flags) {
  70         case F_ADDRESS:
  71                 (void)printf(pr->fmt, (quad_t)address);
  72                 break;
  73         case F_BPAD:
  74                 (void)printf(pr->fmt, "");
  75                 break;
  76         case F_C:
  77                 conv_c(pr, bp);
  78                 break;
  79         case F_CHAR:
  80                 (void)printf(pr->fmt, *bp);
  81                 break;
  82         case F_DBL:
  83                 switch(pr->bcnt) {
  84                 case 4:
  85                         bcopy(bp, &f4, sizeof(f4));
  86                         (void)printf(pr->fmt, f4);
  87                         break;
  88                 case 8:
  89                         bcopy(bp, &f8, sizeof(f8));
  90                         (void)printf(pr->fmt, f8);
  91                         break;
  92                 default:
  93                         if (pr->bcnt == sizeof(long double)) {
  94                                 bcopy(bp, &ldbl, sizeof(ldbl));
  95                                 (void)printf(pr->fmt, ldbl);
  96                         }
  97                         break;
  98                 }
  99                 break;
 100         case F_INT:
 101                 switch(pr->bcnt) {
 102                 case 1:
 103                         (void)printf(pr->fmt, (quad_t)(signed char)*bp);
 104                         break;
 105                 case 2:
 106                         bcopy(bp, &s2, sizeof(s2));
 107                         (void)printf(pr->fmt, (quad_t)s2);
 108                         break;
 109                 case 4:
 110                         bcopy(bp, &s4, sizeof(s4));
 111                         (void)printf(pr->fmt, (quad_t)s4);
 112                         break;
 113                 case 8:
 114                         bcopy(bp, &s8, sizeof(s8));
 115                         (void)printf(pr->fmt, s8);
 116                         break;
 117                 }
 118                 break;
 119         case F_P:
 120                 (void)printf(pr->fmt, isprint(*bp) ? *bp : '.');
 121                 break;
 122         case F_STR:
 123                 (void)printf(pr->fmt, (char *)bp);
 124                 break;
 125         case F_TEXT:
 126                 (void)printf("%s", pr->fmt);
 127                 break;
 128         case F_U:
 129                 conv_u(pr, bp);
 130                 break;
 131         case F_UINT:
 132                 switch(pr->bcnt) {
 133                 case 1:
 134                         (void)printf(pr->fmt, (u_quad_t)*bp);
 135                         break;
 136                 case 2:
 137                         bcopy(bp, &u2, sizeof(u2));
 138                         (void)printf(pr->fmt, (u_quad_t)u2);
 139                         break;
 140                 case 4:
 141                         bcopy(bp, &u4, sizeof(u4));
 142                         (void)printf(pr->fmt, (u_quad_t)u4);
 143                         break;
 144                 case 8:
 145                         bcopy(bp, &u8, sizeof(u8));
 146                         (void)printf(pr->fmt, u8);
 147                         break;
 148                 }
 149                 break;
 150         }
 151 }
 152
 153 void
 154 display(void)
 155 {
 156
 157         FS *fs;
 158         FU *fu;
 159         PR *pr;
 160         int cnt;
 161         u_char *bp;
 162         off_t saveaddress;
 163         u_char savech = '\0', *savebp;
 164
 165         while ((bp = get()))
 166             for (fs = fshead, savebp = bp, saveaddress = address; fs;
 167                 fs = fs->nextfs, bp = savebp, address = saveaddress)
 168                     for (fu = fs->nextfu; fu; fu = fu->nextfu) {
 169                         if (fu->flags&F_IGNORE)
 170                                 break;
 171                         for (cnt = fu->reps; cnt; --cnt)
 172                             for (pr = fu->nextpr; pr; address += pr->bcnt,
 173                                 bp += pr->bcnt, pr = pr->nextpr) {
 174                                     if (eaddress && address >= eaddress &&
 175                                         !(pr->flags & (F_TEXT|F_BPAD)))
 176                                             bpad(pr);
 177                                     if (cnt == 1 && pr->nospace) {
 178                                         savech = *pr->nospace;
 179                                         *pr->nospace = '\0';
 180                                     }
 181                                     print(pr, bp);
 182                                     if (cnt == 1 && pr->nospace)
 183                                         *pr->nospace = savech;
 184                             }
 185                     }
 186         if (endfu) {
 187                 /*
 188                  * If eaddress not set, error or file size was multiple of
 189                  * blocksize, and no partial block ever found.
 190                  */
 191                 if (!eaddress) {
 192                         if (!address)
 193                                 return;
 194                         eaddress = address;
 195                 }
 196                 for (pr = endfu->nextpr; pr; pr = pr->nextpr)
 197                         switch(pr->flags) {
 198                         case F_ADDRESS:
 199                                 (void)printf(pr->fmt, (quad_t)eaddress);
 200                                 break;
 201                         case F_TEXT:
 202                                 (void)printf("%s", pr->fmt);
 203                                 break;
 204                         }
 205         }
 206 }
 207
 208 void
 209 bpad(PR *pr)
 210 {
 211         static char const *spec = " -0+#";
 212         char *p1, *p2;
 213
 214         /*
 215          * Remove all conversion flags; '-' is the only one valid
 216          * with %s, and it's not useful here.
 217          */
 218         pr->flags = F_BPAD;
 219         pr->cchar[0] = 's';
 220         pr->cchar[1] = '\0';
 221         for (p1 = pr->fmt; *p1 != '%'; ++p1);
 222         for (p2 = ++p1; *p1 && strchr(spec, *p1); ++p1);
 223         while ((*p2++ = *p1++));
 224 }
 225
 226 static char **_argv;
 227
 228 u_char *
 229 get(void)
 230 {
 231         static int ateof = 1;
 232         static u_char *curp, *savp;
 233         int n;
 234         int need, nread;
 235         int valid_save = 0;
 236         u_char *tmpp;
 237
 238         if (!curp) {
 239                 if ((curp = calloc(1, blocksize)) == NULL)
 240                         err(1, NULL);
 241                 if ((savp = calloc(1, blocksize)) == NULL)
 242                         err(1, NULL);
 243         } else {
 244                 tmpp = curp;
 245                 curp = savp;
 246                 savp = tmpp;
 247                 address += blocksize;
 248                 valid_save = 1;
 249         }
 250         for (need = blocksize, nread = 0;;) {
 251                 /*
 252                  * if read the right number of bytes, or at EOF for one file,
 253                  * and no other files are available, zero-pad the rest of the
 254                  * block and set the end flag.
 255                  */
 256                 if (!length || (ateof && !next(NULL))) {
 257                         if (odmode && address < skip)
 258                                 errx(1, "cannot skip past end of input");
 259                         if (need == blocksize)
 260                                 return(NULL);
 261                         if (vflag != ALL &&
 262                             valid_save &&
 263                             bcmp(curp, savp, nread) == 0) {
 264                                 if (vflag != DUP)
 265                                         (void)printf("*\n");
 266                                 return(NULL);
 267                         }
 268                         bzero((char *)curp + nread, need);
 269                         eaddress = address + nread;
 270                         return(curp);
 271                 }
 272                 n = fread((char *)curp + nread, sizeof(u_char),
 273                     length == -1 ? need : MIN(length, need), stdin);
 274                 if (!n) {
 275                         if (ferror(stdin))
 276                                 warn("%s", _argv[-1]);
 277                         ateof = 1;
 278                         continue;
 279                 }
 280                 ateof = 0;
 281                 if (length != -1)
 282                         length -= n;
 283                 if (!(need -= n)) {
 284                         if (vflag == ALL || vflag == FIRST ||
 285                             valid_save == 0 ||
 286                             bcmp(curp, savp, blocksize) != 0) {
 287                                 if (vflag == DUP || vflag == FIRST)
 288                                         vflag = WAIT;
 289                                 return(curp);
 290                         }
 291                         if (vflag == WAIT)
 292                                 (void)printf("*\n");
 293                         vflag = DUP;
 294                         address += blocksize;
 295                         need = blocksize;
 296                         nread = 0;
 297                 }
 298                 else
 299                         nread += n;
 300         }
 301 }
 302
 303 int
 304 next(char **argv)
 305 {
 306         static int done;
 307
 308         if (argv) {
 309                 _argv = argv;
 310                 return(1);
 311         }
 312         for (;;) {
 313                 if (*_argv) {
 314                         if (!(freopen(*_argv, "r", stdin))) {
 315                                 warn("%s", *_argv);
 316                                 exitval = 1;
 317                                 ++_argv;
 318                                 continue;
 319                         }
 320                         done = 1;
 321                 } else {
 322                         if (done++)
 323                                 return(0);
 324                 }
 325                 if (skip)
 326                         doskip(*_argv ? *_argv : "stdin");
 327                 if (*_argv)
 328                         ++_argv;
 329                 if (!skip)
 330                         return(1);
 331         }
 332         /* NOTREACHED */
 333 }
 334
 335 void
 336 doskip(const char *fname)
 337 {
 338         int cnt;
 339         struct stat sb;
 340
 341         if (fstat(fileno(stdin), &sb))
 342                 err(1, "%s", fname);
 343         if (S_ISREG(sb.st_mode) && skip >= sb.st_size) {
 344                 address += sb.st_size;
 345                 skip -= sb.st_size;
 346         } else if (S_ISREG(sb.st_mode)) {
 347                 if (fseeko(stdin, skip, SEEK_SET))
 348                         err(1, "%s", fname);
 349                 address += skip;
 350                 skip = 0;
 351         } else {
 352                 for (cnt = 0; cnt < skip; ++cnt)
 353                         if (getchar() == EOF)
 354                                 break;
 355                 address += cnt;
 356                 skip -= cnt;
 357         }
 358 }