56 lines
1.5 KiB
C++
56 lines
1.5 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
#include <endian.h>
|
|
|
|
#include <simdjson.h>
|
|
#include <lz4.h>
|
|
using namespace simdjson;
|
|
// /home/yorick/.mozilla/firefox/sdgle03g.default/sessionstore-backups/recovery.jsonlz4
|
|
// jq '.windows | map(.tabs | length) | add'
|
|
|
|
padded_string read_mozlz4a_file(const std::string& file) {
|
|
std::ifstream is(file, std::ifstream::binary);
|
|
if (!is) {
|
|
std::cerr << "error opening file" << std::endl;
|
|
exit(1);
|
|
}
|
|
is.seekg(0, is.end);
|
|
size_t insz = is.tellg();
|
|
is.seekg(0, is.beg);
|
|
char * buffer = new char[insz];
|
|
is.read(buffer, insz);
|
|
if (!is) {
|
|
std::cerr << "error reading file" << std::endl;
|
|
exit(1);
|
|
}
|
|
is.close();
|
|
if (memcmp(buffer, "mozLz40", 8)) {
|
|
std::cerr << "not a mozLZa file" << std::endl;
|
|
exit(1);
|
|
}
|
|
size_t outsz = le32toh(*(uint32_t *) (buffer + 8));
|
|
padded_string out(outsz);
|
|
if (LZ4_decompress_safe_partial(buffer + 12, out.data(), insz - 12, outsz, outsz) < 0) {
|
|
std::cerr << "decompression error" << std::endl;
|
|
exit(1);
|
|
}
|
|
return out;
|
|
}
|
|
|
|
int main(int ac, char **av) {
|
|
ondemand::parser parser;
|
|
if (ac < 2) {
|
|
std::cerr << "usage: " << av[0] << " .mozilla/firefox/*.default/sessionstore-backups/recovery.jsonlz4" << std::endl;
|
|
exit(1);
|
|
}
|
|
padded_string json = read_mozlz4a_file(av[1]);
|
|
ondemand::document session = parser.iterate(json);
|
|
ondemand::array windows = session["windows"];
|
|
size_t n = 0;
|
|
for (auto i : windows) {
|
|
ondemand::array tabs = i["tabs"];
|
|
n += tabs.count_elements();
|
|
}
|
|
std::cout << n << std::endl;
|
|
}
|