NodeJS 在目录中递归地哈希文件

v01*_*01d 5 recursion hash integrity node.js

我能够实现目录中的递归文件遍历(即探索目录中的所有子目录和文件)。为此,我使用了有关堆栈溢出的相应帖子的答案。其片段如下:

var fs = require("fs");

var tree = function(dir, done) {
  var results = {
        "path": dir,
        "children": []
      };
  fs.readdir(dir, function(err, list) {
    if (err) { return done(err); }
    var pending = list.length;
    if (!pending) { return done(null, results); }
    list.forEach(function(file) {
      fs.stat(dir + '/' + file, function(err, stat) {
        if (stat && stat.isDirectory()) {
          tree(dir + '/' + file, function(err, res) {
            results.children.push(res);
            if (!--pending){ done(null, results); }
          });
        } else {
          results.children.push({"path": dir + "/" + file});
          if (!--pending) { done(null, results); }
        }
      });
    });
  });
};

module.exports = tree;
Run Code Online (Sandbox Code Playgroud)

当我跑步时:

 tree(someDirectoryPath, function(err, results) {
        if (err) throw err;

        console.log(results);
      });
Run Code Online (Sandbox Code Playgroud)

我得到一个示例结果,例如:

{ path: '/Users/UserName/Desktop/1',
  children: 
   [ { path: '/Users/UserName/Desktop/1/file1' },
     { path: '/Users/UserName/Desktop/1/file2' },
     { path: '/Users/UserName/Desktop/1/file3' },
     { path: '/Users/UserName/Desktop/1/subdir1',
       children: [Object] } ] }
Run Code Online (Sandbox Code Playgroud)

我还可以使用 fs 模块 ReadStream 方法对特定位置的单个文件进行哈希处理。其片段如下:

/**
 * Checking File Integrity
 */
var fs = require('fs'),
      args = process.argv.splice('2'),
      path = require('path'),
      traverse = require('/Users/UserName/Desktop/tree.js'),
      crypto = require('crypto');
//var algorithm = ['md5', 'sha1', 'sha256', 'sha512'];
var algorithm = 'sha512';
var hashTable = new Array();

        var hash = crypto.createHash(algorithm);

        var fileStream = fs.ReadStream(args[0]);

        fileStream.on('data', function(data) {
                hash.update(data);
        fileStream.on('end', function() {
                var digest = hash.digest('hex');
                console.log('algorithm used: ', algorithm);
                console.log('hash for the file: ',digest);
                hashTable[args[0]] = digest;
                console.log(hashTable);
        });
});
Run Code Online (Sandbox Code Playgroud)

其中args[0]存储ReadStream要读取的文件的位置。对特定文件进行哈希处理后,返回的控制台日志如下:

node fileIntegrityChecker.js hello.txt
algorithm used:  sha512
hash for the file:  9b71d224bd62f3785d96d46ad3ea3d73319bfbc2890caadae2dff72519673ca72323c3d99ba5c11d7c7acc6e14b8c5da0c4663475c2e5c3adef46f73bcdec043
the hashtable is: [ 'hello.txt': '9b71d224bd62f3785d96d46ad3ea3d73319bfbc2890caadae2dff72519673ca72323c3d99ba5c11d7c7acc6e14b8c5da0c4663475c2e5c3adef46f73bcdec043' ]
Run Code Online (Sandbox Code Playgroud)

我的问题是我尝试以某种方式将树模块功能集成到哈希相关的 js 文件中。我的想法是,程序将捕获用户的输入,作为目录的路径,并且该输入将被处理以遍历文件夹的整个子目录和文件。另外,fileStream.on方法应包含在树模块的回调中。不过我对回调机制还没有完全了解,希望能从你那里得到一些见解。

这是我尝试过的

/**
 * Checking File Integrity
 */
var fs = require('fs'),
      args = process.argv.splice('2'),
      path = require('path'),
      tree = require('/Users/UserName/Desktop/tree.js'),
      crypto = require('crypto');
//var algorithm = ['md5', 'sha1', 'sha256', 'sha512'];
var algorithm = 'sha512';
var hashTable = new Array();

        var pathString = 'Users/UserName/Desktop/1';
        tree(pathString, function(err, results) {
            if (err) throw err;

            var hash = crypto.createHash(algorithm);
            var fileStream = fs.ReadStream(results.children[1]['path']);
             fileStream.on('data', function(data) {
                hash.update(data);
             fileStream.on('end', function() {
                var digest = hash.digest('hex');
                console.log('algorithm used: ', algorithm);
                console.log('hash for the file: ',digest);
                hashTable[results.children[1]['path']] = digest;
                console.log('The hashtable is: ', hashTable);
                });
            });
        });
Run Code Online (Sandbox Code Playgroud)

现在,我已经取得了一些进展,没有收到错误。基本上我达到了我的范围。但是我只能提取一个结果。由于某种原因,我无法想象如何迭代(例如)获取结果 JSON 对象的每个子对象。如果这个问题解决了,我想问题就彻底解决了。

您能告诉我一种如何成功地将模块和js文件结合起来以递归遍历目录的所有内容并为其中的每个文件创建哈希的方法吗?我需要这个来根据文件的哈希值最终检查文件中是否发生了一些更改。谢谢你!

chr*_*lly 2

最简单的方法是在您已经遍历目录树时生成哈希值。这涉及更新 tree.js 文件,如下所示:

    } else {
      var fname = dir + "/" + file};
      // put your hash generation here
      generateHash(fname, function (e, hash) {
        if (e) done(e);

        results.children.push({"path": fname, "hash" : hash);
        if (!--pending) { 
          done(null, results); 
        }
      });
    }
Run Code Online (Sandbox Code Playgroud)

然后将哈希生成代码放入如下函数中:

function generateHash (filename, callback) {
    var algorithm = 'sha512';
    var hashTable = new Array();

    var hash = crypto.createHash(algorithm);
    var fileStream = fs.ReadStream(filename);

    fileStream.on('data', function(data) {
        hash.update(data);      
    });
    fileStream.on('end', function() {
        var digest = hash.digest('hex');
        callback(null, digest);
    });
}
Run Code Online (Sandbox Code Playgroud)