mirror of
https://github.com/gorhill/uBlock.git
synced 2024-08-31 08:09:38 +02:00
225 lines
6.9 KiB
JavaScript
225 lines
6.9 KiB
JavaScript
/*******************************************************************************
|
|
|
|
µBlock - a Chromium browser extension to block requests.
|
|
Copyright (C) 2014 Raymond Hill
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see {http://www.gnu.org/licenses/}.
|
|
|
|
Home: https://github.com/gorhill/uBlock
|
|
*/
|
|
|
|
/* global chrome, µBlock */
|
|
|
|
/******************************************************************************/
|
|
|
|
// Asset update manager
|
|
|
|
µBlock.assetUpdater = (function() {
|
|
|
|
/******************************************************************************/
|
|
|
|
var getUpdateList = function(callback) {
|
|
var localChecksumsText = '';
|
|
var remoteChecksumsText = '';
|
|
|
|
var compareChecksums = function() {
|
|
var parseChecksumsText = function(text) {
|
|
var result = {};
|
|
var lines = text.split(/\n+/);
|
|
var i = lines.length;
|
|
var fields;
|
|
while ( i-- ) {
|
|
fields = lines[i].trim().split(/\s+/);
|
|
if ( fields.length !== 2 ) {
|
|
continue;
|
|
}
|
|
result[fields[1]] = fields[0];
|
|
}
|
|
return result;
|
|
};
|
|
if ( remoteChecksumsText === 'Error' || localChecksumsText === 'Error' ) {
|
|
remoteChecksumsText = localChecksumsText = '';
|
|
}
|
|
var localAssetChecksums = parseChecksumsText(localChecksumsText);
|
|
var remoteAssetChecksums = parseChecksumsText(remoteChecksumsText);
|
|
|
|
var toUpdate = {};
|
|
var path;
|
|
for ( path in remoteAssetChecksums ) {
|
|
if ( !remoteAssetChecksums.hasOwnProperty(path) ) {
|
|
continue;
|
|
}
|
|
if ( localAssetChecksums[path] === undefined ) {
|
|
toUpdate[path] = {
|
|
status: 'Added',
|
|
remoteChecksum: remoteAssetChecksums[path],
|
|
localChecksum: ''
|
|
};
|
|
continue;
|
|
}
|
|
if ( localAssetChecksums[path] === remoteAssetChecksums[path] ) {
|
|
toUpdate[path] = {
|
|
status: 'Unchanged',
|
|
remoteChecksum: remoteAssetChecksums[path],
|
|
localChecksum: localAssetChecksums[path]
|
|
};
|
|
continue;
|
|
}
|
|
toUpdate[path] = {
|
|
status: 'Changed',
|
|
remoteChecksum: remoteAssetChecksums[path],
|
|
localChecksum: localAssetChecksums[path]
|
|
};
|
|
}
|
|
for ( path in localAssetChecksums ) {
|
|
if ( !localAssetChecksums.hasOwnProperty(path) ) {
|
|
continue;
|
|
}
|
|
if ( remoteAssetChecksums[path] === undefined ) {
|
|
toUpdate[path] = {
|
|
status: 'Removed',
|
|
remoteChecksum: '',
|
|
localChecksum: localAssetChecksums[path]
|
|
};
|
|
}
|
|
}
|
|
|
|
callback({ 'list': toUpdate });
|
|
};
|
|
|
|
var validateChecksums = function(details) {
|
|
if ( details.error || details.content === '' ) {
|
|
return 'Error';
|
|
}
|
|
if ( /^(?:[0-9a-f]{32}\s+\S+(\s+|$))+/.test(details.content) ) {
|
|
return details.content;
|
|
}
|
|
return 'Error';
|
|
};
|
|
|
|
var onLocalChecksumsLoaded = function(details) {
|
|
localChecksumsText = validateChecksums(details);
|
|
if ( remoteChecksumsText !== '' ) {
|
|
compareChecksums();
|
|
}
|
|
};
|
|
|
|
var onRemoteChecksumsLoaded = function(details) {
|
|
remoteChecksumsText = validateChecksums(details);
|
|
if ( localChecksumsText !== '' ) {
|
|
compareChecksums();
|
|
}
|
|
};
|
|
|
|
µBlock.assets.getRemote('assets/checksums.txt', onRemoteChecksumsLoaded);
|
|
µBlock.assets.get('assets/checksums.txt', onLocalChecksumsLoaded);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
// If `list` is null, it will be fetched internally.
|
|
|
|
var update = function(list, callback) {
|
|
var assetChangedCount = 0;
|
|
var assetProcessedCount;
|
|
var updatedAssetChecksums = [];
|
|
|
|
var onCompleted = function() {
|
|
var details = {
|
|
what: 'allLocalAssetsUpdated',
|
|
changedCount: assetChangedCount
|
|
};
|
|
callback(details);
|
|
µBlock.messaging.announce(details);
|
|
};
|
|
|
|
var doCountdown = function() {
|
|
assetProcessedCount -= 1;
|
|
if ( assetProcessedCount > 0 ) {
|
|
return;
|
|
}
|
|
µBlock.assets.put(
|
|
'assets/checksums.txt',
|
|
updatedAssetChecksums.join('\n'),
|
|
onCompleted
|
|
);
|
|
chrome.storage.local.set({ 'assetsUpdateTimestamp': Date.now() });
|
|
};
|
|
|
|
var assetUpdated = function(details) {
|
|
var path = details.path;
|
|
var entry = list[path];
|
|
if ( details.error ) {
|
|
updatedAssetChecksums.push(entry.localChecksum + ' ' + path);
|
|
} else {
|
|
updatedAssetChecksums.push(entry.remoteChecksum + ' ' + path);
|
|
assetChangedCount += 1;
|
|
}
|
|
doCountdown();
|
|
};
|
|
|
|
var processList = function() {
|
|
assetProcessedCount = Object.keys(list).length;
|
|
if ( assetProcessedCount === 0 ) {
|
|
onCompleted();
|
|
return;
|
|
}
|
|
var entry;
|
|
var details = { path: '', md5: '' };
|
|
for ( var path in list ) {
|
|
if ( list.hasOwnProperty(path) === false ) {
|
|
continue;
|
|
}
|
|
entry = list[path];
|
|
if ( entry.status === 'Added' || entry.status === 'Changed' ) {
|
|
details.path = path;
|
|
details.md5 = entry.remoteChecksum;
|
|
µBlock.assets.update(details, assetUpdated);
|
|
continue;
|
|
}
|
|
if ( entry.status === 'Unchanged' ) {
|
|
updatedAssetChecksums.push(entry.localChecksum + ' ' + path);
|
|
}
|
|
doCountdown();
|
|
}
|
|
};
|
|
|
|
var listLoaded = function(details) {
|
|
list = details.list;
|
|
processList();
|
|
};
|
|
|
|
if ( list ) {
|
|
processList();
|
|
} else {
|
|
getUpdateList(listLoaded);
|
|
}
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
// Export API
|
|
|
|
return {
|
|
'getList': getUpdateList,
|
|
'update': update
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
})();
|
|
|
|
/******************************************************************************/
|
|
|