p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Implemented export byte saving to file

This commit is contained in:
VakarisZ 2021-05-31 16:09:10 +03:00
parent 308ae3e169
commit 46408e6d32
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/monkey_island/cc/services/utils/config_encryption.py
View File

@ -4,18 +4,18 @@ from typing import Dict
import pyAesCrypt
# TODO use from pyAesCrypt
BUFFER_SIZE = 64 * 1024
def encrypt_config(config: Dict, password: str) -> bytes:
def encrypt_config(config: Dict, password: str) -> str:
plaintext_config_stream = io.BytesIO(json.dumps(config).encode())
ciphertext_config_stream = io.BytesIO()
pyAesCrypt.encryptStream(
plaintext_config_stream, ciphertext_config_stream, password, BUFFER_SIZE
)
ciphertext_config_bytes = ciphertext_config_stream.getvalue()
ciphertext_config_bytes = str(ciphertext_config_stream.getvalue())
return ciphertext_config_bytes

9
monkey/monkey_island/cc/ui/src/components/configuration-components/ExportConfigModal.tsx
View File

@ -1,6 +1,7 @@
import {Button, Modal, Form} from 'react-bootstrap';
import React, {useState} from 'react';
import FileSaver from 'file-saver';
import AuthComponent from '../AuthComponent';
import '../../styles/components/configuration-components/ExportConfigModal.scss';
@ -33,8 +34,12 @@ const ConfigExportModal = (props: Props) => {
})
}
)
.then(res => {
console.log(res.json());
.then(res => res.json())
.then(res => {
console.log(res);
const configAsBinary = new Blob([res['encrypted_config']]);
FileSaver.saveAs(configAsBinary, 'monkey.conf');
})
}