gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2025-12-12 15:49:46 -08:00
Code Issues Packages Projects Releases Wiki Activity

cache: use zlib to reduce cache size

Browse Source
This commit is contained in:
Willi Ballenthin
2023-01-20 15:20:10 +01:00
parent fbd7c566f4
commit 03f72f498e
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
capa/rules/cache.py
View File

@@ -1,4 +1,5 @@
import sys
import zlib
import pickle
import hashlib
import logging
@@ -7,6 +8,7 @@ from typing import List, Optional
from dataclasses import dataclass
import capa.rules
import capa.version
logger = logging.getLogger(__name__)
@@ -70,14 +72,14 @@ class RuleCache:
ruleset: capa.rules.RuleSet
def dump(self):
return MAGIC + VERSION + self.id.encode("ascii") + pickle.dumps(self)
return MAGIC + VERSION + self.id.encode("ascii") + zlib.compress(pickle.dumps(self))
@staticmethod
def load(data):
assert data.startswith(MAGIC + VERSION)
id = data[0x8:0x48].decode("ascii")
cache = pickle.loads(data[0x48:])
cache = pickle.loads(zlib.decompress(data[0x48:]))
assert isinstance(cache, RuleCache)
assert cache.id == id