[build] Check SHA sum of downloaded node package (elastic#8890)

Backports PR elastic#7746 **Commit 1:** fix elastic#7136 - check SHA of downloaded node binaries * Original sha: 955972b * Authored by ppisljar <[email protected]> on 2016-07-11T19:17:08Z **Commit 2:** only skipping download if --skip-download cli argument is present * Original sha: 325e172 * Authored by ppisljar <[email protected]> on 2016-09-07T10:54:23Z **Commit 3:** updating log messages based on epixas comments * Original sha: 20b5c4d * Authored by ppisljar <[email protected]> on 2016-09-23T06:24:55Z **Commit 4:** updating based on courts review * Original sha: 78c124c * Authored by ppisljar <[email protected]> on 2016-10-29T14:44:43Z Former-commit-id: 8e22a9e
airow · Oct 29, 2016 · 5a0646d · 5a0646d
1 parent e963a79
commit 5a0646d
Show file tree

Hide file tree

Showing 4 changed files with 140 additions and 70 deletions.
diff --git a/package.json b/package.json
@@ -95,6 +95,7 @@
     "boom": "2.8.0",
     "brace": "0.5.1",
     "bunyan": "1.7.1",
+    "check-hash": "1.0.1",
     "commander": "2.8.1",
     "css-loader": "0.17.0",
     "d3": "3.5.6",

diff --git a/tasks/build/download_node_builds.js b/tasks/build/download_node_builds.js
@@ -1,84 +1,151 @@
-module.exports = function (grunt) {
-  let { map, fromNode } = require('bluebird');
-  let { resolve } = require('path');
-  let { pluck } = require('lodash');
-  let { createWriteStream } = require('fs');
-  let { createGunzip } = require('zlib');
-  let { Extract } = require('tar');
-  let { rename } = require('fs');
-  let wreck = require('wreck');
-
-  let platforms = grunt.config.get('platforms');
-  let activeDownloads = [];
-
-  let start = async (platform) => {
-    let finalDir = platform.nodeDir;
-    let downloadDir = `${finalDir}.temp`;
+import { Promise, map, fromNode, promisify } from 'bluebird';
+import { resolve, basename, dirname, join } from 'path';
+import { createReadStream, createWriteStream, writeFile } from 'fs';
+import { createGunzip } from 'zlib';
+import { Extract } from 'tar';
+import { fromFile } from 'check-hash';
+import wreck from 'wreck';
+import { mkdirp } from 'mkdirp';
 
-    if (grunt.file.isDir(platform.nodeDir)) {
-      grunt.log.ok(`${platform.name} exists`);
-      return;
-    }
+const mkdirpAsync = promisify(mkdirp);
+const wreckGetAsync = promisify(wreck.get, wreck);
+const checkHashFromFileAsync = promisify(fromFile);
+const writeFileAsync = promisify(writeFile);
 
-    let resp = await fromNode(cb => {
-      let req = wreck.request('GET', platform.nodeUrl, null, function (err, resp) {
-        if (err) {
-          return cb(err);
-        }
+export default function downloadNodeBuilds(grunt) {
+  const platforms = grunt.config.get('platforms');
+  const downloadLimit = 3;
 
-        if (resp.statusCode !== 200) {
-          return cb(new Error(`${platform.nodeUrl} failed with a ${resp.statusCode} response`));
-        }
+  const shaSums = {};
+  const getShaSums = () => {
+    const nodeVersion = grunt.config.get('nodeVersion');
+    const shaSumsUri = `https://nodejs.org/dist/v${nodeVersion}/SHASUMS256.txt`;
 
-        return cb(null, resp);
+    return wreckGetAsync(shaSumsUri).then(([resp, payload]) => {
+      if (resp.statusCode !== 200) {
+        throw new Error(`${shaSumsUri} failed with a ${resp.statusCode} response`);
+      }
+      payload
+      .toString('utf8')
+      .split('\n')
+      .forEach(line => {
+        const [sha, platform] = line.split('  ');
+        shaSums[platform] = sha;
       });
     });
+  };
 
-    // use an async iife to store promise for download
-    // then store platform in active downloads list
-    // which we will read from in the finish task
-    platform.downloadPromise = (async function () {
-      grunt.file.mkdir(downloadDir);
-
-      if (platform.win) {
-        await fromNode(cb => {
-          resp
-          .pipe(createWriteStream(resolve(downloadDir, 'node.exe')))
-          .on('error', cb)
-          .on('finish', cb);
-        });
-      } else {
-        await fromNode(cb => {
-          resp
-          .pipe(createGunzip())
-          .on('error', cb)
-          .pipe(new Extract({ path: downloadDir, strip: 1 }))
-          .on('error', cb)
-          .on('end', cb);
-        });
+  const checkShaSum = (platform) => {
+    const file = basename(platform.nodeUrl);
+    const downloadDir = join(platform.nodeDir, '..');
+    const filePath = resolve(downloadDir, file);
+    const expected = {
+      hash: 'sha256',
+      expected: platform.win ? shaSums[basename(dirname(platform.nodeUrl)) + '/' + file] : shaSums[file]
+    };
+
+    if (!grunt.file.isFile(filePath)) {
+      return false;
+    }
+
+    return checkHashFromFileAsync(filePath, expected).then(([passed, actual]) => {
+      if (!passed) {
+        grunt.log.error(`${platform.name} shasum check failed`);
       }
+      return passed;
+    });
+  };
 
-      await fromNode(cb => {
-        rename(downloadDir, finalDir, cb);
-      });
-    }());
+  const getNodeBuild = (platform) => {
+    const downloadDir = join(platform.nodeDir, '..');
+    const file = basename(platform.nodeUrl);
+    const filePath = resolve(downloadDir, file);
+
+    if (grunt.file.isFile(filePath)) {
+      grunt.file.delete(filePath);
+    }
 
-    activeDownloads.push(platform);
+    return wreckGetAsync(platform.nodeUrl)
+    .then(([resp, payload]) => {
+      if (resp.statusCode !== 200) {
+        throw new Error(`${platform.nodeUrl} failed with a ${resp.statusCode} response`);
+      }
+      return payload;
+    })
+    .then(payload => writeFileAsync(filePath, payload));
 
-    var bytes = parseInt(resp.headers['content-length'], 10) || 'unknown number of';
-    var mb = ((bytes / 1024) / 1024).toFixed(2);
-    grunt.log.ok(`downloading ${platform.name} - ${mb} mb`);
   };
 
-  grunt.registerTask('_build:downloadNodeBuilds:start', function () {
-    map(platforms, start).nodeify(this.async());
+  const start = async (platform) => {
+    const downloadDir = join(platform.nodeDir, '..');
+    let downloadCounter = 0;
+    let isDownloadValid = false;
+
+    await mkdirpAsync(downloadDir);
+    if (grunt.option('skip-node-download')) {
+      grunt.log.ok(`Verifying sha sum of ${platform.name}`);
+      isDownloadValid = await checkShaSum(platform);
+      if (!isDownloadValid) {
+        throw new Error(`${platform.name} sha verification failed.`);
+      }
+      return;
+    }
+
+    while (!isDownloadValid && (downloadCounter < downloadLimit)) {
+      grunt.log.ok(`Downloading ${platform.name} and corresponding sha`);
+      await getNodeBuild(platform);
+      isDownloadValid = await checkShaSum(platform);
+      ++downloadCounter;
+    }
+
+    if (!isDownloadValid) {
+      throw new Error(`${platform.name} download failed`);
+    }
+
+    grunt.log.ok(`${platform.name} downloaded and verified`);
+  };
+
+  grunt.registerTask('_build:downloadNodeBuilds', function () {
+    const done = this.async();
+    getShaSums()
+    .then(() => map(platforms, start))
+    .nodeify(done);
   });
 
-  grunt.registerTask('_build:downloadNodeBuilds:finish', function () {
-    map(activeDownloads, async (platform) => {
-      await platform.downloadPromise;
-      grunt.log.ok(`${platform.name} download complete`);
-    })
-    .nodeify(this.async());
+  const extractNodeBuild = async (platform) => {
+    const file = basename(platform.nodeUrl);
+    const downloadDir = join(platform.nodeDir, '..');
+    const filePath = resolve(downloadDir, file);
+
+    return new Promise((resolve, reject) => {
+      createReadStream(filePath)
+        .pipe(createGunzip())
+        .on('error', reject)
+        .pipe(new Extract({path: platform.nodeDir, strip: 1}))
+        .on('error', reject)
+        .on('end', resolve);
+    });
+  };
+
+  const extract = async(platform) => {
+    const file = basename(platform.nodeUrl);
+    const downloadDir = join(platform.nodeDir, '..');
+    const filePath = resolve(downloadDir, file);
+
+    if (grunt.file.isDir(platform.nodeDir)) {
+      grunt.file.delete(platform.nodeDir);
+    }
+
+    if (platform.win) {
+      grunt.file.mkdir(platform.nodeDir);
+      grunt.file.copy(filePath, resolve(platform.nodeDir, file));
+    } else {
+      await extractNodeBuild(platform);
+    }
+  };
+
+  grunt.registerTask('_build:extractNodeBuilds', function () {
+    map(platforms, extract).nodeify(this.async());
   });
+
 };
diff --git a/tasks/build/index.js b/tasks/build/index.js
@@ -5,7 +5,8 @@ module.exports = function (grunt) {
     grunt.task.run(flatten([
       'clean:build',
       'clean:target',
-      '_build:downloadNodeBuilds:start',
+      '_build:downloadNodeBuilds',
+      '_build:extractNodeBuilds',
       'copy:devSource',
       'babel:build',
       '_build:babelOptions',
@@ -21,7 +22,6 @@ module.exports = function (grunt) {
       'clean:deepModules',
       'run:optimizeBuild',
       'stop:optimizeBuild',
-      '_build:downloadNodeBuilds:finish',
       '_build:versionedLinks',
       '_build:osShellScripts',
       grunt.option('skip-archives') ? [] : ['_build:archives'],

diff --git a/tasks/config/platforms.js b/tasks/config/platforms.js
@@ -23,6 +23,8 @@ module.exports = function (grunt) {
       ? baseName.replace('-x64', '-x86_64')
       : baseName;
 
+    let nodeShaSums = `${baseUri}/SHASUMS256.txt`;
+
     let buildName = `kibana-${version}-${name}`;
     let buildDir = resolve(rootPath, `build/${buildName}`);
 
@@ -49,7 +51,7 @@ module.exports = function (grunt) {
     }
     return {
       name, win,
-      nodeUrl, nodeDir,
+      nodeUrl, nodeDir, nodeShaSums,
       buildName, buildDir,
       tarName, tarPath,
       zipName, zipPath,