Created
March 24, 2015 02:35
-
-
Save trentm/9968c2fc43d630e3cdad to your computer and use it in GitHub Desktop.
compressed file type detection using magic numbers. Wrote it. Don't need it right now.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/* | |
* CDDL HEADER START | |
* | |
* The contents of this file are subject to the terms of the | |
* Common Development and Distribution License, Version 1.0 only | |
* (the "License"). You may not use this file except in compliance | |
* with the License. | |
* | |
* You can obtain a copy of the license at http://smartos.org/CDDL | |
* | |
* See the License for the specific language governing permissions | |
* and limitations under the License. | |
* | |
* When distributing Covered Code, include this CDDL HEADER in each | |
* file. | |
* | |
* If applicable, add the following below this CDDL HEADER, with the | |
* fields enclosed by brackets "[]" replaced with your own identifying | |
* information: Portions Copyright [yyyy] [name of copyright owner] | |
* | |
* CDDL HEADER END | |
* | |
* Copyright (c) 2015, Joyent, Inc. All rights reserved. | |
* | |
* * * | |
* Magic number sniffing of compression type of files. | |
*/ | |
var p = console.log; | |
var assert = require('assert-plus'); | |
var format = require('util').format; | |
var fs = require('fs'); | |
var vasync = require('vasync'); | |
// ---- compression type sniffing | |
var magicNumbers = { | |
// <compression type> : <magic number> | |
bzip2: new Buffer([0x42, 0x5A, 0x68]), | |
gzip: new Buffer([0x1F, 0x8B, 0x08]), | |
xz: new Buffer([0xFD, 0x37, 0x7A, 0x58, 0x5A, 0x00]) | |
}; | |
var maxMagicLen = 0; | |
Object.keys(magicNumbers).forEach(function (type) { | |
maxMagicLen = Math.max(maxMagicLen, magicNumbers[type].length); | |
}); | |
function bufNEquals(a, b, n) { | |
assert.ok(a.length >= n, format( | |
'buffer "a" length (%d) is shorter than "n" (%d)', a.length, n)); | |
assert.ok(b.length >= n, format( | |
'buffer "b" length (%d) is shorter than "n" (%d)', b.length, n)); | |
for (var i = 0; i < n; i++) { | |
if (a[i] !== b[i]) { | |
return false; | |
} | |
} | |
return true; | |
} | |
function compressionTypeFromBufSync(buf) { | |
var types = Object.keys(magicNumbers); | |
for (var i = 0; i < types.length; i++) { | |
var type = types[i]; | |
var magic = magicNumbers[type]; | |
if (bufNEquals(buf, magic, magic.length)) { | |
return type; | |
} | |
} | |
return null; | |
} | |
function compressionTypeFromPath(path, cb) { | |
fs.open(path, 'r', function (oErr, fd) { | |
if (oErr) { | |
cb(oErr); | |
return; | |
} | |
var buf = new Buffer(maxMagicLen); | |
fs.read(fd, buf, 0, buf.length, 0, function (rErr, bytesRead, buffer) { | |
if (rErr) { | |
cb(rErr); | |
return; | |
} | |
fs.close(fd, function (cErr) { | |
if (cErr) { | |
cb(cErr); | |
return; | |
} | |
cb(null, compressionTypeFromBufSync(buf)); | |
}); | |
}); | |
}); | |
} | |
// ---- exports | |
module.exports = { | |
compressionTypeFromPath: compressionTypeFromPath | |
}; |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment