Question

我使用node和express来处理文件上传，并且我使用multiparty / busboy和请求将它们直接传输到转换服务。

有没有办法在将流发送到相应的提供商之前验证流是否有某些特定的文件类型？我尝试https://github.com/mscdex/mmmagic从第一个块中获取MIME类型，并且它运行良好。我想知道以下工作流程是否可能以某种方式起作用：

缓冲文件上传流并检查Mime类型的传入数据。
检查前几个块并且mime类型正确时，将缓冲区清空到请求流中。
当mime类型结果不正确时，发送错误消息并返回。

我试图让它工作但我似乎有一些流兼容性问题（节点0.8.x与节点0.10.x流，请求库不支持）。

是否有解决此问题的最佳做法？我看错了吗？

编辑：感谢Paul我想出了这段代码：

https://gist.github.com/chmanie/8520572

Answer 1

除了检查客户端请求的Content-Type标头之外，我还不知道检查MIME类型的更好，更聪明的方法。

您可以使用Transform stream实施上述解决方案。在此示例中，转换流缓冲一些任意数量的数据，然后将其发送到MIME检查库。如果一切正常，它会重新发出数据。随后的块将按原样发出。

var stream = require('readable-stream');
var mmm = require('mmmagic');
var mimeChecker = new stream.Transform();
mimeChecker.data = [];
mimeChecker.mimeFound = false;
mimeChecker._transform = function (chunk, encoding, done) {
  var self = this;

  if (self.mimeFound) {
    self.push(chunk);
    return done();
  }

  self.data.push(chunk);
  if (self.data.length < 10) {
    return done();
  }
  else if (self.data.length === 10) {
    var buffered = Buffer.concat(this.data);
    new mmm.Magic(mmm.MAGIC_MIME_TYPE).detect(buffered, function(err, result) {
      if (err) return self.emit('error', err);
      if (result !== 'text/plain') return self.emit('error', new Error('Wrong MIME'));
      self.data.map(self.push.bind(self));
      self.mimeFound = true;
      return done();
    });
  }
};

然后，您可以将此转换流传输到任何其他流，例如请求流（完全支持节点0.10流）。

// Usage example
var fs = require('fs');
fs.createReadStream('input.txt').pipe(mimeChecker).pipe(fs.createWriteStream('output.txt'));

编辑：为了更清楚地了解您在Node 0.8和0.10流之间遇到的不兼容性，当您定义流并将.on('data')侦听器附加到它时，它将切换到流模式（也就是0.8流），这意味着即使目的地没有收听也会发出数据。如果您向Magic.detect()发送异步请求，就会发生这种情况：即使您正在监听数据，数据仍然会流动。

验证node.js中上传文件的mime类型

1 个答案: