1
0
Fork 0
mirror of https://github.com/codedread/bitjs synced 2025-10-03 09:39:16 +02:00

Implement crc32 calculation for issue #29

This commit is contained in:
Jeff Schiller 2021-12-16 15:09:20 -05:00
parent c0afe7637e
commit 596a859bdc
2 changed files with 46 additions and 9 deletions

View file

@ -220,7 +220,8 @@ class ZipLocalFile {
/** /**
* Returns a table of Huffman codes. Each entry's key is its code and its value is a JavaScript * Returns a table of Huffman codes. Each entry's key is its code and its value is a JavaScript
* object containing {length: 6, symbol: X}. * object containing {length: 6, symbol: X}.
* @param {number[]} bitLengths * @param {number[]} bitLengths An array representing the bit lengths of the codes, in order.
* See section 3.2.2 of https://datatracker.ietf.org/doc/html/rfc1951.
* @returns {Map<number, SymbolLengthPair>} * @returns {Map<number, SymbolLengthPair>}
*/ */
function getHuffmanCodes(bitLengths) { function getHuffmanCodes(bitLengths) {
@ -267,7 +268,7 @@ function getHuffmanCodes(bitLengths) {
for (let n = 0; n < numLengths; ++n) { for (let n = 0; n < numLengths; ++n) {
const len = bitLengths[n]; const len = bitLengths[n];
if (len != 0) { if (len != 0) {
table.set(next_code[len], { length: len, symbol: n }); //, bitstring: binaryValueToString(next_code[len],len) }; table.set(next_code[len], { length: len, symbol: n });
next_code[len]++; next_code[len]++;
} }
} }

View file

@ -47,6 +47,7 @@ importScripts('../io/bytestream-worker.js');
const zLocalFileHeaderSignature = 0x04034b50; const zLocalFileHeaderSignature = 0x04034b50;
const zCentralFileHeaderSignature = 0x02014b50; const zCentralFileHeaderSignature = 0x02014b50;
const zEndOfCentralDirSignature = 0x06054b50; const zEndOfCentralDirSignature = 0x06054b50;
const zCRC32MagicNumber = 0xedb88320; // 0xdebb20e3;
/** /**
* @typedef CentralDirectoryFileHeaderInfo An object to be used to construct the central directory. * @typedef CentralDirectoryFileHeaderInfo An object to be used to construct the central directory.
@ -77,9 +78,46 @@ const CompressorState = {
}; };
let state = CompressorState.NOT_STARTED; let state = CompressorState.NOT_STARTED;
let lastFileReceived = false; let lastFileReceived = false;
const crc32Table = createCRC32Table();
/** Helper functions. */ /** Helper functions. */
/**
* Logic taken from https://github.com/nodeca/pako/blob/master/lib/zlib/crc32.js
* @returns {Uint8Array}
*/
function createCRC32Table() {
const table = new Uint32Array(256);
for (let n = 0; n < 256; n++) {
let c = n;
for (let k = 0; k < 8; k++) {
c = ((c & 1) ? (zCRC32MagicNumber ^ (c >>> 1)) : (c >>> 1));
}
table[n] = c;
}
return table;
};
/**
* Logic taken from https://github.com/nodeca/pako/blob/master/lib/zlib/crc32.js
* @param {number} crc
* @param {Uint8Array} bytes
* @returns {number}
*/
function calculateCRC32(crc, bytes) {
const len = bytes.byteLength;
crc ^= -1;
for (const byte of bytes) {
crc = (crc >>> 8) ^ crc32Table[(crc ^ byte) & 0xFF];
}
crc ^= -1;
if (crc < 0) {
crc += 0x100000000;
}
return crc;
}
/** /**
* Logic taken from https://github.com/thejoshwolfe/yazl. * Logic taken from https://github.com/thejoshwolfe/yazl.
* @param {number} lastModTime The number of ms since the Unix epoch (1970-01-01 at midnight). * @param {number} lastModTime The number of ms since the Unix epoch (1970-01-01 at midnight).
@ -129,8 +167,7 @@ function zipOneFile(file) {
compressionMethod: 0, compressionMethod: 0,
lastModFileTime: dateToDosTime(jsDate), lastModFileTime: dateToDosTime(jsDate),
lastModFileDate: dateToDosDate(jsDate), lastModFileDate: dateToDosDate(jsDate),
// TODO: Calculate. See section 4.4.7 of https://pkware.cachefly.net/webdocs/casestudies/APPNOTE.TXT. crc32: calculateCRC32(0, file.fileData),
crc32: 0,
// TODO: For now, this is easy. Later when we do DEFLATE, we will have to calculate. // TODO: For now, this is easy. Later when we do DEFLATE, we will have to calculate.
compressedSize: file.fileData.byteLength, compressedSize: file.fileData.byteLength,
uncompressedSize: file.fileData.byteLength, uncompressedSize: file.fileData.byteLength,
@ -157,10 +194,9 @@ function zipOneFile(file) {
*/ */
function writeCentralFileDirectory() { function writeCentralFileDirectory() {
// Each central directory file header is 46 bytes + the filename. // Each central directory file header is 46 bytes + the filename.
let filenameLengths = filesCompressed.map(f => f.fileName.length).reduce((acc, cur) => acc + cur); let cdsLength = filesCompressed.map(f => f.fileName.length + 46).reduce((a, c) => a + c);
const sizeOfCentralDirectoryEntries = (filesCompressed.length * 46) + filenameLengths; // 22 extra bytes for the end-of-central-dir header.
const buffer = new bitjs.io.ByteBuffer(cdsLength + 22);
const buffer = new bitjs.io.ByteBuffer(sizeOfCentralDirectoryEntries + 22);
for (const cdInfo of centralDirectoryInfos) { for (const cdInfo of centralDirectoryInfos) {
buffer.writeNumber(zCentralFileHeaderSignature, 4); // Magic number. buffer.writeNumber(zCentralFileHeaderSignature, 4); // Magic number.
@ -189,7 +225,7 @@ function writeCentralFileDirectory() {
buffer.writeNumber(0, 2); // Disk where central directory starts. buffer.writeNumber(0, 2); // Disk where central directory starts.
buffer.writeNumber(filesCompressed.length, 2); // Number of central directory records on this disk. buffer.writeNumber(filesCompressed.length, 2); // Number of central directory records on this disk.
buffer.writeNumber(filesCompressed.length, 2); // Total number of central directory records. buffer.writeNumber(filesCompressed.length, 2); // Total number of central directory records.
buffer.writeNumber(sizeOfCentralDirectoryEntries, 4); // Size of central directory. buffer.writeNumber(cdsLength, 4); // Size of central directory.
buffer.writeNumber(numBytesWritten, 4); // Offset of start of central directory. buffer.writeNumber(numBytesWritten, 4); // Offset of start of central directory.
buffer.writeNumber(0, 2); // Comment length. buffer.writeNumber(0, 2); // Comment length.