From 6af0a39e6171735fac9cd3d3f78e46ac95d7bf0e Mon Sep 17 00:00:00 2001
From: Martin Boehm <martin.boehm@1mbsoftware.net>
Date: Wed, 24 Jan 2018 15:10:07 +0100
Subject: [PATCH] accept base58 and hex addresses

---
 db/rocksdb.go | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/db/rocksdb.go b/db/rocksdb.go
index 017d8aca..6de7a713 100644
--- a/db/rocksdb.go
+++ b/db/rocksdb.go
@@ -335,11 +335,24 @@ func unpackVaruint(buf []byte) (uint32, int) {
 }
 
 func packAddress(address string) ([]byte, error) {
-	buf := base58.Decode(address)
-	if len(buf) <= 4 {
-		return nil, ErrInvalidAddress
+	var buf []byte
+	if len(address) == 42 || len(address) == 50 {
+		var err error
+		buf, err = hex.DecodeString(address)
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		buf = base58.Decode(address)
 	}
-	return buf[:len(buf)-4], nil // Slice off the checksum
+	if len(buf) == 25 {
+		return buf[:len(buf)-4], nil // Slice off the checksum
+	}
+	if len(buf) == 21 { // address without checksum
+		return buf, nil
+	}
+	return nil, ErrInvalidAddress
+
 }
 
 func unpackAddress(buf []byte) (string, error) {