From 76c6379a54270d36c8ac0ce19d922fbc1687416e Mon Sep 17 00:00:00 2001
From: Josh Rickmar <jrick@conformal.com>
Date: Wed, 26 Feb 2014 15:22:48 -0500
Subject: [PATCH] Update for btcutil.Address API change.

---
 rpcserver.go | 17 +++--------------
 1 file changed, 3 insertions(+), 14 deletions(-)

diff --git a/rpcserver.go b/rpcserver.go
index a959936..abe7570 100644
--- a/rpcserver.go
+++ b/rpcserver.go
@@ -450,18 +450,7 @@ func GetAccount(icmd btcjson.Cmd) (interface{}, *btcjson.Error) {
 	if err != nil {
 		return nil, &btcjson.ErrInvalidAddressOrKey
 	}
-	var net btcwire.BitcoinNet
-	switch a := addr.(type) {
-	case *btcutil.AddressPubKeyHash:
-		net = a.Net()
-
-	case *btcutil.AddressScriptHash:
-		net = a.Net()
-
-	default:
-		return nil, &btcjson.ErrInvalidAddressOrKey
-	}
-	if net != cfg.Net() {
+	if !addr.IsForNet(cfg.Net()) {
 		return nil, &btcjson.ErrInvalidAddressOrKey
 	}
 
@@ -540,7 +529,7 @@ func GetAddressBalance(icmd btcjson.Cmd) (interface{}, *btcjson.Error) {
 		return nil, &btcjson.ErrInvalidAddressOrKey
 	}
 	apkh, ok := addr.(*btcutil.AddressPubKeyHash)
-	if !ok || apkh.Net() != cfg.Net() {
+	if !ok || !apkh.IsForNet(cfg.Net()) {
 		return nil, &btcjson.ErrInvalidAddressOrKey
 	}
 
@@ -1043,7 +1032,7 @@ func ListAddressTransactions(icmd btcjson.Cmd) (interface{}, *btcjson.Error) {
 			return nil, &btcjson.ErrInvalidAddressOrKey
 		}
 		apkh, ok := addr.(*btcutil.AddressPubKeyHash)
-		if !ok || apkh.Net() != cfg.Net() {
+		if !ok || !apkh.IsForNet(cfg.Net()) {
 			return nil, &btcjson.ErrInvalidAddressOrKey
 		}
 		pkHashMap[string(addr.ScriptAddress())] = struct{}{}