make-fingerprint.c 4.07 KB
Newer Older
Daniel Colascione's avatar
Daniel Colascione committed
1 2
/* Hash inputs and generate C file with the digest.

3 4
Copyright (C) 1985-1986, 1992-1994, 1997, 1999-2016, 2018-2019
Free Software Foundation, Inc.
Daniel Colascione's avatar
Daniel Colascione committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27

This file is part of GNU Emacs.

GNU Emacs is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or (at
your option) any later version.

GNU Emacs is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with GNU Emacs.  If not, see <http://www.gnu.org/licenses/>.  */


/* The arguments given to this program are all the object files that
 go into building GNU Emacs.  There is no special search logic to find
 the files.  */

#include <config.h>

28
#include <limits.h>
Daniel Colascione's avatar
Daniel Colascione committed
29 30 31 32 33
#include <stdarg.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
34 35
#include <sys/stat.h>

Daniel Colascione's avatar
Daniel Colascione committed
36
#include <sysstdio.h>
37 38

#include <fingerprint.h>
Daniel Colascione's avatar
Daniel Colascione committed
39
#include <getopt.h>
40 41 42 43 44 45 46
#include <intprops.h>
#include <min-max.h>
#include <sha256.h>

#ifndef SSIZE_MAX
# define SSIZE_MAX TYPE_MAXIMUM (ssize_t)
#endif
Daniel Colascione's avatar
Daniel Colascione committed
47 48 49 50 51 52

#ifdef WINDOWSNT
/* Defined to be sys_fopen in ms-w32.h, but only #ifdef emacs, so this
   is really just insurance.  */
#undef fopen
#include <direct.h>
53 54 55 56 57

#ifndef MINGW_W64
# undef fseeko
# define fseeko fseeko64
#endif
Daniel Colascione's avatar
Daniel Colascione committed
58 59
#endif /* WINDOWSNT */

60 61 62
/* Static (instead of being local to 'main') to pacify LeakSanitizer.  */
static char *buf;

Daniel Colascione's avatar
Daniel Colascione committed
63 64 65 66 67 68 69 70 71 72 73 74 75
int
main (int argc, char **argv)
{
  int c;
  bool raw = false;
  while (0 <= (c = getopt (argc, argv, "rh")))
    {
      switch (c)
        {
        case 'r':
          raw = true;
          break;
        case 'h':
76 77
          printf ("make-fingerprint [-r] FILE: replace or compute a hash\n");
          return EXIT_SUCCESS;
Daniel Colascione's avatar
Daniel Colascione committed
78
        default:
79
          return EXIT_FAILURE;
Daniel Colascione's avatar
Daniel Colascione committed
80 81 82 83 84 85
        }
    }

  struct sha256_ctx ctx;
  sha256_init_ctx (&ctx);

86
  char *prog = argv[0];
87
  char *file = argv[optind];
88
  if (argc - optind != 1)
Daniel Colascione's avatar
Daniel Colascione committed
89
    {
90
      fprintf (stderr, "%s: missing or extra file operand\n", prog);
91 92
      return EXIT_FAILURE;
    }
Daniel Colascione's avatar
Daniel Colascione committed
93

94
  FILE *f = fopen (file, raw ? "r" FOPEN_BINARY : "r+" FOPEN_BINARY);
95 96 97
  struct stat st;
  if (!f || fstat (fileno (f), &st) != 0)
    {
98
      perror (file);
99
      return EXIT_FAILURE;
Daniel Colascione's avatar
Daniel Colascione committed
100 101
    }

102 103 104
  if (!S_ISREG (st.st_mode))
    {
      fprintf (stderr, "%s: Error: %s is not a regular file\n",
105
	       prog, file);
106 107 108 109 110 111 112
      return EXIT_FAILURE;
    }

  ptrdiff_t maxlen = min (min (TYPE_MAXIMUM (off_t), PTRDIFF_MAX),
			  min (SIZE_MAX, SSIZE_MAX));
  if (maxlen <= st.st_size)
    {
113
      fprintf (stderr, "%s: %s: file too big\n", prog, file);
114 115 116
      return EXIT_FAILURE;
    }

117
  buf = malloc (st.st_size + 1);
118 119 120 121 122 123 124 125 126
  if (!buf)
    {
      perror ("malloc");
      return EXIT_FAILURE;
    }

  size_t chunksz = fread (buf, 1, st.st_size + 1, f);
  if (ferror (f) || chunksz != st.st_size)
    {
127
      fprintf (stderr, "%s: Error: could not read %s\n", prog, file);
128 129 130 131 132
      return EXIT_FAILURE;
    }

  sha256_process_bytes (buf, chunksz, &ctx);

133
  unsigned char digest[32];
Daniel Colascione's avatar
Daniel Colascione committed
134 135 136 137 138 139 140 141 142
  sha256_finish_ctx (&ctx, digest);

  if (raw)
    {
      for (int i = 0; i < 32; ++i)
        printf ("%02X", digest[i]);
    }
  else
    {
143 144 145
      char *finger = memmem (buf, chunksz, fingerprint, sizeof fingerprint);
      if (!finger)
	{
146
	  fprintf (stderr, "%s: %s: missing fingerprint\n", prog, file);
147 148 149 150 151 152
	  return EXIT_FAILURE;
	}
      else if (memmem (finger + 1, buf + chunksz - (finger + 1),
		       fingerprint, sizeof fingerprint))
	{
	  fprintf (stderr, "%s: %s: two occurrences of fingerprint\n",
153
		   prog, file);
154 155 156 157 158
	  return EXIT_FAILURE;
	}

      if (fseeko (f, finger - buf, SEEK_SET) != 0)
	{
159
	  perror (file);
160 161 162 163 164
	  return EXIT_FAILURE;
	}

      if (fwrite (digest, 1, sizeof digest, f) != sizeof digest)
	{
165
	  perror (file);
166 167 168 169 170 171
	  return EXIT_FAILURE;
	}
    }

  if (fclose (f) != 0)
    {
172
      perror (file);
173
      return EXIT_FAILURE;
Daniel Colascione's avatar
Daniel Colascione committed
174 175 176 177 178 179
    }

  return EXIT_SUCCESS;
}

/* make-fingerprint.c ends here */