webscanner/src/monitor/storage.js

Automation scanner for web resources

let started = false;

async function start(context) {
  started = true;
  await context.client.DOMStorage.enable();
  await registerStorage(context.client, context.data.storage);
  //TODO:add indexedDB
}

function stop(context) {
  if (!started) {
    return;
  }
  return context.data.storage;
}

async function registerStorage(client, storage) {
  client.DOMStorage.domStorageItemAdded(
    ({
      key,
      newValue: value,
      storageId: { securityOrigin, isLocalStorage },
    }) => {
      const storageType = isLocalStorage ? 'localStorage' : 'sessionStorage';
      storage[securityOrigin] = storage[securityOrigin] || {};
      storage[securityOrigin][storageType] =
        storage[securityOrigin][storageType] || [];
      storage[securityOrigin][storageType].push({ key, value });
    }
  );
}

module.exports = {
  start,
  stop,
};