1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
const fs = require('fs');
const crypto = require('crypto');
function jwt_secret() {
let file = fs.readFileSync( '../data/jwt_secret.txt', 'utf8' );
if( file.endsWith( '\n' ) ) {
file = file.slice( 0, -1 );
}
return file;
}
function write( content, filename ) {
const chatJson = content;
const salt = crypto.randomBytes( 16 );
const key = crypto.pbkdf2Sync( jwt_secret(), salt, 100000, 32, 'sha512' );
const iv = crypto.randomBytes( 12 );
const cipher = crypto.createCipheriv( 'aes-256-gcm', key, iv );
let encrypted = cipher.update( chatJson, 'utf8', 'hex' );
encrypted += cipher.final( 'hex' );
const authTag = cipher.getAuthTag().toString( 'hex' );
let fullStr = salt.toString( 'hex' ) + iv.toString( 'hex' ) + authTag + encrypted;
fs.writeFileSync( filename, fullStr );
}
function read( filename ) {
let contents = '';
try {
contents = fs.readFileSync( filename, 'utf8' );
} catch( e ) {
console.log( '0' );
return;
}
const salt = Buffer.from( contents.slice( 0, 32 ), 'hex' );
const iv = Buffer.from( contents.slice( 32, 56 ), 'hex' );
const authTag = Buffer.from( contents.slice( 56, 88 ), 'hex' );
const data = contents.slice( 88 );
const key = crypto.pbkdf2Sync( jwt_secret(), salt, 100000, 32, 'sha512' );
const decipher = crypto.createDecipheriv( 'aes-256-gcm', key, iv );
decipher.setAuthTag( authTag );
let decrypted = decipher.update( data, 'hex', 'utf8' );
decrypted += decipher.final( 'utf8' );
return decrypted;
}
function main() {
const args = process.argv.slice(2);
const operation = args[0];
switch( operation ) {
case 'read': {
try {
const data = read( args[1] );
console.log( JSON.parse( JSON.stringify( data ) ) );
} catch( e ) {
console.log( "0" );
console.log( e.message );
}
break;
}
case 'write': {
try {
const data = args[2];
write( data, args[1] );
console.log( "1" );
} catch( e ) {
console.log( "0" );
console.log( e.message );
}
break;
}
}
}
main();
|
