Herbert Poul
5 years ago
15 changed files with 344 additions and 96 deletions
@ -0,0 +1,42 @@ |
|||||||
|
import 'dart:convert'; |
||||||
|
import 'dart:io'; |
||||||
|
import 'dart:typed_data'; |
||||||
|
|
||||||
|
import 'package:kdbx/src/kdbx_header.dart'; |
||||||
|
import 'package:kdbx/src/kdbx_xml.dart'; |
||||||
|
import 'package:meta/meta.dart'; |
||||||
|
import 'package:xml/xml.dart'; |
||||||
|
|
||||||
|
class KdbxBinary { |
||||||
|
KdbxBinary({this.isInline, this.isProtected, this.value}); |
||||||
|
final bool isInline; |
||||||
|
final bool isProtected; |
||||||
|
final Uint8List value; |
||||||
|
|
||||||
|
static KdbxBinary readBinaryInnerHeader(InnerHeaderField field) { |
||||||
|
final flags = field.bytes[0]; |
||||||
|
final isProtected = flags & 0x01 == 0x01; |
||||||
|
final value = Uint8List.sublistView(field.bytes, 1); |
||||||
|
return KdbxBinary( |
||||||
|
isInline: false, |
||||||
|
isProtected: isProtected, |
||||||
|
value: value, |
||||||
|
); |
||||||
|
} |
||||||
|
|
||||||
|
static KdbxBinary readBinaryXml(XmlElement valueNode, |
||||||
|
{@required bool isInline}) { |
||||||
|
assert(isInline != null); |
||||||
|
final isProtected = valueNode.getAttributeBool(KdbxXml.ATTR_PROTECTED); |
||||||
|
final isCompressed = valueNode.getAttributeBool(KdbxXml.ATTR_COMPRESSED); |
||||||
|
var value = base64.decode(valueNode.text.trim()); |
||||||
|
if (isCompressed) { |
||||||
|
value = gzip.decode(value) as Uint8List; |
||||||
|
} |
||||||
|
return KdbxBinary( |
||||||
|
isInline: isInline, |
||||||
|
isProtected: isProtected, |
||||||
|
value: value, |
||||||
|
); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,58 @@ |
|||||||
|
//typedef HashStuff = Pointer<Utf8> Function(Pointer<Utf8> str); |
||||||
|
import 'dart:ffi'; |
||||||
|
import 'dart:io'; |
||||||
|
|
||||||
|
import 'package:ffi/ffi.dart'; |
||||||
|
import 'package:kdbx/kdbx.dart'; |
||||||
|
|
||||||
|
typedef Argon2HashNative = Pointer<Utf8> Function( |
||||||
|
Pointer<Uint8> key, |
||||||
|
IntPtr keyLen, |
||||||
|
Pointer<Uint8> salt, |
||||||
|
Uint64 saltlen, |
||||||
|
Uint32 m_cost, // memory cost |
||||||
|
Uint32 t_cost, // time cost (number iterations) |
||||||
|
Uint32 parallelism, |
||||||
|
IntPtr hashlen, |
||||||
|
Uint8 type, |
||||||
|
Uint32 version, |
||||||
|
); |
||||||
|
typedef Argon2Hash = Pointer<Utf8> Function( |
||||||
|
Pointer<Uint8> key, |
||||||
|
int keyLen, |
||||||
|
Pointer<Uint8> salt, |
||||||
|
int saltlen, |
||||||
|
int m_cost, // memory cost |
||||||
|
int t_cost, // time cost (number iterations) |
||||||
|
int parallelism, |
||||||
|
int hashlen, |
||||||
|
int type, |
||||||
|
int version, |
||||||
|
); |
||||||
|
|
||||||
|
class Argon2Test extends Argon2Base { |
||||||
|
Argon2Test() { |
||||||
|
final argon2lib = Platform.isMacOS |
||||||
|
? DynamicLibrary.open('libargon2_ffi.dylib') |
||||||
|
: DynamicLibrary.open('./libargon2_ffi.so'); |
||||||
|
argon2hash = argon2lib |
||||||
|
.lookup<NativeFunction<Argon2HashNative>>('hp_argon2_hash') |
||||||
|
.asFunction(); |
||||||
|
} |
||||||
|
|
||||||
|
@override |
||||||
|
Argon2Hash argon2hash; |
||||||
|
} |
||||||
|
|
||||||
|
class TestUtil { |
||||||
|
static Future<KdbxFile> readKdbxFile( |
||||||
|
String filePath, { |
||||||
|
String password = 'asdf', |
||||||
|
}) async { |
||||||
|
final kdbxFormat = KdbxFormat(Argon2Test()); |
||||||
|
final data = await File(filePath).readAsBytes(); |
||||||
|
final file = await kdbxFormat.read( |
||||||
|
data, Credentials(ProtectedValue.fromString(password))); |
||||||
|
return file; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,69 @@ |
|||||||
|
import 'dart:convert'; |
||||||
|
import 'dart:typed_data'; |
||||||
|
|
||||||
|
import 'package:kdbx/kdbx.dart'; |
||||||
|
import 'package:logging/logging.dart'; |
||||||
|
import 'package:logging_appenders/logging_appenders.dart'; |
||||||
|
import 'package:test/test.dart'; |
||||||
|
|
||||||
|
import 'internal/test_utils.dart'; |
||||||
|
|
||||||
|
void main() { |
||||||
|
Logger.root.level = Level.ALL; |
||||||
|
PrintAppender().attachToLogger(Logger.root); |
||||||
|
|
||||||
|
group('kdbx3 attachment', () { |
||||||
|
test('read binary', () async { |
||||||
|
final file = await TestUtil.readKdbxFile('test/keepass2binaries.kdbx'); |
||||||
|
final entry = file.body.rootGroup.entries.first; |
||||||
|
final binaries = entry.binaryEntries; |
||||||
|
expect(binaries, hasLength(3)); |
||||||
|
for (final binary in binaries) { |
||||||
|
switch (binary.key.key) { |
||||||
|
case 'example1.txt': |
||||||
|
expect(utf8.decode(binary.value.value), 'content1 example\n\n'); |
||||||
|
break; |
||||||
|
case 'example2.txt': |
||||||
|
expect(utf8.decode(binary.value.value), 'content2 example\n\n'); |
||||||
|
break; |
||||||
|
case 'keepasslogo.jpeg': |
||||||
|
expect(binary.value.value, hasLength(7092)); |
||||||
|
break; |
||||||
|
default: |
||||||
|
fail('invalid key. ${binary.key}'); |
||||||
|
} |
||||||
|
} |
||||||
|
}); |
||||||
|
}); |
||||||
|
group('kdbx4 attachment', () { |
||||||
|
test('read binary', () async { |
||||||
|
final file = |
||||||
|
await TestUtil.readKdbxFile('test/keepass2kdbx4binaries.kdbx'); |
||||||
|
void expectBinary(KdbxEntry entry, String key, dynamic matcher) { |
||||||
|
final binaries = entry.binaryEntries; |
||||||
|
expect(binaries, hasLength(1)); |
||||||
|
final binary = binaries.first; |
||||||
|
expect(binary.key.key, key); |
||||||
|
expect(binary.value.value, matcher); |
||||||
|
} |
||||||
|
|
||||||
|
expect(file.body.rootGroup.entries, hasLength(2)); |
||||||
|
expectBinary(file.body.rootGroup.entries.first, 'example2.txt', |
||||||
|
IsUtf8String('content2 example\n\n')); |
||||||
|
expectBinary(file.body.rootGroup.entries.last, 'keepasslogo.jpeg', |
||||||
|
hasLength(7092)); |
||||||
|
}); |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
class IsUtf8String extends CustomMatcher { |
||||||
|
IsUtf8String(dynamic matcher) : super('is utf8 string', 'utf8', matcher); |
||||||
|
|
||||||
|
@override |
||||||
|
Object featureValueOf(dynamic actual) { |
||||||
|
if (actual is Uint8List) { |
||||||
|
return utf8.decode(actual); |
||||||
|
} |
||||||
|
return super.featureValueOf(actual); |
||||||
|
} |
||||||
|
} |
||||||
Binary file not shown.
Binary file not shown.
Loading…
Reference in new issue