コード例 #1
0
ファイル: utils.py プロジェクト: HerbDavisY2K/vxcage
def get_sample_path(sha256):
    path = os.path.join(Config().api.repository, sha256[0], sha256[1], sha256[2], sha256[3], sha256)
    if Config().api.use_aws:
        s3key = AWSStorage.get_key()
        s3key.name = path
        if not s3key.exists():
            return None
    else:
        if not os.path.exists(path):
            return None

    return path
コード例 #2
0
ファイル: utils.py プロジェクト: HerbDavisY2K/vxcage
def get_sample_content(sha256):
    path = get_sample_path(sha256)
    if Config().api.use_aws:
        s3key = AWSStorage.get_key()
        s3key.name = path
        if s3key.storage_class == 'GLACIER':
            s3key.restore(15)
            return (503, jsonize({'error': 'sample_not_online'}))
        else:
            return (200, s3key.get_contents_as_string())
    else:
        return open(path, "rb").read()
コード例 #3
0
ファイル: utils.py プロジェクト: HerbDavisY2K/vxcage
def store_sample(data):
    sha256 = File(data=data).get_sha256()
    
    folder = os.path.join(Config().api.repository, sha256[0], sha256[1], sha256[2], sha256[3])
    path = os.path.join(folder, sha256)

    if Config().api.use_aws:
        s3key = AWSStorage().get_key()
        s3key.name = path
        if not s3key.exists():
            s3key.set_contents_from_string(data)
    else:
        if not os.path.exists(folder):
            os.makedirs(folder, 0750)

        if not os.path.exists(path):
            sample = open(path, "wb")
            sample.write(data)
            sample.close()
    
    return path
コード例 #4
0
ファイル: objects.py プロジェクト: HerbDavisY2K/vxcage
    def __init__(self, path=None, data=None):
        self.path = path
        self.s3key = None

        if path:
            if Config().api.use_aws:
                self.s3key = AWSStorage.get_key()
                self.s3key.name = self.path
                self.data = self.s3key.get_contents_as_string()
            else:
                self.data = open(self.path, "rb").read()
        else:
            self.data = data